PIDNeuralNetworkParameters.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <TObject.h>
 
namespace Belle2 {
  typedef std::vector<std::tuple<size_t, float>> PIDNNMissingInputs;
 
  typedef std::vector< std::tuple<size_t, size_t, double, double, double> > PIDNNInputsToCut;
 
  class PIDNeuralNetworkParameters: public TObject {
 
  public:
    PIDNeuralNetworkParameters() {}
 
    PIDNeuralNetworkParameters(const std::string& description,
                               const std::vector<std::string>& inputNames,
                               const std::string& modelDefinition,
                               const std::vector<int>& outputSpeciesPdg,
                               const std::vector<float>& meanValues,
                               const std::vector<float>& standardDeviations,
                               const PIDNNMissingInputs& handleMissingInputs,
                               const PIDNNInputsToCut& inputsToCut
                              ):
      m_description(description),
      m_inputNames(inputNames),
      m_modelDefinition(modelDefinition),
      m_outputSpeciesPdg(outputSpeciesPdg),
      m_meanValues(meanValues),
      m_standardDeviations(standardDeviations),
      m_handleMissingInputs(handleMissingInputs),
      m_inputsToCut(inputsToCut)
    {}
 
    int pdg2OutputIndex(const int pdg, const bool throwException = true) const;
 
    bool hasPdgCode(const int pdg, const bool throwException = false) const {return pdg2OutputIndex(pdg, throwException) >= 0;}
 
    const std::string& getModelDefinition() const {return m_modelDefinition;}
    const std::vector<float>& getMeanValues() const {return m_meanValues;}
    const std::vector<float>& getStandardDeviations() const {return m_standardDeviations;}
    const PIDNNMissingInputs& getHandleMissingInputs() const {return m_handleMissingInputs;}
    const PIDNNInputsToCut& getInputsToCut() const {return m_inputsToCut;}
    size_t getInputSize() const {return m_inputNames.size();}
    const std::vector<std::string>& getInputNames() const {return m_inputNames;}
    size_t getInputIndex(const std::string& name) const;
 
    const std::string& getDescription() const {return m_description;}
 
    const std::vector<int>& getOutputSpeciesPdg() const {return m_outputSpeciesPdg;}
 
  private:
 
    std::string m_description;                 
    std::vector<std::string> m_inputNames;     
    std::string m_modelDefinition;             
    std::vector<int> m_outputSpeciesPdg;       
    std::vector<float> m_meanValues;           
    std::vector<float> m_standardDeviations;   
    PIDNNMissingInputs m_handleMissingInputs;  
    PIDNNInputsToCut m_inputsToCut;            
    ClassDef(PIDNeuralNetworkParameters, 1); 
  };
 
} // Belle2 namespace