AthTruthSelectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETPHYSVALMONITORING_ATHTRUTHSELECTIONTOOL_H
#define INDETPHYSVALMONITORING_ATHTRUTHSELECTIONTOOL_H
 
// STL includes
#include <string>
#include "TrkTruthTrackInterfaces/IAthSelectionTool.h"
#include "xAODTruth/TruthParticle.h" // typedef, can't fwd declare
#include "AthenaBaseComps/AthAlgTool.h"
#include "InDetPhysValMonitoring/CutFlow.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkSurfaces/CylinderSurface.h"
#include "TrkSurfaces/DiscSurface.h"
#include "TrkExInterfaces/IExtrapolator.h"
 
 
class AthTruthSelectionTool: virtual public IAthSelectionTool, public AthAlgTool {
public:
  AthTruthSelectionTool(const std::string& type, const std::string& name, const IInterface* parent);
  virtual ~AthTruthSelectionTool () {
    /*nop*/
  };
  StatusCode initialize() final;
  StatusCode finalize() final;
 
  virtual IAthSelectionTool::CutResult
  accept(const xAOD::IParticle* particle) const final;
 
  virtual IAthSelectionTool::CutResult
  testAllCuts(const xAOD::IParticle * p, std::vector<unsigned int> &counter) const final;
 
  unsigned int nCuts() const final {
    return m_cutList.size();
  }
 
  std::vector<std::string> names() const final;
 
private:
  CutList<xAOD::TruthParticle> m_cutList;
  // Cut values;
  float m_maxEta;
  float m_maxPt;
  float m_minPt;
  bool m_requireOnlyPrimary;
  bool m_requireCharged;
  int m_selectedCharge;
  bool m_requireStable;
  int m_requireSiHit;
  // max decay radius for secondaries [mm];
  // set to within (Run2) pixel by default
  double m_maxProdVertRadius;
  int m_pdgId;
  bool m_grandparent;
  bool m_poselectronfromgamma;
  std::vector<unsigned int> m_counters;
  std::vector<int> m_ancestors;
  
  /* \defgroup Selection on extrapolated particle to cylinder or disk surface
   * @{
   */
  float m_radiusCylinder; 
  float m_minZCylinder; 
  float m_maxZCylinder; 
  float m_zDisc; 
  float m_minRadiusDisc; 
  float m_maxRadiusDisc; 
 
  //cache surfaces
  std::unique_ptr<Trk::CylinderSurface> m_cylinder;
  std::unique_ptr<Trk::DiscSurface>     m_disc1;
  std::unique_ptr<Trk::DiscSurface>     m_disc2;
 
 
  /* @} */
 
  PublicToolHandle<Trk::IExtrapolator> m_extrapolator
     {this,"Extrapolator","Trk::Extrapolator/AtlasExtrapolator",""};
 
};
 
 
#endif