PunchThroughG4Classifier.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

// PunchThroughG4Classifier.h, (c) ATLAS Detector software
 
#ifndef G4ATLASTOOLS_PUNCHTHROUGHG4CLASSIFIER_H
#define G4ATLASTOOLS_PUNCHTHROUGHG4CLASSIFIER_H
 
//
#include "AthenaBaseComps/AthAlgTool.h"
#include "G4AtlasInterfaces/IPunchThroughG4Classifier.h"
 
#include <map>
#include <memory> //for unique_ptr
 
//LWTNN
#include "lwtnn/LightweightGraph.hh"
 
class PunchThroughG4Classifier : virtual public extends<AthAlgTool, IPunchThroughG4Classifier>{
public:

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

    virtual ~PunchThroughG4Classifier() = default;

    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;

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

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

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

    virtual double computePunchThroughProbability(const G4FastTrack& fastTrack, const double simE, const std::vector<double> & simEfrac) const override;

    static std::map<std::string, std::map<std::string, double> > computeInputs(const G4FastTrack& fastTrack, const double simE, const std::vector<double> & simEfrac);

    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
     *---------------------------------------------------------------------*/
    StringProperty m_scalerConfigFileName{this,     "ScalerConfigFileName",     "", ""};
    StringProperty m_networkConfigFileName{this,    "NetworkConfigFileName",    "", ""};
    StringProperty m_calibratorConfigFileName{this, "CalibratorConfigFileName", "", ""};
 
}; // class PunchThroughG4Classifier
 
#endif //G4ATLASTOOLS_PUNCHTHROUGHG4CLASSIFIER_H