A lot of changes here:
[u/mrichter/AliRoot.git] / PHOS / AliPHOSPIDv1.h
1 #ifndef ALIPHOSPIDV1_H
2 #define ALIPHOSPIDV1_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8
9 //_________________________________________________________________________
10 // Implementation version v1 of the PHOS particle identifier 
11 // Identification is based on information from PPSD and EMC
12 // Oh yeah                 
13 //*-- Author: Yves Schutz (SUBATECH)
14
15 // --- ROOT system ---
16 class TFormula ;
17 class TVector3 ;
18
19 // --- Standard library ---
20
21 // --- AliRoot header files ---
22 class AliPHOSEmcRecPoint ;
23 class AliPHOSRecPoint ;
24
25 #include "AliPHOSPID.h"
26
27 class  AliPHOSPIDv1 : public AliPHOSPID {
28
29 public:
30
31   AliPHOSPIDv1() ;          // ctor            
32   AliPHOSPIDv1(const char* headerFile, const char * tsBranch = 0) ;
33   virtual ~AliPHOSPIDv1() ; // dtor
34
35   virtual void Exec(Option_t * option);
36   virtual char * GetRecParticlesBranch()const {return (char*) fRecParticlesTitle.Data() ;}      
37   virtual char * GetTrackSegmentsBranch()const{return (char*) fTrackSegmentsTitle.Data(); }
38
39   virtual void Init() ;
40   virtual void PlotDispersionCuts()const ;
41   virtual void Print(Option_t * option)const ; 
42   virtual void SetIdentificationMethod(char * option = "CPV DISP" ){fIDOptions = option ;} 
43   virtual void SetShowerProfileCut(char * formula = "0.35*0.35 - (x-1.386)*(x-1.386) - 1.707*1.707*(y-1.008)*(y-1.008)") ;
44   virtual void SetDispersionCut(Float_t cut){fDispersion = cut ; } 
45   virtual void SetCpvtoEmcDistanceCut(Float_t cut ) {fCpvEmcDistance = cut ;}
46   virtual void SetTrackSegmentsBranch(const char* title) { fTrackSegmentsTitle = title;}
47   virtual void SetRecParticlesBranch (const char* title) { fRecParticlesTitle = title;} 
48   virtual const char * Version() const { return "pid-v1" ; }  
49                      
50  private:
51
52   void     MakeRecParticles(void ) ;
53   Float_t  GetDistance(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, Option_t * Axis)const ; // Relative Distance PPSD-EMC
54   TVector3 GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, AliPHOSRecPoint * ppsd)const ;
55   void     PrintRecParticles(Option_t * option) ;
56   virtual Bool_t ReadTrackSegments(Int_t event) ;
57   virtual void WriteRecParticles(Int_t event) ; 
58
59  private:
60
61   TString                fHeaderFileName ;    // file name with event header
62   TString                fTrackSegmentsTitle; // branch name with track segments
63   TString                fRecPointsTitle ;    // branch name with rec points
64   TString                fRecParticlesTitle ; // branch name with rec particles
65   TString                fIDOptions ;         // PID option
66   Int_t                  fNEvent ;            // current event number
67   TObjArray            * fEmcRecPoints ;      // ! initial EMC RecPoints
68   TObjArray            * fCpvRecPoints ;      // ! initial CPV RecPoints
69   TClonesArray         * fTrackSegments;      // ! initial list of TrackSegments
70   TClonesArray         * fRecParticles ;      // ! output
71
72   AliPHOSClusterizer   * fClusterizer ;       // !
73   AliPHOSTrackSegmentMaker * fTSMaker ;       // !
74
75   TFormula             * fFormula ;           // formula to define cut on the shouer elips axis
76   Float_t                fDispersion ;        // dispersion cut
77   Float_t                fCpvEmcDistance ;    // Max EMC-CPV distance
78
79   ClassDef( AliPHOSPIDv1,1)  // Particle identifier implementation version 1
80
81 };
82
83 #endif // AliPHOSPIDV1_H