PseudoTrackSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// PseudoTrackSelector.h, (c) ATLAS Detector software
#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 AthAlgTool, public IAugmentationTool
    {
    public:      
 
 
      PseudoTrackSelector(const std::string& t, const std::string& n, const IInterface* p);
      virtual ~PseudoTrackSelector() {}
      StatusCode initialize();
      StatusCode execute();
      StatusCode finalize();
      virtual StatusCode addBranches() const;
 
    protected:
 
      SG::ReadHandleKey<xAOD::TrackParticleContainer>  m_in_recoTrackParticleLocation;       
      SG::ReadHandleKey<xAOD::TrackParticleContainer>  m_in_pseudoTrackParticleLocation;     
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoReplacedWithPseudo;         
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoReplacedWithPseudoFromB;    
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoReplacedWithPseudoNotFromB; 
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoPlusPseudo;                 
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoPlusPseudoFromB;            
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoPlusPseudoNotFromB;         
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoNoFakes;                    
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoNoFakesFromB;               
      SG::WriteHandleKey<xAOD::TrackParticleContainer> m_out_recoNoFakesNotFromB;            
 
 
      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