PseudoTrackSelector.h

Go to the documentation of this file.

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

// PseudoTrackSelector.h
#ifndef DERIVATIONFRAMEWORK_PseudoTrackSelector_H
#define DERIVATIONFRAMEWORK_PseudoTrackSelector_H
 
#include <string>
#include <map>
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "InDetTrackSystematicsTools/IInDetTrackTruthOriginTool.h"
 
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackStateValidationContainer.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "ExpressionEvaluation/ExpressionParserUser.h"
 
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadHandleKeyArray.h"
 
namespace DerivationFramework {

  class PseudoTrackSelector : public extends<AthAlgTool, IAugmentationTool>
    {
    public:

 
      using base_class::base_class;
 
      virtual StatusCode initialize() override final;
      virtual StatusCode addBranches(const EventContext& ctx) const override final;
 
    protected:

      SG::ReadHandleKey<xAOD::TrackParticleContainer>  m_in_recoTrackParticleLocation{this, "RecoTrackParticleLocation", ""};       
      SG::ReadHandleKey<xAOD::TrackParticleContainer>  m_in_pseudoTrackParticleLocation{this, "PseudoTrackParticleLocation", ""};     
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoReplacedWithPseudo{this, "OutputRecoReplacedWithPseudo", ""};         
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoReplacedWithPseudoFromB{this, "OutputRecoReplacedWithPseudoFromB", ""};    
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoReplacedWithPseudoNotFromB{this, "OutputRecoReplacedWithPseudoNotFromB", ""}; 
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoPlusPseudo{this, "OutputRecoPlusPseudo", ""};                 
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoPlusPseudoFromB{this, "OutputRecoPlusPseudoFromB", ""};            
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoPlusPseudoNotFromB{this, "OutputRecoPlusPseudoNotFromB", ""};         
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoNoFakes{this, "OutputRecoNoFakes", ""};                    
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoNoFakesFromB{this, "OutputRecoNoFakesFromB", ""};               
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoNoFakesNotFromB{this, "OutputRecoNoFakesNotFromB", ""};            



      void fillRecoReplacedWithPseudo(const xAOD::TrackParticleContainer* recoTrackParticleCol,
                                      const xAOD::TrackParticleContainer* pseudoTrackParticleCol,
                                      xAOD::TrackParticleContainer* outputCol,
                                      bool onlyFromB = false,
                                      bool onlyNotFromB = false) const;
      void fillRecoPlusPseudo(const xAOD::TrackParticleContainer* recoTrackParticleCol,
                              const xAOD::TrackParticleContainer* pseudoTrackParticleCol,
                              xAOD::TrackParticleContainer* outputCol,
                              bool onlyFromB = false,
                              bool onlyNotFromB = false) const;
      void fillRecoNoFakes(const xAOD::TrackParticleContainer* recoTrackParticleCol,
                           xAOD::TrackParticleContainer* outputCol,
                           bool onlyFromB = false,
                           bool onlyNotFromB = false) const;
      // Get truth particle associated to a given track particle
      static const xAOD::TruthParticle* getTruth( const xAOD::TrackParticle* track ) ;
 
    private:
        // convience types
        typedef ElementLink<xAOD::TruthParticleContainer> TruthLink;
        // Track origin tool
        ToolHandle< InDet::IInDetTrackTruthOriginTool > m_trackOriginTool{this, "trackTruthOriginTool", "InDet::InDetTrackTruthOriginTool", "truth track origin tool"};
    };
}
#endif // DERIVATIONFRAMEWORK_PseudoTrackSelector_H