/* * 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" namespace Goals { class CaptureObjectsBehavior : public CGoal { private: std::vector objectTypes; std::vector objectSubTypes; std::vector objectsToCapture; bool specificObjects; public: CaptureObjectsBehavior() :CGoal(CAPTURE_OBJECTS) { objectTypes = std::vector(); specificObjects = false; } CaptureObjectsBehavior(std::vector objectsToCapture) :CGoal(CAPTURE_OBJECTS) { this->objectsToCapture = objectsToCapture; specificObjects = true; } CaptureObjectsBehavior(const CGObjectInstance * objectToCapture) :CGoal(CAPTURE_OBJECTS) { objectsToCapture = std::vector(); objectsToCapture.push_back(objectToCapture); specificObjects = true; } virtual Goals::TGoalVec decompose() const override; virtual 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; } virtual bool operator==(const CaptureObjectsBehavior & other) const override; private: bool shouldVisitObject(ObjectIdRef obj) const; }; }