MartyInterface.h

Go to the documentation of this file.

#ifndef MARTY_INTERFACE_H
#define MARTY_INTERFACE_H
 
#include <memory>
#include <filesystem>
#include <iostream>
#include <algorithm>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <optional>
 
#include "config.hpp"
#include "FileNameManager.h"
#include "GeneralModelModifier.h"
#include "CodeGenerator.h"
#include "GppCompilerStrategy.h"
#include "MakeCompilerStrategy.h"
#include "Logger.h"
#include "GeneralNumModelModifier.h"
#include "ICoreAPI.h"
#include "IMartyParameterProxy.h"
 
class MartyInterface {
public:
    MartyInterface(std::shared_ptr<ICoreAPI<Model>> core_api, std::shared_ptr<IMartyParameterProxy<std::string, LhaID>> sm_proxy, std::shared_ptr<IInterpreterPortsFactory> ports, std::shared_ptr<IMartyParameterProxy<std::string, LhaID>> bsm_proxy = nullptr) : core_api(core_api), param_proxy_sm(sm_proxy), ports(ports), param_proxy_bsm(bsm_proxy)
     {}
    
    MartyInterface();
 
    void generate(std::string wilson, std::string model, std::string model_path);
 
    void generate(std::string wilson,
                  std::string output_model,
                  std::string target_model,
                  std::string model_path,
                  bool sm_like_filter,
                  bool bsm_split_generation = false,
                  bool full_target_generation = false);
 
    void compile_run(std::string wilson, std::string model);
 
    void generate_numlib(std::string wilson, std::string model);
 
    void generate_numlib(std::string wilson,
                         std::string output_model,
                         std::string target_model,
                         bool bsm_split_generation = false,
                         bool full_target_generation = false);
 
    void compile_run_libs(std::string wilson, std::string model, double Q_match);
 
    void calculate(std::string wilson, std::string model, double Q_match, std::string model_path);
 
    void calculate(std::string wilson,
                   std::string output_model,
                   std::string target_model,
                   double Q_match,
                   std::string model_path,
                   bool sm_like_filter,
                  bool bsm_split_generation = false,
                  bool full_target_generation = false);
 
    std::string calculate_isolated(std::string wilson,
                                   std::string output_model,
                                   std::string target_model,
                                   double Q_match,
                                   std::string model_path,
                                   bool sm_like_filter,
                                   bool bsm_split_generation = false,
                                   bool full_target_generation = false);
 
    std::unordered_set<InterpretedParam> get_dependencies(std::string wilson);
 
    std::set<std::string> get_special_blocks();
 
private:
    bool already_run(std::string&& outputBinary);
 
    std::string output_binary_name(std::string& wilson, std::string& model);
 
    std::optional<int> resolve_model_template_index(const std::string& model) const;
 
    void invalidate_template_model_cache_if_needed(const std::string& wilson,
                                                   const std::string& output_model,
                                                   const std::string& target_model,
                                                   const std::string& model_path,
                                                   std::optional<int> model_template_index,
                                                   bool sm_like_filter,
                                                   bool bsm_split_generation,
                                                   bool full_target_generation) const;
 
    bool artifacts_ready(const std::string& wilson,
                         const std::string& output_model,
                         const std::string& target_model,
                         const std::string& model_path,
                         bool sm_like_filter,
                         bool bsm_split_generation,
                         bool full_target_generation) const;
 
    void ensure_built(const std::string& wilson,
                      const std::string& output_model,
                      const std::string& target_model,
                      const std::string& model_path,
                      bool sm_like_filter,
                      bool bsm_split_generation,
                      bool full_target_generation);
 
    std::unordered_map<std::string, double> snapshot_numeric_params(
        const std::string& wilson,
        const std::string& output_model,
        const std::string& target_model,
        bool bsm_split_generation,
        bool full_target_generation
    );
 
    void compile_numlib(const std::string& wilson, const std::string& model);
 
    std::set<std::string> specials_block {"KIN", "WEIN", "Finite", "REGPROP", "BETA"};
 
    std::string num_file_path{};
 
    std::unordered_map<std::string, std::unordered_set<InterpretedParam>> dependencies;
 
    std::shared_ptr<ICoreAPI<Model>> core_api;
 
    std::shared_ptr<IMartyParameterProxy<std::string, LhaID>> param_proxy_sm;
    
    std::shared_ptr<IInterpreterPortsFactory> ports;
 
 
    std::shared_ptr<IMartyParameterProxy<std::string, LhaID>> param_proxy_bsm = nullptr;
};
 
#endif