EtaPtFilterTool.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// EtaPtFilterTool.h 
// Header file for class EtaPtFilterTool
// Author: S.Binet<binet@cern.ch>
#ifndef MCPARTICLETOOLS_ETAPTFILTERTOOL_H 
#define MCPARTICLETOOLS_ETAPTFILTERTOOL_H 1
 
// STL includes
#include <string>
#include <map>
 
// McParticleTools includes
#include "TruthParticleFilterBaseTool.h"
 
// Forward declaration
 
class EtaPtFilterTool : public TruthParticleFilterBaseTool
{ 
 
 public: 
 
  // Copy constructor: 
 
  EtaPtFilterTool( const std::string& type,
           const std::string& name, 
           const IInterface* parent );
 
  virtual ~EtaPtFilterTool(); 
 
 
  // Non-const methods: 
 
  StatusCode buildMcAod( const McEventCollection* in, McEventCollection* out );
 
  // Protected methods: 
 protected:
  using ParticleMap_t = std::unordered_map<const HepMC::GenParticle*,
                                           HepMC::GenParticlePtr>;
  using VertexMap_t = std::unordered_map<const HepMC::GenVertex*,
                                         HepMC::GenVertexPtr>;
 
  EtaPtFilterTool();
 
  // Const methods: 
 
  StatusCode buildGenEvent( const HepMC::GenEvent* in, HepMC::GenEvent* out );
 
  bool isAccepted( const HepMC::ConstGenParticlePtr& mcPart ) const;
 
  bool isAccepted( const HepMC::ConstGenVertexPtr& vtx ) const;
 
  StatusCode addVertex( const HepMC::ConstGenVertexPtr& srcVtx, HepMC::GenEvent* evt,
                        VertexMap_t& vmap,
                        ParticleMap_t& pmap,
            bool isSignalVertex = false ) const;
 
 
  bool isPartonVertex( const HepMC::ConstGenVertexPtr& vtx ) const;
  bool isFromHardScattering( const HepMC::ConstGenVertexPtr& vtx ) const;
  
  // Non-const methods: 
 
  StatusCode initializeTool();
 
  void setupInnerEtaRegionCuts( Gaudi::Details::PropertyBase& innerEtaRegionCuts );
 
  void setupOuterEtaRegionCuts( Gaudi::Details::PropertyBase& outerEtaRegionCuts );
 
  // Protected data: 
 protected: 
 
  DoubleArrayProperty m_innerEtaRegionCuts;
 
  DoubleArrayProperty m_outerEtaRegionCuts;
 
  BooleanProperty m_onlyGenerator;
 
  BooleanProperty m_butKeepAllGeneratorStable;
 
  BooleanProperty m_keepDocumentaries;
 
  BooleanProperty m_keepAllLeptons;
 
  
  /* For the isolation */
  std::map<int,double> m_isol_energies;
 
}; 
 
 
#endif //> MCPARTICLETOOLS_ETAPTFILTERTOOL_H