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

#ifndef AliHMPIDReconHTA_h
#define AliHMPIDReconHTA_h

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

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


#include <TTask.h>        //base class
#include <TH2F.h>         //InitDatabase()

class TClonesArray; //CkovAngle()
class AliESDtrack;  //CkovAngle()
class AliHMPIDParam;//General pourpose

class AliHMPIDReconHTA : public TTask 
{
public : 
             AliHMPIDReconHTA();
    virtual ~AliHMPIDReconHTA();

// HTA hidden track algorithm
  void     InitVars         (Int_t n);                                                             //init space for variables
  void     DeleteVars       ()const;                                                               //delete variables
  void     InitDatabase     ();                                                                    //initialization of database
  TH2F*    DBHTA            ()     {return fgDatabase;}                                            //pointer for HTA database of rings
  void     FillZeroChan     ()const;                                                               //complete the DB
  Bool_t   CkovHiddenTrk    (AliESDtrack *pTrk,TClonesArray *pClu,Int_t index, Double_t nmean);    //Pattern recognition without trackinf information
  Bool_t   CluPreFilter     (TClonesArray *pClu               );                                   //Pre clustering filter to cut bkg clusters
  Bool_t   DoRecHiddenTrk   (                                 );                                   //Calling to the fitted procedures
  Bool_t   FindShape        (Double_t &thTrkRec,Double_t &phiTrkRec,Double_t &thetaCRec);          //Find shape of the ring
  Bool_t   ShapeModel       (Int_t np,Double_t *phiphot,Double_t *dist,Double_t &xA,Double_t &xB,Double_t &phiStart);//initial shape model for the cluster candidates
  Double_t VertParab        (Double_t x1,Double_t y1,Double_t x2, Double_t y2, Double_t x3, Double_t y3)const;//calculate the coord. of the min. for a parabole for 3 points
  Bool_t   FitFree          (Double_t thTrkRec,Double_t phiTrkRec);                                //Fit (th,ph) of the track and ckovFit as result
  Double_t FindSimmPhi      ();                                                                    //find phi of the ring with min. dist. algorithm 
  Int_t    r2               (Double_t *coef, Double_t &x1, Double_t &x2);                          // solution of 2nd degree equation
  void     SetNClu          (Int_t nclu                          ) {fNClu=nclu;}                   //Setter for # of clusters
  void     SetClCk          (Int_t i,Bool_t what                 ) {fClCk[i]=what;}                //Setter for cluster flags 
  void     SetCkovFit       (Double_t ckov                       ) {fCkovFit=ckov;}                //Setter for ckov fitted
  void     SetCkovSig2      (Double_t rms                        ) {fCkovSig2=rms;}                //Setter for sigma2 ckov fitted
  void     SetTrkFit        (Double_t th,Double_t ph             ) {fThTrkFit = th;fPhTrkFit = ph;}//Setter for (th,ph) of the track
  void     SetRadXY         (Double_t  x,Double_t y              ) {fRadX = x;fRadY = y;}          //Setter for (th,ph) of the track
  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
  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
  Double_t MipQ         ()const {return fMipQ;}                                                    //Getter of Q MIP
  Double_t RadX         ()const {return fRadX;}                                                    //Getter of x at RAD in LORS
  Double_t RadY         ()const {return fRadY;}                                                    //Getter of y at RAD 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
  Bool_t   ClCk         (Int_t i)const {return fClCk[i];}                                          //Getter of cluster flags
  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 fMipX;                              //mip X position for Hidden Track Algorithm  
  Double_t fMipY;                              //mip Y position for Hidden Track Algorithm
  Double_t fMipQ;                              //mip Q          for Hidden Track Algorithm
  Double_t fRadX;                              //rad X position for Hidden Track Algorithm  
  Double_t fRadY;                              //rad 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;                             //container for x clus position
  Double_t *fYClu;                             //container for y clus position
  Bool_t   *fClCk;                             //flag if cluster is used in fitting
  Double_t fThTrkFit;                          //theta fitted of the track
  Double_t fPhTrkFit;                          //phi   fitted of the track
  Double_t fCkovFit;                           //estimated ring Cherenkov angle
  Double_t fCkovSig2;                          //estimated error^2 on ring Cherenkov angle
  
  AliHMPIDParam *fParam;                       //Pointer to AliHMPIDParam
  static TH2F* fgDatabase;                     //database for ring shapes
//
private:
  AliHMPIDReconHTA(const AliHMPIDReconHTA& r);              //dummy copy constructor
  AliHMPIDReconHTA &operator=(const AliHMPIDReconHTA& r);   //dummy assignment operator
//
  ClassDef(AliHMPIDReconHTA,1)
};

#endif // #ifdef AliHMPIDReconHTA_cxx
Commit	Line	Data
5a3482a0	1	#ifndef AliHMPIDReconHTA_h
	2	#define AliHMPIDReconHTA_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	// AliHMPIDReconHTA //
	10	// //
	11	// HMPID class to perfom pattern recognition based on Hough transfrom //
	12	// //
	13	//////////////////////////////////////////////////////////////////////////
	14
	15
	16	#include <TTask.h> //base class
9785d5fb	17	#include <TH2F.h> //InitDatabase()
5a3482a0	18
	19	class TClonesArray; //CkovAngle()
	20	class AliESDtrack; //CkovAngle()
5a3482a0	21	class AliHMPIDParam;//General pourpose
	22
	23	class AliHMPIDReconHTA : public TTask
	24	{
	25	public :
	26	AliHMPIDReconHTA();
	27	virtual ~AliHMPIDReconHTA();
	28
	29	// HTA hidden track algorithm
	30	void InitVars (Int_t n); //init space for variables
aa00f952	31	void DeleteVars ()const; //delete variables
5a3482a0	32	void InitDatabase (); //initialization of database
9785d5fb	33	TH2F* DBHTA () {return fgDatabase;} //pointer for HTA database of rings
	34	void FillZeroChan ()const; //complete the DB
	35	Bool_t CkovHiddenTrk (AliESDtrack pTrk,TClonesArray pClu,Int_t index, Double_t nmean); //Pattern recognition without trackinf information
5a3482a0	36	Bool_t CluPreFilter (TClonesArray *pClu ); //Pre clustering filter to cut bkg clusters
	37	Bool_t DoRecHiddenTrk ( ); //Calling to the fitted procedures
	38	Bool_t FindShape (Double_t &thTrkRec,Double_t &phiTrkRec,Double_t &thetaCRec); //Find shape of the ring
	39	Bool_t ShapeModel (Int_t np,Double_t phiphot,Double_t dist,Double_t &xA,Double_t &xB,Double_t &phiStart);//initial shape model for the cluster candidates
aa00f952	40	Double_t VertParab (Double_t x1,Double_t y1,Double_t x2, Double_t y2, Double_t x3, Double_t y3)const;//calculate the coord. of the min. for a parabole for 3 points
5a3482a0	41	Bool_t FitFree (Double_t thTrkRec,Double_t phiTrkRec); //Fit (th,ph) of the track and ckovFit as result
9a573d52	42	Double_t FindSimmPhi (); //find phi of the ring with min. dist. algorithm
9a573d52	43	Int_t r2 (Double_t *coef, Double_t &x1, Double_t &x2); // solution of 2nd degree equation
5a3482a0	44	void SetNClu (Int_t nclu ) {fNClu=nclu;} //Setter for # of clusters
	45	void SetClCk (Int_t i,Bool_t what ) {fClCk[i]=what;} //Setter for cluster flags
	46	void SetCkovFit (Double_t ckov ) {fCkovFit=ckov;} //Setter for ckov fitted
	47	void SetCkovSig2 (Double_t rms ) {fCkovSig2=rms;} //Setter for sigma2 ckov fitted
	48	void SetTrkFit (Double_t th,Double_t ph ) {fThTrkFit = th;fPhTrkFit = ph;}//Setter for (th,ph) of the track
	49	void SetRadXY (Double_t x,Double_t y ) {fRadX = x;fRadY = y;} //Setter for (th,ph) of the track
	50	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
	51	Int_t IdxMip ()const {return fIdxMip;} //Getter index of MIP
	52	Double_t MipX ()const {return fMipX;} //Getter of x MIP in LORS
	53	Double_t MipY ()const {return fMipY;} //Getter of y MIP in LORS
	54	Double_t MipQ ()const {return fMipQ;} //Getter of Q MIP
	55	Double_t RadX ()const {return fRadX;} //Getter of x at RAD in LORS
	56	Double_t RadY ()const {return fRadY;} //Getter of y at RAD in LORS
	57	Int_t NClu ()const {return fNClu;} //Getter of cluster multiplicity
	58	Double_t XClu (Int_t i)const {return fXClu[i];} //Getter of x clu
	59	Double_t YClu (Int_t i)const {return fYClu[i];} //Getter of y clu
	60	Bool_t ClCk (Int_t i)const {return fClCk[i];} //Getter of cluster flags
	61	Double_t CkovFit ()const {return fCkovFit;} //Getter of ckov angle fitted
	62	Double_t ThTrkFit ()const {return fThTrkFit;} //Getter of theta fitted of the track
	63	Double_t PhTrkFit ()const {return fPhTrkFit;} //Getter of phi fitted of the track
	64	//
	65	protected:
8f05fd11	66	//
5a3482a0	67	Double_t fMipX; //mip X position for Hidden Track Algorithm
	68	Double_t fMipY; //mip Y position for Hidden Track Algorithm
	69	Double_t fMipQ; //mip Q for Hidden Track Algorithm
	70	Double_t fRadX; //rad X position for Hidden Track Algorithm
	71	Double_t fRadY; //rad Y position for Hidden Track Algorithm
	72	Int_t fIdxMip; //mip index in the clus list
	73	Int_t fNClu; //n clusters to fit
	74	Double_t *fXClu; //container for x clus position
	75	Double_t *fYClu; //container for y clus position
	76	Bool_t *fClCk; //flag if cluster is used in fitting
	77	Double_t fThTrkFit; //theta fitted of the track
	78	Double_t fPhTrkFit; //phi fitted of the track
	79	Double_t fCkovFit; //estimated ring Cherenkov angle
	80	Double_t fCkovSig2; //estimated error^2 on ring Cherenkov angle
	81
	82	AliHMPIDParam *fParam; //Pointer to AliHMPIDParam
9785d5fb	83	static TH2F* fgDatabase; //database for ring shapes
8f05fd11	84	//
	85	private:
	86	AliHMPIDReconHTA(const AliHMPIDReconHTA& r); //dummy copy constructor
	87	AliHMPIDReconHTA &operator=(const AliHMPIDReconHTA& r); //dummy assignment operator
5a3482a0	88	//
9785d5fb	89	ClassDef(AliHMPIDReconHTA,1)
5a3482a0	90	};
	91
	92	#endif // #ifdef AliHMPIDReconHTA_cxx
	93