[u/mrichter/AliRoot.git] / HLT / kalman / AliHLTKalman.h

// @(#) $Id$

#ifndef ALIL3_KALMAN
#define ALIL3_KALMAN


class AliHLTSpacePointData;
class AliHLTTrackArray;
class AliHLTBenchmark;
class AliHLTTrackSegmentData;
class AliHLTKalmanTrack;
class AliHLTTrack;

class AliHLTKalman {

 private:

  Int_t fMinSlice;
  Int_t fMaxSlice;
  AliHLTSpacePointData *fClusters[36][6];
  Char_t fPath[1024];
  UInt_t fNcl[36][6];
  AliHLTTrackArray *fTracks;
  AliHLTTrackArray *fKalmanTracks;
  AliHLTTrackArray *fSeeds;

  AliHLTBenchmark *fBenchmark;
  Int_t fMinPointsOnTrack;
  Int_t fRow[6][2];
  Char_t fWriteOutPath[256];
  Bool_t fWriteOut;
  Int_t fEvent;

 public:

  AliHLTKalman(Char_t *datapath, Int_t *slice, Int_t min_clusters);
  virtual ~AliHLTKalman();
  void Init();
  void LoadTracks(Int_t event, Bool_t sp);
  void ProcessTracks();
  Int_t MakeKalmanSeed(AliHLTKalmanTrack *kalmantrack, AliHLTTrack *track);
  Int_t InitKalmanTrack(AliHLTKalmanTrack *kalmantrack, AliHLTTrack *track);
  Int_t Propagate(AliHLTKalmanTrack *kalmantrack, AliHLTTrack *track);
  Int_t Update(AliHLTSpacePointData *points, UInt_t pos, AliHLTKalmanTrack *kalmantrack);
  void WriteFiles(Char_t *path="data"){fWriteOut = kTRUE; sprintf(fWriteOutPath,"%s",path);}
  Double_t GetCpuTime();
  AliHLTTrackArray *GetTracks() {return fKalmanTracks;}
};

typedef AliHLTKalman AliL3Kalman; // for backward compatibility

#endif
Commit	Line	Data
4aa41877	1	// @(#) $Id$
	2
	3	#ifndef ALIL3_KALMAN
	4	#define ALIL3_KALMAN
	5
	6
	7	class AliHLTSpacePointData;
	8	class AliHLTTrackArray;
	9	class AliHLTBenchmark;
	10	class AliHLTTrackSegmentData;
	11	class AliHLTKalmanTrack;
	12	class AliHLTTrack;
	13
	14	class AliHLTKalman {
	15
	16	private:
	17
	18	Int_t fMinSlice;
	19	Int_t fMaxSlice;
	20	AliHLTSpacePointData *fClusters[36][6];
	21	Char_t fPath[1024];
	22	UInt_t fNcl[36][6];
	23	AliHLTTrackArray *fTracks;
	24	AliHLTTrackArray *fKalmanTracks;
	25	AliHLTTrackArray *fSeeds;
	26
	27	AliHLTBenchmark *fBenchmark;
	28	Int_t fMinPointsOnTrack;
	29	Int_t fRow[6][2];
	30	Char_t fWriteOutPath[256];
	31	Bool_t fWriteOut;
	32	Int_t fEvent;
	33
	34	public:
	35
	36	AliHLTKalman(Char_t datapath, Int_t slice, Int_t min_clusters);
	37	virtual ~AliHLTKalman();
	38	void Init();
	39	void LoadTracks(Int_t event, Bool_t sp);
	40	void ProcessTracks();
	41	Int_t MakeKalmanSeed(AliHLTKalmanTrack kalmantrack, AliHLTTrack track);
	42	Int_t InitKalmanTrack(AliHLTKalmanTrack kalmantrack, AliHLTTrack track);
	43	Int_t Propagate(AliHLTKalmanTrack kalmantrack, AliHLTTrack track);
	44	Int_t Update(AliHLTSpacePointData points, UInt_t pos, AliHLTKalmanTrack kalmantrack);
	45	void WriteFiles(Char_t *path="data"){fWriteOut = kTRUE; sprintf(fWriteOutPath,"%s",path);}
	46	Double_t GetCpuTime();
	47	AliHLTTrackArray *GetTracks() {return fKalmanTracks;}
	48	};
	49
	50	typedef AliHLTKalman AliL3Kalman; // for backward compatibility
	51
	52	#endif