TrackRetriever.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JIVEXML_TRACKRETRIEVER_H
#define JIVEXML_TRACKRETRIEVER_H
 
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "JiveXML/IDataRetriever.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkToolInterfaces/ITrackSummaryTool.h"
#include "TrkToolInterfaces/IResidualPullCalculator.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkTruthData/TrackTruthCollection.h"
 
//Forward declarations
class AtlasDetectorID;
 
namespace JiveXML{
 
  class TrackRetriever : virtual public IDataRetriever,
             public AthAlgTool {
 
  public:
 
    TrackRetriever(const std::string& type,const std::string& name,const IInterface* parent);
 
    virtual StatusCode retrieve(ToolHandle<IFormatTool> &FormatTool);
    const DataMap getData(const TrackCollection* trackCollection, const std::string &collectionName);
    virtual std::string dataTypeName() const { return m_typeName; };
 
    StatusCode initialize();
 
  private:
    const std::string m_typeName = "Track";
 
    SG::ReadHandleKeyArray<TrackCollection> m_keys {this,"TrackCollections", {"CombinedInDetTracks","DisappearingTracks","ExtrapolatedStauTracks","ExtraPolatedMuonsLRTTracks","MSOnlyExtraPolatedMuonsLRTTrackParticlesTracks","ResolvedForwardTracks","CombinedMuonsLRTTracks","CombinedStauTracks","MuonSpectrometerTracks","MSOnlyExtrapolatedTracks","ExtrapolatedMuonTracks","CombinedMuonTracks","GSFTracks"}, "Name of the Trk TrackCollection containers that will be written out to the JiveXML file"};
    SG::ReadHandleKeyArray<TrackTruthCollection> m_TrackTruthCollections {this,"TruthCollections", {"CombinedInDetTracksTruthCollection"}, "Name of the Truth Track containers that will be written out to the JiveXML file"};
    
    Gaudi::Property<bool> m_doWriteResiduals {this,"DoWriteResiduals", true,"Whether to write TrackResiduals"};
    Gaudi::Property<bool> m_doHitsSorting {this, "DoHitsSorting", false, "Whether to perform hits TrackStateOnSurfaces (TSoS) sorting"};
    Gaudi::Property<bool> m_doHitsDetails {this, "DoHitsDetails", true, "Whether to write hits TrackStateOnSurfaces (TSoS) details"};
    Gaudi::Property<bool> m_isMC {this, "isMC", false, "Whether file is MC"};
 
    ToolHandle<Trk::IResidualPullCalculator> m_residualPullCalculator {this, "ResidualPullCalculator","Trk::ResidualPullCalculator/ResidualPullCalculator", "ToolHandle to ResidualPullCaclulator" };
    ToolHandle< Trk::ITrackSummaryTool > m_trackSumTool{this, "TrackSummaryTool", "Trk::TrackSummaryTool/InDetTrackSummaryTool", "Track Summary Tool for number of Pix/SCT/TRT hits"};
 
    const AtlasDetectorID* m_idHelper{};
 
  };
 
}
#endif