TOF/AliTOFPID.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   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  **************************************************************************/
  15
  16 //_________________________________________________________________________
  17 // TTask class for TOF PID.
  18 // Use case: start root and execute the following macro
  19 /*
  20 {
  21   // Dynamically link some shared libs
  22   if (gClassTable->GetID("AliRun") < 0) {
  23     gROOT->LoadMacro("loadlibs.C");
  24     loadlibs();
  25   }
  26   // create an instance of the AliTOFPID class
  27   // You have to pass the ntuple and cuts filenames
  28  AliTOFPID* tofpid=new AliTOFPID("ntuple.root","cuts.root");
  29
  30  // option "pp" for pp events (it includes also electron in the analysis)
  31  // option "visual" to shows interactively histos
  32  // option "asC" or "asEPS" to save canvas in the current dir in .C or .eps format
  33
  34  // make a choice: uncomment one of these lines
  35  // tofpid->Exec("pp","visual","asC");
  36  // tofpid->Exec("pp","novisual","asC");
  37  // tofpid->Exec("Pb-Pb","visual","asC");
  38  // tofpid->Exec("pp","visual","asC");
  39  // tofpid->Exec("pp","novisual","asEPS");
  40 }
  41 */
  42 //
  43 //
  44 //
  45 //
  46 //
  47 //
  48 //
  49 //
  50 //-- Authors: B. Zagreev , F. Pierella
  51 //////////////////////////////////////////////////////////////////////////////
  52
  53
  54 #include "TROOT.h"
  55 #include "TStyle.h"
  56 #include "TTask.h"
  57 #include "TTree.h"
  58 #include "TSystem.h"
  59 #include "TFile.h"
  60 #include "TCanvas.h"
  61 #include "TPad.h"
  62 #include "TText.h"
  63 #include "TLine.h"
  64 #include "TPaveLabel.h"
  65 #include "TPaveText.h"
  66 #include "AliConst.h"
  67 #include "AliTOFConstants.h"
  68 #include "AliTOFPID.h"
  69 #include <TClonesArray.h>
  70 #include "TFile.h"
  71 #include <TF1.h>
  72 #include <TF2.h>
  73 #include "TTask.h"
  74 #include "TTree.h"
  75 #include "TSystem.h"
  76 #include "TROOT.h"
  77 #include "TFolder.h"
  78 #include "TNtuple.h"
  79 #include "TLeaf.h"
  80 #include <stdlib.h>
  81 #include <iostream.h>
  82 #include <fstream.h>
  83
  84 ClassImp(AliTOFPID)
  85
  86 //____________________________________________________________________________
  87   AliTOFPID::AliTOFPID():TTask("AliTOFPID","")
  88 {
  89   // default ctor - set the pointer member vars to zero
  90   felectron = 0;
  91   fpion     = 0;
  92   fkaon     = 0;
  93   fproton   = 0;
  94   fcut      = 0;
  95   fhfile    = 0;
  96   fNtuple   = 0;
  97   fgen      = 0;
  98   foutfileName  = 0;
  99 }
 100
 101 //____________________________________________________________________________
 102   AliTOFPID::AliTOFPID(char* headerFile, char *cutsFile, const Option_t* opt):TTask("AliTOFPID","")
 103 {
 104   fhfile = TFile::Open(headerFile); // connect file with ntuple
 105   fcut = TFile::Open(cutsFile); // connect file for cuts
 106   foutfileName=headerFile;
 107
 108   Init(opt);
 109   // add Task to //root/Tasks folder
 110   TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ;
 111   roottasks->Add(this) ;
 112 }
 113 //____________________________________________________________________________
 114 void AliTOFPID::Init(const Option_t* opt)
 115 {
 116   if(strstr(opt,"pp")){
 117     if(fcut->GetKey("electron")) felectron = (TCutG*)fcut->Get("electron");
 118     fcut->Print();
 119     if(fcut->GetKey("pion")) fpion = (TCutG*)fcut->Get("pion");
 120     fcut->Print();
 121   }
 122   if(fcut->GetKey("kaon")) fkaon = (TCutG*)fcut->Get("kaon");
 123   fcut->Print();
 124   if(fcut->GetKey("proton")) fproton = (TCutG*)fcut->Get("proton");
 125
 126   gFile->ls();
 127   fNtuple= (TNtuple*)fhfile->Get("Ntuple"); // get ntuple from file
 128   Int_t nvar = fNtuple->GetNvar(); cout <<"N of var.="<< nvar << endl;
 129   fNtuple->GetEvent(0);
 130
 131 }
 132
 133 //____________________________________________________________________________
 134   AliTOFPID::~AliTOFPID()
 135 {
 136   //
 137   // dtor (free used memory)
 138   //
 139
 140   if (felectron)
 141     {
 142       delete felectron;
 143       felectron = 0;
 144     }
 145
 146   if (fpion)
 147     {
 148       delete fpion;
 149       fpion = 0;
 150     }
 151
 152   if (fkaon)
 153     {
 154       delete fkaon;
 155       fkaon = 0;
 156     }
 157
 158   if (fproton)
 159     {
 160       delete fproton;
 161       fproton = 0;
 162     }
 163
 164   if (fcut)
 165     {
 166       delete fcut;
 167       fcut = 0;
 168     }
 169
 170
 171   if (fhfile)
 172     {
 173       delete fhfile;
 174       fhfile = 0;
 175     }
 176
 177
 178   if (fNtuple)
 179     {
 180       delete fNtuple;
 181       fNtuple = 0;
 182     }
 183
 184   if (fgen)
 185     {
 186       delete fgen;
 187       fgen = 0;
 188     }
 189
 190   if (foutfileName)
 191     {
 192       delete foutfileName;
 193       foutfileName = 0;
 194     }
 195 }
 196
 197
 198 //____________________________________________________________________________
 199 void AliTOFPID::Exec(const Option_t *eventType, const Option_t *outputmode, const Option_t *outputsavemode)
 200 {
 201   //
 202   // Performs PID for TOF detector
 203   //
 204
 205   fTask=1;
 206   TAxis *xaxis;
 207   ////////// Create histograms /////////////////
 208   // for electron only in pp case
 209   TH1F* eleff=0;
 210   TH1F *elcon=0;
 211   TH1F *elid=0;
 212   TH1F *elall=0;
 213
 214   if(strstr(eventType,"pp")){
 215     eleff = new TH1F("eleff","",10,0,0.6);
 216     xaxis=eleff->GetYaxis();
 217     xaxis->SetLabelSize(.08);
 218     elcon = new TH1F("elcon","",10,0,0.6);
 219     xaxis=elcon->GetXaxis();
 220     xaxis->SetLabelSize(.09);
 221     xaxis=elcon->GetYaxis();
 222     xaxis->SetLabelSize(.08);
 223     elid = new TH1F("elid","Identified electrons",10,0,0.6);
 224     elall = new TH1F("elall","Electrons",10,0,0.6);
 225   }
 226
 227   // pions
 228   TH1F *pit  = new TH1F("pit","",15,0,2.5); //part. with tracks
 229   TH1F *pig  = new TH1F("pig","",15,0,2.5); //part. in geometry acceptance
 230   TH1F *pieff = new TH1F("pieff","",15,0,2.5); //efficiency
 231   xaxis=pieff->GetYaxis();
 232   xaxis->SetLabelSize(.08);
 233   TH1F *picon = new TH1F("picon","",15,0,2.5); //contamination
 234   xaxis=picon->GetXaxis();
 235   xaxis->SetLabelSize(.09);
 236   xaxis=picon->GetYaxis();
 237   xaxis->SetLabelSize(.08);
 238   TH1F *piid  = new TH1F("piid","Identified pions",15,0,2.5);
 239   TH1F *piall = new TH1F("piall","Pions",15,0,2.5);
 240   TH1F *pigen = new TH1F("pigen","Pions",15,0,2.5);
 241   xaxis=pigen->GetXaxis();
 242   xaxis->SetLabelSize(.09);
 243   pigen->SetXTitle("P?t! (GeV/c)");
 244   xaxis->SetTitleSize(.09);
 245   xaxis=pigen->GetYaxis();
 246   xaxis->SetLabelSize(.08);
 247   //pigen->SetYTitle("1/P?t!dN/dP?t! (GeV/c)^-2!");
 248   xaxis->SetTitleSize(.09);
 249
 250   // kaons
 251   TH1F *kat  = new TH1F("kat","",15,0,2.5);
 252   TH1F *kag  = new TH1F("kag","",15,0,2.5);
 253   TH1F *kaeff = new TH1F("kaeff","",15,0,2.5);
 254   xaxis=kaeff->GetYaxis();
 255   xaxis->SetLabelSize(.08);
 256   TH1F *kacon = new TH1F("kacon","",15,0,2.5);
 257   xaxis=kacon->GetXaxis();
 258   xaxis->SetLabelSize(.09);
 259   xaxis=kacon->GetYaxis();
 260   xaxis->SetLabelSize(.08);
 261   TH1F *kaid  = new TH1F("kaid","Identified kaons",15,0,2.5);
 262   TH1F *kaall = new TH1F("kaall","Kaons",15,0,2.5);
 263   TH1F *kagen = new TH1F("kagen","Kaons",15,0,2.5);
 264   xaxis=kagen->GetXaxis();
 265   xaxis->SetLabelSize(.09);
 266   kagen->SetXTitle("P?t! (GeV/c)");
 267   xaxis->SetTitleSize(.09);
 268   xaxis=kagen->GetYaxis();
 269   xaxis->SetLabelSize(.08);
 270   //kagen->SetYTitle("1/P?t!dN/dP?t! (GeV/c)^-2!");
 271   xaxis->SetTitleSize(.09);
 272
 273   // protons
 274   TH1F *prt  = new TH1F("prt","",15,0,4.4);
 275   TH1F *prg  = new TH1F("prg","",15,0,4.4);
 276   TH1F *preff = new TH1F("preff","",15,0,4.4);
 277   xaxis=preff->GetYaxis();
 278   xaxis->SetLabelSize(.08);
 279   TH1F *prcon = new TH1F("prcon","",15,0,4.4);
 280   xaxis=prcon->GetXaxis();
 281   xaxis->SetLabelSize(.09);
 282   xaxis=prcon->GetYaxis();
 283   xaxis->SetLabelSize(.08);
 284   TH1F *prid  = new TH1F("prid","Identified protons",15,0,4.4);
 285   TH1F *prall = new TH1F("prall","Protons",15,0,4.4);
 286   TH1F *prgen = new TH1F("prgen","Protons",15,0,4.4);
 287   xaxis=prgen->GetXaxis();
 288   xaxis->SetLabelSize(.09);
 289   prgen->SetXTitle("P?t! (GeV/c)");
 290   xaxis->SetTitleSize(.09);
 291   xaxis=prgen->GetYaxis();
 292   xaxis->SetLabelSize(.08);
 293   //prgen->SetYTitle("1/P?t!dN/dP?t! (GeV/c)^-2!");
 294   xaxis->SetTitleSize(.09);
 295
 296   // 2-D histos (extrapolated mass vs momentum)
 297   TH2F* hel=0;
 298   if(strstr(eventType,"pp")){
 299     hel = new TH2F("hel","",1000,-.2,1.2,1000,-4.2,0.);
 300     hel->SetXTitle("Mass (GeV/c^{2})");
 301     hel->SetYTitle("Momentum (GeV/c)");
 302   }
 303   TH2F *hpi = new TH2F("hpi","",1000,-.2,1.2,1000,-4.2,0.);
 304   hpi->SetXTitle("Mass (GeV/c^{2})");
 305   hpi->SetYTitle("Momentum (GeV/c)");
 306   TH2F *hka = new TH2F("hka","",1000,-.2,1.2,1000,-4.2,0.);
 307   hka->SetXTitle("Mass (GeV/c^{2})");
 308   hka->SetYTitle("Momentum (GeV/c)");
 309   TH2F *hpr = new TH2F("hpr","",1000,-.2,1.2,1000,-4.2,0.);
 310   hpr->SetXTitle("Mass (GeV/c^{2})");
 311   hpr->SetYTitle("Momentum (GeV/c)");
 312
 313
 314   fhfile->cd();
 315   Int_t nparticles = (Int_t)fNtuple->GetEntries();
 316   cout << " Number of nparticles =" << nparticles << endl;
 317   if (nparticles <= 0) return;
 318
 319   Float_t ka=0, pi=0, pr=0, kaal=0, pial=0, pral=0;
 320   Float_t pitrack=0, pimag=0, pigeom=0;
 321   Float_t katrack=0, kamag=0, kageom=0;
 322   Float_t prtrack=0, prmag=0, prgeom=0;
 323   Float_t pif=0, kaf=0, prf=0, pin=0, kan=0, prn=0;
 324   Float_t px, py, pz, x, y, z, pdgcode, mass;
 325   Int_t event, matc, imam;
 326   Int_t indexOfFile=0, numfile=0;
 327   //////// Loop over tracks (particles)///////////////////////
 328
 329   for (Int_t i=0; i < nparticles; i++) {
 330     fNtuple->GetEvent(i);
 331     event=fNtuple->GetLeaf("event")->GetValue();
 332     pdgcode=fNtuple->GetLeaf("ipart")->GetValue();
 333     mass=fNtuple->GetLeaf("mext")->GetValue(0);
 334     matc=fNtuple->GetLeaf("matc")->GetValue(0);
 335     imam=fNtuple->GetLeaf("imam")->GetValue(0);
 336     px=fNtuple->GetLeaf("pxvtx")->GetValue(0);
 337     py=fNtuple->GetLeaf("pyvtx")->GetValue(0);
 338     pz=fNtuple->GetLeaf("pzvtx")->GetValue(0);
 339     x=fNtuple->GetLeaf("xvtx")->GetValue(0);
 340     y=fNtuple->GetLeaf("yvtx")->GetValue(0);
 341     z=fNtuple->GetLeaf("zvtx")->GetValue(0);
 342     Float_t pvtx=TMath::Sqrt(px*px+py*py+pz*pz);
 343     Float_t ptvtx=TMath::Sqrt(px*px+py*py);
 344     Float_t mt=0.;
 345     Int_t abspdgcode=TMath::Abs(pdgcode);
 346     switch(abspdgcode){
 347     case 321:
 348       mt=TMath::Sqrt(AliTOFConstants::fgkKaonMass*AliTOFConstants::fgkKaonMass+px*px+py*py);
 349       break;
 350     case 2212:
 351       mt=TMath::Sqrt(AliTOFConstants::fgkProtonMass*AliTOFConstants::fgkProtonMass+px*px+py*py);
 352       break;
 353     case 11:
 354       mt=TMath::Sqrt(AliTOFConstants::fgkElectronMass*AliTOFConstants::fgkElectronMass+px*px+py*py);
 355       break;
 356     default:
 357       mt=TMath::Sqrt(AliTOFConstants::fgkPionMass*AliTOFConstants::fgkPionMass+px*px+py*py);
 358       break;
 359     }
 360
 361     if (imam == 0 && pz !=0 && TMath::ATan(TMath::Abs(ptvtx/pz))>TMath::Pi()*45./180.)
 362       {//only primary +/-45
 363         if (fkaon->IsInside(mass,-pvtx) && matc>2) {
 364           ka++;
 365           if (fTask!=2) kaid->Fill(pvtx); else {kaid->Fill(ptvtx);}
 366           if (TMath::Abs(pdgcode)==321) {kaf++; kaeff->Fill(pvtx);} else {kan++; kacon->Fill(pvtx);}
 367         } else if (fproton->IsInside(mass,-pvtx) && matc>1) {
 368           pr++;
 369           if (fTask!=2) prid->Fill(pvtx); else
 370             {prid->Fill(ptvtx);}
 371           if (TMath::Abs(pdgcode)==2212) {prf++; preff->Fill(pvtx);} else {prn++; prcon->Fill(pvtx);}
 372         } else  if (strstr(eventType,"pp") && felectron->IsInside(mass,-pvtx) && matc>2) {elid->Fill(pvtx);
 373         if (strstr(eventType,"pp") && TMath::Abs(pdgcode)==11) eleff->Fill(pvtx); else elcon->Fill(pvtx);
 374         } else if (matc>0) {
 375           //||matc==-4&&fpion->IsInside(mass,-pvtx)
 376           pi++;
 377           if (fTask!=2) piid->Fill(pvtx); else {piid->Fill(ptvtx);}
 378           if (TMath::Abs(pdgcode)==211) {pif++; pieff->Fill(pvtx);} else {pin++; picon->Fill(pvtx);}
 379         }
 380
 381         //////////////// Normalization histograms ////////////////////
 382         if (strstr(eventType,"pp") && TMath::Abs(pdgcode)==11) {
 383           if (fTask!=2) elall->Fill(pvtx); else elall->Fill(ptvtx);
 384           if (fTask==1) hel->Fill(mass,-pvtx);
 385
 386         } else if (TMath::Abs(pdgcode)==211) {
 387           pial++;
 388           if (matc!=0) {
 389             pitrack++;
 390             pit->Fill(pvtx);
 391             if (matc!=-1) {
 392               pimag++;
 393               if (matc>-2 || matc==-4) {
 394                 pigeom++;
 395                 pig->Fill(pvtx);
 396               }
 397             }
 398           }
 399           if (fTask!=2) piall->Fill(pvtx);
 400           else {
 401             piall->Fill(ptvtx,1/ptvtx);
 402           }
 403           if (fTask==1) hpi->Fill(mass,-pvtx);
 404
 405         } else if (TMath::Abs(pdgcode)==321) {
 406           kaal++;
 407           if (matc!=0) {
 408             katrack++;
 409             kat->Fill(pvtx);
 410             if (matc!=-1) {
 411               kamag++;
 412               if (matc>-2 || matc==-4) {
 413                 kageom++;
 414                 kag->Fill(pvtx);
 415               }
 416             }
 417           }
 418           if (fTask!=2) kaall->Fill(pvtx);
 419           else {
 420             kaall->Fill(ptvtx,1/ptvtx);
 421           }
 422           if (fTask==1) hka->Fill(mass,-pvtx);
 423
 424         } else if (TMath::Abs(pdgcode)==2212) {
 425           pral++;
 426           if (matc!=0) {
 427             prtrack++;
 428             prt->Fill(pvtx);
 429             if (matc!=-1) {
 430               prmag++;
 431               if (matc>-2 || matc==-4) {
 432                 prgeom++;
 433                 prg->Fill(pvtx);
 434               }
 435             }
 436           }
 437           if (fTask!=2) prall->Fill(pvtx);
 438           else {
 439             prall->Fill(ptvtx,1/ptvtx);
 440           }
 441           if (fTask==1) hpr->Fill(mass,-pvtx);}
 442
 443       }// End of cuts appling
 444   }// End of loop over particles
 445
 446   // display results
 447   cout<< "Pions in 45-135 deg.     "<< pial <<" (100%)"<< endl;
 448   cout<< "Pions that have track    "<< pitrack/pial*100 <<" %"<<endl;
 449   cout<< "Magnetic field           "<< pimag/pial*100 <<" %"<<endl;
 450   cout<< "Geometry efficiency      "<< pigeom/pial*100 <<" %"<<endl;
 451   cout<< "PID procedure            "<< pif/pial*100 <<" %"<<endl;
 452   cout<< "Contamination            "<< pin/pi*100 <<" %"<<endl;
 453   cout<<endl;
 454   cout<< "Kaons in 45-135 deg.     "<< kaal <<" (100%)"<< endl;
 455   cout<< "Kaons that have track    "<< katrack/kaal*100 <<" %"<<endl;
 456   cout<< "Magnetic field           "<< kamag/kaal*100 <<" %"<<endl;
 457   cout<< "Geometry efficiency      "<< kageom/kaal*100 <<" %"<<endl;
 458   cout<< "PID procedure(+decays)   "<< kaf/kaal*100 <<" %"<<endl;
 459   cout<< "Contamination            "<< kan/ka*100 <<" %"<<endl;
 460   cout<<endl;
 461   cout<< "Protons in 45-135 deg.   "<< pral <<" (100%)"<< endl;
 462   cout<< "Protons that have track  "<< prtrack/pral*100 <<" %"<<endl;
 463   cout<< "Magnetic field           "<< prmag/pral*100 <<" %"<<endl;
 464   cout<< "Geometry efficiency      "<< prgeom/pral*100 <<" %"<<endl;
 465   cout<< "PID procedure            "<< prf/pral*100 <<" %"<<endl;
 466   cout<< "Contamination            "<< prn/pr*100 <<" %"<<endl;
 467   cout<<endl;
 468   cout<< "All part. in 45-135 deg.  "<< pial+kaal+pral <<" (100%)"<< endl;
 469   cout<< "All part. that have track "<< (pitrack+katrack+prtrack)/(pial+kaal+pral)*100 <<" %"<<endl;
 470   cout<< "Magnetic field            "<< (pimag+kamag+prmag)/(pial+kaal+pral)*100 <<" %"<<endl;
 471   cout<< "Geometry efficiency       "<< (pigeom+kageom+prgeom)/(pial+kaal+pral)*100 <<" %"<<endl;
 472   cout<< "PID procedure             "<< (pif+kaf+prf)/(pial+kaal+pral)*100 <<" %"<<endl;
 473   cout<< "Contamination             "<< (pin+kan+prn)/(pi+ka+pr)*100 <<" %"<<endl;
 474   cout<<endl;
 475
 476   TCanvas *pidCanvas=0;
 477   TCanvas *momvsmassCanvas=0;
 478   TPad  *tp=0;
 479   TPad  *pad1=0;
 480   TPad  *pad2=0;
 481   TPad  *pad3=0;
 482   TPad  *pad4=0;
 483   TPad  *pad5=0;
 484   TPad  *pad6=0;
 485   TPad  *pad7=0;
 486   TPad  *pad8=0;
 487
 488   //////////////////////// For fTask 1 ///////////////////////////
 489   if (fTask==1) {
 490     if (strstr(eventType,"pp")){
 491       eleff->Divide(elall);
 492     }
 493     pieff->Divide(piall);
 494     kaeff->Divide(kaall);
 495     preff->Divide(prall);
 496     pit->Divide(piall);
 497     kat->Divide(kaall);
 498     prt->Divide(prall);
 499     pig->Divide(piall);
 500     kag->Divide(kaall);
 501     prg->Divide(prall);
 502     if (strstr(eventType,"pp")){
 503       elcon->Divide(elid);
 504     }
 505     picon->Divide(piid);
 506     kacon->Divide(kaid);
 507     prcon->Divide(prid);
 508     //Create a canvas, set the view range, show histograms
 509     if (indexOfFile==0) {
 510       pidCanvas = new TCanvas("pidCanvas","PID ",10,100,800,500);
 511       pidCanvas->SetBorderMode(0);
 512       pidCanvas->SetBorderSize(0);
 513       pidCanvas->SetFillColor(0);
 514       pidCanvas->SetFillStyle(0);
 515       if (strstr(outputmode,"visual")) pidCanvas->Draw();
 516       Float_t pxs=0.25+0.125; //X size of pad
 517       Float_t pys=0.5+0.055; //y size of pad
 518       tp = new TPad("histo","Histograms",.1,.1,.9,.9);
 519       if (strstr(eventType,"Pb-Pb")){
 520         //pad1 = new TPad("pad1","electron efficiency",0.,.5-.055,0.+pxs,.5-.055+pys-.00001,0,0,0);
 521         pad2 = new TPad("pad2","pion efficiency",0.,0.5-.055,0.+pxs,0.5-.055+pys-.00001,0,0,0);
 522         pad3 = new TPad("pad3","kaon efficiency",0.3,0.5-.055,0.3+pxs,0.5-.055+pys-.00001,0,0,0);
 523         pad4 = new TPad("pad4","proton efficiency",0.6,0.5-.055,0.6+pxs,0.5-.055+pys-.00001,0,0,0);
 524         //pad5 = new TPad("pad5","electron contamination",0.,0.,0.+pxs,0.+pys,0,0,0);
 525         pad6 = new TPad("pad6","pion contamination",0.,0.,0.+pxs,0.+pys,0,0,0);
 526         pad7 = new TPad("pad7","kaon contamination",.3,0.,0.3+pxs,0.+pys,0,0,0);
 527         pad8 = new TPad("pad8","proton contamination",.6,0.,0.6+pxs,0.+pys,0,0,0);
 528       }
 529
 530       if (strstr(eventType,"pp")){
 531         pad1 = new TPad("pad1","electron efficiency",0.,.5-.055,0.25+0.045,1.,0,0,0);
 532         pad2 = new TPad("pad2","pion efficiency",0.25-0.015,0.5-.055,0.5+0.03,1.,0,0,0);
 533         pad3 = new TPad("pad3","kaon efficiency",0.5-0.03,0.5-.055,0.75+0.015,1.,0,0,0);
 534         pad4 = new TPad("pad4","proton efficiency",0.75-0.045,0.5-.055,1.,1.,0,0,0);
 535         pad5 = new TPad("pad5","electron contamination",0.,0.,.25+.045,.5+.055,0,0,0);
 536         pad6 = new TPad("pad6","pion contamination",.25-.015,0.,.5+.03,.5+.055,0,0,0);
 537         pad7 = new TPad("pad7","kaon contamination",.5-.03,0.,.75+.015,.5+.055,0,0,0);
 538         pad8 = new TPad("pad8","proton contamination",.75-.045,0.,1.,.5+.055,0,0,0);
 539       }
 540
 541       gStyle->SetOptStat(0);
 542       tp->SetFillStyle(0);
 543       tp->SetFillColor(0);
 544       tp->SetBorderSize(0);
 545       pidCanvas->cd();
 546       TText *text1= new TText(.1,.2,"Contamination              Efficiency");
 547       text1->SetTextAngle(90);
 548       if (strstr(outputmode,"visual")) text1->Draw();
 549       //tp->DrawText(.3,.0,"p (GeV/c");
 550       pidCanvas->cd();
 551       TText *text2= new TText(.8,.0,"p (GeV/c)");
 552       if (strstr(outputmode,"visual")) {
 553         text2->Draw();
 554         tp->Draw();
 555       }
 556     }
 557
 558
 559     if (strstr(eventType,"pp")){
 560       pad1->SetFillStyle(0);
 561       pad1->SetFillColor(10);
 562       tp->cd();
 563       if (strstr(outputmode,"visual")) pad1->Draw();
 564       pad1->cd();
 565       //eleff->SetLineWidth(15);
 566       eleff->SetLineWidth(3);
 567       eleff->SetMaximum(1.);
 568       if (indexOfFile==0) {
 569         //eleff->SetFillColor(33);
 570         //eleff->SetFillColor(30);
 571         //eleff->SetFillColor(0);
 572         //eleff->SetLineColor(4);
 573         eleff->SetLineColor(1);
 574         if (strstr(outputmode,"visual")) eleff->Draw();}
 575       else {
 576         eleff->SetFillStyle(0);
 577         eleff->SetFillColor(30);
 578         if (indexOfFile==2) {
 579           eleff->SetLineColor(3);
 580           eleff->SetLineWidth(3);
 581         } else {
 582           //eleff->SetLineColor(2);
 583           eleff->SetLineColor(1);
 584           eleff->SetLineStyle(2);}
 585         if (strstr(outputmode,"visual")) eleff->Draw("same");}
 586       //   eleff->Fit("pol1");
 587       TPaveLabel *ellab = new TPaveLabel(.42,.85,.52,1.05,"e");
 588       if (strstr(outputmode,"visual")) ellab->Draw();
 589     }
 590
 591     pad2->SetFillStyle(0);
 592     pad2->SetFillColor(10);
 593     tp->cd();
 594     if (strstr(outputmode,"visual")) pad2->Draw();
 595     pad2->cd();
 596     pieff->SetLineWidth(3);
 597     pieff->SetMaximum(1.);
 598     if (indexOfFile==0) {
 599       pieff->SetLineColor(1);
 600       if (strstr(outputmode,"visual")) pieff->Draw();
 601     } else {
 602       pieff->SetFillStyle(0);
 603       pieff->SetFillColor(30);
 604       if (indexOfFile==1) {
 605         pieff->SetLineStyle(2);
 606       } else if (indexOfFile==2) {
 607         pieff->SetLineStyle(3);
 608       } else {
 609         pieff->SetLineStyle(4);}
 610       if (strstr(outputmode,"visual")) pieff->Draw("same");}
 611     TPaveLabel *pilab = new TPaveLabel(1.7,.85,2.2,1.05,"#pi");
 612     if (strstr(outputmode,"visual")) pilab->Draw();
 613
 614     pad3->SetFillStyle(0);
 615     pad3->SetFillColor(10);
 616     tp->cd();
 617     if (strstr(outputmode,"visual")) pad3->Draw();
 618     pad3->cd();
 619     kaeff->SetLineWidth(3);
 620     kaeff->SetMaximum(1.);
 621     if (indexOfFile==0) {
 622       kaeff->SetLineColor(1);
 623       if (strstr(outputmode,"visual")) kaeff->Draw();
 624     } else {
 625       kaeff->SetFillStyle(0);
 626       kaeff->SetFillColor(30);
 627       if (indexOfFile==1) {
 628         kaeff->SetLineStyle(2);
 629       } else if (indexOfFile==2) {
 630         kaeff->SetLineStyle(3);
 631       } else {
 632         kaeff->SetLineStyle(4);}
 633       if (strstr(outputmode,"visual")) kaeff->Draw("same");}
 634     TPaveLabel *kalab = new TPaveLabel(1.7,.85,2.2,1.05,"K");
 635     if (strstr(outputmode,"visual")) kalab->Draw();
 636
 637     pad4->SetFillStyle(0);
 638     pad4->SetFillColor(10);
 639     tp->cd();
 640     if (strstr(outputmode,"visual")) pad4->Draw();
 641     pad4->cd();
 642     preff->SetLineWidth(3);
 643     preff->SetMaximum(1.);
 644     if (indexOfFile==0) {
 645       preff->SetLineColor(1);
 646       if (strstr(outputmode,"visual")) preff->Draw();
 647     } else {
 648       preff->SetFillStyle(0);
 649       preff->SetFillColor(30);
 650       if (indexOfFile==1) {
 651         preff->SetLineStyle(2);
 652       } else if (indexOfFile==2) {
 653         preff->SetLineStyle(3);
 654       } else {
 655         preff->SetLineStyle(4);}
 656       if (strstr(outputmode,"visual")) preff->Draw("same");}
 657     TPaveLabel *prlab = new TPaveLabel(3.2,.85,4.1,1.05,"p");
 658     if (strstr(outputmode,"visual")) prlab->Draw();
 659
 660     if (strstr(eventType,"pp")){
 661       pad5->SetFillStyle(0);
 662       pad5->SetFillColor(10);
 663       tp->cd();
 664       if (strstr(outputmode,"visual")) pad5->Draw();
 665       pad5->cd();
 666       //elcon->SetLineWidth(5);
 667       elcon->SetLineWidth(3);
 668       elcon->SetMaximum(1.);
 669       if (indexOfFile==0) {
 670         //elcon->SetFillColor(33);
 671         //elcon->SetFillColor(30);
 672         //elcon->SetLineColor(4);
 673         elcon->SetLineColor(1);
 674         if (strstr(outputmode,"visual")) elcon->Draw();}
 675       else {
 676         elcon->SetFillStyle(4000);
 677         elcon->SetFillColor(30);
 678         if (indexOfFile==2) {
 679           elcon->SetLineColor(3);
 680           elcon->SetLineWidth(3);
 681         } else {
 682           elcon->SetLineColor(2);
 683           elcon->SetLineStyle(2);}
 684         if (strstr(outputmode,"visual")) elcon->Draw("same");}
 685     }
 686
 687
 688     pad6->SetFillStyle(0);
 689     pad6->SetFillColor(10);
 690     tp->cd();
 691     if (strstr(outputmode,"visual")) pad6->Draw();
 692     pad6->cd();
 693     picon->SetLineWidth(3);
 694     picon->SetMaximum(1.);
 695     if (indexOfFile==0) {
 696       picon->SetLineColor(1);
 697       if (strstr(outputmode,"visual")) picon->Draw();}
 698     else {
 699       picon->SetFillStyle(0);
 700       picon->SetFillColor(30);
 701       if (indexOfFile==1) {
 702         picon->SetLineStyle(2);
 703       } else if (indexOfFile==2) {
 704         picon->SetLineStyle(3);
 705       } else {
 706         picon->SetLineStyle(4);
 707         TLine* line;
 708         line = new TLine(0.2,0.85,0.9,0.85);
 709         line->SetLineStyle(2);
 710         line->SetLineWidth(1);
 711         if (strstr(outputmode,"visual")) line->Draw();
 712         line = new TLine(0.2,0.65,0.9,0.65);
 713         line->SetLineWidth(2);
 714         if (strstr(outputmode,"visual")) line->Draw();
 715         line = new TLine(0.2,0.45,0.9,0.45);
 716         line->SetLineStyle(3);
 717         line->SetLineWidth(1);
 718         if (strstr(outputmode,"visual")) line->Draw();
 719         line = new TLine(0.2,0.25,0.9,0.25);
 720         line->SetLineStyle(4);
 721         line->SetLineWidth(1);
 722         if (strstr(outputmode,"visual")) line->Draw();
 723         TPaveLabel *pl = new TPaveLabel(1.1,0.8,1.9,0.9,"100 ps","br");
 724         pl->SetFillColor(18);
 725         pl->SetTextSize(0.99);
 726         if (strstr(outputmode,"visual")) pl->Draw();
 727         pl = new TPaveLabel(1.1,0.6,1.9,0.7,"150 ps","br");
 728         pl->SetFillColor(18);
 729         pl->SetTextSize(0.99);
 730         if (strstr(outputmode,"visual")) pl->Draw();
 731         pl = new TPaveLabel(1.1,0.4,1.9,0.5,"200 ps","br");
 732         pl->SetFillColor(18);
 733         pl->SetTextSize(0.99);
 734         if (strstr(outputmode,"visual")) pl->Draw();
 735         pl = new TPaveLabel(1.1,0.2,1.9,0.3,"300 ps","br");
 736         pl->SetFillColor(18);
 737         pl->SetTextSize(0.99);
 738         if (strstr(outputmode,"visual")) pl->Draw();
 739       }
 740       if (strstr(outputmode,"visual")) picon->Draw("same");}
 741
 742     pad7->SetFillStyle(0);
 743     pad7->SetFillColor(10);
 744     tp->cd();
 745     if (strstr(outputmode,"visual")) pad7->Draw();
 746     pad7->cd();
 747     kacon->SetLineWidth(3);
 748     kacon->SetMaximum(1.);
 749     if (indexOfFile==0) {
 750       kacon->SetLineColor(1);
 751       if (strstr(outputmode,"visual")) kacon->Draw();}
 752     else {
 753       kacon->SetFillStyle(0);
 754       kacon->SetFillColor(30);
 755       if (indexOfFile==1) {
 756         kacon->SetLineStyle(2);
 757       } else if (indexOfFile==2) {
 758         kacon->SetLineStyle(3);
 759       } else {
 760         kacon->SetLineStyle(4);}
 761       if (strstr(outputmode,"visual")) kacon->Draw("same");}
 762
 763     pad8->SetFillStyle(0);
 764     pad8->SetFillColor(10);
 765     tp->cd();
 766     if (strstr(outputmode,"visual")) pad8->Draw();
 767     pad8->cd();
 768     prcon->SetLineWidth(3);
 769     prcon->SetMaximum(1.);
 770     if (indexOfFile==0) {
 771       prcon->SetLineColor(1);
 772       if (strstr(outputmode,"visual")) prcon->Draw();}
 773     else {
 774       prcon->SetFillStyle(0);
 775       prcon->SetFillColor(30);
 776       if (indexOfFile==1) {
 777         prcon->SetLineStyle(2);
 778       } else if (indexOfFile==2) {
 779         prcon->SetLineStyle(3);
 780       } else {
 781         prcon->SetLineStyle(4);}
 782       if (strstr(outputmode,"visual")) prcon->Draw("same");}
 783
 784     if (indexOfFile==0) {
 785       momvsmassCanvas = new TCanvas("momvsmassCanvas","Momentum vs mass disribution",500,10,700,700);
 786       momvsmassCanvas->SetFillColor(0);
 787       momvsmassCanvas->SetBorderMode(0);
 788       gPad->SetFillStyle(0);
 789       gPad->SetBorderMode(0);
 790       gPad->SetFillColor(0);
 791       //   gStyle->SetOptStat(11);
 792       if (numfile==4) momvsmassCanvas->Divide(1,2,0,0);
 793     } else if (indexOfFile==1 && numfile == 4) {
 794       momvsmassCanvas->cd(1);
 795       TPaveLabel *pl = new TPaveLabel(-0.0376218,-3.03586,0.0979277,-2.70158,"100 ps","br");
 796       pl->SetFillColor(18);
 797       pl->SetTextSize(0.99);
 798       if (strstr(outputmode,"visual")) pl->Draw();
 799     } else if (indexOfFile==3 && numfile == 4) {
 800       momvsmassCanvas->cd(2);
 801       TPaveLabel *pl = new TPaveLabel(-0.0591866,-3.17077,0.076363,-2.86857,"300 ps","br");
 802       pl->SetFillColor(18);
 803       pl->SetTextSize(0.99);
 804       if (strstr(outputmode,"visual")) pl->Draw();
 805     }
 806     if (numfile !=4) momvsmassCanvas->cd();
 807     if (numfile !=4 || indexOfFile==1 || indexOfFile==3) {
 808       //   hpi->PaintStat2(01);
 809       hpi->SetMarkerColor(5);
 810       if (strstr(outputmode,"visual")) hpi->Draw();
 811       if(strstr(eventType,"pp")){
 812         hel->SetMarkerColor(2);
 813         if (strstr(outputmode,"visual")) hel->Draw("same");
 814       }
 815       hka->SetMarkerColor(4);
 816       if (strstr(outputmode,"visual")) hka->Draw("same");
 817       hpr->SetMarkerColor(3);
 818       if (strstr(outputmode,"visual")) hpr->Draw("same");
 819       if (strstr(outputmode,"visual")) {
 820         fkaon->Draw();
 821         fproton->Draw();
 822         if(strstr(eventType,"pp")){
 823           felectron->Draw();
 824           fpion->Draw();
 825         }
 826       }
 827       if(strstr(eventType,"pp")){
 828         //TPaveText *ep = new TPaveText(-0.05,-0.5,0.05,-0.3);
 829         //ep->AddText("e");
 830         TPaveLabel *ep = new TPaveLabel(.42,.85,.52,1.05,"e");
 831         if (strstr(outputmode,"visual")) ep->Draw();
 832       }
 833
 834       TPaveText *pip = new TPaveText(0.15,-1.0,0.25,-0.8);
 835       pip->AddText("#pi");
 836       if (strstr(outputmode,"visual")) pip->Draw();
 837       TPaveText *kp = new TPaveText(0.5,-2.0,0.6,-1.8);
 838       kp->AddText("K");
 839       if (strstr(outputmode,"visual")) kp->Draw();
 840       TPaveText *prp = new TPaveText(0.9,-2.7,1.0,-2.5);
 841       prp->AddText("p");
 842       if (strstr(outputmode,"visual")) prp->Draw();
 843       //   TText *text2= new TText(.59,.06,"Momentum");
 844       //   text1->SetTextAngle(90);
 845       //   text1->Draw();
 846       //pidCanvas->DrawText(.1,.2,"Contamination               Efficiency");
 847       momvsmassCanvas->Update();
 848       if(strstr(outputsavemode,"asC")) momvsmassCanvas->Print("momvsmassCanvas.C");
 849       if(strstr(outputsavemode,"asEPS")) momvsmassCanvas->Print("momvsmassCanvas.eps");
 850       pidCanvas->cd();
 851       pidCanvas->Update();
 852       if(strstr(outputsavemode,"asC")) pidCanvas->Print("pidCanvas.C");
 853       if(strstr(outputsavemode,"asEPS")) pidCanvas->Print("pidCanvas.eps");
 854       char outFileName[100];
 855       strcpy(outFileName,"histos");
 856       strcat(outFileName,foutfileName);
 857       TFile *houtfile = new TFile(outFileName,"recreate");
 858       houtfile->cd();
 859       // saving canvas
 860       pidCanvas->Write(0,TObject::kOverwrite);
 861       momvsmassCanvas->Write(0,TObject::kOverwrite);
 862       // saving histos
 863       pit->Write(0,TObject::kOverwrite);
 864       pig->Write(0,TObject::kOverwrite);
 865       pieff->Write(0,TObject::kOverwrite);
 866       picon->Write(0,TObject::kOverwrite);
 867       piid->Write(0,TObject::kOverwrite);
 868       piall->Write(0,TObject::kOverwrite);
 869       pigen->Write(0,TObject::kOverwrite);
 870       kat->Write(0,TObject::kOverwrite);
 871       kag->Write(0,TObject::kOverwrite);
 872       kaeff->Write(0,TObject::kOverwrite);
 873       kaid->Write(0,TObject::kOverwrite);
 874       kaall->Write(0,TObject::kOverwrite);
 875       kagen->Write(0,TObject::kOverwrite);
 876       kacon->Write(0,TObject::kOverwrite);
 877       prt->Write(0,TObject::kOverwrite);
 878       prg->Write(0,TObject::kOverwrite);
 879       preff->Write(0,TObject::kOverwrite);
 880       prcon->Write(0,TObject::kOverwrite);
 881       prid->Write(0,TObject::kOverwrite);
 882       prall->Write(0,TObject::kOverwrite);
 883       prgen->Write(0,TObject::kOverwrite);
 884       // 2-D
 885       hpi->Write(0,TObject::kOverwrite);
 886       hka->Write(0,TObject::kOverwrite);
 887       hpr->Write(0,TObject::kOverwrite);
 888       // electron histos
 889       if (hel && eleff && elcon && elid && elall){
 890         hel->Write(0,TObject::kOverwrite);
 891         eleff->Write(0,TObject::kOverwrite);
 892         elcon->Write(0,TObject::kOverwrite);
 893         elid->Write(0,TObject::kOverwrite);
 894         elall->Write(0,TObject::kOverwrite);
 895       }
 896       cout << "file " << houtfile << " has been created" << endl;
 897       cout << "it contains PID histos and canvas" << endl;
 898       houtfile->Close();
 899       houtfile->Write(0,TObject::kOverwrite);
 900     }
 901   }
 902
 903   if (strstr(outputmode,"novisual")){
 904     // free used memory
 905     delete pit  ; pit=0;
 906     delete pig  ; pig=0;
 907     delete pieff; pieff=0;
 908     delete picon; picon=0;
 909     delete piid ; piid=0;
 910     delete piall; piall=0;
 911     delete pigen; pigen=0;
 912     delete kat  ; kat=0;
 913     delete kag  ; kag=0;
 914     delete kaeff; kaeff=0;
 915     delete kaid;  kaid=0;
 916     delete kaall; kaall=0;
 917     delete kagen; kagen=0;
 918     delete kacon; kacon=0;
 919     delete prt;   prt=0;
 920     delete prg;   prg=0;
 921     delete preff; preff=0;
 922     delete prcon; prcon=0;
 923     delete prid;  prid=0;
 924     delete prall; prall=0;
 925     delete prgen; prgen=0;
 926     // 2-D
 927     delete hpi; hpi=0;
 928     delete hka; hka=0;
 929     delete hpr; hpr=0;
 930     if (hel){
 931       delete hel;
 932       hel=0;
 933     }
 934     if (eleff){
 935       delete eleff;
 936       eleff=0;
 937     }
 938     if (elcon){
 939       delete elcon;
 940       elcon=0;
 941     }
 942     if (elid){
 943       delete elid;
 944       elid=0;
 945     }
 946     if (elall){
 947       delete elall;
 948       elall=0;
 949     }
 950   }
 951 }
 952
 953
 954 //__________________________________________________________________
 955 Bool_t AliTOFPID::operator==( AliTOFPID const & tofrec)const
 956 {
 957   // dummy version of Equal operator.
 958   // requested by coding conventions
 959   return kTRUE;
 960
 961 }