cloned Nullkiller into Nullkiller2

AI/Nullkiller2/Behaviors/CaptureObjectsBehavior.h

@@ -0,0 +1,87 @@
+/*
+* CaptureObjectsBehavior.h, part of VCMI engine
+*
+* Authors: listed in file AUTHORS in main folder
+*
+* License: GNU General Public License v2.0 or later
+* Full text of license available in license.txt file, in main folder
+*
+*/
+#pragma once
+
+#include "../../../lib/GameLibrary.h"
+#include "../AIUtility.h"
+#include "../Goals/CGoal.h"
+#include "../Pathfinding/AINodeStorage.h"
+
+namespace NKAI
+{
+namespace Goals
+{
+	class CaptureObjectsBehavior : public CGoal<CaptureObjectsBehavior>
+	{
+	private:
+		std::vector<int> objectTypes;
+		std::vector<int> objectSubTypes;
+		std::vector<const CGObjectInstance *> objectsToCapture;
+		bool specificObjects;
+
+	public:
+		CaptureObjectsBehavior()
+			:CGoal(CAPTURE_OBJECTS)
+		{
+			objectTypes = std::vector<int>();
+			specificObjects = false;
+		}
+
+		CaptureObjectsBehavior(std::vector<const CGObjectInstance *> objectsToCapture)
+			:CGoal(CAPTURE_OBJECTS)
+		{
+			this->objectsToCapture = objectsToCapture;
+			specificObjects = true;
+		}
+
+		CaptureObjectsBehavior(const CGObjectInstance * objectToCapture)
+			:CGoal(CAPTURE_OBJECTS)
+		{
+			objectsToCapture = std::vector<const CGObjectInstance *>();
+			objectsToCapture.push_back(objectToCapture);
+			specificObjects = true;
+		}
+
+		Goals::TGoalVec decompose(const Nullkiller2 * ai) const override;
+		std::string toString() const override;
+
+		CaptureObjectsBehavior & ofType(int type)
+		{
+			objectTypes.push_back(type);
+
+			return *this;
+		}
+		CaptureObjectsBehavior & ofType(int type, int subType)
+		{
+			objectTypes.push_back(type);
+			objectSubTypes.push_back(subType);
+
+			return *this;
+		}
+
+		bool operator==(const CaptureObjectsBehavior & other) const override;
+
+		static Goals::TGoalVec getVisitGoals(
+			const std::vector<AIPath> & paths,
+			const Nullkiller2 * nullkiller,
+			const CGObjectInstance * objToVisit = nullptr,
+			bool force = false);
+
+	private:
+		bool objectMatchesFilter(const CGObjectInstance * obj) const;
+		void decomposeObjects(
+			Goals::TGoalVec & result,
+			const std::vector<const CGObjectInstance *> & objs,
+			const Nullkiller2 * nullkiller) const;
+	};
+}
+
+
+}