1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
18 //_________________________________________________________________________
19 // This is a TTask that made the calculation of the Time zero using TOF.
20 // Description: The algorithm used to calculate the time zero of
21 // interaction using TOF detector is the following.
22 // We select in the MonteCarlo some primary particles - or tracks in
23 // the following - that strike the TOF detector (the larger part are
24 // pions, kaons or protons).
25 // We choose a set of 10 selected tracks, for each track You have the
26 // length of the track when the TOF is reached (a standard TOF hit
27 // does not contain this additional information, this is the reason
28 // why we implemented a new time zero dedicated TOF hit class
29 // AliTOFhitT0; in order to store this type of hit You have to use the
30 // AliTOFv4T0 as TOF class in Your Config.C. In AliTOFv4T0 the
31 // StepManager was modified in order to fill the TOF hit branch with
32 // this type of hits; in fact the AliTOF::AddT0Hit is called rather
33 // that the usual AliTOF::AddHit), the momentum at generation (from
34 // TreeK) and the time of flight given by the TOF detector.
35 // (Observe that the ctor of the AliTOF class, when the AliTOFv4T0
36 // class is used, is called with the "tzero" option: it is in order
37 // create the fHits TClonesArray filled with AliTOFhitT0 objects,
38 // rather than with normal AliTOFhit)
39 // Then Momentum and time of flight for each track are smeared
40 // according to known experimental resolution (all sources of error
41 // have been token into account).
42 // Let consider now only one set of 10 tracks (the algorithm is the
43 // same for all sets).
44 // Assuming the (mass) hypothesis that each track can be AUT a pion,
45 // AUT a kaon, AUT a proton, we consider all the 3 at 10 possible
47 // For each track in each (mass) configuration
48 // (a configuration can be
49 // e.g. pion/pion/kaon/proton/pion/proton/kaon/kaon/pion/pion)
50 // we calculate the time zero (we know in fact the velocity of the
51 // track after the assumption about its mass, the time of flight given
52 // by the TOF, and the corresponding path travelled till the TOF
53 // detector). Then for each mass configuration we have 10 time zero
54 // and we can calculate the ChiSquare for the current configuration
55 // using the weighted mean over all 10 time zero.
56 // We call the best assignment the mass configuration that gives the
57 // minimum value of the ChiSquare.
58 // We plot the weighted mean over all 10 time zero for the best
59 // assignment, the ChiSquare for the best assignment and the
60 // corresponding confidence level.
61 // The strong assumption is the MC selection of primary particles. It
62 // will be introduced in the future also some more realistic
63 // simulation about this point.
66 // root [0] AliTOFT0 * tzero = new AliTOFT0("galice.root")
67 // Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
68 // root [1] tzero->ExecuteTask()
69 // root [2] tzero->ExecuteTask("tim")
70 // // available parameters:
71 // tim - print benchmarking information
72 // all - print usefull informations about the number of
73 // misidentified tracks and a comparison about the
74 // true configuration (known from MC) and the best
76 //-- Author: F. Pierella
77 //////////////////////////////////////////////////////////////////////////////
79 #include <Riostream.h>
82 #include <TBenchmark.h>
84 #include <TClonesArray.h>
89 #include <TParticle.h>
94 #include <TVirtualMC.h>
96 #include "AliDetector.h"
101 #include "AliTOFT0.h"
102 #include "AliTOFhitT0.h"
103 #include "AliTOFv4T0.h"
104 #include "AliTOFv5T0.h"
108 //____________________________________________________________________________
109 AliTOFT0::AliTOFT0():TTask("AliTOFT0","")
115 //____________________________________________________________________________
116 AliTOFT0::AliTOFT0(char* headerFile, Int_t nEvents):TTask("AliTOFT0","")
122 fNevents=nEvents ; // Number of events for which calculate the T0,
123 // default 0: it means all evens in current file
124 fLowerMomBound=1.5; // [GeV/c] default value
125 fUpperMomBound=2. ; // [GeV/c] default value
126 fTimeResolution = 1.2e-10; // 120 ps by default
127 fHeadersFile = headerFile ;
129 TFile * file = (TFile*) gROOT->GetFile(fHeadersFile.Data() ) ;
131 //File was not opened yet
133 if(fHeadersFile.Contains("rfio"))
134 file = TFile::Open(fHeadersFile,"update") ;
136 file = new TFile(fHeadersFile.Data(),"update") ;
137 gAlice = (AliRun *) file->Get("gAlice") ;
140 // add Task to //root/Tasks folder
141 TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ;
142 roottasks->Add(this) ;
145 //____________________________________________________________________________
146 AliTOFT0::AliTOFT0(const AliTOFT0 & tzero):TTask("AliTOFT0","")
150 ( (AliTOFT0 &)tzero ).Copy(*this);
153 //____________________________________________________________________________
154 AliTOFT0::~AliTOFT0()
159 //____________________________________________________________________________
160 void AliTOFT0::Exec(Option_t *option)
163 // calculate T0 distribution for all events using chisquare
166 Int_t nmisidentified=0;
167 Int_t nmisidentified0=0;
168 Int_t nmisidentified1=0;
169 Int_t nmisidentified2=0;
170 Int_t nmisidentified3=0;
171 Int_t nmisidentified4=0;
172 Int_t nmisidentified5=0;
173 Int_t nmisidentified6=0;
174 Int_t nmisidentified7=0;
175 Int_t nmisidentified8=0;
176 Int_t nmisidentified9=0;
181 Float_t timeresolutioninns=fTimeResolution*(1.e+9); // convert in [ns]
182 const Int_t kUPDATE = 5; // for visual option
188 if(strstr(option,"visual")){
189 // Create a new canvas.
190 //c1 = new TCanvas("c1","Dynamic Visual Filling of time zero histo",10,10,500,500);
191 c1 = new TCanvas("c1","Dynamic Visual Filling of time zero histo",10,10,370,370);
192 c1->SetFillColor(35);
193 c1->GetFrame()->SetFillColor(21);
194 c1->GetFrame()->SetBorderSize(6);
195 c1->GetFrame()->SetBorderMode(-1);
197 //c2 = new TCanvas("c2","Dynamic Visual Filling of chisquare histo",550,10,500,500);
198 c2 = new TCanvas("c2","Dynamic Visual Filling of chisquare histo",380,10,370,370);
199 c2->SetFillColor(35);
200 c2->GetFrame()->SetFillColor(21);
201 c2->GetFrame()->SetBorderSize(6);
202 c2->GetFrame()->SetBorderMode(-1);
204 //c3 = new TCanvas("c3","Dynamic Visual Filling of confidence level histo",280,550,500,500);
205 c3 = new TCanvas("c3","Dynamic Visual Filling of confidence level histo",760,10,370,370);
206 c3->SetFillColor(35);
207 c3->GetFrame()->SetFillColor(21);
208 c3->GetFrame()->SetBorderSize(6);
209 c3->GetFrame()->SetBorderMode(-1);
212 if(strstr(option,"tim") || strstr(option,"all"))
213 gBenchmark->Start("TOFT0");
215 TH1F *htzerobest= new TH1F("htzerobest","T0 for best assignment",200,-1.,1.);
216 TH1F* hchibest = new TH1F("hchibest","ChiSquare Min Distribution",80,0.,40.);
217 TH1F* hchibestconflevel = new TH1F("hchibestconflevel","ChiSquare Min Confidence Level",10,0.,1.);
219 // setting histo colors
220 if(strstr(option,"visual")){
221 htzerobest->SetFillColor(48);
222 hchibest->SetFillColor(50);
223 hchibestconflevel->SetFillColor(52);
226 Int_t assparticle[10]={3,3,3,3,3,3,3,3,3,3};
227 Int_t truparticle[10]={3,3,3,3,3,3,3,3,3,3};
229 Float_t timeofflight[10]={0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};
230 Float_t momentum[10]={0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};
231 Float_t timezero[10];
232 Float_t weightedtimezero[10];
233 Float_t beta[10]={0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};
234 Float_t sqMomError[10]={0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};
235 Float_t sqTrackError[10]={0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};
236 Float_t massarray[3]={0.13957,0.493677,0.9382723};
237 Float_t dummychisquare=0.;
238 Float_t chisquare=999.;
239 Float_t tracktoflen[10]={0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};
241 AliTOF *detTOF = (AliTOF *) gAlice->GetDetector ("TOF");
244 Error("AliTOFT0","TOF not found");
248 if(strstr(option,"all")){
249 cout << "Selecting primary tracks with momentum between " << fLowerMomBound << " GeV/c and " << fUpperMomBound << " GeV/c" << endl;
250 cout << "Memorandum: 0 means PION | 1 means KAON | 2 means PROTON" << endl;
253 if (fNevents == 0) fNevents = (Int_t) gAlice->TreeE()->GetEntries();
255 for (Int_t ievent = 0; ievent < fNevents; ievent++) {
256 gAlice->GetEvent(ievent);
257 TTree *hitTree = detTOF->TreeH ();
262 TClonesArray* tofHits = detTOF->Hits();
267 hitTree->SetBranchStatus("*",0); // switch off all branches
268 hitTree->SetBranchStatus("TOF*",1); // switch on only TOF
270 // Start loop on primary tracks in the hits containers
272 Int_t ntracks = static_cast<Int_t>(hitTree->GetEntries());
273 for (Int_t track = 0; track < ntracks; track++)
275 if(nset>=5) break; // check on the number of set analyzed
278 hitTree->GetEvent(track);
279 particle = gAlice->GetMCApp()->Particle(track);
280 Int_t nhits = tofHits->GetEntriesFast();
282 for (Int_t hit = 0; hit < nhits; hit++)
284 tofHit = (AliTOFhitT0 *) tofHits->UncheckedAt(hit);
285 ipart = tofHit->GetTrack();
286 // check to discard the case when the same particle is selected more than one
289 if (ipart != ipartold){
291 particle = (TParticle*)gAlice->GetMCApp()->Particle(ipart);
293 Float_t idealtime=tofHit->GetTof();
294 // Float_t time=idealtime;
295 Float_t time = gRandom->Gaus(idealtime, fTimeResolution);
296 Float_t toflen=tofHit->GetLen();
297 toflen=toflen/100.; // toflen given in m
298 Int_t pdg = particle->GetPdgCode();
299 Int_t abspdg =TMath::Abs(pdg);
300 Float_t idealmom = particle->P();
301 Float_t momres=idealmom*0.025; // 2.5% res token into account for all momenta
302 Float_t mom =gRandom->Gaus(idealmom,momres);
304 Bool_t isgoodpart=(abspdg==211 || abspdg==2212 || abspdg==321);
306 time*=1.E+9; // tof given in nanoseconds
307 if (particle->GetFirstMother() < 0 && isgoodpart && mom<=fUpperMomBound && mom>=fLowerMomBound){
311 Int_t index=selected-1;
312 timeofflight[index]=time;
313 tracktoflen[index]=toflen;
315 // cout << timeofflight[index] << " " << tracktoflen[index] << " " << momentum[index] << endl;
318 truparticle[index]=0;
321 truparticle[index]=1;
324 truparticle[index]=2;
331 if(istop==10){ // start analysis on current set
336 //cout << "starting t0 calculation for current set" << endl;
337 for (Int_t i1=0; i1<3;i1++) {
338 beta[0]=momentum[0]/sqrt(massarray[i1]*massarray[i1]+momentum[0]*momentum[0]);
339 for (Int_t i2=0; i2<3;i2++) {
340 beta[1]=momentum[1]/sqrt(massarray[i2]*massarray[i2]+momentum[1]*momentum[1]);
341 for (Int_t i3=0; i3<3;i3++) {
342 beta[2]=momentum[2]/sqrt(massarray[i3]*massarray[i3]+momentum[2]*momentum[2]);
343 for (Int_t i4=0; i4<3;i4++) {
344 beta[3]=momentum[3]/sqrt(massarray[i4]*massarray[i4]+momentum[3]*momentum[3]);
345 for (Int_t i5=0; i5<3;i5++) {
346 beta[4]=momentum[4]/sqrt(massarray[i5]*massarray[i5]+momentum[4]*momentum[4]);
347 for (Int_t i6=0; i6<3;i6++) {
348 beta[5]=momentum[5]/sqrt(massarray[i6]*massarray[i6]+momentum[5]*momentum[5]);
349 for (Int_t i7=0; i7<3;i7++) {
350 beta[6]=momentum[6]/sqrt(massarray[i7]*massarray[i7]+momentum[6]*momentum[6]);
351 for (Int_t i8=0; i8<3;i8++) {
352 beta[7]=momentum[7]/sqrt(massarray[i8]*massarray[i8]+momentum[7]*momentum[7]);
353 for (Int_t i9=0; i9<3;i9++) {
354 beta[8]=momentum[8]/sqrt(massarray[i9]*massarray[i9]+momentum[8]*momentum[8]);
355 for (Int_t i10=0; i10<3;i10++) {
356 beta[9]=momentum[9]/sqrt(massarray[i10]*massarray[i10]+momentum[9]*momentum[9]);
358 Float_t meantzero=0.;
359 Float_t sumAllweights=0.;
360 for (Int_t itz=0; itz<10;itz++) {
361 sqMomError[itz]=((1.-beta[itz]*beta[itz])*0.025)*((1.-beta[itz]*beta[itz])*0.025)*(tracktoflen[itz]/(0.299792*beta[itz]))*(tracktoflen[itz]/(0.299792*beta[itz])); // this gives the square of the momentum error in nanoseconds
362 sqTrackError[itz]=(timeresolutioninns*timeresolutioninns+sqMomError[itz]); // total error for the current track
363 sumAllweights+=1./sqTrackError[itz];
365 timezero[itz]=(tracktoflen[itz]/(beta[itz]*0.299792))-timeofflight[itz];
366 weightedtimezero[itz]=((tracktoflen[itz]/(beta[itz]*0.299792))-timeofflight[itz])/sqTrackError[itz];// weighted time zero for current track
367 meantzero+=weightedtimezero[itz];
368 } // end loop for (Int_t itz=0; itz<10;itz++)
369 meantzero=meantzero/sumAllweights; // it is given in [ns]
372 // calculate the chisquare for the current assignment
373 for (Int_t icsq=0; icsq<10;icsq++) {
374 dummychisquare+=(timezero[icsq]-meantzero)*(timezero[icsq]-meantzero)/sqTrackError[icsq];
375 } // end loop for (Int_t icsq=0; icsq<10;icsq++)
377 if(dummychisquare<=chisquare){
388 chisquare=dummychisquare;
390 } // close if(dummychisquare<=chisquare)
403 if(truparticle[0]==assparticle[0] && truparticle[1]==assparticle[1] && truparticle[2]==assparticle[2] && truparticle[3]==assparticle[3] && truparticle[4]==assparticle[4]&& truparticle[5]==assparticle[5] && truparticle[6]==assparticle[6] && truparticle[7]==assparticle[7] && truparticle[8]==assparticle[8] && truparticle[9]==assparticle[9]) ngood+=1;
404 if(truparticle[0]!=assparticle[0]) nmisidentified0+=1;
405 if(truparticle[1]!=assparticle[1]) nmisidentified1+=1;
406 if(truparticle[2]!=assparticle[2]) nmisidentified2+=1;
407 if(truparticle[3]!=assparticle[3]) nmisidentified3+=1;
408 if(truparticle[4]!=assparticle[4]) nmisidentified4+=1;
409 if(truparticle[5]!=assparticle[5]) nmisidentified5+=1;
410 if(truparticle[6]!=assparticle[6]) nmisidentified6+=1;
411 if(truparticle[7]!=assparticle[7]) nmisidentified7+=1;
412 if(truparticle[8]!=assparticle[8]) nmisidentified8+=1;
413 if(truparticle[9]!=assparticle[9]) nmisidentified9+=1;
415 htzerobest->Fill(t0best);
416 hchibest->Fill(chisquare);
417 Double_t dblechisquare=(Double_t)chisquare;
418 Float_t confLevel=(Float_t)TMath::Prob(dblechisquare,9); // ndf 10-1=9
419 hchibestconflevel->Fill(confLevel);
421 if(strstr(option,"all")){
422 cout << "True Assignment " << truparticle[0] << truparticle[1] << truparticle[2] << truparticle[3] << truparticle[4] << truparticle[5] << truparticle[6] << truparticle[7] << truparticle[8] << truparticle[9] <<endl;
423 cout << "Best Assignment " << assparticle[0] << assparticle[1] << assparticle[2] << assparticle[3] << assparticle[4] << assparticle[5] << assparticle[6] << assparticle[7] << assparticle[8] << assparticle[9] << endl;
424 cout << "Minimum ChiSquare for current set " << chisquare << endl;
425 cout << "Confidence Level (Minimum ChiSquare) " << confLevel << endl;
427 if (strstr(option,"visual") && itimes && (itimes%kUPDATE) == 0) {
428 if (itimes == kUPDATE){
434 hchibestconflevel->Draw();
442 if (gSystem->ProcessEvents())
448 } // end for the current set. close if(istop==5)
449 } // end condition on ipartold
450 } // end loop on hits for the current track
452 } // end loop on ntracks
455 if(strstr(option,"all")){
456 nmisidentified=(nmisidentified0+nmisidentified1+nmisidentified2+nmisidentified3+nmisidentified4+nmisidentified5+nmisidentified6+nmisidentified7+nmisidentified8+nmisidentified9);
457 cout << "total number of tracks token into account " << 10*5*fNevents << endl;
458 Float_t badPercentage=100.*(Float_t)nmisidentified/(10*5*fNevents);
459 cout << "total misidentified " << nmisidentified << "("<< badPercentage << "%)" <<endl;
460 cout << "Total Number of set token into account " << 5*fNevents << endl;
461 Float_t goodSetPercentage=100.*(Float_t)ngood/(5*fNevents);
462 cout << "Number of set with no misidentified tracks " << ngood << "("<< goodSetPercentage << "%)" <<endl;
465 // free used memory for canvas
470 // generating output filename only if not previously specified using SetTZeroFile
471 char outFileName[70];
472 strcpy(outFileName,"ht010tr120ps"); // global time resolution has to be converted from Int_t to char
473 // in order to have in the output filename this parameter
474 strcat(outFileName,fHeadersFile);
476 if(fT0File.IsNull()) fT0File=outFileName;
478 TFile* houtfile = new TFile(fT0File,"recreate");
480 htzerobest->Write(0,TObject::kOverwrite);
481 hchibest->Write(0,TObject::kOverwrite);
482 hchibestconflevel->Write(0,TObject::kOverwrite);
486 if(strstr(option,"tim") || strstr(option,"all")){
487 gBenchmark->Stop("TOFT0");
488 cout << "AliTOFT0:" << endl ;
489 cout << " took " << gBenchmark->GetCpuTime("TOFT0") << " seconds in order to calculate T0 "
490 << gBenchmark->GetCpuTime("TOFT0")/fNevents << " seconds per event " << endl ;
495 //__________________________________________________________________
496 void AliTOFT0::SetTZeroFile(char * file )
501 printf("Destination file : %s \n", file) ;
506 //__________________________________________________________________
507 void AliTOFT0::Print(Option_t* /*option*/)const
512 printf("------------------- %s -------------\n", GetName()) ;
513 if(!fT0File.IsNull())
514 printf(" Writing T0 Distribution to file %s \n",(char*) fT0File.Data());
518 //__________________________________________________________________
519 Bool_t AliTOFT0::operator==( AliTOFT0 const &tzero )const
525 if( (fTimeResolution==tzero.fTimeResolution)&&(fLowerMomBound==tzero.fLowerMomBound)&&(fUpperMomBound==tzero.fUpperMomBound))