CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
Functions
DeDxTools Namespace Reference

Functions

void buildDiscrimMap (edm::Run const &run, const edm::EventSetup &iSetup, std::string Reccord, std::string ProbabilityMode, TH3F *&Prob_ChargePath)
 
int getCharge (const SiStripCluster *cluster, int &nSatStrip, const GeomDetUnit &detUnit, const std::vector< std::vector< float > > &calibGains, const unsigned int &m_off)
 
bool IsFarFromBorder (const TrajectoryStateOnSurface &trajState, const GeomDetUnit *it)
 
bool IsSpanningOver2APV (unsigned int FirstStrip, unsigned int ClusterSize)
 
void makeCalibrationMap (const std::string &m_calibrationPath, const TrackerGeometry &tkGeom, std::vector< std::vector< float > > &calibGains, const unsigned int &m_off)
 
bool shapeSelection (const SiStripCluster &ampls)
 

Function Documentation

void DeDxTools::buildDiscrimMap ( edm::Run const &  run,
const edm::EventSetup iSetup,
std::string  Reccord,
std::string  ProbabilityMode,
TH3F *&  Prob_ChargePath 
)

Definition at line 221 of file DeDxTools.cc.

References Exception, edm::EventSetup::get(), i, j, relval_2017::k, cmsLHEtoEOSManager::l, tmp, SiStripMonitorClusterAlca_cfi::xmax, SiStripMonitorClusterAlca_cfi::xmin, SiStripMonitorClusterAlca_cfi::ymax, SiStripMonitorClusterAlca_cfi::ymin, SiStripMonitorClusterAlca_cfi::zmax, and SiStripMonitorClusterAlca_cfi::zmin.

Referenced by ASmirnovDeDxDiscriminator::beginRun(), SmirnovDeDxDiscriminator::beginRun(), ProductDeDxDiscriminator::beginRun(), and BTagLikeDeDxDiscriminator::beginRun().

222 {
224  if( strcmp(Reccord.c_str(),"SiStripDeDxMip_3D_Rcd")==0){
225  iSetup.get<SiStripDeDxMip_3D_Rcd>() .get(deDxMapHandle);
226  }else if(strcmp(Reccord.c_str(),"SiStripDeDxPion_3D_Rcd")==0){
227  iSetup.get<SiStripDeDxPion_3D_Rcd>().get(deDxMapHandle);
228  }else if(strcmp(Reccord.c_str(),"SiStripDeDxKaon_3D_Rcd")==0){
229  iSetup.get<SiStripDeDxKaon_3D_Rcd>().get(deDxMapHandle);
230  }else if(strcmp(Reccord.c_str(),"SiStripDeDxProton_3D_Rcd")==0){
231  iSetup.get<SiStripDeDxProton_3D_Rcd>().get(deDxMapHandle);
232  }else if(strcmp(Reccord.c_str(),"SiStripDeDxElectron_3D_Rcd")==0){
233  iSetup.get<SiStripDeDxElectron_3D_Rcd>().get(deDxMapHandle);
234  }else{
235  throw cms::Exception("WrongReccord for dEdx") << "The reccord : " << Reccord << "is unknown\n";
236  }
237 
238  float xmin = deDxMapHandle->rangeX().min;
239  float xmax = deDxMapHandle->rangeX().max;
240  float ymin = deDxMapHandle->rangeY().min;
241  float ymax = deDxMapHandle->rangeY().max;
242  float zmin = deDxMapHandle->rangeZ().min;
243  float zmax = deDxMapHandle->rangeZ().max;
244 
245  if(Prob_ChargePath)delete Prob_ChargePath;
246  Prob_ChargePath = new TH3F ("Prob_ChargePath" , "Prob_ChargePath" , deDxMapHandle->numberOfBinsX(), xmin, xmax, deDxMapHandle->numberOfBinsY() , ymin, ymax, deDxMapHandle->numberOfBinsZ(), zmin, zmax);
247 
248  if(strcmp(ProbabilityMode.c_str(),"Accumulation")==0){
249  for(int i=0;i<=Prob_ChargePath->GetXaxis()->GetNbins()+1;i++){
250  for(int j=0;j<=Prob_ChargePath->GetYaxis()->GetNbins()+1;j++){
251  float Ni = 0;
252  for(int k=0;k<=Prob_ChargePath->GetZaxis()->GetNbins()+1;k++){ Ni+=deDxMapHandle->binContent(i,j,k);}
253  for(int k=0;k<=Prob_ChargePath->GetZaxis()->GetNbins()+1;k++){
254  float tmp = 0;
255  for(int l=0;l<=k;l++){ tmp+=deDxMapHandle->binContent(i,j,l);}
256  if(Ni>0){
257  Prob_ChargePath->SetBinContent (i, j, k, tmp/Ni);
258  }else{
259  Prob_ChargePath->SetBinContent (i, j, k, 0);
260  }
261  }
262  }
263  }
264  }else if(strcmp(ProbabilityMode.c_str(),"Integral")==0){
265  for(int i=0;i<=Prob_ChargePath->GetXaxis()->GetNbins()+1;i++){
266  for(int j=0;j<=Prob_ChargePath->GetYaxis()->GetNbins()+1;j++){
267  float Ni = 0;
268  for(int k=0;k<=Prob_ChargePath->GetZaxis()->GetNbins()+1;k++){ Ni+=deDxMapHandle->binContent(i,j,k);}
269  for(int k=0;k<=Prob_ChargePath->GetZaxis()->GetNbins()+1;k++){
270  float tmp = deDxMapHandle->binContent(i,j,k);
271  if(Ni>0){
272  Prob_ChargePath->SetBinContent (i, j, k, tmp/Ni);
273  }else{
274  Prob_ChargePath->SetBinContent (i, j, k, 0);
275  }
276  }
277  }
278  }
279  }else{
280  throw cms::Exception("WrongConfig for dEdx") << "The ProbabilityMode: " << ProbabilityMode << "is unknown\n";
281  }
282 }
int i
Definition: DBlmapReader.cc:9
int j
Definition: DBlmapReader.cc:9
const T & get() const
Definition: EventSetup.h:56
std::vector< std::vector< double > > tmp
Definition: MVATrainer.cc:100
tuple ProbabilityMode
int DeDxTools::getCharge ( const SiStripCluster cluster,
int &  nSatStrip,
const GeomDetUnit detUnit,
const std::vector< std::vector< float > > &  calibGains,
const unsigned int &  m_off 
)

Definition at line 161 of file DeDxTools.cc.

References SiStripCluster::amplitudes(), RecoTauCleanerPlugins::charge, SiStripCluster::firstStrip(), i, and GeomDet::index().

Referenced by DeDxDiscriminatorLearner::processHit(), HSCPDeDxInfoProducer::processHit(), DeDxHitInfoProducer::processHit(), and DeDxEstimatorProducer::processHit().

162 {
163  const auto & Ampls = cluster->amplitudes();
164 
165  nSatStrip = 0;
166  int charge = 0;
167 
168  if ( calibGains.empty() ) {
169  for(unsigned int i=0;i<Ampls.size();i++){
170  int calibratedCharge = Ampls[i];
171  charge+=calibratedCharge;
172  if(calibratedCharge>=254)nSatStrip++;
173  }
174  }
175  else{
176  for(unsigned int i=0;i<Ampls.size();i++){
177  int calibratedCharge = Ampls[i];
178 
179  auto & gains = calibGains[detUnit.index()-m_off];
180  calibratedCharge = (int)(calibratedCharge / gains[(cluster->firstStrip()+i)/128] );
181  if ( calibratedCharge>=255 ) {
182  if ( calibratedCharge>=1025 )
183  calibratedCharge=255;
184  else
185  calibratedCharge=254;
186  }
187 
188  charge+=calibratedCharge;
189  if(calibratedCharge>=254)nSatStrip++;
190  }
191  }
192  return charge;
193 }
int i
Definition: DBlmapReader.cc:9
uint16_t firstStrip() const
int index() const
Definition: GeomDet.h:97
const std::vector< uint8_t > & amplitudes() const
bool DeDxTools::IsFarFromBorder ( const TrajectoryStateOnSurface trajState,
const GeomDetUnit it 
)

Definition at line 311 of file DeDxTools.cc.

References Surface::bounds(), Bounds::length(), TrajectoryStateOnSurface::localError(), TrajectoryStateOnSurface::localPosition(), HLT_FULL_cff::parameters, LocalTrajectoryError::positionError(), GeomDet::surface(), and LocalError::yy().

Referenced by DeDxDiscriminatorLearner::processHit().

312 {
313  if (dynamic_cast<const StripGeomDetUnit*>(it)==0 && dynamic_cast<const PixelGeomDetUnit*>(it)==0) {
314  edm::LogInfo("DeDxTools::IsFarFromBorder") << "this detID doesn't seem to belong to the Tracker" << std::endl;
315  return false;
316  }
317 
318  LocalPoint HitLocalPos = trajState.localPosition();
319  LocalError HitLocalError = trajState.localError().positionError() ;
320 
321  const BoundPlane plane = it->surface();
322  const TrapezoidalPlaneBounds* trapezoidalBounds( dynamic_cast<const TrapezoidalPlaneBounds*>(&(plane.bounds())));
323  const RectangularPlaneBounds* rectangularBounds( dynamic_cast<const RectangularPlaneBounds*>(&(plane.bounds())));
324 
325  double DistFromBorder = 1.0;
326  //double HalfWidth = it->surface().bounds().width() /2.0;
327  double HalfLength = it->surface().bounds().length() /2.0;
328 
329  if(trapezoidalBounds)
330  {
331  std::array<const float, 4> const & parameters = (*trapezoidalBounds).parameters();
332  HalfLength = parameters[3];
333  //double t = (HalfLength + HitLocalPos.y()) / (2*HalfLength) ;
334  //HalfWidth = parameters[0] + (parameters[1]-parameters[0]) * t;
335  }else if(rectangularBounds){
336  //HalfWidth = it->surface().bounds().width() /2.0;
337  HalfLength = it->surface().bounds().length() /2.0;
338  }else{return false;}
339 
340 // if (fabs(HitLocalPos.x())+HitLocalError.xx() >= (HalfWidth - DistFromBorder) ) return false;//Don't think is really necessary
341  if (fabs(HitLocalPos.y())+HitLocalError.yy() >= (HalfLength - DistFromBorder) ) return false;
342 
343  return true;
344 }
virtual float length() const =0
T y() const
Definition: PV3DBase.h:63
const Bounds & bounds() const
Definition: Surface.h:120
const Plane & surface() const
The nominal surface of the GeomDet.
Definition: GeomDet.h:40
LocalError positionError() const
float yy() const
Definition: LocalError.h:26
const LocalTrajectoryError & localError() const
bool DeDxTools::IsSpanningOver2APV ( unsigned int  FirstStrip,
unsigned int  ClusterSize 
)

Definition at line 286 of file DeDxTools.cc.

Referenced by DeDxDiscriminatorLearner::processHit().

287 {
288  if(FirstStrip==0 ) return true;
289  if(FirstStrip==128 ) return true;
290  if(FirstStrip==256 ) return true;
291  if(FirstStrip==384 ) return true;
292  if(FirstStrip==512 ) return true;
293  if(FirstStrip==640 ) return true;
294 
295  if(FirstStrip<=127 && FirstStrip+ClusterSize>127) return true;
296  if(FirstStrip<=255 && FirstStrip+ClusterSize>255) return true;
297  if(FirstStrip<=383 && FirstStrip+ClusterSize>383) return true;
298  if(FirstStrip<=511 && FirstStrip+ClusterSize>511) return true;
299  if(FirstStrip<=639 && FirstStrip+ClusterSize>639) return true;
300 
301  if(FirstStrip+ClusterSize==127 ) return true;
302  if(FirstStrip+ClusterSize==255 ) return true;
303  if(FirstStrip+ClusterSize==383 ) return true;
304  if(FirstStrip+ClusterSize==511 ) return true;
305  if(FirstStrip+ClusterSize==639 ) return true;
306  if(FirstStrip+ClusterSize==767 ) return true;
307 
308  return false;
309 }
void DeDxTools::makeCalibrationMap ( const std::string &  m_calibrationPath,
const TrackerGeometry tkGeom,
std::vector< std::vector< float > > &  calibGains,
const unsigned int &  m_off 
)

Definition at line 196 of file DeDxTools.cc.

References TrackerGeometry::detUnits(), TrackerGeometry::idToDetUnit(), and GeomDet::index().

Referenced by DeDxDiscriminatorLearner::algoBeginJob(), HSCPDeDxInfoProducer::beginRun(), DeDxHitInfoProducer::beginRun(), and DeDxEstimatorProducer::beginRun().

197 {
198  auto const & dus = tkGeom.detUnits();
199  calibGains.resize(dus.size()-m_off);
200 
201  TChain* t1 = new TChain("SiStripCalib/APVGain");
202  t1->Add(m_calibrationPath.c_str());
203 
204  unsigned int tree_DetId;
205  unsigned char tree_APVId;
206  double tree_Gain;
207  t1->SetBranchAddress("DetId" ,&tree_DetId );
208  t1->SetBranchAddress("APVId" ,&tree_APVId );
209  t1->SetBranchAddress("Gain" ,&tree_Gain );
210 
211  for (unsigned int ientry = 0; ientry < t1->GetEntries(); ientry++) {
212  t1->GetEntry(ientry);
213  auto& gains = calibGains[tkGeom.idToDetUnit(DetId(tree_DetId))->index()-m_off];
214  if(gains.size()<(size_t)(tree_APVId+1)){gains.resize(tree_APVId+1);}
215  gains[tree_APVId] = tree_Gain;
216  }
217  t1->Delete();
218 }
virtual const TrackerGeomDet * idToDetUnit(DetId) const
Return the pointer to the GeomDetUnit corresponding to a given DetId.
virtual const DetUnitContainer & detUnits() const
Returm a vector of all GeomDetUnit.
int index() const
Definition: GeomDet.h:97
Definition: DetId.h:18
bool DeDxTools::shapeSelection ( const SiStripCluster ampls)

Definition at line 10 of file DeDxTools.cc.

References SiStripCluster::amplitudes(), i, and HLT_FULL_cff::noise.

Referenced by DeDxDiscriminatorLearner::processHit(), and DeDxEstimatorProducer::processHit().

11 {
12  // ---------------- COMPTAGE DU NOMBRE DE MAXIMA --------------------------
13  //----------------------------------------------------------------------------
14 // printf("ShapeTest \n");
15  auto const & ampls = clus.amplitudes();
16  Int_t NofMax=0; Int_t recur255=1; Int_t recur254=1;
17  bool MaxOnStart=false;bool MaxInMiddle=false, MaxOnEnd =false;
18  Int_t MaxPos=0;
19  // Début avec max
20  if(ampls.size()!=1 && ((ampls[0]>ampls[1])
21  || (ampls.size()>2 && ampls[0]==ampls[1] && ampls[1]>ampls[2] && ampls[0]!=254 && ampls[0]!=255)
22  || (ampls.size()==2 && ampls[0]==ampls[1] && ampls[0]!=254 && ampls[0]!=255)) ){
23  NofMax=NofMax+1; MaxOnStart=true; }
24 
25  // Maximum entouré
26  if(ampls.size()>2){
27  for (unsigned int i =1; i < ampls.size()-1U; i++) {
28  if( (ampls[i]>ampls[i-1] && ampls[i]>ampls[i+1])
29  || (ampls.size()>3 && i>0 && i<ampls.size()-2U && ampls[i]==ampls[i+1] && ampls[i]>ampls[i-1] && ampls[i]>ampls[i+2] && ampls[i]!=254 && ampls[i]!=255) ){
30  NofMax=NofMax+1; MaxInMiddle=true; MaxPos=i;
31  }
32  if(ampls[i]==255 && ampls[i]==ampls[i-1]) {
33  recur255=recur255+1;
34  MaxPos=i-(recur255/2);
35  if(ampls[i]>ampls[i+1]){NofMax=NofMax+1;MaxInMiddle=true;}
36  }
37  if(ampls[i]==254 && ampls[i]==ampls[i-1]) {
38  recur254=recur254+1;
39  MaxPos=i-(recur254/2);
40  if(ampls[i]>ampls[i+1]){NofMax=NofMax+1;MaxInMiddle=true;}
41  }
42  }
43  }
44  // Fin avec un max
45  if(ampls.size()>1){
46  if(ampls[ampls.size()-1]>ampls[ampls.size()-2]
47  || (ampls.size()>2 && ampls[ampls.size()-1]==ampls[ampls.size()-2] && ampls[ampls.size()-2]>ampls[ampls.size()-3] )
48  || ampls[ampls.size()-1]==255){
49  NofMax=NofMax+1; MaxOnEnd=true; }
50  }
51  // Si une seule strip touchée
52  if(ampls.size()==1){ NofMax=1;}
53 
54 
55 
56 
57  // --- SELECTION EN FONCTION DE LA FORME POUR LES MAXIMA UNIQUES ---------
58  //------------------------------------------------------------------------
59  /*
60  ____
61  | |____
62  ____| | |
63  | | | |____
64  ____| | | | |
65  | | | | | |____
66  __|____|____|____|____|____|____|__
67  C_Mnn C_Mn C_M C_D C_Dn C_Dnn
68  */
69 // bool shapetest=true;
70  bool shapecdtn=false;
71 
72 // Float_t C_M; Float_t C_D; Float_t C_Mn; Float_t C_Dn; Float_t C_Mnn; Float_t C_Dnn;
73  Float_t C_M=0.0; Float_t C_D=0.0; Float_t C_Mn=10000; Float_t C_Dn=10000; Float_t C_Mnn=10000; Float_t C_Dnn=10000;
74  Int_t CDPos;
75  Float_t coeff1=1.7; Float_t coeff2=2.0;
76  Float_t coeffn=0.10; Float_t coeffnn=0.02; Float_t noise=4.0;
77 
78  if(NofMax==1){
79 
80  if(MaxOnStart==true){
81  C_M=(Float_t)ampls[0]; C_D=(Float_t)ampls[1];
82  if(ampls.size()<3) shapecdtn=true ;
83  else if(ampls.size()==3){C_Dn=(Float_t)ampls[2] ; if(C_Dn<=coeff1*coeffn*C_D+coeff2*coeffnn*C_M+2*noise || C_D==255) shapecdtn=true;}
84  else if(ampls.size()>3){ C_Dn=(Float_t)ampls[2]; C_Dnn=(Float_t)ampls[3] ;
85  if((C_Dn<=coeff1*coeffn*C_D+coeff2*coeffnn*C_M+2*noise || C_D==255)
86  && C_Dnn<=coeff1*coeffn*C_Dn+coeff2*coeffnn*C_D+2*noise){
87  shapecdtn=true;}
88  }
89  }
90 
91  if(MaxOnEnd==true){
92  C_M=(Float_t)ampls[ampls.size()-1]; C_D=(Float_t)ampls[ampls.size()-2];
93  if(ampls.size()<3) shapecdtn=true ;
94  else if(ampls.size()==3){C_Dn=(Float_t)ampls[0] ; if(C_Dn<=coeff1*coeffn*C_D+coeff2*coeffnn*C_M+2*noise || C_D==255) shapecdtn=true;}
95  else if(ampls.size()>3){C_Dn=(Float_t)ampls[ampls.size()-3] ; C_Dnn=(Float_t)ampls[ampls.size()-4] ;
96  if((C_Dn<=coeff1*coeffn*C_D+coeff2*coeffnn*C_M+2*noise || C_D==255)
97  && C_Dnn<=coeff1*coeffn*C_Dn+coeff2*coeffnn*C_D+2*noise){
98  shapecdtn=true;}
99  }
100  }
101 
102  if(MaxInMiddle==true){
103  C_M=(Float_t)ampls[MaxPos];
104  int LeftOfMaxPos=MaxPos-1;if(LeftOfMaxPos<=0)LeftOfMaxPos=0;
105  int RightOfMaxPos=MaxPos+1;if(RightOfMaxPos>=(int)ampls.size())RightOfMaxPos=ampls.size()-1;
106  //int after = RightOfMaxPos; int before = LeftOfMaxPos; if (after>=(int)ampls.size() || before<0) std::cout<<"invalid read MaxPos:"<<MaxPos <<"size:"<<ampls.size() <<std::endl;
107  if(ampls[LeftOfMaxPos]<ampls[RightOfMaxPos]){ C_D=(Float_t)ampls[RightOfMaxPos]; C_Mn=(Float_t)ampls[LeftOfMaxPos];CDPos=RightOfMaxPos;} else{ C_D=(Float_t)ampls[LeftOfMaxPos]; C_Mn=(Float_t)ampls[RightOfMaxPos];CDPos=LeftOfMaxPos;}
108  if(C_Mn<coeff1*coeffn*C_M+coeff2*coeffnn*C_D+2*noise || C_M==255){
109  if(ampls.size()==3) shapecdtn=true ;
110  else if(ampls.size()>3){
111  if(CDPos>MaxPos){
112  if(ampls.size()-CDPos-1==0){
113  C_Dn=0; C_Dnn=0;
114  }
115  if(ampls.size()-CDPos-1==1){
116  C_Dn=(Float_t)ampls[CDPos+1];
117  C_Dnn=0;
118  }
119  if(ampls.size()-CDPos-1>1){
120  C_Dn=(Float_t)ampls[CDPos+1];
121  C_Dnn=(Float_t)ampls[CDPos+2];
122  }
123  if(MaxPos>=2){
124  C_Mnn=(Float_t)ampls[MaxPos-2];
125  }
126  else if(MaxPos<2) C_Mnn=0;
127  }
128  if(CDPos<MaxPos){
129  if(CDPos==0){
130  C_Dn=0; C_Dnn=0;
131  }
132  if(CDPos==1){
133  C_Dn=(Float_t)ampls[0];
134  C_Dnn=0;
135  }
136  if(CDPos>1){
137  C_Dn=(Float_t)ampls[CDPos-1];
138  C_Dnn=(Float_t)ampls[CDPos-2];
139  }
140  if(ampls.size()-LeftOfMaxPos>1 && MaxPos+2<(int)(ampls.size())-1){
141  C_Mnn=(Float_t)ampls[MaxPos+2];
142  }else C_Mnn=0;
143  }
144  if((C_Dn<=coeff1*coeffn*C_D+coeff2*coeffnn*C_M+2*noise || C_D==255)
145  && C_Mnn<=coeff1*coeffn*C_Mn+coeff2*coeffnn*C_M+2*noise
146  && C_Dnn<=coeff1*coeffn*C_Dn+coeff2*coeffnn*C_D+2*noise) {
147  shapecdtn=true;
148  }
149 
150  }
151  }
152  }
153  }
154  if(ampls.size()==1){shapecdtn=true;}
155 
156  return shapecdtn;
157 }
int i
Definition: DBlmapReader.cc:9
const std::vector< uint8_t > & amplitudes() const