DerivationFramework::PhotonVertexSelectionWrapper Class Reference

#include <PhotonVertexSelectionWrapper.h>

Inheritance diagram for DerivationFramework::PhotonVertexSelectionWrapper:

Collaboration diagram for DerivationFramework::PhotonVertexSelectionWrapper:

Public Member Functions
	PhotonVertexSelectionWrapper (const std::string &t, const std::string &n, const IInterface *p)
StatusCode	initialize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
ToolHandle< CP::IPhotonPointingTool >	m_photonPointingTool
	PhotonPointingTool. More...
std::string	m_decPrefix
	Prefix for the decorations. 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", "", "" }
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.

Constructor & Destructor Documentation

PhotonVertexSelectionWrapper()

DerivationFramework::PhotonVertexSelectionWrapper::PhotonVertexSelectionWrapper	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 11 of file PhotonVertexSelectionWrapper.cxx.

  : base_class(t, n, p)
  , m_decPrefix("")
{
  declareProperty(
    "DecorationPrefix", m_decPrefix, "Prefix for the decoration name");
}

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 52 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 ( )

finaloverride

Definition at line 22 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());
 
  const std::string baseName = m_vertexContainer.key();
  std::string prefix = "";
  if(!m_decPrefix.empty()) prefix = m_decPrefix + "_";
  m_vtxPt = baseName + "." + prefix + "pt";
  m_vtxEta = baseName + "." + prefix + "eta";
  m_vtxPhi = baseName + "." + prefix + "phi";
  m_vtxSumPt = baseName + "." + prefix + "sumPt";
  m_vtxSumPt2 = baseName + "." + prefix + "sumPt2";
 
  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_decPrefix

std::string DerivationFramework::PhotonVertexSelectionWrapper::m_decPrefix

private

Prefix for the decorations.

Definition at line 50 of file PhotonVertexSelectionWrapper.h.

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 53 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 42 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 61 of file PhotonVertexSelectionWrapper.h.

m_vtxEta

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

private

Initial value:

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

Definition at line 71 of file PhotonVertexSelectionWrapper.h.

m_vtxPhi

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

private

Initial value:

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

Definition at line 75 of file PhotonVertexSelectionWrapper.h.

m_vtxPt

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

private

Definition at line 70 of file PhotonVertexSelectionWrapper.h.

m_vtxSumPt

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

private

Initial value:

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

Definition at line 79 of file PhotonVertexSelectionWrapper.h.

m_vtxSumPt2

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

private

Initial value:

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

Definition at line 83 of file PhotonVertexSelectionWrapper.h.

---

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

• PhotonVertexSelectionWrapper.h
• PhotonVertexSelectionWrapper.cxx