]> git.uio.no Git - u/mrichter/AliRoot.git/blob - HLT/src/AliL3MemHandler.h
Bug correction (energy recalculation when adding new incarnation of the particle).
[u/mrichter/AliRoot.git] / HLT / src / AliL3MemHandler.h
1 // @(#) $Id$
2
3 #ifndef ALIL3_MEMHANDLER_H
4 #define ALIL3_MEMHANDLER_H
5
6 #include "AliL3RootTypes.h"
7 #include "AliL3DigitData.h"
8
9 class AliL3SpacePointData;
10 class AliL3DigitRowData;
11 class AliL3TrackSegmentData;
12 class AliL3TrackArray;
13 class AliL3RandomPointData;
14
15 class AliL3MemHandler{
16  private:
17   
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
51   Int_t fEtaMinTimeBin[159]; //for ROI in eta only
52   Int_t fEtaMaxTimeBin[159];
53   
54   FILE *fInBinary;//!
55   FILE *fOutBinary;//!
56   
57  public:
58   AliL3MemHandler();
59   virtual ~AliL3MemHandler();
60   
61   void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}  
62   void Init(Int_t s,Int_t p, Int_t *r=0);
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
93   virtual AliL3DigitRowData *CompBinary2Memory(UInt_t & nrow);
94   virtual Bool_t Memory2CompBinary(UInt_t nrow,AliL3DigitRowData *data);
95   
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,Int_t slice);
102   static void UpdateRowPointer(AliL3DigitRowData *&tempPt);
103   
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);
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);
112     
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);
123   Byte_t *GetDataPointer(UInt_t &size) {size = fSize; return fPt;}
124   FILE *GetFilePointer() {return fInBinary;}
125   void   Free();
126   
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:
134   virtual void FreeDigitsTree() {return;}
135   virtual Bool_t SetAliInput(char *name){return 0;}
136   virtual void CloseAliInput(){return;} 
137   virtual Bool_t IsDigit(Int_t i=0){return 0;}
138   virtual Bool_t SetMCOutput(char *name){return 0;}
139   virtual Bool_t SetMCOutput(FILE *file){return 0;}
140   virtual void CloseMCOutput(){return;}
141   virtual Bool_t AliDigits2Binary(Int_t event=0,Bool_t altro=kFALSE){return 0;}
142   virtual Bool_t AliDigits2CompBinary(Int_t event=0,Bool_t altro=kFALSE){return 0;}  
143   virtual AliL3DigitRowData *AliDigits2Memory(UInt_t & nrow,Int_t event=0){return 0;}
144   virtual AliL3DigitRowData *AliAltroDigits2Memory(UInt_t & nrow,Int_t event=0,Bool_t eventmerge=kFALSE){return 0;}
145   virtual void AliDigits2RootFile(AliL3DigitRowData *rowPt,Char_t *new_digitsfile){return;}
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;}
172   virtual Bool_t RawData2CompMemory(Int_t event=-1){return 0;}
173
174   //AliL3DDLDataFileHandler
175   virtual Bool_t SetReaderInput(Char_t *name,Bool_t add=kTRUE){return 0;}
176   virtual void CloseReaderInput(){};
177
178   virtual AliL3DigitRowData* DDLData2Memory(UInt_t &nrow,Int_t event=-1){return 0;}
179   virtual Bool_t DDLData2CompBinary(Int_t event=-1){return 0;}
180
181   ClassDef(AliL3MemHandler,1) // Memory handler class
182 };
183
184 inline Int_t  AliL3MemHandler::ComparePoints(UInt_t row,UShort_t pad,UShort_t time){
185   if(fNUsed>=fNDigits) return -2;
186
187   if(pad==fDPt[fNUsed]->fPad&&time==fDPt[fNUsed]->fTime) return 0;
188
189   if(pad<fDPt[fNUsed]->fPad) return -1;
190   if(pad==fDPt[fNUsed]->fPad&&time<fDPt[fNUsed]->fTime)  return -1;
191
192   return 1;
193 }
194
195 inline Int_t AliL3MemHandler::CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b){
196   if(a->fPad==b->fPad && a->fTime == b->fTime) return 0;
197
198   if(a->fPad<b->fPad) return -1;
199   if(a->fPad==b->fPad && a->fTime<b->fTime) return -1;
200   
201   return 1;
202 }
203
204 #endif