CMS 3D CMS Logo

OMTFinput.cc
Go to the documentation of this file.
1 #include <cassert>
2 #include <iostream>
3 #include <cmath>
4 
8 
12 
13  myOmtfConfig = omtfConfig;
14  clear();
15 
16 }
19 const OMTFinput::vector1D & OMTFinput::getLayerData(unsigned int iLayer, bool giveEta) const{
20  assert(iLayer<measurementsPhi.size());
21 
22  if(giveEta) return measurementsEta[iLayer];
23  return measurementsPhi[iLayer];
24 }
27 std::bitset<128> OMTFinput::getRefHits(unsigned int iProcessor) const{
28 
29  std::bitset<128> refHits;
30 
31  unsigned int iRefHit = 0;
32  for(auto iRefHitDef:myOmtfConfig->getRefHitsDefs()[iProcessor]){
33  int iPhi = getLayerData(myOmtfConfig->getRefToLogicNumber()[iRefHitDef.iRefLayer])[iRefHitDef.iInput];
34  int iEta = getLayerData(myOmtfConfig->getRefToLogicNumber()[iRefHitDef.iRefLayer],true)[iRefHitDef.iInput];
35  if(iPhi<(int)myOmtfConfig->nPhiBins()){
36  refHits.set(iRefHit, iRefHitDef.fitsRange(iPhi));
37  refHitsEta[iRefHit] = iEta;
38  }
39  iRefHit++;
40  }
41 
42  return refHits;
43 }
46 bool OMTFinput::addLayerHit(unsigned int iLayer,
47  unsigned int iInput,
48  int iPhi, int iEta, bool allowOverwrite){
49 
50  bool overwrite = false;
51  assert(iLayer<myOmtfConfig->nLayers());
52  assert(iInput<14);
53 
54  if(iPhi>=(int)myOmtfConfig->nPhiBins()) return true;
55 
56  if(allowOverwrite && measurementsPhi[iLayer][iInput]==iPhi && measurementsEta[iLayer][iInput]==iEta) return true;
57 
58  if(measurementsPhi[iLayer][iInput]!=(int)myOmtfConfig->nPhiBins()) ++iInput;
59  if(measurementsPhi[iLayer][iInput]!=(int)myOmtfConfig->nPhiBins()) overwrite = true;
60 
61  if(iInput>13) return true;
62 
63  measurementsPhi[iLayer][iInput] = iPhi;
64  measurementsEta[iLayer][iInput] = iEta;
65 
66  return overwrite;
67 }
71  unsigned int iEvent,
72  unsigned int iProcessor){
73 
74  measurementsPhi = aReader->readEvent(iEvent, iProcessor);
75  measurementsEta = aReader->readEvent(iEvent, iProcessor, true);
76 
77 }
80 void OMTFinput::mergeData(const OMTFinput *aInput){
81 
82  for(unsigned int iLayer=0;iLayer<myOmtfConfig->nLayers();++iLayer){
83  const OMTFinput::vector1D & aPhiVec = aInput->getLayerData(iLayer,false);
84  const OMTFinput::vector1D & aEtaVec = aInput->getLayerData(iLayer,true);
85  if(aPhiVec.empty()) continue;
86 
87  OMTFinput::vector1D layerData = getLayerData(iLayer, false);
88  for(unsigned int iInput=0;iInput<14;++iInput){
89  addLayerHit(iLayer,iInput,aPhiVec[iInput],aEtaVec[iInput]);
90  }
91  }
92 }
96 
97  vector1D aLayer1D(14,myOmtfConfig->nPhiBins());
98  measurementsPhi.assign(myOmtfConfig->nLayers(),aLayer1D);
99  measurementsEta.assign(myOmtfConfig->nLayers(),aLayer1D);
100  refHitsEta.assign(128,myOmtfConfig->nPhiBins());
101 
102 }
105 void OMTFinput::shiftMyPhi(int phiShift){
106 
107  int lowScaleEnd = std::pow(2,myOmtfConfig->nPhiBits()-1);
108  int highScaleEnd = lowScaleEnd-1;
109 
110 for(unsigned int iLogicLayer=0;iLogicLayer<measurementsPhi.size();++iLogicLayer){
111  for(unsigned int iHit=0;iHit<measurementsPhi[iLogicLayer].size();++iHit){
112  if(!myOmtfConfig->getBendingLayers().count(iLogicLayer) &&
113  measurementsPhi[iLogicLayer][iHit]<(int)myOmtfConfig->nPhiBins()){
114  if(measurementsPhi[iLogicLayer][iHit]<0) measurementsPhi[iLogicLayer][iHit]+=myOmtfConfig->nPhiBins();
115  measurementsPhi[iLogicLayer][iHit]-=phiShift;
116  if(measurementsPhi[iLogicLayer][iHit]<0) measurementsPhi[iLogicLayer][iHit]+=myOmtfConfig->nPhiBins();
117  measurementsPhi[iLogicLayer][iHit]+=-lowScaleEnd;
118  if(measurementsPhi[iLogicLayer][iHit]<-lowScaleEnd ||
119  measurementsPhi[iLogicLayer][iHit]>highScaleEnd) measurementsPhi[iLogicLayer][iHit] = (int)myOmtfConfig->nPhiBins();
120  }
121  }
122  }
123 }
126 std::ostream & operator << (std::ostream &out, const OMTFinput & aInput){
127 
128 for(unsigned int iLogicLayer=0;iLogicLayer<aInput.measurementsPhi.size();++iLogicLayer){
129  out<<"Logic layer: "<<iLogicLayer<<" Hits: ";
130  for(unsigned int iHit=0;iHit<aInput.measurementsPhi[iLogicLayer].size();++iHit){
131  out<<aInput.measurementsPhi[iLogicLayer][iHit]<<"\t";
132  }
133  out<<std::endl;
134  }
135  return out;
136 }
const std::set< int > & getBendingLayers() const
bool addLayerHit(unsigned int iLayer, unsigned int iInput, int iPhi, int iEta, bool allowOverwrite=true)
Definition: OMTFinput.cc:46
void clear()
Reset vectors with data.
Definition: OMTFinput.cc:95
vector2D measurementsEta
Definition: OMTFinput.h:59
void mergeData(const OMTFinput *aInput)
Definition: OMTFinput.cc:80
const OMTFConfiguration * myOmtfConfig
Definition: OMTFinput.h:64
void shiftMyPhi(int phiShift)
Apply shift to all data.
Definition: OMTFinput.cc:105
friend std::ostream & operator<<(std::ostream &out, const OMTFinput &aInput)
Definition: OMTFinput.cc:126
unsigned int nLayers() const
unsigned int nPhiBins() const
std::vector< std::vector< int > > readEvent(unsigned int iEvent=0, unsigned int iProcessor=0, bool readEta=false)
std::vector< int > vector1D
Definition: OMTFinput.h:15
const OMTFinput::vector1D & getLayerData(unsigned int iLayer, bool giveEta=false) const
Definition: OMTFinput.cc:19
vector1D refHitsEta
RefHitsEta.
Definition: OMTFinput.h:62
unsigned int nPhiBits() const
int iEvent
Definition: GenABIO.cc:230
const std::vector< std::vector< RefHitDef > > & getRefHitsDefs() const
OMTFinput(const OMTFConfiguration *)
Definition: OMTFinput.cc:11
std::bitset< 128 > getRefHits(unsigned int iProcessor) const
Definition: OMTFinput.cc:27
const std::vector< int > & getRefToLogicNumber() const
void readData(XMLConfigReader *aReader, unsigned int iEvent=0, unsigned int iProcessor=0)
Read data from a XML file.
Definition: OMTFinput.cc:70
Power< A, B >::type pow(const A &a, const B &b)
Definition: Power.h:40
vector2D measurementsPhi
Definition: OMTFinput.h:54