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(const std::string& constraints_file_path)
    {
        constraintsFilePath_ = constraints_file_path;
    }
 
    void ReadConstraintsFile();
    void addVariableToBinarise(const std::string& oldVarName_p, const std::string& binVarName_p);
 
    void addVariablesToBinarise(const std::map<std::string, std::string>& variables_section);
 
    [[nodiscard]] const std::map<std::string, std::string>& 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;
};