EventToTrackLinkNtupleTool.h

Go to the documentation of this file.

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

// EventToTrackLinkNtupleTool.h
//   Header file for EventToTrackLinkNtupleTool
// (c) ATLAS Detector software
// Wolfgang.Liebig -at- cern.ch
 
#ifndef TRK_EVENTTRACKLINKNTUPLETOOL_H
#define TRK_EVENTTRACKLINKNTUPLETOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "GaudiKernel/ToolHandle.h"
#include "TrkValInterfaces/IEventPropertyNtupleTool.h"
#include <vector>
class TTree;

 
namespace Trk {
 
  class EventToTrackLinkNtupleTool : virtual public IEventPropertyNtupleTool, public AthAlgTool {
  public:
 
    // standard AlgToolmethods
    EventToTrackLinkNtupleTool(const std::string&,const std::string&,const IInterface*);
    ~EventToTrackLinkNtupleTool();
 
    // standard Athena methods
    StatusCode initialize();
    StatusCode finalize();

    virtual void registerTrackCollections( const std::vector<std::string>& collections, bool doTruth);
    virtual void setTrackTreeIndices( unsigned int, int, int);
    virtual void setGenParticleTreeIndices( int, int);

    virtual StatusCode addNtupleItems ( TTree*, const std::string &);

    virtual StatusCode fillEventData ( );

    virtual StatusCode  resetVariables ( );
       
   inline  virtual bool isTrackLinkTool( ) const;
   inline  virtual bool isTrkParticleLinkTool( ) const;
   inline  virtual bool isTrkParticleTrigLinkTool( ) const;
   inline  virtual bool isInDetTrackTrigLinkTool( ) const;
   inline  virtual bool isEvtPropertyTool( ) const;
 
  private:
    StatusCode checkCollectionType() const;
 
    std::string                       m_collectionType;
    TTree*                    m_eventLinkTree;         
    bool                      m_doTruth;               
    std::vector<std::string>  m_trackCollections;      
 
    // --- ntuple items --- 
    std::vector<int>  m_trackIndexBegin;       
    std::vector<int>  m_nTracksPerEvent;       
    int               m_genParticleIndexBegin; 
    int               m_nGenParticlesPerEvent; 
    
    static const std::string s_trackTypeName;            
    static const std::string s_trkParticleTypeName;      
    static const std::string s_trkParticleTrigTypeName;  
    static const std::string s_inDetTrackTrigTypeName;   
  };
 
}
 
 
inline bool Trk::EventToTrackLinkNtupleTool::isTrackLinkTool( ) const {
                                                                     if( m_collectionType == s_trackTypeName) return true;
                                             return false;
                                                                            }
   
inline bool Trk::EventToTrackLinkNtupleTool::isTrkParticleLinkTool( ) const {
                                                                       if( m_collectionType == s_trkParticleTypeName) return true;
                                       return false;
                                                                            }
    
inline bool Trk::EventToTrackLinkNtupleTool::isTrkParticleTrigLinkTool( ) const{
                                                                         if( m_collectionType == s_trkParticleTrigTypeName) return true;
                                     return false;
                                                                               }
 
inline bool Trk::EventToTrackLinkNtupleTool::isInDetTrackTrigLinkTool( ) const{
                                                                         if( m_collectionType == s_inDetTrackTrigTypeName) return true;
                                     return false;
                                                                               }
 
inline bool Trk::EventToTrackLinkNtupleTool::isEvtPropertyTool( ) const {
                                                                      return false;
                                                                        } 
 
 
#endif // TRK_EVENTTRACKLINKNTUPLETOOL_H