- use_newio switch removed from libAliHLTTPC, always on; not removed in
[u/mrichter/AliRoot.git] / HLT / TPCLib / AliHLTTPCMemHandler.h
1 // @(#) $Id$
2 // Original: AliHLTMemHandler.h,v 1.30 2004/10/06 08:51:20 cvetan 
3 #ifndef ALIHLTTPC_MEMHANDLER_H
4 #define ALIHLTTPC_MEMHANDLER_H
5
6 //_____________________________________________________________
7 // AliHLTTPCMemHandler
8 //
9 // The HLT Binary File handler 
10 //
11 //  This class does all the memory I/O handling of HLT binary files.
12 //  
13 // Author: Uli Frankenfeld <mailto:franken@fi.uib.no>, 
14 //         Anders Vestbo <mailto:vestbo$fi.uib.no>, 
15 //         Constantin Loizides <mailto:loizides@ikf.uni-frankfurt.de>
16 // *-- Copyright &copy ALICE HLT Group 
17
18 class AliHLTTPCDigitData;
19 class AliHLTTPCSpacePointData;
20 class AliHLTTPCDigitRowData;
21 class AliHLTTPCTrackSegmentData;
22 class AliHLTTPCTrackArray;
23 class AliHLTTPCRandomPointData;
24 class AliHLTTPCRandomDigitData;
25
26 class AliRunLoader;
27 class AliRawEvent;
28 class AliTPCRawStream;
29
30 class AliHLTTPCMemHandler { 
31
32  public:
33   AliHLTTPCMemHandler();
34   virtual ~AliHLTTPCMemHandler();
35   AliHLTTPCMemHandler(const AliHLTTPCMemHandler& /*m*/){};
36   AliHLTTPCMemHandler& operator=(const AliHLTTPCMemHandler& /*&m*/)
37     {return (*this);}
38    
39   void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}  
40   void Init(Int_t s,Int_t p, Int_t *r=0);
41
42   Bool_t SetBinaryInput(char *name);
43   Bool_t SetBinaryInput(FILE *file);
44   void CloseBinaryInput();
45   
46   Bool_t SetBinaryOutput(char *name);
47   Bool_t SetBinaryOutput(FILE *file);
48   void CloseBinaryOutput();
49
50   //Random cluster
51   void SetRandomCluster(Int_t maxnumber);
52   void SetRandomSeed(UInt_t seed){srand(seed);}
53   void SetRandomSeed();
54
55   void ResetRandom(){fNDigits = 0; fNUsed = 0;}
56   void Generate(Int_t row);
57   void SetNGenerate(Int_t number){(number>fNRandom)?fNGenerate=fNRandom:fNGenerate = number;}
58
59   void SetROI(Float_t *eta,Int_t *slice);
60   void ResetROI();
61
62   //Digit IO
63   Bool_t Memory2Binary(UInt_t nrow,AliHLTTPCDigitRowData *data);
64   Bool_t Binary2Memory(UInt_t & nrow,AliHLTTPCDigitRowData *data, UInt_t& sz);
65   Bool_t Binary2Memory(UInt_t & nrow,AliHLTTPCDigitRowData *data){UInt_t tmp;return Binary2Memory(nrow,data,tmp);};
66
67   Int_t Memory2CompMemory(UInt_t nrow,AliHLTTPCDigitRowData *data,UInt_t *comp);
68   Int_t CompMemory2Memory(UInt_t nrow,AliHLTTPCDigitRowData *data,UInt_t *comp,UInt_t& sz);
69   Int_t CompMemory2Memory(UInt_t nrow,AliHLTTPCDigitRowData *data,UInt_t *comp) {UInt_t tmp;return CompMemory2Memory(nrow,data,comp,tmp);};
70   Bool_t CompMemory2CompBinary(UInt_t nrow,UInt_t *comp, UInt_t size=0);
71   Bool_t CompBinary2CompMemory(UInt_t & nrow,UInt_t *comp);
72
73   virtual AliHLTTPCDigitRowData *CompBinary2Memory(UInt_t & nrow, UInt_t& sz);
74   virtual AliHLTTPCDigitRowData *CompBinary2Memory(UInt_t & nrow){UInt_t tmp;return CompBinary2Memory(nrow,tmp);};
75   virtual Bool_t Memory2CompBinary(UInt_t nrow,AliHLTTPCDigitRowData *data);
76   
77   UInt_t GetNRow(UInt_t *comp,UInt_t size);
78
79   //Point IO
80   Bool_t Memory2Binary(UInt_t npoint,AliHLTTPCSpacePointData *data);
81   Bool_t Binary2Memory(UInt_t & npoint,AliHLTTPCSpacePointData *data, UInt_t& sz);
82   Bool_t Binary2Memory(UInt_t & npoint,AliHLTTPCSpacePointData *data) {UInt_t tmp; return Binary2Memory(npoint,data,tmp);};
83   Bool_t Transform(UInt_t npoint,AliHLTTPCSpacePointData *data,Int_t slice);
84   static void UpdateRowPointer(AliHLTTPCDigitRowData *&tempPt);
85   
86   //Track IO
87   Bool_t Memory2Binary(UInt_t ntrack,AliHLTTPCTrackSegmentData *data);
88   Bool_t Binary2Memory(UInt_t & ntrack,AliHLTTPCTrackSegmentData *data);
89   Bool_t TrackArray2Binary(AliHLTTPCTrackArray *array);
90   Bool_t Binary2TrackArray(AliHLTTPCTrackArray *array);
91   Bool_t TrackArray2Memory(UInt_t & ntrack,AliHLTTPCTrackSegmentData *data,AliHLTTPCTrackArray *array) const;
92   Bool_t Memory2TrackArray(UInt_t ntrack,AliHLTTPCTrackSegmentData *data,AliHLTTPCTrackArray *array) const;
93   Bool_t Memory2TrackArray(UInt_t ntrack,AliHLTTPCTrackSegmentData *data,AliHLTTPCTrackArray *array,Int_t slice) const;
94     
95   //Memory Allocation
96   UInt_t GetAllocatedSize() const {return fSize;}  
97   UInt_t GetFileSize();
98   UInt_t GetMemorySize(UInt_t nrow,UInt_t *comp) const;
99   UInt_t GetCompMemorySize(UInt_t nrow,AliHLTTPCDigitRowData *data) const;
100   UInt_t GetRandomSize() const;
101
102   Byte_t *Allocate(UInt_t size);
103   Byte_t *Allocate();  // allocate size of Binary Input File
104   Byte_t *Allocate(AliHLTTPCTrackArray *array);
105   Byte_t *GetDataPointer(UInt_t &size) {size = fSize; return fPt;}
106   FILE *GetFilePointer() {return fInBinary;}
107   void   Free();
108   
109   //Getters:
110   Int_t GetRowMin() const {return fRowMin;}
111   Int_t GetRowMax() const {return fRowMax;}
112   Int_t GetSlice() const {return fSlice;}
113   Int_t GetPatch() const {return fPatch;}
114   
115   //virtual functions:
116   virtual void FreeDigitsTree() {fDummy=0; return;}
117   virtual Bool_t SetAliInput(char */*name*/){fDummy=0; return 0;}
118   virtual Bool_t SetAliInput(AliRunLoader */*runloader*/){fDummy=0; return 0;}
119   virtual void CloseAliInput(){fDummy=0; return;} 
120   virtual Bool_t IsDigit(Int_t /*i*/=0){fDummy=0; return 0;}
121   virtual Bool_t SetMCOutput(char */*name*/){fDummy=0; return 0;}
122   virtual Bool_t SetMCOutput(FILE */*file*/){fDummy=0; return 0;}
123   virtual void CloseMCOutput(){fDummy=0; return;}
124   virtual Bool_t AliDigits2Binary(Int_t /*event*/=0,Bool_t /*altro*/=kFALSE){fDummy=0; return 0;}
125   virtual Bool_t AliDigits2CompBinary(Int_t /*event*/=0,Bool_t /*altro*/=kFALSE){fDummy=0; return 0;}  
126   virtual AliHLTTPCDigitRowData *AliDigits2Memory(UInt_t & /*nrow*/,Int_t /*event*/=0){fDummy=0; return 0;}
127   virtual AliHLTTPCDigitRowData *AliAltroDigits2Memory(UInt_t & /*nrow*/,Int_t /*event*/=0,Bool_t /*eventmerge*/=kFALSE){fDummy=0; return 0;}
128   virtual void AliDigits2RootFile(AliHLTTPCDigitRowData */*rowPt*/,Char_t */*new_digitsfile*/){fDummy=0; return;}
129   virtual Bool_t AliPoints2Binary(Int_t /*eventn*/=0){fDummy=0; return 0;}
130   virtual AliHLTTPCSpacePointData *AliPoints2Memory(UInt_t & /*npoint*/,Int_t /*eventn*/=0){fDummy=0; return 0;}
131
132   //AliHLTTPCRawDataFileHandler
133   virtual Bool_t SetRawInput(Char_t */*name*/){fDummy=0; return 0;}
134   virtual Bool_t SetRawInput(ifstream */*file*/){fDummy=0; return 0;}
135   virtual void CloseRawInput(){} 
136   virtual Int_t ReadRawInput(){fDummy=0; return 0;}
137   virtual Short_t** GetRawData(Int_t &/*channels*/, Int_t & /*timebins*/){fDummy=0; return 0;}
138
139   virtual Bool_t SetRawOutput(Char_t */*name*/){fDummy=0; return 0;}
140   virtual Bool_t SetRawOutput(ofstream */*file*/){fDummy=0; return 0;}
141   virtual void CloseRawOutput(){} 
142   virtual Bool_t SaveRawOutput(){fDummy=0; return 0;}
143
144   virtual Bool_t SetMappingFile(Char_t */*name*/){fDummy=0; return 0;}
145   virtual Bool_t SetMappingFile(FILE */*file*/){fDummy=0; return 0;}
146   virtual void CloseMappingFile(){} 
147   virtual Int_t ReadMappingFile(){fDummy=0; return 0;}
148   
149   virtual Bool_t SetRawPedestalsInput(Char_t */*name*/){fDummy=0; return 0;}
150   virtual Bool_t SetRawPedestalsInput(ifstream */*file*/){fDummy=0; return 0;}
151   virtual void CloseRawPedestalsInput(){} 
152   virtual Int_t ReadRawPedestalsInput(){fDummy=0; return 0;}
153
154   virtual AliHLTTPCDigitRowData* RawData2Memory(UInt_t &/*nrow*/,Int_t /*event*/=-1){fDummy=0; return 0;}
155   virtual Bool_t RawData2CompMemory(Int_t /*event*/=-1){fDummy=0; return 0;}
156
157   //AliHLTTPCDDLDataFileHandler
158   virtual Bool_t SetReaderInput(AliRawEvent */*rawevent*/){fDummy=0; return 0;}
159   virtual Bool_t SetReaderInput(Char_t */*name*/,Int_t /*event*/=0){fDummy=0; return 0;}
160   virtual void CloseReaderInput(){};
161
162   virtual AliHLTTPCDigitRowData* DDLData2Memory(UInt_t &/*nrow*/,Int_t /*event*/=-1){fDummy=0; return 0;}
163   virtual Bool_t DDLData2CompBinary(Int_t /*event*/=-1){fDummy=0; return 0;}
164
165   virtual AliTPCRawStream* GetTPCRawStream(){fDummy=0; return 0;}
166
167  protected:
168   Int_t fRowMin; //min row
169   Int_t fRowMax; //max row
170   Int_t fSlice;  //slice
171   Int_t fPatch;  //patch
172
173   Int_t fEtaMinTimeBin[159]; //for ROI in eta only
174   Int_t fEtaMaxTimeBin[159]; //for ROI in eta only
175   
176   FILE *fInBinary;//!
177   FILE *fOutBinary;//!
178
179  private:
180   
181   Byte_t *fPt;//!
182   UInt_t fSize; //size of allocated data structure
183
184   Bool_t fIsRandom; //random data generated
185   Int_t fNRandom;   //count random digits 
186   Int_t fNGenerate; //count generated digits
187   Int_t fNUsed;     //count used digits
188   Int_t fNDigits;   //count digits from digitstree
189
190   AliHLTTPCRandomDigitData **fDPt;//!
191   AliHLTTPCRandomDigitData *fRandomDigits;//!
192
193   Int_t fDummy; // to fool the virtual const problem 
194                 // of the coding conventions tool
195
196   void Write(UInt_t *comp, UInt_t & index, UInt_t & subindex, UShort_t value) const;
197   UShort_t Read(UInt_t *comp, UInt_t & index, UInt_t & subindex) const;
198   UShort_t Test(UInt_t *comp, UInt_t index, UInt_t subindex) const; 
199   
200   void DigitizePoint(Int_t row,Int_t pad, Int_t time,Int_t charge);
201   void QSort(AliHLTTPCRandomDigitData **a, Int_t first, Int_t last);
202   Int_t ComparePoints(UInt_t row,UShort_t pad,UShort_t time) const ;
203   Int_t CompareDigits(AliHLTTPCRandomDigitData *a,AliHLTTPCRandomDigitData *b) const;
204   void AddData(AliHLTTPCDigitData *data,UInt_t & ndata,
205                       UInt_t row,UShort_t pad,UShort_t time,UShort_t charge) const;
206   void AddRandom(AliHLTTPCDigitData *data,UInt_t & ndata);
207   void MergeDataRandom(AliHLTTPCDigitData *data,UInt_t & ndata,
208                       UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
209   void AddDataRandom(AliHLTTPCDigitData *data,UInt_t & ndata,
210                       UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
211
212
213   ClassDef(AliHLTTPCMemHandler,1) // Memory handler class
214 };
215 #endif