local_optimisation.hpp

Go to the documentation of this file.

#include "../../graph/assignment.hpp"
#include "../../utils/common.hpp"
#include "assigner.hpp"

#include <bitset>
#include <cstdint>
#include <map>
#include <vector>

#ifndef INDIGOX_ALGORITHM_ELECTRON_ASSIGNMENT_LOCAL_OPTIMISATION_HPP
#define INDIGOX_ALGORITHM_ELECTRON_ASSIGNMENT_LOCAL_OPTIMISATION_HPP

namespace indigox::algorithm {

  class IXLocalOptimisation : public IXElectronAssigner::AssignAlgorithm {
    enum _lo_bool_opts {
      __all_mins,  
      __cache,     
      __cache_inf, 
      __num_opts   
    };

  public:
    struct Settings {
      static utils::Option OPTIMISE_LEVEL;

      static uint32_t TIMEOUT;

      static utils::Option USE_CACHE;
    };

  public:
    IXLocalOptimisation() = delete; // no default constructor

    IXLocalOptimisation(const ScoreTable &t);

    virtual void Initalise(const Molecule &mol) override;

    virtual void Run() override;

  private:
    void BuildLocationMasks();

    void GetNeighbourAssignments(const AssignMask &m,
                                 std::vector<AssignMask> &nbrs);

  private:
    using LocMasks_t = std::map<graph::AGVertex, AssignMask>;
    LocMasks_t _loc_masks;
    uint32_t _timeout;
    std::bitset<__num_opts> _lo_opts;
  };

} // namespace indigox::algorithm

#endif /* INDIGOX_ALGORITHM_ELECTRON_ASSIGNMENT_LOCAL_OPTIMISATION_HPP */