BasicValTrkParticleNtupleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef Trk_BasicValTrkParticleNtupleTool_h
#define Trk_BasicValTrkParticleNtupleTool_h
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkValInterfaces/IValidationNtupleTool.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODEventInfo/EventInfo.h"
#include <vector>
 
class TTree;
 
namespace Trk {
class TrackSummary;
class TrackParticleBase;

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

    virtual StatusCode addNtupleItems (
        TTree* tree );

    virtual StatusCode writeTrackData (
        const Trk::Track&,
        const int iterationIndex,
        const unsigned int fitStatCode );

    virtual StatusCode writeTrackParticleData (
        const Trk::TrackParticleBase& );
    
    virtual StatusCode fillTrackData (
        const Trk::Track&,
        const int iterationIndex,
        const unsigned int fitStatCode );

    virtual StatusCode fillTrackParticleData (
        const Trk::TrackParticleBase&);
        
    virtual StatusCode fillTrackParameter (
        const Trk::TrackParameters*,
        const int iterationIndex );
    
    virtual StatusCode fillMeasurementData (
        const Trk::MeasurementBase*,
        const Trk::TrackParameters* );
        
    virtual StatusCode fillOutlierData (
        const Trk::MeasurementBase*,
        const Trk::TrackParameters*,
        const Trk::FitQualityOnSurface* );

    virtual StatusCode fillHoleData (
        const Trk::TrackStateOnSurface&);

    virtual StatusCode fillTrackTruthData ( const TrackParameters*&,
                                            const TrackTruth&, 
                                            const int truthIndex = -1 );

    virtual StatusCode writeRecord( TTree* tree );
private:
    StatusCode fillTrkParticlePerigee(const Trk::Perigee* perigee);
    StatusCode fillTrkParticleSummary (const Trk::TrackSummary* summary);
    StatusCode fillFitQualityData (const Trk::FitQuality* fitQuality);
    void resetVariables();
 
    // jobOptions
    bool m_bookNewNtuple;               
    std::string m_ntupleFileName;       
    std::string m_ntupleDirName;        
    std::string m_ntupleTreeName;       
 
    TTree* m_nt; 
    
    int             m_TrackIDcounter;  
    unsigned int    m_lastEventNumber; 
 
    // --- ntuple variables, doxygen comments are also basis for wiki texts --- 
    int              m_runNumber;   
    int              m_eventNumber; 
    unsigned char        m_TrackID;     
    
    //----  extraction from  Track Perigee -----
    float       m_Rec_d0;      
    float       m_Rec_z0;      
    float       m_Rec_phi0;    
    float       m_Rec_eta;     
    float       m_Rec_qOverP;  
 
    float       m_errord0;     
    float       m_errorz0;     
    float       m_errorphi0;   
    float       m_errortheta0; 
    float       m_errorqoverp; 
    
    //----  extraction from  Fit Quality -----
    float             m_chi2;                          
 
    //----  extraction from TrackSummary -----
    unsigned char         m_numberOfPixelHits;             
    unsigned char         m_numberOfContribPixelLayers;    
    unsigned char         m_numberOfPixelHoles;            
    unsigned char         m_numberOfPixelDeadSensors;      
    unsigned char         m_numberOfSCTHits;               
    unsigned char         m_numberOfSCTHoles;              
    unsigned char         m_numberOfSCTDeadSensors;        
    unsigned char         m_numberOfTRTHits;               
    unsigned char         m_numberOfTRTHoles;              
    unsigned char         m_numberOfTRTDeadStraws;         
    unsigned char         m_numberOfTRTHighThresholdHits;  
 
    unsigned long    m_idHitPattern;                  
     
    static const float  s_errorEntry;        
 
    SG::ReadHandleKey<xAOD::EventInfo>    m_evt  {this, "EvtInfo", "EventInfo", "EventInfo name"};
};
 
 
} // end of namespace Trk
 
#endif // BasicsValTrkParticleNtupleTool_h