MuonTrackStatisticsAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONTRACKSTATISTICSALG_MUONTRACKSTATISTICSALG_H
#define MUONTRACKSTATISTICSALG_MUONTRACKSTATISTICSALG_H
 
#include <fstream>
#include <string>
 
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkTruthData/DetailedTrackTruthCollection.h"
 
class MuonTrackStatisticsTool;
 
class MuonTrackStatisticsAlg : public AthAlgorithm {
public:
    // Algorithm Constructor
    MuonTrackStatisticsAlg(const std::string &name, ISvcLocator *pSvcLocator);
    virtual ~MuonTrackStatisticsAlg();
 
    // Gaudi algorithm hooks
    virtual StatusCode initialize() override;
    virtual StatusCode execute() override;
    virtual StatusCode finalize() override;
 
    virtual unsigned int cardinality() const override final { return 1;}
 
private:
    ToolHandle<MuonTrackStatisticsTool> m_statisticsTool{this, "StatTool", "" };
 
 

    std::string m_fileName;

    std::ofstream m_fileOutput;
 
    bool m_writeToFile;
    bool m_doTruth;
 
    SG::ReadHandleKeyArray<TrackCollection> m_trackKeys{
        this, "TrackLocationList", {"MuonSpectrometerTracks"}, "track collections to track"};
    SG::ReadHandleKeyArray<DetailedTrackTruthCollection> m_truthKeys{
        this, "TruthTrackLocationList", {"MuonSpectrometerTracksTruth"}, "truth track collections"};
 
    void storeTruthTracks(void);
};
 
#endif  // MUONTRACKSTATISTICSALG_MUONTRACKSTATISTICSALG_H