Belle II Software development
BKLMTrackingModule.cc
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8
9/* Own header. */
10#include <klm/bklm/modules/bklmTracking/BKLMTrackingModule.h>
11
12/* KLM headers. */
13#include <klm/bklm/geometry/GeometryPar.h>
14#include <klm/bklm/modules/bklmTracking/BKLMTrackFinder.h>
15
16/* Basf2 headers. */
17#include <framework/dataobjects/EventMetaData.h>
18#include <framework/datastore/StoreObjPtr.h>
19#include <framework/logging/Logger.h>
20#include <framework/utilities/MathHelpers.h>
21
22using namespace Belle2;
23using namespace Belle2::bklm;
24using namespace CLHEP;
25
26REG_MODULE(BKLMTracking);
27
29 m_effiYX(nullptr),
30 m_effiYZ(nullptr),
31 m_passYX(nullptr),
32 m_totalYX(nullptr),
33 m_passYZ(nullptr),
34 m_totalYZ(nullptr),
37{
38 for (int i = 0; i < 8; ++i) {
39 m_total[0][i] = nullptr;
40 m_total[1][i] = nullptr;
41 m_pass[0][i] = nullptr;
42 m_pass[1][i] = nullptr;
43 m_effiVsLayer[0][i] = nullptr;
44 m_effiVsLayer[1][i] = nullptr;
45 }
46 setDescription("Perform standard-alone straight line tracking for BKLM");
47 addParam("MatchToRecoTrack", m_MatchToRecoTrack, "[bool], whether match BKLMTrack to RecoTrack; (default is false)", false);
48 addParam("MaxAngleRequired", m_maxAngleRequired,
49 "[degree], match BKLMTrack to RecoTrack; angle between them is required to be smaller than (default 10)", double(10.0));
50 addParam("MaxDistance", m_maxDistance,
51 "[cm], During efficiency calculation, distance between track and 2dhit must be smaller than (default 10)", double(10.0));
52 addParam("MaxSigma", m_maxSigma,
53 "[sigma], During efficiency calculation, uncertainty of 2dhit must be smaller than (default 5); ", double(5));
54 addParam("MinHitList", m_minHitList,
55 ", During track finding, a good track after initial seed hits must be larger than is (default 2); ", unsigned(2));
56 addParam("MaxHitList", m_maxHitList,
57 ", During track finding, a good track after initial seed hits must be smaller than is (default 60); ", unsigned(60));
58 addParam("fitGlobalBKLMTrack", m_globalFit,
59 "[bool], do the BKLMTrack fitting in global system (multi-sectors track) or local system (sector by sector) (default is false, local sys.)",
60 false);
61 addParam("StudyEffiMode", m_studyEffi, "[bool], run in efficiency study mode (default is false)", false);
62 addParam("outputName", m_outPath, "[string], output file name containing efficiencies plots ", std::string("bklmEffi.root"));
63}
64
69
71{
72
73 hits2D.isRequired();
74 m_storeTracks.registerInDataStore();
75 m_storeTracks.registerRelationTo(hits2D);
76 m_storeTracks.registerRelationTo(recoTracks);
77 recoHitInformation.registerRelationTo(hits2D);
78 hits2D.registerRelationTo(recoTracks);
79
80 if (m_studyEffi)
81 B2INFO("BKLMTrackingModule:: this module is running in efficiency study mode!");
82
83 m_file = new TFile(m_outPath.c_str(), "recreate");
84 TString hname;
85 std::string labelFB[2] = {"BB", "BF"};
86 int Nbin = 16;
87 float gmin = -350;
88 float gmax = 350;
89 int gNbin = 150;
90
91 m_totalYX = new TH2F("totalYX", " denominator Y vs. X", gNbin, gmin, gmax, gNbin, gmin, gmax);
92 m_passYX = new TH2F("passYX", " numerator Y vs. X", gNbin, gmin, gmax, gNbin, gmin, gmax);
93 m_totalYZ = new TH2F("totalYZ", " denominator Y vs. Z", gNbin, gmin, gmax, gNbin, gmin, gmax);
94 m_passYZ = new TH2F("passYZ", " numerator Y vs. Z", gNbin, gmin, gmax, gNbin, gmin, gmax);
95 m_effiYX = new TH2F("effiYX", " effi. Y vs. X", gNbin, gmin, gmax, gNbin, gmin, gmax);
96 m_effiYZ = new TH2F("effiYZ", " effi. Y vs. X", gNbin, gmin, gmax, gNbin, gmin, gmax);
97 m_effiYX->GetXaxis()->SetTitle("x (cm)");
98 m_effiYX->GetYaxis()->SetTitle("y (cm)");
99 m_effiYZ->GetXaxis()->SetTitle("z (cm)");
100 m_effiYZ->GetYaxis()->SetTitle("y (cm)");
101 for (int iF = 0; iF < 2; iF++) {
102 for (int iS = 0; iS < 8; iS++) {
103 hname.Form("effi_%s%i", labelFB[iF].c_str(), iS);
104 m_effiVsLayer[iF][iS] = new TEfficiency(hname, hname, Nbin, 0, 16);
105 hname.Form("total_%s%i", labelFB[iF].c_str(), iS);
106 m_total[iF][iS] = new TH1F(hname, hname, Nbin, 0, 16);
107 hname.Form("pass_%s%i", labelFB[iF].c_str(), iS);
108 m_pass[iF][iS] = new TH1F(hname, hname, Nbin, 0, 16);
109 }
110 }
111
112}
113
115{
116 StoreObjPtr<EventMetaData> eventMetaData("EventMetaData", DataStore::c_Event);
117 m_runNumber.push_back((int)eventMetaData->getRun());
120}
121
123{
124 m_storeTracks.clear();
125 bool thereIsATrack = false;
126
127 if (!m_studyEffi) {
128 runTracking(0, -1, -1, -1);
129 if (m_storeTracks.getEntries() > 0)
130 thereIsATrack = true;
131 } else {
132 for (int iSection = 0; iSection < 2; iSection++) {
133 for (int iSector = 0; iSector < 8; iSector++) {
134 for (int iLayer = 0; iLayer < 15; iLayer++) {
135 runTracking(1, iSection, iSector, iLayer);
136 if (m_storeTracks.getEntries() > 0)
137 thereIsATrack = true;
138 generateEffi(iSection, iSector, iLayer);
139 //clear tracks so prepare for the next layer efficiency study
140 m_storeTracks.clear();
141 }
142 }
143 }
144 }
145
147 if (thereIsATrack)
149}
150
151void BKLMTrackingModule::runTracking(int mode, int iSection, int iSector, int iLayer)
152{
153 m_storeTracks.clear();
154 //std::list<BKLMTrack*> tracks;
155 //tracks.clear();
156
157 BKLMTrackFitter* m_fitter = new BKLMTrackFitter();
158 BKLMTrackFinder* m_finder = new BKLMTrackFinder();
159 m_finder->setGlobalFit(m_globalFit);
160 if (mode == 1)
161 m_finder->setGlobalFit(false);
162 m_finder->registerFitter(m_fitter);
163
164 if (hits2D.getEntries() < 1)
165 return;
166 if (mode == 1) { //efficiency study
167 for (int j = 0; j < hits2D.getEntries(); j++) {
168 if (hits2D[j]->getSubdetector() != KLMElementNumbers::c_BKLM)
169 continue;
170 hits2D[j]->isOnStaTrack(false);
171 }
172 }
173
174 for (int hi = 0; hi < hits2D.getEntries() - 1; ++hi) {
175 if (hits2D[hi]->getSubdetector() != KLMElementNumbers::c_BKLM)
176 continue;
177
178 if (mode == 1 && isLayerUnderStudy(iSection, iSector, iLayer, hits2D[hi]))
179 continue;
180 if (mode == 1 && !isSectorUnderStudy(iSection, iSector, hits2D[hi]))
181 continue;
182 if (hits2D[hi]->isOnStaTrack())
183 continue;
184 if (hits2D[hi]->isOutOfTime())
185 continue;
186 for (int hj = hi + 1; hj < hits2D.getEntries(); ++hj) {
187
188 if (hits2D[hj]->isOnStaTrack())
189 continue;
190 if (hits2D[hj]->isOutOfTime())
191 continue;
192 if (!m_globalFit && !sameSector(hits2D[hi], hits2D[hj]))
193 continue;
194 if (sameSector(hits2D[hi], hits2D[hj]) &&
195 std::abs(hits2D[hi]->getLayer() - hits2D[hj]->getLayer()) < 3)
196 continue;
197
198 std::list<KLMHit2d*> sectorHitList;
199 //sectorHitList.push_back(hits2D[hi]);
200 //sectorHitList.push_back(hits2D[hj]);
201
202 std::list<KLMHit2d*> seed;
203 seed.push_back(hits2D[hi]);
204 seed.push_back(hits2D[hj]);
205
206 for (int ho = 0; ho < hits2D.getEntries(); ++ho) {
207
208 // Exclude seed hits.
209 if (ho == hi || ho == hj)
210 continue;
211 if (mode == 1 && isLayerUnderStudy(iSection, iSector, iLayer, hits2D[hj]))
212 continue;
213 if (mode == 1 && !isSectorUnderStudy(iSection, iSector, hits2D[hj]))
214 continue;
215 if (hits2D[ho]->isOnStaTrack())
216 continue;
217 if (!m_globalFit && !sameSector(hits2D[ho], hits2D[hi]))
218 continue;
219 // if (hits2D[ho]->getLayer() == hits2D[hi]->getLayer() || hits2D[ho]->getLayer() == hits2D[hj]->getLayer())
220 // continue;
221 if (hits2D[ho]->isOutOfTime())
222 continue;
223 sectorHitList.push_back(hits2D[ho]);
224 }
225
226 /* Require at least four hits (minimum for good track, already two as seed, so here we require 2) but
227 * no more than 60 (most likely noise, 60 would be four good tracks).
228 */
229 if (sectorHitList.size() < m_minHitList || sectorHitList.size() > m_maxHitList)
230 continue;
231
232 std::list<KLMHit2d*> m_hits;
233 if (m_finder->filter(seed, sectorHitList, m_hits)) {
234 BKLMTrack* m_track = m_storeTracks.appendNew();
235 m_track->setTrackParam(m_fitter->getTrackParam());
236 m_track->setTrackParamErr(m_fitter->getTrackParamErr());
237 m_track->setLocalTrackParam(m_fitter->getTrackParamSector());
238 m_track->setLocalTrackParamErr(m_fitter->getTrackParamSectorErr());
239 m_track->setTrackChi2(m_fitter->getChi2());
240 m_track->setNumHitOnTrack(m_fitter->getNumHit());
241 m_track->setIsValid(m_fitter->isValid());
242 m_track->setIsGood(m_fitter->isGood());
243 m_hits.sort(sortByLayer);
244 for (KLMHit2d* hit2d : m_hits) {
245 hit2d->isOnStaTrack(true);
246 m_track->addRelationTo(hit2d);
247 }
248 //tracks.push_back(m_track);
249 //m_track->getTrackParam().Print();
250 //m_track->getTrackParamErr().Print();
251 //match BKLMTrack to RecoTrack
252 if (mode == 0) {
253 RecoTrack* closestTrack = nullptr;
254 if (m_MatchToRecoTrack) {
255 if (findClosestRecoTrack(m_track, closestTrack)) {
256 m_track->addRelationTo(closestTrack);
257 for (KLMHit2d* hit2d : m_hits) {
258 unsigned int sortingParameter = closestTrack->getNumberOfTotalHits();
259 closestTrack->addBKLMHit(hit2d, sortingParameter, RecoHitInformation::OriginTrackFinder::c_LocalTrackFinder);
260 }
261 }
262 }//end match
263 }
264 }
265 }
266 }
267
268 delete m_fitter;
269 delete m_finder;
270
271}
272
278
280{
281 for (long unsigned int i = 0; i < m_runNumber.size(); i++) {
282 float ratio = (float)m_totalEventsWithTracks.at(i) / (float)m_totalEvents.at(i);
283 B2INFO("BKLMTrackingModule:: run " << m_runNumber.at(i) << " --> " << ratio * 100 << "% of events has 1+ BKLMTracks");
284 }
285
286 m_file->cd();
287 for (int iF = 0; iF < 2; iF++) {
288 for (int iS = 0; iS < 8; iS++) {
289 m_effiVsLayer[iF][iS]->Write();
290 m_total[iF][iS]->Write();
291 m_pass[iF][iS]->Write();
292 }
293 }
294
295 for (int i = 0; i < m_totalYX->GetNbinsX(); i++) {
296 for (int j = 0; j < m_totalYX->GetNbinsY(); j++) {
297 float num = m_passYX->GetBinContent(i + 1, j + 1);
298 float denom = m_totalYX->GetBinContent(i + 1, j + 1);
299 if (num > 0) {
300 m_effiYX->SetBinContent(i + 1, j + 1, num / denom);
301 m_effiYX->SetBinError(i + 1, j + 1, sqrt(num * (denom - num) / (denom * denom * denom)));
302 } else {
303 m_effiYX->SetBinContent(i + 1, j + 1, 0);
304 m_effiYX->SetBinError(i + 1, j + 1, 0);
305 }
306
307 num = m_passYZ->GetBinContent(i + 1, j + 1);
308 denom = m_totalYZ->GetBinContent(i + 1, j + 1);
309 if (num > 0) {
310 m_effiYZ->SetBinContent(i + 1, j + 1, num / denom);
311 m_effiYZ->SetBinError(i + 1, j + 1, sqrt(num * (denom - num) / (denom * denom * denom)));
312 } else {
313 m_effiYZ->SetBinContent(i + 1, j + 1, 0);
314 m_effiYZ->SetBinError(i + 1, j + 1, 0);
315 }
316 }
317 }
318
319 m_totalYX->Write();
320 m_passYX->Write();
321 m_totalYZ->Write();
322 m_passYZ->Write();
323 m_effiYX->Write();
324 m_effiYZ->Write();
325 m_file->Close();
326
327}
328
330{
331 if (hit1->getSection() == hit2->getSection() && hit1->getSector() == hit2->getSector())
332 return true;
333 else return false;
334}
335
336
338{
339
340 //StoreArray<RecoTrack> recoTracks;
341 RelationVector<KLMHit2d> bklmHits = bklmTrk->getRelationsTo<KLMHit2d> ();
342
343 if (bklmHits.size() < 1) {
344 B2INFO("BKLMTrackingModule::something is wrong! there is BKLMTrack but no bklmHits");
345 return false;
346 }
347 if (recoTracks.getEntries() < 1) {
348 B2INFO("BKLMTrackingModule::there is no recoTrack");
349 return false;
350 }
351 double oldDistanceSq = INFINITY;
352 double oldAngle = INFINITY;
353 closestTrack = nullptr;
354 //bklmHits are already sorted by layer
355 //possible two hits in one layer?
356 //genfit requires TVector3 rather than XYZVector
357 TVector3 firstBKLMHitPosition(bklmHits[0]->getPosition().X(),
358 bklmHits[0]->getPosition().Y(),
359 bklmHits[0]->getPosition().Z());
360
361 // To get direction (angle) below, we have two points on the bklmTrk:
362 // (x1, TrackParam[0]+TrackParam[1]*x1, TrackParam[2]+TrackParam[3]*x1)
363 // (x2, TrackParam[0]+TrackParam[1]*x2, TrackParam[2]+TrackParam[3]*x2)
364 // the difference vector is
365 // (x2-x1, TrackParam[1]*(x2-x1), TrackParam[3]*(x2-x1))
366 // which is proportional to
367 // (1, TrackParam[1], TrackParam[3]).
368 TVector3 bklmTrkVec(1.0, bklmTrk->getTrackParam()[1], bklmTrk->getTrackParam()[3]);
369
370 TMatrixDSym cov(6);
371 TVector3 pos; // initializes to (0,0,0)
372 TVector3 mom; // initializes to (0,0,0)
373
374 for (RecoTrack& track : recoTracks) {
375 try {
376 genfit::MeasuredStateOnPlane state = track.getMeasuredStateOnPlaneFromLastHit();
378 state.getPosMomCov(pos, mom, cov);
379 if (mom.Y() * pos.Y() < 0) {
380 state = track.getMeasuredStateOnPlaneFromFirstHit();
381 }
382 const TVector3& distanceVec = firstBKLMHitPosition - pos;
383 state.extrapolateToPoint(firstBKLMHitPosition);
384 double newDistanceSq = distanceVec.Mag2();
385 double angle = bklmTrkVec.Angle(mom);
386 // choose closest distance or minimum open angle ?
387 // overwrite old distance
388 if (newDistanceSq < oldDistanceSq) {
389 oldDistanceSq = newDistanceSq;
390 closestTrack = &track;
391 oldAngle = angle;
392 }
393
394 /* if(angle<oldAngle)
395 {
396 oldAngle=angle;
397 closestTrack = &track;
398 }
399 */
400 } catch (genfit::Exception& e) {
401 }// try
402 }
403
404 // can not find matched RecoTrack
405 // problem here is the errors of the track parameters are not considered!
406 // best way is the position or vector direction are required within 5/10 sigma ?
407 if (oldAngle > m_maxAngleRequired)
408 return false;
409 // found matched RecoTrack
410 else return true;
411}
412
413void BKLMTrackingModule::generateEffi(int iSection, int iSector, int iLayer)
414{
415
416 std::set<int> m_pointUsed;
417 m_pointUsed.clear();
418 if (m_storeTracks.getEntries() < 1)
419 return;
420 B2DEBUG(10, "BKLMTracking:generateEffi: " << iSection << " " << iSector << " " << iLayer);
421
422
423 for (int it = 0; it < m_storeTracks.getEntries(); it++) {
424 //if(m_storeTracks[it]->getTrackChi2()>10) continue;
425 //if(m_storeTracks[it]->getNumHitOnTrack()<6) continue;
426 int cnt1 = 0;
427 int cnt2 = 0;
428
429 RelationVector<KLMHit2d> relatedHit2D = m_storeTracks[it]->getRelationsTo<KLMHit2d>();
430 for (const KLMHit2d& hit2D : relatedHit2D) {
431 if (hit2D.getLayer() > iLayer + 1)
432 cnt1++;
433 if (hit2D.getLayer() < iLayer + 1)
434 cnt2++;
435 if (hit2D.getLayer() == iLayer + 1) {
436 B2DEBUG(10, "generateEffi: Hit info. Secti/sector/Lay = " << hit2D.getSection()
437 << "/" << hit2D.getSector() - 1 << "/" << hit2D.getLayer() - 1);
438 B2DEBUG(11, "generateEffi: Hit info. x/y/z = " << hit2D.getPositionX()
439 << "/" << hit2D.getPositionY() << "/" << hit2D.getPositionZ());
440 }
441 }
442
443 if (iLayer != 0 && cnt2 < 1)
444 return;
445 if (iLayer != 14 && cnt1 < 1)
446 return;
448 const bklm::Module* module = m_GeoPar->findModule(iSection, iSector + 1, iLayer + 1);
449 int minPhiStrip = module->getPhiStripMin();
450 int maxPhiStrip = module->getPhiStripMax();
451 int minZStrip = module->getZStripMin();
452 int maxZStrip = module->getZStripMax();
453
454 CLHEP::Hep3Vector local = module->getLocalPosition(minPhiStrip, minZStrip);
455 CLHEP::Hep3Vector local2 = module->getLocalPosition(maxPhiStrip, maxZStrip);
456 float minLocalY, maxLocalY;
457 float minLocalZ, maxLocalZ;
458 if (local[1] > local2[1]) {
459 maxLocalY = local[1];
460 minLocalY = local2[1];
461 } else {
462 maxLocalY = local2[1];
463 minLocalY = local[1];
464 }
465 if (local[2] > local2[2]) {
466 maxLocalZ = local[2];
467 minLocalZ = local2[2];
468 } else {
469 maxLocalZ = local2[2];
470 minLocalZ = local[2];
471 }
472
473 TVectorD trkPar = m_storeTracks[it]->getLocalTrackParam();
474
475 //first layer is the reference layer
476 //if (iSection == 1 && (iSector + 1 ) == 5)
477 // cout<<" local X "<<m_GeoPar->getActiveMiddleRadius(iSection, iSector + 1, iLayer + 1) - m_GeoPar->getActiveMiddleRadius(iSection, iSector + 1, 1) << endl;
478 float reflocalX = fabs(m_GeoPar->getActiveMiddleRadius(iSection, iSector + 1,
479 iLayer + 1) - m_GeoPar->getActiveMiddleRadius(iSection, iSector + 1, 1));
480 //if (iSection == 1 && (iSector + 1 ) == 5)
481 // cout<<" local X "<<m_GeoPar->getActiveMiddleRadius(iSection, iSector + 1, iLayer + 1) - m_GeoPar->getActiveMiddleRadius(iSection, iSector + 1, 1) << endl;
482
483 float reflocalY = trkPar[0] + trkPar[1] * reflocalX;
484 float reflocalZ = trkPar[2] + trkPar[3] * reflocalX;
485
486 //reference module is the first layer
487 //module = m_GeoPar->findModule(iSection, iSector + 1, 1);
488 reflocalX = 0.0;
489 Hep3Vector reflocal(reflocalX, reflocalY, reflocalZ);
490 //Hep3Vector global(localX, localY, localZ);
491 Hep3Vector global(0, 0, 0);
492 module = m_GeoPar->findModule(iSection, iSector + 1, iLayer + 1);
493 global = module->localToGlobal(reflocal);
494 //float localX = module->globalToLocal(global)[0];
495 float localY = module->globalToLocal(global)[1];
496 float localZ = module->globalToLocal(global)[2];
497
498 B2DEBUG(10, "BKLM:generateEffi: RefLocal " << reflocalX << " " << reflocalY << " " << reflocalZ);
499 B2DEBUG(10, "BKLM:generateEffi: Global " << global[0] << " " << global[1] << " " << global[2]);
500 B2DEBUG(10, "BKLM:generateEffi: Local " << 0 << " " << localY << " " << localZ);
501
502
503
504 //geometry cut
505 if (localY > minLocalY && localY < maxLocalY && localZ > minLocalZ && localZ < maxLocalZ) {
506
507 bool m_iffound = false;
508 m_total[iSection][iSector]->Fill(iLayer + 1);
509 m_totalYX->Fill(global[0], global[1]);
510 m_totalYZ->Fill(global[2], global[1]);
511
512 for (int he = 0; he < hits2D.getEntries(); ++he) {
513 if (!isLayerUnderStudy(iSection, iSector, iLayer, hits2D[he]))
514 continue;
515 if (hits2D[he]->isOutOfTime())
516 continue;
517 //if already used, skip
518 if (m_pointUsed.find(he) != m_pointUsed.end())
519 continue;
520
521 double error, sigma;
522 float distance = distanceToHit(m_storeTracks[it], hits2D[he], error, sigma);
523 B2DEBUG(10, "BKLM Dist = " << distance << ", error = " << error);
524 if (distance < m_maxDistance && sigma < m_maxSigma) {
525 m_iffound = true;
526 B2DEBUG(10, "BKLMTracking:generateEffi: Hit found!");;
527 }
528 if (m_iffound) {
529 m_pointUsed.insert(he);
530 //global[0] = hits2D[he]->getPosition()[0];
531 //global[1] = hits2D[he]->getPosition()[1];
532 //global[2] = hits2D[he]->getPosition()[2];
533 m_pass[iSection][iSector]->Fill(iLayer + 1);
534 m_passYX->Fill(global[0], global[1]);
535 m_passYZ->Fill(global[2], global[1]);
536 break;
537 }
538 }
539
540 m_effiVsLayer[iSection][iSector]->Fill(m_iffound, iLayer + 1);
541 //cout<<" global "<<global[0]<<", "<< global[1]<<" "<<global[2]<<endl;
542 //m_effiYX->Fill(m_iffound, global[1], global[0]);
543 //m_effiYZ->Fill(m_iffound, global[1], global[2]);
544 //m_effiYX->SetPassedHistogram(*m_passYX);
545 //m_effiYX->SetTotalHistogram(*m_totalYX);
546 //m_effiYZ->SetPassedHistogram(*m_passYZ);
547 //m_effiYZ->SetTotalHistogram(*m_totalYZ);
548 }
549 }//end of loop tracks
550
551}
552
554{
555
556 return hit1->getLayer() < hit2->getLayer();
557
558}
559
560bool BKLMTrackingModule::isLayerUnderStudy(int section, int iSector, int iLayer, KLMHit2d* hit)
561{
562 if (hit->getSection() == section && hit->getSector() == iSector + 1 && hit->getLayer() == iLayer + 1)
563 return true;
564 else return false;
565}
566
567bool BKLMTrackingModule::isSectorUnderStudy(int section, int iSector, KLMHit2d* hit)
568{
569 if (hit->getSection() == section && hit->getSector() == iSector + 1)
570 return true;
571 else return false;
572}
573
575 double& error,
576 double& sigma)
577{
578
579 double x, y, z, dy, dz;
580
581 error = DBL_MAX;
582 sigma = DBL_MAX;
583
584 TVectorD m_SectorPar = track->getLocalTrackParam();
585
587 const Belle2::bklm::Module* refMod = m_GeoPar->findModule(hit->getSection(), hit->getSector(), 1);
588 const Belle2::bklm::Module* corMod = m_GeoPar->findModule(hit->getSection(), hit->getSector(), hit->getLayer());
589
590 CLHEP::Hep3Vector globalPos(hit->getPositionX(), hit->getPositionY(),
591 hit->getPositionZ());
592 CLHEP::Hep3Vector local = refMod->globalToLocal(globalPos);
593
594 x = local[0] ;
595
596 y = m_SectorPar[ 0 ] + x * m_SectorPar[ 1 ];
597 z = m_SectorPar[ 2 ] + x * m_SectorPar[ 3 ];
598
599 dy = y - local[1];
600 dz = z - local[2];
601
602 double distance = sqrt(dy * dy + dz * dz);
603
604 double hit_localPhiErr = corMod->getPhiStripWidth() / sqrt(12);
605 double hit_localZErr = corMod->getZStripWidth() / sqrt(12);
606
607 //error from tracking is ignored here
608 error = sqrt(square(hit_localPhiErr) + square(hit_localZErr));
609
610 if (error != 0.0) {
611 sigma = distance / error;
612 } else {
613 sigma = DBL_MAX;
614 }
615
616 return (distance);
617
618}
track finding procedure
bool filter(const std::list< KLMHit2d * > &seed, std::list< KLMHit2d * > &hits, std::list< KLMHit2d * > &track)
find associated hits and do fit.
void registerFitter(BKLMTrackFitter *fitter)
Register a fitter if not constructed with one.
void setGlobalFit(bool localOrGlobal)
set the fitting mode, local system or global system
track fitting procedure
CLHEP::HepVector getTrackParamSector()
Get track parameters in the sector locan system, y = p0 + p1 * x, z = p2 + p3 *x, where the first lay...
float getChi2()
Chi square of the fit.
bool isGood()
Is fit good.
CLHEP::HepSymMatrix getTrackParamSectorErr()
Get invariance matrix of track parameters in the sector local system, where the first layer of the se...
int getNumHit()
number of the hits on this track
CLHEP::HepSymMatrix getTrackParamErr()
Get invariance matrix of track parameters in the global system.
CLHEP::HepVector getTrackParam()
Get track parameters in the global system. y = p0 + p1 * x; y = p2 + p3 * z, if in local sector fit m...
bool isValid()
Is fit valid.
Store one BKLM Track as a ROOT object.
Definition BKLMTrack.h:35
void setIsValid(const bool valid)
set the fit valid status
Definition BKLMTrack.h:120
void setTrackChi2(const float chi2)
Set the fitted chi2 of the track.
Definition BKLMTrack.h:108
void setLocalTrackParam(const CLHEP::HepVector &trkPar)
Set track parameters in the sector local system, where the first layer of the sector is used as refer...
Definition BKLMTrack.cc:140
void setTrackParamErr(const CLHEP::HepSymMatrix &trkParErr)
Set invariance matrix of track parameters in the global system.
Definition BKLMTrack.cc:130
void setNumHitOnTrack(const int NumHit)
Set the number of 2d hits on the track.
Definition BKLMTrack.h:114
TVectorD getTrackParam()
Get track parameters in the global system. y = p0 + p1 * x; z = p2 + p3 * x.
Definition BKLMTrack.cc:71
void setIsGood(const bool good)
set the fit good status
Definition BKLMTrack.h:126
void setLocalTrackParamErr(const CLHEP::HepSymMatrix &trkParErr)
Set invariance matrix of track parameters in the sector local system, where the first layer of the se...
Definition BKLMTrack.cc:150
void setTrackParam(const CLHEP::HepVector &trkPar)
Set track parameters in the global system. y = p0 + p1 * x; z = p2 + p3 * x.
Definition BKLMTrack.cc:120
bool m_MatchToRecoTrack
whether match BKLMTrack to RecoTrack
TEfficiency * m_effiVsLayer[2][8]
Efficiency of each layer.
std::vector< int > m_runNumber
run number
bool findClosestRecoTrack(BKLMTrack *bklmTrk, RecoTrack *&closestTrack)
find the closest RecoTrack, match BKLMTrack to RecoTrack, if the matched RecoTrack is found,...
TH2F * m_passYZ
passed event at global position Y vs Z
TH2F * m_effiYX
Efficiency at global position Y vs X.
bool m_studyEffi
option for efficiency study mode, in this mode, the layer under study should not be used in tracking
double m_maxSigma
maximum sigma for hit acceptance during efficiency calculation
double m_maxAngleRequired
angle required between RecoTrack and BKLMTrack, if openangle is larger than m_maxAngleRequired,...
double distanceToHit(BKLMTrack *track, KLMHit2d *hit, double &error, double &sigma)
calculate distance from track to hit
void initialize() override
Initialize at start of job.
unsigned int m_minHitList
minimum number of hits in sector for track finder to run (-2 from initial seed)
std::string m_outPath
output file name containing efficiencies plots
void event() override
Unpack one event and create digits.
void endRun() override
end run stuff
StoreArray< RecoTrack > recoTracks
RecoTrack StoreArray.
void runTracking(int mode, int section, int sector, int layer)
run the track finding and fitting
void terminate() override
Terminate at the end of job.
bklm::GeometryPar * m_GeoPar
bklm GeometryPar
TH1F * m_pass[2][8]
Numerator of each layer.
TH2F * m_totalYX
total event at global position Y vs X
StoreArray< RecoHitInformation > recoHitInformation
RecoHitInformation StoreArray.
int m_runTotalEventsWithTracks
total number of processed events in the run with at lease one BKLMTrack
void beginRun() override
begin run stuff
double m_maxDistance
maximum distance required between track and KLMHit2d to be accepted for efficiency calculation
StoreArray< KLMHit2d > hits2D
KLMHit2d StoreArray.
TFile * m_file
TFile that store efficiency plots.
std::vector< int > m_totalEvents
total number of processed events
bool isLayerUnderStudy(int section, int iSector, int iLayer, KLMHit2d *hit)
judge whether the current layer is understudy
TH1F * m_total[2][8]
Denominator of each layer.
StoreArray< BKLMTrack > m_storeTracks
BKLMTrack StoreArray.
bool m_globalFit
do the BKLMTrack fitting in global system (multi-sectors track) or local system (sector by sector)
unsigned int m_maxHitList
max number of hits in sector for track finder to run
std::vector< int > m_totalEventsWithTracks
total number of processed events with at least one BKLMTrack
TH2F * m_effiYZ
Efficiency at global position Y vs Z.
bool isSectorUnderStudy(int section, int iSector, KLMHit2d *hit)
judge whether the hits come from the sctor understudy
static bool sortByLayer(KLMHit2d *hit1, KLMHit2d *hit2)
my defined sort function using layer number
TH2F * m_totalYZ
total event at global position Y vs Z
bool sameSector(KLMHit2d *hit1, KLMHit2d *hit2)
Judge if two hits come from the same sector.
int m_runTotalEvents
total number of processed events in the run
TH2F * m_passYX
passed event at global position Y vs X
void generateEffi(int section, int sector, int layer)
calculate efficiency
@ c_Event
Different object in each event, all objects/arrays are invalidated after event() function has been ca...
Definition DataStore.h:59
KLM 2d hit.
Definition KLMHit2d.h:33
int getLayer() const
Get layer number.
Definition KLMHit2d.h:132
int getSection() const
Get section number.
Definition KLMHit2d.h:96
float getPositionZ() const
Get hit global position z coordinate.
Definition KLMHit2d.h:306
int getSector() const
Get sector number.
Definition KLMHit2d.h:114
float getPositionX() const
Get hit global position x coordinate.
Definition KLMHit2d.h:288
float getPositionY() const
Get hit global position y coordinate.
Definition KLMHit2d.h:297
void setDescription(const std::string &description)
Sets the description of the module.
Definition Module.cc:214
Module()
Constructor.
Definition Module.cc:30
This is the Reconstruction Event-Data Model Track.
Definition RecoTrack.h:79
bool addBKLMHit(const UsedBKLMHit *bklmHit, const unsigned int sortingParameter, OriginTrackFinder foundByTrackFinder=OriginTrackFinder::c_undefinedTrackFinder)
Adds a bklm hit with the given information to the reco track.
Definition RecoTrack.h:286
unsigned int getNumberOfTotalHits() const
Return the number of cdc + svd + pxd + bklm + eklm hits.
Definition RecoTrack.h:436
Class for type safe access to objects that are referred to in relations.
size_t size() const
Get number of relations.
void addRelationTo(const RelationsInterface< BASE > *object, float weight=1.0, const std::string &namedRelation="") const
Add a relation from this object to another object (with caching).
RelationVector< TO > getRelationsTo(const std::string &name="", const std::string &namedRelation="") const
Get the relations that point from this object to another store array.
Type-safe access to single objects in the data store.
Definition StoreObjPtr.h:96
static GeometryPar * instance(void)
Static method to get a reference to the singleton GeometryPar instance.
Define the geometry of a BKLM module Each sector [octant] contains Modules.
Definition Module.h:76
const CLHEP::Hep3Vector globalToLocal(const CLHEP::Hep3Vector &v, bool reco=false) const
Transform space-point within this module from global to local coordinates.
Definition Module.cc:339
double getPhiStripWidth() const
Get phi-strip width.
Definition Module.h:137
double getZStripWidth() const
Get z-strip width.
Definition Module.h:155
void addParam(const std::string &name, T &paramVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module.
Definition Module.h:559
constexpr T square(const T &x)
Calculate the square of the input.
Definition MathHelpers.h:21
#define REG_MODULE(moduleName)
Register the given module (without 'Module' suffix) with the framework.
Definition Module.h:649
double sqrt(double a)
sqrt for double
Definition beamHelpers.h:28
ExpRunEvt getPosition(const std::vector< Evt > &events, double tEdge)
Get the exp-run-evt number from the event time [hours].
Definition Splitter.h:341
Abstract base class for different kinds of events.