1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
/*
* 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;
};
}
}
|
