#include <RecoTracker/DeDx/plugins/DeDxHitInfoProducer.cc>

Inheritance diagram for DeDxHitInfoProducer:

Public Member Functions
	DeDxHitInfoProducer (const edm::ParameterSet &)

	~DeDxHitInfoProducer () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	beginRun (edm::Run const &run, const edm::EventSetup &) override

void	makeCalibrationMap (const TrackerGeometry &tkGeom)

void	processHit (const TrackingRecHit *recHit, const float trackMomentum, const float cosine, reco::DeDxHitInfo &hitDeDxInfo, const LocalPoint &hitLocalPos)

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

uint64_t	xorshift128p (uint64_t state[2])

Private Attributes
std::vector< std::vector< float > >	calibGains

const float	lowPtTracksDeDxThreshold

GenericTruncatedAverageDeDxEstimator	lowPtTracksEstimator

const unsigned int	lowPtTracksPrescaleFail

const unsigned int	lowPtTracksPrescalePass

const std::string	m_calibrationPath

unsigned int	m_off

edm::EDGetTokenT< reco::TrackCollection >	m_tracksTag

const float	maxTrackEta

const float	MeVperADCPixel

const float	MeVperADCStrip

const unsigned int	minTrackHits

const float	minTrackPt

const float	minTrackPtPrescale

const bool	shapetest

edm::ESHandle< TrackerGeometry >	tkGeom

const bool	useCalibration

const bool	usePixel

const bool	useStrip

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

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 34 of file DeDxHitInfoProducer.h.

Constructor & Destructor Documentation

DeDxHitInfoProducer::DeDxHitInfoProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 27 of file DeDxHitInfoProducer.cc.

References edm::ParameterSet::getParameter(), m_tracksTag, usePixel, and useStrip.

                                                                       :
    usePixel          ( iConfig.getParameter<bool>    ("usePixel")       ),
    useStrip          ( iConfig.getParameter<bool>    ("useStrip")       ),
    MeVperADCPixel    ( iConfig.getParameter<double>  ("MeVperADCPixel") ),
    MeVperADCStrip    ( iConfig.getParameter<double>  ("MeVperADCStrip") ),
    minTrackHits      ( iConfig.getParameter<unsigned>("minTrackHits")   ),
    minTrackPt        ( iConfig.getParameter<double>  ("minTrackPt"  )   ),
    minTrackPtPrescale( iConfig.getParameter<double>  ("minTrackPtPrescale") ),
    maxTrackEta       ( iConfig.getParameter<double>  ("maxTrackEta" )   ),
    m_calibrationPath ( iConfig.getParameter<string>  ("calibrationPath")),
    useCalibration    ( iConfig.getParameter<bool>    ("useCalibration") ),
    shapetest         ( iConfig.getParameter<bool>    ("shapeTest")      ),
    lowPtTracksPrescalePass( iConfig.getParameter<uint32_t>("lowPtTracksPrescalePass") ),
    lowPtTracksPrescaleFail( iConfig.getParameter<uint32_t>("lowPtTracksPrescaleFail") ),
    lowPtTracksEstimator(iConfig.getParameter<edm::ParameterSet>("lowPtTracksEstimatorParameters")),
    lowPtTracksDeDxThreshold(iConfig.getParameter<double>("lowPtTracksDeDxThreshold"))
 {
    produces<reco::DeDxHitInfoCollection >();
    produces<reco::DeDxHitInfoAss >();
    produces<edm::ValueMap<int> >("prescale");
 
    m_tracksTag = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
 
    if(!usePixel && !useStrip)
    edm::LogError("DeDxHitsProducer") << "No Pixel Hits NOR Strip Hits will be saved.  Running this module is useless";
 }

DeDxHitInfoProducer::~DeDxHitInfoProducer ( )

override

Definition at line 55 of file DeDxHitInfoProducer.cc.

55 {}

Member Function Documentation

void DeDxHitInfoProducer::beginRun	(	edm::Run const &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 58 of file DeDxHitInfoProducer.cc.

References calibGains, edm::EventSetup::get(), m_calibrationPath, m_off, DeDxTools::makeCalibrationMap(), TrackerGeometry::offsetDU(), GeomDetEnumerators::PixelBarrel, tkGeom, and useCalibration.

 {
    iSetup.get<TrackerDigiGeometryRecord>().get( tkGeom );
    if(useCalibration && calibGains.empty()){
       m_off = tkGeom->offsetDU(GeomDetEnumerators::PixelBarrel); //index start at the first pixel
 
       DeDxTools::makeCalibrationMap(m_calibrationPath, *tkGeom, calibGains, m_off);
    }
 }

void DeDxHitInfoProducer::makeCalibrationMap ( const TrackerGeometry & tkGeom )

private

void DeDxHitInfoProducer::processHit	(	const TrackingRecHit *	recHit,
		const float	trackMomentum,
		const float	cosine,
		reco::DeDxHitInfo &	hitDeDxInfo,
		const LocalPoint &	hitLocalPos
	)

private

Definition at line 165 of file DeDxHitInfoProducer.cc.

References funct::abs(), reco::DeDxHitInfo::addHit(), Surface::bounds(), calibGains, DEFINE_FWK_MODULE, TrackerSingleRecHit::detUnit(), TrackingRecHit::detUnit(), f, TrackingRecHit::geographicalId(), DeDxTools::getCharge(), TrackerGeometry::idToDet(), m_off, SiStripPI::max, SiStripMatchedRecHit2D::monoHit(), rpcPointValidation_cfi::recHit, SiStripMatchedRecHit2D::stereoHit(), TrackerSingleRecHit::stripCluster(), GeomDet::surface(), Bounds::thickness(), tkGeom, usePixel, and useStrip.

Referenced by produce().

                                                                                                                                                                              {
       auto const & thit = static_cast<BaseTrackerRecHit const&>(*recHit);
       if(!thit.isValid())return;
 
       float cosineAbs = std::max(0.00000001f,std::abs(cosine));//make sure cosine is not 0
 
       auto const & clus = thit.firstClusterRef();
       if(!clus.isValid())return;
 
 
       if(clus.isPixel()){
           if(!usePixel) return;
 
           const auto * detUnit = recHit->detUnit();
           if (detUnit == nullptr) detUnit = tkGeom->idToDet(thit.geographicalId());
           float pathLen     = detUnit->surface().bounds().thickness()/cosineAbs;
           float chargeAbs   = clus.pixelCluster().charge();
           hitDeDxInfo.addHit(chargeAbs, pathLen, thit.geographicalId(), hitLocalPos, clus.pixelCluster() );
        }else if(clus.isStrip() && !thit.isMatched()){
           if(!useStrip) return;
 
           const auto * detUnit = recHit->detUnit();
           if (detUnit == nullptr) detUnit = tkGeom->idToDet(thit.geographicalId());
           int   NSaturating = 0;
           float pathLen     = detUnit->surface().bounds().thickness()/cosineAbs;
           float chargeAbs   = DeDxTools::getCharge(&(clus.stripCluster()),NSaturating, *detUnit, calibGains, m_off);
           hitDeDxInfo.addHit(chargeAbs, pathLen, thit.geographicalId(), hitLocalPos, clus.stripCluster() );
        }else if(clus.isStrip() && thit.isMatched()){
           if(!useStrip) return;
           const SiStripMatchedRecHit2D* matchedHit=dynamic_cast<const SiStripMatchedRecHit2D*>(recHit);
           if(!matchedHit)return;
 
           const auto * detUnitM = matchedHit->monoHit().detUnit();
           if (detUnitM == nullptr) detUnitM = tkGeom->idToDet(matchedHit->monoHit().geographicalId());
           int   NSaturating = 0;
           float pathLen     = detUnitM->surface().bounds().thickness()/cosineAbs;
           float chargeAbs   = DeDxTools::getCharge(&(matchedHit->monoHit().stripCluster()),NSaturating, *detUnitM, calibGains, m_off);
           hitDeDxInfo.addHit(chargeAbs, pathLen, thit.geographicalId(), hitLocalPos, matchedHit->monoHit().stripCluster() );
 
           const auto * detUnitS = matchedHit->stereoHit().detUnit();
           if (detUnitS == nullptr) detUnitS = tkGeom->idToDet(matchedHit->stereoHit().geographicalId());
           NSaturating = 0;
           pathLen     = detUnitS->surface().bounds().thickness()/cosineAbs;
           chargeAbs   = DeDxTools::getCharge(&(matchedHit->stereoHit().stripCluster()),NSaturating, *detUnitS, calibGains, m_off);
           hitDeDxInfo.addHit(chargeAbs, pathLen, thit.geographicalId(), hitLocalPos, matchedHit->stereoHit().stripCluster() );          
        }
 }

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

overrideprivate

Definition at line 70 of file DeDxHitInfoProducer.cc.

References funct::abs(), reco::DeDxHitInfo::charge(), GenericTruncatedAverageDeDxEstimator::dedx(), reco::DeDxHitInfo::detId(), reco::TrackBase::eta(), edm::EventID::event(), reco::Track::extra(), objects.autophobj::filler, edm::Event::getByToken(), h, hcalSimParameters_cfi::hb, hfClusterShapes_cfi::hits, mps_fire::i, edm::EventBase::id(), trackerHitRTTI::isFromDet(), lowPtTracksDeDxThreshold, lowPtTracksEstimator, lowPtTracksPrescaleFail, lowPtTracksPrescalePass, edm::EventID::luminosityBlock(), m_tracksTag, maxTrackEta, MeVperADCPixel, MeVperADCStrip, minTrackHits, minTrackPt, minTrackPtPrescale, eostools::move(), gen::n, reco::TrackBase::numberOfValidHits(), reco::TrackBase::p(), reco::DeDxHitInfo::pathlength(), processHit(), edm::Handle< T >::product(), reco::TrackBase::pt(), edm::Event::put(), rpcPointValidation_cfi::recHit, reco::Track::recHitsBegin(), reco::Track::recHitsSize(), DeDxTools::shapeSelection(), shapetest, reco::DeDxHitInfo::size(), reco::DeDxHitInfo::stripCluster(), DetId::subdetId(), HiIsolationCommonParameters_cff::track, findElectronsInSiStrips_cfi::trackCollection, and xorshift128p().

 {  
   edm::Handle<reco::TrackCollection> trackCollectionHandle;
   iEvent.getByToken(m_tracksTag,trackCollectionHandle);
   const TrackCollection& trackCollection(*trackCollectionHandle.product());
 
   // creates the output collection
   auto resultdedxHitColl = std::make_unique<reco::DeDxHitInfoCollection>();
 
   std::vector<int> indices; std::vector<int> prescales;
   uint64_t state[2] = { iEvent.id().event(), iEvent.id().luminosityBlock() };
   for(unsigned int j=0;j<trackCollection.size();j++){            
      const reco::Track& track = trackCollection[j];
 
      //track selection
      bool passPt = (track.pt() >= minTrackPt), passLowDeDx = false, passHighDeDx = false, pass = passPt;
      if (!pass && (track.pt() >= minTrackPtPrescale)) {
         if (lowPtTracksPrescalePass > 0) {
             passHighDeDx = ((xorshift128p(state) % lowPtTracksPrescalePass) == 0);
         } 
         if (lowPtTracksPrescaleFail > 0) {
             passLowDeDx = ((xorshift128p(state) % lowPtTracksPrescaleFail) == 0);
         }
         pass = passHighDeDx || passLowDeDx;
      }
      if(!pass || std::abs(track.eta())>maxTrackEta ||track.numberOfValidHits()<minTrackHits){
         indices.push_back(-1);
         continue;
      }
 
      reco::DeDxHitInfo hitDeDxInfo;
      auto const & trajParams = track.extra()->trajParams();
      auto hb = track.recHitsBegin();
         for(unsigned int h=0;h<track.recHitsSize();h++){
            auto recHit = *(hb+h);
            if (!trackerHitRTTI::isFromDet(*recHit) ) continue;
 
            auto trackDirection = trajParams[h].direction();
            float cosine = trackDirection.z()/trackDirection.mag();
  
            processHit(recHit, track.p(), cosine, hitDeDxInfo, trajParams[h].position());
      }
 
      if (!passPt) {
         std::vector<DeDxHit> hits; hits.reserve(hitDeDxInfo.size());
         for (unsigned int i = 0, n = hitDeDxInfo.size(); i < n; ++i) {
            if (hitDeDxInfo.detId(i).subdetId() <= 2) {
                hits.push_back( DeDxHit(hitDeDxInfo.charge(i)/hitDeDxInfo.pathlength(i) * MeVperADCPixel, 0,0,0) );
            } else {
                if (shapetest && !DeDxTools::shapeSelection(*hitDeDxInfo.stripCluster(i))) continue;
                hits.push_back( DeDxHit(hitDeDxInfo.charge(i)/hitDeDxInfo.pathlength(i) * MeVperADCStrip, 0,0,0) );
            }
         }
         std::sort(hits.begin(), hits.end(), std::less<DeDxHit>());
         if (lowPtTracksEstimator.dedx(hits).first < lowPtTracksDeDxThreshold) {
             if (passLowDeDx) {
                 prescales.push_back(lowPtTracksPrescaleFail);
             } else {
                 indices.push_back(-1);
                 continue;
             }
         } else {
             if (passHighDeDx) {
                 prescales.push_back(lowPtTracksPrescalePass);
             } else {
                 indices.push_back(-1);
                 continue;
             }
 
         }
      } else {
         prescales.push_back(1);
      }
      indices.push_back(resultdedxHitColl->size());
      resultdedxHitColl->push_back(hitDeDxInfo);
   }
  
 
   edm::OrphanHandle<reco::DeDxHitInfoCollection> dedxHitCollHandle = iEvent.put(std::move(resultdedxHitColl));
 
   //create map passing the handle to the matched collection
   auto dedxMatch = std::make_unique<reco::DeDxHitInfoAss>(dedxHitCollHandle);
   reco::DeDxHitInfoAss::Filler filler(*dedxMatch);  
   filler.insert(trackCollectionHandle, indices.begin(), indices.end()); 
   filler.fill();
   iEvent.put(std::move(dedxMatch));
 
   auto dedxPrescale = std::make_unique<edm::ValueMap<int>>();
   edm::ValueMap<int>::Filler pfiller(*dedxPrescale);
   pfiller.insert(dedxHitCollHandle, prescales.begin(), prescales.end());
   pfiller.fill();
   iEvent.put(std::move(dedxPrescale), "prescale");
 }

uint64_t DeDxHitInfoProducer::xorshift128p ( uint64_t state[2] )

inlineprivate

Definition at line 73 of file DeDxHitInfoProducer.h.

References x, and y.

Referenced by produce().

                                            {
       uint64_t x = state[0];
       uint64_t const y = state[1];
       state[0] = y;
       x ^= x << 23; // a
       state[1] = x ^ y ^ (x >> 17) ^ (y >> 26); // b, c
       return state[1] + y;
   }

Member Data Documentation

std::vector< std::vector<float> > DeDxHitInfoProducer::calibGains

private

Definition at line 68 of file DeDxHitInfoProducer.h.

Referenced by beginRun(), and processHit().

const float DeDxHitInfoProducer::lowPtTracksDeDxThreshold

private

Definition at line 66 of file DeDxHitInfoProducer.h.

Referenced by produce().

GenericTruncatedAverageDeDxEstimator DeDxHitInfoProducer::lowPtTracksEstimator

private

Definition at line 65 of file DeDxHitInfoProducer.h.

Referenced by produce().

const unsigned int DeDxHitInfoProducer::lowPtTracksPrescaleFail

private

Definition at line 64 of file DeDxHitInfoProducer.h.

Referenced by produce().

const unsigned int DeDxHitInfoProducer::lowPtTracksPrescalePass

private

Definition at line 64 of file DeDxHitInfoProducer.h.

Referenced by produce().

const std::string DeDxHitInfoProducer::m_calibrationPath

private

Definition at line 60 of file DeDxHitInfoProducer.h.

Referenced by beginRun().

unsigned int DeDxHitInfoProducer::m_off

private

Definition at line 69 of file DeDxHitInfoProducer.h.

Referenced by beginRun(), and processHit().

edm::EDGetTokenT<reco::TrackCollection> DeDxHitInfoProducer::m_tracksTag

private

Definition at line 48 of file DeDxHitInfoProducer.h.

Referenced by DeDxHitInfoProducer(), and produce().

const float DeDxHitInfoProducer::maxTrackEta

private

Definition at line 58 of file DeDxHitInfoProducer.h.

Referenced by produce().

const float DeDxHitInfoProducer::MeVperADCPixel

private

Definition at line 52 of file DeDxHitInfoProducer.h.

Referenced by produce().

const float DeDxHitInfoProducer::MeVperADCStrip

private

Definition at line 53 of file DeDxHitInfoProducer.h.

Referenced by produce().

const unsigned int DeDxHitInfoProducer::minTrackHits

private

Definition at line 55 of file DeDxHitInfoProducer.h.

Referenced by produce().

const float DeDxHitInfoProducer::minTrackPt

private

Definition at line 56 of file DeDxHitInfoProducer.h.

Referenced by produce().

const float DeDxHitInfoProducer::minTrackPtPrescale

private

Definition at line 57 of file DeDxHitInfoProducer.h.

Referenced by produce().

const bool DeDxHitInfoProducer::shapetest

private

Definition at line 62 of file DeDxHitInfoProducer.h.

Referenced by produce().

edm::ESHandle<TrackerGeometry> DeDxHitInfoProducer::tkGeom

private

Definition at line 71 of file DeDxHitInfoProducer.h.

Referenced by beginRun(), and processHit().

const bool DeDxHitInfoProducer::useCalibration

private

Definition at line 61 of file DeDxHitInfoProducer.h.

Referenced by beginRun().

const bool DeDxHitInfoProducer::usePixel

private

Definition at line 50 of file DeDxHitInfoProducer.h.

Referenced by DeDxHitInfoProducer(), and processHit().

const bool DeDxHitInfoProducer::useStrip

private

Definition at line 51 of file DeDxHitInfoProducer.h.

Referenced by DeDxHitInfoProducer(), and processHit().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation