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         void updateTuner(SearchTuner *_newtuner) {tuner = _newtuner;}
  26         long long getTime(Problem *problem);
  27         void setTime(Problem *problem, long long time);
  28 private:
  29         SearchTuner *tuner;
  30         Vector<Problem *> problems;
  31         Hashtable<Problem *, long long, uint64_t> timetaken;
  32         friend class MultiTuner;
  33         friend void clearVector(Vector<TunerRecord *> *tunerV);
  34 };
  35
  36 class MultiTuner {
  37 public:
  38         MultiTuner(uint budget, uint rounds, uint timeout);
  39         ~MultiTuner();
  40         void addProblem(const char *filename);
  41         void addTuner(SearchTuner *tuner);
  42         void tuneK();
  43         void tuneComp();
  44         void printData();
  45         CMEMALLOC;
  46 protected:
  47         long long evaluate(Problem *problem, TunerRecord *tuner);
  48         double evaluateAll(TunerRecord *tuner);
  49         SearchTuner *mutateTuner(SearchTuner *oldTuner, uint k);
  50         void mapProblemsToTuners(Vector<TunerRecord *> *tunerV);
  51         void improveTuners(Vector<TunerRecord *> *tunerV);
  52         TunerRecord *tune(TunerRecord *tuner);
  53
  54         Vector<TunerRecord *> allTuners;
  55         Vector<Problem *> problems;
  56         Vector<TunerRecord *> tuners;
  57         uint budget;
  58         uint rounds;
  59         uint timeout;
  60         int execnum;
  61 };
  62 #endif