TCCHelpers::WeightsCollector Struct Reference

Implement a concrete CombinedUFOLoop dedicated to collection of energy sharing weights for UFO see TCCHelpers.h in TrackCaloClusterRecTools/. More...

Inheritance diagram for TCCHelpers::WeightsCollector:

Collaboration diagram for TCCHelpers::WeightsCollector:

Public Member Functions
virtual void	processPFO (const xAOD::TrackParticle *trk, const xAOD::FlowElement *pfo)
virtual void	processTrk (const xAOD::TrackParticle *)
virtual void	combinedUFOLoop (const TrackCaloClusterInfo *tccInfo, const xAOD::FlowElementContainer *pfos)

Public Attributes
std::multimap< const xAOD::IParticle *, const xAOD::TrackParticle * >	pfoToTracksMap
TrackCaloClusterInfo *	tccInfo_nonconst = nullptr
const xAOD::IParticle::FourMom_t	nullV = {0,0,0,0}
std::string	m_orig_pfoK
std::string	m_orig_FlowElemK
std::string	m_clustersLinkK
const CP::ITrackVertexAssociationTool *	m_trackVertexAssoTool
float	m_clusterEcut
bool	m_useEnergy
const SG::ReadDecorHandleKey< xAOD::TrackParticleContainer > *	m_linkdecorkey =nullptr

Detailed Description

Implement a concrete CombinedUFOLoop dedicated to collection of energy sharing weights for UFO see TCCHelpers.h in TrackCaloClusterRecTools/.

Definition at line 108 of file TrackCaloClusterInfoAlg.cxx.

Member Function Documentation

combinedUFOLoop()

virtual void TCCHelpers::CombinedUFOLoop::combinedUFOLoop ( const TrackCaloClusterInfo * tccInfo, const xAOD::FlowElementContainer * pfos )

inlinevirtualinherited

Definition at line 54 of file TCCHelpers.h.

                                                                                                           {
      SG::AuxElement::ConstAccessor<ElementLink<xAOD::IParticleContainer> > orig_pfo(m_orig_pfoK);
      SG::AuxElement::ConstAccessor< std::vector<ElementLink<xAOD::CaloClusterContainer> > > clusterLinks(m_clustersLinkK);
      
      
      
      // For performance reasons, we create a map<track, bool> before looping on the tracks.
      std::map<const xAOD::TrackParticle*, bool> isIsolatedMatchedTrack;
      for(const xAOD::FlowElement* pfo: *pfos ){
    if( ! pfo->isCharged() ) continue;
            
    static const SG::AuxElement::ConstAccessor<int> acc_isInDenseEnvironment("IsInDenseEnvironment");
    if(!acc_isInDenseEnvironment.isAvailable(*pfo)){
        asg::msgUserCode::ANA_MSG_ERROR("TCCHelpers.h: Dense environ variable not found, skipping given FE");
        continue;
    }
    if(acc_isInDenseEnvironment(*pfo)==1){
        continue;
    }
    const xAOD::IParticle* pfo_iparticle=pfo->chargedObjects().at(0);
    //cast to track, since output of chargedObjects is an IParticle
    const xAOD::TrackParticle* pfoTrack=dynamic_cast<const xAOD::TrackParticle*>(pfo_iparticle);
    isIsolatedMatchedTrack[pfoTrack] = true;
      }
 
      if(m_linkdecorkey==nullptr){
      asg::msgUserCode::ANA_MSG_ERROR("TCCHelpers.h m_linkdecorkey is not initialised in the struct to do anything, not sure what this means? Skip this case");
      return;
      }
      const EventContext& ctx =Gaudi::Hive::currentContext();
      SG::ReadDecorHandle<xAOD::TrackParticleContainer, std::vector<ElementLink<xAOD::CaloClusterContainer>> > trackcalodecor(*m_linkdecorkey,ctx); //init the handle
 
      if(!trackcalodecor.isPresent()){      
    asg::msgUserCode::ANA_MSG_ERROR("TCCHelpers.h: no link between track and cluster. We need these links : "<<m_linkdecorkey->key());
      }
      
      for(const xAOD::TrackParticle * trk: *tccInfo->allTracks){
    // Only include tracks which aren't associated to a charged PFO -- don't want to double count!
    if (isIsolatedMatchedTrack.find(trk) != isIsolatedMatchedTrack.end() ) continue;
 
    // if not matched to PV0, ignore track
    if (! m_trackVertexAssoTool->isCompatible(*trk, *tccInfo->pv0) ) continue ;
    //access the link ReadDecorHandle (saved as struct), and then apply it
    
    
    
    
    
        const auto& clustLinks=trackcalodecor(*trk);
    if(clustLinks.empty() ){ 
      continue;
    }   
 
    
    // follow the link to the calorimeter clusters
    for( const auto& clLink : clustLinks) {
        
      const xAOD::CaloCluster* cluster = *clLink;
      
      // See if any of the PFO are matched to this cluster
      for ( const xAOD::FlowElement* pfo : *pfos ){
        if(pfo->pt()<=0.) continue;
        
        const xAOD::IParticle* pfo_orig = nullptr;
        if ( orig_pfo.isAvailable(*pfo) ) pfo_orig = *orig_pfo( *pfo ) ;
        else pfo_orig = pfo;
        
        if( !(std::abs(pfo_orig->eta() - cluster->rawEta() )<0.01 && std::abs( pfo_orig->phi()  - cluster->rawPhi() )<0.01)) continue; // temp disable origin logic. Matt A: not 100% sure if its needed now anyway       
        
        if( ! pfo->isCharged() ){
          // Only want to apply this to PFO which aren't able to be subtracted with E/p
          if(cluster->rawE() < m_clusterEcut) continue;
        }
 
        // this pfo is matched to a cluster associated to the track -->
        // Call the specialized function to accumulate either :
        // - E sharing weights (ex: in TrackCaloClusterInfoAlg.cxx)
        // - clusters/pfo mometum associated to the current trk (ex: TrackCaloClusterTool)
        processPFO(trk, pfo);
 
      } // pfo loop
    } // cluster associated to trk loop
 
    // When building UFO, the specialized func below will create a combined UFO from this
    // track and the accumulated momentum from the associated cluster/pfo
    processTrk(trk);
    
      } // track loop
 
      
    }

processPFO()

virtual void TCCHelpers::WeightsCollector::processPFO ( const xAOD::TrackParticle * trk, const xAOD::FlowElement * pfo )

inlinevirtual

collect pfo momentum and register the pair (trk,pfo) in the relevant tccInfo map

Implements TCCHelpers::CombinedUFOLoop.

Definition at line 116 of file TrackCaloClusterInfoAlg.cxx.

                                                                                        {
      pfoToTracksMap.insert(std::make_pair(pfo, trk));
      // get the total p4 for this track, creating it from nullV  if not existing.
      xAOD::IParticle::FourMom_t & totalP4 = ( tccInfo_nonconst->trackTotalClusterPt.insert( {trk, nullV} ) ).first->second;
      totalP4 += pfo->p4(); // this modifies the totalP4 stored inside the map (reference access)
    }

processTrk()

virtual void TCCHelpers::WeightsCollector::processTrk ( const xAOD::TrackParticle * )

inlinevirtual

Implements TCCHelpers::CombinedUFOLoop.

Definition at line 124 of file TrackCaloClusterInfoAlg.cxx.

{ }

Member Data Documentation

m_clusterEcut

float TCCHelpers::CombinedUFOLoop::m_clusterEcut

inherited

Definition at line 47 of file TCCHelpers.h.

m_clustersLinkK

std::string TCCHelpers::CombinedUFOLoop::m_clustersLinkK

inherited

Definition at line 42 of file TCCHelpers.h.

m_linkdecorkey

const SG::ReadDecorHandleKey<xAOD::TrackParticleContainer>* TCCHelpers::CombinedUFOLoop::m_linkdecorkey =nullptr

inherited

Definition at line 49 of file TCCHelpers.h.

m_orig_FlowElemK

std::string TCCHelpers::CombinedUFOLoop::m_orig_FlowElemK

inherited

Definition at line 41 of file TCCHelpers.h.

m_orig_pfoK

std::string TCCHelpers::CombinedUFOLoop::m_orig_pfoK

inherited

Definition at line 40 of file TCCHelpers.h.

m_trackVertexAssoTool

const CP::ITrackVertexAssociationTool* TCCHelpers::CombinedUFOLoop::m_trackVertexAssoTool

inherited

Definition at line 46 of file TCCHelpers.h.

m_useEnergy

bool TCCHelpers::CombinedUFOLoop::m_useEnergy

inherited

Definition at line 48 of file TCCHelpers.h.

nullV

const xAOD::IParticle::FourMom_t TCCHelpers::WeightsCollector::nullV = {0,0,0,0}

Definition at line 114 of file TrackCaloClusterInfoAlg.cxx.

{0,0,0,0};

pfoToTracksMap

std::multimap<const xAOD::IParticle*, const xAOD::TrackParticle*> TCCHelpers::WeightsCollector::pfoToTracksMap

Definition at line 111 of file TrackCaloClusterInfoAlg.cxx.

tccInfo_nonconst

TrackCaloClusterInfo* TCCHelpers::WeightsCollector::tccInfo_nonconst = nullptr

Definition at line 112 of file TrackCaloClusterInfoAlg.cxx.

---

The documentation for this struct was generated from the following file:

• TrackCaloClusterInfoAlg.cxx