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 "Behavior.h"
#include "../AIUtility.h"

class CaptureObjectsBehavior : public Behavior {
private:
	std::vector<int> objectTypes;
	std::vector<int> objectSubTypes;
	std::vector<const CGObjectInstance *> objectsToCapture;
	bool specificObjects;
public:
	CaptureObjectsBehavior()
	{
		objectTypes = std::vector<int>();
		specificObjects = false;
	}

	CaptureObjectsBehavior(std::vector<const CGObjectInstance *> objectsToCapture)
	{
		this->objectsToCapture = objectsToCapture;
		specificObjects = true;
	}

	CaptureObjectsBehavior(const CGObjectInstance * objectToCapture)
	{
		objectsToCapture = std::vector<const CGObjectInstance *>();
		objectsToCapture.push_back(objectToCapture);
		specificObjects = true;
	}

	virtual Goals::TGoalVec getTasks() 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;
	}

private:
	bool shouldVisitObject(ObjectIdRef obj) const;
};