#include <RecoTracker/DeDxEstimatorProducer/src/DeDxEstimatorProducer.cc>

Inheritance diagram for DeDxEstimatorProducer:

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

	~DeDxEstimatorProducer ()

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

void	makeCalibrationMap (const TrackerGeometry &tkGeom)

void	processHit (const TrackingRecHit *recHit, float trackMomentum, float &cosine, reco::DeDxHitCollection &dedxHits, int &NClusterSaturating)

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

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

std::string	m_calibrationPath

BaseDeDxEstimator *	m_estimator

unsigned int	m_off

edm::EDGetTokenT < reco::TrackCollection >	m_tracksTag

edm::EDGetTokenT < TrajTrackAssociationCollection >	m_trajTrackAssociationTag

unsigned int	MaxNrStrips

float	meVperADCPixel

float	meVperADCStrip

bool	shapetest

bool	useCalibration

bool	usePixel

bool	useStrip

bool	useTrajectory

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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

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

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

Definition at line 47 of file DeDxEstimatorProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 53 of file DeDxEstimatorProducer.cc.

References edm::ParameterSet::getParameter().

 {
 
    produces<ValueMap<DeDxData> >();
 
    string estimatorName = iConfig.getParameter<string>("estimator");
    if     (estimatorName == "median")              m_estimator = new MedianDeDxEstimator(iConfig);
    else if(estimatorName == "generic")             m_estimator = new GenericAverageDeDxEstimator  (iConfig);
    else if(estimatorName == "truncated")           m_estimator = new TruncatedAverageDeDxEstimator(iConfig);
    else if(estimatorName == "unbinnedFit")         m_estimator = new UnbinnedFitDeDxEstimator(iConfig);
    else if(estimatorName == "productDiscrim")      m_estimator = new ProductDeDxDiscriminator(iConfig);
    else if(estimatorName == "btagDiscrim")         m_estimator = new BTagLikeDeDxDiscriminator(iConfig);
    else if(estimatorName == "smirnovDiscrim")      m_estimator = new SmirnovDeDxDiscriminator(iConfig);
    else if(estimatorName == "asmirnovDiscrim")     m_estimator = new ASmirnovDeDxDiscriminator(iConfig);
 
   //Commented for now, might be used in the future
 //   MaxNrStrips         = iConfig.getUntrackedParameter<unsigned>("maxNrStrips"        ,  255);
 
    m_tracksTag = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
    m_trajTrackAssociationTag   = consumes<TrajTrackAssociationCollection>(iConfig.getParameter<edm::InputTag>("trajectoryTrackAssociation"));
    useTrajectory = iConfig.getParameter<bool>("UseTrajectory"); 
 
    usePixel = iConfig.getParameter<bool>("UsePixel"); 
    useStrip = iConfig.getParameter<bool>("UseStrip");
    meVperADCPixel = iConfig.getParameter<double>("MeVperADCPixel"); 
    meVperADCStrip = iConfig.getParameter<double>("MeVperADCStrip"); 
 
    shapetest = iConfig.getParameter<bool>("ShapeTest");
    useCalibration = iConfig.getParameter<bool>("UseCalibration");
    m_calibrationPath = iConfig.getParameter<string>("calibrationPath");
 
    if(!usePixel && !useStrip)
    edm::LogWarning("DeDxHitsProducer") << "Pixel Hits AND Strip Hits will not be used to estimate dEdx --> BUG, Please Update the config file";
 
 }

DeDxEstimatorProducer::~DeDxEstimatorProducer ( )

Definition at line 90 of file DeDxEstimatorProducer.cc.

 {
   delete m_estimator;
 }

Member Function Documentation

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

overrideprivatevirtual

Reimplemented from edm::stream::EDProducerBase.

Definition at line 96 of file DeDxEstimatorProducer.cc.

References edm::EventSetup::get(), DeDxTools::makeCalibrationMap(), and GeomDetEnumerators::PixelBarrel.

 {
    if(useCalibration && calibGains.size()==0){
       edm::ESHandle<TrackerGeometry> tkGeom;
       iSetup.get<TrackerDigiGeometryRecord>().get( tkGeom );
       m_off = tkGeom->offsetDU(GeomDetEnumerators::PixelBarrel); //index start at the first pixel
 
       DeDxTools::makeCalibrationMap(m_calibrationPath, *tkGeom, calibGains, m_off);
    }
 
    m_estimator->beginRun(run, iSetup);
 }

void DeDxEstimatorProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 31 of file DeDxEstimatorProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), and HLT_25ns14e33_v1_cff::InputTag.

                                                                                        {
   edm::ParameterSetDescription desc;
   desc.add<string>("estimator","generic");
   desc.add<edm::InputTag>("tracks",edm::InputTag("generalTracks"));
   desc.add<edm::InputTag>("trajectoryTrackAssociation",edm::InputTag("generalTracks"));
   desc.add<bool>("UseTrajectory",true);  
   desc.add<bool>("UsePixel",false); 
   desc.add<bool>("UseStrip",true); 
   desc.add<double>("MeVperADCPixel",3.61e-06*265);
   desc.add<double>("MeVperADCStrip",3.61e-06);
   desc.add<bool>("ShapeTest",true);      
   desc.add<bool>("UseCalibration",false);  
   desc.add<string>("calibrationPath", "");
   desc.add<string>("Reccord", "SiStripDeDxMip_3D_Rcd");
   desc.add<string>("ProbabilityMode", "Accumulation");
   desc.add<double>("fraction", 0.4);
   desc.add<double>("exponent",-2.0);
 
   descriptions.add("DeDxEstimatorProducer",desc);
 }

void DeDxEstimatorProducer::makeCalibrationMap ( const TrackerGeometry & tkGeom )

private

void DeDxEstimatorProducer::processHit	(	const TrackingRecHit *	recHit,
		float	trackMomentum,
		float &	cosine,
		reco::DeDxHitCollection &	dedxHits,
		int &	NClusterSaturating
	)

private

Definition at line 180 of file DeDxEstimatorProducer.cc.

References Surface::bounds(), RecoTauCleanerPlugins::charge, compareJSON::const, TrackingRecHit::det(), TrackingRecHit::detUnit(), DeDxTools::getCharge(), SiStripMatchedRecHit2D::monoCluster(), GluedGeomDet::monoDet(), SiStripMatchedRecHit2D::monoId(), DeDxTools::shapeSelection(), SiStripMatchedRecHit2D::stereoCluster(), GluedGeomDet::stereoDet(), SiStripMatchedRecHit2D::stereoId(), GeomDet::surface(), and Bounds::thickness().

                                                                                                                                                                 {
       auto const & thit = static_cast<BaseTrackerRecHit const&>(*recHit);
       if(!thit.isValid())return;
 
       auto const & clus = thit.firstClusterRef();
       if(!clus.isValid())return;
 
       if(clus.isPixel()){
           if(!usePixel) return;
 
           auto& detUnit     = *(recHit->detUnit());
           float pathLen     = detUnit.surface().bounds().thickness()/fabs(cosine);
           float chargeAbs   = clus.pixelCluster().charge();
           float charge      = meVperADCPixel*chargeAbs/pathLen;
           dedxHits.push_back( DeDxHit( charge, trackMomentum, pathLen, thit.geographicalId()) );
        }else if(clus.isStrip() && !thit.isMatched()){
           if(!useStrip) return;
 
           auto& detUnit     = *(recHit->detUnit());
           int   NSaturating = 0;
           float pathLen     = detUnit.surface().bounds().thickness()/fabs(cosine);
           float chargeAbs   = DeDxTools::getCharge(&(clus.stripCluster()),NSaturating, detUnit, calibGains, m_off);
           float charge      = meVperADCStrip*chargeAbs/pathLen;
           if(!shapetest || (shapetest && DeDxTools::shapeSelection(clus.stripCluster()))){
              dedxHits.push_back( DeDxHit( charge, trackMomentum, pathLen, thit.geographicalId()) );
              if(NSaturating>0)NClusterSaturating++;
           }
        }else if(clus.isStrip() && thit.isMatched()){
           if(!useStrip) return;
           const SiStripMatchedRecHit2D* matchedHit=dynamic_cast<const SiStripMatchedRecHit2D*>(recHit);
           if(!matchedHit)return;
           const GluedGeomDet* gdet = static_cast<const GluedGeomDet*>(matchedHit->det());
 
           auto& detUnitM     = *(gdet->monoDet());
           int   NSaturating = 0;
           float pathLen     = detUnitM.surface().bounds().thickness()/fabs(cosine);
           float chargeAbs   = DeDxTools::getCharge(&(matchedHit->monoCluster()),NSaturating, detUnitM, calibGains, m_off);
           float charge      = meVperADCStrip*chargeAbs/pathLen;
           if(!shapetest || (shapetest && DeDxTools::shapeSelection(matchedHit->monoCluster()))){
              dedxHits.push_back( DeDxHit( charge, trackMomentum, pathLen, matchedHit->monoId()) );
              if(NSaturating>0)NClusterSaturating++;
           }
 
           auto& detUnitS     = *(gdet->stereoDet());
           NSaturating = 0;
           pathLen     = detUnitS.surface().bounds().thickness()/fabs(cosine);
           chargeAbs   = DeDxTools::getCharge(&(matchedHit->stereoCluster()),NSaturating, detUnitS, calibGains, m_off);
           charge      = meVperADCStrip*chargeAbs/pathLen;
           if(!shapetest || (shapetest && DeDxTools::shapeSelection(matchedHit->stereoCluster()))){
              dedxHits.push_back( DeDxHit( charge, trackMomentum, pathLen, matchedHit->stereoId()) );
              if(NSaturating>0)NClusterSaturating++;
           }      
        }
 }

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

overrideprivatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 111 of file DeDxEstimatorProducer.cc.

References compareJSON::const, edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, TrackingRecHit::detUnit(), edm::helper::Filler< Map >::fill(), edm::Event::getByToken(), h, edm::helper::Filler< Map >::insert(), TrajectoryStateOnSurface::isValid(), TrackingRecHit::isValid(), j, edm::Ref< C, T, F >::key(), TrajectoryStateOnSurface::localDirection(), TrajectoryStateOnSurface::localMomentum(), PV3DBase< T, PVType, FrameType >::mag(), Plane::normalVector(), edm::Handle< T >::product(), edm::Event::put(), python.multivaluedict::sort(), GeomDet::surface(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

 {
   auto_ptr<ValueMap<DeDxData> > trackDeDxEstimateAssociation(new ValueMap<DeDxData> );  
   ValueMap<DeDxData>::Filler filler(*trackDeDxEstimateAssociation);
 
   edm::Handle<reco::TrackCollection> trackCollectionHandle;
   iEvent.getByToken(m_tracksTag,trackCollectionHandle);
 
   Handle<TrajTrackAssociationCollection> trajTrackAssociationHandle;
   if(useTrajectory)iEvent.getByToken(m_trajTrackAssociationTag, trajTrackAssociationHandle);
 
   std::vector<DeDxData> dedxEstimate( trackCollectionHandle->size() );
 
   TrajTrackAssociationCollection::const_iterator cit;
   if(useTrajectory)cit = trajTrackAssociationHandle->begin();
   for(unsigned int j=0;j<trackCollectionHandle->size();j++){            
      const reco::TrackRef track = reco::TrackRef( trackCollectionHandle.product(), j );
 
      int NClusterSaturating = 0; 
      DeDxHitCollection dedxHits;
    
      if(useTrajectory){  //trajectory allows to take into account the local direction of the particle on the module sensor --> muc much better 'dx' measurement
         const edm::Ref<std::vector<Trajectory> > traj = cit->key; cit++;
         const vector<TrajectoryMeasurement> & measurements = traj->measurements();
         dedxHits.reserve(measurements.size()/2);
         for(vector<TrajectoryMeasurement>::const_iterator it = measurements.begin(); it!=measurements.end(); it++){
            TrajectoryStateOnSurface trajState=it->updatedState();
            if( !trajState.isValid()) continue;
      
            const TrackingRecHit * recHit=(*it->recHit()).hit();
            if(!recHit || !recHit->isValid())continue;
            LocalVector trackDirection = trajState.localDirection();
            float cosine = trackDirection.z()/trackDirection.mag();
 
            processHit(recHit, trajState.localMomentum().mag(), cosine, dedxHits, NClusterSaturating);
         }
      }else{ //assume that the particles trajectory is a straight line originating from the center of the detector  (can be improved)
         dedxHits.reserve(track->recHitsSize()/2);
         for(unsigned int h=0;h<track->recHitsSize();h++){
            const TrackingRecHit* recHit = &(*(track->recHit(h)));
            if(!recHit || !recHit->isValid())continue;
            auto const & thit = static_cast<BaseTrackerRecHit const&>(*recHit);
            if(!thit.isValid())continue;//make sure it's a tracker hit
 
            const GlobalVector& ModuleNormal = recHit->detUnit()->surface().normalVector();         
            float cosine = (track->px()*ModuleNormal.x()+track->py()*ModuleNormal.y()+track->pz()*ModuleNormal.z())/track->p();
 
            processHit(recHit, track->p(), cosine, dedxHits, NClusterSaturating);
         } 
      }
 
      sort(dedxHits.begin(),dedxHits.end(),less<DeDxHit>());   
      std::pair<float,float> val_and_error = m_estimator->dedx(dedxHits);
 
      //WARNING: Since the dEdX Error is not properly computed for the moment
      //It was decided to store the number of saturating cluster in that dataformat
      val_and_error.second = NClusterSaturating; 
      dedxEstimate[j] = DeDxData(val_and_error.first, val_and_error.second, dedxHits.size() );
   }
 
   filler.insert(trackCollectionHandle, dedxEstimate.begin(), dedxEstimate.end());
 
   // fill the association map and put it into the event
   filler.fill();
   iEvent.put(trackDeDxEstimateAssociation);   
 }

Member Data Documentation

std::vector< std::vector<float> > DeDxEstimatorProducer::calibGains

private

Definition at line 78 of file DeDxEstimatorProducer.h.

std::string DeDxEstimatorProducer::m_calibrationPath

private

Definition at line 74 of file DeDxEstimatorProducer.h.

BaseDeDxEstimator* DeDxEstimatorProducer::m_estimator

private

Definition at line 61 of file DeDxEstimatorProducer.h.

unsigned int DeDxEstimatorProducer::m_off

private

Definition at line 79 of file DeDxEstimatorProducer.h.

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

private

Definition at line 64 of file DeDxEstimatorProducer.h.

edm::EDGetTokenT<TrajTrackAssociationCollection> DeDxEstimatorProducer::m_trajTrackAssociationTag

private

Definition at line 63 of file DeDxEstimatorProducer.h.

unsigned int DeDxEstimatorProducer::MaxNrStrips

private

Definition at line 72 of file DeDxEstimatorProducer.h.

float DeDxEstimatorProducer::meVperADCPixel

private

Definition at line 69 of file DeDxEstimatorProducer.h.

float DeDxEstimatorProducer::meVperADCStrip

private

Definition at line 70 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::shapetest

private

Definition at line 76 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::useCalibration

private

Definition at line 75 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::usePixel

private

Definition at line 67 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::useStrip

private

Definition at line 68 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::useTrajectory

private

Definition at line 66 of file DeDxEstimatorProducer.h.

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation