EGElectronAmbiguityTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// EGElectronAmbiguityTool.h, (c) ATLAS Detector software
 
#ifndef DERIVATIONFRAMEWORK_EGELECTRONAMBIGUITYTOOL_H
#define DERIVATIONFRAMEWORK_EGELECTRONAMBIGUITYTOOL_H
 
#include "GaudiKernel/ToolHandle.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
#include "GaudiKernel/EventContext.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandle.h"
 
#include "AthContainers/ConstDataVector.h"
#include "xAODEgamma/Electron.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/Vertex.h"
#include "xAODTracking/VertexContainer.h"
 
#include <string>
 
namespace DerivationFramework {
 
class EGElectronAmbiguityTool : public extends<AthAlgTool, IAugmentationTool>
{
 
public:
  EGElectronAmbiguityTool(const std::string& t,
                          const std::string& n,
                          const IInterface* p);
 
  virtual StatusCode initialize() override final;
  virtual StatusCode addBranches() const override final;
 
private:
  SG::ReadHandleKey<xAOD::ElectronContainer> m_containerName{
    this,
    "ContainerName",
    "Electrons",
    "SG key of electron container"
  };
  SG::ReadHandleKey<xAOD::VertexContainer> m_VtxContainerName{
    this,
    "VtxContainerName",
    "PrimaryVertices",
    "SG key of vertex container"
  };
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_tpContainerName{
    this,
    "tpContainerName",
    "InDetTrackParticles",
    "SG key of track particles container"
  };
 
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_tpCName{
    this,
    "tpCName",
    "GSFTrackParticles",
    "SG key of TrackParticleInputContainer"
  };
 
  // Write decoration handle keys
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_drv{ this, "DFCommonSimpleConvRadius", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dphiv{ this, "DFCommonSimpleConvPhi", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dmee{ this, "DFCommonSimpleMee", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dmeeVtx{ this, "DFCommonSimpleMeeAtVtx", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dsep{ this, "DFCommonSimpleSeparation", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dambi{ this, "DFCommonAddAmbiguity", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dtrv{ this, "DFCommonProdTrueRadius", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dtpv{ this, "DFCommonProdTruePhi", "", "" };
 
  SG::WriteDecorHandleKey<xAOD::ElectronContainer>
    m_dtzv{ this, "DFCommonProdTrueZ", "", "" };
 
  struct DecorHandles {
    DecorHandles (const EGElectronAmbiguityTool& tool, const EventContext& ctx);
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> drv;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dphiv;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dmee;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dmeeVtx;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dsep;
    SG::WriteDecorHandle<xAOD::ElectronContainer, int>   dambi;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dtrv;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dtpv;
    SG::WriteDecorHandle<xAOD::ElectronContainer, float> dtzv;
  };
  friend struct DecorHandles;
 
  StatusCode decorateSimple(
    DecorHandles& dh,
    std::unique_ptr<ConstDataVector<xAOD::TrackParticleContainer>>& tpC,
    const xAOD::Electron* ele,
    const xAOD::Vertex* pvtx) const;
 
  bool m_isMC{};
 
  // cuts to select the electron to run on
  double m_elepTCut;
  std::string m_idCut;
 
  // cuts to select the other track
  unsigned int m_nSiCut;
  double m_dctCut, m_sepCut, m_dzCut;
 
  // cuts to define the various types :
  // ambi = -1 : no other track, 0 : other track exists but no good gamma reco,
  // 1 : gamma*, 2 : material conversion
  double m_rvECCut, m_meeAtVtxECCut, m_meeICCut;
};
}
 
#endif // DERIVATIONFRAMEWORK_EGCONVERSIONINFOTOOL_H