AdditionalConstraints.h

#pragma once
 
#include <map>
#include <set>
#include <string>
#include <utility>
 
#include "antares-xpansion/lpnamer/helper/ProblemGenerationLogger.h"
 
class AdditionalConstraint : public std::map<std::string, double> {
 private:
  std::string _sectionName;
  std::string _name;
  std::string _sign;
  double _rhs = 0.0;
 
 public:
  AdditionalConstraint() = default;
 
  AdditionalConstraint(std::string sectionName_p, std::string constraintName_p,
                       std::string sign_p, double rhs_p)
      : _sectionName(std::move(sectionName_p)),
        _name(std::move(constraintName_p)),
        _sign(std::move(sign_p)),
        _rhs(rhs_p) {}
 
  virtual ~AdditionalConstraint() = default;
 
  [[nodiscard]] std::string getName() const { return _name; }
 
  [[nodiscard]] double getRHS() const { return _rhs; }
 
  [[nodiscard]] std::string getSign() const { return _sign; }
 
  void setCoeff(const std::string& varName_p, const double varCoeff_p) {
    (*this)[varName_p] = varCoeff_p;
  }
};
 
class AdditionalConstraints
    : public std::map<std::string, AdditionalConstraint> {
 private:
  // set of variables to which a binary corresponding variable will be created
  std::map<std::string, std::string> _variablesToBinarise;
  std::set<std::string> _binaryVariables;
  std::string constraintsFilePath_;
  std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger_;
 
 public:
  explicit AdditionalConstraints(
      std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger)
      : logger_(std::move(logger)) {}
 
  explicit AdditionalConstraints(
      std::string constraints_file_path,
      std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger);
 
  void SetConstraintsFile(std::string const& constraints_file_path) {
    constraintsFilePath_ = constraints_file_path;
  }
  void ReadConstraintsFile();
  void addVariableToBinarise(const std::string& oldVarName_p,
                             const std::string& binVarName_p);
 
  void addVariablesToBinarise(
      std::map<std::string, std::string> const& variables_section);
 
  [[nodiscard]] std::map<std::string, std::string> const&
  getVariablesToBinarise() const;
 
  void constructAdditionalConstraints(
      const std::string& sectionName_l,
      const std::map<std::string, std::string>& constarintsSection_l);
 
  [[nodiscard]] std::string checkAndReturnConstraintName(
      const std::string& sectionName_l,
      const std::map<std::string, std::string>& constarintsSection_l) const;
 
  [[nodiscard]] std::string checkAndReturnSectionSign(
      const std::string& sectionName_l,
      const std::map<std::string, std::string>& constarintsSection_l) const;
 
  [[nodiscard]] double checkAndReturnSectionRhs(
      const std::string& sectionName_l,
      const std::map<std::string, std::string>& constarintsSection_l) const;
};