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

// @(#) $Id$

#ifndef ALIL3_HOUGH_Eval
#define ALIL3_HOUGH_Eval

#include "AliL3RootTypes.h"

class AliL3TrackArray;
class AliL3HoughBaseTransformer;
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; //!
  Float_t fZVertex;

  //Flags
  Bool_t fRemoveFoundTracks;
  
  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 &nrows_crossed,Int_t *rowrange,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;}
  void SetSlice(Int_t i) {fSlice=i;}
  void SetZVertex(Float_t zvertex) {fZVertex=zvertex;}

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

};

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