ProblemGeneration.h

//
// Created by marechaljas on 27/10/23.
//
 
#pragma once
 
#include <filesystem>
#include <optional>
#include <string>
 
#include <antares/solver/lps/LpsFromAntares.h>
 
#include "ConfigurationManager.h"
#include "ProblemGenerationOptions.h"
#include "antares-xpansion/helpers/ArchiveReader.h"
#include "antares-xpansion/lpnamer/helper/ProblemGenerationLogger.h"
#include "antares-xpansion/lpnamer/input_reader/MpsTxtWriter.h"
#include "antares-xpansion/lpnamer/main/ProblemGenerationExeOptions.h"
#include "antares-xpansion/lpnamer/model/Problem.h"
#include "antares-xpansion/lpnamer/model/SimulationInputMode.h"
#include "antares-xpansion/multisolver_interface/SolverAbstract.h"
#include "antares-xpansion/multisolver_interface/SolverConfig.h"
 
class ProblemGeneration
{
public:
    explicit ProblemGeneration(ProblemGenerationOptions& options);
    virtual void performAntaresSimulation(const std::filesystem::path& study_dir);
    virtual ~ProblemGeneration() = default;
    std::filesystem::path updateProblems();
    const ProblemGenerationOptions& options_;
 
private:
    virtual void RunProblemGeneration(
      const std::filesystem::path& xpansion_output_dir,
      const std::string& master_formulation,
      const std::string& additionalConstraintFilename_l,
      const std::filesystem::path& archive_path,
      std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger,
      const std::filesystem::path& log_file_path,
      const std::filesystem::path& weights_file,
      bool unnamed_problems);
 
    void ExtractUtilsFiles(const std::filesystem::path& antares_archive_path,
                           const std::filesystem::path& xpansion_output_dir,
                           std::shared_ptr<ProblemGenerationLog::ProblemGenerationLogger> logger);
    std::vector<std::shared_ptr<Problem>> getXpansionProblems(
      SolverLogManager& solver_log_manager,
      const std::vector<ProblemData>& mpsList,
      std::filesystem::path& lpDir_,
      std::shared_ptr<ArchiveReader> reader);
    virtual void set_solver(std::filesystem::path study_dir,
                            ProblemGenerationLog::ProblemGenerationLogger* logger);
 
    Antares::Solver::LpsFromAntares lps_;
    std::optional<SimulationInputMode> mode_;
    virtual void generate_antares_problems(const std::filesystem::path& study_dir,
                                           const std::filesystem::path& output_dir);
    SolverConfig solver_config_{"Coin"};
 
protected:
    void loadProblemsFromAntares(const std::filesystem::path& study_dir,
                                 const std::filesystem::path& simulation_dir,
                                 ProblemGenerationLog::ProblemGenerationLogger* logger);
    ConfigurationManager configuration_manager_;
    ConfigurationManager::ConfigDirectories directories_;
};