DerivationFramework::PFlowAugmentationTool Class Reference

#include <PFlowAugmentationTool.h>

Inheritance diagram for DerivationFramework::PFlowAugmentationTool:

Collaboration diagram for DerivationFramework::PFlowAugmentationTool:

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

Private Attributes
PublicToolHandle< CP::IWeightPFOTool >	m_weightPFOTool {this, "WeightPFOTool", "CP::WeightPFOTool/WeightPFOTool"}
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainer_key {this, "VertexContainer", "PrimaryVertices", "Input vertex container"}
SG::ReadHandleKey< xAOD::FlowElementContainer >	m_pfoContainer_key {this, "GlobalChargedParticleFlowObjects", "GlobalChargedParticleFlowObjects", "Input charged PFO"}
SG::WriteDecorHandleKey< xAOD::FlowElementContainer >	m_corrP4_ptKey {this, "m_corrP4_ptKey", m_pfoContainer_key, "DFCommonPFlow_CaloCorrectedPt", "Decoration for weighted charged PFO pt"}
SG::WriteDecorHandleKey< xAOD::FlowElementContainer >	m_z0Key {this, "m_z0Key", m_pfoContainer_key, "DFCommonPFlow_z0", "Decoration for track z0"}
SG::WriteDecorHandleKey< xAOD::FlowElementContainer >	m_vzKey {this, "m_vzKey", m_pfoContainer_key, "DFCommonPFlow_vz", "Decoration for track vz"}
SG::WriteDecorHandleKey< xAOD::FlowElementContainer >	m_d0Key {this, "m_d0Key", m_pfoContainer_key, "DFCommonPFlow_d0", "Decoration for track d0"}
SG::WriteDecorHandleKey< xAOD::FlowElementContainer >	m_thetaKey {this, "m_thetaKey", m_pfoContainer_key, "DFCommonPFlow_theta", "Decoration for track theta"}
SG::WriteDecorHandleKey< xAOD::FlowElementContainer >	m_envWeightKey {this, "m_envWeightKey", m_pfoContainer_key, "DFCommonPFlow_envWeight", "Decoration for weight for dense environments"}

Detailed Description

Definition at line 29 of file PFlowAugmentationTool.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 32 of file PFlowAugmentationTool.cxx.

  {
    // Get the vertex.
    const xAOD::Vertex* pv{};
 
    auto vertexContainer = SG::makeHandle (m_vertexContainer_key, ctx);
    if (!vertexContainer.isValid()){
      ATH_MSG_WARNING("Invalid  xAOD::VertexContainer datahandle"
                      << m_vertexContainer_key.key());
      return StatusCode::FAILURE;
    }
    auto pvcont = vertexContainer.cptr();
    if ( pvcont == 0 || pvcont->size()==0 ) {
      ATH_MSG_WARNING(" Failed to retrieve PrimaryVertices collection" );
      return StatusCode::FAILURE;
    }
    for (const auto vx : *pvcont) {
      if (vx->vertexType() == xAOD::VxType::PriVtx) {
        pv = vx;
        break;
      }//If we have a vertex of type primary vertex
    }//iterate over the vertices and check their type
 
    // Use NoVtx as fall-back in case no PV is found, but the events should be rejected by the user
    // If there is no such then mark all CPFOs as unmatched
    if (pv == nullptr) {
      ATH_MSG_DEBUG("Could not find a primary vertex in this event" );
      for (auto theVertex : *pvcont) {
        if (xAOD::VxType::NoVtx == theVertex->vertexType() ) {
          pv = theVertex;
          break;
        }
      }
      if (nullptr == pv) {
        ATH_MSG_WARNING("Found neither PriVtx nor NoVtx in this event" );
      }
    }
 
    SG::WriteDecorHandle<xAOD::FlowElementContainer,float> dec_corrP4_pt(m_corrP4_ptKey, ctx);
    SG::WriteDecorHandle<xAOD::FlowElementContainer,float> dec_z0(m_z0Key, ctx);
    SG::WriteDecorHandle<xAOD::FlowElementContainer,float> dec_vz(m_vzKey, ctx);
    SG::WriteDecorHandle<xAOD::FlowElementContainer,float> dec_d0(m_d0Key, ctx);
    SG::WriteDecorHandle<xAOD::FlowElementContainer,float> dec_theta(m_thetaKey, ctx);
    SG::WriteDecorHandle<xAOD::FlowElementContainer,float> dec_envWeight(m_envWeightKey, ctx);
 
    auto pfoContainer = SG::makeHandle (m_pfoContainer_key, ctx);
    if (!pfoContainer.isValid()){
      ATH_MSG_WARNING("Invalid  xAOD::PFOContainer datahandle"
                      << m_pfoContainer_key.key());
      return StatusCode::FAILURE;
    }
    auto cpfos = pfoContainer.cptr();
 
    for ( const xAOD::FlowElement* cpfo : *cpfos ) {
      if ( cpfo == 0 ) {
        ATH_MSG_WARNING("Have NULL pointer to charged PFO");
        continue;
      }
      const xAOD::TrackParticle* ptrk = dynamic_cast<const xAOD::TrackParticle*>(cpfo->chargedObject(0));
      if ( ptrk == 0 ) {
        ATH_MSG_WARNING("Skipping charged PFO with null track pointer.");
        continue;
      }
 
      // decorate the track properties
      dec_z0(*cpfo) = ptrk->z0();
      dec_vz(*cpfo) = ptrk->vz();
      dec_d0(*cpfo) = ptrk->d0();
      dec_theta(*cpfo) = ptrk->theta();
 
      //find the weights from the tool
      float weight = 1.0;
      const static SG::AuxElement::ConstAccessor<int> accIsInDE("IsInDenseEnvironment");
      if(accIsInDE.isAvailable(*cpfo)){
        ATH_CHECK( m_weightPFOTool->fillWeight( *cpfo, weight ) );
      }
 
      // decorate the computed variables
      dec_corrP4_pt(*cpfo) = weight*cpfo->pt();
      dec_envWeight(*cpfo) = weight;
    }
 
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode DerivationFramework::PFlowAugmentationTool::initialize ( )

finaloverridevirtual

Definition at line 17 of file PFlowAugmentationTool.cxx.

  {
 
    ATH_CHECK(m_vertexContainer_key.initialize());
    ATH_CHECK(m_pfoContainer_key.initialize());
    ATH_CHECK(m_corrP4_ptKey.initialize());
    ATH_CHECK(m_z0Key.initialize());
    ATH_CHECK(m_vzKey.initialize());
    ATH_CHECK(m_d0Key.initialize());
    ATH_CHECK(m_thetaKey.initialize());
    ATH_CHECK(m_envWeightKey.initialize());
 
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_corrP4_ptKey

SG::WriteDecorHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_corrP4_ptKey {this, "m_corrP4_ptKey", m_pfoContainer_key, "DFCommonPFlow_CaloCorrectedPt", "Decoration for weighted charged PFO pt"}

private

Definition at line 44 of file PFlowAugmentationTool.h.

m_d0Key

SG::WriteDecorHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_d0Key {this, "m_d0Key", m_pfoContainer_key, "DFCommonPFlow_d0", "Decoration for track d0"}

private

Definition at line 47 of file PFlowAugmentationTool.h.

m_envWeightKey

SG::WriteDecorHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_envWeightKey {this, "m_envWeightKey", m_pfoContainer_key, "DFCommonPFlow_envWeight", "Decoration for weight for dense environments"}

private

Definition at line 49 of file PFlowAugmentationTool.h.

m_pfoContainer_key

SG::ReadHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_pfoContainer_key {this, "GlobalChargedParticleFlowObjects", "GlobalChargedParticleFlowObjects", "Input charged PFO"}

private

Definition at line 42 of file PFlowAugmentationTool.h.

m_thetaKey

SG::WriteDecorHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_thetaKey {this, "m_thetaKey", m_pfoContainer_key, "DFCommonPFlow_theta", "Decoration for track theta"}

private

Definition at line 48 of file PFlowAugmentationTool.h.

m_vertexContainer_key

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::PFlowAugmentationTool::m_vertexContainer_key {this, "VertexContainer", "PrimaryVertices", "Input vertex container"}

private

Definition at line 41 of file PFlowAugmentationTool.h.

m_vzKey

SG::WriteDecorHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_vzKey {this, "m_vzKey", m_pfoContainer_key, "DFCommonPFlow_vz", "Decoration for track vz"}

private

Definition at line 46 of file PFlowAugmentationTool.h.

m_weightPFOTool

PublicToolHandle<CP::IWeightPFOTool> DerivationFramework::PFlowAugmentationTool::m_weightPFOTool {this, "WeightPFOTool", "CP::WeightPFOTool/WeightPFOTool"}

private

Definition at line 39 of file PFlowAugmentationTool.h.

m_z0Key

SG::WriteDecorHandleKey<xAOD::FlowElementContainer> DerivationFramework::PFlowAugmentationTool::m_z0Key {this, "m_z0Key", m_pfoContainer_key, "DFCommonPFlow_z0", "Decoration for track z0"}

private

Definition at line 45 of file PFlowAugmentationTool.h.

---

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

• PFlowAugmentationTool.h
• PFlowAugmentationTool.cxx