[u/mrichter/AliRoot.git] / HLT / TPCLib / Ref / AliHLTTPCMerger.h

// @(#) $Id$
#ifndef ALIHLTTPCMERGER_H
#define ALIHLTTPCMERGER_H
#define PI 3.14159265358979312

#include "AliHLTTPCRootTypes.h"

class AliHLTTPCTrack;
class AliHLTTPCTrackSegmentData;
class AliHLTTPCVertex;
class AliHLTTPCTrackArray;

class AliHLTTPCMerger {
 private:
  Double_t fMaxY;
  Double_t fMaxZ;
  Double_t fMaxKappa;
  Double_t fMaxPsi;
  Double_t fMaxTgl;
  void SetArray(Int_t nin);
  void DeleteArray();
  Char_t fTrackType;
  
  AliHLTTPCTrackArray **fInTrack;//!
  Int_t fNIn;

  AliHLTTPCTrackArray *fOutTrack;//!

 protected:
  Int_t fCurrentTracks;
  Int_t fSlice;
  AliHLTTPCVertex *fVertex;//!
  Bool_t f2Global;
  Bool_t Is2Global(Bool_t is){f2Global=is;return f2Global;}
  void InitMerger(Int_t ntrackarrays,Char_t *tracktype="AliHLTTPCTrack");
  
 public:
  AliHLTTPCMerger();
  virtual ~AliHLTTPCMerger();

  Int_t GetNIn(){return fNIn;}
  AliHLTTPCTrackArray *GetInTracks(Int_t in){return fInTrack[in];}
  AliHLTTPCTrackArray *GetOutTracks(){return fOutTrack;}

  Bool_t Is2Global(){return f2Global;}
  void SetVertex(AliHLTTPCVertex *vertex){fVertex=vertex;}
  void Reset();
  void SetParameter(Double_t maxy=1., Double_t maxz=1., Double_t maxkappa=0.001, Double_t maxpsi=0.05, Double_t maxtgl=0.1);
  void FillTracks(Int_t ntracks, AliHLTTPCTrackSegmentData* tr); //Fill tracks in fTrackArray[fCurrentTracks] 
  Double_t GetAngle(Double_t a1,Double_t a2);
  void* GetNtuple();
  void* GetNtuple(char *varlist);
  Bool_t WriteNtuple(char *filename,void* nt);
  void FillNtuple(void* nt,Float_t *data);
  void FillNtuple(void* nt,AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack);
  void AddAllTracks();//Copy all Tracks to Output Array
  void SortGlobalTracks(AliHLTTPCTrack **tracks, Int_t ntrack);
  virtual void SortTracks(AliHLTTPCTrack **tracks, Int_t ntrack);
  virtual void AddTrack(AliHLTTPCTrackArray *mergedtrack,AliHLTTPCTrack *track);
  virtual AliHLTTPCTrack * MultiMerge(AliHLTTPCTrackArray *mergedtrack,AliHLTTPCTrack **tracks, Int_t ntrack);
  AliHLTTPCTrack * MergeTracks(AliHLTTPCTrackArray *mergedtrack,AliHLTTPCTrack *t0,AliHLTTPCTrack *t1);
  virtual Bool_t IsTrack(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack);
  Bool_t IsRTrack(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack);
  Double_t TrackDiff(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack);
  void Print();
  void PrintDiff(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack);
  void PrintTrack(AliHLTTPCTrack *track);
//  Int_t WriteTracks(Char_t *file);
//  Int_t WriteInTracks(Char_t *file);
//  Int_t WriteAllTracks(Char_t *file);
  
  ClassDef(AliHLTTPCMerger,1) //Merging base class
};

#endif
Commit	Line	Data
78001a73	1	// @(#) $Id$
	2	#ifndef ALIHLTTPCMERGER_H
	3	#define ALIHLTTPCMERGER_H
	4	#define PI 3.14159265358979312
	5
	6	#include "AliHLTTPCRootTypes.h"
	7
	8	class AliHLTTPCTrack;
	9	class AliHLTTPCTrackSegmentData;
	10	class AliHLTTPCVertex;
	11	class AliHLTTPCTrackArray;
	12
	13	class AliHLTTPCMerger {
	14	private:
	15	Double_t fMaxY;
	16	Double_t fMaxZ;
	17	Double_t fMaxKappa;
	18	Double_t fMaxPsi;
	19	Double_t fMaxTgl;
	20	void SetArray(Int_t nin);
	21	void DeleteArray();
	22	Char_t fTrackType;
	23
	24	AliHLTTPCTrackArray **fInTrack;//!
	25	Int_t fNIn;
	26
	27	AliHLTTPCTrackArray *fOutTrack;//!
	28
	29	protected:
	30	Int_t fCurrentTracks;
	31	Int_t fSlice;
	32	AliHLTTPCVertex *fVertex;//!
	33	Bool_t f2Global;
	34	Bool_t Is2Global(Bool_t is){f2Global=is;return f2Global;}
	35	void InitMerger(Int_t ntrackarrays,Char_t *tracktype="AliHLTTPCTrack");
	36
	37	public:
	38	AliHLTTPCMerger();
	39	virtual ~AliHLTTPCMerger();
	40
	41	Int_t GetNIn(){return fNIn;}
	42	AliHLTTPCTrackArray *GetInTracks(Int_t in){return fInTrack[in];}
	43	AliHLTTPCTrackArray *GetOutTracks(){return fOutTrack;}
	44
	45	Bool_t Is2Global(){return f2Global;}
	46	void SetVertex(AliHLTTPCVertex *vertex){fVertex=vertex;}
	47	void Reset();
	48	void SetParameter(Double_t maxy=1., Double_t maxz=1., Double_t maxkappa=0.001, Double_t maxpsi=0.05, Double_t maxtgl=0.1);
	49	void FillTracks(Int_t ntracks, AliHLTTPCTrackSegmentData* tr); //Fill tracks in fTrackArray[fCurrentTracks]
	50	Double_t GetAngle(Double_t a1,Double_t a2);
	51	void* GetNtuple();
	52	void* GetNtuple(char *varlist);
	53	Bool_t WriteNtuple(char filename,void nt);
	54	void FillNtuple(void* nt,Float_t *data);
	55	void FillNtuple(void* nt,AliHLTTPCTrack innertrack,AliHLTTPCTrack outertrack);
	56	void AddAllTracks();//Copy all Tracks to Output Array
	57	void SortGlobalTracks(AliHLTTPCTrack **tracks, Int_t ntrack);
	58	virtual void SortTracks(AliHLTTPCTrack **tracks, Int_t ntrack);
	59	virtual void AddTrack(AliHLTTPCTrackArray mergedtrack,AliHLTTPCTrack track);
	60	virtual AliHLTTPCTrack * MultiMerge(AliHLTTPCTrackArray mergedtrack,AliHLTTPCTrack *tracks, Int_t ntrack);
	61	AliHLTTPCTrack * MergeTracks(AliHLTTPCTrackArray mergedtrack,AliHLTTPCTrack t0,AliHLTTPCTrack *t1);
	62	virtual Bool_t IsTrack(AliHLTTPCTrack innertrack,AliHLTTPCTrack outertrack);
	63	Bool_t IsRTrack(AliHLTTPCTrack innertrack,AliHLTTPCTrack outertrack);
	64	Double_t TrackDiff(AliHLTTPCTrack innertrack,AliHLTTPCTrack outertrack);
65	void Print();
66	void PrintDiff(AliHLTTPCTrack innertrack,AliHLTTPCTrack outertrack);
67	void PrintTrack(AliHLTTPCTrack *track);
68	// Int_t WriteTracks(Char_t *file);
69	// Int_t WriteInTracks(Char_t *file);
70	// Int_t WriteAllTracks(Char_t *file);
71
72	ClassDef(AliHLTTPCMerger,1) //Merging base class
73	};
74
75	#endif