Adding macros to create Calibration objects
[u/mrichter/AliRoot.git] / TRD / TRDbase / AliTRDrawData.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id$ */
17
18 ///////////////////////////////////////////////////////////////////////////////
19 //                                                                           //
20 //  TRD raw data conversion class                                            //
21 //                                                                           //
22 ///////////////////////////////////////////////////////////////////////////////
23
24
25 #include <TMath.h>
26 #include "TClass.h"
27
28 #include "AliDAQ.h"
29 #include "AliRawDataHeaderSim.h"
30 #include "AliRawReader.h"
31 #include "AliLog.h"
32 #include "AliFstream.h"
33 #include "AliLoader.h"
34 #include "AliTreeLoader.h"
35
36 #include "AliTRDrawData.h"
37 #include "AliTRDdigitsManager.h"
38 #include "AliTRDgeometry.h"
39 #include "AliTRDarrayDictionary.h"
40 #include "AliTRDarrayADC.h"
41 #include "AliTRDrawStream.h"
42 #include "AliTRDcalibDB.h"
43 #include "AliTRDSignalIndex.h"
44 #include "AliTRDfeeParam.h"
45 #include "AliTRDmcmSim.h"
46 #include "AliTRDtrackletWord.h"
47 #include "AliTRDtrackGTU.h"
48 #include "AliESDTrdTrack.h"
49 #include "AliTRDdigitsParam.h"
50
51 ClassImp(AliTRDrawData)
52
53 Int_t AliTRDrawData::fgDataSuppressionLevel = 0;
54
55 //_____________________________________________________________________________
56 AliTRDrawData::AliTRDrawData()
57   :TObject()
58   ,fRunLoader(NULL)
59   ,fGeo(NULL)
60   ,fFee(NULL)
61   ,fNumberOfDDLs(0)
62   ,fTrackletTree(NULL)
63   ,fTracklets(new TClonesArray("AliTRDtrackletWord", 1000))
64   ,fTracks(new TClonesArray("AliESDTrdTrack", 500))
65   ,fSMindexPos(0)
66   ,fStackindexPos(0)
67   ,fEventCounter(0)
68   ,fTrgFlags()
69   ,fMcmSim(new AliTRDmcmSim)
70   ,fDigitsParam(NULL)
71 {
72   //
73   // Default constructor
74   //
75
76   fFee = AliTRDfeeParam::Instance();
77   fNumberOfDDLs = AliDAQ::NumberOfDdls("TRD");
78
79 }
80
81 //_____________________________________________________________________________
82 AliTRDrawData::AliTRDrawData(const AliTRDrawData &r)
83   :TObject(r)
84   ,fRunLoader(NULL)
85   ,fGeo(NULL)
86   ,fFee(NULL)
87   ,fNumberOfDDLs(0)
88   ,fTrackletTree(NULL)
89   ,fTracklets(new TClonesArray("AliTRDtrackletWord", 1000))
90   ,fTracks(new TClonesArray("AliESDTrdTrack", 500))
91   ,fSMindexPos(0)
92   ,fStackindexPos(0)
93   ,fEventCounter(0)
94   ,fTrgFlags()
95   ,fMcmSim(new AliTRDmcmSim)
96   ,fDigitsParam(NULL)
97 {
98   //
99   // Copy constructor
100   //
101
102   fFee = AliTRDfeeParam::Instance();
103   fNumberOfDDLs = AliDAQ::NumberOfDdls("TRD");
104
105 }
106
107 //_____________________________________________________________________________
108 AliTRDrawData::~AliTRDrawData()
109 {
110   //
111   // Destructor
112   //
113
114   delete fTracklets;
115   delete fTracks;
116   delete fMcmSim;
117
118 }
119
120 //_____________________________________________________________________________
121 Bool_t AliTRDrawData::Digits2Raw(TTree *digitsTree, const TTree *tracks )
122 {
123   //
124   // Initialize necessary parameters and call one
125   // of the raw data simulator selected by SetRawVersion.
126   //
127   // Currently tracklet output is not spported yet and it
128   // will be supported in higher version simulator.
129   //
130
131   AliTRDdigitsManager* const digitsManager = new AliTRDdigitsManager();
132
133   if (!digitsManager->ReadDigits(digitsTree)) {
134     delete digitsManager;
135     return kFALSE;
136   }
137
138   if (tracks != NULL) {
139     delete digitsManager;
140     AliError("Tracklet input is not supported yet.");
141     return kFALSE;
142   }
143
144   fGeo = new AliTRDgeometry();
145
146   if (!AliTRDcalibDB::Instance()) {
147     AliError("Could not get calibration object");
148     delete fGeo;
149     delete digitsManager;
150     return kFALSE;
151   }
152
153   Int_t retval = kTRUE;
154   Int_t rv     = fFee->GetRAWversion();
155
156   // Call appropriate Raw Simulator
157   if ( rv > 0 && rv <= 3 ) retval = Digits2Raw(digitsManager); 
158   else {
159     retval = kFALSE;
160     AliWarning(Form("Unsupported raw version (%d).", rv));
161   }
162
163   // Cleanup
164   delete fGeo;
165   delete digitsManager;
166
167   return retval;
168
169 }
170
171 //_____________________________________________________________________________
172 Bool_t AliTRDrawData::Digits2Raw(AliTRDdigitsManager *digitsManager)
173 {
174   //
175   // Raw data simulator for all versions > 0. This is prepared for real data.
176   // This version simulate only raw data with ADC data and not with tracklet.
177   //
178
179   const Int_t kMaxHcWords = (fGeo->TBmax()/3)
180                           * fGeo->ADCmax()
181                           * fGeo->MCMmax()
182                           * fGeo->ROBmaxC1()/2 
183                           + 100 + 20;
184
185   // Buffer to temporary store half chamber data
186   UInt_t     *hcBuffer    = new UInt_t[kMaxHcWords];
187
188   Bool_t newEvent = kFALSE;  // only for correct readout tree
189   Bool_t newSM    = kFALSE;  // new SM flag, for writing SM index words
190   // Coverity
191   //Bool_t newStack = kFALSE;  // new stack flag, for writing stack index words
192
193   // Get digits parameter
194   fDigitsParam = digitsManager->GetDigitsParam();
195
196   // sect is same as iDDL, so I use only sect here.
197   for (Int_t sect = 0; sect < fGeo->Nsector(); sect++) { 
198
199     char name[1024];
200     snprintf(name,1024,"TRD_%d.ddl",sect + AliTRDrawStream::kDDLOffset);
201
202     AliFstream* of = new AliFstream(name);
203
204     // Write a dummy data header
205     AliRawDataHeaderSim  header;  // the event header
206     UInt_t hpos = of->Tellp();
207     of->WriteBuffer((char *) (& header), sizeof(header));
208     
209     // Reset payload byte size (payload does not include header).
210     Int_t npayloadbyte = 0;
211
212     // check the existance of the data
213     // SM index word and Stack index word
214     UInt_t *iwbuffer = new UInt_t[109]; // index word buffer; max 109 = 2 SM headers + 67 dummy headers + 5*8 stack headers
215     Int_t nheader = 0;
216     UInt_t bStackMask = 0x0;
217     Bool_t bStackHasData = kFALSE;
218     Bool_t bSMHasData = kFALSE;
219     
220     //iwbuffer[nheader++] = 0x0001a020;   // SM index words 
221     iwbuffer[nheader++] = 0x0044b020;   // SM index words | additional SM header:48 = 1 SM header + 47 dummy words(for future use)
222     iwbuffer[nheader++] = 0x10404071;   // SM header
223     for ( Int_t i=0; i<66; i++ ) iwbuffer[nheader++] = 0x00000000;  // dummy words 
224     iwbuffer[nheader++] = 0x10000000;   // end of dummy words
225     
226     for ( Int_t stack= 0; stack < fGeo->Nstack(); stack++) {
227       UInt_t linkMask = 0x0;
228       for( Int_t layer = 0; layer < fGeo->Nlayer(); layer++) {
229         Int_t iDet = fGeo->GetDetector(layer,stack,sect);
230         AliTRDarrayADC *digits = (AliTRDarrayADC *) digitsManager->GetDigits(iDet);
231         if ( fgDataSuppressionLevel==0 || digits->HasData() ) {
232           bStackMask = bStackMask | ( 1 << stack ); // active stack mask for new stack
233           linkMask = linkMask | ( 3 << (2*layer) );    // 3 = 0011
234           bStackHasData = kTRUE;
235           bSMHasData = kTRUE;
236         } // has data
237       } // loop over layer
238       
239       if ( fgDataSuppressionLevel==0 || bStackHasData ){
240         iwbuffer[nheader++] = 0x0007a000 | linkMask;    // stack index word + link masks
241         if (fgDataSuppressionLevel==0) iwbuffer[nheader-1] = 0x0007afff;  // no suppression
242         iwbuffer[nheader++] = 0x04045b01;               // stack header
243         for (Int_t i=0;i<6;i++) iwbuffer[nheader++] = 0x00000000; // 6 dummy words
244         bStackHasData = kFALSE;
245       }
246     } // loop over stack
247     
248     if ( fgDataSuppressionLevel==0 || bSMHasData ){
249       iwbuffer[0] = iwbuffer[0] | bStackMask;  // add stack masks to SM index word
250       if (fgDataSuppressionLevel==0) iwbuffer[0] = 0x0044b03f;    // no suppression : all stacks are active
251       of->WriteBuffer((char *) iwbuffer, nheader*4);
252       AliDebug(11, Form("SM %d index word: %08x", sect, iwbuffer[0]));
253       AliDebug(11, Form("SM %d header: %08x", sect, iwbuffer[1]));
254     }
255     // end of SM & stack header ------------------------------------------------------------------------
256     // -------------------------------------------------------------------------------------------------
257     
258     // Prepare chamber data
259     for( Int_t stack = 0; stack < fGeo->Nstack(); stack++) {
260       for( Int_t layer = 0; layer < fGeo->Nlayer(); layer++) {
261         
262         Int_t iDet = fGeo->GetDetector(layer,stack,sect);
263         if (iDet == 0){
264           newEvent = kTRUE; // it is expected that each event has at least one tracklet; 
265           // this is only needed for correct readout tree
266           fEventCounter++;
267           AliDebug(11, Form("New event!! Event counter: %d",fEventCounter));
268         }
269         
270         if ( stack==0 && layer==0 ) newSM = kTRUE;  // new SM flag
271         // Coverity
272         //if ( layer==0 ) newStack = kTRUE;           // new stack flag
273         AliDebug(15, Form("stack : %d, layer : %d, iDec : %d\n",stack,layer,iDet));
274         // Get the digits array
275         AliTRDarrayADC *digits = (AliTRDarrayADC *) digitsManager->GetDigits(iDet);
276         if (fgDataSuppressionLevel==0 || digits->HasData() ) {  // second part is new!! and is for indicating a new event
277           
278           if (digits->HasData()) digits->Expand();
279           
280           Int_t hcwords = 0;
281           
282           // Process A side of the chamber
283           hcwords = ProduceHcData(digits,0,iDet,hcBuffer,kMaxHcWords,newEvent,newSM);
284           if ( newEvent ) newEvent = kFALSE;
285           //AssignLinkMask(hcBuffer, layer);  // active link mask for this layer(2*HC)
286           of->WriteBuffer((char *) hcBuffer, hcwords*4);
287           npayloadbyte += hcwords*4;
288           
289           // Process B side of the chamber
290           hcwords = ProduceHcData(digits,1,iDet,hcBuffer,kMaxHcWords,newEvent,newSM);
291           of->WriteBuffer((char *) hcBuffer, hcwords*4);
292           npayloadbyte += hcwords*4;
293         } // has data
294         
295       } // loop over layer
296     } // loop over stack
297     
298     // Complete header
299     header.fSize = UInt_t(of->Tellp()) - hpos;
300     header.SetAttribute(0);  // Valid data
301     of->Seekp(hpos);         // Rewind to header position
302     of->WriteBuffer((char *) (& header), sizeof(header));
303     delete of;
304
305     delete [] iwbuffer;
306
307   } // loop over sector(SM)
308   
309   delete [] hcBuffer;
310   
311   return kTRUE;
312 }
313
314 //_____________________________________________________________________________
315 void AliTRDrawData::ProduceSMIndexData(UInt_t *buf, Int_t& nw){
316         // 
317         // This function generates 
318         //       1) SM index words : ssssssss ssssssss vvvv rrrr r d t mmmmm
319         //          - s : size of SM header (number of header, default = 0x0001)
320         //          - v : SM header version (default = 0xa)
321         //          - r : reserved for future use (default = 00000)
322         //          - d : track data enabled bit (default = 0)
323         //          - t : tracklet data enabled bit (default = 1)
324         //          - m : stack mask (each bit corresponds a stack, default = 11111)
325         //
326         //       2) SM header : rrr c vvvv vvvvvvvv vvvv rrrr bbbbbbbb
327         //          - r : reserved for future use (default = 000)
328         //          - c : clean check out flag (default = 1)
329         //          - v : hardware design revision (default = 0x0404)
330         //          - r : reserved for future use (default = 0x0)
331         //          - b : physical board ID (default = 0x71)
332         //
333         //       3) stack index words : ssssssss ssssssss vvvv mmmm mmmmmmmm
334         //          - s : size of stack header (number of header, (default = 0x0007)
335         //          - v : header version (default = 0xa)
336         //          - m : link mask (default = 0xfff)
337         //
338         //       4) stack header : vvvvvvvv vvvvvvvv bbbbbbbb rrrr rrr c
339         //          - v : hardware design revision (default = 0x0404)
340         //          - b : physical board ID (default = 0x5b)
341         //          - r : reserved for future use (default = 0000 000)
342         //          - c : clean checkout flag (default = 1)
343         //      
344         //       and 6 dummy words(0x00000000)
345         //
346         
347     //buf[nw++] = 0x0001a03f;   // SM index words
348     fSMindexPos = nw;       // memorize position of the SM index word for re-allocating stack mask
349     buf[nw++] = 0x0001b020; // SM index words
350     buf[nw++] = 0x10404071; // SM header
351
352     fStackindexPos = nw;    // memorize position of the stack index word for future adding
353         /*  
354     for (Int_t istack=0; istack<5; istack++){
355         buf[nw++] = 0x0007afff; // stack index words
356         buf[nw++] = 0x04045b01; // stack header
357         for (Int_t i=0;i<6;i++) buf[nw++] = 0x00000000; // 6 dummy words
358     } // loop over 5 stacks
359         */
360 }
361
362 //_____________________________________________________________________________
363 void AliTRDrawData::AssignStackMask(UInt_t *buf, Int_t nStack){
364     //
365     // This function re-assign stack mask active(from 0 to 1) in the SM index word
366     //   
367     buf[fSMindexPos] = buf[fSMindexPos] | ( 1 << nStack );
368 }
369
370 //_____________________________________________________________________________  
371 Int_t AliTRDrawData::AddStackIndexWords(UInt_t *buf, Int_t /*nStack*/, Int_t nMax){
372     // 
373     // This function add stack index words and stack header when there is data for the stack
374     //
375     //   1) stack index words : ssssssss ssssssss vvvv mmmm mmmmmmmm 
376     //      - s : size of stack header (number of header, (default = 0x0007)       
377     //      - v : header version (default = 0xa)
378     //      - m : link mask (default = 0xfff)
379     //      - m : link mask (starting value = 0x000)
380     //
381     //   2) stack header : vvvvvvvv vvvvvvvv bbbbbbbb rrrr rrr c
382     //      - v : hardware design revision (default = 0x0404)
383     //      - b : physical board ID (default = 0x5b)
384     //      - r : reserved for future use (default = 0000 000)
385     //      - c : clean checkout flag (default = 1)
386     //  
387     //   and 6 dummy words(0x00000000)
388     //
389
390     Int_t nAddedWords = 0;  // Number of added words
391     if ( ShiftWords(buf, fStackindexPos, 8, nMax)== kFALSE ){
392         AliError("Adding stack header failed.");
393         return 0;
394     }
395
396     buf[fStackindexPos++] = 0x0007a000; // stack index words
397     buf[fStackindexPos++] = 0x04045b01; // stack header
398     for (Int_t i=0;i<6;i++) buf[fStackindexPos++] = 0x00000000; // 6 dummy words 
399     nAddedWords += 8;
400
401     return nAddedWords;
402 }
403
404 //_____________________________________________________________________________
405 void AliTRDrawData::AssignLinkMask(UInt_t *buf, Int_t nLayer){
406     //
407     // This function re-assign link mask active(from 0 to 1) in the stack index word
408     //   
409     buf[fStackindexPos-8] = buf[fStackindexPos-8] | ( 3 << (2*nLayer) );    // 3 = 0011 
410 }
411
412 //_____________________________________________________________________________ 
413 Bool_t AliTRDrawData::ShiftWords(UInt_t *buf, Int_t nStart, Int_t nWords, Int_t nMax){
414     //  
415     // This function shifts n words
416     //
417     //if ( nStart+nWords > sizeof(buf)/sizeof(UInt_t) ){
418     //  AliError("Words shift failed. No more buffer space.");
419     //  return kFALSE;
420     //}
421
422     for ( Int_t iw=nMax; iw>nStart-1; iw--){
423         buf[iw+nWords] = buf[iw];
424     }
425     return kTRUE;
426 }
427
428 //_____________________________________________________________________________
429 Int_t AliTRDrawData::ProduceHcData(AliTRDarrayADC *digits, Int_t side, Int_t det, UInt_t *buf, Int_t maxSize, Bool_t /*newEvent = kFALSE*/, Bool_t /*newSM = kFALSE*/){
430         //
431         // This function produces the raw data for one HC, i.e. tracklets, tracklet endmarkers, 
432         // raw data, raw data endmarkers. 
433         // This function can be used for both ZS and NZS data
434         //
435
436         Int_t           nw = 0;                       // Number of written    words
437         Int_t           of = 0;                       // Number of overflowed words
438         Int_t      *tempnw = &nw;                     // Number of written    words for temp. buffer
439         Int_t      *tempof = &of;                     // Number of overflowed words for temp. buffer
440         Int_t        layer = fGeo->GetLayer( det );   // Layer
441         Int_t        stack = fGeo->GetStack( det );   // Stack
442         Int_t         sect = fGeo->GetSector( det );  // Sector (=iDDL)
443         const Int_t kCtype = fGeo->GetStack(det) == 2 ? 0 : 1;                       // Chamber type (0:C0, 1:C1)
444
445         Bool_t trackletOn = fFee->GetTracklet();     // tracklet simulation active?
446
447         AliDebug(1,Form("Producing raw data for sect=%d layer=%d stack=%d side=%d",sect,layer,stack,side));
448         
449         UInt_t *tempBuffer = buf; // tempBuffer used to write ADC data
450                                   // different in case of tracklet writing
451         
452         if (trackletOn) {
453           tempBuffer = new UInt_t[maxSize];
454           tempnw     = new Int_t(0);
455           tempof     = new Int_t(0);
456         }
457           
458         WriteIntermediateWords(tempBuffer,*tempnw,*tempof,maxSize,det,side);
459
460         if (digits->HasData()) {
461           // scanning direction such, that tracklet-words are sorted in ascending z and then in ascending y order
462           // ROB numbering on chamber and MCM numbering on ROB increase with decreasing z and increasing y
463           for (Int_t iRobRow = 0; iRobRow <= (kCtype + 3)-1; iRobRow++ ) {
464             // ROB number should be increasing
465             Int_t iRob = iRobRow * 2 + side;
466             // MCM on one ROB
467             for (Int_t iMcmRB = 0; iMcmRB < fGeo->MCMmax(); iMcmRB++ ) {
468               Int_t iMcm = 16 - 4*(iMcmRB/4 + 1) + (iMcmRB%4);
469               
470               fMcmSim->Init(det, iRob, iMcm);
471               fMcmSim->SetData(digits);     // no filtering done here (already done in digitizer)
472               if (trackletOn) {
473                 fMcmSim->Tracklet();
474                 Int_t tempNw = fMcmSim->ProduceTrackletStream(&buf[nw], maxSize - nw);
475                 if(  tempNw < 0 ) {
476                   of += tempNw;
477                   nw += maxSize - nw;
478                   AliError(Form("Buffer overflow detected. Please increase the buffer size and recompile."));
479                 } else {
480                   nw += tempNw;
481                 }
482               }
483               fMcmSim->ZSMapping();  // Calculate zero suppression mapping
484               // at the moment it has to be rerun here
485               // Write MCM data to temp. buffer
486               Int_t tempNw = fMcmSim->ProduceRawStream( &tempBuffer[*tempnw], maxSize - *tempnw, fEventCounter );
487               if ( tempNw < 0 ) {
488                 *tempof += tempNw;
489                 *tempnw += maxSize - nw;
490                 AliError(Form("Buffer overflow detected. Please increase the buffer size and recompile."));
491               } else {
492                 *tempnw += tempNw;
493               }
494             }
495           }
496         }
497           
498         // in case of tracklet writing copy temp data to final buffer
499         if (trackletOn) {
500           if (nw + *tempnw < maxSize) {
501             memcpy(&buf[nw], tempBuffer, *tempnw * sizeof(UInt_t));
502             nw += *tempnw;
503           }
504           else {
505             AliError("Buffer overflow detected");
506           }
507
508           delete [] tempBuffer;
509           delete tempof;
510           delete tempnw;
511         }
512     
513         // Write end of raw data marker
514         if (nw+3 < maxSize) {
515           buf[nw++] = fgkEndOfDataMarker;
516           buf[nw++] = fgkEndOfDataMarker;
517           buf[nw++] = fgkEndOfDataMarker;
518           buf[nw++] = fgkEndOfDataMarker;
519         } else {
520           of += 4;
521         }
522         
523         if (of != 0) {
524           AliError("Buffer overflow. Data is truncated. Please increase buffer size and recompile.");
525         }
526
527         return nw;
528 }
529
530 //_____________________________________________________________________________
531 AliTRDdigitsManager *AliTRDrawData::Raw2Digits(AliRawReader *rawReader)
532 {
533   //
534   // Vx of the raw data reading
535   //
536
537   rawReader->Select("TRD"); //[mj]
538
539   AliTRDarrayADC *digits = 0;
540   AliTRDarrayDictionary *track0 = 0;
541   AliTRDarrayDictionary *track1 = 0;
542   AliTRDarrayDictionary *track2 = 0;  
543
544   //AliTRDSignalIndex *indexes = 0;
545   // Create the digits manager
546   AliTRDdigitsManager* digitsManager = new AliTRDdigitsManager();
547   digitsManager->CreateArrays();
548
549   AliTRDrawStream input(rawReader);
550
551   AliRunLoader *runLoader = AliRunLoader::Instance();
552
553   // ----- preparing tracklet output -----
554   AliDataLoader *trklLoader = runLoader->GetLoader("TRDLoader")->GetDataLoader("tracklets");
555   if (!trklLoader) {
556     trklLoader = new AliDataLoader("TRD.Tracklets.root","tracklets", "tracklets");
557     runLoader->GetLoader("TRDLoader")->AddDataLoader(trklLoader);
558   }
559   if (!trklLoader) {
560     AliError("Could not get the tracklet data loader!");
561     return 0x0;
562   }
563   // trklLoader->SetEvent();
564   trklLoader->Load("update");
565   AliTreeLoader *trklTreeLoader = dynamic_cast<AliTreeLoader*> (trklLoader->GetBaseLoader("tracklets-raw"));
566   if (!trklTreeLoader) {
567     trklTreeLoader = new AliTreeLoader("tracklets-raw", trklLoader);
568     trklLoader->AddBaseLoader(trklTreeLoader);
569   }
570
571   TClonesArray *fTrackletArray = new TClonesArray("AliTRDtrackletWord", 256);
572   Int_t lastHC = -1;
573   if (trklLoader->Tree())
574     trklLoader->MakeTree();
575   TTree *trackletTree = trklTreeLoader->Tree();
576   if (!trackletTree) {
577     trklTreeLoader->MakeTree();
578     trackletTree = trklTreeLoader->Tree();
579   }
580   if (!trackletTree->GetBranch("hc"))
581     trackletTree->Branch("hc", &lastHC, "hc/I");
582   else
583     trackletTree->SetBranchAddress("hc", &lastHC);
584
585   if (!trackletTree->GetBranch("trkl"))
586     trackletTree->Branch("trkl", &fTrackletArray);
587   else
588     trackletTree->SetBranchAddress("trkl", &fTrackletArray);
589
590   // ----- preparing track output -----
591   AliDataLoader *trkLoader = 0x0;
592   trkLoader = runLoader->GetLoader("TRDLoader")->GetDataLoader("gtutracks");
593   if (!trkLoader) {
594     trkLoader = new AliDataLoader("TRD.GtuTracks.root","gtutracks", "gtutracks");
595     runLoader->GetLoader("TRDLoader")->AddDataLoader(trkLoader);
596   }
597   if (!trkLoader) {
598     AliError("Could not get the GTU-track data loader!");
599     return 0x0;
600   }
601   trkLoader->Load("UPDATE");
602
603   TTree *trackTree = trkLoader->Tree();
604   if (!trackTree) {
605     trkLoader->MakeTree();
606     trackTree = trkLoader->Tree();
607   }
608
609   AliTRDtrackGTU trk;
610   if (!trackTree->GetBranch("TRDtrackGTU"))
611     trackTree->Branch("TRDtrackGTU", "AliTRDtrackGTU", &trk, 32000);
612
613   // ----- read the raw data -----
614   // write the tracklets to arrays while reading raw data
615   input.SetTrackletArray(fTracklets);
616   input.SetTrackArray(fTracks);
617
618   // Loop through the digits
619   Int_t det    = 0;
620
621   while (det >= 0)
622     {
623       det = input.NextChamber(digitsManager);
624
625       if (det >= 0)
626         {
627           // get...
628           digits = (AliTRDarrayADC *) digitsManager->GetDigits(det);
629           track0 = (AliTRDarrayDictionary *) digitsManager->GetDictionary(det,0);
630           track1 = (AliTRDarrayDictionary *) digitsManager->GetDictionary(det,1);
631           track2 = (AliTRDarrayDictionary *) digitsManager->GetDictionary(det,2);
632           // and compress
633           if (digits) digits->Compress();  
634           if (track0) track0->Compress();   
635           if (track1) track1->Compress();     
636           if (track2) track2->Compress();
637         }
638     }
639
640   for (Int_t iSector = 0; iSector < fGeo->Nsector(); iSector++) {
641     fTrgFlags[iSector] = input.GetTriggerFlags(iSector);
642   }
643
644   // ----- tracklet output -----
645   fTrackletArray->Clear();
646   Int_t nTracklets = fTracklets ? fTracklets->GetEntries() : 0;
647   AliDebug(1, Form("Writing %i tracklets to loader", nTracklets));
648   for (Int_t iTracklet = 0; iTracklet < nTracklets; ++iTracklet) {
649     AliTRDtrackletWord *trkl = (AliTRDtrackletWord*) fTracklets->At(iTracklet);
650     if (trkl->GetHCId() != lastHC) {
651       if (fTrackletArray->GetEntriesFast() > 0) {
652         trackletTree->Fill();
653         fTrackletArray->Clear();
654       }
655       lastHC = trkl->GetHCId();
656     }
657     new ((*fTrackletArray)[fTrackletArray->GetEntriesFast()]) AliTRDtrackletWord(*trkl);
658   }
659   if (fTrackletArray->GetEntriesFast() > 0) {
660       trackletTree->Fill();
661       fTrackletArray->Clear();
662   }
663   trklTreeLoader->WriteData("OVERWRITE");
664   trklLoader->WriteData("OVERWRITE");
665   trklLoader->UnloadAll();
666   trklLoader->CloseFile();
667
668   // ----- track output -----
669   Int_t nTracks = fTracks ? fTracks->GetEntriesFast() : 0;
670   AliDebug(1, Form("Writing %i tracks to loader", nTracks));
671   for (Int_t iTrack = 0; iTrack < nTracks; ++iTrack) {
672     AliESDTrdTrack *trkEsd = (AliESDTrdTrack*) fTracks->At(iTrack);
673     trk = *trkEsd;
674     trackTree->Fill();
675   }
676   trkLoader->WriteData("OVERWRITE");
677   trkLoader->UnloadAll();
678   trkLoader->CloseFile();
679
680   return digitsManager;
681 }
682
683 //_____________________________________________________________________________
684 void AliTRDrawData::WriteIntermediateWords(UInt_t* buf, Int_t& nw, Int_t& of, const Int_t& maxSize, const Int_t& det, const Int_t& side) {
685     // 
686     // write tracklet end marker(0x10001000) 
687     // and half chamber headers(H[0] and H[1])
688     //
689     
690     Int_t        layer = fGeo->GetLayer( det );   // Layer
691     Int_t        stack = fGeo->GetStack( det );   // Stack
692     Int_t         sect = fGeo->GetSector( det );  // Sector (=iDDL)
693     Int_t           rv = fFee->GetRAWversion();
694     const Int_t kNTBin = fDigitsParam->GetNTimeBins(det);
695     Bool_t  trackletOn = fFee->GetTracklet();
696     UInt_t           x = 0;
697
698     // Write end of tracklet marker
699     if (nw < maxSize){
700       buf[nw++] = fgkEndOfTrackletMarker;
701       buf[nw++] = fgkEndOfTrackletMarker;     // the number of tracklet end marker should be more than 2
702     }
703     else {
704       of++;
705     }
706    
707     // Half Chamber header
708     // h[0] (there are 2 HC headers) xmmm mmmm nnnn nnnq qqss sssp ppcc ci01
709     //  , where  x : Raw version speacial number (=1)
710     //               m : Raw version major number (test pattern, ZS, disable tracklet, 0, options)
711     //               n : Raw version minor number
712     //               q : number of addtional header words (default = 1)
713     //               s : SM sector number (ALICE numbering)
714     //               p : plane(layer) number
715     //                   c : chamber(stack) number
716     //                   i : side number (0:A, 1:B)
717     Int_t majorv = 0;   // The major version number 
718     Int_t minorv = 0;   // The minor version number
719     Int_t add    = 1;   // The number of additional header words to follow : now 1, previous 2
720     Int_t tp     = 0;   // test pattern (default=0)
721     Int_t zs     = (rv==3) ? 1 : 0;                     // zero suppression
722     Int_t dt     = (trackletOn) ? 0 : 1;        // disable tracklet 
723     
724     majorv = (tp<<6) | (zs<<5) | (dt<<4) | 1;   // major version
725     
726     x = (1<<31) | (majorv<<24) | (minorv<<17) | (add<<14) | (sect<<9) | (layer<<6) | (stack<<3) | (side<<2) | 1;
727     if (nw < maxSize) buf[nw++] = x; else of++;
728     
729     // h[1]             tttt ttbb bbbb bbbb bbbb bbpp pphh hh01
730     // , where  t : number of time bins
731     //          b : bunch crossing number
732     //          p : pretrigger counter
733     //          h : pretrigger phase
734     Int_t bcCtr   = 99; // bunch crossing counter. Here it is set to 99 always for no reason
735     Int_t ptCtr   = 15; // pretrigger counter. Here it is set to 15 always for no reason
736     Int_t ptPhase = 11; // pretrigger phase. Here it is set to 11 always for no reason
737     //x = (bcCtr<<16) | (ptCtr<<12) | (ptPhase<<8) | ((kNTBin-1)<<2) | 1;       // old format
738     x = ((kNTBin)<<26) | (bcCtr<<10) | (ptCtr<<6) | (ptPhase<<2) | 1;
739     if (nw < maxSize) buf[nw++] = x; else of++;
740 }