vcmi/AI/Nullkiller/Behaviors/CaptureObjectsBehavior.h

/*
* 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/VCMI_Lib.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 Nullkiller * 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 Nullkiller * 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 Nullkiller * nullkiller) const;
	};
}


}