TrackPositionNtupleHelper.h

Go to the documentation of this file.

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

// TrackPositionNtupleHelper.h
//   Header file for TrackPositionNtupleHelper
// (c) ATLAS Detector software
// Anna.Franckowiak@cern.ch, Sebastian.Fleischmann@cern.ch
 
#ifndef TRK_TRACKPOSITIONNTUPLEHELPER_H
#define TRK_TRACKPOSITIONNTUPLEHELPER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "TrkValInterfaces/IValidationNtupleHelperTool.h"
#include "TrkParameters/TrackParameters.h"
 
class INTupleSvc;
class TTree;
 
namespace Trk {
 
class MeasurementBase;

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

    virtual StatusCode fillMeasurementData (
        const Trk::MeasurementBase*,
        const Trk::TrackParameters*,
        //const NTuple::Item<long>& numberOfHits,
        //const std::bitset<8>& detectorType
        const int& detectorType,
        const bool& isOutlier );

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

    virtual StatusCode addNtupleItems (
        TTree* tree,
        const int& detectorType );

    virtual StatusCode resetVariables (
        const int& detectorType );
 
private:
 
    bool      m_trkParametersWarning;
 
    std::vector<float>*         m_trkX;  
    std::vector<float>*         m_trkY;  
    std::vector<float>*         m_trkZ;  
    std::vector<float>* m_trkMomentumX;  
    std::vector<float>* m_trkMomentumY;  
    std::vector<float>* m_trkMomentumZ;  
    std::vector<float>*      m_trkLoc1;  
    std::vector<float>*      m_trkLoc2;  
 
};
 
} // end of namespace
 
#endif // TRK_TRACKPOSITIONNTUPLEHELPER_H