1162e821c7c8ce38f91b41c0ccc544938c9b3f14
[satune.git] / src / Tuner / multituner.h
1 #ifndef MULTITUNER_H
2 #define MULTITUNER_H
3 #include "classlist.h"
4 #include "structs.h"
5
6 #define AUTOTUNERFACTOR 0.3
7
8 class Problem {
9 public:
10         Problem(const char *problem);
11         char *getProblem() {return problem;}
12         ~Problem();
13         CMEMALLOC;
14 private:
15         int result;
16         char *problem;
17         friend class MultiTuner;
18 };
19
20 class TunerRecord {
21 public:
22         TunerRecord(SearchTuner *_tuner) : tuner(_tuner) {}
23         SearchTuner *getTuner() {return tuner;}
24         TunerRecord *changeTuner(SearchTuner *_newtuner);
25         long long getTime(Problem *problem);
26         void setTime(Problem *problem, long long time);
27 private:
28         SearchTuner *tuner;
29         Vector<Problem *> problems;
30         Hashtable<Problem *, long long, uint64_t> timetaken;
31         friend class MultiTuner;
32         friend void clearVector(Vector<TunerRecord *> *tunerV);
33 };
34
35 class MultiTuner {
36 public:
37         MultiTuner(uint budget, uint rounds, uint timeout);
38         ~MultiTuner();
39         void addProblem(const char *filename);
40         void addTuner(SearchTuner *tuner);
41         void tuneK();
42         void tuneComp();
43         CMEMALLOC;
44 protected:
45         long long evaluate(Problem *problem, SearchTuner *tuner);
46         double evaluateAll(TunerRecord *tuner);
47         SearchTuner *mutateTuner(SearchTuner *oldTuner, uint k);
48         void mapProblemsToTuners(Vector<TunerRecord *> *tunerV);
49         void improveTuners(Vector<TunerRecord *> *tunerV);
50         TunerRecord *tune(TunerRecord *tuner);
51
52
53         Vector<TunerRecord *> allTuners;
54         Vector<Problem *> problems;
55         Vector<TunerRecord *> tuners;
56         uint budget;
57         uint rounds;
58         uint timeout;
59         int execnum;
60 };
61 #endif