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);
    const std::vector<std::string> getKeys();
    virtual std::string dataTypeName() const { return m_typeName; };
 
    StatusCode initialize();
 
  private:
    const std::string m_typeName = "Track";
 
    Gaudi::Property<std::string> m_priorityKey {this,"PriorityTrackCollection","CombinedInDetTracks", "First track collections to retrieve, shown as default in Atlantis"};
    Gaudi::Property<std::vector<std::string>> m_otherKeys {this, "OtherTrackCollections", {}, "Track collections to retrieve, all if empty"};
    Gaudi::Property<bool> m_doWriteHLT {this, "DoWriteHLT", false, "Write out other collections that have HLT in the name"};
    Gaudi::Property<bool> m_doWriteAllCollections {this, "DoWriteAllCollections", false, "Write out all Track collections"};   
    Gaudi::Property<std::string> m_TrackTruthCollection {this, "TrackTruthColName", "TrackTruthCollection", "Track collection from which to retrieve the truth associations for the priority track collection"};
    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"};
 
    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