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
|
#ifndef WISHLIST_H
#define WISHLIST_H
#include <vector>
#include <queue>
#include <map>
#include <algorithm>
#include "headers/Defines.h"
class AIClasses;
class UnitType;
class CWishList {
public:
CWishList(AIClasses *ai);
~CWishList() {}
/* Insert a unit in the wishlist, sorted by priority p */
void push(unsigned int categories, buildPriority p);
/* Remove the top unit from the wishlist */
void pop(int factory);
/* Is empty ? */
bool empty(int factory);
/* View the top unit from the wishlist */
UnitType* top(int factory);
private:
struct Wish {
buildPriority p;
UnitType *ut;
Wish(UnitType *ut, buildPriority p) {
this->ut = ut;
this->p = p;
}
bool operator< (const Wish &w) const {
return p > w.p;
}
bool operator== (const Wish &w) const {
return p == w.p;
}
};
void unique(std::vector<Wish> &vector);
std::map<int, std::vector<Wish> > wishlist; /* <factory type, wishes> */
AIClasses *ai;
};
#endif
|
