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 **************************************************************************/
16 //_________________________________________________________________________
17 // TTask class for TOF Probabilities (ALICE-Offline week June 2002).
18 // Use case: start root and execute the following macro
19 // ntuple.root is assumed to be the ntuple with the reconstructed
23 // Dynamically link some shared libs
24 if (gClassTable->GetID("AliRun") < 0) {
25 gROOT->LoadMacro("loadlibs.C");
28 // create an instance of the AliTOFProb class
29 // You have to pass the ntuple
30 AliTOFProb* tofprob=new AliTOFProb("ntuple.root");
35 //-- Author: F. Pierella | pierella@bo.infn.it
36 //////////////////////////////////////////////////////////////////////////////
55 #include "AliTOFProb.h"
58 #include <Riostream.h>
59 #include <Riostream.h>
63 //____________________________________________________________________________
64 AliTOFProb::AliTOFProb():TTask("AliTOFProb","")
66 // default ctor - set the pointer member vars to zero
74 //____________________________________________________________________________
75 AliTOFProb::AliTOFProb(char* headerFile):TTask("AliTOFProb","")
78 fhfile = TFile::Open(headerFile); // connect file with ntuple
79 foutfileName=headerFile;
84 // add Task to //root/Tasks folder
85 TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ;
86 roottasks->Add(this) ;
88 //____________________________________________________________________________
89 void AliTOFProb::Init()
92 fNtuple= (TNtuple*)fhfile->Get("Ntuple"); // get ntuple from file
93 Int_t nvar = fNtuple->GetNvar(); cout <<"N of var.="<< nvar << endl;
98 //____________________________________________________________________________
99 AliTOFProb::~AliTOFProb()
102 // dtor (free used memory)
132 //____________________________________________________________________________
133 void AliTOFProb::Exec(const Option_t * /*dummyOpt*/)
136 // Performs Prob for TOF detector
141 TH3* h3D=new TH3F("h3D", "", 120, -0.1, 1.1, 120, -0.1, 1.1, 120, -0.1, 1.1);
142 TH3* h3Ddt=new TH3F("h3Ddt", "", 360, -120., 120., 360, -120., 120., 360, -120., 120.);
144 TH1* hpion =new TH1F("hpion", "", 120, -0.1, 1.1);
145 TH1* hkaon =new TH1F("hkaon", "", 120, -0.1, 1.1);
146 TH1* hproton=new TH1F("hproton", "", 120, -0.1, 1.1);
148 TH1* hdtpion =new TH1F("hdtpion", "", 600, -120., 120.);
149 TH1* hdtkaon =new TH1F("hdtkaon", "", 600, -120., 120.);
150 TH1* hdtproton=new TH1F("hdtproton", "", 600, -120., 120.);
152 const Float_t timeResolution=0.12;
154 Int_t nparticles = (Int_t)fNtuple->GetEntries();
155 cout << " Number of nparticles =" << nparticles << endl;
156 if (nparticles <= 0) return;
158 for (Int_t i=0; i < nparticles; i++) {
159 fNtuple->GetEvent(i);
160 //Int_t event=(Int_t)(fNtuple->GetLeaf("event")->GetValue());
161 Int_t pdgcode=(Int_t)(fNtuple->GetLeaf("ipart")->GetValue());
162 // unused for the time being in the context of probabilities
163 // probably it will be included in AliESD approach
164 //Float_t mass=fNtuple->GetLeaf("mext")->GetValue(0);
165 Int_t matc=(Int_t)(fNtuple->GetLeaf("matc")->GetValue(0));
166 Int_t imam=(Int_t)(fNtuple->GetLeaf("imam")->GetValue(0));
167 Float_t time=fNtuple->GetLeaf("text")->GetValue(0); // [ns]
168 // generate a gaussian around time with sigma timeresolution [ns]
169 Float_t lowTime=time-1000.*timeResolution;
170 Float_t upTime =time+1000.*timeResolution;
171 TF1* gauss=new TF1("tg","gaus",lowTime,upTime);
174 Double_t p2=timeResolution;
175 //Float_t twopi=2*TMath::Pi();
176 //Float_t sqrtwopi=TMath::Sqrt(twopi);
177 //Double_t p0=1./(p2*sqrtwopi);
178 // choising a different normalization
180 gauss->SetParameters(p0,p1,p2);
182 Float_t px=fNtuple->GetLeaf("pxvtx")->GetValue(0);
183 Float_t py=fNtuple->GetLeaf("pyvtx")->GetValue(0);
184 Float_t pz=fNtuple->GetLeaf("pzvtx")->GetValue(0);
186 Float_t toflen=fNtuple->GetLeaf("leng")->GetValue(0);
188 Float_t pvtx=TMath::Sqrt(px*px+py*py+pz*pz);
189 Float_t ptvtx=TMath::Sqrt(px*px+py*py);
191 Bool_t isSelected=(imam == 0 && pz !=0 && TMath::ATan(TMath::Abs(ptvtx/pz))>TMath::Pi()*45./180.);
192 Bool_t trackWithTime=(matc==2 || matc==3 || matc==4);
194 Int_t abspdgcode=TMath::Abs(pdgcode);
206 if (isSelected && trackWithTime && toflen>370.)
207 {//only primary +/-45
208 Float_t beta[3]={0.,0.,0.};
209 Float_t timeth[3]={0.,0.,0.};
210 Float_t massarray[3]={0.13957,0.493677,0.9382723};
211 for (Int_t j=0; j < 3; j++) {
212 Float_t dummy=(pvtx/massarray[j])*(pvtx/massarray[j]);
213 beta[j]=TMath::Sqrt(dummy/(1.+dummy));
214 timeth[j]=toflen/(29.9792458*beta[j]);
216 Float_t pPion =(Float_t)gauss->Eval(timeth[0],0.,0.);
217 Float_t pKaon =(Float_t)gauss->Eval(timeth[1],0.,0.);
218 Float_t pProton =(Float_t)gauss->Eval(timeth[2],0.,0.);
220 Float_t dtpion=time-timeth[0];
221 Float_t dtkaon=time-timeth[1];
222 Float_t dtproton=time-timeth[2];
223 hdtpion->Fill(dtpion);
224 hdtkaon->Fill(dtkaon);
225 hdtproton->Fill(dtproton);
227 // normalizing probabilities
228 Float_t sumProb=pPion+pKaon+pProton;
231 pProton=pProton/sumProb;
234 hproton->Fill(pProton);
235 //cout << "pPion " << pPion << " pKaon " << pKaon << " pProton " << pProton << endl;
236 h3D->Fill(pPion,pKaon,pProton);
237 h3Ddt->Fill(dtpion,dtkaon,dtproton);
238 //cout << "pion " << timeth[0] << " kaon " << timeth[1] << " proton " << timeth[2] << endl;
239 }// End of cuts appling
243 }// End of loop over particles
245 char outFileName[100];
246 strcpy(outFileName,"3DProbHisto.root");
247 TFile *houtfile = new TFile(outFileName,"recreate");
249 h3D->Write(0,TObject::kOverwrite);
250 h3Ddt->Write(0,TObject::kOverwrite);
251 hpion->Write(0,TObject::kOverwrite);
252 hkaon->Write(0,TObject::kOverwrite);
253 hproton->Write(0,TObject::kOverwrite);
254 hdtpion->Write(0,TObject::kOverwrite);
255 hdtkaon->Write(0,TObject::kOverwrite);
256 hdtproton->Write(0,TObject::kOverwrite);
259 houtfile->Write(0,TObject::kOverwrite);
263 //__________________________________________________________________
264 Bool_t AliTOFProb::operator==( AliTOFProb const & /*tofrec*/)const
266 // dummy version of Equal operator.
267 // requested by coding conventions