--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/API/Requirements/AbstractUnitRequirement.cs	Sun Apr 03 18:50:32 2011 +0000
@@ -0,0 +1,50 @@
+// This file (AbstractUnitRequirement.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2008, 2009 IBBoard.
+//
+// The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.
+
+using System;
+using IBBoard.WarFoundry.API.Objects;
+
+namespace IBBoard.WarFoundry.API.Requirements
+{
+	/// <summary>
+	/// Base abstract class for all requirements related to adding/removing something from a Unit (e.g. adding equipment or abilities)
+	/// </summary>
+	public abstract class AbstractUnitRequirement : AbstractRequirement
+	{		
+		protected abstract AbstractFailedRequirement CanAddToUnit(Unit unit);
+		protected abstract AbstractFailedRequirement CanRemoveFromUnit(Unit unit);
+		
+		public override AbstractFailedRequirement CanAddToWarFoundryObject (WarFoundryObject obj)
+		{
+			AbstractFailedRequirement fail = null;
+			
+			if (obj is Unit)
+			{
+				fail = CanAddToUnit((Unit)obj);
+			}
+			else
+			{
+				fail = new FailedRequirement(this);
+			}
+			
+			return fail;
+		}
+		
+		public override AbstractFailedRequirement CanRemoveFromWarFoundryObject (WarFoundryObject obj)
+		{
+			AbstractFailedRequirement fail = null;
+			
+			if (obj is Unit)
+			{
+				fail = CanRemoveFromUnit((Unit)obj);
+			}
+			else
+			{
+				fail = new FailedRequirement(this);
+			}
+			
+			return fail;
+		}
+	}
+}