PassThroughSaltModel.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef FLAVORTAGINFERENCE_PASSTHROUGHSALTMODEL_H
#define FLAVORTAGINFERENCE_PASSTHROUGHSALTMODEL_H
 
#include "FlavorTagInference/ISaltModel.h"
#include "FlavorTagInference/SaltModelGraphConfig.h"
#include "nlohmann/json.hpp"
 
#include <string>
#include <vector>
#include <map>
 
namespace FlavorTagInference {

  class PassThroughSaltModel : public ISaltModel {
  public:
    PassThroughSaltModel(const nlohmann::json& config);
    virtual ~PassThroughSaltModel() = default;
 
    InferenceOutput runInference(InputMap& gnn_inputs) const override;
 
    const SaltModelGraphConfig::GraphConfig getGraphConfig() const override;
    const OutputConfig& getOutputConfig() const override;
    SaltModelVersion getSaltModelVersion() const override;
    const std::string& getModelName() const override;
 
  private:
    std::string m_model_name;
    SaltModelGraphConfig::GraphConfig m_graph_config;
    OutputConfig m_output_config;

    std::vector<std::string> m_jet_input_names;
    std::vector<std::string> m_jet_output_names;

    struct ConstituentNode {
      std::string node_name;       
      std::string input_key;       
      size_t num_vars = 0;         
      std::vector<std::string> output_names; 
      std::vector<std::string> var_types;  
    };
    std::vector<ConstituentNode> m_constituent_nodes;
  };
 
} // namespace FlavorTagInference
 
#endif