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