TTClusterAssociator< T > Class Template Reference

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
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

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)

Private Attributes
edm::EDGetTokenT < edm::DetSetVector < PixelDigiSimLink > >	digisimLinkToken_
edm::Handle< edm::DetSetVector < PixelDigiSimLink > >	thePixelDigiSimLinkHandle_
	Data members. More...
edm::ESGetToken < TrackerGeometry, TrackerDigiGeometryRecord >	theTrackerGeometryToken_
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<>
using	CacheTypes = CacheContexts< T...>
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T...>
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::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 (tidy up: Ian Tomalin, 2020)

Definition at line 43 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 82 of file TTClusterAssociator.h.

References DeDxTools::esConsumes(), and edm::ParameterSet::getParameter().

                                                                          {
  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 (const auto& iTag : ttClustersInputTags_) {
    ttClustersTokens_.push_back(consumes<edmNew::DetSetVector<TTCluster<T> > >(iTag));

    produces<TTClusterAssociationMap<T> >(iTag.instance());
  }

  theTrackerGeometryToken_ = esConsumes();
}

template<typename T >

TTClusterAssociator< T >::~TTClusterAssociator ( )

override

Destructor.

Definition at line 100 of file TTClusterAssociator.h.

{}

Member Function Documentation

template<typename T >

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

overrideprivate

Mandatory methods.

Begin run.

Print some information when loaded

Definition at line 104 of file TTClusterAssociator.h.

                                                                                    {
  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 111 of file TTClusterAssociator.h.

{}

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

Use 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(), HLT_FULL_cff::clusters, edm::DetSet< T >::data, TrackerGeometry::dets(), edmNew::DetSet< T >::end(), edm::Event::getByToken(), edm::EventSetup::getData(), mps_fire::i, edm::EventBase::isRealData(), edmNew::makeRefTo(), eostools::move(), AlCaHLTBitMon_ParallelJobs::p, edm::Event::put(), cscDigiValidation_cfi::simTrack, 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 = &iSetup.getData(theTrackerGeometryToken_);

  std::map<std::pair<unsigned int, EncodedEventId>, TrackingParticlePtr> simTrackUniqueToTPMap;

  if (not trackingParticleHandle_->empty()) {
    for (unsigned int tpCnt = 0; tpCnt < trackingParticleHandle_->size(); tpCnt++) {
      TrackingParticlePtr tempTPPtr(trackingParticleHandle_, tpCnt);

      EncodedEventId eventId = EncodedEventId(tempTPPtr->eventId());

      for (const auto& simTrack : tempTPPtr->g4Tracks()) {
        simTrackUniqueToTPMap.emplace(std::make_pair(simTrack.trackId(), eventId), tempTPPtr);
      }
    }  
  }


  int ncont1 = 0;

  for (const auto& iTag : ttClustersTokens_) {
    auto associationMapForOutput = std::make_unique<TTClusterAssociationMap<Ref_Phase2TrackerDigi_>>();

    edm::Handle<TTClusterDetSetVec> TTClusterHandle;

    iEvent.getByToken(iTag, TTClusterHandle);

    std::map<TTClusterRef, std::vector<TrackingParticlePtr>> clusterToTrackingParticleVectorMap;
    std::map<TrackingParticlePtr, std::vector<TTClusterRef>> trackingParticleToClusterVectorMap;

    for (const auto& gd : theTrackerGeom->dets()) {
      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) {
        TTClusterRef tempCluRef = edmNew::makeRefTo(TTClusterHandle, contentIter);

        if (clusterToTrackingParticleVectorMap.find(tempCluRef) == clusterToTrackingParticleVectorMap.end()) {
          std::vector<TrackingParticlePtr> tpVector;
          clusterToTrackingParticleVectorMap.emplace(tempCluRef, tpVector);
        }

        if (thePixelDigiSimLinkHandle_->find(detid) == thePixelDigiSimLinkHandle_->end()) {

          std::vector<Ref_Phase2TrackerDigi_> theseHits = tempCluRef->getHits();
          for (unsigned int i = 0; i < theseHits.size(); i++) {
            TrackingParticlePtr tempTPPtr;
            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()) {
              TrackingParticlePtr thisTrackingParticle = simTrackUniqueToTPMap.find(thisUniqueId)->second;

              clusterToTrackingParticleVectorMap.find(tempCluRef)->second.push_back(thisTrackingParticle);

              if (trackingParticleToClusterVectorMap.find(thisTrackingParticle) ==
                  trackingParticleToClusterVectorMap.end()) {
                std::vector<TTClusterRef> clusterVector;
                trackingParticleToClusterVectorMap.emplace(thisTrackingParticle, clusterVector);
              }
              trackingParticleToClusterVectorMap.find(thisTrackingParticle)
                  ->second.push_back(tempCluRef);  
            } else {

              TrackingParticlePtr tempTPPtr;
              clusterToTrackingParticleVectorMap.find(tempCluRef)->second.push_back(tempTPPtr);
            }
          }  
        }    

        const std::vector<TrackingParticlePtr>& 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);  
        }
      }
    }  

    for (auto& p : trackingParticleToClusterVectorMap) {
      std::vector<TTClusterRef>& tempVector = p.second;

      std::sort(tempVector.begin(), tempVector.end());
      tempVector.erase(std::unique(tempVector.begin(), tempVector.end()), tempVector.end());
    }

    associationMapForOutput->setTTClusterToTrackingParticlesMap(clusterToTrackingParticleVectorMap);
    associationMapForOutput->setTrackingParticleToTTClustersMap(trackingParticleToClusterVectorMap);

    iEvent.put(std::move(associationMapForOutput), ttClustersInputTags_.at(ncont1).instance());

    ++ncont1;

  }  
}

template<typename T >

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

overrideprivate

template<>

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

private

Member Data Documentation

template<typename T >

edm::EDGetTokenT<edm::DetSetVector<PixelDigiSimLink> > TTClusterAssociator< T >::digisimLinkToken_

private

Definition at line 60 of file TTClusterAssociator.h.

template<typename T >

edm::Handle<edm::DetSetVector<PixelDigiSimLink> > TTClusterAssociator< T >::thePixelDigiSimLinkHandle_

private

Data members.

Definition at line 55 of file TTClusterAssociator.h.

template<typename T >

edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> TTClusterAssociator< T >::theTrackerGeometryToken_

private

Definition at line 64 of file TTClusterAssociator.h.

template<typename T >

edm::EDGetTokenT<std::vector<TrackingParticle> > TTClusterAssociator< T >::tpToken_

private

Definition at line 61 of file TTClusterAssociator.h.

template<typename T >

edm::Handle<std::vector<TrackingParticle> > TTClusterAssociator< T >::trackingParticleHandle_

private

Definition at line 56 of file TTClusterAssociator.h.

template<typename T >

std::vector<edm::InputTag> TTClusterAssociator< T >::ttClustersInputTags_

private

Definition at line 58 of file TTClusterAssociator.h.

template<typename T >

std::vector<edm::EDGetTokenT<edmNew::DetSetVector<TTCluster<T> > > > TTClusterAssociator< T >::ttClustersTokens_

private

Definition at line 62 of file TTClusterAssociator.h.