Included also 3 and 4 prong decays; added variables to ntuple

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

diff --git a/TRD/AliTRDcluster.h b/TRD/AliTRDcluster.h
index e4cbe448e2bd2930254fa3684ef2c13c0d40308b..52a46b279092a5585664981bc8f5e4c225124fe1 100644 (file)
--- a/TRD/AliTRDcluster.h
+++ b/TRD/AliTRDcluster.h
@@ -13,6 +13,8 @@
 
 #include "AliCluster.h"
 
+class AliTRDtrackletWord;
+
 class AliTRDcluster : public AliCluster {
 public:
   enum ETRDclusterStatus { 
@@ -35,6 +37,7 @@ public:
     Int_t *tracks, Char_t npads, Short_t *signals,
     UChar_t col, UChar_t row, UChar_t time,
     Char_t timebin, Float_t center, UShort_t volid);
+  AliTRDcluster(const AliTRDtrackletWord *const tracklet, Int_t det, UShort_t volid);
   AliTRDcluster(const AliTRDcluster &c);
   virtual ~AliTRDcluster() {};
 
@@ -62,7 +65,9 @@ public:
   Short_t *GetSignals()                    { return fSignals;       }
   Float_t  GetSumS() const;
   static Double_t  GetSX(Int_t tb, Double_t z=-1);
-  static Double_t  GetSY(Int_t tb, Double_t z=-1);
+  static Double_t  GetSYdrift(Int_t tb, Int_t ly=0, Double_t z=-1);
+  static Double_t  GetSYcharge(Float_t q);
+  static Double_t  GetSYprf(Int_t ly, Double_t center, Double_t s2);
   static Double_t  GetXcorr(Int_t tb, Double_t z=-1);
   static Double_t  GetYcorr(Int_t ly, Float_t y);
   Float_t  GetXloc(Double_t t0, Double_t vd, Double_t *const q=0x0, Double_t *const xq=0x0, Double_t z = 0.2);
@@ -83,7 +88,8 @@ public:
   void     SetQ(Float_t inQ){ fQ = inQ;}
   void     SetClusterMasking(UChar_t inClusterMasking){ fClusterMasking = inClusterMasking;}
   void     SetShared(Bool_t sh  = kTRUE)   { SetBit(AliCluster::kShared,sh);    }
-  void     Use(Int_t = 0)                  { SetBit(AliCluster::kUsed, kTRUE);              }
+  void     SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
+  void     Use(Int_t u = 1)                  { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE);              }
   void     SetFivePad(Bool_t b = kTRUE) { SetBit(kFivePad,b);}
 
 protected:
@@ -123,11 +129,12 @@ inline Bool_t AliTRDcluster::IsRPhiMethod(ETRDclusterStatus m) const
 //________________________________________________
 inline void AliTRDcluster::SetRPhiMethod(ETRDclusterStatus m)
 {
+  SetBit(kCOG,0);SetBit(kLUT,0);SetBit(kGAUS,0);
   switch(m){
   case kCOG: SetBit(kCOG); break;
   case kLUT: SetBit(kLUT); break;
   case kGAUS: SetBit(kGAUS); break;
-  default: break;
+  default: SetBit(kLUT); break;
   }
 }