DerivationFramework::PhotonsDirectionTool Class Reference

#include <PhotonsDirectionTool.h>

Inheritance diagram for DerivationFramework::PhotonsDirectionTool:

Collaboration diagram for DerivationFramework::PhotonsDirectionTool:

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

Private Attributes
SG::ReadHandleKey< xAOD::PhotonContainer >	m_collName
SG::WriteHandleKey< std::vector< float > >	m_sgEta
SG::WriteHandleKey< std::vector< float > >	m_sgPhi
SG::WriteHandleKey< std::vector< float > >	m_sgEt
SG::WriteHandleKey< std::vector< float > >	m_sgE
bool	m_doEta = false
bool	m_doPhi = false
bool	m_doEt = false
bool	m_doE = false

Detailed Description

Definition at line 17 of file PhotonsDirectionTool.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 41 of file PhotonsDirectionTool.cxx.

{
  // Retrieve photon container
  SG::ReadHandle<xAOD::PhotonContainer> photons(m_collName, ctx);
  // define the pointers to vectors which will hold the additional payloads
  auto recEta = std::make_unique<std::vector<float>>();
  auto recPhi = std::make_unique<std::vector<float>>();
  auto recEt = std::make_unique<std::vector<float>>();
  auto recE = std::make_unique<std::vector<float>>();
 
  // Loop over electrons and fill the vectors
  for (const xAOD::Photon* ph : *photons) {
    // Prepare variables
    float eta(0.0), phi(0.0), e(0.0), et(0.0);
 
    if (ph->nCaloClusters() > 0) {
      const xAOD::CaloCluster* gCluster = ph->caloCluster(0);
      eta = gCluster->etaBE(2);
      phi = gCluster->phi();
      e = gCluster->e();
      et = e / cosh(eta);
    } else {
      ATH_MSG_WARNING(
        "Couldn't retrieve photon cluster, will use photon 4-momentum");
      eta = ph->eta();
      phi = ph->phi();
      e = ph->e();
      et = ph->pt();
    }
    recEta->push_back(eta);
    recPhi->push_back(phi);
    recEt->push_back(et);
    recE->push_back(e);
  } // end of loop over photons
 
  // Write payload to StoreGate for downstream client access
  if (m_doEta) {
    SG::WriteHandle<std::vector<float>> etas(m_sgEta, ctx);
    ATH_CHECK(etas.record(std::move(recEta)));
  }
  if (m_doPhi) {
    SG::WriteHandle<std::vector<float>> phis(m_sgPhi, ctx);
    ATH_CHECK(phis.record(std::move(recPhi)));
  }
  if (m_doEt) {
    SG::WriteHandle<std::vector<float>> ets(m_sgEt, ctx);
    ATH_CHECK(ets.record(std::move(recEt)));
  }
  if (m_doE) {
    SG::WriteHandle<std::vector<float>> es(m_sgE, ctx);
    ATH_CHECK(es.record(std::move(recE)));
  }
  return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::PhotonsDirectionTool::initialize ( )

finaloverridevirtual

Definition at line 20 of file PhotonsDirectionTool.cxx.

{
  ATH_CHECK(m_collName.initialize());
 
  m_doEta = !(m_sgEta.key().empty());
  m_doPhi = !(m_sgPhi.key().empty());
  m_doEt = !(m_sgEt.key().empty());
  m_doE = !(m_sgE.key().empty());
  if (!m_doEta && !m_doPhi && !m_doE && !m_doEt) {
    ATH_MSG_ERROR("You are requesting the PhotonsDirectionTool but have "
                  "provided no SG names for any of the results");
    return StatusCode::FAILURE;
  }
  ATH_CHECK(m_sgEta.initialize(m_doEta));
  ATH_CHECK(m_sgPhi.initialize(m_doPhi));
  ATH_CHECK(m_sgEt.initialize(m_doEt));
  ATH_CHECK(m_sgE.initialize(m_doE));
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_collName

SG::ReadHandleKey<xAOD::PhotonContainer> DerivationFramework::PhotonsDirectionTool::m_collName

private

Initial value:

{ this,
        "PhotonContainer",
        "Photons",
        "Input Photons" }

Definition at line 27 of file PhotonsDirectionTool.h.

m_doE

bool DerivationFramework::PhotonsDirectionTool::m_doE = false

private

Definition at line 55 of file PhotonsDirectionTool.h.

m_doEt

bool DerivationFramework::PhotonsDirectionTool::m_doEt = false

private

Definition at line 54 of file PhotonsDirectionTool.h.

m_doEta

bool DerivationFramework::PhotonsDirectionTool::m_doEta = false

private

Definition at line 52 of file PhotonsDirectionTool.h.

m_doPhi

bool DerivationFramework::PhotonsDirectionTool::m_doPhi = false

private

Definition at line 53 of file PhotonsDirectionTool.h.

m_sgE

SG::WriteHandleKey<std::vector<float> > DerivationFramework::PhotonsDirectionTool::m_sgE

private

Initial value:

{ this,
      "ESGEntry",
      "",
      "output E vector" }

Definition at line 47 of file PhotonsDirectionTool.h.

m_sgEt

SG::WriteHandleKey<std::vector<float> > DerivationFramework::PhotonsDirectionTool::m_sgEt

private

Initial value:

{ this,
      "EtSGEntry",
      "",
      "output E vector" }

Definition at line 42 of file PhotonsDirectionTool.h.

m_sgEta

SG::WriteHandleKey<std::vector<float> > DerivationFramework::PhotonsDirectionTool::m_sgEta

private

Initial value:

{ this,
      "EtaSGEntry",
      "",
      "output Eta vector" }

Definition at line 32 of file PhotonsDirectionTool.h.

m_sgPhi

SG::WriteHandleKey<std::vector<float> > DerivationFramework::PhotonsDirectionTool::m_sgPhi

private

Initial value:

{ this,
      "PhiSGEntry",
      "",
      "output Phi vector" }

Definition at line 37 of file PhotonsDirectionTool.h.

---

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

• PhotonsDirectionTool.h
• PhotonsDirectionTool.cxx