InDetTrackPerfMonTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_INDETTRACKPERFMONTOOL_H
#define INDETTRACKPERFMON_INDETTRACKPERFMONTOOL_H


#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Service.h"

#include "AthenaMonitoring/ManagedMonitorToolBase.h"
#include "TrigDecisionTool/TrigDecisionTool.h"

#include "xAODCore/BaseContainer.h"
#include "xAODCore/AuxContainerBase.h"

#include "InDetTrackPerfMon/ITrackAnalysisDefinitionSvc.h"
#include "TrackAnalysisCollections.h"
#include "RoiSelectionTool.h"
#include "TrackRoiSelectionTool.h"
#include "VertexRoiSelectionTool.h"
#include "ITrackSelectionTool.h"
#include "IVertexSelectionTool.h"
#include "ITrackMatchingTool.h"
#include "TrackAnalysisPlotsMgr.h"
#include "TrackAnalysisInfoWriteTool.h"

#include <string>
#include <vector>
 
namespace IDTPM {
  
class InDetTrackPerfMonTool : public ManagedMonitorToolBase {
 
public :

    InDetTrackPerfMonTool( const std::string& type, const std::string& name, const IInterface* parent );

    virtual ~InDetTrackPerfMonTool();
 
    virtual StatusCode initialize();
    virtual StatusCode bookHistograms();
    virtual StatusCode fillHistograms(const EventContext& ctx);
    virtual StatusCode procHistograms();
 
private :

    InDetTrackPerfMonTool();

    StatusCode loadCollections( IDTPM::TrackAnalysisCollections& trkAnaColls );

    SG::ReadHandleKey< xAOD::TrackParticleContainer > m_offlineTrkParticleName {
        this, "OfflineTrkParticleContainerName", "InDetTrackParticles", "Name of container of offline tracks" };

    SG::ReadHandleKey< xAOD::TrackParticleContainer > m_triggerTrkParticleName {
        this, "TriggerTrkParticleContainerName", "", "Name of container of trigger tracks" };

    SG::ReadHandleKey< xAOD::TruthParticleContainer > m_truthParticleName {
        this, "TruthParticleContainerName",  "TruthParticles", "Name of container of TruthParticles" };

    SG::ReadHandleKey< xAOD::TruthEventContainer > m_truthEventName {
        this, "TruthEvents", "TruthEvents", "Name of the truth events container probably either TruthEvent or TruthEvents" };

    SG::ReadHandleKey< xAOD::TruthPileupEventContainer > m_truthPileUpEventName {
        this, "TruthPileupEvents", "TruthPileupEvents", "Name of the truth pileup events container probably TruthPileupEvent(s)" };

    SG::ReadHandleKey< xAOD::EventInfo > m_eventInfoContainerName {
        this, "EventInfoContainerName", "EventInfo", "event info" };

    SG::ReadHandleKey< xAOD::VertexContainer > m_offlineVertexContainerName {
        this, "OfflineVertexContainerName", "PrimaryVertices", "" };

    SG::ReadHandleKey< xAOD::VertexContainer > m_triggerVertexContainerName {
        this, "TriggerVertexContainerName", "", "" };

    SG::ReadHandleKey< xAOD::TruthVertexContainer > m_truthVertexContainerName {
        this, "TruthVertexContainerName",  "TruthVertices", "" };

    SG::WriteHandleKey< xAOD::BaseContainer > m_trkAnaInfoKey {
        this, "TrkAnaInfoKey", "TrackAnalysisInfo", "Dedicated TrackAnalysis Info written out" };

 
    ToolHandle< IDTPM::ITrackSelectionTool > m_trackQualitySelectionTool {
        this, "TrackQualitySelectionTool", "IDTPM::InDetTrackPerfMon/ITrackSelectionTool", "Wrapper-tool to perform general quality-based track(truth) selection" };
 
    ToolHandle< IDTPM::IVertexSelectionTool > m_vertexQualitySelectionTool {
        this, "VertexQualitySelectionTool", "IDTPM::InDetTrackPerfMon/IVertexSelectionTool", "Wrapper-tool to perform general quality-based (truth) vertex selection" };
 
    ToolHandle< IDTPM::RoiSelectionTool > m_roiSelectionTool {
        this, "RoiSelectionTool", "IDTPM::InDetTrackPerfMon/RoiSelectionTool", "Tool to retrieve and select RoIs" };
 
    ToolHandle< IDTPM::TrackRoiSelectionTool > m_trackRoiSelectionTool {
        this, "TrackRoiSelectionTool", "IDTPM::InDetTrackPerfMon/TrackRoiSelectionTool", "Tool to select track within a RoI" };
 
    ToolHandle< IDTPM::VertexRoiSelectionTool > m_vertexRoiSelectionTool {
        this, "VertexRoiSelectionTool", "IDTPM::InDetTrackPerfMon/VertexRoiSelectionTool", "Tool to select vertices within a RoI" };
 
    ToolHandle< IDTPM::ITrackMatchingTool > m_trackMatchingTool {
        this, "TrackMatchingTool", "IDTPM::InDetTrackPerfMon/ITrackMatchingTool", "Tool to match test to reference tracks and viceversa" };
 
    ToolHandle< IDTPM::TrackAnalysisInfoWriteTool > m_trkAnaInfoWriteTool {
        this, "TrackAnalysisInfoWriteTool", "IDTPM::InDetTrackPerfMon/TrackAnalysisInfoWriteTool", "Tool to write TrackAnalysisInfo to StoreGate" };
 
    StringProperty m_anaTag{ this, "AnaTag", "", "Track analysis tag" }; 
 
    BooleanProperty m_doMatch{ this, "doMatch", false, "Enable TrackMatchingTool" };
 
    BooleanProperty m_writeOut{ this, "writeOut", false, "Write TrkAnaInfo Collection to AOD_IDTPM" };

    SmartIF<IDTPM::ITrackAnalysisDefinitionSvc> m_trkAnaDefSvc;

    std::vector< std::unique_ptr< IDTPM::TrackAnalysisPlotsMgr > >  m_trkAnaPlotsMgrVec;
};
 
}
  
#endif