HitPositionNtupleHelper.h

Go to the documentation of this file.

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

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

 
class HitPositionNtupleHelper : virtual public Trk::IValidationNtupleHelperTool, public AthAlgTool {
public:
 
    // standard AlgToolmethods
    HitPositionNtupleHelper(const std::string&,const std::string&,const IInterface*);
    ~HitPositionNtupleHelper();
 
    // 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:
 
    // --- member variables for ntuple items ---
    std::vector<float>*             m_hitX; 
    std::vector<float>*             m_hitY; 
    std::vector<float>*             m_hitZ; 
    std::vector<float>*        m_hitRadius; 
    std::vector<float>*           m_hitPhi; 
    std::vector<float>*          m_hitLoc1; 
    std::vector<float>*          m_hitLoc2; 
    std::vector<float>*      m_hitSurfaceX; 
    std::vector<float>*      m_hitSurfaceY; 
    std::vector<float>*      m_hitSurfaceZ; 
    std::vector<float>* m_hitSurfaceRadius; 
    std::vector<float>*    m_hitSurfacePhi; 
    std::vector<float>*  m_hitSurfaceTheta; 
};
 
 
} // end of namespace
 
#endif // TRK_HITPOSITIONNTUPLEHELPER_H