mirror of
https://github.com/vcmi/vcmi.git
synced 2024-11-28 08:48:48 +02:00
79 lines
1.8 KiB
C++
79 lines
1.8 KiB
C++
/*
|
|
* 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;
|
|
}
|
|
|
|
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;
|
|
|
|
static Goals::TGoalVec getVisitGoals(const std::vector<AIPath> & paths, const CGObjectInstance * objToVisit = nullptr);
|
|
|
|
private:
|
|
bool objectMatchesFilter(const CGObjectInstance * obj) const;
|
|
};
|
|
}
|
|
|
|
|
|
}
|