Backward compatibility using typedef, removing some debug printout
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDHelix.h
1 #ifndef AliHMPIDHelix_h
2 #define AliHMPIDHelix_h
3
4 #include <TObject.h>              //base class
5 #include <TVector3.h>             //used extensively
6
7 class AliHMPIDHelix: public TObject
8 {
9 public:
10   AliHMPIDHelix(                                                            ):TObject(),fX0(TVector3(0,0,0)),fP0(TVector3(0,0,0)),fQ(0),fBz(0 )  {}
11   AliHMPIDHelix(const TVector3 &x0,const TVector3 &p0,Int_t q=1,Double_t b=2):TObject(),fX0(x0             ),fP0(p0             ),fQ(q),fBz(b )  {}
12   AliHMPIDHelix(Double_t p,Double_t theta,Double_t phi,Double_t bz=2        ):TObject(),fX0(TVector3(0,0,0)),fP0(TVector3(0,0,0)),fQ(0),fBz(bz)  
13                    {fP0.SetMagThetaPhi(p,theta*TMath::DegToRad(),phi*TMath::DegToRad());} //p [GeV], theta,phi [deg], Bz [Tesla];
14   virtual ~AliHMPIDHelix()                                                                                                                   {}        
15            
16          void     Draw          (const Option_t *opt=""              );                              //from TObject, draw helix
17          void     Print    (const Option_t *opt=""              )const;                         //from TObject, print status
18 //private part++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
19   inline Bool_t   Intersect(            TVector3 &pnt,TVector3 &norm );          //intersection with plane given by point and normal vector
20   inline void     Propagate(Float_t len,TVector3 &x,  TVector3 &p    );          //propogate helix by given length along it            
21 protected:
22   TVector3 fX0;            //helix position in point of definition, [cm]    in MARS
23   TVector3 fP0;            //helix momentum in point of definition, [GeV/c] in MARS
24   Int_t    fQ;             //sign of track charge  
25   Float_t  fBz;            //magnetic field along z, [kGaus] 
26   ClassDef(AliHMPIDHelix,0) //General helix
27 };//class AliHMPIDHelix
28
29 typedef AliHMPIDHelix AliRICHHelix; // for backward compatibility
30
31 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
32 void AliHMPIDHelix::Propagate(Float_t len,TVector3 &x,TVector3 &p)
33 {
34 // Propogates the helix from inintial point by a given distance along helix. Assumes uniform magnetic field along z direction.  
35 // Arguments: len - distance to propagate by, [cm]
36 //   Returns: none    
37   if(fBz==0){//no magnetic field->straight line
38     x=fX0+fP0.Unit()*len;   
39     p=fP0;
40   }else{
41     const Float_t c = 0.000299792458;//this speed of light value provides that coordinates are in [cm] momentum in [GeV/c] field in [kGaus]
42     Float_t a = -c*fBz*fQ;
43     Float_t rho = a/fP0.Mag();
44     x.SetX( fX0.X()+fP0.X()*TMath::Sin(rho*len)/a-fP0.Y()*(1-TMath::Cos(rho*len))/a  );
45     x.SetY( fX0.Y()+fP0.Y()*TMath::Sin(rho*len)/a+fP0.X()*(1-TMath::Cos(rho*len))/a  ); 
46     x.SetZ( fX0.Z()+fP0.Z()*len/fP0.Mag()                                            );
47     x.SetX( fP0.X()*TMath::Cos(rho*len)-fP0.Y()*TMath::Sin(rho*len)                  );
48     p.SetY( fP0.Y()*TMath::Cos(rho*len)+fP0.X()*TMath::Sin(rho*len)                  );
49     p.SetZ( fP0.Z()                                                                  );
50   }
51 }//Propagate()
52 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
53 Bool_t AliHMPIDHelix::Intersect(TVector3 &pnt,TVector3 &norm)
54 {
55 // Finds point of intersection (if exists) of the helix with the plane. Stores result in fX and fP.   
56 // Arguments: pnt  - arbitrary point of the plane, [cm] in MARS
57 //            norm - vector, normal to the plane, [cm] in MARS
58 //   Returns:      - kTrue if helix intersects the plane, kFALSE otherwise.
59 //                 - pnt contains the point of intersection, [cm] in MARS  
60   TVector3 x,p;                                                    //current helix position and momentum
61   Double_t s=(pnt-fX0)*norm,dist=99999,distPrev=dist;              //estimates initial distance to plane
62   while(TMath::Abs(dist)>0.00001){                                 //loop while the distance is less then precision   
63     Propagate(s,x,p);                                              //calculates helix at the distance s from x0 ALONG the helix
64     dist=(x-pnt)*norm;                                             //distance between current helix position and plane
65     if(TMath::Abs(dist) >= TMath::Abs(distPrev)) { return kFALSE;} //if distance increases then no intersection 
66     distPrev=dist;
67     s-=dist;
68   }  
69   norm=p;
70   pnt=x;                                    
71   return kTRUE;
72 }//Intersect()
73 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
74 #endif