DerivationFramework::PhotonVertexSelectionWrapper Class Reference

#include <PhotonVertexSelectionWrapper.h>

Inheritance diagram for DerivationFramework::PhotonVertexSelectionWrapper:

Collaboration diagram for DerivationFramework::PhotonVertexSelectionWrapper:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
ToolHandle< CP::IPhotonPointingTool >	m_photonPointingTool
	PhotonPointingTool. More...
SG::ReadHandleKey< xAOD::PhotonContainer >	m_photonContainer
	Input photon container. More...
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainer
	Input primary vertex container. More...
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_vtxPt { this, "pt", m_vertexContainer, "", "" }
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_vtxEta
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_vtxPhi
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_vtxSumPt
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_vtxSumPt2

Detailed Description

Definition at line 30 of file PhotonVertexSelectionWrapper.h.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::PhotonVertexSelectionWrapper::addBranches ( const EventContext &  ctx ) const

finaloverridevirtual

Definition at line 33 of file PhotonVertexSelectionWrapper.cxx.

{
  // retrieve the input containers
  SG::ReadHandle<xAOD::PhotonContainer> photons{ m_photonContainer, ctx };
  SG::ReadHandle<xAOD::VertexContainer> vertices{ m_vertexContainer, ctx };
 
  // Update calo pointing auxdata for photons
  if (m_photonPointingTool->updatePointingAuxdata(*photons).isFailure()) {
    ATH_MSG_ERROR("Couldn't update photon calo pointing auxdata");
    return StatusCode::FAILURE;
  }
 
  // create the decorators
  SG::WriteDecorHandle<xAOD::VertexContainer, float> vtxPt(m_vtxPt, ctx);
  SG::WriteDecorHandle<xAOD::VertexContainer, float> vtxEta(m_vtxEta, ctx);
  SG::WriteDecorHandle<xAOD::VertexContainer, float> vtxPhi(m_vtxPhi, ctx);
  SG::WriteDecorHandle<xAOD::VertexContainer, float> vtxSumPt(m_vtxSumPt, ctx);
  SG::WriteDecorHandle<xAOD::VertexContainer, float> vtxSumPt2(m_vtxSumPt2, ctx);
  bool isMomentum_available = vtxPt.isAvailable();
  bool isSumPt_available = vtxSumPt.isAvailable();
  bool isSumPt2_available = vtxSumPt2.isAvailable();
  bool found_PV = false;
 
  // Loop over vertices and update auxdata
  for (const auto *vertex : *vertices) {
 
    float pt = -999.;
    float eta = -999.;
    float phi = -999.;
    float sumPt = -999.;
    float sumPt2 = -999.;
 
    if(vertex->vertexType() == xAOD::VxType::VertexType::PriVtx or
       vertex->vertexType() == xAOD::VxType::VertexType::PileUp){
 
      found_PV = true;
      // Get momentum vector of vertex and add it as a decoration
      TLorentzVector vmom = xAOD::PVHelpers::getVertexMomentum(vertex, isMomentum_available);
      pt = sqrt(vmom.Px() * vmom.Px() + vmom.Py() * vmom.Py());
      if(pt>0.){
    eta = asinh(vmom.Pz() / pt);
    phi = acos(vmom.Px() / pt); // in [0,Pi]
      }
      // Calculate additional quantities
      sumPt = xAOD::PVHelpers::getVertexSumPt(vertex, 1, isSumPt_available);
      sumPt2 = xAOD::PVHelpers::getVertexSumPt(vertex, 2, isSumPt2_available);
 
    }
 
    // write decorations
    if(!isMomentum_available){
      vtxPt(*vertex) = pt;
      vtxEta(*vertex) = eta;
      vtxPhi(*vertex) = phi;
    }
    if(!isSumPt_available) vtxSumPt(*vertex) = sumPt;
    // For events where no PV is reconstructed, beamspot is saved and isSumPt2_available = False
    // Need to also check found_PV to avoid modifying locked store
    if(!isSumPt2_available && found_PV) vtxSumPt2(*vertex) = sumPt2;
 
  } // end loop o vertices
 
  return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::PhotonVertexSelectionWrapper::initialize ( )

finaloverridevirtual

Definition at line 12 of file PhotonVertexSelectionWrapper.cxx.

{
  ATH_MSG_DEBUG("Initializing " << name() << "...");
  ATH_CHECK(m_photonPointingTool.retrieve());
  ATH_MSG_DEBUG("Retrieved tool " << m_photonPointingTool);
 
  ATH_CHECK(m_photonContainer.initialize());
  ATH_CHECK(m_vertexContainer.initialize());
 
  ATH_CHECK(m_vtxPt.initialize());
  ATH_CHECK(m_vtxEta.initialize());
  ATH_CHECK(m_vtxPhi.initialize());
  ATH_CHECK(m_vtxSumPt.initialize());
  ATH_CHECK(m_vtxSumPt2.initialize());
 
  ATH_MSG_DEBUG("Initialization successful");
 
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_photonContainer

SG::ReadHandleKey<xAOD::PhotonContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_photonContainer

private

Initial value:

{
      this,
      "PhotonContainer",
      "Photons",
      "SG key of input photon container"
    }

Input photon container.

Definition at line 49 of file PhotonVertexSelectionWrapper.h.

m_photonPointingTool

ToolHandle<CP::IPhotonPointingTool> DerivationFramework::PhotonVertexSelectionWrapper::m_photonPointingTool

private

Initial value:

{
      this,
        "PhotonPointingTool",
        "",
        "Handle to the photon pointing tool"
        }

PhotonPointingTool.

Definition at line 41 of file PhotonVertexSelectionWrapper.h.

m_vertexContainer

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_vertexContainer

private

Initial value:

{
      this,
      "VertexContainer",
      "PrimaryVertices",
      "SG key of input vertex container"
    }

Input primary vertex container.

Definition at line 57 of file PhotonVertexSelectionWrapper.h.

m_vtxEta

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_vtxEta

private

Initial value:

{ this,
      "eta",
      m_vertexContainer, "",
      "" }

Definition at line 67 of file PhotonVertexSelectionWrapper.h.

m_vtxPhi

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_vtxPhi

private

Initial value:

{ this,
      "phi",
      m_vertexContainer, "",
      "" }

Definition at line 71 of file PhotonVertexSelectionWrapper.h.

m_vtxPt

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_vtxPt { this, "pt", m_vertexContainer, "", "" }

private

Definition at line 66 of file PhotonVertexSelectionWrapper.h.

m_vtxSumPt

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_vtxSumPt

private

Initial value:

{ this,
      "sumPt",
      m_vertexContainer,  "",
      "" }

Definition at line 75 of file PhotonVertexSelectionWrapper.h.

m_vtxSumPt2

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::PhotonVertexSelectionWrapper::m_vtxSumPt2

private

Initial value:

{ this,
      "sumPt2",
      m_vertexContainer, "",
      "" }

Definition at line 79 of file PhotonVertexSelectionWrapper.h.

---

The documentation for this class was generated from the following files:

• PhotonVertexSelectionWrapper.h
• PhotonVertexSelectionWrapper.cxx