Too wide range for charge.
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDRecon.h
1 #ifndef AliHMPIDRecon_h
2 #define AliHMPIDRecon_h
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 //////////////////////////////////////////////////////////////////////////
8 //                                                                      //
9 // AliHMPIDRecon                                                         //
10 //                                                                      //
11 // HMPID class to perfom pattern recognition based on Hough transfrom    //
12 //                                                                      //
13 //////////////////////////////////////////////////////////////////////////
14
15
16 #include <TTask.h>        //base class
17 #include <TVector3.h>     //fields 
18
19 class TClonesArray; //CkovAngle()
20 class AliESDtrack;  //CkovAngle()
21
22 class AliHMPIDRecon : public TTask 
23 {
24 public : 
25              AliHMPIDRecon();
26     virtual ~AliHMPIDRecon()                                                          {}
27
28
29   void     CkovAngle    (AliESDtrack *pTrk,TClonesArray *pCluLst,Double_t nmean, Double_t qthre);  //reconstructed Theta Cerenkov
30   Bool_t   FindPhotCkov (Double_t cluX,Double_t cluY,Double_t &thetaCer,Double_t &phiCer    );     //find ckov angle for single photon candidate
31   Double_t FindRingCkov (Int_t iNclus                                                       );     //best ckov for ring formed by found photon candidates
32   Double_t FindRingArea (Double_t ckovAng                                                   )const;//estimated area of delta ring in cm^2 to weight Hough Transform
33   TVector2 IntWithEdge  (TVector2 p1,TVector2 p2                                            )const;//find intercection between plane and lines of 2 thetaC
34   Int_t    FlagPhot     (Double_t ckov                                                      );     //is photon ckov near most probable track ckov
35   Double_t HoughResponse(                                                                   );     //most probable track ckov angle
36   void     Propagate    (const TVector3  dir,      TVector3 &pos,Double_t z                 )const;//propagate photon alogn the line  
37   void     Refract      (      TVector3 &dir,                    Double_t n1,    Double_t n2)const;//refract photon on the boundary
38   TVector2 TracePhot    (Double_t ckovTh,Double_t ckovPh                                    )const;//trace photon created by track to PC 
39   TVector2 TraceForward (TVector3 dirCkov                                                   )const;//tracing forward a photon from (x,y) to PC
40   void     RecPhot      (TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer               );     //theta,phi cerenkov reconstructed
41   TVector2 GetMip       (                                                                   ) 
42                         {return fMipPos;}                                                          //mip coordinates
43   void     SetTrack     (Double_t xRad,Double_t yRad,Double_t theta,Double_t phi            )
44                                 {fTrkDir.SetMagThetaPhi(1,theta,phi);  fTrkPos.Set(xRad,yRad);}    //set track parameter at RAD
45   void     SetImpPC     (Double_t xPc,Double_t yPc                                          )
46                                 {fPc.Set(xPc,yPc);}                                                //set track impact to PC 
47   void     SetMip       (Double_t xmip,Double_t ymip                                        )
48                                 {fMipPos.Set(xmip,ymip);}                                          //set track impact to PC 
49   Double_t SigLoc       (Double_t ckovTh,Double_t ckovPh,Double_t beta                      )const;//error due to cathode segmetation
50   Double_t SigGeom      (Double_t ckovTh,Double_t ckovPh,Double_t beta                      )const;//error due to unknown photon origin
51   Double_t SigCrom      (Double_t ckovTh,Double_t ckovPh,Double_t beta                      )const;//error due to unknonw photon energy
52   Double_t Sigma2       (Double_t ckovTh,Double_t ckovPh                                    )const;//photon candidate sigma^2
53   enum ETrackingFlags {kMipDistCut=-9,kMipQdcCut=-5,kNoPhotAccept=-11};
54 // HTA hidden track algorithm
55   Bool_t   CkovHiddenTrk    (AliESDtrack *pTrk,TClonesArray *pCluLst,Double_t nmean,Double_t qthre);//Pattern recognition without trackinf information
56   Bool_t   CluPreFilter     (TClonesArray *pClu               );                                   //Pre clustering filter to cut bkg clusters
57   Bool_t   DoRecHiddenTrk   (TClonesArray *pClu               );                                   //Calling to the fitted procedures
58   Bool_t   FitEllipse       (Double_t &phiRec                    );                                //Fit clusters with a conical section (kTRUE only for ellipses)
59   Bool_t   FitFree          (Double_t phiRec                     );                                //Fit (th,ph) of the track and ckovFit as result
60   Double_t FunConSect       (Double_t *c,Double_t x,Double_t y   );                                //Function of a general conical section
61   void     SetNClu          (Int_t nclu                          ) {fNClu=nclu;}                   //Setter for # of clusters
62   void     SetClCk          (Int_t i,Bool_t what                 ) {fClCk[i]=what;}                //Setter for cluster flags 
63   void     SetCkovFit       (Double_t ckov                       ) {fCkovFit=ckov;}                //Setter for ckov fitted
64   void     SetCkovSig2      (Double_t rms                       ) {fCkovSig2=rms;}                 //Setter for sigma2 ckov fitted
65   void     SetTrkFit        (Double_t th,Double_t ph             ) {fThTrkFit = th;fPhTrkFit = ph;}//Setter for (th,ph) of the track
66   void     SetRadXY         (Double_t  x,Double_t y              ) {fRadX = x;fRadY = y;}          //Setter for (th,ph) of the track
67   static void     FunMinEl  (Int_t&/* */,Double_t* /* */,Double_t &f,Double_t *par,Int_t /* */);   //Fit function to find ellipes parameters
68   static void     FunMinPhot(Int_t&/* */,Double_t* /* */,Double_t &f,Double_t *par,Int_t iflag);   //Fit function to minimize thetaCer RMS/Sqrt(n) of n clusters
69   Int_t    IdxMip       ()const {return fIdxMip;}                                                  //Getter index of MIP
70   Double_t MipX         ()const {return fMipX;}                                                    //Getter of x MIP in LORS
71   Double_t MipY         ()const {return fMipY;}                                                    //Getter of y MIP in LORS
72   Double_t MipQ         ()const {return fMipQ;}                                                    //Getter of Q MIP
73   Double_t RadX         ()const {return fRadX;}                                                    //Getter of x at RAD in LORS
74   Double_t RadY         ()const {return fRadY;}                                                    //Getter of y at RAD in LORS
75   Int_t    NClu         ()const {return fNClu;}                                                    //Getter of cluster multiplicity
76   Double_t XClu         (Int_t i)const {return fXClu[i];}                                          //Getter of x clu
77   Double_t YClu         (Int_t i)const {return fYClu[i];}                                          //Getter of y clu
78   Bool_t   ClCk         (Int_t i)const {return fClCk[i];}                                          //Getter of cluster flags
79   Double_t CkovFit      ()const {return fCkovFit;}                                                 //Getter of ckov angle fitted
80   Double_t ThTrkFit     ()const {return fThTrkFit;}                                                //Getter of theta fitted of the track
81   Double_t PhTrkFit     ()const {return fPhTrkFit;}                                                //Getter of phi fitted of the track
82 //
83 protected:
84   Double_t fRadNmean;                          //C6F14 mean refractive index
85   Int_t    fPhotCnt;                           // counter of photons candidate
86   Int_t    fPhotFlag[3000];                    // flags of photon candidates
87   Double_t fPhotCkov[3000];                    // Ckov angles of photon candidates, [rad]
88   Double_t fPhotPhi [3000];                    // phis of photons candidates, [rad]
89   Double_t fPhotWei [3000];                    // weigths of photon candidates
90   Double_t fCkovSigma2;                        // sigma2 of the reconstructed ring
91
92   Bool_t   fIsWEIGHT;                          // flag to consider weight procedure
93   Float_t  fDTheta;                            // Step for sliding window
94   Float_t  fWindowWidth;                       // Hough width of sliding window
95   
96   TVector3 fTrkDir;                            //track direction in LORS at RAD
97   TVector2 fTrkPos;                            //track positon in LORS at RAD
98   TVector2 fMipPos;                            //mip positon for a given track
99   TVector2 fPc;                                //track position at PC
100 // HTA hidden track algorithm
101   Double_t fMipX;                              //mip X position for Hidden Track Algorithm  
102   Double_t fMipY;                              //mip Y position for Hidden Track Algorithm
103   Double_t fMipQ;                              //mip Q          for Hidden Track Algorithm
104   Double_t fRadX;                              //rad X position for Hidden Track Algorithm  
105   Double_t fRadY;                              //rad Y position for Hidden Track Algorithm
106   Int_t    fIdxMip;                            //mip index in the clus list
107   Int_t    fNClu;                              //n clusters to fit
108   Double_t fXClu[100];                         //container for x clus position
109   Double_t fYClu[100];                         //container for y clus position
110   Bool_t   fClCk[100];                         //flag if cluster is used in fitting
111   Double_t fThTrkFit;                          //theta fitted of the track
112   Double_t fPhTrkFit;                          //phi   fitted of the track
113   Double_t fCkovFit;                           //estimated ring Cherenkov angle
114   Double_t fCkovSig2;                          //estimated error^2 on ring Cherenkov angle
115 //
116 private:
117   static const Double_t fgkRadThick;                      //radiator thickness
118   static const Double_t fgkWinThick;                      //window thickness
119   static const Double_t fgkGapThick;                      //proximity gap thickness
120   static const Double_t fgkWinIdx;                        //mean refractive index of WIN material (SiO2) 
121   static const Double_t fgkGapIdx;                        //mean refractive index of GAP material (CH4)
122
123   ClassDef(AliHMPIDRecon,0)
124 };
125
126 #endif // #ifdef AliHMPIDRecon_cxx
127