ITRT_ElectronPidTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 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 "GaudiKernel/ThreadLocalContext.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkTrackSummary/TrackSummary.h"
#include <vector>
 
namespace Trk {
  class Track;
 
  static const InterfaceID IID_ITRT_ElectronPidTool("Trk::ITRT_ElectronPidTool", 1, 0);
 
  class ITRT_ElectronPidTool : virtual public IAlgTool {
  public:
    static const InterfaceID& interfaceID( ) ;
 
    virtual std::vector<float> electronProbability(
      const EventContext& ctx,
      const Trk::Track& track) const = 0;
 
    std::vector<float> electronProbability(
      const Trk::Track& track) const {
      return  electronProbability(Gaudi::Hive::currentContext(),track);
    }
 
 
    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;
 
    double probHTRun2(float pTrk,
                      Trk::ParticleHypothesis hypothesis,
                      int TrtPart,
                      int GasType,
                      int StrawLayer,
                      float ZR,
                      float rTrkWire,
                      float Occupancy) const
    {
      return probHTRun2(Gaudi::Hive::currentContext(),
                        pTrk,
                        hypothesis,
                        TrtPart,
                        GasType,
                        StrawLayer,
                        ZR,
                        rTrkWire,
                        Occupancy);
    }
  };
 
  inline const InterfaceID& Trk::ITRT_ElectronPidTool::interfaceID()
    { 
      return IID_ITRT_ElectronPidTool; 
    }
 
} // end of namespace
 
#endif