Helper classes for Et em corrections from Marcelo

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

diff --git a/TRD/AliTRDclusterizer.h b/TRD/AliTRDclusterizer.h
index 0b305f4978519bc06870b6b51f5ae0a9bb2a0384..55a97d84337d3647727b3b7f29804abfd1bba943 100644 (file)
--- a/TRD/AliTRDclusterizer.h
+++ b/TRD/AliTRDclusterizer.h
@@ -49,6 +49,7 @@ class AliTRDclusterizer : public TNamed
     ,kLUT    = BIT(18)  //  using look up table for cluster's r-phi position
     ,kGAUS   = BIT(19)  //  using gauss approx. for cluster's r-phi position
     ,knewDM  = BIT(20)  //  was the digitsmanger created by raw2clusters?
+    ,kTracksOwner = BIT(21) //  toggle GTU track ownership
   };
 
   struct MaxStruct
@@ -82,11 +83,14 @@ class AliTRDclusterizer : public TNamed
   Bool_t   ReadDigits(AliRawReader *rawReader);
   Bool_t   ReadDigits(TTree *digitsTree);
 
+  Bool_t   ReadTracklets();
+  Bool_t   ReadTracks();
+
   Bool_t   WriteClusters(Int_t det);
   void     ResetRecPoints();
   virtual TClonesArray    *RecPoints();
-  virtual TClonesArray    *TrackletsArray();
-  Bool_t   WriteTracklets(Int_t det);
+  virtual TClonesArray    *TrackletsArray(const TString &trkltype = "");
+  virtual TClonesArray    *TracksArray();
 
   Bool_t   Raw2Clusters(AliRawReader *rawReader);
   Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);
@@ -105,11 +109,14 @@ class AliTRDclusterizer : public TNamed
   Bool_t           IsClustersOwner() const {return TestBit(kClOwner);}
   virtual void     SetClustersOwner(Bool_t own=kTRUE) {SetBit(kClOwner, own); if(!own) {fRecPoints = 0x0; fNoOfClusters=0;} }
   void             SetTrackletsOwner(Bool_t own=kTRUE) {SetBit(kTrOwner, own); if(!own) {fTracklets = 0x0; } }
+  void             SetTracksOwner(Bool_t own=kTRUE) {SetBit(kTracksOwner, own); if(!own) {fTracks = 0x0; } }
   void             SetSkipTransform(Bool_t b=kTRUE) {SetBit(kSkipTrafo, b); }
 
 protected:
 
+  void             ApplyTCTM(Short_t *const arr, const Int_t nTime, const Int_t nexp);
   void             DeConvExp (Short_t *const arr, const Int_t nTime, const Int_t nexp);
+  void             ConvExp(Short_t *const arr, const Int_t nTime);
   void             TailCancelation(const AliTRDrecoParam* const recoParam);
 
   Float_t          Unfold(Double_t eps, Int_t layer, const Double_t *const padSignal) const;
@@ -123,7 +130,6 @@ protected:
   void             CreateCluster(const MaxStruct &Max); 
 
   virtual void     AddClusterToArray(AliTRDcluster* cluster);
-  virtual void     AddTrackletsToArray();
 
 private:
   inline void      CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount);
@@ -134,12 +140,15 @@ protected:
   TTree               *fClusterTree;         //! Tree with the cluster
   TClonesArray        *fRecPoints;           //! Array of clusters
   TClonesArray        *fTracklets;           //! Array of online tracklets
+  TClonesArray        *fTracks;              //! Array of GTU tracks
 
   TTree               *fTrackletTree;        //! Tree for tracklets
 
   AliTRDdigitsManager *fDigitsManager;       //! TRD digits manager
 
   UInt_t              **fTrackletContainer;  //! tracklet container
+                                            // legacy code to avoid breakint AliHLTTRDClusterizer
+                                            // but it's useless
 
   Int_t                fRawVersion;          //  Expected raw version of the data - default is 2