TruthPreselectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_ALGORITHMS_DROPQSPARTICLEDAUGHTERSTOOL_H
#define ISF_ALGORITHMS_DROPQSPARTICLEDAUGHTERSTOOL_H 1
 
// STL includes
#include <string>
 
// FrameWork includes
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
// ISF_Interfaces includes
#include "ISF_HepMC_Interfaces/IGenParticleFilter.h"
#include "ISF_Interfaces/IGenEventFilter.h"
 
namespace ISF {
 
  class TruthPreselectionTool : public extends<AthAlgTool, IGenEventFilter> {
 
  public:
    TruthPreselectionTool( const std::string& t, const std::string& n, const IInterface* p );

    virtual ~TruthPreselectionTool() = default;

    virtual StatusCode  initialize() override final;
    virtual std::unique_ptr<HepMC::GenEvent> filterGenEvent(const HepMC::GenEvent& inputEvent) const override final;
 
  private:
#ifdef HEPMC3
    bool passesFilters(HepMC::ConstGenParticlePtr& part, const ToolHandleArray<IGenParticleFilter>& filters) const;
    bool identifiedQuasiStableParticleForSim(HepMC::ConstGenParticlePtr& part) const;
    bool hasQuasiStableAncestorParticle(HepMC::ConstGenParticlePtr& part) const;
    bool isPostQuasiStableParticleVertex(HepMC::ConstGenVertexPtr& vtx) const;
#else
    bool passesFilters(HepMC::ConstGenParticlePtr part, const ToolHandleArray<IGenParticleFilter>& filters) const;
    bool identifiedQuasiStableParticleForSim(HepMC::ConstGenParticlePtr part) const;
    bool hasQuasiStableAncestorParticle(HepMC::ConstGenParticlePtr part) const;
    bool isPostQuasiStableParticleVertex(HepMC::ConstGenVertexPtr vtx) const;
#endif

    ToolHandleArray<IGenParticleFilter>  m_genParticleFilters{this, "GenParticleFilters", {}, "Tools for filtering out GenParticles"};    
    ToolHandle<IGenParticleFilter>  m_quasiStableFilter{this, "QuasiStableParticleFilter", "", "Tools for finding quasi-stable particles"};
  };
}
 
#endif //> !ISF_ALGORITHMS_DROPQSPARTICLEDAUGHTERSTOOL_H