EgammaMonitoring.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef egammaValidation_EgammaMonitoring_H
#define egammaValidation_EgammaMonitoring_H
 
#include "xAODEventInfo/EventInfo.h"
#include "xAODTracking/TrackParticle.h"
 
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEgamma/PhotonContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTracking/TrackParticleContainer.h"
 
#include "AsgTools/ToolHandle.h"
#include "AthenaBaseComps/AthAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "EgammaAnalysisInterfaces/IAsgElectronLikelihoodTool.h"
#include "EgammaAnalysisInterfaces/IAsgPhotonIsEMSelector.h"
#include "IsolationSelection/IIsolationSelectionTool.h"
#include "MCTruthClassifier/IMCTruthClassifier.h"
 
#include "EfficiencyPlot.h"
#include "IHistograms.h"
#include "DiObjectHistograms.h"
#include "RecoElectronHistograms.h"
#include "RecoPhotonHistograms.h"
#include "RecoClusterHistograms.h"
#include "ShowerShapesHistograms.h"
#include "TrackHistograms.h"
#include "ClusterHistograms.h"
#include "TruthElectronHistograms.h"
#include "TruthPhotonHistograms.h"
#include "WidthPlot.h"
#include "IsolationHistograms.h"
 
#include <memory>
#include <string>
#include <vector>
 
class TH1D;
 
class ITHistSvc;
 
class EgammaMonitoring : public AthAlgorithm
{
public:
  ITHistSvc* rootHistSvc = nullptr;
 
  // samples from simulation
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterAll;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> cluster10GeV;
 
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterPromptAll;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterPrompt10GeV;
 
  // Whatever samples
  std::unique_ptr<egammaMonitoring::ShowerShapesHistograms> showerShapesAll;
  std::unique_ptr<egammaMonitoring::ShowerShapesHistograms> showerShapes10GeV;
  std::unique_ptr<egammaMonitoring::IsolationHistograms> isolationAll;
 
  // di-electron data
  std::unique_ptr<egammaMonitoring::RecoClusterHistograms> m_clusterReco;
  std::unique_ptr<egammaMonitoring::DiObjectHistograms> m_diElectron;
 
  // electrons
  std::unique_ptr<egammaMonitoring::RecoElectronHistograms> recoElectronAll;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms> truthElectronAll;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthPromptElectronAll;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthElectronRecoElectronAll;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthPromptElectronWithTrack;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthPromptElectronWithGSFTrack;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthPromptElectronWithReco;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthPromptElectronWithRecoTrack;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthRecoElectronLoose;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthRecoElectronMedium;
  std::unique_ptr<egammaMonitoring::TruthElectronHistograms>
    truthRecoElectronTight;
 
  // photons
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterConvPhoton;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterConvPhotonSi;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterConvPhotonSiSi;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterConvPhotonTRT;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterConvPhotonTRTTRT;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterConvPhotonSiTRT;
  std::unique_ptr<egammaMonitoring::ClusterHistograms> clusterUnconvPhoton;
 
  std::unique_ptr<egammaMonitoring::RecoPhotonHistograms> recoPhotonAll;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonAll;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonAllUnconv;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonAllConv;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonAllLateConv;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonRecoPhoton;
  std::unique_ptr<egammaMonitoring::IHistograms>
    truthPhotonRecoPhotonOrElectron;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvPhoton;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoConv;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoConv1Si;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoConv1TRT;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoConv2Si;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoConv2TRT;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoConv2SiTRT;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonConvRecoUnconv;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvPhoton;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvRecoConv;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvRecoConv1Si;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvRecoConv1TRT;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvRecoConv2Si;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvRecoConv2TRT;
  std::unique_ptr<egammaMonitoring::IHistograms>
    truthPhotonUnconvRecoConv2SiTRT;
  std::unique_ptr<egammaMonitoring::IHistograms> truthPhotonUnconvRecoUnconv;
  std::unique_ptr<egammaMonitoring::IHistograms> recoPhotonUnconvLoose;
  std::unique_ptr<egammaMonitoring::IHistograms> recoPhotonUnconvTight;
  std::unique_ptr<egammaMonitoring::IHistograms> recoPhotonConvLoose;
  std::unique_ptr<egammaMonitoring::IHistograms> recoPhotonConvTight;
  std::unique_ptr<egammaMonitoring::IHistograms>
    recoPhotonUnconvIsoFixedCutTight;
  std::unique_ptr<egammaMonitoring::IHistograms>
    recoPhotonUnconvIsoFixedCutTightCaloOnly;
  std::unique_ptr<egammaMonitoring::IHistograms>
    recoPhotonUnconvIsoFixedCutLoose;
  std::unique_ptr<egammaMonitoring::IHistograms> recoPhotonConvIsoFixedCutTight;
  std::unique_ptr<egammaMonitoring::IHistograms>
    recoPhotonConvIsoFixedCutTightCaloOnly;
  std::unique_ptr<egammaMonitoring::IHistograms> recoPhotonConvIsoFixedCutLoose;
 
  std::map<std::string,std::unique_ptr<egammaMonitoring::TrackHistograms>> mapTrkHistograms;
 
  // Histos
  // General Info
 
  EgammaMonitoring(const std::string& name, ISvcLocator* pSvcLocator);
  ~EgammaMonitoring(){};
 
  virtual StatusCode initialize();
  virtual StatusCode execute();
  virtual StatusCode finalize();
 
private:
  Gaudi::Property<std::string> m_sampleType
    { this, "sampleType", "Unknown", "electron or gamma" };
 
  Gaudi::Property<bool> m_hasTRT
    { this, "hasTRT", true, "is there a TRT in the ID" };
 
  // electron ID
  ToolHandle<IAsgElectronLikelihoodTool> m_Loose_Ele
    { this, "Loose_Ele", "", "" };
  ToolHandle<IAsgElectronLikelihoodTool> m_Medium_Ele
    { this, "Medium_Ele", "", "" };
  ToolHandle<IAsgElectronLikelihoodTool> m_Tight_Ele
    { this, "Tight_Ele", "", "" };
  ToolHandle<IAsgElectronLikelihoodTool> m_Loose_FwdEle
    { this, "Loose_FwdEle", "", "" };
  ToolHandle<IAsgElectronLikelihoodTool> m_Medium_FwdEle
    { this, "Medium_FwdEle", "", "" };
  ToolHandle<IAsgElectronLikelihoodTool> m_Tight_FwdEle
    { this, "Tight_FwdEle", "", "" };
 
  // photon ID
  ToolHandle<IAsgPhotonIsEMSelector> m_Loose_Photon
    { this, "Loose_Photon", "", "" };
  ToolHandle<IAsgPhotonIsEMSelector> m_Tight_Photon
    { this, "Tight_Photon", "", "" };
 
  // photon isolation
  ToolHandle<CP::IIsolationSelectionTool> m_IsoFixedCutTight
    { this, "IsoFixedCutTight", "", "" };
  ToolHandle<CP::IIsolationSelectionTool> m_IsoFixedCutTightCaloOnly
    { this, "IsoTightCaloOnly", "", "" };
  ToolHandle<CP::IIsolationSelectionTool> m_IsoFixedCutLoose
    { this, "IsoFixedCutLoose", "", "" };
 
  // Truth classifier
  ToolHandle<IMCTruthClassifier> m_mcTruthClassifier
    { this, "MCTruthClassifier", "", "" };
 
  // Collections to read
  SG::ReadHandleKey<xAOD::EventInfo> m_eventInfoKey
    { this, "EventInfoKey", "EventInfo", "" };
 
  SG::ReadHandleKey<xAOD::TruthParticleContainer> m_egTruthParticlesKey
    { this, "egammaTruthParticlesKey", "", "" };
 
  SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthParticlesKey
    { this, "truthParticlesKey", "", "" };
 
  SG::ReadHandleKey<xAOD::ElectronContainer> m_ElectronsKey
    { this, "ElectronsKey", "", "" };
 
  SG::ReadHandleKey<xAOD::ElectronContainer> m_FwdElectronsKey
    { this, "FwdElectronsKey", "", "" };
 
  SG::ReadHandleKey<xAOD::PhotonContainer> m_PhotonsKey
    { this, "PhotonsKey", "", "" };
 
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_InDetTrackParticlesKey
    { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" };
 
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_GSFTrackParticlesKey
    { this, "GSFTrackParticlesKey", "", "" };
 
  static bool matchedToElectron(const xAOD::TrackParticle& tp);
  static bool matchedToPion(const xAOD::TrackParticle& tp);
  static bool notMatchedToTruth(const xAOD::TrackParticle& tp);
 
  StatusCode ZeeSelection(float mu, const EventContext& ctx);
  StatusCode ZeePostProc();
 
  int m_CenFwdOverlap[2] = { 0, 0 };
 
  TH1D *m_dR1{}; 
  TH1D *m_dR2{}; 
};
 
#endif