MeasurementVectorNtupleTool.h

Go to the documentation of this file.

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

// PerigeeParametersNtupleTool.h
//   Header file for PerigeeParametersNtupleTool
// (c) ATLAS Detector software
// Sebastian.Fleischmann -at- cern.ch, Wolfgang.Liebig -at- cern.ch
#ifndef TRK_MEASUREMENTVECTORNTUPLETOOL_H
#define TRK_MEASUREMENTVECTORNTUPLETOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkValInterfaces/ITrackValidationNtupleTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkParameters/TrackParameters.h"
 
class AtlasDetectorID;
 
namespace Trk {
 
  class IUpdator;
  class ITrackHoleSearchTool;
  class IValidationNtupleHelperTool;
  class IResidualPullCalculator;
  class MeasurementBase;
  class MeasurementTypeID;

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

    virtual StatusCode addNtupleItems ( TTree* tree );

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

    virtual StatusCode fillTrackParticleData ( const Trk::TrackParticleBase& );

    virtual StatusCode fillTrackTruthData ( const TrackParameters*&,
                                            const TrackTruth&,
                                            const int);

    virtual void resetVariables( );
 
private:

    StatusCode fillMeasurementData (
                    const Trk::MeasurementBase*,
                    const Trk::TrackParameters*,
                    const bool& isOutlier,
                    const int& detectorType);
    StatusCode callHelperTools (
                    const Trk::MeasurementBase*,
                    const Trk::TrackParameters*,
                    const bool& isOutlier,
                    const int& detectorType,
                    const int& nCurrentHit);
    virtual StatusCode fillHoleData (
        const Trk::TrackStateOnSurface&);
 
    const AtlasDetectorID*  m_idHelper;   
    MeasurementTypeID*      m_detTypeHelper;
 
    ToolHandle<Trk::IUpdator>                   m_updatorHandle;            
    Trk::IUpdator*                              m_updator;                  
    
    ToolHandle<Trk::IResidualPullCalculator>    m_residualPullCalculator;   
    ToolHandle<Trk::ITrackHoleSearchTool>       m_holeSearchTool;           
    ToolHandleArray<IValidationNtupleHelperTool> m_PixelNtupleHelperToolHandles;    
    ToolHandleArray<IValidationNtupleHelperTool> m_SCTNtupleHelperToolHandles;      
    ToolHandleArray<IValidationNtupleHelperTool> m_TRTNtupleHelperToolHandles;      
    ToolHandleArray<IValidationNtupleHelperTool> m_MDTNtupleHelperToolHandles;      
    ToolHandleArray<IValidationNtupleHelperTool> m_CSCNtupleHelperToolHandles;      
    ToolHandleArray<IValidationNtupleHelperTool> m_RPCNtupleHelperToolHandles;      
    ToolHandleArray<IValidationNtupleHelperTool> m_TGCNtupleHelperToolHandles;      
    ToolHandleArray<IValidationNtupleHelperTool> m_GeneralNtupleHelperToolHandles;  
 
    bool m_useROTwithMaxAssgnProb;      
    bool m_ignoreMissTrkCov;          
    std::vector<IValidationNtupleHelperTool*> m_PixelHelperTools;     
    std::vector<IValidationNtupleHelperTool*> m_SCTHelperTools;       
    std::vector<IValidationNtupleHelperTool*> m_TRTHelperTools;       
    std::vector<IValidationNtupleHelperTool*> m_MDTHelperTools;       
    std::vector<IValidationNtupleHelperTool*> m_CSCHelperTools;       
    std::vector<IValidationNtupleHelperTool*> m_RPCHelperTools;       
    std::vector<IValidationNtupleHelperTool*> m_TGCHelperTools;       
    std::vector<IValidationNtupleHelperTool*> m_GeneralHelperTools;   
    // DNA     ToolHandle< ITrackBreakpointAnalyser >  m_brempointAnalyser;
    bool      m_pullWarning;    
    bool      m_UpdatorWarning; 
    bool      m_trkParametersWarning; 
    bool              m_doTruth;        
    bool              m_doHoleSearch;   
    std::map<const IValidationNtupleHelperTool*, bool>  m_helperToolWarning;
 
    int                 m_isUnbiased;  
    std::vector<float>* m_residualLocX; 
    std::vector<float>* m_residualLocY; 
    std::vector<float>* m_pullLocX; 
    std::vector<float>* m_pullLocY; 
    std::vector<int>*   m_DetectorType; 
    std::vector<int>*   m_isOutlier; 
 
    int         m_nPixelHits; 
    int         m_nSCTHits;   
    int         m_nTRTHits;   
    int         m_nMDTHits;   
    int         m_nCSCHits;   
    int         m_nRPCHits;   
    int         m_nTGCHits;   
 
    std::vector<int>*   m_pixelHitIndex; 
    std::vector<int>*   m_sctHitIndex; 
    std::vector<int>*   m_trtHitIndex; 
    std::vector<int>*   m_mdtHitIndex; 
    std::vector<int>*   m_cscHitIndex; 
    std::vector<int>*   m_rpcHitIndex; 
    std::vector<int>*   m_tgcHitIndex; 
    
    // hole data:
    int         m_nHoles;                   
    int         m_nPixelHoles;                   
    int         m_nSctHoles;                   
    std::vector<int>*   m_HoleDetectorType; 
 
 
};
 
 
} // end of namespace
 
 
#endif // TRK_PERIGEEPARAMETERSNTUPLETOOL_H