Mip-track matching improved. Now Last point of HMPID stored and track re-fitted.
[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 class AliHMPIDParam;//general pourpose
22
23 class AliHMPIDRecon : public TTask 
24 {
25 public : 
26              AliHMPIDRecon();
27     virtual ~AliHMPIDRecon() {;} //dtor
28
29   void     InitVars     (Int_t n);                                                                 //init space for variables
30   void     DeleteVars   ()const;                                                                   //delete variables
31   void     CkovAngle    (AliESDtrack *pTrk,TClonesArray *pCluLst,Int_t index,Double_t nmean );     //reconstructed Theta Cerenkov
32   Bool_t   FindPhotCkov (Double_t cluX,Double_t cluY,Double_t &thetaCer,Double_t &phiCer    );     //find ckov angle for single photon candidate
33   Double_t FindRingCkov (Int_t iNclus                                                       );     //best ckov for ring formed by found photon candidates
34   void     FindRingGeom (Double_t ckovAng,Int_t level=1                                     );     //estimated area of ring in cm^2 and portion accepted by geometry
35   TVector2 IntWithEdge  (TVector2 p1,TVector2 p2                                            )const;//find intercection between plane and lines of 2 thetaC
36   Int_t    FlagPhot     (Double_t ckov                                                      );     //is photon ckov near most probable track ckov
37   Double_t HoughResponse(                                                                   );     //most probable track ckov angle
38   void     Propagate    (const TVector3  dir,      TVector3 &pos,Double_t z                 )const;//propagate photon alogn the line  
39   void     Refract      (      TVector3 &dir,                    Double_t n1,    Double_t n2)const;//refract photon on the boundary
40   TVector2 TracePhot    (Double_t ckovTh,Double_t ckovPh                                    )const;//trace photon created by track to PC 
41   TVector2 TraceForward (TVector3 dirCkov                                                   )const;//tracing forward a photon from (x,y) to PC
42   void     Lors2Trs     (TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer               )const;//LORS to TRS 
43   void     Trs2Lors     (TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer               )const;//TRS to LORS
44   TVector2 GetMip       (                                                                   )const 
45                         {return fMipPos;}                                                          //mip coordinates
46   Double_t GetRingArea  (                                                                   )const
47                         {return fRingArea;}                                                        //area of the current ring in cm^2 
48   Double_t GetRingAcc   (                                                                   )const
49                         {return fRingAcc;}                                                         //portion of the ring ([0,1]) accepted by geometry.To scale n. of photons 
50   void     SetTrack     (Double_t xRad,Double_t yRad,Double_t theta,Double_t phi            )
51                                 {fTrkDir.SetMagThetaPhi(1,theta,phi);  fTrkPos.Set(xRad,yRad);}    //set track parameter at RAD
52   void     SetImpPC     (Double_t xPc,Double_t yPc                                          )
53                                 {fPc.Set(xPc,yPc);}                                                //set track impact to PC 
54   void     SetMip       (Double_t xmip,Double_t ymip                                        )
55                                 {fMipPos.Set(xmip,ymip);}                                          //set track impact to PC
56   enum ETrackingFlags {kNotPerformed=-20,kMipDistCut=-9,kMipQdcCut=-5,kNoPhotAccept=-11,kNoRad = -22};
57 //
58 protected:
59   Int_t     fPhotCnt;                           // counter of photons candidate
60   Int_t    *fPhotFlag;                          // flags of photon candidates
61   Double_t *fPhotCkov;                          // Ckov angles of photon candidates, [rad]
62   Double_t *fPhotPhi;                           // phis of photons candidates, [rad]
63   Double_t *fPhotWei;                           // weigths of photon candidates
64   Double_t  fCkovSigma2;                        // sigma2 of the reconstructed ring
65
66   Bool_t    fIsWEIGHT;                          // flag to consider weight procedure
67   Float_t   fDTheta;                            // Step for sliding window
68   Float_t   fWindowWidth;                       // Hough width of sliding window
69   
70   Double_t  fRingArea;                          // area of a given ring
71   Double_t  fRingAcc;                           // fraction of the ring accepted by geometry
72   TVector3  fTrkDir;                            // track direction in LORS at RAD
73   TVector2  fTrkPos;                            // track positon in LORS at RAD
74   TVector2  fMipPos;                            // mip positon for a given track
75   TVector2  fPc;                                // track position at PC
76   
77   AliHMPIDParam *fParam;                        // Pointer to AliHMPIDParam
78   
79 private:
80   AliHMPIDRecon(const AliHMPIDRecon& r);              //dummy copy constructor
81   AliHMPIDRecon &operator=(const AliHMPIDRecon& r);   //dummy assignment operator
82 //
83   ClassDef(AliHMPIDRecon,1)
84 };
85
86 #endif // #ifdef AliHMPIDRecon_cxx
87