Check on existing of primary added
[u/mrichter/AliRoot.git] / PHOS / AliPHOSPIDv1.cxx
CommitLineData
6ad0bfa0 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
b2a60966 16/* $Id$ */
17
6ad0bfa0 18//_________________________________________________________________________
b2a60966 19// Implementation version v1 of the PHOS particle identifier
7acf6008 20// Particle identification based on the
148b2bba 21// - RCPV: distance from CPV recpoint to EMCA recpoint.
22// - TOF
23// - PCA: Principal Components Analysis..
24// The identified particle has an identification number corresponding
25// to a 9 bits number:
26// -Bit 0 to 2: bit set if RCPV > fCpvEmcDistance (each bit corresponds
27// to a different efficiency-purity point of the photon identification)
28// -Bit 3 to 5: bit set if TOF < fTimeGate (each bit corresponds
29// to a different efficiency-purity point of the photon identification)
30// -Bit 6 to 9: bit set if Principal Components are
50739f15 31// inside an ellipse defined by the parameters a, b, c, x0 and y0.
148b2bba 32// (each bit corresponds to a different efficiency-purity point of the
50739f15 33// photon identification)
34// The PCA (Principal components analysis) needs a file that contains
35// a previous analysis of the correlations between the particles. This
36// file is $ALICE_ROOT/PHOS/PCA8pa15_0.5-100.root. Analysis don for
37// energies between 0.5 and 100 GeV.
9fa5f1d0 38// A calibrated energy is calculated. The energy of the reconstructed
50739f15 39// cluster is corrected with the formula A + B * E + C * E^2, whose
40// parameters where obtained thourgh the study of the reconstructed
41// energy distribution of monoenergetic photons.
a4e98857 42//
50739f15 43// All the parameters (RCPV(6 rows-3 columns),TOF(6r-3c),PCA(5r-4c)
44// and calibration(1r-3c))are stored in a file called
45// $ALICE_ROOT/PHOS/Parameters.dat. Each time that AliPHOSPIDv1 is
46// initialized, this parameters are copied to a Matrix (18,4), a
47// TMatrixD object.
7acf6008 48//
a4e98857 49// use case:
50739f15 50// root [0] AliPHOSPIDv1 * p = new AliPHOSPIDv1("galice1.root")
a4e98857 51// Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
50739f15 52// // reading headers from file galice1.root and create RecParticles
53 // TrackSegments and RecPoints are used
54// // set file name for the branch RecParticles
f0a4c9e9 55// root [1] p->ExecuteTask("deb all time")
50739f15 56// // available options
57// // "deb" - prints # of reconstructed particles
58// // "deb all" - prints # and list of RecParticles
59// // "time" - prints benchmarking results
7acf6008 60//
50739f15 61// root [2] AliPHOSPIDv1 * p2 = new AliPHOSPIDv1("galice1.root","v1",kTRUE)
148b2bba 62// Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
50739f15 63// //Split mode.
f0a4c9e9 64// root [3] p2->ExecuteTask()
65//
50739f15 66
f0a4c9e9 67
7acf6008 68//*-- Author: Yves Schutz (SUBATECH) & Gines Martinez (SUBATECH) &
148b2bba 69// Gustavo Conesa April 2002
50739f15 70// PCA redesigned by Gustavo Conesa October 2002:
71// The way of using the PCA has changed. Instead of 2
72// files with the PCA, each one with different energy ranges
73// of application, we use the wide one (0.5-100 GeV), and instead
74// of fixing 3 elipses for different ranges of energy, it has been
75// studied the dependency of the ellipses parameters with the
76// energy, and they are implemented in the code as a funtion
77// of the energy.
78//
79//
80//
6ad0bfa0 81// --- ROOT system ---
7acf6008 82#include "TROOT.h"
83#include "TTree.h"
84#include "TFile.h"
85#include "TF2.h"
86#include "TFormula.h"
87#include "TCanvas.h"
88#include "TFolder.h"
89#include "TSystem.h"
90#include "TBenchmark.h"
148b2bba 91#include "TMatrixD.h"
92#include "TPrincipal.h"
93#include "TSystem.h"
94
6ad0bfa0 95// --- Standard library ---
96
75a6835b 97#include <Riostream.h>
98
6ad0bfa0 99// --- AliRoot header files ---
100
7acf6008 101#include "AliRun.h"
102#include "AliGenerator.h"
103#include "AliPHOS.h"
26d4b141 104#include "AliPHOSPIDv1.h"
7acf6008 105#include "AliPHOSClusterizerv1.h"
6ad0bfa0 106#include "AliPHOSTrackSegment.h"
7acf6008 107#include "AliPHOSTrackSegmentMakerv1.h"
6ad0bfa0 108#include "AliPHOSRecParticle.h"
7b7c1533 109#include "AliPHOSGeometry.h"
110#include "AliPHOSGetter.h"
6ad0bfa0 111
26d4b141 112ClassImp( AliPHOSPIDv1)
6ad0bfa0 113
1cb7c1ee 114//____________________________________________________________________________
115AliPHOSPIDv1::AliPHOSPIDv1():AliPHOSPID()
116{
a4e98857 117 // default ctor
148b2bba 118
8d0f3f77 119 InitParameters() ;
92f521a9 120 fDefaultInit = kTRUE ;
121
7acf6008 122}
123
124//____________________________________________________________________________
fbf811ec 125AliPHOSPIDv1::AliPHOSPIDv1(const char * headerFile,const char * name, const Bool_t toSplit)
126:AliPHOSPID(headerFile, name,toSplit)
7acf6008 127{
a4e98857 128 //ctor with the indication on where to look for the track segments
7b7c1533 129
8d0f3f77 130 InitParameters() ;
131
2bd5457f 132 Init() ;
92f521a9 133 fDefaultInit = kFALSE ;
7acf6008 134
135}
7b7c1533 136
7acf6008 137//____________________________________________________________________________
138AliPHOSPIDv1::~AliPHOSPIDv1()
139{
79bb1b62 140 // dtor
92f521a9 141 // fDefaultInit = kTRUE if PID created by default ctor (to get just the parameters)
9fa5f1d0 142
a496c46c 143 delete [] fX ; // Principal input
148b2bba 144 delete [] fP ; // Principal components
6f969528 145// delete fParameters ; // Matrix of Parameters
50739f15 146
9fa5f1d0 147
8d0f3f77 148
92f521a9 149 if (!fDefaultInit) {
fbf811ec 150// AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
92f521a9 151 // remove the task from the folder list
fbf811ec 152// gime->RemoveTask("P",GetName()) ;
153// TString name(GetName()) ;
154// name.ReplaceAll("pid", "clu") ;
155// gime->RemoveTask("C",name) ;
92f521a9 156
fbf811ec 157// // remove the data from the folder list
158// name = GetName() ;
159// name.Remove(name.Index(":")) ;
160// gime->RemoveObjects("RE", name) ; // EMCARecPoints
161// gime->RemoveObjects("RC", name) ; // CPVRecPoints
162// gime->RemoveObjects("T", name) ; // TrackSegments
163// gime->RemoveObjects("P", name) ; // RecParticles
92f521a9 164
fbf811ec 165// // Delete gAlice
166// gime->CloseFile() ;
92f521a9 167
168 fSplitFile = 0 ;
79bb1b62 169 }
7acf6008 170}
2bd5457f 171
a496c46c 172//____________________________________________________________________________
173const TString AliPHOSPIDv1::BranchName() const
174{
175 TString branchName(GetName() ) ;
176 branchName.Remove(branchName.Index(Version())-1) ;
177 return branchName ;
178}
179
148b2bba 180//____________________________________________________________________________
181void AliPHOSPIDv1::Init()
182{
183 // Make all memory allocations that are not possible in default constructor
184 // Add the PID task to the list of PHOS tasks
a496c46c 185
148b2bba 186 if ( strcmp(GetTitle(), "") == 0 )
187 SetTitle("galice.root") ;
148b2bba 188
fbf811ec 189 TString branchname(GetName()) ;
190 branchname.Remove(branchname.Index(Version())-1) ;
191 AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(),branchname.Data(),fToSplit ) ;
192
193 // gime->SetRecParticlesTitle(BranchName()) ;
148b2bba 194 if ( gime == 0 ) {
21cd0c07 195 Error("Init", "Could not obtain the Getter object !" ) ;
148b2bba 196 return ;
197 }
fbf811ec 198
199 fSplitFile = 0 ;
200 if(fToSplit){
201 //First - extract full path if necessary
202 TString fileName(GetTitle()) ;
203 Ssiz_t islash = fileName.Last('/') ;
204 if(islash<fileName.Length())
205 fileName.Remove(islash+1,fileName.Length()) ;
206 else
207 fileName="" ;
208 fileName+="PHOS.RecData." ;
209 if((strcmp(branchname.Data(),"Default")!=0)&&(strcmp(branchname.Data(),"")!=0)){
210 fileName+=branchname.Data() ;
211 fileName+="." ;
212 }
213 fileName+="root" ;
214 fSplitFile = static_cast<TFile*>(gROOT->GetFile(fileName.Data()));
215 if(!fSplitFile)
216 fSplitFile = TFile::Open(fileName.Data(),"update") ;
217 }
a496c46c 218
148b2bba 219 gime->PostPID(this) ;
220 // create a folder on the white board //YSAlice/WhiteBoard/RecParticles/PHOS/recparticlesName
fbf811ec 221 gime->PostRecParticles(branchname) ;
148b2bba 222
223}
8d0f3f77 224
225//____________________________________________________________________________
226void AliPHOSPIDv1::InitParameters()
227{
fbf811ec 228// fFrom = "" ;
229// fHeaderFileName = GetTitle() ;
230// TString name(GetName()) ;
231// if (name.IsNull())
232// name = "Default" ;
233// fTrackSegmentsTitle = name ;
234// fRecPointsTitle = name ;
235// fRecParticlesTitle = name ;
236// name.Append(":") ;
237// name.Append(Version()) ;
238// SetName(name) ;
8d0f3f77 239 fRecParticlesInRun = 0 ;
8d0f3f77 240 fNEvent = 0 ;
fbf811ec 241 // fClusterizer = 0 ;
242 // fTSMaker = 0 ;
8d0f3f77 243 fRecParticlesInRun = 0 ;
fbf811ec 244 TString pidName( GetName()) ;
245 if (pidName.IsNull() )
246 pidName = "Default" ;
247 pidName.Append(":") ;
248 pidName.Append(Version()) ;
249 SetName(pidName) ;
9fa5f1d0 250 SetParameters() ; // fill the parameters matrix from parameters file
8d0f3f77 251}
252
69183710 253//____________________________________________________________________________
50739f15 254const Double_t AliPHOSPIDv1::GetCpvtoEmcDistanceCut(const Float_t Cluster_En, const TString Eff_Pur) const
148b2bba 255{
256 // Get CpvtoEmcDistanceCut parameter depending on the cluster energy and
257 // Purity-Efficiency point (possible options "HIGH EFFICIENCY"
258 // "MEDIUM EFFICIENCY" "LOW EFFICIENCY" and 3 more options changing
259 // EFFICIENCY by PURITY)
148b2bba 260
9fa5f1d0 261 Int_t eff_pur = GetEffPurOption(Eff_Pur);
50739f15 262 Int_t cluster = GetClusterOption(Cluster_En) ;
263 if((cluster!= -1)&&(eff_pur != -1))
264 return (*fParameters)(cluster,eff_pur) ;
9fa5f1d0 265 else
266 return 0.0;
148b2bba 267}
268//____________________________________________________________________________
269
50739f15 270const Double_t AliPHOSPIDv1::GetTimeGate(const Float_t Cluster_En, const TString Eff_Pur) const
148b2bba 271{
272 // Get TimeGate parameter depending on the cluster energy and
273 // Purity-Efficiency point (possible options "HIGH EFFICIENCY"
274 // "MEDIUM EFFICIENCY" "LOW EFFICIENCY" and 3 more options changing
275 // EFFICIENCY by PURITY)
9fa5f1d0 276
f0a4c9e9 277 Int_t eff_pur = GetEffPurOption(Eff_Pur);
50739f15 278 Int_t cluster = GetClusterOption(Cluster_En) ;
279 if((cluster!= -1)&&(eff_pur != -1))
280 return (*fParameters)(cluster+6,eff_pur) ;
9fa5f1d0 281 else
282 return 0.0;
f0a4c9e9 283
148b2bba 284}
285//_____________________________________________________________________________
50739f15 286const Float_t AliPHOSPIDv1::GetDistance(AliPHOSEmcRecPoint * emc,AliPHOSRecPoint * cpv, Option_t * Axis)const
69183710 287{
288 // Calculates the distance between the EMC RecPoint and the PPSD RecPoint
148b2bba 289
7b7c1533 290 const AliPHOSGeometry * geom = AliPHOSGetter::GetInstance()->PHOSGeometry() ;
69183710 291 TVector3 vecEmc ;
7acf6008 292 TVector3 vecCpv ;
148b2bba 293 if(cpv){
294 emc->GetLocalPosition(vecEmc) ;
295 cpv->GetLocalPosition(vecCpv) ;
296 if(emc->GetPHOSMod() == cpv->GetPHOSMod()){
297 // Correct to difference in CPV and EMC position due to different distance to center.
298 // we assume, that particle moves from center
299 Float_t dCPV = geom->GetIPtoOuterCoverDistance();
300 Float_t dEMC = geom->GetIPtoCrystalSurface() ;
301 dEMC = dEMC / dCPV ;
302 vecCpv = dEMC * vecCpv - vecEmc ;
303 if (Axis == "X") return vecCpv.X();
304 if (Axis == "Y") return vecCpv.Y();
305 if (Axis == "Z") return vecCpv.Z();
306 if (Axis == "R") return vecCpv.Mag();
7acf6008 307 }
148b2bba 308
309 return 100000000 ;
310 }
7acf6008 311 return 100000000 ;
69183710 312}
313
6ad0bfa0 314//____________________________________________________________________________
50739f15 315const Double_t AliPHOSPIDv1::GetCalibratedEnergy(const Float_t e) const
316{
317// It calibrates Energy depending on the recpoint energy.
318// The energy of the reconstructed cluster is corrected with
319// the formula A + B* E + C* E^2, whose parameters where obtained
320// through the study of the reconstructed energy distribution of
321// monoenergetic photons.
322
323 Double_t p[]={0.,0.,0.};
324 Int_t i;
325 for(i=0;i<3;i++) p[i]= (*fParameters)(17,i);
326 Double_t enerec = p[0] + p[1]* e+ p[2] * e * e;
9fa5f1d0 327 return enerec ;
328
329}
330//____________________________________________________________________________
50739f15 331const Int_t AliPHOSPIDv1::GetPrincipalSign(const Double_t* P,const Int_t eff_pur, const Float_t E)const
148b2bba 332{
50739f15 333 //Is the particle inside de PCA ellipse?
334
335 Int_t prinsign= 0 ;
336 Double_t A = GetEllipseParameter("a", E);
337 Double_t B = GetEllipseParameter("b", E);
338 Double_t C = GetEllipseParameter("c", E);
339 Double_t X_center = GetEllipseParameter("x0", E);
340 Double_t Y_center = GetEllipseParameter("y0", E);
148b2bba 341
50739f15 342 Double_t R = TMath::Power((P[0] - X_center)/A,2) +
343 TMath::Power((P[1] - Y_center)/B,2) +
344 C*(P[0] - X_center)*(P[1] - Y_center)/(A*B) ;
345 //3 different ellipses defined
346 if((eff_pur==2)&&(R <1./2.)) prinsign= 1;
347 if((eff_pur==1)&&(R <2. )) prinsign= 1;
348 if((eff_pur==0)&&(R <9./2.)) prinsign= 1;
349
21cd0c07 350 if(R<0)
351 Error("GetPrincipalSign", "Negative square?") ;
148b2bba 352 return prinsign;
148b2bba 353
148b2bba 354}
148b2bba 355
148b2bba 356//_____________________________________________________________________________
357void AliPHOSPIDv1::SetCpvtoEmcDistanceCut(Float_t Cluster_En, TString Eff_Pur, Float_t cut)
358{
359
360 // Set the parameter Cpvto EmcDistanceCut depending on the cluster energy and
361 // Purity-Efficiency point (possible options "HIGH EFFICIENCY"
362 // "MEDIUM EFFICIENCY" "LOW EFFICIENCY" and 3 more options changing
363 // EFFICIENCY by PURITY)
364
92f521a9 365
9fa5f1d0 366 Int_t eff_pur = GetEffPurOption(Eff_Pur);
50739f15 367 Int_t cluster = GetClusterOption(Cluster_En) ;
368 if((cluster!= -1)&&(eff_pur != -1))
369 (*fParameters)(cluster,eff_pur) = cut ;
f0a4c9e9 370}
371//_____________________________________________________________________________
372void AliPHOSPIDv1::SetTimeGate(Float_t Cluster_En, TString Eff_Pur, Float_t gate)
373{
a496c46c 374
f0a4c9e9 375 // Set the parameter TimeGate depending on the cluster energy and
376 // Purity-Efficiency point (possible options "HIGH EFFICIENCY"
377 // "MEDIUM EFFICIENCY" "LOW EFFICIENCY" and 3 more options changing
378 // EFFICIENCY by PURITY)
379
f0a4c9e9 380 Int_t eff_pur = GetEffPurOption(Eff_Pur);
50739f15 381 Int_t cluster = GetClusterOption(Cluster_En) ;
382 if((cluster!= -1)&&(eff_pur != -1))
383 (*fParameters)(cluster+6,eff_pur) = gate ;
f0a4c9e9 384}
a496c46c 385//_____________________________________________________________________________
50739f15 386void AliPHOSPIDv1::SetParameters()
a496c46c 387{
388 // PCA : To do the Principal Components Analysis it is necessary
389 // the Principal file, which is opened here
390 fX = new double[7]; // Data for the PCA
f0a4c9e9 391 fP = new double[7]; // Eigenvalues of the PCA
e0ed2e49 392
50739f15 393 // Open principal and parameters files to be used
394
395 fFileName = "$ALICE_ROOT/PHOS/PCA8pa15_0.5-100.root" ;
396 fFileNamePar = gSystem->ExpandPathName("$ALICE_ROOT/PHOS/Parameters.dat");
397 TFile f( fFileName.Data(), "read" ) ;
a496c46c 398 fPrincipal = dynamic_cast<TPrincipal*> (f.Get("principal")) ;
399 f.Close() ;
a496c46c 400
50739f15 401 // Initialization of the Parameters matrix. In the File Parameters.dat
402 // are all the parameters. These are introduced in a matrix of 18x4
403 //
404 // All the parameters defined in this file are, in order of row:
405 // CpvtoEmcDistanceCut (6 rows, each one depends on the energy range of the
406 // particle, and 3 columns, each one depending on the efficiency-purity
407 // point), TimeGate (the same) and the parameters of the functions that
408 // calculate the ellipse parameters, x0,y0,a, b, c. These 5 parameters
409 // (5 rows) depend on 4 parameters (columns). Finally there is a row with
410 // the energy calibration parameters, 3 parameters.
9fa5f1d0 411
50739f15 412 fParameters = new TMatrixD(18,4) ;
9fa5f1d0 413
50739f15 414 ifstream paramFile(fFileNamePar) ;
9fa5f1d0 415 Int_t h,n;
50739f15 416 for(h = 0; h< 18; h++){
417 for(n = 0; n< 4; n++){
9fa5f1d0 418 paramFile >> (*fParameters)(h,n) ;
a496c46c 419 }
420 }
9fa5f1d0 421 paramFile.close();
a496c46c 422}
148b2bba 423//_____________________________________________________________________________
50739f15 424const Int_t AliPHOSPIDv1::GetClusterOption(const Float_t Cluster_En) const
9fa5f1d0 425{
50739f15 426 // Gives the cluster energy range, for each range there is associated a TOF or RCPV
427 // parameter.
428 Int_t cluster = -1;
429 if((Cluster_En > 0.0 )&&(Cluster_En <= 2.0 )) cluster = 0 ;
430 if((Cluster_En > 2.0 )&&(Cluster_En <= 5.0 )) cluster = 1 ;
431 if((Cluster_En > 5.0 )&&(Cluster_En <= 10.0)) cluster = 2 ;
432 if((Cluster_En > 10.0)&&(Cluster_En <= 20.0)) cluster = 3 ;
433 if((Cluster_En > 20.0)&&(Cluster_En <= 30.0)) cluster = 4 ;
434 if( Cluster_En > 30.0) cluster = 5 ;
f0a4c9e9 435
50739f15 436 return cluster;
f0a4c9e9 437}
438//____________________________________________________________________________
50739f15 439const Int_t AliPHOSPIDv1::GetEffPurOption(const TString Eff_Pur) const
f0a4c9e9 440{
441
442 // Looks for the Purity-Efficiency point (possible options "HIGH EFFICIENCY"
148b2bba 443 // "MEDIUM EFFICIENCY" "LOW EFFICIENCY" and 3 more options changing
444 // EFFICIENCY by PURITY)
445
148b2bba 446 Int_t eff_pur = -1 ;
447
f0a4c9e9 448 if(Eff_Pur.Contains("HIGH EFFICIENCY") ||Eff_Pur.Contains("LOW PURITY") )
449 eff_pur = 0 ;
450 else if(Eff_Pur.Contains("MEDIUM EFFICIENCY") ||Eff_Pur.Contains("MEDIUM PURITY") )
451 eff_pur = 1 ;
452 else if(Eff_Pur.Contains("LOW EFFICIENCY")||Eff_Pur.Contains("HIGH PURITY") )
453 eff_pur = 2 ;
148b2bba 454 else{
f0a4c9e9 455 eff_pur = -1;
21cd0c07 456 TString message ;
457 message = "Invalid Efficiency-Purity option\n";
458 message += "Possible options: HIGH EFFICIENCY = LOW PURITY\n" ;
459 message += " MEDIUM EFFICIENCY = MEDIUM PURITY\n" ;
460 message += " LOW EFFICIENCY = HIGH PURITY\n" ;
148b2bba 461 }
f0a4c9e9 462
463 return eff_pur;
464}
50739f15 465//________________________________________________________________________
466void AliPHOSPIDv1::SetEllipseParameter(TString Param, Int_t i, Double_t par)
467{
468 // Set the parameter "i" that is needed to calculate the ellipse
469 // parameter "Param".
470
471 Int_t p= -1;
472
473 if(Param.Contains("a"))p=12;
474 if(Param.Contains("b"))p=13;
475 if(Param.Contains("c"))p=14;
476 if(Param.Contains("x0"))p=15;
477 if(Param.Contains("y0"))p=16;
478 if((i>4)||(i<0))
21cd0c07 479 Error("SetEllipseParameter", "No parameter with index %d", i) ;
50739f15 480 else if(p==-1)
21cd0c07 481 Error("SetEllipseParameter", "No parameter with name %s", Param.Data() ) ;
50739f15 482 else
483 (*fParameters)(p,i) = par ;
484}
485//________________________________________________________________________
486const Double_t AliPHOSPIDv1::GetParameterToCalculateEllipse(const TString Param, const Int_t i) const
487{
488 // Get the parameter "i" that is needed to calculate the ellipse
489 // parameter "Param".
490
491 Int_t p= -1;
492 Double_t par = -1;
493
494 if(Param.Contains("a"))p=12;
495 if(Param.Contains("b"))p=13;
496 if(Param.Contains("c"))p=14;
497 if(Param.Contains("x0"))p=15;
498 if(Param.Contains("y0"))p=16;
499
500 if((i>4)||(i<0))
21cd0c07 501 Error("GetParameterToCalculateEllipse", "No parameter with index", i) ;
50739f15 502 else if(p==-1)
21cd0c07 503 Error("GetParameterToCalculateEllipse", "No parameter with name %s", Param.Data() ) ;
50739f15 504 else
505 par = (*fParameters)(p,i) ;
506
507 return par;
508
509}
510//____________________________________________________________________________
511void AliPHOSPIDv1::SetCalibrationParameter(Int_t i,Double_t param)
512{
513 (*fParameters)(17,i) = param ;
514}
515//____________________________________________________________________________
516const Double_t AliPHOSPIDv1::GetCalibrationParameter(const Int_t i) const
517{
518 Float_t param = (*fParameters)(17,i);
519 return param;
520}
521//____________________________________________________________________________
522const Double_t AliPHOSPIDv1::GetEllipseParameter(const TString Param,Float_t E) const
523{
524 Double_t p[4]={0.,0.,0.,0.};
525 Double_t value = 0.0;
526 Int_t i;
527
528 if(Param.Contains("a")){
529 for(i=0;i<4;i++)p[i]=(*fParameters)(12,i);
530 if(E>70.)E=70.;
531 }
532
533 if(Param.Contains("b")){
534 for(i=0;i<4;i++)p[i]=(*fParameters)(13,i);
535 if(E>70.)E=70.;
536 }
537
538 if(Param.Contains("c"))
539 for(i=0;i<4;i++)p[i]=(*fParameters)(14,i);
540
541 if(Param.Contains("x0")){
542 for(i=0;i<4;i++)p[i]=(*fParameters)(15,i);
543 if(E<1.)E=1.1;
544 }
545 if(Param.Contains("y0"))
546 for(i=0;i<4;i++)p[i]=(*fParameters)(16,i);
547
548 value = p[0]/TMath::Sqrt(E)+p[1]*E+p[2]*E*E+p[3];
549 return value;
550}
148b2bba 551//____________________________________________________________________________
552
7acf6008 553void AliPHOSPIDv1::Exec(Option_t * option)
6ad0bfa0 554{
f035f6ce 555 //Steering method
9688c1dd 556
bf8f1fbd 557 if( strcmp(GetName(), "")== 0 )
7acf6008 558 Init() ;
bf8f1fbd 559
560 if(strstr(option,"tim"))
7acf6008 561 gBenchmark->Start("PHOSPID");
bf8f1fbd 562
563 if(strstr(option,"print")) {
7b7c1533 564 Print("") ;
565 return ;
bf8f1fbd 566 }
9688c1dd 567
148b2bba 568
fbf811ec 569// gAlice->GetEvent(0) ;
148b2bba 570
fbf811ec 571// //check, if the branch with name of this" already exits?
572// if (gAlice->TreeR()) {
573// TObjArray * lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
574// TIter next(lob) ;
575// TBranch * branch = 0 ;
576// Bool_t phospidfound = kFALSE, pidfound = kFALSE ;
8d0f3f77 577
fbf811ec 578// TString taskName(GetName()) ;
579// taskName.Remove(taskName.Index(Version())-1) ;
8d0f3f77 580
fbf811ec 581// while ( (branch = (TBranch*)next()) && (!phospidfound || !pidfound) ) {
582// if ( (strcmp(branch->GetName(), "PHOSPID")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) )
583// phospidfound = kTRUE ;
8d0f3f77 584
fbf811ec 585// else if ( (strcmp(branch->GetName(), "AliPHOSPID")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) )
586// pidfound = kTRUE ;
587// }
7b7c1533 588
fbf811ec 589// if ( phospidfound || pidfound ) {
21cd0c07 590// Error("Exec", "RecParticles and/or PIDtMaker branch with name %s already exists", taskName.Data() ) ;
fbf811ec 591// return ;
592// }
593// }
594
595// Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ;
596// Int_t ievent ;
597// AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
9688c1dd 598
fbf811ec 599 AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
600 if(gime->BranchExists("RecParticles") )
601 return ;
602 Int_t nevents = gime->MaxEvent() ; //(Int_t) gAlice->TreeE()->GetEntries() ;
7b7c1533 603 Int_t ievent ;
fbf811ec 604
605
7b7c1533 606 for(ievent = 0; ievent < nevents; ievent++){
a496c46c 607 gime->Event(ievent,"R") ;
148b2bba 608
7acf6008 609 MakeRecParticles() ;
9688c1dd 610
7b7c1533 611 WriteRecParticles(ievent);
9688c1dd 612
7acf6008 613 if(strstr(option,"deb"))
614 PrintRecParticles(option) ;
94de8339 615
616 //increment the total number of rec particles per run
a496c46c 617 fRecParticlesInRun += gime->RecParticles(BranchName())->GetEntriesFast() ;
94de8339 618
7acf6008 619 }
9688c1dd 620
7acf6008 621 if(strstr(option,"tim")){
622 gBenchmark->Stop("PHOSPID");
21cd0c07 623 Info("Exec", "took %f seconds for PID %f seconds per event",
624 gBenchmark->GetCpuTime("PHOSPID"),
625 gBenchmark->GetCpuTime("PHOSPID")/nevents) ;
626 }
7b7c1533 627}
628
7acf6008 629//____________________________________________________________________________
630void AliPHOSPIDv1::MakeRecParticles(){
631
b2a60966 632 // Makes a RecParticle out of a TrackSegment
148b2bba 633
7b7c1533 634 AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
fbf811ec 635 TObjArray * emcRecPoints = gime->EmcRecPoints() ;
636 TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
637 TClonesArray * trackSegments = gime->TrackSegments() ;
148b2bba 638 if ( !emcRecPoints || !cpvRecPoints || !trackSegments ) {
f1611b7c 639 Fatal("MakeRecParticles", "RecPoints or TrackSegments not found !") ;
148b2bba 640 }
fbf811ec 641 TClonesArray * recParticles = gime->RecParticles() ;
01a599c9 642 recParticles->Clear();
148b2bba 643
644
7b7c1533 645 TIter next(trackSegments) ;
7acf6008 646 AliPHOSTrackSegment * ts ;
6ad0bfa0 647 Int_t index = 0 ;
09fc14a0 648 AliPHOSRecParticle * rp ;
7acf6008 649 while ( (ts = (AliPHOSTrackSegment *)next()) ) {
fad3e5b9 650
7b7c1533 651 new( (*recParticles)[index] ) AliPHOSRecParticle() ;
652 rp = (AliPHOSRecParticle *)recParticles->At(index) ;
f0a4c9e9 653 rp->SetTrackSegment(index) ;
9688c1dd 654 rp->SetIndexInList(index) ;
148b2bba 655
7acf6008 656 AliPHOSEmcRecPoint * emc = 0 ;
657 if(ts->GetEmcIndex()>=0)
7b7c1533 658 emc = (AliPHOSEmcRecPoint *) emcRecPoints->At(ts->GetEmcIndex()) ;
fad3e5b9 659
7acf6008 660 AliPHOSRecPoint * cpv = 0 ;
661 if(ts->GetCpvIndex()>=0)
7b7c1533 662 cpv = (AliPHOSRecPoint *) cpvRecPoints->At(ts->GetCpvIndex()) ;
fad3e5b9 663
148b2bba 664 // Now set type (reconstructed) of the particle
665
666 // Choose the cluster energy range
9fa5f1d0 667
fbf811ec 668 // YK: check if (emc != 0) !!!
669 if (!emc) {
f1611b7c 670 Fatal("MakeRecParticles", "-> emc(%d) = %d", ts->GetEmcIndex(), emc ) ;
fbf811ec 671 }
9fa5f1d0 672 Float_t e = emc->GetEnergy() ;
50739f15 673 Int_t cluster = GetClusterOption(e) ;// Gives value to cluster that defines the energy range parameter to be used in de RCPV, TOF and used in the PCA.
674 if(cluster== -1) continue ;
675
6f969528 676 Float_t lambda[2] ;
677 emc->GetElipsAxis(lambda) ;
50739f15 678
679 if((lambda[0]>0.01) && (lambda[1]>0.01)){
680 // Looking PCA. Define and calculate the data (X),
681 // introduce in the function
682 // X2P that gives the components (P).
683 Float_t Spher = 0. ;
684 Float_t Emaxdtotal = 0. ;
685
686 if((lambda[0]+lambda[1])!=0) Spher=fabs(lambda[0]-lambda[1])/(lambda[0]+lambda[1]);
687
688 Emaxdtotal=emc->GetMaximalEnergy()/emc->GetEnergy();
689
690 fX[0] = lambda[0] ;
691 fX[1] = lambda[1] ;
692 fX[2] = emc->GetDispersion() ;
693 fX[3] = Spher ;
694 fX[4] = emc->GetMultiplicity() ;
695 fX[5] = Emaxdtotal ;
696 fX[6] = emc->GetCoreEnergy() ;
697
698 fPrincipal->X2P(fX,fP);
699 }
700 else{
701 fP[0]=-100.0; //We do not accept clusters with
702 fP[1]=-100.0; //one cell as a photon-like
703 }
704
6f969528 705 Float_t time =emc->GetTime() ;
9fa5f1d0 706
50739f15 707 // Loop of Efficiency-Purity (the 3 points of purity or efficiency are taken
708 // into account to set the particle identification)
709 for(Int_t eff_pur = 0; eff_pur < 3 ; eff_pur++){
710
711 // Looking at the CPV detector. If RCPV greater than CpvEmcDistance, 1st,
712 // 2nd or 3rd bit (depending on the efficiency-purity point )is set to 1 .
713
714 if(GetDistance(emc, cpv, "R") > (*fParameters)(cluster,eff_pur) )
715 rp->SetPIDBit(eff_pur) ;
f0a4c9e9 716
50739f15 717 // Looking the TOF. If TOF smaller than gate, 4th, 5th or 6th
718 // bit (depending on the efficiency-purity point )is set to 1
719 if(time< (*fParameters)(cluster+6,eff_pur)) {
720 rp->SetPIDBit(eff_pur+3) ;
9fa5f1d0 721 }
50739f15 722
723 //If we are inside the ellipse, 7th, 8th or 9th
724 // bit (depending on the efficiency-purity point )is set to 1
725 if(GetPrincipalSign(fP,eff_pur,e) == 1)
726 rp->SetPIDBit(eff_pur+6) ;
f0a4c9e9 727 }
50739f15 728
9fa5f1d0 729 //Set momentum, energy and other parameters
50739f15 730 Float_t encal = GetCalibratedEnergy(e);
9fa5f1d0 731 TVector3 dir = GetMomentumDirection(emc,cpv) ;
732 dir.SetMag(encal) ;
733 rp->SetMomentum(dir.X(),dir.Y(),dir.Z(),encal) ;
734 rp->SetCalcMass(0);
e0ed2e49 735 rp->Name(); //If photon sets the particle pdg name to gamma
e747b8da 736 rp->SetProductionVertex(0,0,0,0);
737 rp->SetFirstMother(-1);
738 rp->SetLastMother(-1);
739 rp->SetFirstDaughter(-1);
740 rp->SetLastDaughter(-1);
741 rp->SetPolarisation(0,0,0);
6ad0bfa0 742 index++ ;
743 }
7acf6008 744
6ad0bfa0 745}
746
09fc14a0 747//____________________________________________________________________________
a496c46c 748void AliPHOSPIDv1:: Print()
09fc14a0 749{
b2a60966 750 // Print the parameters used for the particle type identification
21cd0c07 751 TString message ;
752 message = "\n=============== AliPHOSPID1 ================\n" ;
753 message += "Making PID\n";
754 message += " Pricipal analysis file from 0.5 to 100 %s\n" ;
755 message += " Name of parameters file %s\n" ;
756 message += " Matrix of Parameters: 18x4\n" ;
757 message += " RCPV 6x3 [High Eff-Low Pur,Medium Eff-Pur, Low Eff-High Pur]\n" ;
758 message += " TOF 6x3 [High Eff-Low Pur,Medium Eff-Pur, Low Eff-High Pur]\n" ;
759 message += " PCA 5x4 [5 ellipse parametres and 4 parametres to calculate them: A/Sqrt(E) + B* E + C * E^2 + D]\n" ;
760 message += " Energy Calibration 1x3 [3 parametres to calibrate energy: A + B* E + C * E^2]\n" ;
761 Info("Print", message.Data(), fFileName.Data(), fFileNamePar.Data() ) ;
50739f15 762 fParameters->Print() ;
09fc14a0 763}
764
7acf6008 765//____________________________________________________________________________
7b7c1533 766void AliPHOSPIDv1::WriteRecParticles(Int_t event)
09fc14a0 767{
7b7c1533 768
769 AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ;
a496c46c 770
fbf811ec 771 TClonesArray * recParticles = gime->RecParticles() ;
bf8f1fbd 772 recParticles->Expand(recParticles->GetEntriesFast() ) ;
fbf811ec 773 TTree * treeR ;
774
775 if(fToSplit){
776 if(!fSplitFile)
777 return ;
778 fSplitFile->cd() ;
779 char name[10] ;
780 sprintf(name,"%s%d", "TreeR",event) ;
781 treeR = dynamic_cast<TTree*>(fSplitFile->Get(name));
782 }
783 else{
784 treeR = gAlice->TreeR();
785 }
786
787 if(!treeR){
788 gAlice->MakeTree("R", fSplitFile);
789 treeR = gAlice->TreeR() ;
790 }
7acf6008 791
792 //First rp
793 Int_t bufferSize = 32000 ;
8d0f3f77 794 TBranch * rpBranch = treeR->Branch("PHOSRP",&recParticles,bufferSize);
fbf811ec 795 rpBranch->SetTitle(BranchName());
8d0f3f77 796
9688c1dd 797
7acf6008 798 //second, pid
799 Int_t splitlevel = 0 ;
800 AliPHOSPIDv1 * pid = this ;
8d0f3f77 801 TBranch * pidBranch = treeR->Branch("AliPHOSPID","AliPHOSPIDv1",&pid,bufferSize,splitlevel);
fbf811ec 802 pidBranch->SetTitle(BranchName());
7acf6008 803
761e34c0 804 rpBranch->Fill() ;
a496c46c 805 pidBranch->Fill() ;
9688c1dd 806
fbf811ec 807 treeR->AutoSave() ; //Write(0,kOverwrite) ;
808 if(gAlice->TreeR()!=treeR){
809 treeR->Delete();
810 }
7acf6008 811}
69183710 812
7acf6008 813//____________________________________________________________________________
9688c1dd 814TVector3 AliPHOSPIDv1::GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv)const
7acf6008 815{
816 // Calculates the momentum direction:
817 // 1. if only a EMC RecPoint, direction is given by IP and this RecPoint
9688c1dd 818 // 2. if a EMC RecPoint and CPV RecPoint, direction is given by the line through the 2 recpoints
819 // However because of the poor position resolution of PPSD the direction is always taken as if we were
7acf6008 820 // in case 1.
821
822 TVector3 dir(0,0,0) ;
823
824 TVector3 emcglobalpos ;
825 TMatrix dummy ;
826
827 emc->GetGlobalPosition(emcglobalpos, dummy) ;
828
148b2bba 829
7acf6008 830 dir = emcglobalpos ;
831 dir.SetZ( -dir.Z() ) ; // why ?
832 dir.SetMag(1.) ;
833
834 //account correction to the position of IP
835 Float_t xo,yo,zo ; //Coordinates of the origin
836 gAlice->Generator()->GetOrigin(xo,yo,zo) ;
837 TVector3 origin(xo,yo,zo);
838 dir = dir - origin ;
839
840 return dir ;
841}
842//____________________________________________________________________________
a4e98857 843void AliPHOSPIDv1::PrintRecParticles(Option_t * option)
844{
dd5c4038 845 // Print table of reconstructed particles
846
7b7c1533 847 AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ;
bf8f1fbd 848
a496c46c 849 TClonesArray * recParticles = gime->RecParticles(BranchName()) ;
21cd0c07 850
851 TString message ;
3bf72d32 852 message = "\nevent " ;
853 message += gAlice->GetEvNumber() ;
854 message += " found " ;
855 message += recParticles->GetEntriesFast();
856 message += " RecParticles\n" ;
857
7acf6008 858 if(strstr(option,"all")) { // printing found TS
3bf72d32 859 message += "\n PARTICLE Index \n" ;
7acf6008 860
861 Int_t index ;
7b7c1533 862 for (index = 0 ; index < recParticles->GetEntries() ; index++) {
21cd0c07 863 AliPHOSRecParticle * rp = (AliPHOSRecParticle * ) recParticles->At(index) ;
3bf72d32 864 message += "\n" ;
865 message += rp->Name().Data() ;
866 message += " " ;
867 message += rp->GetIndexInList() ;
868 message += " " ;
869 message += rp->GetType() ;
7acf6008 870 }
3bf72d32 871 }
872 Info("Print", message.Data() ) ;
69183710 873}
874
7acf6008 875
876