DerivationFramework::AugmentationToolExample Class Reference

#include <AugmentationToolExample.h>

Inheritance diagram for DerivationFramework::AugmentationToolExample:

Collaboration diagram for DerivationFramework::AugmentationToolExample:

Public Member Functions
virtual StatusCode	addBranches () const

Detailed Description

Definition at line 15 of file AugmentationToolExample.h.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::AugmentationToolExample::addBranches ( ) const

virtual

Definition at line 24 of file AugmentationToolExample.cxx.

  {
 
      // Set up the vector
      std::unique_ptr<std::vector<float> > track_z0_PV(new std::vector<float>());
 
      // Set up the decorators 
      SG::AuxElement::Decorator< float > decorator("DFDecoratorExample"); 
 
      // CALCULATION OF THE NEW VARIABLE
      // Get Primary vertex
      const xAOD::VertexContainer* vertices =  evtStore()->retrieve< const xAOD::VertexContainer >("PrimaryVertices");
    
      if(!vertices) {
        ATH_MSG_ERROR ("Couldn't retrieve VertexContainer with key PrimaryVertices");
        return StatusCode::FAILURE;
      }
 
      const xAOD::Vertex* pv{nullptr};
      for (const xAOD::Vertex* vx : *vertices) {
        if (vx->vertexType() == xAOD::VxType::PriVtx) {
          pv = vx;
          break;
        }
      }
 
      // Get the track container
      const xAOD::TrackParticleContainer* tracks = evtStore()->retrieve< const xAOD::TrackParticleContainer >("InDetTrackParticles");
      
      if(!tracks) {
        ATH_MSG_ERROR ("Couldn't retrieve TrackParticleContainer with key InDetTrackParticles");
        return StatusCode::FAILURE;
      }
 
      // Get track z0 w.r.t PV: this is what we're adding
      for (xAOD::TrackParticleContainer::const_iterator trackIt=tracks->begin(); trackIt!=tracks->end(); ++trackIt) {
        if (pv) {
          float z0wrtPV = (*trackIt)->z0() + (*trackIt)->vz() - pv->z(); // CALCULATE THE QUANTITY
          track_z0_PV->push_back(z0wrtPV); // ADD TO VECTOR
          decorator(**trackIt) = z0wrtPV; // DECORATE THE TRACK
        } else {
          track_z0_PV->push_back(999.);
          decorator(**trackIt) = 999.;
        }
      }
 
      // Write decision to SG for access by downstream algs 
      if (evtStore()->contains<std::vector<float> >("DFAugmentationExample")) {
          ATH_MSG_ERROR("Tool is attempting to write StoreGate keys which already exists. Please use a different key");
          return StatusCode::FAILURE;
      } else {
        ATH_CHECK(evtStore()->record(std::move(track_z0_PV), "DFAugmentationExample"));
      }
 
      return StatusCode::SUCCESS;
  }

---

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

• AugmentationToolExample.h
• AugmentationToolExample.cxx