DerivationFramework::JetMSVAugmentation Class Reference

#include <JetMSVAugmentation.h>

Inheritance diagram for DerivationFramework::JetMSVAugmentation:

Collaboration diagram for DerivationFramework::JetMSVAugmentation:

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

Private Attributes
SG::ReadHandleKey< xAOD::JetContainer >	m_jetCollectionName {this, "JetCollectionName", "AntiKt4EMTopoJets"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxmass {this, "vtxmassDecorKey", m_jetCollectionName, "MSV_vtxmass"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxpt {this, "vtxptDecorKey", m_jetCollectionName, "MSV_vtxpt"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxeta {this, "vtxetaDecorKey", m_jetCollectionName, "MSV_vtxeta"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxphi {this, "vtxphiDecorKey", m_jetCollectionName, "MSV_vtxphi"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxefrac {this, "vtxefracDecorKey", m_jetCollectionName, "MSV_vtxefrac"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxx {this, "vtxxDecorKey", m_jetCollectionName, "MSV_vtxx"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxy {this, "vtxyDecorKey", m_jetCollectionName, "MSV_vtxy"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxz {this, "vtxzDecorKey", m_jetCollectionName, "MSV_vtxz"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxntrk {this, "vtxntrkDecorKey", m_jetCollectionName, "MSV_vtxntrk"}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_dec_vtxdls {this, "vtxdlsDecorKey", m_jetCollectionName, "MSV_vtxdls"}
Gaudi::Property< std::string >	m_vtxAlgName {this, "vertexAlgName", "MSV"}

Detailed Description

Definition at line 27 of file JetMSVAugmentation.h.

Constructor & Destructor Documentation

JetMSVAugmentation()

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

Definition at line 17 of file JetMSVAugmentation.cxx.

                                                                                                 :
  base_class(t,n,p)
{
}

~JetMSVAugmentation()

DerivationFramework::JetMSVAugmentation::~JetMSVAugmentation ( )

default

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 38 of file JetMSVAugmentation.cxx.

                                                                       {
 
  SG::ReadHandle<xAOD::JetContainer> jets{m_jetCollectionName, ctx};
  if ( !jets.isValid() ) {
    ATH_MSG_ERROR ("Couldn't retrieve jets with key: " << m_jetCollectionName );
    return StatusCode::FAILURE;
  }
 
 
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxmass(m_dec_vtxmass, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxpt(m_dec_vtxpt, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxeta(m_dec_vtxeta, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxphi(m_dec_vtxphi, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxefrac(m_dec_vtxefrac, ctx);
 
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxx(m_dec_vtxx, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxy(m_dec_vtxy, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxz(m_dec_vtxz, ctx);
 
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<int>> dec_vtxntrk(m_dec_vtxntrk, ctx);
  SG::WriteDecorHandle<xAOD::JetContainer, std::vector<float>> dec_vtxdls(m_dec_vtxdls, ctx);
 
 
  for (auto jet : *jets) {
    const xAOD::BTagging* bjet = xAOD::BTaggingUtilities::getBTagging( *jet );
    if (!bjet) {
      ATH_MSG_WARNING("btagging information not available" );
      continue;
    }
 
    std::vector< ElementLink< xAOD::VertexContainer > > msvVertices;
    bjet->variable<std::vector<ElementLink<xAOD::VertexContainer> > >(m_vtxAlgName, "vertices", msvVertices);
 
    std::vector<float> vtx_mass;
    std::vector<float> vtx_pt;
    std::vector<float> vtx_eta;
    std::vector<float> vtx_phi;
    std::vector<float> vtx_efrac;
    std::vector<float> vtx_x;
    std::vector<float> vtx_y;
    std::vector<float> vtx_z;
    std::vector<int> vtx_ntrk;
    std::vector<float> vtx_dls;
 
 
    for (auto vtx : msvVertices) {//loop in vertices
 
      int   ntrk = xAOD::SecVtxHelper::VtxNtrk(*vtx);
      float mass = xAOD::SecVtxHelper::VertexMass(*vtx);
      float efrc = xAOD::SecVtxHelper::EnergyFraction(*vtx);
      float pt   = xAOD::SecVtxHelper::Vtxpt(*vtx);
      float eta  = xAOD::SecVtxHelper::Vtxeta(*vtx);
      float phi  = xAOD::SecVtxHelper::Vtxphi(*vtx);
      float dls  = xAOD::SecVtxHelper::VtxnormDist(*vtx);
      float xp   = (*vtx)->x();
      float yp   = (*vtx)->y();
      float zp   = (*vtx)->z();
      // float chi  = (*vtx)->chiSquared();
      // float ndf  = (*vtx)->numberDoF();
 
      TLorentzVector p;
      p.SetPtEtaPhiM(pt,eta,phi,mass);
 
      vtx_mass.push_back(mass);
      vtx_pt.push_back(pt);
      vtx_eta.push_back(eta);
      vtx_phi.push_back(phi);
      vtx_efrac.push_back(efrc);
      vtx_x.push_back(xp);
      vtx_y.push_back(yp);
      vtx_z.push_back(zp);
      vtx_ntrk.push_back(ntrk);
      vtx_dls.push_back(dls);
    }
 
    dec_vtxmass(*bjet)=vtx_mass;
    dec_vtxpt(*bjet)=vtx_pt;
    dec_vtxeta(*bjet)=vtx_eta;
    dec_vtxphi(*bjet)=vtx_phi;
    dec_vtxefrac(*bjet)=vtx_efrac;
    dec_vtxx(*bjet)=vtx_x;
    dec_vtxy(*bjet)=vtx_y;
    dec_vtxz(*bjet)=vtx_z;
    dec_vtxdls(*bjet)=vtx_dls;
    dec_vtxntrk(*bjet)=vtx_ntrk;
 
  }
 
  return StatusCode::SUCCESS;
 
}

initialize()

StatusCode DerivationFramework::JetMSVAugmentation::initialize ( )

finaloverridevirtual

Definition at line 28 of file JetMSVAugmentation.cxx.

                                         {
  ATH_MSG_DEBUG("Initialize " );
  ATH_CHECK(m_jetCollectionName.initialize());
 
  return StatusCode::SUCCESS;
 
}

Member Data Documentation

m_dec_vtxdls

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxdls {this, "vtxdlsDecorKey", m_jetCollectionName, "MSV_vtxdls"}

private

Definition at line 50 of file JetMSVAugmentation.h.

m_dec_vtxefrac

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxefrac {this, "vtxefracDecorKey", m_jetCollectionName, "MSV_vtxefrac"}

private

Definition at line 43 of file JetMSVAugmentation.h.

m_dec_vtxeta

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxeta {this, "vtxetaDecorKey", m_jetCollectionName, "MSV_vtxeta"}

private

Definition at line 41 of file JetMSVAugmentation.h.

m_dec_vtxmass

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxmass {this, "vtxmassDecorKey", m_jetCollectionName, "MSV_vtxmass"}

private

Definition at line 39 of file JetMSVAugmentation.h.

m_dec_vtxntrk

SG::WriteDecorHandleKey<xAOD::JetContainer > DerivationFramework::JetMSVAugmentation::m_dec_vtxntrk {this, "vtxntrkDecorKey", m_jetCollectionName, "MSV_vtxntrk"}

private

Definition at line 49 of file JetMSVAugmentation.h.

m_dec_vtxphi

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxphi {this, "vtxphiDecorKey", m_jetCollectionName, "MSV_vtxphi"}

private

Definition at line 42 of file JetMSVAugmentation.h.

m_dec_vtxpt

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxpt {this, "vtxptDecorKey", m_jetCollectionName, "MSV_vtxpt"}

private

Definition at line 40 of file JetMSVAugmentation.h.

m_dec_vtxx

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxx {this, "vtxxDecorKey", m_jetCollectionName, "MSV_vtxx"}

private

Definition at line 45 of file JetMSVAugmentation.h.

m_dec_vtxy

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxy {this, "vtxyDecorKey", m_jetCollectionName, "MSV_vtxy"}

private

Definition at line 46 of file JetMSVAugmentation.h.

m_dec_vtxz

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_dec_vtxz {this, "vtxzDecorKey", m_jetCollectionName, "MSV_vtxz"}

private

Definition at line 47 of file JetMSVAugmentation.h.

m_jetCollectionName

SG::ReadHandleKey<xAOD::JetContainer> DerivationFramework::JetMSVAugmentation::m_jetCollectionName {this, "JetCollectionName", "AntiKt4EMTopoJets"}

private

Definition at line 37 of file JetMSVAugmentation.h.

m_vtxAlgName

Gaudi::Property<std::string> DerivationFramework::JetMSVAugmentation::m_vtxAlgName {this, "vertexAlgName", "MSV"}

private

Definition at line 52 of file JetMSVAugmentation.h.

---

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

• JetMSVAugmentation.h
• JetMSVAugmentation.cxx