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

Inheritance diagram for DeDxEstimatorProducer:

Classes
class	isEqual

struct	stModInfo

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
	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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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

int	getCharge (const SiStripCluster *Cluster, int &Saturating_Strips, const uint32_t &)

void	MakeCalibrationMap ()

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

Private Attributes
std::string	m_calibrationPath

BaseDeDxEstimator *	m_estimator

edm::EDGetTokenT < reco::TrackCollection >	m_tracksTag

edm::EDGetTokenT < TrajTrackAssociationCollection >	m_trajTrackAssociationTag

unsigned int	MaxNrStrips

double	MeVperADCPixel

double	MeVperADCStrip

unsigned int	MinTrackHits

__gnu_cxx::hash_map< unsigned int, stModInfo *, __gnu_cxx::hash< unsigned int >, isEqual >	MODsColl

bool	shapetest

bool	useCalibration

bool	usePixel

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

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

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

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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 34 of file DeDxEstimatorProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 50 of file DeDxEstimatorProducer.cc.

References edm::ParameterSet::getParameter(), and edm::ParameterSet::getUntrackedParameter().

 {
 
    produces<ValueMap<DeDxData> >();
 
 
    string estimatorName = iConfig.getParameter<string>("estimator");
    if(estimatorName == "median")      m_estimator = new MedianDeDxEstimator(-2.);
    if(estimatorName == "generic")     m_estimator = new GenericAverageDeDxEstimator  (iConfig.getParameter<double>("exponent"));
    if(estimatorName == "truncated")   m_estimator = new TruncatedAverageDeDxEstimator(iConfig.getParameter<double>("fraction"));
    if(estimatorName == "unbinnedFit") m_estimator = new UnbinnedFitDeDxEstimator();
 
    MaxNrStrips         = iConfig.getUntrackedParameter<unsigned>("maxNrStrips"        ,  255);
    MinTrackHits        = iConfig.getUntrackedParameter<unsigned>("MinTrackHits"       ,  4);
 
    m_tracksTag = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
    m_trajTrackAssociationTag   = consumes<TrajTrackAssociationCollection>(iConfig.getParameter<edm::InputTag>("trajectoryTrackAssociation"));
 
    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 82 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 88 of file DeDxEstimatorProducer.cc.

References compareJSON::const, DeDxEstimatorProducer::stModInfo::DetId, DeDxEstimatorProducer::stModInfo::Distance, DeDxEstimatorProducer::stModInfo::Gain, edm::EventSetup::get(), i, DeDxEstimatorProducer::stModInfo::Normalization, DetId::rawId(), and DeDxEstimatorProducer::stModInfo::Thickness.

 {
    if(MODsColl.size()!=0)return;
 
 
    edm::ESHandle<TrackerGeometry> tkGeom;
    iSetup.get<TrackerDigiGeometryRecord>().get( tkGeom );
 
    auto const & Det = tkGeom->dets();
    for(unsigned int i=0;i<Det.size();i++){
       DetId  Detid  = Det[i]->geographicalId();
 
        auto StripDetUnit = dynamic_cast<StripGeomDetUnit const*> (Det[i]);
        auto PixelDetUnit = dynamic_cast<PixelGeomDetUnit const*> (Det[i]);
 
        double Thick=-1, Dist=-1, Norma=-1;
        if(StripDetUnit){
           Dist      = StripDetUnit->position().mag();
           Thick     = StripDetUnit->surface().bounds().thickness();
           Norma     = MeVperADCStrip/Thick;
        }else if(PixelDetUnit){
           Dist      = PixelDetUnit->position().mag();
           Thick     = PixelDetUnit->surface().bounds().thickness();
           Norma     = MeVperADCPixel/Thick;
        }
 
        stModInfo* MOD       = new stModInfo;
        MOD->DetId           = Detid.rawId();
        MOD->Thickness       = Thick;
        MOD->Distance        = Dist;
        MOD->Normalization   = Norma;
        MOD->Gain            = 1;
        MODsColl[MOD->DetId] = MOD;
    }
 
    MakeCalibrationMap();
 }

int DeDxEstimatorProducer::getCharge	(	const SiStripCluster *	Cluster,
		int &	Saturating_Strips,
		const uint32_t &	DetId
	)

private

Definition at line 282 of file DeDxEstimatorProducer.cc.

References SiStripCluster::amplitudes(), DeDxEstimatorProducer::stModInfo::Gain, and i.

 {
    //const vector<uint8_t>&  Ampls       = Cluster->amplitudes();
    const vector<uint8_t>&  Ampls       = Cluster->amplitudes();
    //uint32_t                DetId       = Cluster->geographicalId();
 
 //   float G=1.0f;
 
    int toReturn = 0;
    Saturating_Strips = 0;
    for(unsigned int i=0;i<Ampls.size();i++){
       int CalibratedCharge = Ampls[i];
 
       if(useCalibration){
          stModInfo* MOD = MODsColl[DetId];
 //         G = MOD->Gain;
          CalibratedCharge = (int)(CalibratedCharge / MOD->Gain );
          if(CalibratedCharge>=1024){
             CalibratedCharge = 255;
          }else if(CalibratedCharge>=255){
             CalibratedCharge = 254;
          } 
       }
 
       toReturn+=CalibratedCharge;
       if(CalibratedCharge>=254)Saturating_Strips++;
    }
 
 //   printf("Charge = %i --> %i  (Gain=%f)\n", accumulate(Ampls.begin(), Ampls.end(), 0), toReturn, G);         
 
 
    return toReturn;
 }

void DeDxEstimatorProducer::MakeCalibrationMap ( )

private

Definition at line 254 of file DeDxEstimatorProducer.cc.

References DeDxEstimatorProducer::stModInfo::Gain.

 {
    if(!useCalibration)return;
 
    TChain* t1 = new TChain("SiStripCalib/APVGain");
    t1->Add(m_calibrationPath.c_str());
 
    unsigned int  tree_DetId;
    unsigned char tree_APVId;
    double        tree_Gain;
 
    t1->SetBranchAddress("DetId"             ,&tree_DetId      );
    t1->SetBranchAddress("APVId"             ,&tree_APVId      );
    t1->SetBranchAddress("Gain"              ,&tree_Gain       );
 
 
 
    for (unsigned int ientry = 0; ientry < t1->GetEntries(); ientry++) {
        t1->GetEntry(ientry);
        stModInfo* MOD  = MODsColl[tree_DetId];
        MOD->Gain = tree_Gain;
    }
 
    delete t1;
 
 }

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

overrideprivatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 128 of file DeDxEstimatorProducer.cc.

References funct::abs(), DeDxTools::RawHits::angleCosine, edm::AssociationMap< Tag >::begin(), DeDxTools::RawHits::charge, DeDxDiscriminatorTools::charge(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, DeDxTools::RawHits::detId, DeDxEstimatorProducer::stModInfo::Distance, edm::AssociationMap< Tag >::end(), edm::helper::Filler< Map >::fill(), edm::Event::getByToken(), DeDxTools::GetCluster(), i, edm::helper::Filler< Map >::insert(), TrajectoryStateOnSurface::isValid(), j, edm::Ref< C, T, F >::key(), TrajectoryStateOnSurface::localDirection(), PV3DBase< T, PVType, FrameType >::mag(), n, DeDxEstimatorProducer::stModInfo::Normalization, DeDxTools::RawHits::NSaturating, edm::Handle< T >::product(), edm::Event::put(), DeDxTools::shapeSelection(), edm::AssociationMap< Tag >::size(), python.multivaluedict::sort(), DeDxEstimatorProducer::stModInfo::Thickness, DeDxTools::RawHits::trajectoryMeasurement, and PV3DBase< T, PVType, FrameType >::z().

 {
   auto_ptr<ValueMap<DeDxData> > trackDeDxEstimateAssociation(new ValueMap<DeDxData> );  
   ValueMap<DeDxData>::Filler filler(*trackDeDxEstimateAssociation);
 
   Handle<TrajTrackAssociationCollection> trajTrackAssociationHandle;
   iEvent.getByToken(m_trajTrackAssociationTag, trajTrackAssociationHandle);
   const TrajTrackAssociationCollection & TrajToTrackMap = *trajTrackAssociationHandle.product();
 
   edm::Handle<reco::TrackCollection> trackCollectionHandle;
   iEvent.getByToken(m_tracksTag,trackCollectionHandle);
 
   size_t n =  TrajToTrackMap.size();
   std::vector<DeDxData> dedxEstimate(n);
 
   //assume trajectory collection size is equal to track collection size and that order is kept
   int j=0;
   for(TrajTrackAssociationCollection::const_iterator cit=TrajToTrackMap.begin(); cit!=TrajToTrackMap.end(); cit++,j++){
      
      const edm::Ref<std::vector<Trajectory> > traj = cit->key;
      const reco::TrackRef track = cit->val;
 
      DeDxHitCollection dedxHits;
      vector<DeDxTools::RawHits> hits; 
 //     DeDxTools::trajectoryRawHits(traj, hits, usePixel, useStrip);
 
     const vector<TrajectoryMeasurement> & measurements = traj->measurements();
     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();
       LocalVector trackDirection = trajState.localDirection();
       double cosine = trackDirection.z()/trackDirection.mag();
               
        if(const SiStripMatchedRecHit2D* matchedHit=dynamic_cast<const SiStripMatchedRecHit2D*>(recHit)){
        if(!useStrip) continue;
        DeDxTools::RawHits mono,stereo; 
        mono.trajectoryMeasurement = &(*it);
        stereo.trajectoryMeasurement = &(*it);
        mono.angleCosine = cosine; 
        stereo.angleCosine = cosine;
 
        mono.charge = getCharge(DeDxTools::GetCluster(matchedHit->monoHit()),mono.NSaturating,matchedHit->monoId());
            stereo.charge = getCharge(DeDxTools::GetCluster(matchedHit->stereoHit()),stereo.NSaturating,matchedHit->stereoId());
 
        mono.detId= matchedHit->monoId();
        stereo.detId= matchedHit->stereoId();
 
            if(shapetest && !(DeDxTools::shapeSelection((DeDxTools::GetCluster(matchedHit->stereoHit()))->amplitudes()))) hits.push_back(stereo);
        if(shapetest && !(DeDxTools::shapeSelection((DeDxTools::GetCluster(matchedHit->  monoHit()))->amplitudes()))) hits.push_back(mono);
         }else if(const ProjectedSiStripRecHit2D* projectedHit=dynamic_cast<const ProjectedSiStripRecHit2D*>(recHit)) {
            if(!useStrip) continue;
            DeDxTools::RawHits mono;
 
            mono.trajectoryMeasurement = &(*it);
            mono.angleCosine = cosine;
            mono.charge = getCharge(DeDxTools::GetCluster(projectedHit->originalHit()),mono.NSaturating,projectedHit->originalId());
            mono.detId= projectedHit->originalId();
        if(shapetest && !(DeDxTools::shapeSelection((DeDxTools::GetCluster(projectedHit->originalHit()))->amplitudes()))) continue;
            hits.push_back(mono);
         }else if(const SiStripRecHit2D* singleHit=dynamic_cast<const SiStripRecHit2D*>(recHit)){
            if(!useStrip) continue;
            DeDxTools::RawHits mono;
            
            mono.trajectoryMeasurement = &(*it);
            mono.angleCosine = cosine; 
            mono.charge = getCharge(DeDxTools::GetCluster(singleHit),mono.NSaturating,singleHit->geographicalId());
            mono.detId= singleHit->geographicalId();
        if(shapetest && !(DeDxTools::shapeSelection((DeDxTools::GetCluster(singleHit))->amplitudes()))) continue;
            hits.push_back(mono); 
         }else if(const SiStripRecHit1D* single1DHit=dynamic_cast<const SiStripRecHit1D*>(recHit)){
            if(!useStrip) continue;
            DeDxTools::RawHits mono;
                
            mono.trajectoryMeasurement = &(*it);
            mono.angleCosine = cosine; 
            mono.charge = getCharge(DeDxTools::GetCluster(single1DHit),mono.NSaturating,single1DHit->geographicalId());
            mono.detId= single1DHit->geographicalId();
        if(shapetest && !(DeDxTools::shapeSelection((DeDxTools::GetCluster(single1DHit))->amplitudes()))) continue;
            hits.push_back(mono); 
         }else if(const SiPixelRecHit* pixelHit=dynamic_cast<const SiPixelRecHit*>(recHit)){
            if(!usePixel) continue;
 
            DeDxTools::RawHits pixel;
 
            pixel.trajectoryMeasurement = &(*it);
            pixel.angleCosine = cosine; 
            pixel.charge = pixelHit->cluster()->charge();
            pixel.NSaturating=-1;
            pixel.detId= pixelHit->geographicalId();
            hits.push_back(pixel);
        } 
     }
 
      int NClusterSaturating = 0; 
      for(size_t i=0; i < hits.size(); i++)
      {
          stModInfo* MOD = MODsColl[hits[i].detId];
          float pathLen = MOD->Thickness/std::abs(hits[i].angleCosine);
          float charge  = MOD->Normalization*hits[i].charge*std::abs(hits[i].angleCosine);
          dedxHits.push_back( DeDxHit( charge, MOD->Distance, pathLen, hits[i].detId) );
 
          if(hits[i].NSaturating>0)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());
 
   // really fill the association map
   filler.fill();
    // put into the event 
   iEvent.put(trackDeDxEstimateAssociation);   
 }

Member Data Documentation

std::string DeDxEstimatorProducer::m_calibrationPath

private

Definition at line 64 of file DeDxEstimatorProducer.h.

BaseDeDxEstimator* DeDxEstimatorProducer::m_estimator

private

Definition at line 51 of file DeDxEstimatorProducer.h.

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

private

Definition at line 54 of file DeDxEstimatorProducer.h.

edm::EDGetTokenT<TrajTrackAssociationCollection> DeDxEstimatorProducer::m_trajTrackAssociationTag

private

Definition at line 53 of file DeDxEstimatorProducer.h.

unsigned int DeDxEstimatorProducer::MaxNrStrips

private

Definition at line 61 of file DeDxEstimatorProducer.h.

double DeDxEstimatorProducer::MeVperADCPixel

private

Definition at line 58 of file DeDxEstimatorProducer.h.

double DeDxEstimatorProducer::MeVperADCStrip

private

Definition at line 59 of file DeDxEstimatorProducer.h.

unsigned int DeDxEstimatorProducer::MinTrackHits

private

Definition at line 62 of file DeDxEstimatorProducer.h.

__gnu_cxx::hash_map<unsigned int, stModInfo*, __gnu_cxx::hash<unsigned int>, isEqual > DeDxEstimatorProducer::MODsColl

private

Definition at line 76 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::shapetest

private

Definition at line 66 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::useCalibration

private

Definition at line 65 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::usePixel

private

Definition at line 56 of file DeDxEstimatorProducer.h.

bool DeDxEstimatorProducer::useStrip

private

Definition at line 57 of file DeDxEstimatorProducer.h.

Classes

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation