da/d4b/MeasurementVectorNtupleTool_8h_source.html

/*

  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