GeneralModelModifier.h

Go to the documentation of this file.

#ifndef GENERAL_MODEL_MODIFIER_H
#define GENERAL_MODEL_MODIFIER_H
 
#include <unordered_map>
#include <cctype>
#include <algorithm>
#include <optional>
#include <string>
#include <utility>
#include <vector>
 
#include "ModelModifier.h"
#include "ModelFileChecker.h"
#include "config.hpp"
 
class GeneralModelModifier : public ModelModifier {
public:
    GeneralModelModifier(std::string wilson,
                         std::string model,
                         std::string model_path,
                         std::optional<int> model_template_index = std::nullopt);
 
    GeneralModelModifier(std::string wilson,
                         std::string output_model,
                         std::string target_model,
                         std::string model_path,
                         std::optional<int> model_template_index,
                         bool disable_non_sm_particles,
                         bool bsm_split_generation = false,
                         bool full_target_generation = false,
                         bool tree_first_fallback = false);
 
    void modifyLine(std::string& line) override;
 
    void addLine(std::ofstream& outputFile, const std::string& currentLine) override;
 
    static std::string resolveModelInstantiation(const std::string& model,
                                                 const std::string& model_path,
                                                 std::optional<int> model_template_index = std::nullopt);
 
    static std::string modelSignature(const std::string& model,
                                      const std::string& model_path,
                                      std::optional<int> model_template_index = std::nullopt);
 
private:
    static std::string makeSmFilterHelper();
    bool usesRegPropSplit() const;
    bool usesGenericTreeFirst() const;
    static void replaceWilsonOrderArgument(std::string& line);
    bool consumeTreeSafeWilsonCall(std::ofstream& outputFile,
                                   const std::string& currentLine,
                                   bool pair_return,
                                   bool count_graphs);
    void emitTreeSafeWilsonCall(std::ofstream& outputFile,
                                bool pair_return,
                                bool count_graphs);
 
    std::string output_model{}; 
    std::string target_model{}; 
    std::string model_path{};   
    std::string marty_path{};   
    std::optional<int> model_template_index{}; 
    bool disable_non_sm_particles{false}; 
    bool bsm_split_generation{false}; 
    bool full_target_generation{false}; 
    bool tree_first_fallback{false}; 
    bool inside_calculate_function{false}; 
    bool skip_old_main{false}; 
    bool expression_returned{false}; 
    std::string generic_builder_name{}; 
    bool buffering_tree_safe_wilson_call{false}; 
    std::vector<std::string> tree_safe_wilson_call_lines{}; 
    ModelClassInfo model_class{}; 
    std::string model_instantiation{}; 
};
 
#endif