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

#ifndef ALIL3MEMHANDLER_H
#define ALIL3MEMHANDLER_H

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