[u/mrichter/AliRoot.git] / HLT / hough / AliL3HoughEval.h

#ifndef ALIL3_HOUGH_Eval
#define ALIL3_HOUGH_Eval

#include "AliL3RootTypes.h"


class AliL3HoughBaseTransformer;
class AliL3Transform;
class AliL3HoughTrack;
class AliL3DigitRowData;
class AliL3Histogram;
class AliL3Histogram1D;

class AliL3HoughEval {
  
 private:

  Int_t fSlice;
  Int_t fPatch;
  Int_t fNrows;
  Int_t fNEtaSegments;
  Double_t fEtaMin;
  Double_t fEtaMax;
  Int_t fNumOfPadsToLook;
  Int_t fNumOfRowsToMiss;
  AliL3Histogram1D **fEtaHistos; //!

  //Flags
  Bool_t fRemoveFoundTracks;
  
  AliL3Transform *fTransform; //!
  AliL3HoughBaseTransformer *fHoughTransformer; //!
  AliL3DigitRowData **fRowPointers; //!
  
 public:
  AliL3HoughEval(); 
  virtual ~AliL3HoughEval();
  
  void InitTransformer(AliL3HoughBaseTransformer *transformer);
  void GenerateLUT();
  void DisplayEtaSlice(Int_t eta_index,AliL3Histogram *hist);
  Bool_t LookInsideRoad(AliL3HoughTrack *track,Int_t eta_index,Bool_t remove=kFALSE);
#ifdef use_root
  void CompareMC(AliL3TrackArray *tracks,Char_t *goodtracks="good_tracks",Int_t treshold=0);
#endif
  void FindEta(AliL3TrackArray *tracks);
  
  //Getters
  AliL3Histogram1D *GetEtaHisto(Int_t i) {if(!fEtaHistos) return 0; if(!fEtaHistos[i]) return 0; return fEtaHistos[i];}

  //Setters:
  void RemoveFoundTracks() {fRemoveFoundTracks = kTRUE;}
  void SetNumOfRowsToMiss(Int_t i) {fNumOfRowsToMiss = i;}
  void SetNumOfPadsToLook(Int_t i) {fNumOfPadsToLook = i;}

  ClassDef(AliL3HoughEval,1) //Hough transform verfication class

};

#endif
Commit	Line	Data
4de874d1	1	#ifndef ALIL3_HOUGH_Eval
	2	#define ALIL3_HOUGH_Eval
	3
	4	#include "AliL3RootTypes.h"
	5
162e2e5a	6
237d3f5c	7	class AliL3HoughBaseTransformer;
99e7186b	8	class AliL3Transform;
	9	class AliL3HoughTrack;
	10	class AliL3DigitRowData;
	11	class AliL3Histogram;
b46b21a5	12	class AliL3Histogram1D;
162e2e5a	13
95a00d93	14	class AliL3HoughEval {
4de874d1	15
4de874d1	16	private:
99e7186b	17
	18	Int_t fSlice;
	19	Int_t fPatch;
	20	Int_t fNrows;
	21	Int_t fNEtaSegments;
	22	Double_t fEtaMin;
	23	Double_t fEtaMax;
f80b98cb	24	Int_t fNumOfPadsToLook;
f80b98cb	25	Int_t fNumOfRowsToMiss;
b46b21a5	26	AliL3Histogram1D **fEtaHistos; //!
b46b21a5	27
99e7186b	28	//Flags
	29	Bool_t fRemoveFoundTracks;
	30
	31	AliL3Transform *fTransform; //!
237d3f5c	32	AliL3HoughBaseTransformer *fHoughTransformer; //!
99e7186b	33	AliL3DigitRowData **fRowPointers; //!
f80b98cb	34
4de874d1	35	public:
4de874d1	36	AliL3HoughEval();
4de874d1	37	virtual ~AliL3HoughEval();
4de874d1	38
237d3f5c	39	void InitTransformer(AliL3HoughBaseTransformer *transformer);
99e7186b	40	void GenerateLUT();
	41	void DisplayEtaSlice(Int_t eta_index,AliL3Histogram *hist);
	42	Bool_t LookInsideRoad(AliL3HoughTrack *track,Int_t eta_index,Bool_t remove=kFALSE);
95a00d93	43	#ifdef use_root
b1886074	44	void CompareMC(AliL3TrackArray tracks,Char_t goodtracks="good_tracks",Int_t treshold=0);
95a00d93	45	#endif
b46b21a5	46	void FindEta(AliL3TrackArray *tracks);
99e7186b	47
b46b21a5	48	//Getters
	49	AliL3Histogram1D *GetEtaHisto(Int_t i) {if(!fEtaHistos) return 0; if(!fEtaHistos[i]) return 0; return fEtaHistos[i];}
	50
99e7186b	51	//Setters:
99e7186b	52	void RemoveFoundTracks() {fRemoveFoundTracks = kTRUE;}
4fc9a6a4	53	void SetNumOfRowsToMiss(Int_t i) {fNumOfRowsToMiss = i;}
	54	void SetNumOfPadsToLook(Int_t i) {fNumOfPadsToLook = i;}
	55
b1886074	56	ClassDef(AliL3HoughEval,1) //Hough transform verfication class
4de874d1	57
	58	};
	59
	60	#endif