DerivationFramework::JetExternalAssocTool Class Reference

#include <JetExternalAssocTool.h>

Inheritance diagram for DerivationFramework::JetExternalAssocTool:

Collaboration diagram for DerivationFramework::JetExternalAssocTool:

Public Member Functions
	JetExternalAssocTool (const std::string &t, const std::string &n, const IInterface *p)
StatusCode	initialize ()
StatusCode	finalize ()
virtual StatusCode	addBranches () const
bool	TransferLink (const xAOD::Jet &jet, const xAOD::Jet &jet_external) const

Private Types
typedef ElementLink< xAOD::IParticleContainer >	type_el
	decoration pointers More...
typedef std::vector< type_el >	type_ghostlink

Private Attributes
std::string	m_momentPrefix
	Properties. More...
std::string	m_containerName
std::string	m_ExternalJetCollectionName
std::vector< std::string >	m_VectorOfOldLinkNames
std::vector< std::string >	m_VectorOfNewLinkNames
bool	m_dRMatch
double	m_dRCut
SG::WriteDecorHandleKeyArray< xAOD::JetContainer >	m_dec_keys {this, "DecKeys", {}, "SG keys for external decorations"}

Detailed Description

Definition at line 31 of file JetExternalAssocTool.h.

Member Typedef Documentation

type_el

typedef ElementLink<xAOD::IParticleContainer> DerivationFramework::JetExternalAssocTool::type_el

private

decoration pointers

Definition at line 55 of file JetExternalAssocTool.h.

type_ghostlink

typedef std::vector<type_el> DerivationFramework::JetExternalAssocTool::type_ghostlink

private

Definition at line 56 of file JetExternalAssocTool.h.

Constructor & Destructor Documentation

JetExternalAssocTool()

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

Definition at line 14 of file JetExternalAssocTool.cxx.

                                                                                                     :
  base_class(t,n,p),
  m_momentPrefix(""),
  m_containerName(""),
  m_ExternalJetCollectionName(""),
  m_VectorOfOldLinkNames({}),
  m_VectorOfNewLinkNames({}),
  m_dRMatch(false),
  m_dRCut(0.01)
{
 
  declareProperty("MomentPrefix"             , m_momentPrefix);
  declareProperty("InputJets"                , m_containerName);
 
  declareProperty("ExternalJetCollectionName", m_ExternalJetCollectionName);
  declareProperty("ListOfOldLinkNames"       , m_VectorOfOldLinkNames);
  declareProperty("ListOfNewLinkNames"       , m_VectorOfNewLinkNames);
 
  declareProperty("DeltaRMatch"              , m_dRMatch);
  declareProperty("DeltaRCut"                , m_dRCut);
}

Member Function Documentation

addBranches()

StatusCode DerivationFramework::JetExternalAssocTool::addBranches ( ) const

virtual

Definition at line 66 of file JetExternalAssocTool.cxx.

                                                  {
 
  // get jet collection to be decorated
  const xAOD::JetContainer* jets = 0;
  if(evtStore()->retrieve(jets, m_containerName).isFailure()){
    ATH_MSG_ERROR("Unable to retrieve jet collection: " << m_containerName << "!");
    return StatusCode::FAILURE;
  }
 
  // get external jet collection
  const xAOD::JetContainer* ExternalJetCollection = 0;
  if(evtStore()->retrieve(ExternalJetCollection, m_ExternalJetCollectionName).isFailure()){
    ATH_MSG_ERROR("Unable to find external jet collection: " << m_ExternalJetCollectionName << "!");
    return StatusCode::FAILURE;
  }
 
  if(!m_dRMatch){
    // Here we are assuming there is a strict one-to-one correspondence between two jet collections and both jet collections have been sorted by pT
 
    // basic sanity check
    if(jets->size() != ExternalJetCollection->size()){
      ATH_MSG_ERROR("Size not the same between collection " << m_containerName << " and " << m_ExternalJetCollectionName << ". Please either double check your jet collections, or turn on dRMatch option");
      return StatusCode::FAILURE;
    }
 
    for(unsigned int index = 0; index < jets->size(); index++){
      auto jet = jets->at(index);
      auto jet_external = ExternalJetCollection->at(index);
 
      if( std::abs(jet->pt() - jet_external->pt()) > 0.1 ){ // 0.1 is hard-coded in
        ATH_MSG_WARNING("Potential inconsistency between two jet collections: " << jet->pt() << " v.s. " << jet_external->pt());
      }
 
      if(!TransferLink(*jet, *jet_external)){
        ATH_MSG_ERROR("Failure when transferring link from external jet to current jet!");
        return StatusCode::FAILURE;
      }
    }
  }
  else{
    // simple dR matching
 
    // initialize list of un-assigned external jets
    std::vector<const xAOD::Jet*> UnAssignedExternalJets;
    for(auto jet_external : *ExternalJetCollection){
      UnAssignedExternalJets.push_back(jet_external);
    }
 
    // loop
    for(auto jet : *jets){
      // get associated jet
      double mindR = 9e9;
      const xAOD::Jet* associated_jet = 0;
      std::vector<const xAOD::Jet*>::iterator associated_iter = UnAssignedExternalJets.begin();  // random assignment
 
      for(auto external_iter = UnAssignedExternalJets.begin(); external_iter != UnAssignedExternalJets.end(); external_iter++){
        auto jet_external = (*external_iter);
 
        double dR = jet->p4().DeltaR(jet_external->p4());
        if(dR > m_dRCut) continue;
        if(dR < mindR){
          mindR = dR;
          associated_jet = jet_external;
          associated_iter = external_iter;
        }
      }
 
      if(associated_jet == 0){
        ATH_MSG_WARNING("Unable to find associated external jet! This jet will be skipped");
        continue;
      }
      else{
        // sanity check
        if(associated_jet != (*associated_iter)){
          ATH_MSG_ERROR("Sanity check after association fails!");
          return StatusCode::FAILURE;
        }
 
        // remove associated jet from unasigned jet list
        UnAssignedExternalJets.erase(associated_iter);
      }
 
      // transfer the link
      if(!TransferLink(*jet, *associated_jet)){
        ATH_MSG_ERROR("Failure when transferring link from external jet to current jet!");
        return StatusCode::FAILURE;
      }
    }
  }
 
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::JetExternalAssocTool::finalize

(

)

Definition at line 59 of file JetExternalAssocTool.cxx.

                                         {
 
  return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::JetExternalAssocTool::initialize

(

)

Definition at line 38 of file JetExternalAssocTool.cxx.

                                            {
 
  // get link names
  if(m_VectorOfNewLinkNames.empty()) m_VectorOfNewLinkNames = m_VectorOfOldLinkNames;
 
  // sanity check
  if(m_VectorOfNewLinkNames.size() != m_VectorOfOldLinkNames.size()){
    ATH_MSG_ERROR("Vector size between old and new link names does not agree!");
    return StatusCode::FAILURE;
  }
 
  // setup vector of decorator
  for(const std::string& NewLinkName : m_VectorOfNewLinkNames){
    m_dec_keys.emplace_back(  m_containerName + "." + m_momentPrefix + NewLinkName);
  }
 
  ATH_CHECK(m_dec_keys.initialize());
 
  return StatusCode::SUCCESS;
}

TransferLink()

bool DerivationFramework::JetExternalAssocTool::TransferLink	(	const xAOD::Jet &	jet,
		const xAOD::Jet &	jet_external
	)		const

Definition at line 159 of file JetExternalAssocTool.cxx.

                                                                                            {
 
  for(unsigned int index_link = 0; index_link < m_VectorOfOldLinkNames.size(); index_link++){
    auto OldLinkName = m_VectorOfOldLinkNames[index_link];
    auto NewLinkName = m_VectorOfNewLinkNames[index_link];
 
    // assume we are always dealing a list of IParticles
    std::vector<const xAOD::IParticle*> targetObjs;
    if(!jet_external.getAssociatedObjects<xAOD::IParticle>(OldLinkName, targetObjs)){
      ATH_MSG_WARNING("Unable to fetch link " << OldLinkName << " under associated external jet");
      continue;
    }
 
    // put it under my jet
    type_ghostlink targetLinks;
    for(auto obj : targetObjs){
      type_el el_obj;
      el_obj.toIndexedElement(*(static_cast<const xAOD::IParticleContainer*>(obj->container())), obj->index());
 
      targetLinks.push_back(el_obj);
    }
 
    SG::WriteDecorHandle<xAOD::JetContainer, type_ghostlink*> dec_handle(m_dec_keys.at(index_link));
    *dec_handle(jet) = targetLinks;
 
  }
 
  return true;
}

Member Data Documentation

m_containerName

std::string DerivationFramework::JetExternalAssocTool::m_containerName

private

Definition at line 45 of file JetExternalAssocTool.h.

m_dec_keys

SG::WriteDecorHandleKeyArray<xAOD::JetContainer> DerivationFramework::JetExternalAssocTool::m_dec_keys {this, "DecKeys", {}, "SG keys for external decorations"}

private

Definition at line 58 of file JetExternalAssocTool.h.

m_dRCut

double DerivationFramework::JetExternalAssocTool::m_dRCut

private

Definition at line 52 of file JetExternalAssocTool.h.

m_dRMatch

bool DerivationFramework::JetExternalAssocTool::m_dRMatch

private

Definition at line 51 of file JetExternalAssocTool.h.

m_ExternalJetCollectionName

std::string DerivationFramework::JetExternalAssocTool::m_ExternalJetCollectionName

private

Definition at line 47 of file JetExternalAssocTool.h.

m_momentPrefix

std::string DerivationFramework::JetExternalAssocTool::m_momentPrefix

private

Properties.

Definition at line 44 of file JetExternalAssocTool.h.

m_VectorOfNewLinkNames

std::vector<std::string> DerivationFramework::JetExternalAssocTool::m_VectorOfNewLinkNames

private

Definition at line 49 of file JetExternalAssocTool.h.

m_VectorOfOldLinkNames

std::vector<std::string> DerivationFramework::JetExternalAssocTool::m_VectorOfOldLinkNames

private

Definition at line 48 of file JetExternalAssocTool.h.

---

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

• JetExternalAssocTool.h
• JetExternalAssocTool.cxx