// -*- C++ -*-

 #ifndef LEMON_SIMANN_H

 #define LEMON_SIMANN_H

 /// \ingroup experimental

 /// \file

 /// \brief Simulated annealing framework.

 ///

 /// \todo A test and some demo should be added

 /// \todo Doc should be improved

 /// \author Akos Ladanyi

 #include <cstdlib>

 #include <cmath>

 #include <limits>

 #include <lemon/time_measure.h>

 namespace lemon {

 /// \addtogroup experimental

 /// @{

   class SimAnnBase;

   /// \brief A base class for controllers.

   class ControllerBase {

   public:

     friend class SimAnnBase;

     /// \brief Pointer to the simulated annealing base class.

     SimAnnBase *simann;

     /// \brief Initializes the controller.

     virtual void init() {}

     /// \brief This is called by the simulated annealing class when a

     /// neighbouring state gets accepted.

     virtual void acceptEvent() {}

     /// \brief This is called by the simulated annealing class when the

     /// accepted neighbouring state's cost is less than the best found one's.

     virtual void improveEvent() {}

     /// \brief This is called by the simulated annealing class when a

     /// neighbouring state gets rejected.

     virtual void rejectEvent() {}

     /// \brief Decides whether to continue the annealing process or not.

     virtual bool next() = 0;

     /// \brief Decides whether to accept the current solution or not.

     virtual bool accept() = 0;

     /// \brief Destructor.

     virtual ~ControllerBase() {}

   };

   /// \brief Skeleton of an entity class.

   class EntityBase {

   public:

     /// \brief Makes a minor change to the entity.

     /// \return the new cost

     virtual double mutate() = 0;

     /// \brief Restores the entity to its previous state i.e. reverts the

     /// effects of the last mutate().

     virtual void revert() = 0;

     /// \brief Makes a copy of the entity.

     virtual EntityBase* clone() = 0;

     /// \brief Makes a major change to the entity.

     virtual void randomize() = 0;

     /// \brief Destructor.

     virtual ~EntityBase() {}

   };

   /// \brief Simulated annealing abstract base class.

   /// Can be used to derive a custom simulated annealing class if \ref SimAnn

   /// doesn't fit your needs.

   class SimAnnBase {

   private:

     /// \brief Pointer to the controller.

     ControllerBase *controller;

     /// \brief Cost of the current solution.

     double curr_cost;

     /// \brief Cost of the best solution.

     double best_cost;

     /// \brief Cost of the previous solution.

     double prev_cost;

     /// \brief Cost of the solution preceding the previous one.

     double prev_prev_cost;

     /// \brief Number of iterations.

     long iter;

     /// \brief Number of iterations which did not improve the solution since

     /// the last improvement.

     long last_impr;

   protected:

     /// \brief Step to a neighbouring state.

     virtual double mutate() = 0;

     /// \brief Reverts the last mutate().

     virtual void revert() = 0;

     /// \brief Saves the current solution as the best one.

     virtual void saveAsBest() = 0;

     /// \brief Does initializations before each run.

     virtual void init() {

       controller->init();

       curr_cost = prev_cost = prev_prev_cost = best_cost =

         std::numeric_limits<double>::infinity();

       iter = last_impr = 0;

     }

   public:

     /// \brief Sets the controller class to use.

     void setController(ControllerBase &_controller) {

       controller = &_controller;

       controller->simann = this;

     }

     /// \brief Returns the cost of the current solution.

     double getCurrCost() const { return curr_cost; }

     /// \brief Returns the cost of the previous solution.

     double getPrevCost() const { return prev_cost; }

     /// \brief Returns the cost of the best solution.

     double getBestCost() const { return best_cost; }

     /// \brief Returns the number of iterations done.

     long getIter() const { return iter; }

     /// \brief Returns the ordinal number of the last iteration when the

     /// solution was improved.

     long getLastImpr() const { return last_impr; }

     /// \brief Performs one iteration.

     bool step() {

       iter++;

       prev_prev_cost = prev_cost;

       prev_cost = curr_cost;

       curr_cost = mutate();

       if (controller->accept()) {

         controller->acceptEvent();

         last_impr = iter;

         if (curr_cost < best_cost) {

           best_cost = curr_cost;

           saveAsBest();

           controller->improveEvent();

         }

       }

       else {

         revert();

         curr_cost = prev_cost;

         prev_cost = prev_prev_cost;

         controller->rejectEvent();

       }

       return controller->next();

     }

     /// \brief Performs a given number of iterations.

     /// \param n the number of iterations

     bool step(int n) {

       for(; n > 0 && step(); --n) ;

       return !n;

     }

     /// \brief Starts the annealing process.

     void run() {

       init();

       do { } while (step());

     }

     /// \brief Destructor.

     virtual ~SimAnnBase() {}

   };

   /// \brief Simulated annealing class.

   class SimAnn : public SimAnnBase {

   private:

     /// \brief Pointer to the current entity.

     EntityBase *curr_ent;

     /// \brief Pointer to the best entity.

     EntityBase *best_ent;

     /// \brief Does initializations before each run.

     void init() {

       SimAnnBase::init();

       if (best_ent) delete best_ent;

       best_ent = NULL;

       curr_ent->randomize();

     }

   public:

     /// \brief Constructor.

     SimAnn() : curr_ent(NULL), best_ent(NULL) {}

     /// \brief Destructor.

     virtual ~SimAnn() {

       if (best_ent) delete best_ent;

     }

     /// \brief Step to a neighbouring state.

     double mutate() {

       return curr_ent->mutate();

     }

     /// \brief Reverts the last mutate().

     void revert() {

       curr_ent->revert();

     }

     /// \brief Saves the current solution as the best one.

     void saveAsBest() { 

       if (best_ent) delete best_ent;

       best_ent = curr_ent->clone();

     }

     /// \brief Sets the current entity.

     void setEntity(EntityBase &_ent) {

       curr_ent = &_ent;

     }

     /// \brief Returns a copy of the best found entity.

     EntityBase* getBestEntity() { return best_ent->clone(); }

   };

   /// \brief A simple controller for the simulated annealing class.

   /// This controller starts from a given initial temperature and evenly

   /// decreases it.

   class SimpleController : public ControllerBase {

   private:

     /// \brief Maximum number of iterations.

     long max_iter;

     /// \brief Maximum number of iterations which do not improve the

     /// solution.

     long max_no_impr;

     /// \brief Temperature.

     double temp;

     /// \brief Annealing factor.

     double ann_fact;

     /// \brief Constructor.

     /// \param _max_iter maximum number of iterations

     /// \param _max_no_impr maximum number of consecutive iterations which do

     ///        not yield a better solution

     /// \param _temp initial temperature

     /// \param _ann_fact annealing factor

   public:

     SimpleController(long _max_iter = 500000, long _max_no_impr = 20000,

     double _temp = 1000.0, double _ann_fact = 0.9999) : max_iter(_max_iter),

       max_no_impr(_max_no_impr), temp(_temp), ann_fact(_ann_fact)

     {

       srand48(time(0));

     }

     /// \brief This is called when a neighbouring state gets accepted.

     void acceptEvent() {}

     /// \brief This is called when the accepted neighbouring state's cost is

     /// less than the best found one's.

     void improveEvent() {}

     /// \brief This is called when a neighbouring state gets rejected.

     void rejectEvent() {}

     /// \brief Decides whether to continue the annealing process or not. Also

     /// decreases the temperature.

     bool next() {

       temp *= ann_fact;

       bool quit = (simann->getIter() > max_iter) ||

         (simann->getIter() - simann->getLastImpr() > max_no_impr);

       return !quit;

     }

     /// \brief Decides whether to accept the current solution or not.

     bool accept() {

       double cost_diff = simann->getCurrCost() - simann->getPrevCost();

       return (drand48() <= exp(-(cost_diff / temp)));

     }

     /// \brief Destructor.

     virtual ~SimpleController() {}

   };

   /// \brief A controller with preset running time for the simulated annealing

   /// class.

   /// With this controller you can set the running time of the annealing

   /// process in advance. It works the following way: the controller measures

   /// a kind of divergence. The divergence is the difference of the average

   /// cost of the recently found solutions the cost of the best found one. In

   /// case this divergence is greater than a given threshold, then we decrease

   /// the annealing factor, that is we cool the system faster. In case the

   /// divergence is lower than the threshold, then we increase the temperature.

   /// The threshold is a function of the elapsed time which reaches zero at the

   /// desired end time.

   class AdvancedController : public ControllerBase {

   private:

     /// \brief Timer class to measure the elapsed time.

     Timer timer;

     /// \brief Calculates the threshold value.

     /// \param time the elapsed time in seconds

     virtual double threshold(double time) {

       return (-1.0) * start_threshold / end_time * time + start_threshold;

     }

     /// \brief Parameter used to calculate the running average.

     double alpha;

     /// \brief Parameter used to decrease the annealing factor.

     double beta;

     /// \brief Parameter used to increase the temperature.

     double gamma;

     /// \brief The time at the end of the algorithm.

     double end_time;

     /// \brief The time at the start of the algorithm.

     double start_time;

     /// \brief Starting threshold.

     double start_threshold;

     /// \brief Average cost of recent solutions.

     double avg_cost;

     /// \brief Temperature.

     double temp;

     /// \brief Annealing factor.

     double ann_fact;

     /// \brief Initial annealing factor.

     double init_ann_fact;

     /// \brief True when the annealing process has been started.

     bool start;

   public:

     /// \brief Constructor.

     /// \param _end_time running time in seconds

     /// \param _alpha parameter used to calculate the running average

     /// \param _beta parameter used to decrease the annealing factor

     /// \param _gamma parameter used to increase the temperature

     /// \param _ann_fact initial annealing factor

     AdvancedController(double _end_time, double _alpha = 0.2,

     double _beta = 0.9, double _gamma = 1.6, double _ann_fact = 0.9999) :

     alpha(_alpha), beta(_beta), gamma(_gamma), end_time(_end_time),

     ann_fact(_ann_fact), init_ann_fact(_ann_fact), start(false)

     {

       srand48(time(0));

     }

     /// \brief Does initializations before each run.

     void init() {

       avg_cost = simann->getCurrCost();

     }

     /// \brief This is called when a neighbouring state gets accepted.

     void acceptEvent() {

       avg_cost = alpha * simann->getCurrCost() + (1.0 - alpha) * avg_cost;

       if (!start) {

         static int cnt = 0;

         cnt++;

         if (cnt >= 100) {

           // calculate starting threshold and starting temperature

           start_threshold = 5.0 * fabs(simann->getBestCost() - avg_cost);

           temp = 10000.0;

           start = true;

           timer.restart();

         }

       }

     }

     /// \brief Decides whether to continue the annealing process or not.

     bool next() {

       if (!start) {

         return true;

       }

       else {

         double elapsed_time = timer.realTime();

         if (fabs(avg_cost - simann->getBestCost()) > threshold(elapsed_time)) {

           // decrease the annealing factor

           ann_fact *= beta;

         }

         else {

           // increase the temperature

           temp *= gamma;

           // reset the annealing factor

           ann_fact = init_ann_fact;

         }

         temp *= ann_fact;

         return elapsed_time < end_time;

       }

     }

     /// \brief Decides whether to accept the current solution or not.

     bool accept() {

       if (!start) {

         return true;

       }

       else {

         double cost_diff = simann->getCurrCost() - simann->getPrevCost();

         return (drand48() <= exp(-(cost_diff / temp)));

       }

     }

     /// \brief Destructor.

     virtual ~AdvancedController() {}

   };

 /// @}

 }

 #endif