LinkProblemsGenerator.h

#pragma once
 
#include <filesystem>
#include <memory>
#include <mutex>
#include <utility>
 
#include "antares-xpansion/helpers/ArchiveReader.h"
#include "antares-xpansion/helpers/ArchiveWriter.h"
#include "antares-xpansion/helpers/FileInBuffer.h"
#include "antares-xpansion/lpnamer/helper/ProblemGenerationLogger.h"
#include "antares-xpansion/lpnamer/input_reader/MpsTxtWriter.h"
#include "antares-xpansion/lpnamer/input_reader/VariableFileReader.h"
#include "antares-xpansion/lpnamer/problem_modifier/IProblemProviderPort.h"
#include "antares-xpansion/lpnamer/problem_modifier/IProblemVariablesProviderPort.h"
#include "antares-xpansion/lpnamer/problem_modifier/IProblemWriter.h"
#include "antares-xpansion/lpnamer/problem_modifier/ProblemModifier.h"
#include "antares-xpansion/multisolver_interface/SolverAbstract.h"
#include "antares-xpansion/multisolver_interface/SolverConfig.h"
#include "antares-xpansion/xpansion_interfaces/StringManip.h"
 
const std::string CANDIDATES_INI{"candidates.ini"};
const std::string STRUCTURE_FILE{"structure.txt"};
const std::string STUDY_FILE{"study.antares"};
using CandidateNameAndProblemName = std::pair<std::string, std::string>;
using ColId = unsigned int;
using Couplings = std::map<CandidateNameAndProblemName, ColId>;
 
class LinkProblemsGenerator {
 public:
  LinkProblemsGenerator(
      std::filesystem::path& lpDir, const std::vector<ActiveLink>& links,
      SolverConfig solver_config,
      std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger,
      SolverLogManager& solver_log_manager, bool rename_problems)
      : _links(links),
        solver_config_(std::move(solver_config)),
        lpDir_(lpDir),
        logger_(std::move(logger)),
        rename_problems_(rename_problems),
        solver_log_manager_(solver_log_manager) {}
 
  void treatloop(const std::filesystem::path& root, Couplings& couplings,
                 const std::vector<ProblemData>& mps_list,
                 IProblemWriter* writer);
  void treat(const std::string& problem_name, Couplings& couplings,
             Problem* problem, IProblemVariablesProviderPort* variable_provider,
             IProblemWriter* writer);
  void treat(const std::string& problem_name, Couplings& couplings,
             IProblemProviderPort* problem_provider,
             IProblemVariablesProviderPort* variable_provider,
             IProblemWriter* writer);
 
 private:
  const std::vector<ActiveLink>& _links;
  SolverConfig solver_config_;
  std::filesystem::path lpDir_ = "";
  std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger_;
  mutable std::mutex coupling_mutex_;
  bool rename_problems_ = false;
  SolverLogManager& solver_log_manager_;
};