added slewing correction by data
[u/mrichter/AliRoot.git] / TRD / 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   Bool_t newStack = kFALSE;  // new stack flag, for writing stack index words
191
192   // Get digits parameter
193   fDigitsParam = digitsManager->GetDigitsParam();
194
195   // sect is same as iDDL, so I use only sect here.
196   for (Int_t sect = 0; sect < fGeo->Nsector(); sect++) { 
197
198     char name[1024];
199     snprintf(name,1024,"TRD_%d.ddl",sect + AliTRDrawStream::kDDLOffset);
200
201     AliFstream* of = new AliFstream(name);
202
203     // Write a dummy data header
204     AliRawDataHeaderSim  header;  // the event header
205     UInt_t hpos = of->Tellp();
206     of->WriteBuffer((char *) (& header), sizeof(header));
207     
208     // Reset payload byte size (payload does not include header).
209     Int_t npayloadbyte = 0;
210
211     // check the existance of the data
212     // SM index word and Stack index word
213     UInt_t *iwbuffer = new UInt_t[109]; // index word buffer; max 109 = 2 SM headers + 67 dummy headers + 5*8 stack headers
214     Int_t nheader = 0;
215     UInt_t bStackMask = 0x0;
216     Bool_t bStackHasData = kFALSE;
217     Bool_t bSMHasData = kFALSE;
218     
219     //iwbuffer[nheader++] = 0x0001a020;   // SM index words 
220     iwbuffer[nheader++] = 0x0044b020;   // SM index words | additional SM header:48 = 1 SM header + 47 dummy words(for future use)
221     iwbuffer[nheader++] = 0x10404071;   // SM header
222     for ( Int_t i=0; i<66; i++ ) iwbuffer[nheader++] = 0x00000000;  // dummy words 
223     iwbuffer[nheader++] = 0x10000000;   // end of dummy words
224     
225     for ( Int_t stack= 0; stack < fGeo->Nstack(); stack++) {
226       UInt_t linkMask = 0x0;
227       for( Int_t layer = 0; layer < fGeo->Nlayer(); layer++) {
228         Int_t iDet = fGeo->GetDetector(layer,stack,sect);
229         AliTRDarrayADC *digits = (AliTRDarrayADC *) digitsManager->GetDigits(iDet);
230         if ( fgDataSuppressionLevel==0 || digits->HasData() ) {
231           bStackMask = bStackMask | ( 1 << stack ); // active stack mask for new stack
232           linkMask = linkMask | ( 3 << (2*layer) );    // 3 = 0011
233           bStackHasData = kTRUE;
234           bSMHasData = kTRUE;
235         } // has data
236       } // loop over layer
237       
238       if ( fgDataSuppressionLevel==0 || bStackHasData ){
239         iwbuffer[nheader++] = 0x0007a000 | linkMask;    // stack index word + link masks
240         if (fgDataSuppressionLevel==0) iwbuffer[nheader-1] = 0x0007afff;  // no suppression
241         iwbuffer[nheader++] = 0x04045b01;               // stack header
242         for (Int_t i=0;i<6;i++) iwbuffer[nheader++] = 0x00000000; // 6 dummy words
243         bStackHasData = kFALSE;
244       }
245     } // loop over stack
246     
247     if ( fgDataSuppressionLevel==0 || bSMHasData ){
248       iwbuffer[0] = iwbuffer[0] | bStackMask;  // add stack masks to SM index word
249       if (fgDataSuppressionLevel==0) iwbuffer[0] = 0x0044b03f;    // no suppression : all stacks are active
250       of->WriteBuffer((char *) iwbuffer, nheader*4);
251       AliDebug(11, Form("SM %d index word: %08x", sect, iwbuffer[0]));
252       AliDebug(11, Form("SM %d header: %08x", sect, iwbuffer[1]));
253     }
254     // end of SM & stack header ------------------------------------------------------------------------
255     // -------------------------------------------------------------------------------------------------
256     
257     // Prepare chamber data
258     for( Int_t stack = 0; stack < fGeo->Nstack(); stack++) {
259       for( Int_t layer = 0; layer < fGeo->Nlayer(); layer++) {
260         
261         Int_t iDet = fGeo->GetDetector(layer,stack,sect);
262         if (iDet == 0){
263           newEvent = kTRUE; // it is expected that each event has at least one tracklet; 
264           // this is only needed for correct readout tree
265           fEventCounter++;
266           AliDebug(11, Form("New event!! Event counter: %d",fEventCounter));
267         }
268         
269         if ( stack==0 && layer==0 ) newSM = kTRUE;  // new SM flag
270         if ( layer==0 ) newStack = kTRUE;           // new stack flag
271         AliDebug(15, Form("stack : %d, layer : %d, iDec : %d\n",stack,layer,iDet));
272         // Get the digits array
273         AliTRDarrayADC *digits = (AliTRDarrayADC *) digitsManager->GetDigits(iDet);
274         if (fgDataSuppressionLevel==0 || digits->HasData() ) {  // second part is new!! and is for indicating a new event
275           
276           if (digits->HasData()) digits->Expand();
277           
278           Int_t hcwords = 0;
279           
280           // Process A side of the chamber
281           hcwords = ProduceHcData(digits,0,iDet,hcBuffer,kMaxHcWords,newEvent,newSM);
282           if ( newEvent ) newEvent = kFALSE;
283           //AssignLinkMask(hcBuffer, layer);  // active link mask for this layer(2*HC)
284           of->WriteBuffer((char *) hcBuffer, hcwords*4);
285           npayloadbyte += hcwords*4;
286           
287           // Process B side of the chamber
288           hcwords = ProduceHcData(digits,1,iDet,hcBuffer,kMaxHcWords,newEvent,newSM);
289           of->WriteBuffer((char *) hcBuffer, hcwords*4);
290           npayloadbyte += hcwords*4;
291         } // has data
292         
293       } // loop over layer
294     } // loop over stack
295     
296     // Complete header
297     header.fSize = UInt_t(of->Tellp()) - hpos;
298     header.SetAttribute(0);  // Valid data
299     of->Seekp(hpos);         // Rewind to header position
300     of->WriteBuffer((char *) (& header), sizeof(header));
301     delete of;
302
303     delete [] iwbuffer;
304
305   } // loop over sector(SM)
306   
307   delete [] hcBuffer;
308   
309   return kTRUE;
310 }
311
312 //_____________________________________________________________________________
313 void AliTRDrawData::ProduceSMIndexData(UInt_t *buf, Int_t& nw){
314         // 
315         // This function generates 
316         //       1) SM index words : ssssssss ssssssss vvvv rrrr r d t mmmmm
317         //          - s : size of SM header (number of header, default = 0x0001)
318         //          - v : SM header version (default = 0xa)
319         //          - r : reserved for future use (default = 00000)
320         //          - d : track data enabled bit (default = 0)
321         //          - t : tracklet data enabled bit (default = 1)
322         //          - m : stack mask (each bit corresponds a stack, default = 11111)
323         //
324         //       2) SM header : rrr c vvvv vvvvvvvv vvvv rrrr bbbbbbbb
325         //          - r : reserved for future use (default = 000)
326         //          - c : clean check out flag (default = 1)
327         //          - v : hardware design revision (default = 0x0404)
328         //          - r : reserved for future use (default = 0x0)
329         //          - b : physical board ID (default = 0x71)
330         //
331         //       3) stack index words : ssssssss ssssssss vvvv mmmm mmmmmmmm
332         //          - s : size of stack header (number of header, (default = 0x0007)
333         //          - v : header version (default = 0xa)
334         //          - m : link mask (default = 0xfff)
335         //
336         //       4) stack header : vvvvvvvv vvvvvvvv bbbbbbbb rrrr rrr c
337         //          - v : hardware design revision (default = 0x0404)
338         //          - b : physical board ID (default = 0x5b)
339         //          - r : reserved for future use (default = 0000 000)
340         //          - c : clean checkout flag (default = 1)
341         //      
342         //       and 6 dummy words(0x00000000)
343         //
344         
345     //buf[nw++] = 0x0001a03f;   // SM index words
346     fSMindexPos = nw;       // memorize position of the SM index word for re-allocating stack mask
347     buf[nw++] = 0x0001b020; // SM index words
348     buf[nw++] = 0x10404071; // SM header
349
350     fStackindexPos = nw;    // memorize position of the stack index word for future adding
351         /*  
352     for (Int_t istack=0; istack<5; istack++){
353         buf[nw++] = 0x0007afff; // stack index words
354         buf[nw++] = 0x04045b01; // stack header
355         for (Int_t i=0;i<6;i++) buf[nw++] = 0x00000000; // 6 dummy words
356     } // loop over 5 stacks
357         */
358 }
359
360 //_____________________________________________________________________________
361 void AliTRDrawData::AssignStackMask(UInt_t *buf, Int_t nStack){
362     //
363     // This function re-assign stack mask active(from 0 to 1) in the SM index word
364     //   
365     buf[fSMindexPos] = buf[fSMindexPos] | ( 1 << nStack );
366 }
367
368 //_____________________________________________________________________________  
369 Int_t AliTRDrawData::AddStackIndexWords(UInt_t *buf, Int_t /*nStack*/, Int_t nMax){
370     // 
371     // This function add stack index words and stack header when there is data for the stack
372     //
373     //   1) stack index words : ssssssss ssssssss vvvv mmmm mmmmmmmm 
374     //      - s : size of stack header (number of header, (default = 0x0007)       
375     //      - v : header version (default = 0xa)
376     //      - m : link mask (default = 0xfff)
377     //      - m : link mask (starting value = 0x000)
378     //
379     //   2) stack header : vvvvvvvv vvvvvvvv bbbbbbbb rrrr rrr c
380     //      - v : hardware design revision (default = 0x0404)
381     //      - b : physical board ID (default = 0x5b)
382     //      - r : reserved for future use (default = 0000 000)
383     //      - c : clean checkout flag (default = 1)
384     //  
385     //   and 6 dummy words(0x00000000)
386     //
387
388     Int_t nAddedWords = 0;  // Number of added words
389     if ( ShiftWords(buf, fStackindexPos, 8, nMax)== kFALSE ){
390         AliError("Adding stack header failed.");
391         return 0;
392     }
393
394     buf[fStackindexPos++] = 0x0007a000; // stack index words
395     buf[fStackindexPos++] = 0x04045b01; // stack header
396     for (Int_t i=0;i<6;i++) buf[fStackindexPos++] = 0x00000000; // 6 dummy words 
397     nAddedWords += 8;
398
399     return nAddedWords;
400 }
401
402 //_____________________________________________________________________________
403 void AliTRDrawData::AssignLinkMask(UInt_t *buf, Int_t nLayer){
404     //
405     // This function re-assign link mask active(from 0 to 1) in the stack index word
406     //   
407     buf[fStackindexPos-8] = buf[fStackindexPos-8] | ( 3 << (2*nLayer) );    // 3 = 0011 
408 }
409
410 //_____________________________________________________________________________ 
411 Bool_t AliTRDrawData::ShiftWords(UInt_t *buf, Int_t nStart, Int_t nWords, Int_t nMax){
412     //  
413     // This function shifts n words
414     //
415     //if ( nStart+nWords > sizeof(buf)/sizeof(UInt_t) ){
416     //  AliError("Words shift failed. No more buffer space.");
417     //  return kFALSE;
418     //}
419
420     for ( Int_t iw=nMax; iw>nStart-1; iw--){
421         buf[iw+nWords] = buf[iw];
422     }
423     return kTRUE;
424 }
425
426 //_____________________________________________________________________________
427 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*/){
428         //
429         // This function produces the raw data for one HC, i.e. tracklets, tracklet endmarkers, 
430         // raw data, raw data endmarkers. 
431         // This function can be used for both ZS and NZS data
432         //
433
434         Int_t           nw = 0;                       // Number of written    words
435         Int_t           of = 0;                       // Number of overflowed words
436         Int_t      *tempnw = &nw;                     // Number of written    words for temp. buffer
437         Int_t      *tempof = &of;                     // Number of overflowed words for temp. buffer
438         Int_t        layer = fGeo->GetLayer( det );   // Layer
439         Int_t        stack = fGeo->GetStack( det );   // Stack
440         Int_t         sect = fGeo->GetSector( det );  // Sector (=iDDL)
441         const Int_t kCtype = fGeo->GetStack(det) == 2 ? 0 : 1;                       // Chamber type (0:C0, 1:C1)
442
443         Bool_t trackletOn = fFee->GetTracklet();     // tracklet simulation active?
444
445         AliDebug(1,Form("Producing raw data for sect=%d layer=%d stack=%d side=%d",sect,layer,stack,side));
446         
447         UInt_t *tempBuffer = buf; // tempBuffer used to write ADC data
448                                   // different in case of tracklet writing
449         
450         if (trackletOn) {
451           tempBuffer = new UInt_t[maxSize];
452           tempnw     = new Int_t(0);
453           tempof     = new Int_t(0);
454         }
455           
456         WriteIntermediateWords(tempBuffer,*tempnw,*tempof,maxSize,det,side);
457
458         if (digits->HasData()) {
459           // scanning direction such, that tracklet-words are sorted in ascending z and then in ascending y order
460           // ROB numbering on chamber and MCM numbering on ROB increase with decreasing z and increasing y
461           for (Int_t iRobRow = 0; iRobRow <= (kCtype + 3)-1; iRobRow++ ) {
462             // ROB number should be increasing
463             Int_t iRob = iRobRow * 2 + side;
464             // MCM on one ROB
465             for (Int_t iMcmRB = 0; iMcmRB < fGeo->MCMmax(); iMcmRB++ ) {
466               Int_t iMcm = 16 - 4*(iMcmRB/4 + 1) + (iMcmRB%4);
467               
468               fMcmSim->Init(det, iRob, iMcm);
469               fMcmSim->SetData(digits);     // no filtering done here (already done in digitizer)
470               if (trackletOn) {
471                 fMcmSim->Tracklet();
472                 Int_t tempNw = fMcmSim->ProduceTrackletStream(&buf[nw], maxSize - nw);
473                 if(  tempNw < 0 ) {
474                   of += tempNw;
475                   nw += maxSize - nw;
476                   AliError(Form("Buffer overflow detected. Please increase the buffer size and recompile."));
477                 } else {
478                   nw += tempNw;
479                 }
480               }
481               fMcmSim->ZSMapping();  // Calculate zero suppression mapping
482               // at the moment it has to be rerun here
483               // Write MCM data to temp. buffer
484               Int_t tempNw = fMcmSim->ProduceRawStream( &tempBuffer[*tempnw], maxSize - *tempnw, fEventCounter );
485               if ( tempNw < 0 ) {
486                 *tempof += tempNw;
487                 *tempnw += maxSize - nw;
488                 AliError(Form("Buffer overflow detected. Please increase the buffer size and recompile."));
489               } else {
490                 *tempnw += tempNw;
491               }
492             }
493           }
494           
495           // in case of tracklet writing copy temp data to final buffer
496           if (trackletOn) {
497             if (nw + *tempnw < maxSize) {
498               memcpy(&buf[nw], tempBuffer, *tempnw * sizeof(UInt_t));
499               nw += *tempnw;
500             }
501             else {
502               AliError("Buffer overflow detected");
503             }
504           }
505         }
506
507         if (trackletOn) {
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 }