Coverity
[u/mrichter/AliRoot.git] / TRD / AliTRDrawData.cxx
index 53f6a3b..81fcf7f 100644 (file)
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
-#include <Riostream.h>
+
 #include <TMath.h>
+#include "TClass.h"
 
 #include "AliDAQ.h"
-#include "AliRawDataHeader.h"
+#include "AliRawDataHeaderSim.h"
 #include "AliRawReader.h"
 #include "AliLog.h"
+#include "AliFstream.h"
+#include "AliLoader.h"
+#include "AliTreeLoader.h"
 
 #include "AliTRDrawData.h"
 #include "AliTRDdigitsManager.h"
 #include "AliTRDgeometry.h"
-#include "AliTRDdataArrayI.h"
-#include "AliTRDRawStream.h"
-
+#include "AliTRDarrayDictionary.h"
+#include "AliTRDarrayADC.h"
+#include "AliTRDrawStream.h"
 #include "AliTRDcalibDB.h"
-#include "AliFstream.h"
-
 #include "AliTRDSignalIndex.h"
+#include "AliTRDfeeParam.h"
+#include "AliTRDmcmSim.h"
+#include "AliTRDtrackletWord.h"
+#include "AliTRDdigitsParam.h"
+
 ClassImp(AliTRDrawData)
 
+Int_t AliTRDrawData::fgDataSuppressionLevel = 0;
+
 //_____________________________________________________________________________
 AliTRDrawData::AliTRDrawData()
   :TObject()
-  ,fRawVersion(2)    // Default Raw Data version set here
-  ,fGeo(0)
+  ,fRunLoader(NULL)
+  ,fGeo(NULL)
+  ,fFee(NULL)
   ,fNumberOfDDLs(0)
+  ,fTrackletTree(NULL)
+  ,fTracklets(NULL)
+  ,fTracks(NULL)
+  ,fSMindexPos(0)
+  ,fStackindexPos(0)
+  ,fEventCounter(0)
+  ,fTrgFlags()
+  ,fMcmSim(new AliTRDmcmSim)
+  ,fDigitsParam(NULL)
 {
   //
   // Default constructor
   //
 
+  fFee = AliTRDfeeParam::Instance();
+  fNumberOfDDLs = AliDAQ::NumberOfDdls("TRD");
+
 }
 
 //_____________________________________________________________________________
 AliTRDrawData::AliTRDrawData(const AliTRDrawData &r)
   :TObject(r)
-  ,fRawVersion(2)    // Default Raw Data version set here
-  ,fGeo(0)
+  ,fRunLoader(NULL)
+  ,fGeo(NULL)
+  ,fFee(NULL)
   ,fNumberOfDDLs(0)
+  ,fTrackletTree(NULL)
+  ,fTracklets(NULL)
+  ,fTracks(NULL)
+  ,fSMindexPos(0)
+  ,fStackindexPos(0)
+  ,fEventCounter(0)
+  ,fTrgFlags()
+  ,fMcmSim(new AliTRDmcmSim)
+  ,fDigitsParam(NULL)
 {
   //
   // Copy constructor
   //
 
+  fFee = AliTRDfeeParam::Instance();
+  fNumberOfDDLs = AliDAQ::NumberOfDdls("TRD");
+
 }
 
 //_____________________________________________________________________________
@@ -74,26 +109,22 @@ AliTRDrawData::~AliTRDrawData()
   // Destructor
   //
 
-}
-
-//_____________________________________________________________________________
-Bool_t AliTRDrawData::SetRawVersion(Int_t v)
-{
-  //
-  // Set the raw data version (Currently only version 0, 1 and 2 are available)
-  //
+  if (fTracklets){
+    fTracklets->Delete();
+    delete fTracklets;
+  }
 
-  if ( (v >= 0) && (v <= 2) ) {
-    fRawVersion = v;
-    return kTRUE;
+  if (fTracks){
+    fTracks->Delete();
+    delete fTracks;
   }
 
-  return kFALSE;
+  delete fMcmSim;
 
 }
 
 //_____________________________________________________________________________
-Bool_t AliTRDrawData::Digits2Raw(TTree *digitsTree, TTree *tracks )
+Bool_t AliTRDrawData::Digits2Raw(TTree *digitsTree, const TTree *tracks )
 {
   //
   // Initialize necessary parameters and call one
@@ -103,9 +134,7 @@ Bool_t AliTRDrawData::Digits2Raw(TTree *digitsTree, TTree *tracks )
   // will be supported in higher version simulator.
   //
 
-  fNumberOfDDLs = AliDAQ::NumberOfDdls("TRD");
-
-  AliTRDdigitsManager* digitsManager = new AliTRDdigitsManager();
+  AliTRDdigitsManager* const digitsManager = new AliTRDdigitsManager();
 
   if (!digitsManager->ReadDigits(digitsTree)) {
     delete digitsManager;
@@ -128,12 +157,13 @@ Bool_t AliTRDrawData::Digits2Raw(TTree *digitsTree, TTree *tracks )
   }
 
   Int_t retval = kTRUE;
+  Int_t rv     = fFee->GetRAWversion();
 
   // Call appropriate Raw Simulator
-  if ( fRawVersion > 0 && fRawVersion <= 2 ) retval = Digits2Raw(digitsManager); 
+  if ( rv > 0 && rv <= 3 ) retval = Digits2Raw(digitsManager); 
   else {
     retval = kFALSE;
-    AliWarning(Form("Unsupported raw version (fRawVersion=%d).",fRawVersion));
+    AliWarning(Form("Unsupported raw version (%d).", rv));
   }
 
   // Cleanup
@@ -152,261 +182,355 @@ Bool_t AliTRDrawData::Digits2Raw(AliTRDdigitsManager *digitsManager)
   // This version simulate only raw data with ADC data and not with tracklet.
   //
 
-  const Int_t kMaxHcWords = (fGeo->TBmax()/3)*fGeo->ADCmax()*fGeo->MCMmax()*fGeo->ROBmaxC1()/2 + 100 + 20;
+  const Int_t kMaxHcWords = (fGeo->TBmax()/3)
+                          * fGeo->ADCmax()
+                          * fGeo->MCMmax()
+                          * fGeo->ROBmaxC1()/2 
+                          + 100 + 20;
 
   // Buffer to temporary store half chamber data
-  UInt_t     *hc_buffer   = new UInt_t[kMaxHcWords];
+  UInt_t     *hcBuffer    = new UInt_t[kMaxHcWords];
+
+  Bool_t newEvent = kFALSE;  // only for correct readout tree
+  Bool_t newSM    = kFALSE;  // new SM flag, for writing SM index words
+  Bool_t newStack = kFALSE;  // new stack flag, for writing stack index words
+
+  // Get digits parameter
+  fDigitsParam = digitsManager->GetDigitsParam();
 
   // sect is same as iDDL, so I use only sect here.
-  for (Int_t sect = 0; sect < fGeo->Nsect(); sect++) { 
+  for (Int_t sect = 0; sect < fGeo->Nsector(); sect++) { 
 
     char name[1024];
-    sprintf(name,"TRD_%d.ddl",sect + AliTRDRawStream::kDDLOffset);
+    snprintf(name,1024,"TRD_%d.ddl",sect + AliTRDrawStream::kDDLOffset);
 
     AliFstream* of = new AliFstream(name);
 
     // Write a dummy data header
-    AliRawDataHeader  header;  // the event header
+    AliRawDataHeaderSim  header;  // the event header
     UInt_t hpos = of->Tellp();
     of->WriteBuffer((char *) (& header), sizeof(header));
-
+    
     // Reset payload byte size (payload does not include header).
     Int_t npayloadbyte = 0;
 
-    // GTU common data header (5x4 bytes per super module, shows link mask)
-    for( Int_t cham = 0; cham < fGeo->Ncham(); cham++ ) {
-      UInt_t GtuCdh = (UInt_t)(0xe << 28);
-      for( Int_t plan = 0; plan < fGeo->Nplan(); plan++) {
-       Int_t iDet = fGeo->GetDetector(plan, cham, sect);
-       // If chamber status is ok, we assume that the optical link is also OK.
-        // This is shown in the GTU link mask.
-       if ( AliTRDcalibDB::Instance()->GetChamberStatus(iDet) )
-         GtuCdh = GtuCdh | (3 << (2*plan));
+    // check the existance of the data
+    // SM index word and Stack index word
+    UInt_t *iwbuffer = new UInt_t[109]; // index word buffer; max 109 = 2 SM headers + 67 dummy headers + 5*8 stack headers
+    Int_t nheader = 0;
+    UInt_t bStackMask = 0x0;
+    Bool_t bStackHasData = kFALSE;
+    Bool_t bSMHasData = kFALSE;
+    
+    //iwbuffer[nheader++] = 0x0001a020;   // SM index words 
+    iwbuffer[nheader++] = 0x0044b020;   // SM index words | additional SM header:48 = 1 SM header + 47 dummy words(for future use)
+    iwbuffer[nheader++] = 0x10404071;   // SM header
+    for ( Int_t i=0; i<66; i++ ) iwbuffer[nheader++] = 0x00000000;  // dummy words 
+    iwbuffer[nheader++] = 0x10000000;   // end of dummy words
+    
+    for ( Int_t stack= 0; stack < fGeo->Nstack(); stack++) {
+      UInt_t linkMask = 0x0;
+      for( Int_t layer = 0; layer < fGeo->Nlayer(); layer++) {
+       Int_t iDet = fGeo->GetDetector(layer,stack,sect);
+       AliTRDarrayADC *digits = (AliTRDarrayADC *) digitsManager->GetDigits(iDet);
+       if ( fgDataSuppressionLevel==0 || digits->HasData() ) {
+         bStackMask = bStackMask | ( 1 << stack ); // active stack mask for new stack
+         linkMask = linkMask | ( 3 << (2*layer) );    // 3 = 0011
+         bStackHasData = kTRUE;
+         bSMHasData = kTRUE;
+       } // has data
+      } // loop over layer
+      
+      if ( fgDataSuppressionLevel==0 || bStackHasData ){
+       iwbuffer[nheader++] = 0x0007a000 | linkMask;    // stack index word + link masks
+       if (fgDataSuppressionLevel==0) iwbuffer[nheader-1] = 0x0007afff;  // no suppression
+       iwbuffer[nheader++] = 0x04045b01;               // stack header
+       for (Int_t i=0;i<6;i++) iwbuffer[nheader++] = 0x00000000; // 6 dummy words
+       bStackHasData = kFALSE;
       }
-      of->WriteBuffer((char *) (& GtuCdh), sizeof(GtuCdh));
-      npayloadbyte += 4;
+    } // loop over stack
+    
+    if ( fgDataSuppressionLevel==0 || bSMHasData ){
+      iwbuffer[0] = iwbuffer[0] | bStackMask;  // add stack masks to SM index word
+      if (fgDataSuppressionLevel==0) iwbuffer[0] = 0x0044b03f;    // no suppression : all stacks are active
+      of->WriteBuffer((char *) iwbuffer, nheader*4);
+      AliDebug(11, Form("SM %d index word: %08x", sect, iwbuffer[0]));
+      AliDebug(11, Form("SM %d header: %08x", sect, iwbuffer[1]));
     }
-
+    // end of SM & stack header ------------------------------------------------------------------------
+    // -------------------------------------------------------------------------------------------------
+    
     // Prepare chamber data
-    for( Int_t cham = 0; cham < fGeo->Ncham(); cham++) {
-      for( Int_t plan = 0; plan < fGeo->Nplan(); plan++) {
-
-        Int_t iDet = fGeo->GetDetector(plan,cham,sect);
-
-        // Get the digits array
-        AliTRDdataArrayI *digits = digitsManager->GetDigits(iDet);
-        digits->Expand();
-
-        Int_t hcwords = 0;
-
-        // Process A side of the chamber
-       if ( fRawVersion >= 1 && fRawVersion <= 2 ) hcwords = ProduceHcDataV1andV2(digits,0,iDet,hc_buffer,kMaxHcWords);
-        of->WriteBuffer((char *) hc_buffer, hcwords*4);
-        npayloadbyte += hcwords*4;
-
-        // Process B side of the chamber
-       if ( fRawVersion >= 1 && fRawVersion <= 2 ) hcwords = ProduceHcDataV1andV2(digits,1,iDet,hc_buffer,kMaxHcWords);
-        of->WriteBuffer((char *) hc_buffer, hcwords*4);
-        npayloadbyte += hcwords*4;
-
-      }
-    }
-
+    for( Int_t stack = 0; stack < fGeo->Nstack(); stack++) {
+      for( Int_t layer = 0; layer < fGeo->Nlayer(); layer++) {
+       
+       Int_t iDet = fGeo->GetDetector(layer,stack,sect);
+       if (iDet == 0){
+         newEvent = kTRUE; // it is expected that each event has at least one tracklet; 
+         // this is only needed for correct readout tree
+         fEventCounter++;
+         AliDebug(11, Form("New event!! Event counter: %d",fEventCounter));
+       }
+       
+       if ( stack==0 && layer==0 ) newSM = kTRUE;  // new SM flag
+       if ( layer==0 ) newStack = kTRUE;           // new stack flag
+       AliDebug(15, Form("stack : %d, layer : %d, iDec : %d\n",stack,layer,iDet));
+       // Get the digits array
+       AliTRDarrayADC *digits = (AliTRDarrayADC *) digitsManager->GetDigits(iDet);
+       if (fgDataSuppressionLevel==0 || digits->HasData() ) {  // second part is new!! and is for indicating a new event
+         
+         if (digits->HasData()) digits->Expand();
+         
+         Int_t hcwords = 0;
+         
+         // Process A side of the chamber
+         hcwords = ProduceHcData(digits,0,iDet,hcBuffer,kMaxHcWords,newEvent,newSM);
+         if ( newEvent ) newEvent = kFALSE;
+         //AssignLinkMask(hcBuffer, layer);  // active link mask for this layer(2*HC)
+         of->WriteBuffer((char *) hcBuffer, hcwords*4);
+         npayloadbyte += hcwords*4;
+         
+         // Process B side of the chamber
+         hcwords = ProduceHcData(digits,1,iDet,hcBuffer,kMaxHcWords,newEvent,newSM);
+         of->WriteBuffer((char *) hcBuffer, hcwords*4);
+         npayloadbyte += hcwords*4;
+       } // has data
+       
+      } // loop over layer
+    } // loop over stack
+    
     // Complete header
     header.fSize = UInt_t(of->Tellp()) - hpos;
     header.SetAttribute(0);  // Valid data
     of->Seekp(hpos);         // Rewind to header position
     of->WriteBuffer((char *) (& header), sizeof(header));
     delete of;
-  }
 
-  delete [] hc_buffer;
+    delete [] iwbuffer;
+
+  } // loop over sector(SM)
+  
+  delete [] hcBuffer;
+  
   return kTRUE;
+}
 
+//_____________________________________________________________________________
+void AliTRDrawData::ProduceSMIndexData(UInt_t *buf, Int_t& nw){
+       // 
+       // This function generates 
+       //       1) SM index words : ssssssss ssssssss vvvv rrrr r d t mmmmm
+       //          - s : size of SM header (number of header, default = 0x0001)
+       //          - v : SM header version (default = 0xa)
+       //          - r : reserved for future use (default = 00000)
+       //          - d : track data enabled bit (default = 0)
+       //          - t : tracklet data enabled bit (default = 1)
+       //          - m : stack mask (each bit corresponds a stack, default = 11111)
+       //
+       //       2) SM header : rrr c vvvv vvvvvvvv vvvv rrrr bbbbbbbb
+       //          - r : reserved for future use (default = 000)
+       //          - c : clean check out flag (default = 1)
+       //          - v : hardware design revision (default = 0x0404)
+       //          - r : reserved for future use (default = 0x0)
+       //          - b : physical board ID (default = 0x71)
+       //
+       //       3) stack index words : ssssssss ssssssss vvvv mmmm mmmmmmmm
+       //          - s : size of stack header (number of header, (default = 0x0007)
+       //          - v : header version (default = 0xa)
+       //          - m : link mask (default = 0xfff)
+       //
+       //       4) stack header : vvvvvvvv vvvvvvvv bbbbbbbb rrrr rrr c
+       //          - v : hardware design revision (default = 0x0404)
+       //          - b : physical board ID (default = 0x5b)
+       //          - r : reserved for future use (default = 0000 000)
+       //          - c : clean checkout flag (default = 1)
+       //      
+       //       and 6 dummy words(0x00000000)
+       //
+       
+    //buf[nw++] = 0x0001a03f;   // SM index words
+    fSMindexPos = nw;       // memorize position of the SM index word for re-allocating stack mask
+    buf[nw++] = 0x0001b020; // SM index words
+    buf[nw++] = 0x10404071; // SM header
+
+    fStackindexPos = nw;    // memorize position of the stack index word for future adding
+       /*  
+    for (Int_t istack=0; istack<5; istack++){
+        buf[nw++] = 0x0007afff; // stack index words
+        buf[nw++] = 0x04045b01; // stack header
+        for (Int_t i=0;i<6;i++) buf[nw++] = 0x00000000; // 6 dummy words
+    } // loop over 5 stacks
+       */
 }
 
 //_____________________________________________________________________________
-Int_t AliTRDrawData::ProduceHcDataV1andV2(AliTRDdataArrayI *digits, Int_t side
-                                        , Int_t det, UInt_t *buf, Int_t maxSize)
-{
-  //
-  // This function simulates: 1) SM-I commissiong data Oct. 06 (fRawVersion == 1).
-  //                          2) Full Raw Production Version   (fRawVersion == 2)
-  //
-  // Produce half chamber data (= an ORI data) for the given chamber (det) and side (side)
-  // where
-  //
-  //   side=0 means A side with ROB positions 0, 2, 4, 6.
-  //   side=1 means B side with ROB positions 1, 3, 5, 7.
-  //
-  // Chamber type (C0 orC1) is determined by "det" automatically.
-  // Appropriate size of buffer (*buf) must be prepared prior to calling this function.
-  // Pointer to the buffer and its size must be given to "buf" and "maxSize".
-  // Return value is the number of valid data filled in the buffer in unit of 32 bits
-  // UInt_t words.
-  // If buffer size if too small, the data is truncated with the buffer size however
-  // the function will finish without crash (this behaviour is similar to the MCM).
-  //
+void AliTRDrawData::AssignStackMask(UInt_t *buf, Int_t nStack){
+    //
+    // This function re-assign stack mask active(from 0 to 1) in the SM index word
+    //   
+    buf[fSMindexPos] = buf[fSMindexPos] | ( 1 << nStack );
+}
 
-  Int_t          nw = 0;                       // Number of written    words
-  Int_t          of = 0;                       // Number of overflowed words
-  Int_t        plan = fGeo->GetPlane( det );   // Plane
-  Int_t        cham = fGeo->GetChamber( det ); // Chamber
-  Int_t        sect = fGeo->GetSector( det );  // Sector (=iDDL)
-  Int_t        nRow = fGeo->GetRowMax( plan, cham, sect );
-  Int_t        nCol = fGeo->GetColMax( plan );
-  const Int_t nTBin = AliTRDcalibDB::Instance()->GetNumberOfTimeBins();
-  Int_t      kCtype = 0;                       // Chamber type (0:C0, 1:C1)
-  Int_t         iEv = 0xA;                     // Event ID. Now fixed to 10, how do I get event id?
-  UInt_t          x = 0;                       // General used number
-
-  // Check the nCol and nRow.
-  if ((nCol == 144) && 
-      (nRow == 16 || nRow == 12)) {
-    kCtype = (nRow-12) / 4;
-  } 
-  else {
-    AliError(Form("This type of chamber is not supported (nRow=%d, nCol=%d)."
-                 ,nRow,nCol));
-    return 0;
-  }
+//_____________________________________________________________________________  
+Int_t AliTRDrawData::AddStackIndexWords(UInt_t *buf, Int_t /*nStack*/, Int_t nMax){
+    // 
+    // This function add stack index words and stack header when there is data for the stack
+    //
+    //   1) stack index words : ssssssss ssssssss vvvv mmmm mmmmmmmm 
+    //      - s : size of stack header (number of header, (default = 0x0007)       
+    //      - v : header version (default = 0xa)
+    //      - m : link mask (default = 0xfff)
+    //      - m : link mask (starting value = 0x000)
+    //
+    //   2) stack header : vvvvvvvv vvvvvvvv bbbbbbbb rrrr rrr c
+    //      - v : hardware design revision (default = 0x0404)
+    //      - b : physical board ID (default = 0x5b)
+    //      - r : reserved for future use (default = 0000 000)
+    //      - c : clean checkout flag (default = 1)
+    //  
+    //   and 6 dummy words(0x00000000)
+    //
+
+    Int_t nAddedWords = 0;  // Number of added words
+    if ( ShiftWords(buf, fStackindexPos, 8, nMax)== kFALSE ){
+        AliError("Adding stack header failed.");
+        return 0;
+    }
 
-  AliDebug(1,Form("Producing raw data for sect=%d plan=%d cham=%d side=%d"
-                 ,sect,plan,cham,side));
+    buf[fStackindexPos++] = 0x0007a000; // stack index words
+    buf[fStackindexPos++] = 0x04045b01; // stack header
+    for (Int_t i=0;i<6;i++) buf[fStackindexPos++] = 0x00000000; // 6 dummy words 
+    nAddedWords += 8;
 
-  // Tracklet should be processed here but not implemented yet
+    return nAddedWords;
+}
 
-  // Write end of tracklet marker
-  if (nw < maxSize) {
-    buf[nw++] = kEndoftrackletmarker;
-  } 
-  else {
-    of++;
-  }
+//_____________________________________________________________________________
+void AliTRDrawData::AssignLinkMask(UInt_t *buf, Int_t nLayer){
+    //
+    // This function re-assign link mask active(from 0 to 1) in the stack index word
+    //   
+    buf[fStackindexPos-8] = buf[fStackindexPos-8] | ( 3 << (2*nLayer) );    // 3 = 0011 
+}
 
-  // Half Chamber header
-  if      ( fRawVersion == 1 ) {
-    // Now it is the same version as used in SM-I commissioning.
-    Int_t  dcs = det+100;      // DCS Serial (in simulation, it is meaningless
-    x = (dcs<<20) | (sect<<15) | (plan<<12) | (cham<<9) | (side<<8) | 1;
-    if (nw < maxSize) {
-      buf[nw++] = x; 
-    }
-    else {
-      of++;
-    }
-  } 
-  else if ( fRawVersion == 2 ) {
-    // h[0] (there are 3 HC header)
-    Int_t minorv = 0;      // The minor version number
-    Int_t add    = 1;      // The number of additional header words to follow
-    x = (1<<31) | (fRawVersion<<24) | (minorv<<17) | (add<<14) | (sect<<9) | (plan<<6) | (cham<<3) | (side<<2) | 1;
-    if (nw < maxSize) {
-      buf[nw++] = x; 
-    }
-    else {
-      of++;
-    }
-    // h[1]
-    Int_t bc_ctr   = 99; // bunch crossing counter. Here it is set to 99 always for no reason
-    Int_t pt_ctr   = 15; // pretrigger counter. Here it is set to 15 always for no reason
-    Int_t pt_phase = 11; // pretrigger phase. Here it is set to 11 always for no reason
-    x = (bc_ctr<<16) | (pt_ctr<<12) | (pt_phase<<8) | ((nTBin-1)<<2) | 1;
-    if (nw < maxSize) {
-      buf[nw++] = x; 
-    }
-    else {
-      of++;
-    }
-    // h[2]
-    Int_t ped_setup       = 1;    // Pedestal filter setup (0:1). Here it is always 1 for no reason
-    Int_t gain_setup      = 1;    // Gain filter setup (0:1). Here it is always 1 for no reason
-    Int_t tail_setup      = 1;    // Tail filter setup (0:1). Here it is always 1 for no reason
-    Int_t xt_setup        = 0;    // Cross talk filter setup (0:1). Here it is always 0 for no reason
-    Int_t nonlin_setup    = 0;    // Nonlinearity filter setup (0:1). Here it is always 0 for no reason
-    Int_t bypass_setup    = 0;    // Filter bypass (for raw data) setup (0:1). Here it is always 0 for no reason
-    Int_t common_additive = 10;   // Digital filter common additive (0:63). Here it is always 10 for no reason
-    x = (ped_setup<<31) | (gain_setup<<30) | (tail_setup<<29) | (xt_setup<<28) | (nonlin_setup<<27)
-      | (bypass_setup<<26) | (common_additive<<20) | 1;
-    if (nw < maxSize) {
-      buf[nw++] = x; 
-    }
-    else {
-      of++;
+//_____________________________________________________________________________ 
+Bool_t AliTRDrawData::ShiftWords(UInt_t *buf, Int_t nStart, Int_t nWords, Int_t nMax){
+    //  
+    // This function shifts n words
+    //
+    //if ( nStart+nWords > sizeof(buf)/sizeof(UInt_t) ){
+    //  AliError("Words shift failed. No more buffer space.");
+    //  return kFALSE;
+    //}
+
+    for ( Int_t iw=nMax; iw>nStart-1; iw--){
+        buf[iw+nWords] = buf[iw];
     }
-  }
-
-  // Scan for ROB and MCM
-  for (Int_t iRobRow = 0; iRobRow < (kCtype + 3); iRobRow++ ) {
-    Int_t iRob = iRobRow * 2 + side;
-    for (Int_t iMcm = 0; iMcm < fGeo->MCMmax(); iMcm++ ) {
-      Int_t padrow = iRobRow * 4 + iMcm / 4;
-
-      // MCM header
-      x = ((iRob * fGeo->MCMmax() + iMcm) << 24) | ((iEv % 0x100000) << 4) | 0xC;
-      if (nw < maxSize) {
-        buf[nw++] = x; 
-      }
-      else {
-        of++;
-      }
+    return kTRUE;
+}
 
-      // ADC data
-      for (Int_t iAdc = 0; iAdc < 21; iAdc++ ) {
-       Int_t padcol = fGeo->GetPadColFromADC(iRob, iMcm, iAdc);
-       UInt_t aa = !(iAdc & 1) + 2;
-        UInt_t *a = new UInt_t[nTBin+2];
-        // 3 timebins are packed into one 32 bits word
-        for (Int_t iT = 0; iT < nTBin; iT+=3) { 
-          if ((padcol >=    0) && (padcol <  nCol)) {
-           a[iT  ] = ((iT    ) < nTBin ) ? digits->GetDataUnchecked(padrow,padcol,iT    ) : 0;
-           a[iT+1] = ((iT + 1) < nTBin ) ? digits->GetDataUnchecked(padrow,padcol,iT + 1) : 0;
-           a[iT+2] = ((iT + 2) < nTBin ) ? digits->GetDataUnchecked(padrow,padcol,iT + 2) : 0; 
-         } 
-         else {
-           a[iT] = a[iT+1] = a[iT+2] = 0; // This happenes at the edge of chamber (should be pedestal! How?)
-         }
-         x = (a[iT+2] << 22) | (a[iT+1] << 12) | (a[iT] << 2) | aa;
-         if (nw < maxSize) {
-           buf[nw++] = x; 
+//_____________________________________________________________________________
+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*/){
+       //
+        // This function produces the raw data for one HC, i.e. tracklets, tracklet endmarkers, 
+        // raw data, raw data endmarkers. 
+       // This function can be used for both ZS and NZS data
+       //
+
+       Int_t           nw = 0;                       // Number of written    words
+       Int_t           of = 0;                       // Number of overflowed words
+       Int_t      *tempnw = &nw;                     // Number of written    words for temp. buffer
+       Int_t      *tempof = &of;                     // Number of overflowed words for temp. buffer
+       Int_t        layer = fGeo->GetLayer( det );   // Layer
+       Int_t        stack = fGeo->GetStack( det );   // Stack
+       Int_t         sect = fGeo->GetSector( det );  // Sector (=iDDL)
+       const Int_t kCtype = fGeo->GetStack(det) == 2 ? 0 : 1;                       // Chamber type (0:C0, 1:C1)
+
+       Bool_t trackletOn = fFee->GetTracklet();     // tracklet simulation active?
+
+       AliDebug(1,Form("Producing raw data for sect=%d layer=%d stack=%d side=%d",sect,layer,stack,side));
+        
+       UInt_t *tempBuffer = buf; // tempBuffer used to write ADC data
+                                 // different in case of tracklet writing
+       
+       if (trackletOn) {
+         tempBuffer = new UInt_t[maxSize];
+         tempnw     = new Int_t(0);
+         tempof     = new Int_t(0);
+       }
+         
+       WriteIntermediateWords(tempBuffer,*tempnw,*tempof,maxSize,det,side);
+
+       if (digits->HasData()) {
+         // scanning direction such, that tracklet-words are sorted in ascending z and then in ascending y order
+         // ROB numbering on chamber and MCM numbering on ROB increase with decreasing z and increasing y
+         for (Int_t iRobRow = 0; iRobRow <= (kCtype + 3)-1; iRobRow++ ) {
+           // ROB number should be increasing
+           Int_t iRob = iRobRow * 2 + side;
+           // MCM on one ROB
+           for (Int_t iMcmRB = 0; iMcmRB < fGeo->MCMmax(); iMcmRB++ ) {
+             Int_t iMcm = 16 - 4*(iMcmRB/4 + 1) + (iMcmRB%4);
+             
+             fMcmSim->Init(det, iRob, iMcm);
+             fMcmSim->SetData(digits);     // no filtering done here (already done in digitizer)
+             if (trackletOn) {
+               fMcmSim->Tracklet();
+               Int_t tempNw = fMcmSim->ProduceTrackletStream(&buf[nw], maxSize - nw);
+               if(  tempNw < 0 ) {
+                 of += tempNw;
+                 nw += maxSize - nw;
+                 AliError(Form("Buffer overflow detected. Please increase the buffer size and recompile."));
+               } else {
+                 nw += tempNw;
+               }
+             }
+             fMcmSim->ZSMapping();  // Calculate zero suppression mapping
+             // at the moment it has to be rerun here
+             // Write MCM data to temp. buffer
+             Int_t tempNw = fMcmSim->ProduceRawStream( &tempBuffer[*tempnw], maxSize - *tempnw, fEventCounter );
+             if ( tempNw < 0 ) {
+               *tempof += tempNw;
+               *tempnw += maxSize - nw;
+               AliError(Form("Buffer overflow detected. Please increase the buffer size and recompile."));
+             } else {
+               *tempnw += tempNw;
+             }
+           }
          }
-         else {
-           of++;
+         
+         // in case of tracklet writing copy temp data to final buffer
+         if (trackletOn) {
+           if (nw + *tempnw < maxSize) {
+             memcpy(&buf[nw], tempBuffer, *tempnw * sizeof(UInt_t));
+             nw += *tempnw;
+           }
+           else {
+             AliError("Buffer overflow detected");
+           }
          }
        }
-        // Diagnostics
-        Float_t avg = 0;
-        Float_t rms = 0;
-        for (Int_t iT = 0; iT < nTBin; iT++) {
-          avg += (Float_t) (a[iT]);
-       }
-        avg /= (Float_t) nTBin;
-        for (Int_t iT = 0; iT < nTBin; iT++) {
-          rms += ((Float_t) (a[iT]) - avg) * ((Float_t) (a[iT]) - avg);
-       }
-        rms = TMath::Sqrt(rms / (Float_t) nTBin);
-        if (rms > 1.7) {
-          AliDebug(2,Form("Large RMS (>1.7)  (ROB,MCM,ADC)=(%02d,%02d,%02d), avg=%03.1f, rms=%03.1f"
-                         ,iRob,iMcm,iAdc,avg,rms));
-       }
-        delete [] a;
-      }
-    }
-  }
-
-  // Write end of raw data marker
-  if (nw < maxSize) {
-    buf[nw++] = kEndofrawdatamarker; 
-  }
-  else {
-    of++;
-  }
-  if (of != 0) {
-    AliWarning("Buffer overflow. Data is truncated. Please increase buffer size and recompile.");
-  }
-
-  return nw;
 
+        if (trackletOn) {
+         delete [] tempBuffer;
+         delete tempof;
+          delete tempnw;
+       }
+    
+       // Write end of raw data marker
+       if (nw+3 < maxSize) {
+          buf[nw++] = fgkEndOfDataMarker;
+          buf[nw++] = fgkEndOfDataMarker;
+          buf[nw++] = fgkEndOfDataMarker;
+          buf[nw++] = fgkEndOfDataMarker;
+       } else {
+          of += 4;
+       }
+       
+       if (of != 0) {
+          AliError("Buffer overflow. Data is truncated. Please increase buffer size and recompile.");
+       }
+
+       return nw;
 }
 
 //_____________________________________________________________________________
@@ -416,90 +540,151 @@ AliTRDdigitsManager *AliTRDrawData::Raw2Digits(AliRawReader *rawReader)
   // Vx of the raw data reading
   //
 
-  AliTRDdataArrayI *digits = 0;
-  AliTRDdataArrayI *track0 = 0;
-  AliTRDdataArrayI *track1 = 0;
-  AliTRDdataArrayI *track2 = 0; 
+  rawReader->Select("TRD"); //[mj]
 
-  AliTRDSignalIndex *indexes = 0;
+  AliTRDarrayADC *digits = 0;
+  AliTRDarrayDictionary *track0 = 0;
+  AliTRDarrayDictionary *track1 = 0;
+  AliTRDarrayDictionary *track2 = 0;  
+
+  //AliTRDSignalIndex *indexes = 0;
   // Create the digits manager
   AliTRDdigitsManager* digitsManager = new AliTRDdigitsManager();
   digitsManager->CreateArrays();
 
-  AliTRDRawStream input(rawReader);
-  input.SetRawVersion( fRawVersion );
-  input.Init();
+  AliTRDrawStream input(rawReader);
 
-  // Loop through the digits
-  Int_t lastdet = -1;
-  Int_t det    = 0;
-  Int_t it = 0;
-  while (input.Next()) 
-    {
+  // ----- preparing tracklet output -----
+  AliDataLoader *trklLoader = AliRunLoader::Instance()->GetLoader("TRDLoader")->GetDataLoader("tracklets");
+  if (!trklLoader) {
+    trklLoader = new AliDataLoader("TRD.Tracklets.root","tracklets", "tracklets");
+    AliRunLoader::Instance()->GetLoader("TRDLoader")->AddDataLoader(trklLoader);
+  }
+  if (!trklLoader) {
+    return 0x0;
+  }
+  AliTreeLoader *trklTreeLoader = dynamic_cast<AliTreeLoader*> (trklLoader->GetBaseLoader("tracklets-raw"));
+  if (!trklTreeLoader) {
+    trklTreeLoader = new AliTreeLoader("tracklets-raw", trklLoader);
+    trklLoader->AddBaseLoader(trklTreeLoader);
+  }
+  if (!trklTreeLoader) {
+    return 0x0;
+  }
 
-      det    = input.GetDet();
+  if (!trklTreeLoader->Tree())
+    trklTreeLoader->MakeTree();
 
-      if (det != lastdet) 
-       {       
-       
-         lastdet = det;
+  input.SetTrackletArray(TrackletsArray());
+  input.SetTrackArray(TracksArray());
 
-         if (digits) digits->Compress(1,0);
-         if (track0) track0->Compress(1,0);
-         if (track1) track1->Compress(1,0);
-         if (track2) track2->Compress(1,0);
-       
-         // Add a container for the digits of this detector
-         digits = digitsManager->GetDigits(det);
-         track0 = digitsManager->GetDictionary(det,0);
-         track1 = digitsManager->GetDictionary(det,1);
-         track2 = digitsManager->GetDictionary(det,2);
-
-         // Allocate memory space for the digits buffer
-         if (digits->GetNtime() == 0) 
-           {
-             digits->Allocate(input.GetMaxRow(),input.GetMaxCol(), input.GetNumberOfTimeBins());
-             track0->Allocate(input.GetMaxRow(),input.GetMaxCol(), input.GetNumberOfTimeBins());
-             track1->Allocate(input.GetMaxRow(),input.GetMaxCol(), input.GetNumberOfTimeBins());
-             track2->Allocate(input.GetMaxRow(),input.GetMaxCol(), input.GetNumberOfTimeBins());
-           }
+  // Loop through the digits
+  Int_t det    = 0;
 
-         indexes = digitsManager->GetIndexes(det);
-         indexes->SetSM(input.GetSM());
-         indexes->SetStack(input.GetStack());
-         indexes->SetLayer(input.GetLayer());
-         indexes->SetDetNumber(det);
-         if (indexes->IsAllocated() == kFALSE)
-           indexes->Allocate(input.GetMaxRow(), input.GetMaxCol(), input.GetNumberOfTimeBins());
-       }
-    
-      for (it = 0; it < 3; it++)
+  while (det >= 0)
+    {
+      det = input.NextChamber(digitsManager);
+
+      if (det >= 0)
        {
-         if ( input.GetTimeBin() + it < input.GetNumberOfTimeBins() )
-           {
-             if (input.GetSignals()[it] > 0)
-               {
-                 digits->SetDataUnchecked(input.GetRow(), input.GetCol(),
-                                          input.GetTimeBin() + it, input.GetSignals()[it]);
-
-                 indexes->AddIndexTBin(input.GetRow(), input.GetCol(),
-                                       input.GetTimeBin() + it);
-                 track0->SetDataUnchecked(input.GetRow(), input.GetCol(),
-                                          input.GetTimeBin() + it, 0);
-                 track1->SetDataUnchecked(input.GetRow(), input.GetCol(),
-                                          input.GetTimeBin() + it, 0);
-                 track2->SetDataUnchecked(input.GetRow(), input.GetCol(),
-                                          input.GetTimeBin() + it, 0);
-               }
-           }
+         // get...
+         digits = (AliTRDarrayADC *) digitsManager->GetDigits(det);
+         track0 = (AliTRDarrayDictionary *) digitsManager->GetDictionary(det,0);
+         track1 = (AliTRDarrayDictionary *) digitsManager->GetDictionary(det,1);
+         track2 = (AliTRDarrayDictionary *) digitsManager->GetDictionary(det,2);
+         // and compress
+         if (digits) digits->Compress();  
+         if (track0) track0->Compress();   
+         if (track1) track1->Compress();     
+         if (track2) track2->Compress();
        }
+    }
+
+  for (Int_t iSector = 0; iSector < fGeo->Nsector(); iSector++) {
+    fTrgFlags[iSector] = input.GetTriggerFlags(iSector);
   }
 
-  if (digits) digits->Compress(1,0);
-  if (track0) track0->Compress(1,0);
-  if (track1) track1->Compress(1,0);
-  if (track2) track2->Compress(1,0);
+  trklTreeLoader->WriteData("OVERWRITE");
+  trklLoader->UnloadAll();
 
   return digitsManager;
+}
 
+//_____________________________________________________________________________
+void AliTRDrawData::WriteIntermediateWords(UInt_t* buf, Int_t& nw, Int_t& of, const Int_t& maxSize, const Int_t& det, const Int_t& side) {
+    // 
+    // write tracklet end marker(0x10001000) 
+    // and half chamber headers(H[0] and H[1])
+    //
+    
+    Int_t        layer = fGeo->GetLayer( det );   // Layer
+    Int_t        stack = fGeo->GetStack( det );   // Stack
+    Int_t         sect = fGeo->GetSector( det );  // Sector (=iDDL)
+    Int_t           rv = fFee->GetRAWversion();
+    const Int_t kNTBin = fDigitsParam->GetNTimeBins(det);
+    Bool_t  trackletOn = fFee->GetTracklet();
+    UInt_t           x = 0;
+
+    // Write end of tracklet marker
+    if (nw < maxSize){
+      buf[nw++] = fgkEndOfTrackletMarker;
+      buf[nw++] = fgkEndOfTrackletMarker;     // the number of tracklet end marker should be more than 2
+    }
+    else {
+      of++;
+    }
+   
+    // Half Chamber header
+    // h[0] (there are 2 HC headers) xmmm mmmm nnnn nnnq qqss sssp ppcc ci01
+    //         , where  x : Raw version speacial number (=1)
+    //                      m : Raw version major number (test pattern, ZS, disable tracklet, 0, options)
+    //                      n : Raw version minor number
+    //                      q : number of addtional header words (default = 1)
+    //                      s : SM sector number (ALICE numbering)
+    //                      p : plane(layer) number
+    //                  c : chamber(stack) number
+    //                  i : side number (0:A, 1:B)
+    Int_t majorv = 0;  // The major version number 
+    Int_t minorv = 0;  // The minor version number
+    Int_t add    = 1;  // The number of additional header words to follow : now 1, previous 2
+    Int_t tp    = 0;   // test pattern (default=0)
+    Int_t zs    = (rv==3) ? 1 : 0;                     // zero suppression
+    Int_t dt    = (trackletOn) ? 0 : 1;        // disable tracklet 
+    
+    majorv = (tp<<6) | (zs<<5) | (dt<<4) | 1;  // major version
+    
+    x = (1<<31) | (majorv<<24) | (minorv<<17) | (add<<14) | (sect<<9) | (layer<<6) | (stack<<3) | (side<<2) | 1;
+    if (nw < maxSize) buf[nw++] = x; else of++;
+    
+    // h[1]            tttt ttbb bbbb bbbb bbbb bbpp pphh hh01
+    // , where  t : number of time bins
+    //          b : bunch crossing number
+    //          p : pretrigger counter
+    //          h : pretrigger phase
+    Int_t bcCtr   = 99; // bunch crossing counter. Here it is set to 99 always for no reason
+    Int_t ptCtr   = 15; // pretrigger counter. Here it is set to 15 always for no reason
+    Int_t ptPhase = 11; // pretrigger phase. Here it is set to 11 always for no reason
+    //x = (bcCtr<<16) | (ptCtr<<12) | (ptPhase<<8) | ((kNTBin-1)<<2) | 1;      // old format
+    x = ((kNTBin)<<26) | (bcCtr<<10) | (ptCtr<<6) | (ptPhase<<2) | 1;
+    if (nw < maxSize) buf[nw++] = x; else of++;
+}
+
+TClonesArray *AliTRDrawData::TrackletsArray()
+{
+  // Returns the array of on-line tracklets
+
+  if (!fTracklets) {
+    fTracklets = new TClonesArray("AliTRDtrackletWord", 200);
+  }
+  return fTracklets;
+}
+
+TClonesArray* AliTRDrawData::TracksArray()
+{
+  // return array of GTU tracks (create TClonesArray if necessary)
+
+  if (!fTracks) {
+    fTracks = new TClonesArray("AliESDTrdTrack",100);
+  }
+  return fTracks;
 }