TRT_ElectronPidToolRun2.h

Go to the documentation of this file.

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

// TRT_ElectronPidToolRun2.h, (c) ATLAS Detector software
 
#ifndef INDETTRT_ELECTRONPIDTOOLRUN2_H
#define INDETTRT_ELECTRONPIDTOOLRUN2_H
 
#include "AthenaBaseComps/AthAlgTool.h"       // Exchange IN
 
#include "GaudiKernel/ITHistSvc.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadCondHandleKey.h"
 
#include "TrkToolInterfaces/ITRT_ElectronPidTool.h"
#include "TRT_ElectronPidTools/ITRT_ElectronToTTool.h"
#include "TRT_ConditionsData/HTcalculator.h"
#include "TRT_ConditionsNN/TRTPIDNN.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
 
#include "TRT_ElectronPidTools/ITRT_LocalOccupancy.h"
 
#include "TRT_ConditionsServices/ITRT_StrawStatusSummaryTool.h"
 
 
#include <vector>
#include <string>
 
class AtlasDetectorID;
class Identifier;
class TRT_ID;
namespace InDetDD{ class TRT_DetectorManager; }
 
namespace InDet{
    class ITRT_LocalOccupancy;
}
 
class ITRT_StrawSummaryTool;
 
class ITRT_ToT_dEdx;
 
namespace Trk {
  class Track;
}
 
namespace InDet 
{

 
  class TRT_ElectronPidToolRun2;
 
  class TRT_ElectronPidToolRun2 : virtual public Trk::ITRT_ElectronPidTool, virtual public ITRT_ElectronToTTool, public AthAlgTool
  {
    //Athena specific, called externally:
  public:
    TRT_ElectronPidToolRun2(const std::string&,const std::string&,const IInterface*);

    virtual ~TRT_ElectronPidToolRun2 ();
      
    virtual StatusCode initialize() override;

    virtual StatusCode finalize() override;

    virtual std::vector<float> electronProbability(
      const EventContext& ctx,
      const Trk::Track& track) const override final;

    static std::vector<float> electronProbability_old(const Trk::Track& track);
 
    virtual double probHT(const double pTrk,
                          const Trk::ParticleHypothesis hypothesis,
                          const int HitPart,
                          const int Layer,
                          const int Strawlayer) const override final;
 
    virtual double probHTRun2(const EventContext& ctx,
                              float pTrk,
                              Trk::ParticleHypothesis hypothesis,
                              int TrtPart,
                              int GasType,
                              int StrawLayer,
                              float ZR,
                              float rTrkWire,
                              float Occupancy) const override final;
 
  private:
 
      //Check valid TRT straw:
    bool CheckGeometry(int BEC, int Layer, int Strawlayer) const;
 
    double inline sqr(double a) {return a*a;} 
 
    const TRT_ID* m_trtId{nullptr}; // TRT ID helper (identifying barrel/wheels and global position)
    UnsignedIntegerProperty m_minTRThits{this, "MinimumTRThitsForIDpid", 5,
      "Minimum number of TRT hits to give PID"};
    FloatProperty m_ptMinNN{this, "MinimumTrackPtForNNPid", 2000.,
      "Minimum track pt to calculate NN response for PID"};
    BooleanProperty m_calculateNN{this, "CalculateNNPid", true,
      "Decide whether to use NN PID"};
 
    ToolHandle<ITRT_ToT_dEdx> m_TRTdEdxTool{ this,
                                             "TRT_ToT_dEdx_Tool",
                                             {},
                                             "TRT ToT dEdx Tool"};
    ToolHandle<InDet::ITRT_LocalOccupancy> m_LocalOccTool{
      this,
      "TRT_LocalOccupancyTool",
      {},
      "TRT Local occupancy tool"
    };
    ToolHandle<ITRT_StrawStatusSummaryTool> m_TRTStrawSummaryTool{
      this,
      "TRTStrawSummaryTool",
      "InDetTRTStrawStatusSummaryTool",
      "TRT straw summary tool"
    };
 
    SG::ReadCondHandleKey<HTcalculator> m_HTReadKey{ this,
                                                     "HTcalculator",
                                                     "HTcalculator",
                                                     "HTcalculator in-key" };
 
    SG::ReadCondHandleKey<InDet::TRTPIDNN> m_TRTPIDNNReadKey{ this,
                                                     "TRTPIDNN",
                                                     "TRTPIDNN",
                                                     "TRTPIDNN in-key" };
   }; 
} // end of namespace
 
#endif