1
0
mirror of https://github.com/vcmi/vcmi.git synced 2024-12-20 20:23:03 +02:00
vcmi/AI/Nullkiller/Analyzers/ObjectClusterizer.h
2021-07-26 21:02:50 +03:00

75 lines
1.6 KiB
C++

/*
* DangerHitMapAnalyzer.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 "../Pathfinding/AINodeStorage.h"
struct ClusterObjectInfo
{
float priority;
float movementCost;
uint64_t danger;
uint8_t turn;
};
typedef tbb::concurrent_hash_map<const CGObjectInstance *, ClusterObjectInfo> ClusterObjects;
struct ObjectCluster
{
public:
ClusterObjects objects;
const CGObjectInstance * blocker;
void reset()
{
objects.clear();
}
void addObject(const CGObjectInstance * object, const AIPath & path, float priority);
ObjectCluster(const CGObjectInstance * blocker)
:objects(), blocker(blocker)
{
}
ObjectCluster() : ObjectCluster(nullptr)
{
}
std::vector<const CGObjectInstance *> getObjects() const;
const CGObjectInstance * calculateCenter() const;
};
typedef tbb::concurrent_hash_map<const CGObjectInstance *, std::shared_ptr<ObjectCluster>> ClusterMap;
class ObjectClusterizer
{
private:
ObjectCluster nearObjects;
ObjectCluster farObjects;
ClusterMap blockedObjects;
const Nullkiller * ai;
public:
void clusterize();
std::vector<const CGObjectInstance *> getNearbyObjects() const;
std::vector<const CGObjectInstance *> getFarObjects() const;
std::vector<std::shared_ptr<ObjectCluster>> getLockedClusters() const;
const CGObjectInstance * getBlocker(const AIPath & path) const;
ObjectClusterizer(const Nullkiller * ai)
:nearObjects(), farObjects(), blockedObjects(), ai(ai)
{
}
private:
bool shouldVisitObject(const CGObjectInstance * obj) const;
};