[u/mrichter/AliRoot.git] / HLT / src / AliL3MemHandler.h

#ifndef ALIL3MEMHANDLER_H
#define ALIL3MEMHANDLER_H

#include "AliL3RootTypes.h"
#include <stdio.h>
#include <stdlib.h>
#include "AliL3DigitData.h"
#include "AliL3Transform.h"
class AliL3SpacePointData;
class AliL3DigitRowData;
class AliL3TrackSegmentData;
class AliL3TrackArray;
class AliL3RandomPointData;

class AliL3MemHandler{
 private:
  AliL3Transform *fTransformer;//!
  FILE *fInBinary;//!
  FILE *fOutBinary;//!
  Byte_t *fPt;//!
  UInt_t fSize;

  AliL3RandomDigitData **fDPt;//!
  AliL3RandomDigitData *fDigits;//!
  Bool_t IsRandom;
  Int_t fNRandom;
  Int_t fNGenerate;
  Int_t fNUsed;
  Int_t fNDigits;

  void Write(UInt_t *comp, UInt_t & index, UInt_t & subindex, UShort_t value);
  UShort_t Read(UInt_t *comp, UInt_t & index, UInt_t & subindex);
  UShort_t Test(UInt_t *comp, UInt_t index, UInt_t subindex); 
  
  void DigitizePoint(Int_t row,Int_t pad, Int_t time,Int_t charge);
  void QSort(AliL3RandomDigitData **a, Int_t first, Int_t last);
  Int_t ComparePoints(UInt_t row,UShort_t pad,UShort_t time);
  Int_t CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b);
  void AddData(AliL3DigitData *data,UInt_t & ndata,
                      UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
  void AddRandom(AliL3DigitData *data,UInt_t & ndata);
  void MergeDataRandom(AliL3DigitData *data,UInt_t & ndata,
                      UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
  void AddDataRandom(AliL3DigitData *data,UInt_t & ndata,
                      UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);

 protected:
  Int_t fRowMin;
  Int_t fRowMax;
  Int_t fSlice;
  Int_t fPatch;

  Int_t fEtaMinTimeBin[176];
  Int_t fEtaMaxTimeBin[176];
  

 public:
  AliL3MemHandler();
  virtual ~AliL3MemHandler();
  
  void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}  
  void Init(AliL3Transform *t){fTransformer = t;}
  void Init(Int_t s,Int_t p,const Int_t* row){fSlice=s;fPatch=p;fRowMin=row[0];fRowMax=row[1]; ResetROI();}

  Bool_t SetBinaryInput(char *name);
  Bool_t SetBinaryInput(FILE *file);
  void CloseBinaryInput();
  
  Bool_t SetBinaryOutput(char *name);
  Bool_t SetBinaryOutput(FILE *file);
  void CloseBinaryOutput();

  //Random cluster
  void SetRandomCluster(Int_t maxnumber);
  void SetRandomSeed(UInt_t seed){srand(seed);}
  void SetRandomSeed();

  void ResetRandom(){fNDigits = 0; fNUsed = 0;}
  void Generate(Int_t row);
  void SetNGenerate(Int_t number){(number>fNRandom)?fNGenerate=fNRandom:fNGenerate = number;}

  void SetROI(Float_t *eta,Int_t *slice);
  void ResetROI();

  //Digit IO
  Bool_t Memory2Binary(UInt_t nrow,AliL3DigitRowData *data);
  Bool_t Binary2Memory(UInt_t & nrow,AliL3DigitRowData *data);

  Int_t Memory2CompMemory(UInt_t nrow,AliL3DigitRowData *data,UInt_t *comp);
  Int_t CompMemory2Memory(UInt_t nrow,AliL3DigitRowData *data,UInt_t *comp);
  Bool_t CompMemory2CompBinary(UInt_t nrow,UInt_t *comp, UInt_t size=0);
  Bool_t CompBinary2CompMemory(UInt_t & nrow,UInt_t *comp);

  AliL3DigitRowData *CompBinary2Memory(UInt_t & nrow);
  Bool_t Memory2CompBinary(UInt_t nrow,AliL3DigitRowData *data);
  UInt_t GetNRow(UInt_t *comp,UInt_t size);

  //Point IO
  Bool_t Memory2Binary(UInt_t npoint,AliL3SpacePointData *data);
  Bool_t Binary2Memory(UInt_t & npoint,AliL3SpacePointData *data);
  Bool_t Transform(UInt_t npoint,AliL3SpacePointData *data,
                              Int_t slice, AliL3Transform* trans);
  static void UpdateRowPointer(AliL3DigitRowData *&tempPt);
  
  //Track IO
  Bool_t Memory2Binary(UInt_t ntrack,AliL3TrackSegmentData *data);
  Bool_t Binary2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data);
  Bool_t TrackArray2Binary(AliL3TrackArray *array);
  Bool_t Binary2TrackArray(AliL3TrackArray *array);
  Bool_t TrackArray2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data,
                                             AliL3TrackArray *array);
  Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data,
                                             AliL3TrackArray *array);
  Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data,
			   AliL3TrackArray *array,Int_t slice, AliL3Transform* trans);
  
  
  //Memory Allocation
  UInt_t GetAllocatedSize(){return fSize;}  
  UInt_t GetFileSize();
  UInt_t GetMemorySize(UInt_t nrow,UInt_t *comp);
  UInt_t GetCompMemorySize(UInt_t nrow,AliL3DigitRowData *data);
  UInt_t GetRandomSize();

  Byte_t *Allocate(UInt_t size);
  Byte_t *Allocate();  // allocate size of Binary Input File
  Byte_t *Allocate(AliL3TrackArray *array);
  Byte_t *GetDataPointer(UInt_t &size) {size = fSize; return fPt;}
  void   Free();
  
  ClassDef(AliL3MemHandler,1) // Memory handler class
};

inline Int_t  AliL3MemHandler::ComparePoints(UInt_t row,UShort_t pad,UShort_t time){
  if(fNUsed>=fNDigits) return -2;

  if(pad==fDPt[fNUsed]->fPad&&time==fDPt[fNUsed]->fTime) return 0;

  if(pad<fDPt[fNUsed]->fPad) return -1;
  if(pad==fDPt[fNUsed]->fPad&&time<fDPt[fNUsed]->fTime)  return -1;

  return 1;
}

inline Int_t AliL3MemHandler::CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b){
  if(a->fPad==b->fPad && a->fTime == b->fTime) return 0;

  if(a->fPad<b->fPad) return -1;
  if(a->fPad==b->fPad && a->fTime<b->fTime) return -1;
  
  return 1;
}

#endif
Commit	Line	Data
108615fc	1	#ifndef ALIL3MEMHANDLER_H
	2	#define ALIL3MEMHANDLER_H
	3
	4	#include "AliL3RootTypes.h"
	5	#include <stdio.h>
	6	#include <stdlib.h>
	7	#include "AliL3DigitData.h"
	8	#include "AliL3Transform.h"
	9	class AliL3SpacePointData;
	10	class AliL3DigitRowData;
	11	class AliL3TrackSegmentData;
	12	class AliL3TrackArray;
	13	class AliL3RandomPointData;
	14
	15	class AliL3MemHandler{
	16	private:
	17	AliL3Transform *fTransformer;//!
	18	FILE *fInBinary;//!
	19	FILE *fOutBinary;//!
	20	Byte_t *fPt;//!
	21	UInt_t fSize;
	22
	23	AliL3RandomDigitData **fDPt;//!
	24	AliL3RandomDigitData *fDigits;//!
	25	Bool_t IsRandom;
	26	Int_t fNRandom;
	27	Int_t fNGenerate;
	28	Int_t fNUsed;
	29	Int_t fNDigits;
	30
	31	void Write(UInt_t *comp, UInt_t & index, UInt_t & subindex, UShort_t value);
	32	UShort_t Read(UInt_t *comp, UInt_t & index, UInt_t & subindex);
	33	UShort_t Test(UInt_t *comp, UInt_t index, UInt_t subindex);
	34
	35	void DigitizePoint(Int_t row,Int_t pad, Int_t time,Int_t charge);
	36	void QSort(AliL3RandomDigitData **a, Int_t first, Int_t last);
	37	Int_t ComparePoints(UInt_t row,UShort_t pad,UShort_t time);
	38	Int_t CompareDigits(AliL3RandomDigitData a,AliL3RandomDigitData b);
	39	void AddData(AliL3DigitData *data,UInt_t & ndata,
	40	UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
	41	void AddRandom(AliL3DigitData *data,UInt_t & ndata);
	42	void MergeDataRandom(AliL3DigitData *data,UInt_t & ndata,
	43	UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
	44	void AddDataRandom(AliL3DigitData *data,UInt_t & ndata,
	45	UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
	46
	47	protected:
	48	Int_t fRowMin;
	49	Int_t fRowMax;
	50	Int_t fSlice;
	51	Int_t fPatch;
	52
ae97a0b9	53	Int_t fEtaMinTimeBin[176];
ae97a0b9	54	Int_t fEtaMaxTimeBin[176];
108615fc	55
	56
	57	public:
	58	AliL3MemHandler();
	59	virtual ~AliL3MemHandler();
	60
	61	void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}
	62	void Init(AliL3Transform *t){fTransformer = t;}
9bd4d591	63	void Init(Int_t s,Int_t p,const Int_t* row){fSlice=s;fPatch=p;fRowMin=row[0];fRowMax=row[1]; ResetROI();}
108615fc	64
	65	Bool_t SetBinaryInput(char *name);
	66	Bool_t SetBinaryInput(FILE *file);
	67	void CloseBinaryInput();
	68
	69	Bool_t SetBinaryOutput(char *name);
	70	Bool_t SetBinaryOutput(FILE *file);
	71	void CloseBinaryOutput();
	72
	73	//Random cluster
	74	void SetRandomCluster(Int_t maxnumber);
	75	void SetRandomSeed(UInt_t seed){srand(seed);}
	76	void SetRandomSeed();
	77
	78	void ResetRandom(){fNDigits = 0; fNUsed = 0;}
	79	void Generate(Int_t row);
	80	void SetNGenerate(Int_t number){(number>fNRandom)?fNGenerate=fNRandom:fNGenerate = number;}
	81
	82	void SetROI(Float_t eta,Int_t slice);
	83	void ResetROI();
	84
	85	//Digit IO
	86	Bool_t Memory2Binary(UInt_t nrow,AliL3DigitRowData *data);
	87	Bool_t Binary2Memory(UInt_t & nrow,AliL3DigitRowData *data);
	88
	89	Int_t Memory2CompMemory(UInt_t nrow,AliL3DigitRowData data,UInt_t comp);
	90	Int_t CompMemory2Memory(UInt_t nrow,AliL3DigitRowData data,UInt_t comp);
	91	Bool_t CompMemory2CompBinary(UInt_t nrow,UInt_t *comp, UInt_t size=0);
	92	Bool_t CompBinary2CompMemory(UInt_t & nrow,UInt_t *comp);
	93
	94	AliL3DigitRowData *CompBinary2Memory(UInt_t & nrow);
	95	Bool_t Memory2CompBinary(UInt_t nrow,AliL3DigitRowData *data);
	96	UInt_t GetNRow(UInt_t *comp,UInt_t size);
	97
	98	//Point IO
	99	Bool_t Memory2Binary(UInt_t npoint,AliL3SpacePointData *data);
	100	Bool_t Binary2Memory(UInt_t & npoint,AliL3SpacePointData *data);
	101	Bool_t Transform(UInt_t npoint,AliL3SpacePointData *data,
	102	Int_t slice, AliL3Transform* trans);
a830c5a6	103	static void UpdateRowPointer(AliL3DigitRowData *&tempPt);
9bd4d591	104
108615fc	105	//Track IO
	106	Bool_t Memory2Binary(UInt_t ntrack,AliL3TrackSegmentData *data);
	107	Bool_t Binary2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data);
	108	Bool_t TrackArray2Binary(AliL3TrackArray *array);
	109	Bool_t Binary2TrackArray(AliL3TrackArray *array);
	110	Bool_t TrackArray2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data,
	111	AliL3TrackArray *array);
	112	Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data,
	113	AliL3TrackArray *array);
	114	Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data,
9bd4d591	115	AliL3TrackArray array,Int_t slice, AliL3Transform trans);
	116
	117
108615fc	118	//Memory Allocation
	119	UInt_t GetAllocatedSize(){return fSize;}
	120	UInt_t GetFileSize();
	121	UInt_t GetMemorySize(UInt_t nrow,UInt_t *comp);
	122	UInt_t GetCompMemorySize(UInt_t nrow,AliL3DigitRowData *data);
	123	UInt_t GetRandomSize();
	124
	125	Byte_t *Allocate(UInt_t size);
	126	Byte_t *Allocate(); // allocate size of Binary Input File
	127	Byte_t Allocate(AliL3TrackArray array);
b661165c	128	Byte_t *GetDataPointer(UInt_t &size) {size = fSize; return fPt;}
108615fc	129	void Free();
108615fc	130
b661165c	131	ClassDef(AliL3MemHandler,1) // Memory handler class
108615fc	132	};
	133
	134	inline Int_t AliL3MemHandler::ComparePoints(UInt_t row,UShort_t pad,UShort_t time){
	135	if(fNUsed>=fNDigits) return -2;
	136
	137	if(pad==fDPt[fNUsed]->fPad&&time==fDPt[fNUsed]->fTime) return 0;
	138
	139	if(pad<fDPt[fNUsed]->fPad) return -1;
	140	if(pad==fDPt[fNUsed]->fPad&&time<fDPt[fNUsed]->fTime) return -1;
	141
	142	return 1;
	143	}
	144
	145	inline Int_t AliL3MemHandler::CompareDigits(AliL3RandomDigitData a,AliL3RandomDigitData b){
	146	if(a->fPad==b->fPad && a->fTime == b->fTime) return 0;
	147
	148	if(a->fPad<b->fPad) return -1;
	149	if(a->fPad==b->fPad && a->fTime<b->fTime) return -1;
	150
	151	return 1;
	152	}
	153
	154	#endif