Plugin to create the MC truth for TTClusters. More...

#include <TTClusterAssociator.h>

Inheritance diagram for TTClusterAssociator< T >:

Public Member Functions
	TTClusterAssociator (const edm::ParameterSet &iConfig)
	Constructors. More...

	~TTClusterAssociator () override
	Destructor. More...

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	beginRun (const edm::Run &run, const edm::EventSetup &iSetup) override
	Mandatory methods. More...

void	endRun (const edm::Run &run, const edm::EventSetup &iSetup) override
	End run. More...

template<>
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup)

void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override

template<>
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup)
	Implement the producer. More...

Private Attributes
edm::EDGetTokenT< edm::DetSetVector< PixelDigiSimLink > >	digisimLinkToken

edm::EDGetTokenT< edm::SimTrackContainer >	simTrackToken

edm::Handle< edm::DetSetVector< PixelDigiSimLink > >	thePixelDigiSimLinkHandle
	Data members. More...

edm::Handle< edm::SimTrackContainer >	theSimTrackHandle

edm::ESHandle< TrackerGeometry >	theTrackerGeometry

edm::ESHandle< TrackerTopology >	theTrackerTopology

edm::EDGetTokenT< std::vector< TrackingParticle > >	tpToken

edm::Handle< std::vector< TrackingParticle > >	TrackingParticleHandle

std::vector< edm::InputTag >	TTClustersInputTags

std::vector< edm::EDGetTokenT< edmNew::DetSetVector< TTCluster< T > > > >	TTClustersTokens

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

template<typename T>
class TTClusterAssociator< T >

Plugin to create the MC truth for TTClusters.

After moving from SimDataFormats to DataFormats, the template structure of the class was maintained in order to accomodate any types other than PixelDigis in case there is such a need in the future.

Author: Nicola Pozzobon

Date: 2013, Jul 19

Definition at line 44 of file TTClusterAssociator.h.

Constructor & Destructor Documentation

template<typename T >

TTClusterAssociator< T >::TTClusterAssociator ( const edm::ParameterSet & iConfig )

explicit

Constructors.

Close class.

NOTE since pattern hit correlation must be performed within a stacked module, one must store Clusters in a proper way, providing easy access to them in a detector/member-wise way

Implementation of methods

Here, in the header file, the methods which do not depend on the specific type <T> that can fit the template. Other methods, with type-specific features, are implemented in the source file.Constructors

Definition at line 93 of file TTClusterAssociator.h.

References TTClusterAssociator< T >::digisimLinkToken, edm::ParameterSet::getParameter(), TTClusterAssociator< T >::simTrackToken, TTClusterAssociator< T >::tpToken, TTClusterAssociator< T >::TTClustersInputTags, and TTClusterAssociator< T >::TTClustersTokens.

 {
   simTrackToken = consumes< edm::SimTrackContainer >(iConfig.getParameter< edm::InputTag >( "simTrackHits" ));
   digisimLinkToken = consumes< edm::DetSetVector< PixelDigiSimLink > >(iConfig.getParameter< edm::InputTag >( "digiSimLinks" ));
   tpToken = consumes< std::vector< TrackingParticle >  >(iConfig.getParameter< edm::InputTag >( "trackingParts" ));
 
 
   TTClustersInputTags = iConfig.getParameter< std::vector< edm::InputTag > >( "TTClusters" );
 
   for ( auto iTag =  TTClustersInputTags.begin(); iTag!=  TTClustersInputTags.end(); iTag++ )
   {
     TTClustersTokens.push_back(consumes< edmNew::DetSetVector< TTCluster< T > > >(*iTag));
 
     produces< TTClusterAssociationMap< T > >( (*iTag).instance() );
   }
 }

template<typename T >

TTClusterAssociator< T >::~TTClusterAssociator ( )

override

Destructor.

Definition at line 112 of file TTClusterAssociator.h.

112 {}

Member Function Documentation

template<typename T >

void TTClusterAssociator< T >::beginRun	(	const edm::Run &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Mandatory methods.

Begin run.

Get the geometry

Print some information when loaded

Definition at line 116 of file TTClusterAssociator.h.

References edm::EventSetup::get(), TTClusterAssociator< T >::theTrackerGeometry, and TTClusterAssociator< T >::theTrackerTopology.

 {
   iSetup.get<TrackerDigiGeometryRecord>().get(theTrackerGeometry);
   iSetup.get<TrackerTopologyRcd>().get(theTrackerTopology);
 
   edm::LogInfo("TTClusterAssociator< ") << templateNameFinder< T >() << " > loaded.";
 }

template<typename T >

void TTClusterAssociator< T >::endRun	(	const edm::Run &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

End run.

Definition at line 128 of file TTClusterAssociator.h.

References iEvent, and TTClusterAssociator< T >::produce().

128 {}

template<>

void TTClusterAssociator< Ref_Phase2TrackerDigi_ >::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

private

Here, in the source file, the methods which do depend on the specific type <T> that can fit the template.

Author: Nicola Pozzobon

Date: 2013, Jul 19Implement the producer

Exit if real data

Get the PixelDigiSimLink

Get the TrackingParticles

Preliminary task: map SimTracks by TrackingParticle Prepare the map

Loop over TrackingParticles

Make the pointer to the TrackingParticle

Get the EncodedEventId

Loop over SimTracks inside TrackingParticle

Build the unique SimTrack Id (which is SimTrack ID + EncodedEventId)

End of loop over TrackingParticles

Loop over InputTags to handle multiple collections

Prepare output

Get the Clusters already stored away

Prepare the necessary maps

Loop over the input Clusters

Get the DetSets of the Clusters

Make the reference to be put in the map

Prepare the maps wrt TTCluster

Get the PixelDigiSimLink Safety check added after new digitizer (Oct 2014)

Sensor is not found in DigiSimLink. Set MC truth to NULL for all hits in this sensor. Period.

Get the Digis and loop over them

No SimLink is found by definition Then store NULL MC truth for all the digis

Go to the next sensor

Get the Digis and loop over them

Loop over PixelDigiSimLink

Find the link and, if there's not, skip

Get SimTrack Id and type

Prepare the SimTrack Unique ID

Get the corresponding TrackingParticle

Store the TrackingParticle

Prepare the maps wrt TrackingParticle

Fill the auxiliary map

In case no TrackingParticle is found, store a NULL pointer

End of loop over PixelDigiSimLink

End of loop over all the hits composing the Cluster

Check that the cluster has a non-NULL TP pointer

In case no TrackingParticle is found at all, drop the map element

Use "erase by key"

End of loop over all the TTClusters of the event

Clean the maps that need cleaning Prepare the output map wrt TrackingParticle

Get the vector of references to TTCluster

Sort and remove duplicates

Put the maps in the association object

Put output in the event

End of loop over input tags

Definition at line 14 of file TTClusterAssociator.cc.

References edmNew::DetSet< T >::begin(), fastPrimaryVertexProducer_cfi::clusters, edm::DetSet< T >::data, TrackerGeometry::dets(), edmNew::DetSet< T >::end(), TrackingParticle::eventId(), funct::false, TrackingParticle::g4Tracks(), edm::Event::getByToken(), mps_fire::i, edm::EventBase::isRealData(), edmNew::makeRefTo(), eostools::move(), edm::Event::put(), DetId::subdetId(), StripSubdetector::TID, StripSubdetector::TOB, and tier0::unique().

 {
   if ( iEvent.isRealData() )
     return;
 
   iEvent.getByToken(digisimLinkToken, thePixelDigiSimLinkHandle);
 
 
   iEvent.getByToken(tpToken, TrackingParticleHandle );
 
   //  const TrackerTopology* const tTopo = theTrackerTopology.product();
   const TrackerGeometry* const theTrackerGeom = theTrackerGeometry.product();
 
 
   std::map< std::pair< unsigned int, EncodedEventId >, edm::Ptr< TrackingParticle > > simTrackUniqueToTPMap;
   simTrackUniqueToTPMap.clear();
 
   if ( !TrackingParticleHandle->empty() )
   {
     unsigned int tpCnt = 0;
     std::vector< TrackingParticle >::const_iterator iterTPart;
     for ( iterTPart = TrackingParticleHandle->begin();
           iterTPart != TrackingParticleHandle->end();
           ++iterTPart )
     {
       edm::Ptr< TrackingParticle > tempTPPtr( TrackingParticleHandle, tpCnt++ );
 
       EncodedEventId eventId = EncodedEventId( tempTPPtr->eventId() );
 
       std::vector< SimTrack >::const_iterator iterSimTrack;
       for ( iterSimTrack = tempTPPtr->g4Tracks().begin();
             iterSimTrack != tempTPPtr->g4Tracks().end();
             ++iterSimTrack )
       {
         std::pair< unsigned int, EncodedEventId > simTrackUniqueId( iterSimTrack->trackId(), eventId );
         simTrackUniqueToTPMap.insert( std::make_pair( simTrackUniqueId, tempTPPtr ) );
       }
     } 
   }
 
 
   int ncont1=0;
 
   for ( auto iTag =  TTClustersTokens.begin(); iTag!=  TTClustersTokens.end(); iTag++ )
   {
     
     auto associationMapForOutput      = std::make_unique<TTClusterAssociationMap<Ref_Phase2TrackerDigi_>>();
 
     edm::Handle< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > > > TTClusterHandle;
 
     iEvent.getByToken( *iTag, TTClusterHandle );
 
     std::map< edm::Ref< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > >, TTCluster< Ref_Phase2TrackerDigi_ > >, std::vector< edm::Ptr< TrackingParticle > > > clusterToTrackingParticleVectorMap;
     std::map< edm::Ptr< TrackingParticle >, std::vector< edm::Ref< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > >, TTCluster< Ref_Phase2TrackerDigi_ > > > > trackingParticleToClusterVectorMap;
     clusterToTrackingParticleVectorMap.clear();
     trackingParticleToClusterVectorMap.clear();
 
     for (auto gd=theTrackerGeom->dets().begin(); gd != theTrackerGeom->dets().end(); gd++) 
     {
       DetId detid = (*gd)->geographicalId();
       if(detid.subdetId()!=StripSubdetector::TOB && detid.subdetId()!=StripSubdetector::TID ) continue; // only run on OT
 
       if (TTClusterHandle->find( detid ) == TTClusterHandle->end() ) continue;
 
       edmNew::DetSet< TTCluster< Ref_Phase2TrackerDigi_ > > clusters = (*TTClusterHandle)[ detid ];
 
       for ( auto contentIter = clusters.begin();contentIter != clusters.end();++contentIter ) 
       {
         edm::Ref< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > >, TTCluster< Ref_Phase2TrackerDigi_ > > tempCluRef = edmNew::makeRefTo( TTClusterHandle, contentIter );
 
         if ( clusterToTrackingParticleVectorMap.find( tempCluRef ) == clusterToTrackingParticleVectorMap.end() )
         {
           std::vector< edm::Ptr< TrackingParticle > > tpVector;
           tpVector.clear();
           clusterToTrackingParticleVectorMap.insert( std::make_pair( tempCluRef, tpVector ) );
         }
 
         if ( thePixelDigiSimLinkHandle->find(detid) == thePixelDigiSimLinkHandle->end() )
         {
 
           std::vector< Ref_Phase2TrackerDigi_ > theseHits = tempCluRef->getHits();
           for ( unsigned int i = 0; i < theseHits.size(); i++ )
           {
             edm::Ptr< TrackingParticle > tempTPPtr; // = new edm::Ptr< TrackingParticle >();
             clusterToTrackingParticleVectorMap.find( tempCluRef )->second.push_back( tempTPPtr );
           }
 
           continue;
         }
 
         edm::DetSet<PixelDigiSimLink> thisDigiSimLink = (*(thePixelDigiSimLinkHandle) )[detid];
         edm::DetSet<PixelDigiSimLink>::const_iterator iterSimLink;
 
         std::vector< Ref_Phase2TrackerDigi_ > theseHits = tempCluRef->getHits();
         for ( unsigned int i = 0; i < theseHits.size(); i++ )
         {
           for ( iterSimLink = thisDigiSimLink.data.begin();
                 iterSimLink != thisDigiSimLink.data.end();
                 iterSimLink++ )
           {
             if ( static_cast<int>(iterSimLink->channel()) != static_cast<int>(theseHits.at(i)->channel()) )
               continue;
 
             unsigned int curSimTrkId = iterSimLink->SimTrackId();
             EncodedEventId curSimEvId = iterSimLink->eventId();
 
             std::pair< unsigned int, EncodedEventId > thisUniqueId = std::make_pair( curSimTrkId, curSimEvId );
 
             if ( simTrackUniqueToTPMap.find( thisUniqueId ) != simTrackUniqueToTPMap.end() )
             {
               edm::Ptr< TrackingParticle > thisTrackingParticle = simTrackUniqueToTPMap.find( thisUniqueId )->second;
 
               clusterToTrackingParticleVectorMap.find( tempCluRef )->second.push_back( thisTrackingParticle );
 
               if ( trackingParticleToClusterVectorMap.find( thisTrackingParticle ) == trackingParticleToClusterVectorMap.end() )
               {
                 std::vector< edm::Ref< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > >, TTCluster< Ref_Phase2TrackerDigi_ > > > clusterVector;
                 clusterVector.clear();
                 trackingParticleToClusterVectorMap.insert( std::make_pair( thisTrackingParticle, clusterVector ) );
               }
               trackingParticleToClusterVectorMap.find( thisTrackingParticle )->second.push_back( tempCluRef ); 
             }
             else 
             {
 
               edm::Ptr< TrackingParticle > tempTPPtr; // = new edm::Ptr< TrackingParticle >();
               clusterToTrackingParticleVectorMap.find( tempCluRef )->second.push_back( tempTPPtr );
             }
           } 
         } 
 
         std::vector< edm::Ptr< TrackingParticle > > theseClusterTrackingParticlePtrs = clusterToTrackingParticleVectorMap.find( tempCluRef )->second;
         bool allOfThemAreNull = true;
         for ( unsigned int tpi = 0; tpi < theseClusterTrackingParticlePtrs.size() && allOfThemAreNull; tpi++ )
         {
           if ( theseClusterTrackingParticlePtrs.at(tpi).isNull() == false )
             allOfThemAreNull = false;
         }
 
         if ( allOfThemAreNull )
         {
           clusterToTrackingParticleVectorMap.erase( tempCluRef ); 
         }
 
       }
     } 
 
     std::map< edm::Ptr< TrackingParticle >, std::vector< edm::Ref< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > >, TTCluster< Ref_Phase2TrackerDigi_ > > > >::iterator iterMapToClean;
     for ( iterMapToClean = trackingParticleToClusterVectorMap.begin();
           iterMapToClean != trackingParticleToClusterVectorMap.end();
           ++iterMapToClean )
     {
       std::vector< edm::Ref< edmNew::DetSetVector< TTCluster< Ref_Phase2TrackerDigi_ > >, TTCluster< Ref_Phase2TrackerDigi_ > > > tempVector = iterMapToClean->second;
 
       std::sort( tempVector.begin(), tempVector.end() );
       tempVector.erase( std::unique( tempVector.begin(), tempVector.end() ), tempVector.end() );
       iterMapToClean->second = tempVector;
     }
 
     associationMapForOutput->setTTClusterToTrackingParticlesMap( clusterToTrackingParticleVectorMap );
     associationMapForOutput->setTrackingParticleToTTClustersMap( trackingParticleToClusterVectorMap );
 
     //   iEvent.put( associationMapForOutput, (*iTag).instance() );
     iEvent.put( std::move(associationMapForOutput), TTClustersInputTags.at(ncont1).instance() );
 
     ++ncont1;
 
   } 
 }