ITRT_ElectronPidTool.h

Go to the documentation of this file.

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

// ITRT_ElectronPidTool.h, (c) ATLAS Detector software
#ifndef TRK_ITRT_ELECTRONPIDTOOL_H
#define TRK_ITRT_ELECTRONPIDTOOL_H
 
#include "GaudiKernel/IAlgTool.h"
#include "GaudiKernel/EventContext.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkTrackSummary/TrackSummary.h"
#include <vector>
 
namespace Trk {
  class Track;

 
  class ITRT_ElectronPidTool : virtual public IAlgTool {
  public:
    DeclareInterfaceID(ITRT_ElectronPidTool, 1, 0);

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

    virtual double probHT(const double p,
                          const Trk::ParticleHypothesis hypothesis,
                          const int HitPart,
                          const int Layer,
                          const int Strawlayer) const = 0;
 
    virtual double probHTRun2(const EventContext& ctx,
                              float pTrk,
                              Trk::ParticleHypothesis hypothesis,
                              int TrtPart,
                              int GasType,
                              int StrawLayer,
                              float ZR,
                              float rTrkWire,
                              float Occupancy) const = 0;
  };
 
} // end of namespace
 
#endif