DirectTrackNtupleWriterTool.h

Go to the documentation of this file.

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

// DirectTrackNtupleWriterTool.h
//   Header file for DirectTrackNtupleWriterTool
// (c) ATLAS Detector software
// Sebastian.Fleischmann@cern.ch
 
#ifndef TRK_DIRECTTRACKVALIDATIONNTUPLETOOL_H
#define TRK_DIRECTTRACKVALIDATIONNTUPLETOOL_H
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkValInterfaces/IDirectTrackNtupleWriter.h"
 
class TTree;
 
namespace Trk {
 
class IExtrapolator;
class ITrackValidationNtupleTool;
class Track;
 
 

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

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

    virtual StatusCode writeTrackParticleData ( const Trk::TrackParticleBase& );
 
    
private:
    StatusCode writeRecord(TTree* tree);
 
    ToolHandleArray<Trk::ITrackValidationNtupleTool> m_ValidationNtupleTools;  
    ToolHandle<Trk::IExtrapolator>                   m_extrapolatorHandle;     
    std::string m_ntupleFileName;       
    std::string m_ntupleDirName;        
    std::string m_ntupleTreeName;       
    TTree* m_nt; 
    bool              m_doTruth;        
};
 
 
} // end of namespace
 
#endif // TRK_DIRECTTRACKVALIDATIONNTUPLETOOL_H