CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
FastTrackDeDxProducer Class Reference

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

Inheritance diagram for FastTrackDeDxProducer:
edm::stream::EDProducer<>

Public Member Functions

 FastTrackDeDxProducer (const edm::ParameterSet &)
 
 ~FastTrackDeDxProducer () override=default
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 

Private Member Functions

void beginRun (edm::Run const &run, const edm::EventSetup &) override
 
void makeCalibrationMap (const TrackerGeometry &tkGeom)
 
void processHit (const FastTrackerRecHit &recHit, float trackMomentum, float &cosine, reco::DeDxHitCollection &dedxHits, int &NClusterSaturating)
 
void produce (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

std::vector< std::vector< float > > calibGains
 
bool convertFromGeV2MeV
 
std::string m_calibrationPath
 
std::unique_ptr< BaseDeDxEstimatorm_estimator
 
unsigned int m_off
 
edm::EDGetTokenT< reco::TrackCollectionm_tracksTag
 
unsigned int MaxNrStrips
 
float meVperADCPixel
 
float meVperADCStrip
 
bool nothick
 
bool shapetest
 
edm::EDGetTokenT< FastTrackerRecHitRefCollectionsimHit2RecHitMapToken
 
edm::EDGetTokenT< edm::PSimHitContainersimHitsToken
 
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, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Detailed Description

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

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

Definition at line 70 of file FastTrackDeDxProducer.cc.

Constructor & Destructor Documentation

FastTrackDeDxProducer::FastTrackDeDxProducer ( const edm::ParameterSet iConfig)
explicit

Definition at line 142 of file FastTrackDeDxProducer.cc.

References convertFromGeV2MeV, Exception, edm::ParameterSet::getParameter(), m_calibrationPath, m_estimator, m_tracksTag, meVperADCPixel, meVperADCStrip, nothick, shapetest, useCalibration, usePixel, and useStrip.

143 : simHitsToken(consumes<edm::PSimHitContainer>(iConfig.getParameter<edm::InputTag>("simHits")))
144 , simHit2RecHitMapToken(consumes<FastTrackerRecHitRefCollection>(iConfig.getParameter<edm::InputTag>("simHit2RecHitMap")))
145 {
146  produces<ValueMap<DeDxData> >();
147 
148  string estimatorName = iConfig.getParameter<string>("estimator");
149  if (estimatorName == "median") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new MedianDeDxEstimator(iConfig));
150  else if(estimatorName == "generic") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new GenericAverageDeDxEstimator (iConfig));
151  else if(estimatorName == "truncated") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new TruncatedAverageDeDxEstimator(iConfig));
152  //else if(estimatorName == "unbinnedFit") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new UnbinnedFitDeDxEstimator(iConfig));//estimator used in FullSimVersion
153  else if(estimatorName == "productDiscrim") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new ProductDeDxDiscriminator(iConfig));
154  else if(estimatorName == "btagDiscrim") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new BTagLikeDeDxDiscriminator(iConfig));
155  else if(estimatorName == "smirnovDiscrim") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new SmirnovDeDxDiscriminator(iConfig));
156  else if(estimatorName == "asmirnovDiscrim") m_estimator = std::unique_ptr<BaseDeDxEstimator> (new ASmirnovDeDxDiscriminator(iConfig));
157  else throw cms::Exception("fastsim::SimplifiedGeometry::FastTrackDeDxProducer.cc") << " estimator name does not exist";
158 
159  //Commented for now, might be used in the future
160  // MaxNrStrips = iConfig.getUntrackedParameter<unsigned>("maxNrStrips" , 255);
161 
162  m_tracksTag = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
163 
164  usePixel = iConfig.getParameter<bool>("UsePixel");
165  useStrip = iConfig.getParameter<bool>("UseStrip");
166  meVperADCPixel = iConfig.getParameter<double>("MeVperADCPixel");
167  meVperADCStrip = iConfig.getParameter<double>("MeVperADCStrip");
168 
169  shapetest = iConfig.getParameter<bool>("ShapeTest");
170  useCalibration = iConfig.getParameter<bool>("UseCalibration");
171  m_calibrationPath = iConfig.getParameter<string>("calibrationPath");
172 
173  convertFromGeV2MeV = iConfig.getParameter<bool>("convertFromGeV2MeV");
174  nothick = iConfig.getParameter<bool>("nothick");
175 
176  if(!usePixel && !useStrip)
177  throw cms::Exception("fastsim::SimplifiedGeometry::FastTrackDeDxProducer.cc") << " neither pixel hits nor strips hits will be used to compute de/dx";
178 
179 
180 }
T getParameter(std::string const &) const
edm::EDGetTokenT< reco::TrackCollection > m_tracksTag
edm::EDGetTokenT< FastTrackerRecHitRefCollection > simHit2RecHitMapToken
edm::EDGetTokenT< edm::PSimHitContainer > simHitsToken
std::unique_ptr< BaseDeDxEstimator > m_estimator
FastTrackDeDxProducer::~FastTrackDeDxProducer ( )
overridedefault

Member Function Documentation

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

Definition at line 183 of file FastTrackDeDxProducer.cc.

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

184 {
185  if(useCalibration && calibGains.empty()){
187  iSetup.get<TrackerDigiGeometryRecord>().get( tkGeom );
188  m_off = tkGeom->offsetDU(GeomDetEnumerators::PixelBarrel); //index start at the first pixel
189 
191  }
192 
193  m_estimator->beginRun(run, iSetup);
194 }
void makeCalibrationMap(const std::string &m_calibrationPath, const TrackerGeometry &tkGeom, std::vector< std::vector< float > > &calibGains, const unsigned int &m_off)
Definition: DeDxTools.cc:196
unsigned int offsetDU(SubDetector sid) const
std::vector< std::vector< float > > calibGains
T get() const
Definition: EventSetup.h:71
std::unique_ptr< BaseDeDxEstimator > m_estimator
void FastTrackDeDxProducer::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 119 of file FastTrackDeDxProducer.cc.

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

119  {
121  desc.add<string>("estimator","generic");
122  desc.add<edm::InputTag>("tracks",edm::InputTag("generalTracks"));
123  desc.add<bool>("UsePixel",false);
124  desc.add<bool>("UseStrip",true);
125  desc.add<double>("MeVperADCPixel",3.61e-06*265);
126  desc.add<double>("MeVperADCStrip",3.61e-06);
127  desc.add<bool>("ShapeTest",true);
128  desc.add<bool>("UseCalibration",false);
129  desc.add<string>("calibrationPath", "");
130  desc.add<string>("Reccord", "SiStripDeDxMip_3D_Rcd");
131  desc.add<string>("ProbabilityMode", "Accumulation");
132  desc.add<double>("fraction", 0.4);
133  desc.add<double>("exponent",-2.0);
134  desc.add<bool>("convertFromGeV2MeV",true);
135  desc.add<bool>("nothick",false);
136  desc.add<edm::InputTag>("simHits");
137  desc.add<edm::InputTag>("simHit2RecHitMap");
138  descriptions.add("FastTrackDeDxProducer",desc);
139 }
ParameterDescriptionBase * add(U const &iLabel, T const &value)
void add(std::string const &label, ParameterSetDescription const &psetDescription)
void FastTrackDeDxProducer::makeCalibrationMap ( const TrackerGeometry tkGeom)
private
void FastTrackDeDxProducer::processHit ( const FastTrackerRecHit recHit,
float  trackMomentum,
float &  cosine,
reco::DeDxHitCollection dedxHits,
int &  NClusterSaturating 
)
private

Definition at line 249 of file FastTrackDeDxProducer.cc.

References ALCARECOTkAlJpsiMuMu_cff::charge, convertFromGeV2MeV, DEFINE_FWK_MODULE, TrackingRecHit::detUnit(), FastTrackerRecHit::energyLoss(), TrackingRecHit::geographicalId(), FastTrackerRecHit::isPixel(), TrackingRecHit::isValid(), nothick, shapetest, usePixel, and useStrip.

Referenced by produce().

249  {
250 
251 
252  if(!recHit.isValid())return;
253 
254  if(recHit.isPixel()){
255  if(!usePixel) return;
256 
257  auto& detUnit = *(recHit.detUnit());
258  float pathLen = detUnit.surface().bounds().thickness()/fabs(cosine);
259  if (nothick) pathLen = 1.0;
260  float charge = recHit.energyLoss()/pathLen;
261  if (convertFromGeV2MeV) charge*=1000;
262  dedxHits.push_back( DeDxHit( charge, trackMomentum, pathLen, recHit.geographicalId()) );
263  }
264  else if(!recHit.isPixel()){// && !recHit.isMatched()){//check what recHit.isMatched is doing
265  if(!useStrip) return;
266  auto& detUnit = *(recHit.detUnit());
267  float pathLen = detUnit.surface().bounds().thickness()/fabs(cosine);
268  if (nothick) pathLen = 1.0;
269  float dedxOfRecHit = recHit.energyLoss()/pathLen;
270  if (convertFromGeV2MeV) dedxOfRecHit*=1000;
271  if(!shapetest ){
272  dedxHits.push_back( DeDxHit( dedxOfRecHit, trackMomentum, pathLen, recHit.geographicalId()) );
273  }
274  }
275 }
float energyLoss() const
bool isValid() const
bool isPixel() const override
pixel or strip?
virtual const GeomDetUnit * detUnit() const
DetId geographicalId() const
void FastTrackDeDxProducer::produce ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 198 of file FastTrackDeDxProducer.cc.

References objects.autophobj::filler, edm::Event::getByToken(), h, hcalSimParameters_cfi::hb, TrackingRecHit::isValid(), m_estimator, m_tracksTag, eostools::move(), processHit(), edm::Handle< T >::product(), edm::Event::put(), rpcPointValidation_cfi::recHit, jetUpdater_cfi::sort, HiIsolationCommonParameters_cff::track, and findElectronsInSiStrips_cfi::trackCollection.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

199 {
200 
201  auto trackDeDxEstimateAssociation = std::make_unique<ValueMap<DeDxData>>();
202  ValueMap<DeDxData>::Filler filler(*trackDeDxEstimateAssociation);
203 
204  edm::Handle<reco::TrackCollection> trackCollectionHandle;
205  iEvent.getByToken(m_tracksTag,trackCollectionHandle);
206  const auto& trackCollection = *trackCollectionHandle;
207  std::vector<DeDxData> dedxEstimate( trackCollection.size() );
208 
209 
210 
211  for(unsigned int j=0;j<trackCollection.size();j++){
212  const reco::TrackRef track = reco::TrackRef( trackCollectionHandle.product(), j );
213 
214  int NClusterSaturating = 0;
215  DeDxHitCollection dedxHits;
216 
217  auto const & trajParams = track->extra()->trajParams();
218  assert(trajParams.size()==track->recHitsSize());
219 
220  auto hb = track->recHitsBegin();
221  dedxHits.reserve(track->recHitsSize()/2);
222 
223 
224 
225  for(unsigned int h=0;h<track->recHitsSize();h++){
226  const FastTrackerRecHit& recHit = static_cast< const FastTrackerRecHit & >(*(*(hb+h)));
227  if(!recHit.isValid()) continue;//FastTrackerRecHit recHit = *(hb+h);
228  auto trackDirection = trajParams[h].direction();
229  float cosine = trackDirection.z()/trackDirection.mag();
230  processHit(recHit, track->p(), cosine, dedxHits, NClusterSaturating);
231  }
232 
233  sort(dedxHits.begin(),dedxHits.end(),less<DeDxHit>());
234  std::pair<float,float> val_and_error = m_estimator->dedx(dedxHits);
235  //WARNING: Since the dEdX Error is not properly computed for the moment
236  //It was decided to store the number of saturating cluster in that dataformat
237  val_and_error.second = NClusterSaturating;
238  dedxEstimate[j] = DeDxData(val_and_error.first, val_and_error.second, dedxHits.size() );
239  }
240 
241 
242  filler.insert(trackCollectionHandle, dedxEstimate.begin(), dedxEstimate.end());
243  // fill the association map and put it into the event
244  filler.fill();
245  iEvent.put(std::move(trackDeDxEstimateAssociation));
246 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:125
void processHit(const FastTrackerRecHit &recHit, float trackMomentum, float &cosine, reco::DeDxHitCollection &dedxHits, int &NClusterSaturating)
FWCore Framework interface EventSetupRecordImplementation h
Helper function to determine trigger accepts.
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:517
edm::EDGetTokenT< reco::TrackCollection > m_tracksTag
std::vector< DeDxHit > DeDxHitCollection
Definition: DeDxHit.h:58
T const * product() const
Definition: Handle.h:74
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
Definition: TrackFwd.h:21
bool isValid() const
std::unique_ptr< BaseDeDxEstimator > m_estimator
def move(src, dest)
Definition: eostools.py:511

Member Data Documentation

std::vector< std::vector<float> > FastTrackDeDxProducer::calibGains
private

Definition at line 99 of file FastTrackDeDxProducer.cc.

Referenced by beginRun().

bool FastTrackDeDxProducer::convertFromGeV2MeV
private

Definition at line 109 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer(), and processHit().

std::string FastTrackDeDxProducer::m_calibrationPath
private

Definition at line 97 of file FastTrackDeDxProducer.cc.

Referenced by beginRun(), and FastTrackDeDxProducer().

std::unique_ptr<BaseDeDxEstimator> FastTrackDeDxProducer::m_estimator
private

Definition at line 86 of file FastTrackDeDxProducer.cc.

Referenced by beginRun(), FastTrackDeDxProducer(), and produce().

unsigned int FastTrackDeDxProducer::m_off
private

Definition at line 100 of file FastTrackDeDxProducer.cc.

Referenced by beginRun().

edm::EDGetTokenT<reco::TrackCollection> FastTrackDeDxProducer::m_tracksTag
private

Definition at line 88 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer(), and produce().

unsigned int FastTrackDeDxProducer::MaxNrStrips
private

Definition at line 95 of file FastTrackDeDxProducer.cc.

float FastTrackDeDxProducer::meVperADCPixel
private

Definition at line 92 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer().

float FastTrackDeDxProducer::meVperADCStrip
private

Definition at line 93 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer().

bool FastTrackDeDxProducer::nothick
private

Definition at line 110 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer(), and processHit().

bool FastTrackDeDxProducer::shapetest
private

Definition at line 108 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer(), and processHit().

edm::EDGetTokenT<FastTrackerRecHitRefCollection> FastTrackDeDxProducer::simHit2RecHitMapToken
private

Definition at line 103 of file FastTrackDeDxProducer.cc.

edm::EDGetTokenT<edm::PSimHitContainer> FastTrackDeDxProducer::simHitsToken
private

Definition at line 102 of file FastTrackDeDxProducer.cc.

bool FastTrackDeDxProducer::useCalibration
private

Definition at line 107 of file FastTrackDeDxProducer.cc.

Referenced by beginRun(), and FastTrackDeDxProducer().

bool FastTrackDeDxProducer::usePixel
private

Definition at line 105 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer(), and processHit().

bool FastTrackDeDxProducer::useStrip
private

Definition at line 106 of file FastTrackDeDxProducer.cc.

Referenced by FastTrackDeDxProducer(), and processHit().