TrackValidationNtupleWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// TrackValidationNtupleWriter.h, (c) ATLAS Detector software
 
#ifndef TRK_TRACKVALIDATIONNTUPLEWRITER_H
#define TRK_TRACKVALIDATIONNTUPLEWRITER_H
 
// Gaudi includes
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "TrkTrack/TrackCollection.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkTruthData/TrackTruthCollection.h"
 
class TrackTruth;
class TTree;
 
#include "AtlasHepMC/GenParticle_fwd.h"
 
namespace Trk {
 
class Track;
class ITruthToTrack;         // Produces perigee track parameters from generated parameters
class IGenParticleJetFinder;
class ITrackSelectorTool;
class ITrackValidationNtupleTool;
class IValidationNtupleTool;
class IGenParticleSelector;
class ITruthNtupleTool;
class IJetTruthNtupleTool;
class ITrackTruthClassifier;
class IEventPropertyNtupleTool;
class ValidationTrackTruthData;
class Vertex;
 
class TrackValidationNtupleWriter : public AthAlgorithm {
 public:
  TrackValidationNtupleWriter(const std::string& name, ISvcLocator* pSvcLocator);
  ~TrackValidationNtupleWriter();
  
  StatusCode          initialize();
  StatusCode          execute();
  StatusCode          finalize();
  
 
protected:
  //const Trk::TrackParameters* associateTruthToTrack( const Track*, const TrackCollection*, const TrackTruth*& );
  StatusCode writeTrackData(unsigned int trackColIndex,
                            std::vector<Trk::ValidationTrackTruthData>&  truthData,
                            const Trk::Vertex* primaryVertex = NULL);
 
  StatusCode writeTrackParticleData(unsigned int trackParticleColIndex); 
 
  ToolHandleArray<Trk::ITrackValidationNtupleTool> m_ValidationNtupleTools;
  ToolHandle<Trk::IValidationNtupleTool> m_ValTrkParticleNtupleTool;      
  ToolHandle<Trk::ITruthNtupleTool>         m_truthNtupleTool;
  ToolHandle<Trk::IJetTruthNtupleTool>      m_jetTruthNtupleTool;
  ToolHandleArray<Trk::ITrackTruthClassifier> m_trackTruthClassifierHandles;    
  std::vector<const Trk::ITrackTruthClassifier*> m_trackTruthClassifiers;     
  ToolHandleArray<Trk::IEventPropertyNtupleTool> m_eventPropertyNtupleHandles;
  std::vector<Trk::IEventPropertyNtupleTool*> m_eventPropertyNtupleTools;
  
  std::vector<std::string>  m_inputTrackCollection;     
  std::vector<std::string>  m_inputTrackParticleCollection;     
  std::vector<std::string>  m_trackTruthCollectionName; 
  std::string               m_McEventCollectionName;    
  std::string               m_inputPrimaryVertexCollection;
  std::string               m_ntupleFileName;           
  std::string               m_ntupleDirName;               
 
  ToolHandle<Trk::ITruthToTrack>        m_truthToTrack;    
  ToolHandle<Trk::ITrackSelectorTool>   m_trackSelector;   
  ToolHandle<Trk::IGenParticleSelector> m_genPartSelector; 
  ToolHandle<Trk::IGenParticleJetFinder> m_genJetFinder;   
  bool                      m_useExternalEventLinkTree;    
  bool                      m_doTruth;                     
  bool                      m_doTrackParticle;             
 
  HepMC::GenParticlePtr m_visibleParticleWithoutTruth; 
  std::vector<unsigned int>           m_nTrackTreeRecords;
  std::vector<TTree*>       m_trees;                 
  TTree*                    m_eventLinkTree;         
};
 
} // close of namespace
 
#endif // TRK_TRACKVALIDATIONNTUPLEWRITER_H