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

// @(#) $Id$

#ifndef ALIL3HOUGHEVAL_H
#define ALIL3HOUGHEVAL_H

#include "AliHLTRootTypes.h"

class AliHLTTrackArray;
class AliHLTHoughBaseTransformer;
class AliHLTHoughTrack;
class AliHLTDigitRowData;
class AliHLTHistogram;
class AliHLTHistogram1D;

class AliHLTHoughEval {
  
 public:
  AliHLTHoughEval(); 
  virtual ~AliHLTHoughEval();
  
  void InitTransformer(AliHLTHoughBaseTransformer *transformer);
  void GenerateLUT();
  void DisplayEtaSlice(Int_t etaindex,AliHLTHistogram *hist);
  Bool_t LookInsideRoad(AliHLTHoughTrack *track,Int_t &nrowscrossed,Int_t *rowrange,Bool_t remove=kFALSE);
#ifdef use_root
  void CompareMC(AliHLTTrackArray *tracks,Char_t *goodtracks="good_tracks",Int_t treshold=0);
#endif
  void FindEta(AliHLTTrackArray *tracks);
  
  //Getters
  AliHLTHistogram1D *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;}

 private:

  Int_t fSlice;//Index of the slice being processed
  Int_t fPatch;//Index of the patch being processed
  Int_t fNrows;//Number of padrows inside the patch
  Int_t fNEtaSegments;//Number of eta slices
  Double_t fEtaMin;//Minimum allowed eta
  Double_t fEtaMax;//Maximum allowed eta
  Int_t fNumOfPadsToLook;//Padrow search window
  Int_t fNumOfRowsToMiss;//Maximum numbers of padrow which could be missed
  AliHLTHistogram1D **fEtaHistos; //!
  Float_t fZVertex;//Z position of the primary vertex

  //Flags
  Bool_t fRemoveFoundTracks;//Remove the found tracks or not?
  
  AliHLTHoughBaseTransformer *fHoughTransformer; //!
  AliHLTDigitRowData **fRowPointers; //!
  
  ClassDef(AliHLTHoughEval,1) //Hough transform verfication class

};

typedef AliHLTHoughEval AliL3HoughEval; // for backward comaptibility

#endif
Commit	Line	Data
4aa41877	1	// @(#) $Id$
	2
	3	#ifndef ALIL3HOUGHEVAL_H
	4	#define ALIL3HOUGHEVAL_H
	5
	6	#include "AliHLTRootTypes.h"
	7
	8	class AliHLTTrackArray;
	9	class AliHLTHoughBaseTransformer;
	10	class AliHLTHoughTrack;
	11	class AliHLTDigitRowData;
	12	class AliHLTHistogram;
	13	class AliHLTHistogram1D;
	14
	15	class AliHLTHoughEval {
	16
	17	public:
	18	AliHLTHoughEval();
	19	virtual ~AliHLTHoughEval();
	20
	21	void InitTransformer(AliHLTHoughBaseTransformer *transformer);
	22	void GenerateLUT();
	23	void DisplayEtaSlice(Int_t etaindex,AliHLTHistogram *hist);
	24	Bool_t LookInsideRoad(AliHLTHoughTrack track,Int_t &nrowscrossed,Int_t rowrange,Bool_t remove=kFALSE);
	25	#ifdef use_root
	26	void CompareMC(AliHLTTrackArray tracks,Char_t goodtracks="good_tracks",Int_t treshold=0);
	27	#endif
	28	void FindEta(AliHLTTrackArray *tracks);
	29
	30	//Getters
	31	AliHLTHistogram1D *GetEtaHisto(Int_t i) {if(!fEtaHistos) return 0; if(!fEtaHistos[i]) return 0; return fEtaHistos[i];}
	32
	33	//Setters:
	34	void RemoveFoundTracks() {fRemoveFoundTracks = kTRUE;}
	35	void SetNumOfRowsToMiss(Int_t i) {fNumOfRowsToMiss = i;}
	36	void SetNumOfPadsToLook(Int_t i) {fNumOfPadsToLook = i;}
	37	void SetSlice(Int_t i) {fSlice=i;}
	38	void SetZVertex(Float_t zvertex) {fZVertex=zvertex;}
	39
	40	private:
	41
	42	Int_t fSlice;//Index of the slice being processed
	43	Int_t fPatch;//Index of the patch being processed
	44	Int_t fNrows;//Number of padrows inside the patch
	45	Int_t fNEtaSegments;//Number of eta slices
	46	Double_t fEtaMin;//Minimum allowed eta
	47	Double_t fEtaMax;//Maximum allowed eta
	48	Int_t fNumOfPadsToLook;//Padrow search window
	49	Int_t fNumOfRowsToMiss;//Maximum numbers of padrow which could be missed
	50	AliHLTHistogram1D **fEtaHistos; //!
	51	Float_t fZVertex;//Z position of the primary vertex
	52
	53	//Flags
	54	Bool_t fRemoveFoundTracks;//Remove the found tracks or not?
	55
	56	AliHLTHoughBaseTransformer *fHoughTransformer; //!
	57	AliHLTDigitRowData **fRowPointers; //!
	58
	59	ClassDef(AliHLTHoughEval,1) //Hough transform verfication class
	60
	61	};
	62
	63	typedef AliHLTHoughEval AliL3HoughEval; // for backward comaptibility
	64
65	#endif