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 | |
53 | Int_t fEtaMinTimeBin[174]; |
54 | Int_t fEtaMaxTimeBin[174]; |
55 | |
56 | |
57 | public: |
58 | AliL3MemHandler(); |
59 | virtual ~AliL3MemHandler(); |
60 | |
61 | void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();} |
62 | void Init(AliL3Transform *t){fTransformer = t;} |
63 | void Init(Int_t s,Int_t p,Int_t* row){fSlice=s;fPatch=p;fRowMin=row[0];fRowMax=row[1]; ResetROI();} |
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); |
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, |
110 | AliL3TrackArray *array); |
111 | Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data, |
112 | AliL3TrackArray *array); |
113 | Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data, |
114 | AliL3TrackArray *array,Int_t slice, AliL3Transform* trans); |
115 | |
116 | //Memory Allocation |
117 | UInt_t GetAllocatedSize(){return fSize;} |
118 | UInt_t GetFileSize(); |
119 | UInt_t GetMemorySize(UInt_t nrow,UInt_t *comp); |
120 | UInt_t GetCompMemorySize(UInt_t nrow,AliL3DigitRowData *data); |
121 | UInt_t GetRandomSize(); |
122 | |
123 | Byte_t *Allocate(UInt_t size); |
124 | Byte_t *Allocate(); // allocate size of Binary Input File |
125 | Byte_t *Allocate(AliL3TrackArray *array); |
126 | void Free(); |
127 | |
128 | ClassDef(AliL3MemHandler,1) |
129 | }; |
130 | |
131 | inline Int_t AliL3MemHandler::ComparePoints(UInt_t row,UShort_t pad,UShort_t time){ |
132 | if(fNUsed>=fNDigits) return -2; |
133 | |
134 | if(pad==fDPt[fNUsed]->fPad&&time==fDPt[fNUsed]->fTime) return 0; |
135 | |
136 | if(pad<fDPt[fNUsed]->fPad) return -1; |
137 | if(pad==fDPt[fNUsed]->fPad&&time<fDPt[fNUsed]->fTime) return -1; |
138 | |
139 | return 1; |
140 | } |
141 | |
142 | inline Int_t AliL3MemHandler::CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b){ |
143 | if(a->fPad==b->fPad && a->fTime == b->fTime) return 0; |
144 | |
145 | if(a->fPad<b->fPad) return -1; |
146 | if(a->fPad==b->fPad && a->fTime<b->fTime) return -1; |
147 | |
148 | return 1; |
149 | } |
150 | |
151 | #endif |