DerivationFramework::DiphotonVertexDecorator Class Reference

#include <DiphotonVertexDecorator.h>

Inheritance diagram for DerivationFramework::DiphotonVertexDecorator:

Collaboration diagram for DerivationFramework::DiphotonVertexDecorator:

Public Member Functions
	DiphotonVertexDecorator (const std::string &t, const std::string &n, const IInterface *p)
	Constructor with parameters. More...
	~DiphotonVertexDecorator ()
	Destructor. More...
StatusCode	initialize ()
StatusCode	finalize ()
virtual StatusCode	addBranches (const EventContext &ctx) const

Private Member Functions
bool	PhotonPreselect (const xAOD::Photon *ph) const
StatusCode	matchFlowElement (const xAOD::Photon *eg, const xAOD::FlowElementContainer *pfoCont) const

Static Private Member Functions
static bool	greaterPtFlowElement (const xAOD::FlowElement *part1, const xAOD::FlowElement *part2)

Private Attributes
ToolHandle< CP::IPhotonVertexSelectionTool >	m_photonVertexSelectionTool {this, "PhotonVertexSelectionTool", "", ""}
SG::ReadHandleKey< xAOD::VertexContainer >	m_primaryVertexKey {this, "PrimaryVertexName", "PrimaryVertices", "" }
SG::ReadHandleKey< xAOD::PhotonContainer >	m_photonKey { this, "PhotonKey", "Photons", "" }
SG::WriteHandleKey< xAOD::VertexContainer >	m_diphotonVertexKey {this, "DiphotonVertexName", "HggPrimaryVertices", "" }
SG::ReadHandleKey< xAOD::FlowElementContainer >	m_FEContainerHandleKey {this,"PFOContainerName","JetETMissChargedParticleFlowObjects","ReadHandleKey for the PFO container"}
double	m_minPhotonPt
bool	m_removeCrack
double	m_maxEta
bool	m_ignoreConv
std::string	m_pfoToolName
double	m_tcMatch_dR
double	m_tcMatch_maxRat

Detailed Description

Author

Bruno Lenzi

Leo Cerda

magda.nosp@m.c@ce.nosp@m.rn.ch

Definition at line 37 of file DiphotonVertexDecorator.h.

Constructor & Destructor Documentation

DiphotonVertexDecorator()

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

Constructor with parameters.

Definition at line 29 of file DiphotonVertexDecorator.cxx.

                                                     : 
  base_class(t, n, p)
{
 
 
  declareProperty("RemoveCrack",           m_removeCrack    = true);
  declareProperty("MaxEta",                m_maxEta         = 2.37);
  declareProperty("MinimumPhotonPt",       m_minPhotonPt    = 20*CLHEP::GeV);
  declareProperty("IgnoreConvPointing",    m_ignoreConv     = false);
  declareProperty("TCMatchMaxRat",         m_tcMatch_maxRat = 1.5    );
  declareProperty("TCMatchDeltaR",         m_tcMatch_dR     = 0.1    );
 
}

~DiphotonVertexDecorator()

DerivationFramework::DiphotonVertexDecorator::~DiphotonVertexDecorator

(

)

Destructor.

Definition at line 46 of file DiphotonVertexDecorator.cxx.

                                                                     {
}  

Member Function Documentation

addBranches()

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

virtual

Definition at line 65 of file DiphotonVertexDecorator.cxx.

{
  ATH_MSG_DEBUG( "DiphotonVertexDecorator::AddingBranches" );
  
  SG::ReadHandle<xAOD::VertexContainer> PV (m_primaryVertexKey, ctx);
 
  if (!PV->empty() && PV->at(0)) {
    ATH_MSG_DEBUG( "Default PV " << PV->at(0) << ", type = " << PV->at(0)->vertexType() << " , z = " << PV->at(0)->z()  );
  } 
  
  // Select the two highest pt photons that pass a preselection
 
  SG::ReadHandle<xAOD::PhotonContainer> photons (m_photonKey, ctx);
  const xAOD::Photon *ph1 = nullptr, *ph2 = nullptr;
 
  for (const xAOD::Photon* ph: *photons)
  {
    if (!PhotonPreselect(ph)) continue;
    if (not ph1 or ph->pt() > ph1->pt()) // new leading photon
    {
      ph2 = ph1;
      ph1 = ph;
    }
    else if (not ph2 or ph->pt() > ph2->pt()) ph2 = ph; // new subleading photon
  }
 
  const ConstDataVector< xAOD::PhotonContainer > vertexPhotons = {ph1, ph2};
 
  CP::IPhotonVertexSelectionTool::FailType vertexFailType;
  CP::IPhotonVertexSelectionTool::yyVtxType yyvertexVtxType = CP::IPhotonVertexSelectionTool::Unknown;
 
  ATH_CHECK( m_photonVertexSelectionTool->decorateInputs(*(vertexPhotons.asDataVector()), &vertexFailType) );
 
  // Get the photon vertex if possible
  std::vector<std::pair<const xAOD::Vertex*, float> > vxResult;
  const xAOD::Vertex *newPV = nullptr;
 
  SG::ReadHandle<xAOD::FlowElementContainer> FEHandle(m_FEContainerHandleKey, ctx);
  SG::Decorator<char> passORDec("passOR");
  for(const auto *const fe : *FEHandle) passORDec(*fe) = true;
  
  if (ph1 and ph2)
  {
    vxResult = m_photonVertexSelectionTool->getVertex( *( vertexPhotons.asDataVector()) , m_ignoreConv, true, &yyvertexVtxType, &vertexFailType );
    if(!vxResult.empty()) {
      newPV = vxResult[0].first; //output of photon vertex selection tool must be sorted according to score
    }
    ATH_CHECK(matchFlowElement(ph1,&*FEHandle));
    ATH_CHECK(matchFlowElement(ph2,&*FEHandle));
  }
 
  // Decorate the vertices with the NN score
  ATH_MSG_DEBUG("PhotonVertexSelection returns vertex " << newPV << " " << (newPV? Form(" with z = %g", newPV->z()) : "") );
  // Create shallow copy of the PrimaryVertices container
  std::pair< xAOD::VertexContainer*, xAOD::ShallowAuxContainer* > HggPV = xAOD::shallowCopyContainer( *PV );
  HggPV.second->setShallowIO(false);
 
  SG::WriteHandle<xAOD::VertexContainer> vertexContainer(m_diphotonVertexKey, ctx);
  ATH_CHECK(vertexContainer.recordNonConst(std::unique_ptr< xAOD::VertexContainer >(HggPV.first),
                                           std::unique_ptr< xAOD::ShallowAuxContainer >(HggPV.second)));
 
 
  static const SG::Accessor<float> vertexScoreAcc("vertexScore");
  static const SG::Accessor<int> vertexFailTypeAcc("vertexFailType");
  static const SG::Accessor<int> vertexCaseAcc("vertexCase");
  static const SG::Accessor<phlink_t> leadingPhotonLinkAcc("leadingPhotonLink");
  static const SG::Accessor<phlink_t> subleadingPhotonLinkAcc("subleadingPhotonLink");
 
  if (newPV) {
    //loop over vertex container; shallow copy has the same order
    for (unsigned int iPV=0; iPV<PV->size(); iPV++) {
      const auto *vx = PV->at(iPV);
      auto yyvx = (HggPV.first)->at(iPV);
      //reset vertex type
      if (vx == newPV) { 
        //is this the diphoton primary vertex returned from the tool?
        yyvx->setVertexType( xAOD::VxType::PriVtx );
      } else if ( vx->vertexType()==xAOD::VxType::PriVtx || vx->vertexType()==xAOD::VxType::PileUp ) {
        //not overriding the type of dummy vertices of type 0 (NoVtx)
        yyvx->setVertexType( xAOD::VxType::PileUp );
      }
      //decorate score
      for (const auto& vxR: vxResult) {
        //find vertex in output from photonVertexSelectionTool
        if ( vx == vxR.first ) {
          vertexScoreAcc(*yyvx) = vxR.second;
          vertexFailTypeAcc(*yyvx) = vertexFailType;
          vertexCaseAcc(*yyvx) = yyvertexVtxType;
          leadingPhotonLinkAcc(*yyvx) = phlink_t(*photons, ph1->index());
          subleadingPhotonLinkAcc(*yyvx) = phlink_t(*photons, ph2->index());
          break;
        }
      }
    }
  }
  else {
    //no vertex returned by photonVertexSelectionTool, decorate default PV with fit information
    xAOD::VertexContainer::iterator yyvx_itr;
    xAOD::VertexContainer::iterator yyvx_end = (HggPV.first)->end();
    for(yyvx_itr = (HggPV.first)->begin(); yyvx_itr != yyvx_end; ++yyvx_itr ) {
      if ( (*yyvx_itr)->vertexType()==xAOD::VxType::PriVtx ) {
        vertexScoreAcc(**yyvx_itr) = -9999;
        vertexFailTypeAcc(**yyvx_itr) = vertexFailType;
        vertexCaseAcc(**yyvx_itr) = yyvertexVtxType;
        leadingPhotonLinkAcc(**yyvx_itr) = (phlink_t()) ;
        subleadingPhotonLinkAcc(**yyvx_itr) = (phlink_t());
      }
    }
  }
 
 
  if( !evtStore()->transientContains< xAOD::VertexContainer >( m_diphotonVertexKey.key() ) ){
    ATH_MSG_WARNING("Unable to find transient xAOD::VertexContainer, \"" << m_diphotonVertexKey.key() << "\"");
  } 
 
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::DiphotonVertexDecorator::finalize

(

)

Definition at line 60 of file DiphotonVertexDecorator.cxx.

{
  return StatusCode::SUCCESS;
}

greaterPtFlowElement()

static bool DerivationFramework::DiphotonVertexDecorator::greaterPtFlowElement ( const xAOD::FlowElement *  part1, const xAOD::FlowElement *  part2  )

inlinestaticprivate

Definition at line 75 of file DiphotonVertexDecorator.h.

                                                                                                          {
        if (part1->charge()==0 && part2->charge()!=0) return false;
        if (part1->charge()!=0 && part2->charge()==0) return true;
        return part1->pt()>part2->pt();
    }

initialize()

StatusCode DerivationFramework::DiphotonVertexDecorator::initialize

(

)

Definition at line 50 of file DiphotonVertexDecorator.cxx.

{
  ATH_CHECK( m_photonVertexSelectionTool.retrieve() );
  ATH_CHECK( m_primaryVertexKey.initialize() );
  ATH_CHECK( m_photonKey.initialize() );
  ATH_CHECK( m_diphotonVertexKey.initialize() );
  ATH_CHECK( m_FEContainerHandleKey.initialize() );
  return StatusCode::SUCCESS;
}

matchFlowElement()

StatusCode DerivationFramework::DiphotonVertexDecorator::matchFlowElement ( const xAOD::Photon *  eg, const xAOD::FlowElementContainer *  pfoCont  ) const

private

Definition at line 220 of file DiphotonVertexDecorator.cxx.

                                                                                                                                         {
  const xAOD::IParticle* swclus = eg->caloCluster();
 
  // Preselect FEs based on proximity: dR<0.4
  std::vector<const xAOD::FlowElement*> nearbyFE;
  nearbyFE.reserve(20);
  for(const auto *const fe : *feCont) {
    if(xAOD::P4Helpers::isInDeltaR(*fe, *swclus, 0.4, true)) {
      if( ( !fe->isCharged() && fe->e() > FLT_MIN )) nearbyFE.push_back(fe);
    } // DeltaR check
  } // FE loop
 
  SG::Decorator<char> passORDec("passOR");
 
  double eg_cl_e = swclus->e();
  bool doSum = true;
  double sumE_fe = 0.;
  const xAOD::IParticle* bestbadmatch = nullptr;
  std::sort(nearbyFE.begin(),nearbyFE.end(),greaterPtFlowElement);
  for(const auto& fe : nearbyFE) {
    if(!xAOD::P4Helpers::isInDeltaR(*fe, *swclus, m_tcMatch_dR, true)) {continue;}
    // Handle neutral FEs like topoclusters
    double fe_e = fe->e();
    // skip cluster if it's above our bad match threshold or outside the matching radius
    if(fe_e>m_tcMatch_maxRat*eg_cl_e) {
      ATH_MSG_VERBOSE("Reject topocluster in sum. Ratio vs eg cluster: " << (fe_e/eg_cl_e));
      if( !bestbadmatch || (std::abs(fe_e/eg_cl_e-1.) < std::abs(bestbadmatch->e()/eg_cl_e-1.)) ) bestbadmatch = fe;
      continue;
    }
 
    ATH_MSG_VERBOSE("E match with new nFE: " << std::abs(sumE_fe+fe_e - eg_cl_e) / eg_cl_e);
    if( (doSum = std::abs(sumE_fe+fe_e-eg_cl_e) < std::abs(sumE_fe - eg_cl_e)) ) {
       passORDec(*fe) = false;
      sumE_fe += fe_e;
    } // if we will retain the topocluster
    else {break;}
  } // loop over nearby clusters
  if(sumE_fe<FLT_MIN && bestbadmatch) {
    passORDec(*bestbadmatch) = false;
  }
 
  return StatusCode::SUCCESS;
}

PhotonPreselect()

bool DerivationFramework::DiphotonVertexDecorator::PhotonPreselect ( const xAOD::Photon *  ph ) const

private

Definition at line 183 of file DiphotonVertexDecorator.cxx.

{
 
  if (!ph) return false;
 
  if (!ph->isGoodOQ(34214)) return false;
 
  bool val(false);
  bool defined(false);
 
  static const SG::ConstAccessor<char> DFCommonPhotonsIsEMLooseAcc("DFCommonPhotonsIsEMLoose");
  if(DFCommonPhotonsIsEMLooseAcc.isAvailable(*ph)){
    defined = true;
    val = static_cast<bool>(DFCommonPhotonsIsEMLooseAcc(*ph));
  }
  else{
    defined = ph->passSelection(val, "Loose");
  }
 
  if(!defined || !val) return false;
 
  // veto topo-seeded clusters
  if (ph->author(xAOD::EgammaParameters::AuthorCaloTopo35)) return false;
 
  // Check which variable versions are best...
  const xAOD::CaloCluster *caloCluster(ph->caloCluster());
  double eta = std::abs(caloCluster->etaBE(2));
 
  if (eta > m_maxEta) return false;
  if (m_removeCrack && 1.37 <= eta && eta <= 1.52) return false;
 
  if (ph->pt() < m_minPhotonPt) return false;
  
  return true;
 
}

Member Data Documentation

m_diphotonVertexKey

SG::WriteHandleKey<xAOD::VertexContainer> DerivationFramework::DiphotonVertexDecorator::m_diphotonVertexKey {this, "DiphotonVertexName", "HggPrimaryVertices", "" }

private

Definition at line 60 of file DiphotonVertexDecorator.h.

m_FEContainerHandleKey

SG::ReadHandleKey<xAOD::FlowElementContainer> DerivationFramework::DiphotonVertexDecorator::m_FEContainerHandleKey {this,"PFOContainerName","JetETMissChargedParticleFlowObjects","ReadHandleKey for the PFO container"}

private

Definition at line 61 of file DiphotonVertexDecorator.h.

m_ignoreConv

bool DerivationFramework::DiphotonVertexDecorator::m_ignoreConv

private

Definition at line 68 of file DiphotonVertexDecorator.h.

m_maxEta

double DerivationFramework::DiphotonVertexDecorator::m_maxEta

private

Definition at line 67 of file DiphotonVertexDecorator.h.

m_minPhotonPt

double DerivationFramework::DiphotonVertexDecorator::m_minPhotonPt

private

Definition at line 65 of file DiphotonVertexDecorator.h.

m_pfoToolName

std::string DerivationFramework::DiphotonVertexDecorator::m_pfoToolName

private

Definition at line 69 of file DiphotonVertexDecorator.h.

m_photonKey

SG::ReadHandleKey<xAOD::PhotonContainer> DerivationFramework::DiphotonVertexDecorator::m_photonKey { this, "PhotonKey", "Photons", "" }

private

Definition at line 59 of file DiphotonVertexDecorator.h.

m_photonVertexSelectionTool

ToolHandle<CP::IPhotonVertexSelectionTool> DerivationFramework::DiphotonVertexDecorator::m_photonVertexSelectionTool {this, "PhotonVertexSelectionTool", "", ""}

private

Definition at line 56 of file DiphotonVertexDecorator.h.

m_primaryVertexKey

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::DiphotonVertexDecorator::m_primaryVertexKey {this, "PrimaryVertexName", "PrimaryVertices", "" }

private

Definition at line 58 of file DiphotonVertexDecorator.h.

m_removeCrack

bool DerivationFramework::DiphotonVertexDecorator::m_removeCrack

private

Definition at line 66 of file DiphotonVertexDecorator.h.

m_tcMatch_dR

double DerivationFramework::DiphotonVertexDecorator::m_tcMatch_dR

private

Definition at line 70 of file DiphotonVertexDecorator.h.

m_tcMatch_maxRat

double DerivationFramework::DiphotonVertexDecorator::m_tcMatch_maxRat

private

Definition at line 71 of file DiphotonVertexDecorator.h.

---

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

• DiphotonVertexDecorator.h
• DiphotonVertexDecorator.cxx