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
|
#ifndef E323_CCATALOGUER_H
#define E323_CCATALOGUER_H
#include <map>
#include "headers/Defines.h"
#include "CUnitTable.h"
class CUnit;
struct CategoryMatcher {
CategoryMatcher(const unitCategory& inc, const unitCategory& exc = 0):include(inc),exclude(exc) {}
bool operator < (const CategoryMatcher& other) const {
UnitCategoryCompare cmp;
if (cmp(include, other.include))
return true;
if (include == other.include)
return cmp(exclude, other.exclude);
return false;
}
bool test(const unitCategory& other) const {
return (include&other).any() && (exclude&other).none();
}
unitCategory include;
unitCategory exclude;
};
class CCataloguer {
public:
void registerMatcher(const CategoryMatcher& m);
void registerMatcher(const unitCategory& inc, const unitCategory& exc = 0) { registerMatcher(CategoryMatcher(inc, exc)); }
void addUnit(UnitType* type, int id);
void addUnit(CUnit* unit);
void removeUnit(int id);
std::map<int, UnitType*>& getUnits(const CategoryMatcher& m);
std::map<int, UnitType*>* getUnits(const unitCategory& inc, const unitCategory& exc = 0);
int size() const { return cache.size(); }
private:
typedef std::map<CategoryMatcher, std::map<int, UnitType*> > CacheMap;
typedef CacheMap::iterator CacheMapIt;
CacheMap cache;
};
#endif
|
