TruthNtupleTool.h

Go to the documentation of this file.

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

// TruthNtupleTool.h
//   Header file for TruthNtupleTool
// (c) ATLAS Detector software
// Sebastian.Fleischmann@cern.ch
 
#ifndef TRK_TRUTHNTUPLETOOL_H
#define TRK_TRUTHNTUPLETOOL_H
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ITHistSvc.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODEventInfo/EventInfo.h"
 
#include "TrkValInterfaces/ITruthNtupleTool.h"
#include "TrkValInterfaces/ITrackTruthClassifier.h"
#include <vector>
 
 
class TH1D;
class TTree;
 
namespace Trk {
 

 
class TruthNtupleTool : virtual public Trk::ITruthNtupleTool, public AthAlgTool {
public:
 
    // standard AlgToolmethods
    TruthNtupleTool(const std::string&,const std::string&,const IInterface*);
    ~TruthNtupleTool();
 
    // standard Athena methods
    StatusCode initialize();
    StatusCode finalize();
 
    virtual StatusCode writeTruthData ( const std::vector< Trk::ValidationTrackTruthData >& truthData );
 
    virtual unsigned int getNumberOfTreeRecords() const;
 
    virtual StatusCode initBranches(const std::vector<const Trk::ITrackTruthClassifier*>& classifiers,
                                    bool,
                                    const std::vector<std::string>& trackCollectionNames);
 
private:
 
    // jobOptions
    std::string m_ntupleTreeName;       
    std::string m_ntupleFileName;       
    std::vector<double> m_etaBins;
    bool m_fillJets;            
 
    ServiceHandle<ITHistSvc> m_histSvc;
    TTree* m_nt; 
 
    unsigned int m_numberOfTreeEntries;
 
    // ntuple variables
    int         m_runNumber;       
    int         m_eventNumber;     
 
    std::vector< std::vector<unsigned int>* >   m_TrackLinkIndex;
    std::vector< std::vector<float>* >          m_mc_prob;
    std::vector<unsigned int>                   m_classifications;
    // Truth information
    float       m_mc_d0;            
    float       m_mc_z0;            
    float       m_mc_phi0;          
    float       m_mc_theta;         
    float       m_mc_qOverP;        
    float       m_mc_qOverPt;       
    float       m_mc_eta;           
 
    int         m_mc_particleID;    
    int         m_mc_uniqueID;       
    float       m_mc_energy;        
    int         m_mc_jetLinkIndex;  
    float       m_mc_prodR; 
    float       m_mc_prodz; 
 
    // statistics
    std::vector< std::vector<TH1D*> >   m_recoTrackCounts;
    std::vector< std::vector<TH1D*> >   m_truthTrackCounts;
 
    std::vector<const Trk::ITrackTruthClassifier*> m_trackTruthClassifiers;     
 
    SG::ReadHandleKey<xAOD::EventInfo>    m_evt  {this, "EvtInfo", "EventInfo", "EventInfo name"};
};
 
 
} // end of namespace
 
#endif // TRK_TRUTHNTUPLETOOL_H