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

// @(#) $Id$

#ifndef ALIL3_MEMHANDLER_H
#define ALIL3_MEMHANDLER_H

#include "AliL3RootTypes.h"
#include "AliL3DigitData.h"

class AliL3SpacePointData;
class AliL3DigitRowData;
class AliL3TrackSegmentData;
class AliL3TrackArray;
class AliL3RandomPointData;

class AliL3MemHandler{
 private:
  
  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[159]; //for ROI in eta only
  Int_t fEtaMaxTimeBin[159];
  
  FILE *fInBinary;//!
  FILE *fOutBinary;//!
  
 public:
  AliL3MemHandler();
  virtual ~AliL3MemHandler();
  
  void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}  
  void Init(Int_t s,Int_t p, Int_t *r=0);

  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);

  virtual AliL3DigitRowData *CompBinary2Memory(UInt_t & nrow);
  virtual 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);
  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);
    
  //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;}
  FILE *GetFilePointer() {return fInBinary;}
  void   Free();
  
  //Getters:
  Int_t GetRowMin(){return fRowMin;}
  Int_t GetRowMax(){return fRowMax;}
  Int_t GetSlice(){return fSlice;}
  Int_t GetPatch(){return fPatch;}
  
  //virtual functions:
  virtual void FreeDigitsTree() {return;}
  virtual Bool_t SetAliInput(char *name){return 0;}
  virtual void CloseAliInput(){return;} 
  virtual Bool_t IsDigit(Int_t i=0){return 0;}
  virtual Bool_t SetMCOutput(char *name){return 0;}
  virtual Bool_t SetMCOutput(FILE *file){return 0;}
  virtual void CloseMCOutput(){return;}
  virtual Bool_t AliDigits2Binary(Int_t event=0,Bool_t altro=kFALSE){return 0;}
  virtual Bool_t AliDigits2CompBinary(Int_t event=0,Bool_t altro=kFALSE){return 0;}  
  virtual AliL3DigitRowData *AliDigits2Memory(UInt_t & nrow,Int_t event=0){return 0;}
  virtual AliL3DigitRowData *AliAltroDigits2Memory(UInt_t & nrow,Int_t event=0,Bool_t eventmerge=kFALSE){return 0;}
  virtual void AliDigits2RootFile(AliL3DigitRowData *rowPt,Char_t *new_digitsfile){return;}
  virtual Bool_t AliPoints2Binary(Int_t eventn=0){return 0;}
  virtual AliL3SpacePointData *AliPoints2Memory(UInt_t & npoint,Int_t eventn=0){return 0;}

  //AliL3RawDataFileHandler
  virtual Bool_t SetRawInput(Char_t *name){return 0;}
  virtual Bool_t SetRawInput(ifstream *file){return 0;}
  virtual void CloseRawInput(){} 
  virtual Int_t ReadRawInput(){return 0;}
  virtual Short_t** GetRawData(Int_t &channels, Int_t & timebins){return 0;}

  virtual Bool_t SetRawOutput(Char_t *name){return 0;}
  virtual Bool_t SetRawOutput(ofstream *file){return 0;}
  virtual void CloseRawOutput(){} 
  virtual Bool_t SaveRawOutput(){return 0;}

  virtual Bool_t SetMappingFile(Char_t *name){return 0;}
  virtual Bool_t SetMappingFile(FILE *file){return 0;}
  virtual void CloseMappingFile(){} 
  virtual Int_t ReadMappingFile(){return 0;}
  
  virtual Bool_t SetRawPedestalsInput(Char_t *name){return 0;}
  virtual Bool_t SetRawPedestalsInput(ifstream *file){return 0;}
  virtual void CloseRawPedestalsInput(){} 
  virtual Int_t ReadRawPedestalsInput(){return 0;}

  virtual AliL3DigitRowData* RawData2Memory(UInt_t &nrow,Int_t event=-1){return 0;}
  virtual Bool_t RawData2CompMemory(Int_t event=-1){return 0;}

  //AliL3DDLDataFileHandler
#ifdef use_newio
  virtual Bool_t SetReaderInput(Char_t *name,Int_t event=0){return 0;}
#else
  virtual Bool_t SetReaderInput(Char_t *name,Bool_t add=kTRUE){return 0;}
#endif
  virtual void CloseReaderInput(){};

  virtual AliL3DigitRowData* DDLData2Memory(UInt_t &nrow,Int_t event=-1){return 0;}
  virtual Bool_t DDLData2CompBinary(Int_t event=-1){return 0;}

  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
3e87ef69	1	// @(#) $Id$
3e87ef69	2
4499ed26	3	#ifndef ALIL3_MEMHANDLER_H
4499ed26	4	#define ALIL3_MEMHANDLER_H
108615fc	5
d2fd868a	6	#include "AliL3RootTypes.h"
108615fc	7	#include "AliL3DigitData.h"
494fad94	8
108615fc	9	class AliL3SpacePointData;
	10	class AliL3DigitRowData;
	11	class AliL3TrackSegmentData;
	12	class AliL3TrackArray;
	13	class AliL3RandomPointData;
	14
	15	class AliL3MemHandler{
	16	private:
b25c64e5	17
108615fc	18	Byte_t *fPt;//!
	19	UInt_t fSize;
	20
	21	AliL3RandomDigitData **fDPt;//!
	22	AliL3RandomDigitData *fDigits;//!
	23	Bool_t IsRandom;
	24	Int_t fNRandom;
	25	Int_t fNGenerate;
	26	Int_t fNUsed;
	27	Int_t fNDigits;
	28
	29	void Write(UInt_t *comp, UInt_t & index, UInt_t & subindex, UShort_t value);
	30	UShort_t Read(UInt_t *comp, UInt_t & index, UInt_t & subindex);
	31	UShort_t Test(UInt_t *comp, UInt_t index, UInt_t subindex);
	32
	33	void DigitizePoint(Int_t row,Int_t pad, Int_t time,Int_t charge);
	34	void QSort(AliL3RandomDigitData **a, Int_t first, Int_t last);
	35	Int_t ComparePoints(UInt_t row,UShort_t pad,UShort_t time);
	36	Int_t CompareDigits(AliL3RandomDigitData a,AliL3RandomDigitData b);
	37	void AddData(AliL3DigitData *data,UInt_t & ndata,
	38	UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
	39	void AddRandom(AliL3DigitData *data,UInt_t & ndata);
	40	void MergeDataRandom(AliL3DigitData *data,UInt_t & ndata,
	41	UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
	42	void AddDataRandom(AliL3DigitData *data,UInt_t & ndata,
	43	UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
	44
	45	protected:
	46	Int_t fRowMin;
	47	Int_t fRowMax;
	48	Int_t fSlice;
	49	Int_t fPatch;
	50
3e87ef69	51	Int_t fEtaMinTimeBin[159]; //for ROI in eta only
6c83647d	52	Int_t fEtaMaxTimeBin[159];
256c7399	53
	54	FILE *fInBinary;//!
	55	FILE *fOutBinary;//!
	56
108615fc	57	public:
	58	AliL3MemHandler();
	59	virtual ~AliL3MemHandler();
	60
	61	void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}
eeddc64d	62	void Init(Int_t s,Int_t p, Int_t *r=0);
108615fc	63
	64	Bool_t SetBinaryInput(char *name);
	65	Bool_t SetBinaryInput(FILE *file);
	66	void CloseBinaryInput();
	67
	68	Bool_t SetBinaryOutput(char *name);
	69	Bool_t SetBinaryOutput(FILE *file);
	70	void CloseBinaryOutput();
	71
	72	//Random cluster
	73	void SetRandomCluster(Int_t maxnumber);
	74	void SetRandomSeed(UInt_t seed){srand(seed);}
	75	void SetRandomSeed();
	76
	77	void ResetRandom(){fNDigits = 0; fNUsed = 0;}
	78	void Generate(Int_t row);
	79	void SetNGenerate(Int_t number){(number>fNRandom)?fNGenerate=fNRandom:fNGenerate = number;}
	80
	81	void SetROI(Float_t eta,Int_t slice);
	82	void ResetROI();
	83
	84	//Digit IO
	85	Bool_t Memory2Binary(UInt_t nrow,AliL3DigitRowData *data);
	86	Bool_t Binary2Memory(UInt_t & nrow,AliL3DigitRowData *data);
	87
	88	Int_t Memory2CompMemory(UInt_t nrow,AliL3DigitRowData data,UInt_t comp);
	89	Int_t CompMemory2Memory(UInt_t nrow,AliL3DigitRowData data,UInt_t comp);
	90	Bool_t CompMemory2CompBinary(UInt_t nrow,UInt_t *comp, UInt_t size=0);
	91	Bool_t CompBinary2CompMemory(UInt_t & nrow,UInt_t *comp);
	92
256c7399	93	virtual AliL3DigitRowData *CompBinary2Memory(UInt_t & nrow);
	94	virtual Bool_t Memory2CompBinary(UInt_t nrow,AliL3DigitRowData *data);
	95
108615fc	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);
c51c6aaa	101	Bool_t Transform(UInt_t npoint,AliL3SpacePointData *data,Int_t slice);
a830c5a6	102	static void UpdateRowPointer(AliL3DigitRowData *&tempPt);
9bd4d591	103
108615fc	104	//Track IO
	105	Bool_t Memory2Binary(UInt_t ntrack,AliL3TrackSegmentData *data);
	106	Bool_t Binary2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data);
	107	Bool_t TrackArray2Binary(AliL3TrackArray *array);
	108	Bool_t Binary2TrackArray(AliL3TrackArray *array);
240d63be	109	Bool_t TrackArray2Memory(UInt_t & ntrack,AliL3TrackSegmentData data,AliL3TrackArray array);
	110	Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData data,AliL3TrackArray array);
	111	Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData data,AliL3TrackArray array,Int_t slice);
4499ed26	112
108615fc	113	//Memory Allocation
	114	UInt_t GetAllocatedSize(){return fSize;}
	115	UInt_t GetFileSize();
	116	UInt_t GetMemorySize(UInt_t nrow,UInt_t *comp);
	117	UInt_t GetCompMemorySize(UInt_t nrow,AliL3DigitRowData *data);
	118	UInt_t GetRandomSize();
	119
	120	Byte_t *Allocate(UInt_t size);
	121	Byte_t *Allocate(); // allocate size of Binary Input File
	122	Byte_t Allocate(AliL3TrackArray array);
b661165c	123	Byte_t *GetDataPointer(UInt_t &size) {size = fSize; return fPt;}
3e87ef69	124	FILE *GetFilePointer() {return fInBinary;}
108615fc	125	void Free();
108615fc	126
f8bc0cc9	127	//Getters:
	128	Int_t GetRowMin(){return fRowMin;}
	129	Int_t GetRowMax(){return fRowMax;}
	130	Int_t GetSlice(){return fSlice;}
	131	Int_t GetPatch(){return fPatch;}
	132
	133	//virtual functions:
a815f9dd	134	virtual void FreeDigitsTree() {return;}
f8bc0cc9	135	virtual Bool_t SetAliInput(char *name){return 0;}
f8bc0cc9	136	virtual void CloseAliInput(){return;}
3e87ef69	137	virtual Bool_t IsDigit(Int_t i=0){return 0;}
f8bc0cc9	138	virtual Bool_t SetMCOutput(char *name){return 0;}
	139	virtual Bool_t SetMCOutput(FILE *file){return 0;}
	140	virtual void CloseMCOutput(){return;}
3e87ef69	141	virtual Bool_t AliDigits2Binary(Int_t event=0,Bool_t altro=kFALSE){return 0;}
240d63be	142	virtual Bool_t AliDigits2CompBinary(Int_t event=0,Bool_t altro=kFALSE){return 0;}
f8bc0cc9	143	virtual AliL3DigitRowData *AliDigits2Memory(UInt_t & nrow,Int_t event=0){return 0;}
3e87ef69	144	virtual AliL3DigitRowData *AliAltroDigits2Memory(UInt_t & nrow,Int_t event=0,Bool_t eventmerge=kFALSE){return 0;}
f8bc0cc9	145	virtual void AliDigits2RootFile(AliL3DigitRowData rowPt,Char_t new_digitsfile){return;}
3e87ef69	146	virtual Bool_t AliPoints2Binary(Int_t eventn=0){return 0;}
	147	virtual AliL3SpacePointData *AliPoints2Memory(UInt_t & npoint,Int_t eventn=0){return 0;}
	148
	149	//AliL3RawDataFileHandler
	150	virtual Bool_t SetRawInput(Char_t *name){return 0;}
	151	virtual Bool_t SetRawInput(ifstream *file){return 0;}
	152	virtual void CloseRawInput(){}
	153	virtual Int_t ReadRawInput(){return 0;}
	154	virtual Short_t** GetRawData(Int_t &channels, Int_t & timebins){return 0;}
	155
	156	virtual Bool_t SetRawOutput(Char_t *name){return 0;}
	157	virtual Bool_t SetRawOutput(ofstream *file){return 0;}
	158	virtual void CloseRawOutput(){}
	159	virtual Bool_t SaveRawOutput(){return 0;}
	160
	161	virtual Bool_t SetMappingFile(Char_t *name){return 0;}
	162	virtual Bool_t SetMappingFile(FILE *file){return 0;}
	163	virtual void CloseMappingFile(){}
	164	virtual Int_t ReadMappingFile(){return 0;}
	165
	166	virtual Bool_t SetRawPedestalsInput(Char_t *name){return 0;}
	167	virtual Bool_t SetRawPedestalsInput(ifstream *file){return 0;}
	168	virtual void CloseRawPedestalsInput(){}
	169	virtual Int_t ReadRawPedestalsInput(){return 0;}
	170
	171	virtual AliL3DigitRowData* RawData2Memory(UInt_t &nrow,Int_t event=-1){return 0;}
240d63be	172	virtual Bool_t RawData2CompMemory(Int_t event=-1){return 0;}
	173
	174	//AliL3DDLDataFileHandler
b2a02bce	175	#ifdef use_newio
a27af97b	176	virtual Bool_t SetReaderInput(Char_t *name,Int_t event=0){return 0;}
b2a02bce	177	#else
	178	virtual Bool_t SetReaderInput(Char_t *name,Bool_t add=kTRUE){return 0;}
	179	#endif
240d63be	180	virtual void CloseReaderInput(){};
	181
	182	virtual AliL3DigitRowData* DDLData2Memory(UInt_t &nrow,Int_t event=-1){return 0;}
	183	virtual Bool_t DDLData2CompBinary(Int_t event=-1){return 0;}
f8bc0cc9	184
b661165c	185	ClassDef(AliL3MemHandler,1) // Memory handler class
108615fc	186	};
	187
	188	inline Int_t AliL3MemHandler::ComparePoints(UInt_t row,UShort_t pad,UShort_t time){
	189	if(fNUsed>=fNDigits) return -2;
	190
	191	if(pad==fDPt[fNUsed]->fPad&&time==fDPt[fNUsed]->fTime) return 0;
	192
	193	if(pad<fDPt[fNUsed]->fPad) return -1;
	194	if(pad==fDPt[fNUsed]->fPad&&time<fDPt[fNUsed]->fTime) return -1;
	195
	196	return 1;
	197	}
	198
	199	inline Int_t AliL3MemHandler::CompareDigits(AliL3RandomDigitData a,AliL3RandomDigitData b){
	200	if(a->fPad==b->fPad && a->fTime == b->fTime) return 0;
	201
	202	if(a->fPad<b->fPad) return -1;
	203	if(a->fPad==b->fPad && a->fTime<b->fTime) return -1;
	204
	205	return 1;
	206	}
	207
	208	#endif