ATLAS Offline Software
Converter.h
Go to the documentation of this file.
1 /* emacs: this is -*- c++ -*- */
12 #ifndef TrigInDetAnalysis_Converter_H
13 #define TrigInDetAnalysis_Converter_H
14 
15 
16 #include "TMath.h"
17 
18 #include "TrigInDetEvent/TrigInDetTrackCollection.h"
19 
20 #include "Particle/TrackParticle.h"
21 #include "Particle/TrackParticleContainer.h"
22 #include "muonEvent/MuonContainer.h"
23 
24 #include "TrigInDetAnalysis/Track.h"
25 
26 
27 class Converter {
28 
29  public:
30 
31  Converter() {
32  m_beamX = 0;
33  m_beamY = 0;
34  m_beamZ = 0;
35  }
36 
37  Converter(double x, double y, double z) {
38  m_beamX = x;
39  m_beamY = y;
40  m_beamZ = z;
41  }
42 
43 
44  // Method adding a track
45  void addTrack(TIDA::Track* t) {
46  m_tracks.push_back(t);
47  }
48 
49  // get the tracks
50  const std::vector<TIDA::Track*>& tracks() const {return m_tracks;}
51 
52  // clear tracks
53  void clear() {m_tracks.clear();}
54 
55  // Method converting TrigInDetTrack objects
56  void selectTracks(const TrigInDetTrackCollection* trigtracks) {
57 
58  TrigInDetTrackCollection::const_iterator trackitr = trigtracks->begin();
59  TrigInDetTrackCollection::const_iterator trackend = trigtracks->end();
60 
61  while ( trackitr!=trackend ) {
62 
63  double eta = (*trackitr)->param()->eta();
64  double phi = (*trackitr)->param()->phi0();
65  double z0 = (*trackitr)->param()->z0();
66  double pT = (*trackitr)->param()->pT();
67  double d0 = (*trackitr)->param()->a0();
68  //double theta = 2*atan2(exp(-(*trackitr)->param()->eta()),1);
69 
70  double deta = (*trackitr)->param()->eeta();
71  double dphi = (*trackitr)->param()->ephi0();
72  double dz0 = (*trackitr)->param()->ez0();
73  double dpT = (*trackitr)->param()->epT();
74  double dd0 = (*trackitr)->param()->ea0();
75 
76  int algoid = (*trackitr)->algorithmId();
77 
78  int nBlayerHits = ((*trackitr)->HitPattern() & 0x1);
79  int nPixelHits = 2*(*trackitr)->NPixelSpacePoints(); // NB: for comparison with offline
80  int nSctHits = 2*(*trackitr)->NSCT_SpacePoints(); // a spacepoint is 2 "hits"
81  int nStrawHits = (*trackitr)->NStrawHits();
82  int nTrHits = (*trackitr)->NTRHits();
83 
84  int nSiHits = nPixelHits + nSctHits;
85  bool expectBL = false; //not available with TrigInDetTrack
86 
87  unsigned hitPattern = (*trackitr)->HitPattern();
88  unsigned multiPattern = 0;
89 
90  double chi2 = (*trackitr)->chi2();
91  double dof = 0;
92 
93  // Shift d0 and z0 according to beam position
94  ipCorr(d0, z0, d0, z0, phi, eta, pT);
95 
96  // Create and save Track
97  TIDA::Track* t = new TIDA::Track(eta, phi, z0, d0, pT, chi2, dof,
98  deta, dphi, dz0, dd0, dpT,
99  nBlayerHits, nPixelHits, nSctHits, nSiHits,
100  nStrawHits, nTrHits, hitPattern, multiPattern,
101  algoid,
102  expectBL ) ;
103 
104  addTrack(t);
105  trackitr++;
106  }
107  }
108 
109 
110  // Method converting TrackParticle objects
111  void selectTracks( const Rec::TrackParticleContainer* trigtracks ) {
112 
113  Rec::TrackParticleContainer::const_iterator trackitr = trigtracks->begin();
114  Rec::TrackParticleContainer::const_iterator trackend = trigtracks->end();
115 
116  static int hpmap[20] = { 0, 1, 2, 7, 8, 9, 3, 4, 5, 6, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 };
117 
118  while ( trackitr!=trackend ) {
119 
120  const Trk::MeasuredPerigee* measPer = (*trackitr)->measuredPerigee();
121 
122  CLHEP::HepVector perigeeParams = measPer->parameters();
123  double pT = measPer->pT();
124  double eta = measPer->eta();
125  double phi = perigeeParams[Trk::phi0];
126  double z0 = perigeeParams[Trk::z0];
127  double d0 = perigeeParams[Trk::d0];
128  //double theta = perigeeParams[Trk::theta];
129 
130  double deta = 1;
131  double dphi = 1;
132  double dz0 = 1;
133  double dd0 = 1;
134  double dpT = 1;
135 
136  // Check number of hits
137  // NB: a spacepoint is two offline "hits", so a pixel spacepoint is really
138  // 2 "hits" and an offline SCT "hit" is really a 1D cluster, so two intersetcting
139  // stereo clusters making a spacepoint are two "hits"
140  const Trk::TrackSummary *summary = (*trackitr)->trackSummary();
141  int nBlayerHits = 2*summary->get(Trk::numberOfBLayerHits);
142  int nPixelHits = 2*summary->get(Trk::numberOfPixelHits);
143  int nSctHits = summary->get(Trk::numberOfSCTHits);
144  int nStrawHits = summary->get(Trk::numberOfTRTHits);
145  int nTrHits = summary->get(Trk::numberOfTRTHighThresholdHits);
146  int nSiHits = nPixelHits + nSctHits;
147  bool expectBL = false; // Not stored for Rec::TrackParticle
148 
149  const Trk::FitQuality *quality = (*trackitr)->fitQuality();
150  double chi2 = quality->chiSquared();
151  double dof = quality->numberDoF();
152 
153  unsigned bitmap = 0;
154 
155  for ( int ih=0 ; ih<20 ; ih++ ) {
156  if ( summary->isHit(Trk::DetectorType(ih)) ) bitmap |= ( 1<<hpmap[ih] );
157  }
158 
159  // Shift d0 and z0 according to beam position
160  ipCorr(d0, z0, d0, z0, phi, eta, pT);
161 
162  // Create and save Track
163  TIDA::Track* t = new TIDA::Track(eta, phi, z0, d0, pT, chi2, dof,
164  deta, dphi, dz0, dd0, dpT,
165  nBlayerHits, nPixelHits, nSctHits, nSiHits,
166  nStrawHits, nTrHits, bitmap, 0,
167  -1,
168  expectBL) ;
169 
170  addTrack(t);
171  trackitr++;
172  }
173 
174  }
175 
176  // Method converting TrackParticle objects
177  void selectTracks( const Analysis::MuonContainer* muontracks ) {
178 
179  static int hpmap[20] = { 0, 1, 2, 7, 8, 9, 3, 4, 5, 6, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 };
180 
181  Analysis::MuonContainer::const_iterator muonitr = muontracks->begin();
182  Analysis::MuonContainer::const_iterator muonend = muontracks->end();
183 
184  while(muonitr!=muonend) {
185 
186  bool hasInDet = (*muonitr)->hasInDetTrackParticle();
187  const Rec::TrackParticle *trackitr=NULL;
188  if(hasInDet) trackitr=(*muonitr)->inDetTrackParticle();
189  muonitr++;
190  if(!hasInDet) continue;
191 
192  const Trk::MeasuredPerigee* measPer = trackitr->measuredPerigee();
193 
194  CLHEP::HepVector perigeeParams = measPer->parameters();
195  double pT = measPer->pT();
196  double eta = measPer->eta();
197  double phi = perigeeParams[Trk::phi0];
198  double z0 = perigeeParams[Trk::z0];
199  double d0 = perigeeParams[Trk::d0];
200  //double theta = perigeeParams[Trk::theta];
201 
202  double deta = 1;
203  double dphi = 1;
204  double dz0 = 1;
205  double dd0 = 1;
206  double dpT = 1;
207 
208  // Check number of hits
209  // NB: a spacepoint is two offline "hits", so a pixel spacepoint is really
210  // 2 "hits" and an offline SCT "hit" is really a 1D cluster, so two intersetcting
211  // stereo clusters making a spacepoint are two "hits"
212  const Trk::TrackSummary *summary = trackitr->trackSummary();
213  int nBlayerHits = 2*summary->get(Trk::numberOfBLayerHits);
214  int nPixelHits = 2*summary->get(Trk::numberOfPixelHits);
215  int nSctHits = summary->get(Trk::numberOfSCTHits);
216  int nStrawHits = summary->get(Trk::numberOfTRTHits);
217  int nTrHits = summary->get(Trk::numberOfTRTHighThresholdHits);
218 
219  int nSiHits = nPixelHits + nSctHits;
220 
221  const Trk::FitQuality *quality = trackitr->fitQuality();
222  double chi2 = quality->chiSquared();
223  double dof = quality->quality->numberDoF();
224 
225  unsigned bitmap = 0;
226 
227  for ( int ih=0 ; ih<20 ; ih++ ) {
228  if ( summary->isHit(Trk::DetectorType(ih)) ) bitmap |= ( 1<<hpmap[ih] );
229  }
230 
231  // Shift d0 and z0 according to beam position
232  ipCorr(d0, z0, d0, z0, phi, eta, pT);
233 
234  // Create and save Track
235  TIDA::Track* t = new TIDA::Track(eta, phi, z0, d0, pT, chi2, dof,
236  deta, dphi, dz0, dd0, dpT,
237  nBlayerHits, nPixelHits, nSctHits, nSiHits,
238  nStrawHits, nTrHits, bitmap, 0,
239  -1) ;
240 
241  addTrack(t);
242  }
243 
244  }
245 
246  protected:
247 
248  double phiCorr(double phi){
249 
250  if (phi < -TMath::Pi()) phi += 2*TMath::Pi();
251  if (phi > TMath::Pi()) phi -= 2*TMath::Pi();
252 
253  return phi;
254  }
255 
256 
257  void ipCorr(double d0, double z0, double& d0c, double& z0c, double phi0, double eta, double pt) {
258 
259  double sn = sin(phi0);
260  double cs = cos(phi0);
261  double sd0 = (d0 != 0 ? d0/fabs(d0) : 1);
262  double spt = (pt != 0 ? pt/fabs(pt) : 1);
263 
264  if (fabs(pt) >= 1*CLHEP::TeV) {
265 
266  d0c = d0 + m_beamX*sn - m_beamY*cs;
267 
268  } else {
269 
270  double rc = fabs(pt)*15.0/(9.0*1.042);
271 
272  double xc = (fabs(d0)-spt*sd0*rc)*cos(phi0+M_PI/2*sd0) - m_beamX;
273  double yc = (fabs(d0)-spt*sd0*rc)*sin(phi0+M_PI/2*sd0) - m_beamY;
274 
275  double newphi;
276  double xd01,yd01,xd02,yd02;
277 
278  if (xc == 0) {
279  xd01 = 0; yd01 = rc+yc;
280  xd02 = 0; yd02 = -rc+yc;
281  } else {
282  xd01 = xc+yc/xc*yc+sqrt(pow((xc+yc/xc*yc),2)-xc*xc-yc*yc+rc*rc); yd01 = yc/xc*xd01;
283  xd02 = xc+yc/xc*yc-sqrt(pow((xc+yc/xc*yc),2)-xc*xc-yc*yc+rc*rc); yd02 = yc/xc*xd02;
284  }
285 
286  double r1 = sqrt(xd01*xd01+yd01*yd01);
287  double r2 = sqrt(xd02*xd02+yd02*yd02);
288 
289  double phiV;
290 
291  if (r1 < r2)
292  phiV = atan2(yd01,xd01);
293  else
294  phiV = atan2(yd02,xd02);
295 
296  double phi1 = phiCorr(phiV+M_PI/2);
297  double phi2 = phiCorr(phiV-M_PI/2);
298 
299  if (fabs(phiCorr(phi1-phi0))<=fabs(phiCorr(phi2-phi0)))
300  newphi = phi1;
301  else
302  newphi = phi2;
303 
304  d0c = fabs(sqrt(xc*xc+yc*yc)-rc)*sin(phiV-newphi);
305 
306  double theta=2*atan2(exp(-eta),1);
307  double theta_save=theta;
308  theta = theta - (1+spt)/2*M_PI;
309  if (theta>0) theta = theta_save;
310 
311  double deltaz0= -spt*rc/tan(theta)*phiCorr(phiCorr(newphi)-phiCorr(phi0));
312  z0c = z0 + deltaz0;
313 
314  }
315  }
316 
317 
318  protected:
319 
320  double m_beamX, m_beamY, m_beamZ;
321  std::vector<TIDA::Track*> m_tracks;
322 
323 };
324 
325 
326 #endif // TrigInDetAnalysis_Converter_H
Converter::selectTracks
void selectTracks(const TrigInDetTrackCollection *trigtracks)
Definition: Converter.h:56
Trk::numberOfPixelHits
@ numberOfPixelHits
number of pixel layers on track with absence of hits
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:57
Converter::Converter
Converter()
Definition: Converter.h:31
CalculateHighPtTerm.pT
pT
Definition: ICHEP2016/CalculateHighPtTerm.py:57
Track.h
Converter::ipCorr
void ipCorr(double d0, double z0, double &d0c, double &z0c, double phi0, double eta, double pt)
Definition: Converter.h:257
DataModel_detail::const_iterator
Const iterator class for DataVector/DataList.
Definition: DVLIterator.h:82
Trk::numberOfTRTHighThresholdHits
@ numberOfTRTHighThresholdHits
total number of TRT hits which pass the high threshold
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:87
Converter::m_beamY
double m_beamY
Definition: Converter.h:320
TrackParticle.h
phi
Scalar phi() const
phi method
Definition: AmgMatrixBasePlugin.h:64
TauGNNUtils::Variables::Track::nPixelHits
bool nPixelHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:542
TrigInDetTrackCollection
Definition: TrigInDetTrackCollection.h:13
eta
Scalar eta() const
pseudorapidity method
Definition: AmgMatrixBasePlugin.h:79
Trk::ParametersT
Dummy class used to allow special convertors to be called for surfaces owned by a detector element.
Definition: EMErrorDetail.h:25
InDetAccessor::phi0
@ phi0
Definition: InDetAccessor.h:33
theta
Scalar theta() const
theta method
Definition: AmgMatrixBasePlugin.h:71
conifer::pow
constexpr int pow(int x)
Definition: conifer.h:20
xAOD::JetInput::Track
@ Track
Definition: JetContainerInfo.h:61
MuonContainer.h
Trk::TrackParticleBase::trackSummary
const TrackSummary * trackSummary() const
accessor function for TrackSummary.
Definition: TrackParticleBase.h:247
test_pyathena.pt
pt
Definition: test_pyathena.py:11
Converter::selectTracks
void selectTracks(const Analysis::MuonContainer *muontracks)
Definition: Converter.h:177
M_PI
#define M_PI
Definition: ActiveFraction.h:11
Trk::z0
@ z0
Definition: ParamDefs.h:70
TrigInDetTrackCollection.h
python.SystemOfUnits.TeV
int TeV
Definition: SystemOfUnits.py:158
read_hist_ntuple.t
t
Definition: read_hist_ntuple.py:5
drawFromPickle.cos
cos
Definition: drawFromPickle.py:36
drawFromPickle.exp
exp
Definition: drawFromPickle.py:36
x
#define x
Converter::selectTracks
void selectTracks(const Rec::TrackParticleContainer *trigtracks)
Definition: Converter.h:111
Trk::numberOfBLayerHits
@ numberOfBLayerHits
these are the hits in the 0th pixel layer?
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:51
Converter::phiCorr
double phiCorr(double phi)
Definition: Converter.h:248
Converter::tracks
const std::vector< TIDA::Track * > & tracks() const
Definition: Converter.h:50
TauGNNUtils::Variables::Track::nSiHits
bool nSiHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:691
z
#define z
Converter::m_beamX
double m_beamX
Definition: Converter.h:320
Trk::numberOfSCTHits
@ numberOfSCTHits
number of SCT holes
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:71
Converter::m_beamZ
double m_beamZ
Definition: Converter.h:320
Trk::TrackParticleBase::fitQuality
const FitQuality * fitQuality() const
accessor function for FitQuality.
Definition: TrackParticleBase.h:259
chi2
double chi2(TH1 *h0, TH1 *h1)
Definition: comparitor.cxx:522
TRT::Track::d0
@ d0
Definition: InnerDetector/InDetCalibEvent/TRT_CalibData/TRT_CalibData/TrackInfo.h:62
drawFromPickle.tan
tan
Definition: drawFromPickle.py:36
Trk::FitQuality
Class to represent and store fit qualities from track reconstruction in terms of and number of degre...
Definition: FitQuality.h:97
TRT::Track::z0
@ z0
Definition: InnerDetector/InDetCalibEvent/TRT_CalibData/TRT_CalibData/TrackInfo.h:63
Analysis::MuonContainer
definition of StoreGate container holding a vector of Analysis::Muon
Definition: Reconstruction/MuonIdentification/muonEvent/muonEvent/MuonContainer.h:38
Converter::addTrack
void addTrack(TIDA::Track *t)
Definition: Converter.h:45
Converter::clear
void clear()
Definition: Converter.h:53
Trk::numberOfTRTHits
@ numberOfTRTHits
number of TRT outliers
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:79
Converter
Definition: Converter.h:27
Trk::TrackSummary
A summary of the information contained by a track.
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:287
Trk::d0
@ d0
Definition: ParamDefs.h:69
Rec::TrackParticleContainer
Definition: Reconstruction/Particle/Particle/TrackParticleContainer.h:33
Rec::TrackParticle
Definition: Reconstruction/Particle/Particle/TrackParticle.h:47
Rec::TrackParticle::measuredPerigee
const Trk::Perigee * measuredPerigee() const
Accessor method for Perigee.
Definition: Reconstruction/Particle/Particle/TrackParticle.h:180
DataVector::end
const_iterator end() const noexcept
Return a const_iterator pointing past the end of the collection.
Trk::DetectorType
DetectorType
enumerates the various detector types currently accessible from the isHit() method.
Definition: Tracking/TrkEvent/TrkTrackSummary/TrkTrackSummary/TrackSummary.h:231
y
#define y
TIDA::Track
Definition: Trigger/TrigAnalysis/TrigInDetAnalysis/TrigInDetAnalysis/Track.h:26
Trk::FitQuality::chiSquared
double chiSquared() const
returns the of the overall track fit
Definition: FitQuality.h:56
Trk::FitQuality::numberDoF
int numberDoF() const
returns the number of degrees of freedom of the overall track or vertex fit as integer
Definition: FitQuality.h:60
drawFromPickle.sin
sin
Definition: drawFromPickle.py:36
Converter::m_tracks
std::vector< TIDA::Track * > m_tracks
Definition: Converter.h:321
Trk::phi0
@ phi0
Definition: ParamDefs.h:71
Converter::Converter
Converter(double x, double y, double z)
Definition: Converter.h:37
DataVector::begin
const_iterator begin() const noexcept
Return a const_iterator pointing at the beginning of the collection.
TrackParticleContainer.h
SCT_Monitoring::summary
@ summary
Definition: SCT_MonitoringNumbers.h:65
Generated on Fri Jun 28 2024 21:12:53 for ATLAS Offline Software by   doxygen 1.8.18