DerivationFramework::AugOriginalCounts Class Reference

Augmentation with primary vertex counts (before thinning) More...

#include <AugOriginalCounts.h>

Inheritance diagram for DerivationFramework::AugOriginalCounts:

Collaboration diagram for DerivationFramework::AugOriginalCounts:

Public Member Functions
	AugOriginalCounts (const std::string &t, const std::string &n, const IInterface *p)
	Main constructor. More...
virtual StatusCode	addBranches () const override
	Main method called for each event. More...
virtual StatusCode	initialize () override

Private Attributes
job options
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigPVNTracks {this, "DO_NOT_SET1", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNTracksKeys {this, "DO_NOT_SET2", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNTracksLRTKeys {this, "DO_NOT_SET22", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNtype0 {this, "DO_NOT_SET3", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNtype1 {this, "DO_NOT_SET4", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNtype2 {this, "DO_NOT_SET5", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNtype3 {this, "DO_NOT_SET6", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_OrigNtypeUnknown {this, "DO_NOT_SET7", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_OrigSqrtPt2Sum {this, "DO_NOT_SET8", "", "internal property"}
SG::WriteDecorHandleKey< xAOD::VertexContainer >	m_d_nPVTracks {this, "DO_NOT_SET9", "", "internal property"}
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_TrackContainername
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_TrackContainerLRTname
SG::ReadHandleKey< xAOD::VertexContainer >	m_PVContainername
bool	m_addPVCountsByType
bool	m_addNTracksToPVs
bool	m_addSqrtPt2SumToPVs

Detailed Description

Augmentation with primary vertex counts (before thinning)

This tool adds primary vertex counts and track counts to the EventInfo container in order to preserve them in case the primary vertex or track collections are thinned.

Job options

Name	Description
TrackContainer	name of the TrackParticle container to be used
VertexContainer	name of the Vertex container to be used
AddPVCountsByType	add PV counts by PV type (default: false)

Definition at line 60 of file AugOriginalCounts.h.

Constructor & Destructor Documentation

AugOriginalCounts()

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

Main constructor.

Definition at line 17 of file AugOriginalCounts.cxx.

                                                            :
    base_class(t,n,p),
    m_TrackContainername("InDetTrackParticles"),
    m_TrackContainerLRTname("InDetLargeD0TrackParticles"),
    m_PVContainername("PrimaryVertices")
  {
 
    declareProperty("TrackContainer", m_TrackContainername);
    declareProperty("TrackLRTContainer", m_TrackContainerLRTname);
    declareProperty("VertexContainer", m_PVContainername);
    declareProperty("AddPVCountsByType", m_addPVCountsByType = false);
    // decorate PVs with track counts and/or sqrt(sum(pt^2))
    // (needed if track collection will be thinned)
    declareProperty("AddNTracksToPVs", m_addNTracksToPVs = false);
    declareProperty("AddSqrtPt2SumToPVs", m_addSqrtPt2SumToPVs = false);
  }

Member Function Documentation

addBranches()

StatusCode DerivationFramework::AugOriginalCounts::addBranches ( ) const

overridevirtual

Main method called for each event.

Definition at line 89 of file AugOriginalCounts.cxx.

  {
 
    const EventContext& ctx = Gaudi::Hive::currentContext();
    
    if(!m_PVContainername.empty()){
 
      SG::WriteDecorHandle<xAOD::EventInfo, int> PV_count(m_OrigPVNTracks, ctx);
      SG::ReadHandle<xAOD::VertexContainer> vertices(m_PVContainername, ctx);
      PV_count(0) = vertices->size();
      
      if ( m_addPVCountsByType ) {
        SG::WriteDecorHandle<xAOD::EventInfo, int> PV0_count(m_OrigNtype0, ctx);
        SG::WriteDecorHandle<xAOD::EventInfo, int> PV1_count(m_OrigNtype1, ctx);
        SG::WriteDecorHandle<xAOD::EventInfo, int> PV2_count(m_OrigNtype2, ctx);
        SG::WriteDecorHandle<xAOD::EventInfo, int> PV3_count(m_OrigNtype3, ctx);
        SG::WriteDecorHandle<xAOD::EventInfo, int> PVUnk_count(m_OrigNtypeUnknown, ctx);
 
        // now count
        constexpr int nvtypes = 5;
        int nvtc[] = {0, 0, 0, 0, 0};
        for (auto vtx : *vertices) {
          VxType::VertexType vt = vtx->vertexType();
          if ( vt >=0 && vt < nvtypes ) {
            nvtc[vt]++; // vertex types 0 - 3
          } else {
            nvtc[nvtypes-1]++; // unknown
          }
        }
        PV0_count(0) = nvtc[0];
        PV1_count(0) = nvtc[1];
        PV2_count(0) = nvtc[2];
        PV3_count(0) = nvtc[3];
        PVUnk_count(0) = nvtc[4];
      } // m_addPVCountsByType
 
      // decorate PVs with track counts
      // (needed if track collection will be thinned)
      if ( m_addNTracksToPVs ) {
        SG::WriteDecorHandle<xAOD::VertexContainer, int> d_nPVTracks(m_d_nPVTracks, ctx);
        for (auto vtx : *vertices) {
          d_nPVTracks(*vtx) = (int)vtx->nTrackParticles();
        }
      } // m_addNTracksToPVs
      
      // decorate PVs with sqrt(sum(pt^2)) of tracks
      // (needed if track collection will be thinned)
      if ( m_addSqrtPt2SumToPVs ) {
        SG::WriteDecorHandle<xAOD::VertexContainer, float> d_pvSqrtPt2Sum(m_OrigSqrtPt2Sum, ctx);
        for (auto vtx : *vertices) {
          float sqrtPt2Sum(0.);
          for (auto tp : vtx->trackParticleLinks()) {
            sqrtPt2Sum += std::sqrt(pow((*tp)->pt(),2));
          }
          d_pvSqrtPt2Sum(*vtx) = sqrtPt2Sum;
        }
      } // m_addSqrtPt2SumToPVs
    }
    
    if(!m_TrackContainername.empty()){
      SG::ReadHandle<xAOD::TrackParticleContainer> tracks(m_TrackContainername, ctx);
      SG::WriteDecorHandle<xAOD::EventInfo, int> track_count(m_OrigNTracksKeys, ctx);
      track_count(0) = tracks->size();
    }
    if(!m_TrackContainerLRTname.empty()){
      SG::ReadHandle<xAOD::TrackParticleContainer> tracks(m_TrackContainerLRTname, ctx);
      SG::WriteDecorHandle<xAOD::EventInfo, int> track_count(m_OrigNTracksLRTKeys, ctx);
      track_count(0) = tracks->size();
    }
 
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode DerivationFramework::AugOriginalCounts::initialize ( )

overridevirtual

Definition at line 36 of file AugOriginalCounts.cxx.

  {
     ATH_CHECK(m_TrackContainername.initialize(SG::AllowEmpty));
     ATH_CHECK(m_TrackContainerLRTname.initialize(SG::AllowEmpty));
     ATH_CHECK(m_PVContainername.initialize(SG::AllowEmpty));
     
     if(!m_PVContainername.empty()){
        std::string pvstring = "EventInfo.OriginalCount_";
        pvstring += m_PVContainername.key();
        m_OrigPVNTracks = std::move(pvstring);
        ATH_CHECK(m_OrigPVNTracks.initialize());
     }
     if ( m_addPVCountsByType ) {
        std::string pv0string = "EventInfo.OriginalCount_type0_"+m_PVContainername.key();
        std::string pv1string = "EventInfo.OriginalCount_type1_"+m_PVContainername.key();
        std::string pv2string = "EventInfo.OriginalCount_type2_"+m_PVContainername.key();
        std::string pv3string = "EventInfo.OriginalCount_type3_"+m_PVContainername.key();
        std::string pvUstring = "EventInfo.OriginalCount_typeUnknown_"+m_PVContainername.key();
        m_OrigNtype0 = std::move(pv0string);
        m_OrigNtype1 = std::move(pv1string);
        m_OrigNtype2 = std::move(pv2string);
        m_OrigNtype3 = std::move(pv3string);
        m_OrigNtypeUnknown = std::move(pvUstring);
        ATH_CHECK(m_OrigNtype0.initialize());
        ATH_CHECK(m_OrigNtype1.initialize());
        ATH_CHECK(m_OrigNtype2.initialize());
        ATH_CHECK(m_OrigNtype3.initialize());
        ATH_CHECK(m_OrigNtypeUnknown.initialize());
     }
     if ( m_addSqrtPt2SumToPVs ) {
        std::string trackcon  = m_PVContainername.key();
        trackcon += ".OriginalCount_";
        trackcon += m_TrackContainername.key();
        m_OrigSqrtPt2Sum = std::move(trackcon);
        ATH_CHECK(m_OrigSqrtPt2Sum.initialize());
     }
     if ( m_addNTracksToPVs ) {
        std::string name = m_PVContainername.key();
        name+= ".OrigNTracks";
        m_d_nPVTracks = std::move(name);
        ATH_CHECK(m_d_nPVTracks.initialize());
     }
     if(!m_TrackContainername.empty()){
        m_OrigNTracksKeys = "EventInfo.OriginalCount_" + m_TrackContainername.key();
        ATH_CHECK(m_OrigNTracksKeys.initialize());
     }
     if(!m_TrackContainerLRTname.empty()){
        m_OrigNTracksLRTKeys = "EventInfo.OriginalCount_" + m_TrackContainerLRTname.key();
        ATH_CHECK(m_OrigNTracksLRTKeys.initialize());
     }
     return StatusCode::SUCCESS;
  }

Member Data Documentation

m_addNTracksToPVs

bool DerivationFramework::AugOriginalCounts::m_addNTracksToPVs

private

Definition at line 87 of file AugOriginalCounts.h.

m_addPVCountsByType

bool DerivationFramework::AugOriginalCounts::m_addPVCountsByType

private

Definition at line 86 of file AugOriginalCounts.h.

m_addSqrtPt2SumToPVs

bool DerivationFramework::AugOriginalCounts::m_addSqrtPt2SumToPVs

private

Definition at line 88 of file AugOriginalCounts.h.

m_d_nPVTracks

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::AugOriginalCounts::m_d_nPVTracks {this, "DO_NOT_SET9", "", "internal property"}

private

Definition at line 82 of file AugOriginalCounts.h.

m_OrigNTracksKeys

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNTracksKeys {this, "DO_NOT_SET2", "", "internal property"}

private

Definition at line 73 of file AugOriginalCounts.h.

m_OrigNTracksLRTKeys

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNTracksLRTKeys {this, "DO_NOT_SET22", "", "internal property"}

private

Definition at line 74 of file AugOriginalCounts.h.

m_OrigNtype0

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNtype0 {this, "DO_NOT_SET3", "", "internal property"}

private

Definition at line 75 of file AugOriginalCounts.h.

m_OrigNtype1

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNtype1 {this, "DO_NOT_SET4", "", "internal property"}

private

Definition at line 76 of file AugOriginalCounts.h.

m_OrigNtype2

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNtype2 {this, "DO_NOT_SET5", "", "internal property"}

private

Definition at line 77 of file AugOriginalCounts.h.

m_OrigNtype3

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNtype3 {this, "DO_NOT_SET6", "", "internal property"}

private

Definition at line 78 of file AugOriginalCounts.h.

m_OrigNtypeUnknown

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigNtypeUnknown {this, "DO_NOT_SET7", "", "internal property"}

private

Definition at line 79 of file AugOriginalCounts.h.

m_OrigPVNTracks

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::AugOriginalCounts::m_OrigPVNTracks {this, "DO_NOT_SET1", "", "internal property"}

private

Definition at line 72 of file AugOriginalCounts.h.

m_OrigSqrtPt2Sum

SG::WriteDecorHandleKey<xAOD::VertexContainer> DerivationFramework::AugOriginalCounts::m_OrigSqrtPt2Sum {this, "DO_NOT_SET8", "", "internal property"}

private

Definition at line 81 of file AugOriginalCounts.h.

m_PVContainername

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::AugOriginalCounts::m_PVContainername

private

Definition at line 85 of file AugOriginalCounts.h.

m_TrackContainerLRTname

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::AugOriginalCounts::m_TrackContainerLRTname

private

Definition at line 84 of file AugOriginalCounts.h.

m_TrackContainername

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::AugOriginalCounts::m_TrackContainername

private

Definition at line 83 of file AugOriginalCounts.h.

---

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

• AugOriginalCounts.h
• AugOriginalCounts.cxx