Fix for memory leaks in digits TClonesArrays (Matevz)

[u/mrichter/AliRoot.git] / TRD / AliTRDrawData.h

diff --git a/TRD/AliTRDrawData.h b/TRD/AliTRDrawData.h
index 4e042f4a578aa79df6bb3c91151eafbeb1b0799d..38fac037f8598c16a2cb3ef8ad6f25356b569bc1 100644 (file)
--- a/TRD/AliTRDrawData.h
+++ b/TRD/AliTRDrawData.h
@@ -14,9 +14,21 @@
 #include "TObject.h"
 
 class TTree;
-class AliTRDdigitsManager;
+
+class AliRunLoader;
+
 class AliRawReader;
 
+class AliTRDdigitsManager;
+class AliTRDgeometry;
+class AliTRDfeeParam;
+class AliTRDarrayADC;
+
+// Some constants:
+const UInt_t kEndoftrackletmarker = 0xAAAAAAAA; /*This marks the end of tracklet data words*/
+const UInt_t kEndofrawdatamarker  = 0x00000000; /*This marks the end of half-chamber-data*/
+const UInt_t kSizeWord            = sizeof(UInt_t);
+
 class AliTRDrawData : public TObject {
 
  public:
@@ -24,19 +36,69 @@ class AliTRDrawData : public TObject {
   AliTRDrawData();
   AliTRDrawData(const AliTRDrawData &r);
   virtual ~AliTRDrawData();
-  AliTRDrawData &operator=(const AliTRDrawData &r);
 
-  virtual void                 Copy(TObject &r);
+  AliTRDrawData &operator=(const AliTRDrawData &/*r*/) { return *this; }
+
+  virtual Bool_t       Digits2Raw(TTree *digits, const TTree *tracks = NULL);
+
+  virtual AliTRDdigitsManager *Raw2Digits(AliRawReader *rawReader);
+  //virtual AliTRDdigitsManager *Raw2DigitsOLD(AliRawReader *rawReader);
+  static void SetRawFormatVersion(Int_t iver){ fgRawFormatVersion=iver; };
+  static void SetSuppressionLevel(Int_t ilevel){ fgDataSuppressionLevel=ilevel; };
+  Bool_t WriteTracklets(Int_t det);
+  Bool_t OpenOutput();
 
-  virtual Bool_t               Digits2Raw(TTree *digits);
-  virtual AliTRDdigitsManager* Raw2Digits(AliRawReader* rawReader);
-  virtual void                 SetDebug(Int_t v = 1) { fDebug = v; };
+  enum FORMATTYPE
+    {
+      kRawOldFormat  =  0,
+      kRawNewFormat  =  1
+    };
 
  protected:
 
-  Int_t                fDebug;          //  Debug level
+  virtual Bool_t       Digits2Raw(AliTRDdigitsManager* digitsManager); // for fRawVersion > 0
+  virtual Int_t        ProduceHcData(AliTRDarrayADC *digits, Int_t side, Int_t det, UInt_t *buf, Int_t maxSize, Bool_t newEvent, Bool_t newSM);
+  virtual Int_t        ProduceHcDataNoSuppression(Int_t side, Int_t det, UInt_t *buf, Int_t maxSize);
+  virtual Int_t        ProduceHcDataV1andV2(AliTRDarrayADC *digits, Int_t side, Int_t det, UInt_t *buf, Int_t maxSize);
+  virtual Int_t        ProduceHcDataV3(AliTRDarrayADC *digits, Int_t side, Int_t det, UInt_t *buf, Int_t maxSize, Bool_t newEvent);
+  //virtual Int_t      ProduceHcDataV3(AliTRDarrayADC *digits, Int_t side, Int_t det, UInt_t *buf, Int_t maxSize);
+                 void         ProduceSMIndexData(UInt_t *buf, Int_t& nw);                              // SM index words and header - real data format
+          void         WriteIntermediateWords(UInt_t *buf, Int_t& nw, Int_t& of, const Int_t& maxSize, const Int_t& det, const Int_t& side); // writes tracklet-endmarker and additional words between tracklet and raw-data
+          void            WriteIntermediateWordsV2(UInt_t *buf, Int_t& nw, Int_t& of, const Int_t& maxSize, const Int_t& det, const Int_t& side); // real data format
+                 void         AssignStackMask(UInt_t *buf, Int_t nStack);  // re-assignment of stack mask in the SM index word
+          void         AssignLinkMask(UInt_t *buf, Int_t nLayer);   // re-assignment of link mask in the stack index word
+          Int_t        AddStackIndexWords(UInt_t *buf, Int_t nStack, Int_t nMax);   // add stack index words and stack header when there is no data for the stack 
+          Bool_t       ShiftWords(UInt_t *buf, Int_t nStart, Int_t nWords, Int_t nMax); // shifts n words
+  
+  AliRunLoader        *fRunLoader;      //! Run Loader
+  AliTRDgeometry      *fGeo;            //! Geometry
+  AliTRDfeeParam      *fFee;            //! Fee Parameters
+  Int_t                fNumberOfDDLs;   //  Number of DDLs
+  TTree               *fTrackletTree;        //! Tree for tracklets
+  UInt_t              **fTrackletContainer;  //! tracklet container
+
+  ClassDef(AliTRDrawData,5)             //  TRD raw data class
+
+ private:
+
+       static       Int_t  fgRawFormatVersion;                   // simulation raw data version - 0:old , 1:new(real data format)
+       static       Int_t  fgDataSuppressionLevel;                 // Data suppression level - 0:no su, 1: su, 2: deep suppression 
+       static const UInt_t fgkEndOfTrackletMarker  = 0x10001000; // This marks the end of tracklet data words
+       static const UInt_t fgkEndOfDataMarker      = 0x00000000; // This marks the end of HC data words
+
+    Int_t   fNTimeBins;                 // number of time bin
+    Int_t   fSMindexPos;                // Position of SM index word
+    Int_t   fStackindexPos;             // Position of SM index word
+    UInt_t  fEventCounter;              // Event counter(starting from 1)
+
+
 
-  ClassDef(AliTRDrawData,1)             //  TRD raw data class
 
 };
 #endif
+
+
+
+
+
+