1 | #ifndef LEMON_SIMANN_H |
---|
2 | #define LEMON_SIMANN_H |
---|
3 | |
---|
4 | namespace lemon { |
---|
5 | |
---|
6 | const double INFTY = 1e24; |
---|
7 | |
---|
8 | class SimAnnBase { |
---|
9 | public: |
---|
10 | class Controller; |
---|
11 | private: |
---|
12 | Controller *controller; |
---|
13 | protected: |
---|
14 | double curr_cost; |
---|
15 | double prev_cost; |
---|
16 | double best_cost; |
---|
17 | |
---|
18 | virtual void mutate() = 0; |
---|
19 | virtual void revert() = 0; |
---|
20 | virtual void saveAsBest() = 0; |
---|
21 | public: |
---|
22 | SimAnnBase() { |
---|
23 | curr_cost = prev_cost = best_cost = INFTY; |
---|
24 | } |
---|
25 | void setController(Controller &_controller) { controller = &_controller; } |
---|
26 | double getBestCost() { return best_cost; } |
---|
27 | void run() { |
---|
28 | while (controller->next()) { |
---|
29 | mutate(); |
---|
30 | if (controller->accept(prev_cost - curr_cost)) { |
---|
31 | controller->acceptEvent(); |
---|
32 | if (curr_cost < best_cost) { |
---|
33 | saveAsBest(); |
---|
34 | controller->improveEvent(); |
---|
35 | } |
---|
36 | } |
---|
37 | else { |
---|
38 | revert(); |
---|
39 | controller->rejectEvent(); |
---|
40 | } |
---|
41 | } |
---|
42 | } |
---|
43 | |
---|
44 | class Controller { |
---|
45 | public: |
---|
46 | virtual void acceptEvent() {} |
---|
47 | virtual void improveEvent() {} |
---|
48 | virtual void rejectEvent() {} |
---|
49 | virtual bool next() = 0; |
---|
50 | virtual bool accept(double cost_diff) = 0; |
---|
51 | }; |
---|
52 | }; |
---|
53 | |
---|
54 | template <typename E> |
---|
55 | class SimAnn : public SimAnnBase { |
---|
56 | private: |
---|
57 | E *curr_ent; |
---|
58 | E *prev_ent; |
---|
59 | E *best_ent; |
---|
60 | public: |
---|
61 | SimAnn() : SimAnnBase() {} |
---|
62 | void setEntity(E &Ent) { |
---|
63 | curr_ent = new E(Ent); |
---|
64 | prev_ent = new E(Ent); |
---|
65 | best_ent = new E(Ent); |
---|
66 | } |
---|
67 | E getBestEntity() { return *best_ent; } |
---|
68 | void mutate() { |
---|
69 | *prev_ent = *curr_ent; |
---|
70 | prev_cost = curr_cost; |
---|
71 | curr_cost = curr_ent->mutate(); |
---|
72 | } |
---|
73 | void revert() { |
---|
74 | E *tmp = curr_ent; |
---|
75 | curr_ent = prev_ent; |
---|
76 | prev_ent = tmp; |
---|
77 | curr_cost = prev_cost; |
---|
78 | } |
---|
79 | void saveAsBest() { |
---|
80 | *best_ent = *curr_ent; |
---|
81 | best_cost = curr_cost; |
---|
82 | } |
---|
83 | }; |
---|
84 | |
---|
85 | class EntitySkeleton { |
---|
86 | public: |
---|
87 | // returns the new cost |
---|
88 | double mutate() { return 0.0; } |
---|
89 | }; |
---|
90 | |
---|
91 | template <typename E> |
---|
92 | class SimAnn2 : public SimAnnBase { |
---|
93 | private: |
---|
94 | E *curr_ent; |
---|
95 | E *best_ent; |
---|
96 | public: |
---|
97 | SimAnn2() : SimAnnBase() {} |
---|
98 | void setEntity(E &Ent) { |
---|
99 | curr_ent = new E(Ent); |
---|
100 | best_ent = new E(Ent); |
---|
101 | } |
---|
102 | E getBestEntity() { return *best_ent; } |
---|
103 | void mutate() { |
---|
104 | curr_ent->mutate(); |
---|
105 | } |
---|
106 | void revert() { |
---|
107 | curr_ent->revert(); |
---|
108 | } |
---|
109 | void saveAsBest() { |
---|
110 | *best_ent = *curr_ent; |
---|
111 | best_cost = curr_cost; |
---|
112 | } |
---|
113 | }; |
---|
114 | |
---|
115 | class EntitySkeleton2 { |
---|
116 | public: |
---|
117 | // returns the new cost |
---|
118 | double mutate() { return 0.0; } |
---|
119 | // restores the entity to its previous state i.e. reverts the effects of |
---|
120 | // the last mutate() |
---|
121 | void revert() {} |
---|
122 | }; |
---|
123 | |
---|
124 | } |
---|
125 | |
---|
126 | #endif |
---|