MtdSimLayerClusterToTPAssociatorByTrackIdImpl Class Reference

#include <MtdSimLayerClusterToTPAssociatorByTrackIdImpl.h>

Inheritance diagram for MtdSimLayerClusterToTPAssociatorByTrackIdImpl:

Public Member Functions
reco::SimToTPCollectionMtd	associateSimToTP (const edm::Handle< MtdSimLayerClusterCollection > &simClusH, const edm::Handle< TrackingParticleCollection > &trackingParticleH) const override
	Associate a MtdSimLayerCluster to TrackingParticle. More...
reco::TPToSimCollectionMtd	associateTPToSim (const edm::Handle< MtdSimLayerClusterCollection > &simClusH, const edm::Handle< TrackingParticleCollection > &trackingParticleH) const override
	Associate a TrackingParticle to MtdSimLayerCluster. More...
	MtdSimLayerClusterToTPAssociatorByTrackIdImpl (edm::EDProductGetter const &)
Public Member Functions inherited from reco::MtdSimLayerClusterToTPAssociatorBaseImpl
	MtdSimLayerClusterToTPAssociatorBaseImpl ()
	Constructor. More...
virtual	~MtdSimLayerClusterToTPAssociatorBaseImpl ()
	Destructor. More...

Private Attributes
edm::EDProductGetter const *	productGetter_

Detailed Description

Definition at line 10 of file MtdSimLayerClusterToTPAssociatorByTrackIdImpl.h.

Constructor & Destructor Documentation

MtdSimLayerClusterToTPAssociatorByTrackIdImpl()

MtdSimLayerClusterToTPAssociatorByTrackIdImpl::MtdSimLayerClusterToTPAssociatorByTrackIdImpl ( edm::EDProductGetter const &  productGetter )

explicit

Definition at line 11 of file MtdSimLayerClusterToTPAssociatorByTrackIdImpl.cc.

    : productGetter_(&productGetter) {}

Member Function Documentation

associateSimToTP()

reco::SimToTPCollectionMtd MtdSimLayerClusterToTPAssociatorByTrackIdImpl::associateSimToTP ( const edm::Handle< MtdSimLayerClusterCollection > &  simClusH, const edm::Handle< TrackingParticleCollection > &  trackingParticleH  ) const

overridevirtual

Associate a MtdSimLayerCluster to TrackingParticle.

Reimplemented from reco::MtdSimLayerClusterToTPAssociatorBaseImpl.

Definition at line 19 of file MtdSimLayerClusterToTPAssociatorByTrackIdImpl.cc.

References edm::AssociationMap< Tag >::insert(), ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, LogDebug, edm::Handle< T >::product(), productGetter_, EncodedEventId::rawId(), cmsswSequenceInfo::tp, and muonClassificationByHits_cfi::trackingParticles.

                                                                          {
  SimToTPCollectionMtd outputCollection(productGetter_);

  // -- get the collections
  const auto& simClusters = *simClusH.product();
  const auto& trackingParticles = *trackingParticleH.product();

  // -- Loop over tracking particles and build a temporary map of trackId, eventId  --> tpRef
  std::map<std::pair<unsigned int, uint32_t>, TrackingParticleRef> tpIdMap;
  for (auto tpIt = trackingParticles.begin(); tpIt != trackingParticles.end(); tpIt++) {
    const auto& tp = *tpIt;
    unsigned int tpTrackId = tp.g4Tracks()[0].trackId();
    EncodedEventId tpEventId = tp.eventId();
    TrackingParticleRef tpRef =
        edm::Ref<TrackingParticleCollection>(trackingParticleH, tpIt - trackingParticles.begin());
    tpIdMap[std::make_pair(tpTrackId, tpEventId.rawId())] = tpRef;
  }

  // -- loop over sim clusters and get the trackId, eventId

  LogDebug("MtdSimLayerClusterToTPAssociator")
      << " Found " << simClusters.size() << " MtdSimLayerClusters in the event";

  for (auto simClusIt = simClusters.begin(); simClusIt != simClusters.end(); simClusIt++) {
    const auto& simClus = *simClusIt;
    size_t simClusIndex = simClusIt - simClusters.begin();
    MtdSimLayerClusterRef simClusterRef = edm::Ref<MtdSimLayerClusterCollection>(simClusH, simClusIndex);
    unsigned int simClusTrackId = simClus.g4Tracks()[0].trackId();
    EncodedEventId simClusEventId = simClus.eventId();

    // -- Check the trackId offset of the sim hits and keep only clusters with "direct" hits (offset == 0)
    /* 
       if (simClus.trackIdOffset() != 0 ) continue; 
    */

    std::pair uniqueId = std::make_pair(simClusTrackId, simClusEventId.rawId());
    auto it = tpIdMap.find(uniqueId);

    if (it != tpIdMap.end()) {
      TrackingParticleRef tpRef = tpIdMap[uniqueId];
      outputCollection.insert(simClusterRef, tpRef);

      LogDebug("MtdSimLayerClusterToTPAssociator::associateSimToTP")
          << "MtdSimLayerCluster: index = " << simClusIndex << "   simClus TrackId = " << simClusTrackId
          << " simClus EventId = " << simClusEventId.rawId() << " simClus Eta = " << simClus.eta()
          << " simClus Phi = " << simClus.phi() << "  simClus Time = " << simClus.simLCTime()
          << "  simClus Energy = " << simClus.simLCEnergy() << std::endl;
      LogDebug("MtdSimLayerClusterToTPAssociator::associateSimToTP")
          << "  --> Found associated tracking particle:  tp TrackId = " << (*tpRef).g4Tracks()[0].trackId()
          << " tp EventId = " << (*tpRef).eventId().rawId() << std::endl;
    }

  }  // -- end loop over sim clus

  return outputCollection;
}

associateTPToSim()

reco::TPToSimCollectionMtd MtdSimLayerClusterToTPAssociatorByTrackIdImpl::associateTPToSim ( const edm::Handle< MtdSimLayerClusterCollection > &  simClusH, const edm::Handle< TrackingParticleCollection > &  trackingParticleH  ) const

overridevirtual

Associate a TrackingParticle to MtdSimLayerCluster.

Reimplemented from reco::MtdSimLayerClusterToTPAssociatorBaseImpl.

Definition at line 78 of file MtdSimLayerClusterToTPAssociatorByTrackIdImpl.cc.

References mps_fire::i, edm::AssociationMap< Tag >::insert(), ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, LogDebug, edm::Handle< T >::product(), productGetter_, push_back(), EncodedEventId::rawId(), cmsswSequenceInfo::tp, and muonClassificationByHits_cfi::trackingParticles.

                                                                          {
  TPToSimCollectionMtd outputCollection(productGetter_);

  // -- get the collections
  const auto& simClusters = *simClusH.product();
  const auto& trackingParticles = *trackingParticleH.product();

  // -- Loop over MtdSimLayerClusters and build a temporary map of trackId, eventId --> simClusterRef
  std::map<std::pair<unsigned int, uint32_t>, std::vector<MtdSimLayerClusterRef>> simClusIdMap;
  for (auto simClusIt = simClusters.begin(); simClusIt != simClusters.end(); simClusIt++) {
    const auto& simClus = *simClusIt;
    unsigned int simClusTrackId = simClus.g4Tracks()[0].trackId();
    EncodedEventId simClusEventId = simClus.eventId();
    MtdSimLayerClusterRef simClusterRef =
        edm::Ref<MtdSimLayerClusterCollection>(simClusH, simClusIt - simClusters.begin());
    simClusIdMap[std::make_pair(simClusTrackId, simClusEventId.rawId())].push_back(simClusterRef);
  }

  // -- Loop over the tracking particles
  for (auto tpIt = trackingParticles.begin(); tpIt != trackingParticles.end(); tpIt++) {
    const auto& tp = *tpIt;
    size_t tpIndex = tpIt - trackingParticles.begin();
    TrackingParticleRef tpRef = edm::Ref<TrackingParticleCollection>(trackingParticleH, tpIndex);
    unsigned int tpTrackId = tp.g4Tracks()[0].trackId();
    EncodedEventId tpEventId = tp.eventId();

    std::pair uniqueId = std::make_pair(tpTrackId, tpEventId.rawId());
    auto it = simClusIdMap.find(uniqueId);

    if (it != simClusIdMap.end()) {
      for (unsigned int i = 0; i < simClusIdMap[uniqueId].size(); i++) {
        MtdSimLayerClusterRef simClusterRef = simClusIdMap[uniqueId][i];
        // -- Check the trackId offset of the sim hits and keep only clusters with "direct" hits (offset == 0)
        /*
      if (simClus.trackIdOffset() != 0 ) continue; 
    */

        outputCollection.insert(tpRef, simClusterRef);

        LogDebug("MtdSimLayerClusterToTPAssociator")
            << "Tracking particle:  index = " << tpIndex << "  tp TrackId = " << tpTrackId
            << "  tp EventId = " << tpEventId.rawId();
        LogDebug("MtdSimLayerClusterToTPAssociator")
            << " --> Found associated MtdSimLayerCluster:  simClus TrackId = "
            << (*simClusterRef).g4Tracks()[0].trackId() << " simClus EventId = " << (*simClusterRef).eventId().rawId()
            << " simClus Eta = " << (*simClusterRef).eta() << " simClus Phi = " << (*simClusterRef).phi()
            << "  simClus Time = " << (*simClusterRef).simLCTime()
            << "  simClus Energy = " << (*simClusterRef).simLCEnergy() << std::endl;
      }
    }
  }

  return outputCollection;
}

Member Data Documentation

productGetter_

edm::EDProductGetter const* MtdSimLayerClusterToTPAssociatorByTrackIdImpl::productGetter_

private

Definition at line 23 of file MtdSimLayerClusterToTPAssociatorByTrackIdImpl.h.

Referenced by associateSimToTP(), and associateTPToSim().