Tool_ModeDiscriminator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PANTAUALGS_TOOL_MODEDISCRIMINATOR_H
#define PANTAUALGS_TOOL_MODEDISCRIMINATOR_H

#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "AsgTools/PropertyWrapper.h"

#include "PanTauAlgs/HelperFunctions.h"
#include "PanTauAlgs/ITool_ModeDiscriminator.h"
#include "PanTauAlgs/ITool_InformationStore.h"

#include <vector>
#include <string>
 
#include "MVAUtils/BDT.h"
 
namespace PanTau {
  class PanTauSeed;
}
 
 
 
namespace PanTau {

  class Tool_ModeDiscriminator : public asg::AsgTool, virtual public PanTau::ITool_ModeDiscriminator {
 
    ASG_TOOL_CLASS1(Tool_ModeDiscriminator, PanTau::ITool_ModeDiscriminator)
 
      public:
        
    Tool_ModeDiscriminator(const std::string& name);
    virtual ~Tool_ModeDiscriminator ();
    
    virtual StatusCode initialize();
 
    virtual double getResponse(PanTau::PanTauSeed* inSeed, bool& isOK) const;
        
  private:
        
    void                                        updateReaderVariables(PanTau::PanTauSeed* inSeed, std::vector<float>& list_BDTVariableValues) const;
        
    ToolHandle<PanTau::ITool_InformationStore>  m_Tool_InformationStore{this, "Tool_InformationStore", "PanTau::Tool_InformationStore/Tool_InformationStore", "Handle to the information store tool"}; 
    Gaudi::Property<std::string> m_calib_path{this, "calibFolder", "", "Location of calib files in cvmfs"};
    Gaudi::Property<std::string> m_Name_ModeCase{this, "Name_ModeCase", "InvalidModeCase", "Name of the two modes to be distinguished for this instance"};
    Gaudi::Property<std::string> m_Tool_InformationStoreName{this, "Tool_InformationStoreName", "PanTau::Tool_InformationStore/Tool_InformationStore", "Handle to the information store tool"};
 
    PanTau::HelperFunctions   m_HelperFunctions;
    std::vector<std::unique_ptr<MVAUtils::BDT> > m_MVABDT_List;
        
    std::vector<double>                         m_BinEdges_Pt;
    std::string                                 m_MethodName;
    std::vector<std::string>                    m_List_BDTVariableNames;
    std::vector<double>                         m_List_BDTVariableDefaultValues;
 
    bool m_init=false;
 
  public:
    inline bool isInitialized(){return m_init;}
        
  };
} // end of namespace PanTau
#endif // PANTAUALGS_TOOL_MODEDISCRIMINATOR_H