PunchThroughClassifier.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_PUNCHTHROUGHTOOLS_SRC_PUNCHTHROUGHCLASSIFIER_H
#define ISF_PUNCHTHROUGHTOOLS_SRC_PUNCHTHROUGHCLASSIFIER_H 1
 
// ISF includes
#include "ISF_Event/ISFParticle.h"
 
#include "ISF_FastCaloSimInterfaces/IPunchThroughClassifier.h"
 
#include "ISF_FastCaloSimEvent/TFCSSimulationState.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
 
 
#include <map>
 
//LWTNN
#include "lwtnn/LightweightGraph.hh"
 
 
namespace ISF {
 
class PunchThroughClassifier : public extends<AthAlgTool, IPunchThroughClassifier> {
public:

    PunchThroughClassifier(const std::string&,const std::string&,const IInterface*);

    virtual ~PunchThroughClassifier() = default;

    virtual StatusCode initialize() override final;

    StatusCode initializeScaler(const std::string & scalerConfigFile);

    StatusCode initializeNetwork(const std::string & networkConfigFile);

    StatusCode initializeCalibrator(const std::string & calibratorConfigFile);

    virtual double computePunchThroughProbability(const ISF::ISFParticle &isfp, const TFCSSimulationState& simulstate) const override;

    static std::map<std::string, std::map<std::string, double> > computeInputs(const ISF::ISFParticle &isfp, const TFCSSimulationState& simulstate) ;

    std::map<std::string, std::map<std::string, double> > scaleInputs(std::map<std::string, std::map<std::string, double> >& inputs) const;

    double calibrateOutput(double& networkOutput) const;
 
private:
    std::unique_ptr<lwt::LightweightGraph> m_graph{};

    double m_scalerMin{};
    double m_scalerMax{};
    std::map<std::string, double> m_scalerMinMap;
    std::map<std::string, double> m_scalerMaxMap;

    std::string m_calibratorConfigFile;
    double m_calibrationMin{};
    double m_calibrationMax{};
    std::map<double, double> m_calibrationMap;
 
    //properties
    std::string m_networkConfigFileName;
    std::string m_scalerConfigFileName;
    std::string m_calibratorConfigFileName;
 
};
}
 
#endif //ISF_PUNCHTHROUGHTOOLS_SRC_PUNCHTHROUGHCLASSIFIER_H