108615fc |
1 | |
2 | //Author: Uli Frankenfeld |
3 | //Last Modified: 17.12.2000 |
4 | |
5 | #include <math.h> |
6 | #include <iostream.h> |
ee4604c0 |
7 | #include <TObject.h> |
108615fc |
8 | #include <TFile.h> |
9 | #include <TTree.h> |
10 | #include <stdio.h> |
11 | |
12 | #include "AliL3Transform.h" |
13 | #include "AliL3Logging.h" |
14 | #include "AliL3MemHandler.h" |
15 | #include "AliL3FileHandler.h" |
16 | |
17 | #include "AliTPCClustersArray.h" |
18 | #include "AliTPCcluster.h" |
19 | #include "AliTPCClustersRow.h" |
20 | #include "AliTPCParam.h" |
21 | #include "AliSimDigits.h" |
22 | |
23 | #include "AliL3DigitData.h" |
24 | #include "AliL3TrackSegmentData.h" |
25 | #include "AliL3SpacePointData.h" |
26 | #include "AliL3TrackArray.h" |
27 | //_____________________________________________________________ |
28 | // |
29 | // The L3 Binary File handler |
30 | // |
31 | |
32 | ClassImp(AliL3FileHandler) |
33 | |
34 | AliL3FileHandler::AliL3FileHandler(){ |
35 | //Default constructor |
36 | fInAli = 0; |
37 | fParam = 0; |
38 | fTransformer = 0; |
39 | fMC =0; |
40 | } |
41 | |
42 | |
43 | AliL3FileHandler::~AliL3FileHandler(){ |
44 | //Destructor |
45 | if(fTransformer) delete fTransformer; |
46 | if(fMC) CloseMCOutput(); |
47 | } |
48 | |
49 | Bool_t AliL3FileHandler::SetMCOutput(char *name){ |
50 | fMC = fopen(name,"w"); |
51 | if(!fMC){ |
52 | LOG(AliL3Log::kWarning,"AliL3FileHandler::SetMCOutput","File Open") |
53 | <<"Pointer to File = 0x0 "<<ENDLOG; |
54 | return kFALSE; |
55 | } |
56 | return kTRUE; |
57 | } |
58 | |
59 | Bool_t AliL3FileHandler::SetMCOutput(FILE *file){ |
60 | fMC = file; |
61 | if(!fMC){ |
62 | LOG(AliL3Log::kWarning,"AliL3FileHandler::SetMCOutput","File Open") |
63 | <<"Pointer to File = 0x0 "<<ENDLOG; |
64 | return kFALSE; |
65 | } |
66 | return kTRUE; |
67 | } |
68 | |
69 | void AliL3FileHandler::CloseMCOutput(){ |
70 | if(!fMC){ |
71 | LOG(AliL3Log::kWarning,"AliL3FileHandler::CloseMCOutPut","File Close") |
72 | <<"Nothing to Close"<<ENDLOG; |
73 | return; |
74 | } |
75 | fclose(fMC); |
76 | fMC =0; |
77 | } |
78 | |
79 | Bool_t AliL3FileHandler::SetAliInput(){ |
80 | if(!fInAli->IsOpen()){ |
81 | LOG(AliL3Log::kError,"AliL3FileHandler::SetAliInput","File Open") |
82 | <<"Ali File "<<fInAli->GetName()<<" does not exist"<<ENDLOG; |
83 | return kFALSE; |
84 | } |
85 | fParam = (AliTPCParam*)fInAli->Get("75x40_100x60"); |
86 | if(!fParam){ |
87 | LOG(AliL3Log::kError,"AliL3FileHandler::SetAliInput","File Open") |
88 | <<"No AliTPCParam 75x40_100x60 in File "<<fInAli->GetName()<<ENDLOG; |
89 | return kFALSE; |
90 | } |
91 | fTransformer = new AliL3Transform(); |
92 | return kTRUE; |
93 | } |
94 | |
95 | Bool_t AliL3FileHandler::SetAliInput(char *name){ |
96 | fInAli= new TFile(name,"READ"); |
97 | if(!fInAli){ |
98 | LOG(AliL3Log::kWarning,"AliL3FileHandler::SetAliInput","File Open") |
99 | <<"Pointer to TFile = 0x0 "<<ENDLOG; |
100 | return kFALSE; |
101 | } |
102 | return SetAliInput(); |
103 | } |
104 | |
105 | Bool_t AliL3FileHandler::SetAliInput(TFile *file){ |
106 | fInAli=file; |
107 | if(!fInAli){ |
108 | LOG(AliL3Log::kWarning,"AliL3FileHandler::SetAliInput","File Open") |
109 | <<"Pointer to TFile = 0x0 "<<ENDLOG; |
110 | return kFALSE; |
111 | } |
112 | return SetAliInput(); |
113 | } |
114 | |
115 | void AliL3FileHandler::CloseAliInput(){ |
116 | if(!fInAli){ |
117 | LOG(AliL3Log::kWarning,"AliL3FileHandler::CloseAliInput","File Close") |
118 | <<"Nothing to Close"<<ENDLOG; |
119 | return; |
120 | } |
121 | if(fInAli->IsOpen()) fInAli->Close(); |
122 | delete fInAli; |
123 | fInAli = 0; |
124 | } |
125 | |
126 | Bool_t AliL3FileHandler::IsDigit(){ |
127 | if(!fInAli){ |
128 | LOG(AliL3Log::kWarning,"AliL3FileHandler::IsDigit","File") |
129 | <<"Pointer to TFile = 0x0 "<<ENDLOG; |
130 | return kTRUE; //may you are use binary input which is Digits!! |
131 | } |
132 | TTree *t=(TTree*)fInAli->Get("TreeD_75x40_100x60"); |
133 | if(t){ |
134 | LOG(AliL3Log::kInformational,"AliL3FileHandler::IsDigit","File Type") |
135 | <<"Found Digit Tree -> Use Fast Cluster Finder"<<ENDLOG; |
136 | return kTRUE; |
137 | } |
138 | else{ |
139 | LOG(AliL3Log::kInformational,"AliL3FileHandler::IsDigit","File Type") |
140 | <<"No Digit Tree -> Use Cluster Tree"<<ENDLOG; |
141 | return kFALSE; |
142 | } |
143 | } |
144 | |
145 | ///////////////////////////////////////// Digit IO |
146 | Bool_t AliL3FileHandler::AliDigits2Binary(){ |
147 | Bool_t out = kTRUE; |
148 | UInt_t nrow; |
149 | AliL3DigitRowData* data = AliDigits2Memory(nrow); |
150 | out = Memory2Binary(nrow,data); |
151 | Free(); |
152 | return out; |
153 | } |
154 | |
155 | |
156 | Bool_t AliL3FileHandler::AliDigits2CompBinary(){ |
157 | Bool_t out = kTRUE; |
158 | UInt_t ndigits=0; |
159 | AliL3DigitRowData* digits = AliDigits2Memory(ndigits); |
160 | out = Memory2CompBinary(ndigits,digits); |
161 | Free(); |
162 | return out; |
163 | } |
164 | |
165 | |
166 | AliL3DigitRowData * AliL3FileHandler::AliDigits2Memory(UInt_t & nrow){ |
167 | AliL3DigitRowData *data = 0; |
168 | nrow=0; |
169 | if(!fInAli){ |
170 | LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Memory","File") |
171 | <<"No Input avalible: no object TFile"<<ENDLOG; |
172 | return 0; |
173 | } |
174 | if(!fInAli->IsOpen()){ |
175 | LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Memory","File") |
176 | <<"No Input avalible: TFile not opend"<<ENDLOG; |
177 | return 0; |
178 | } |
179 | |
180 | TDirectory *savedir = gDirectory; |
181 | fInAli->cd(); |
182 | TTree *t=(TTree*)fInAli->Get("TreeD_75x40_100x60"); |
183 | if(!t){ |
184 | LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Binary","AliRoot") |
185 | <<"No Digit Tree inside!"<<ENDLOG; |
186 | return 0; |
187 | } |
188 | AliSimDigits digarr, *dummy=&digarr; |
189 | t->GetBranch("Segment")->SetAddress(&dummy); |
190 | UShort_t dig; |
191 | Int_t time,pad,sector,row; |
192 | Int_t nrows=0; |
193 | Int_t ndigitcount=0; |
194 | Int_t entries = (Int_t)t->GetEntries(); |
195 | Int_t ndigits[entries]; |
196 | Int_t lslice,lrow; |
197 | for(Int_t n=0; n<t->GetEntries(); n++) |
198 | { |
199 | t->GetEvent(n); |
200 | fParam->AdjustSectorRow(digarr.GetID(),sector,row); |
201 | fTransformer->Sector2Slice(lslice,lrow,sector,row); |
202 | if(fSlice != lslice || lrow<fRowMin || lrow>fRowMax) continue; |
203 | // if(fSlice != lslice) continue; |
204 | |
205 | Float_t xyz[3]; |
206 | ndigits[lrow] = 0; |
207 | digarr.First(); |
208 | do { |
209 | time=digarr.CurrentRow(); |
210 | pad=digarr.CurrentColumn(); |
211 | dig = digarr.GetDigit(time,pad); |
212 | if(dig<=fParam->GetZeroSup()) continue; |
213 | if(time < fParam->GetMaxTBin()-1 && time > 0) |
214 | if(digarr.GetDigit(time+1,pad) <= fParam->GetZeroSup() |
215 | && digarr.GetDigit(time-1,pad) <= fParam->GetZeroSup()) |
216 | continue; |
217 | |
218 | fTransformer->Raw2Local(xyz,sector,row,pad,time); |
219 | if(fParam->GetPadRowRadii(sector,row)<230./250.*fabs(xyz[2])) |
220 | continue; |
221 | |
222 | ndigits[lrow]++; //for this row only |
223 | ndigitcount++; //total number of digits to be published |
224 | |
225 | } while (digarr.Next()); |
226 | |
227 | nrows++; |
228 | } |
229 | Int_t size = sizeof(AliL3DigitData)*ndigitcount |
230 | + nrows*sizeof(AliL3DigitRowData); |
231 | |
232 | LOG(AliL3Log::kDebug,"AliL3FileHandler::AliDigits2Memory","Digits") |
233 | <<AliL3Log::kDec<<"Found "<<ndigitcount<<" Digits"<<ENDLOG; |
234 | |
235 | data=(AliL3DigitRowData*) Allocate(size); |
236 | nrow = (UInt_t)nrows; |
237 | AliL3DigitRowData *tempPt = data; |
238 | for(Int_t n=0; n<t->GetEntries(); n++) |
239 | { |
240 | t->GetEvent(n); |
241 | |
242 | Float_t xyz[3]; |
243 | fParam->AdjustSectorRow(digarr.GetID(),sector,row); |
244 | fTransformer->Sector2Slice(lslice,lrow,sector,row); |
245 | // if(fSlice != lslice) continue; |
246 | if(fSlice != lslice || lrow<fRowMin || lrow>fRowMax) continue; |
247 | tempPt->fRow = lrow; |
248 | tempPt->fNDigit = ndigits[lrow]; |
249 | |
250 | Int_t localcount=0; |
251 | digarr.First(); |
252 | do { |
253 | dig=digarr.CurrentDigit(); |
254 | if (dig<=fParam->GetZeroSup()) continue; |
255 | time=digarr.CurrentRow(); |
256 | pad=digarr.CurrentColumn(); |
257 | if(time < fParam->GetMaxTBin()-1 && time > 0) |
258 | if(digarr.GetDigit(time-1,pad) <= fParam->GetZeroSup() && |
259 | digarr.GetDigit(time+1,pad) <= fParam->GetZeroSup()) continue; |
260 | |
261 | //Exclude data outside cone: |
262 | fTransformer->Raw2Local(xyz,sector,row,pad,time); |
263 | if(fParam->GetPadRowRadii(sector,row)<230./250.*fabs(xyz[2])) |
264 | continue; |
265 | |
266 | if(localcount >= ndigits[lrow]) |
267 | LOG(AliL3Log::kFatal,"AliL3FileHandler::AliDigits2Binary","Memory") |
268 | <<AliL3Log::kDec<<"Mismatch: localcount "<<localcount<<" ndigits " |
269 | <<ndigits[lrow]<<ENDLOG; |
270 | |
271 | tempPt->fDigitData[localcount].fCharge=dig; |
272 | tempPt->fDigitData[localcount].fPad=pad; |
273 | tempPt->fDigitData[localcount].fTime=time; |
274 | localcount++; |
275 | } while (digarr.Next()); |
276 | |
277 | Byte_t *tmp = (Byte_t*)tempPt; |
278 | Int_t size = sizeof(AliL3DigitRowData) |
279 | + ndigits[lrow]*sizeof(AliL3DigitData); |
280 | tmp += size; |
281 | tempPt = (AliL3DigitRowData*)tmp; |
282 | } |
283 | savedir->cd(); |
284 | return data; |
285 | } |
286 | |
287 | ///////////////////////////////////////// Point IO |
288 | Bool_t AliL3FileHandler::AliPoints2Binary(){ |
289 | Bool_t out = kTRUE; |
290 | UInt_t npoint; |
291 | AliL3SpacePointData *data = AliPoints2Memory(npoint); |
292 | out = Memory2Binary(npoint,data); |
293 | Free(); |
294 | return out; |
295 | } |
296 | |
297 | AliL3SpacePointData * AliL3FileHandler::AliPoints2Memory(UInt_t & npoint){ |
298 | AliL3SpacePointData *data = 0; |
299 | npoint=0; |
300 | if(!fInAli){ |
301 | LOG(AliL3Log::kWarning,"AliL3FileHandler::AliPoints2Memory","File") |
302 | <<"No Input avalible: no object TFile"<<ENDLOG; |
303 | return 0; |
304 | } |
305 | if(!fInAli->IsOpen()){ |
306 | LOG(AliL3Log::kWarning,"AliL3FileHandler::AliPoints2Memory","File") |
307 | <<"No Input avalible: TFile not opend"<<ENDLOG; |
308 | return 0; |
309 | } |
310 | TDirectory *savedir = gDirectory; |
311 | fInAli->cd(); |
312 | |
313 | AliTPCClustersArray carray; |
314 | carray.Setup(fParam); |
315 | carray.SetClusterType("AliTPCcluster"); |
316 | Bool_t clusterok = carray.ConnectTree("Segment Tree"); |
317 | if(!clusterok) return 0; |
318 | |
319 | AliTPCClustersRow ** clusterrow = |
320 | new AliTPCClustersRow*[ (int)carray.GetTree()->GetEntries()]; |
321 | Int_t *rows = new int[ (int)carray.GetTree()->GetEntries()]; |
322 | Int_t *sects = new int[ (int)carray.GetTree()->GetEntries()]; |
323 | Int_t sum=0; |
324 | |
325 | Int_t lslice,lrow; |
326 | for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++){ |
327 | AliSegmentID *s = carray.LoadEntry(i); |
328 | Int_t sector,row; |
329 | fParam->AdjustSectorRow(s->GetID(),sector,row); |
330 | rows[i] = row; |
331 | sects[i] = sector; |
332 | clusterrow[i] = 0; |
333 | fTransformer->Sector2Slice(lslice,lrow,sector,row); |
334 | if(fSlice != lslice || lrow<fRowMin || lrow>fRowMax) continue; |
335 | clusterrow[i] = carray.GetRow(sector,row); |
336 | if(clusterrow[i]) |
337 | sum+=clusterrow[i]->GetArray()->GetEntriesFast(); |
338 | } |
339 | UInt_t size = sum*sizeof(AliL3SpacePointData); |
340 | |
341 | LOG(AliL3Log::kDebug,"AliL3FileHandler::AliPoints2Memory","File") |
342 | <<AliL3Log::kDec<<"Found "<<sum<<" SpacePoints"<<ENDLOG; |
343 | |
344 | data = (AliL3SpacePointData *) Allocate(size); |
345 | npoint = sum; |
346 | UInt_t n=0; |
347 | for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++){ |
348 | if(!clusterrow[i]) continue; |
349 | Int_t row = rows[i]; |
350 | Int_t sector = sects[i]; |
351 | fTransformer->Sector2Slice(lslice,lrow,sector,row); |
352 | Int_t entries_in_row = clusterrow[i]->GetArray()->GetEntriesFast(); |
353 | for(Int_t j = 0;j<entries_in_row;j++){ |
354 | AliTPCcluster *c = (AliTPCcluster*)(*clusterrow[i])[j]; |
355 | data[n].fZ = c->GetZ(); |
356 | data[n].fY = c->GetY(); |
357 | data[n].fX = fParam->GetPadRowRadii(sector,row); |
358 | data[n].fID = n+((fSlice&0x7f)<<25)+((fPatch&0x7)<<22);//uli |
359 | data[n].fPadRow = lrow; |
360 | data[n].fXYErr = c->GetSigmaY2(); |
361 | data[n].fZErr = c->GetSigmaZ2(); |
362 | if(fMC) fprintf(fMC,"%d %d\n",data[n].fID,c->GetLabel(0)); |
363 | n++; |
364 | } |
365 | } |
366 | for(Int_t i=0;i<carray.GetTree()->GetEntries();i++){ |
367 | Int_t row = rows[i]; |
368 | Int_t sector = sects[i]; |
369 | if(carray.GetRow(sector,row)) |
370 | carray.ClearRow(sector,row); |
371 | } |
372 | |
373 | delete [] clusterrow; |
374 | delete [] rows; |
375 | delete [] sects; |
376 | savedir->cd(); |
377 | |
378 | return data; |
379 | } |
380 | |