[u/mrichter/AliRoot.git] / HMPID / AliHMPIDRecon.h

#ifndef AliHMPIDRecon_h
#define AliHMPIDRecon_h

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//////////////////////////////////////////////////////////////////////////
//                                                                      //
// AliHMPIDRecon                                                         //
//                                                                      //
// HMPID class to perfom pattern recognition based on Hough transfrom    //
//                                                                      //
//////////////////////////////////////////////////////////////////////////


#include <TTask.h>        //base class
#include <TVector3.h>     //fields 

class TClonesArray; //CkovAngle()
class AliESDtrack;  //CkovAngle()

class AliHMPIDRecon : public TTask 
{
public : 
             AliHMPIDRecon();
    virtual ~AliHMPIDRecon()                                                          {}

  
  void     CkovAngle    (AliESDtrack *pTrk,TClonesArray *pCluLst,Double_t nmean             );     //reconstructed Theta Cerenkov
  Bool_t   FindPhotCkov (Double_t cluX,Double_t cluY,Double_t &thetaCer,Double_t &phiCer    );     //find ckov angle for single photon candidate
  Double_t FindRingCkov (Int_t iNclus                                                       );     //best ckov for ring formed by found photon candidates
  Double_t FindRingArea (Double_t ckov                                                      )const;//estimated area of ring in cm^2
  Int_t    FlagPhot     (Double_t ckov                                                      );     //is photon ckov near most probable track ckov
  Double_t HoughResponse(                                                                   );     //most probable track ckov angle
  void     Propagate    (const TVector3  dir,      TVector3 &pos,Double_t z                 )const;//propagate photon alogn the line  
  void     Refract      (      TVector3 &dir,                    Double_t n1,    Double_t n2)const;//refract photon on the boundary
  TVector2 TracePhot    (Double_t ckovTh,Double_t ckovPh                                    )const;//trace photon created by track to PC 
  TVector2 TraceForward (TVector3 dirCkov                                                   )const;//tracing forward a photon from (x,y) to PC
  void     RecPhot      (TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer               );     //theta,phi cerenkov reconstructed
  void     SetTrack     (Double_t xRad,Double_t yRad,Double_t theta,Double_t phi            )
                                {fTrkDir.SetMagThetaPhi(1,theta,phi);  fTrkPos.Set(xRad,yRad);}    //set track parameter at RAD
  void     SetImpPC     (Double_t xPc,Double_t yPc                                          )
                                {fPc.Set(xPc,yPc);}                                                //set track impact to PC 
  Double_t SigLoc       (Double_t ckovTh,Double_t ckovPh,Double_t beta                      )const;//error due to cathode segmetation
  Double_t SigGeom      (Double_t ckovTh,Double_t ckovPh,Double_t beta                      )const;//error due to unknown photon origin
  Double_t SigCrom      (Double_t ckovTh,Double_t ckovPh,Double_t beta                      )const;//error due to unknonw photon energy
  Double_t Sigma2       (Double_t ckovTh,Double_t ckovPh                                    )const;//photon candidate sigma
  enum ETrackingFlags {kMipDistCut=-9,kMipQdcCut=-5,kNoPhotAccept=-11};
// hidden track algorithm
  void     CkovHiddenTrk    (TClonesArray *pCluLst,Double_t nmean);                                //Pattern recognition without trackinf information
  void     CluPreFilter     (                                    );                                //Pre clustering filter to cut bkg clusters
  void     DoRecHiddenTrk   (                                    );                                //Calling to the fitted procedures
  Bool_t   FitEllipse       (Double_t &phiRec                    );                                //Fit clusters with a conical section (kTRUE only for ellipses)
  void     FitFree          (Double_t phiRec                     );                                //Fit (th,ph) of the track and ckovFit as result
  Double_t FunConSect       (Double_t *c,Double_t x,Double_t y   );                                //Function of a general conical section
  void     SetCkovFit       (Double_t ckov                       ) {fCkovFit=ckov;}                //Setter for ckof fitted
  void     SetTrkFit        (Double_t th,Double_t ph             ) {fThTrkFit = th;fPhTrkFit = ph;}//Setter for (th,ph) of the track
  static void     FunMinEl  (Int_t&/* */,Double_t* /* */,Double_t &f,Double_t *par,Int_t /* */);   //Fit function to find ellipes parameters
  static void     FunMinPhot(Int_t&/* */,Double_t* /* */,Double_t &f,Double_t *par,Int_t /* */);   //Fit function to minimize thetaCer RMS/Sqrt(n) of n clusters
  Int_t    IdxMip       ()const {return fIdxMip;}                                                  //Getter index of MIP
  Double_t MipX         ()const {return fMipX;}                                                    //Getter of x MIP in LORS
  Double_t MipY         ()const {return fMipY;}                                                    //Getter of y MIP in LORS
  Int_t    NClu         ()const {return fNClu;}                                                    //Getter of cluster multiplicity
  Double_t XClu         (Int_t i)const {return fXClu[i];}                                          //Getter of x clu
  Double_t YClu         (Int_t i)const {return fYClu[i];}                                          //Getter of y clu
  Double_t CkovFit      ()const {return fCkovFit;}                                                 //Getter of ckov angle fitted
  Double_t ThTrkFit     ()const {return fThTrkFit;}                                                //Getter of theta fitted of the track
  Double_t PhTrkFit     ()const {return fPhTrkFit;}                                                //Getter of phi fitted of the track
//
protected:
  Double_t fRadNmean;                          //C6F14 mean refractive index
  Int_t    fPhotCnt;                           // counter of photons candidate
  Int_t    fPhotFlag[3000];                    // flags of photon candidates
  Double_t fPhotCkov[3000];                    // Ckov angles of photon candidates, [rad]
  Double_t fPhotPhi [3000];                    // phis of photons candidates, [rad]
  Double_t fPhotWei [3000];                    // weigths of photon candidates
  Double_t fCkovSigma2;                        // sigma2 of the reconstructed ring

  Bool_t   fIsWEIGHT;                          // flag to consider weight procedure
  Float_t  fDTheta;                            // Step for sliding window
  Float_t  fWindowWidth;                       // Hough width of sliding window
  
  TVector3 fTrkDir;                            //track direction in LORS at RAD
  TVector2 fTrkPos;                            //track positon in LORS at RAD
  TVector2 fPc;                                //track position at PC
// hidden track algorithm
  Double_t fMipX;                              //mip X position for Hidden Track Algorithm  
  Double_t fMipY;                              //mip Y position for Hidden Track Algorithm
  Int_t    fIdxMip;                            //mip index in the clus list
  Int_t    fNClu;                              //n clusters to fit
  Double_t fXClu[1000];                        //container for x clus position
  Double_t fYClu[1000];                        //container for y clus position
  Double_t fThTrkFit;                          //theta fitted of the track
  Double_t fPhTrkFit;                          //phi   fitted of the track
  Double_t fCkovFit;                           //estimated ring Cherenkov angle
//
private:  
  static const Double_t fgkRadThick;                      //radiator thickness
  static const Double_t fgkWinThick;                      //window thickness
  static const Double_t fgkGapThick;                      //proximity gap thickness
  static const Double_t fgkWinIdx;                        //mean refractive index of WIN material (SiO2) 
  static const Double_t fgkGapIdx;                        //mean refractive index of GAP material (CH4)
  
  ClassDef(AliHMPIDRecon,0)
};

#endif // #ifdef AliHMPIDRecon_cxx
Commit	Line	Data
d3da6dc4	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
59280a5a	19	class TClonesArray; //CkovAngle()
	20	class AliESDtrack; //CkovAngle()
	21
d3da6dc4	22	class AliHMPIDRecon : public TTask
	23	{
	24	public :
	25	AliHMPIDRecon();
	26	virtual ~AliHMPIDRecon() {}
	27
	28
a591e55f	29	void CkovAngle (AliESDtrack pTrk,TClonesArray pCluLst,Double_t nmean ); //reconstructed Theta Cerenkov
a591e55f	30	Bool_t FindPhotCkov (Double_t cluX,Double_t cluY,Double_t &thetaCer,Double_t &phiCer ); //find ckov angle for single photon candidate
d3da6dc4	31	Double_t FindRingCkov (Int_t iNclus ); //best ckov for ring formed by found photon candidates
	32	Double_t FindRingArea (Double_t ckov )const;//estimated area of ring in cm^2
	33	Int_t FlagPhot (Double_t ckov ); //is photon ckov near most probable track ckov
	34	Double_t HoughResponse( ); //most probable track ckov angle
a591e55f	35	void Propagate (const TVector3 dir, TVector3 &pos,Double_t z )const;//propagate photon alogn the line
d3da6dc4	36	void Refract ( TVector3 &dir, Double_t n1, Double_t n2)const;//refract photon on the boundary
a591e55f	37	TVector2 TracePhot (Double_t ckovTh,Double_t ckovPh )const;//trace photon created by track to PC
	38	TVector2 TraceForward (TVector3 dirCkov )const;//tracing forward a photon from (x,y) to PC
	39	void RecPhot (TVector3 dirCkov,Double_t &thetaCer,Double_t &phiCer ); //theta,phi cerenkov reconstructed
	40	void SetTrack (Double_t xRad,Double_t yRad,Double_t theta,Double_t phi )
	41	{fTrkDir.SetMagThetaPhi(1,theta,phi); fTrkPos.Set(xRad,yRad);} //set track parameter at RAD
	42	void SetImpPC (Double_t xPc,Double_t yPc )
	43	{fPc.Set(xPc,yPc);} //set track impact to PC
d3da6dc4	44	Double_t SigLoc (Double_t ckovTh,Double_t ckovPh,Double_t beta )const;//error due to cathode segmetation
	45	Double_t SigGeom (Double_t ckovTh,Double_t ckovPh,Double_t beta )const;//error due to unknown photon origin
	46	Double_t SigCrom (Double_t ckovTh,Double_t ckovPh,Double_t beta )const;//error due to unknonw photon energy
	47	Double_t Sigma2 (Double_t ckovTh,Double_t ckovPh )const;//photon candidate sigma
59280a5a	48	enum ETrackingFlags {kMipDistCut=-9,kMipQdcCut=-5,kNoPhotAccept=-11};
43400d2d	49	// hidden track algorithm
	50	void CkovHiddenTrk (TClonesArray *pCluLst,Double_t nmean); //Pattern recognition without trackinf information
	51	void CluPreFilter ( ); //Pre clustering filter to cut bkg clusters
	52	void DoRecHiddenTrk ( ); //Calling to the fitted procedures
	53	Bool_t FitEllipse (Double_t &phiRec ); //Fit clusters with a conical section (kTRUE only for ellipses)
	54	void FitFree (Double_t phiRec ); //Fit (th,ph) of the track and ckovFit as result
	55	Double_t FunConSect (Double_t *c,Double_t x,Double_t y ); //Function of a general conical section
	56	void SetCkovFit (Double_t ckov ) {fCkovFit=ckov;} //Setter for ckof fitted
	57	void SetTrkFit (Double_t th,Double_t ph ) {fThTrkFit = th;fPhTrkFit = ph;}//Setter for (th,ph) of the track
	58	static void FunMinEl (Int_t&/* /,Double_t /* /,Double_t &f,Double_t par,Int_t /* */); //Fit function to find ellipes parameters
	59	static void FunMinPhot(Int_t&/* /,Double_t /* /,Double_t &f,Double_t par,Int_t /* */); //Fit function to minimize thetaCer RMS/Sqrt(n) of n clusters
	60	Int_t IdxMip ()const {return fIdxMip;} //Getter index of MIP
	61	Double_t MipX ()const {return fMipX;} //Getter of x MIP in LORS
	62	Double_t MipY ()const {return fMipY;} //Getter of y MIP in LORS
	63	Int_t NClu ()const {return fNClu;} //Getter of cluster multiplicity
	64	Double_t XClu (Int_t i)const {return fXClu[i];} //Getter of x clu
	65	Double_t YClu (Int_t i)const {return fYClu[i];} //Getter of y clu
	66	Double_t CkovFit ()const {return fCkovFit;} //Getter of ckov angle fitted
	67	Double_t ThTrkFit ()const {return fThTrkFit;} //Getter of theta fitted of the track
	68	Double_t PhTrkFit ()const {return fPhTrkFit;} //Getter of phi fitted of the track
	69	//
394d35c1	70	protected:
abb5f786	71	Double_t fRadNmean; //C6F14 mean refractive index
d3da6dc4	72	Int_t fPhotCnt; // counter of photons candidate
	73	Int_t fPhotFlag[3000]; // flags of photon candidates
	74	Double_t fPhotCkov[3000]; // Ckov angles of photon candidates, [rad]
	75	Double_t fPhotPhi [3000]; // phis of photons candidates, [rad]
	76	Double_t fPhotWei [3000]; // weigths of photon candidates
	77	Double_t fCkovSigma2; // sigma2 of the reconstructed ring
	78
43400d2d	79	Bool_t fIsWEIGHT; // flag to consider weight procedure
	80	Float_t fDTheta; // Step for sliding window
	81	Float_t fWindowWidth; // Hough width of sliding window
e0085eed	82
43400d2d	83	TVector3 fTrkDir; //track direction in LORS at RAD
	84	TVector2 fTrkPos; //track positon in LORS at RAD
	85	TVector2 fPc; //track position at PC
	86	// hidden track algorithm
	87	Double_t fMipX; //mip X position for Hidden Track Algorithm
	88	Double_t fMipY; //mip Y position for Hidden Track Algorithm
	89	Int_t fIdxMip; //mip index in the clus list
	90	Int_t fNClu; //n clusters to fit
	91	Double_t fXClu[1000]; //container for x clus position
	92	Double_t fYClu[1000]; //container for y clus position
	93	Double_t fThTrkFit; //theta fitted of the track
	94	Double_t fPhTrkFit; //phi fitted of the track
	95	Double_t fCkovFit; //estimated ring Cherenkov angle
	96	//
e0085eed	97	private:
	98	static const Double_t fgkRadThick; //radiator thickness
	99	static const Double_t fgkWinThick; //window thickness
	100	static const Double_t fgkGapThick; //proximity gap thickness
	101	static const Double_t fgkWinIdx; //mean refractive index of WIN material (SiO2)
	102	static const Double_t fgkGapIdx; //mean refractive index of GAP material (CH4)
	103
d3da6dc4	104	ClassDef(AliHMPIDRecon,0)
d3da6dc4	105	};
cf7e313e	106
d3da6dc4	107	#endif // #ifdef AliHMPIDRecon_cxx
d3da6dc4	108