Setters added (Magnus)

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

diff --git a/TRD/AliTRDtrackletMCM.h b/TRD/AliTRDtrackletMCM.h
index 6a918c40b2a7a20c77f80ea37c3ab1a2aa41e0f5..884471d8a25a270a1bfb75215803b1bad693cd34 100644 (file)
--- a/TRD/AliTRDtrackletMCM.h
+++ b/TRD/AliTRDtrackletMCM.h
@@ -33,7 +33,9 @@ class AliTRDtrackletMCM : public AliTRDtrackletBase {
   // ----- Getters for MCM-tracklet information -----
   Int_t GetMCM() const { return fMCM; }
   Int_t GetROB() const { return fROB; }
-  Int_t GetLabel() const { return fLabel; }
+  Int_t GetLabel() const { return fLabel[0]; }
+  Int_t GetLabel(const Int_t i) const { return fLabel[i]; }
+  Bool_t HasLabel(const Int_t label) const { return (fLabel[0] == label || fLabel[1] == label || fLabel[2] == label); }
 
   // ----- Getters for offline corresponding values -----
   Bool_t CookPID() { return kFALSE; }
@@ -43,7 +45,10 @@ class AliTRDtrackletMCM : public AliTRDtrackletBase {
   Float_t GetdYdX() const { return (GetdY() * 140e-4 / 3.); }
   Float_t GetX() const { return fGeo->GetTime0((fHCId % 12) / 2); }
   Float_t GetY() const { return (GetYbin() * 160e-4); }
-  Float_t GetZ() const { return fGeo->GetPadPlane((fHCId % 12) / 2, (fHCId / 12) % 5)->GetRowPos( 4 * (fROB / 2) + fMCM / 4); }
+  Float_t GetZ() const { return fGeo->GetPadPlane((fHCId % 12) / 2, (fHCId / 12) % 5)->GetRowPos( 4 * (fROB / 2) + fMCM / 4) - 
+      fGeo->GetPadPlane((fHCId % 12) / 2, (fHCId /12) % 5)->GetRowSize(4 * (fROB / 2) + fMCM / 4) * .5; }
+  Float_t GetLocalZ() const { return GetZ() - 
+      (fGeo->GetPadPlane((fHCId % 12) / 2, (fHCId / 12) % 5)->GetRow0()+fGeo->GetPadPlane((fHCId % 12) / 2, (fHCId / 12) % 5)->GetRowEnd())/2.; }
 
   Int_t GetQ0() const { return fQ0; }
   Int_t GetQ1() const { return fQ1; }
@@ -58,13 +63,25 @@ class AliTRDtrackletMCM : public AliTRDtrackletBase {
   void SetHCId(Int_t id) { fHCId = id; }
   void SetMCM(Int_t mcm) { fMCM = mcm; }
   void SetROB(Int_t rob) { fROB = rob; }
-  void SetLabel(Int_t label) { fLabel = label; }
+  void SetLabel(Int_t label[]);
   void SetQ0(Int_t charge) { fQ0 = charge; }
   void SetQ1(Int_t charge) { fQ1 = charge; }
   void SetNHits(Int_t nhits) { fNHits = nhits; }
   void SetNHits0(Int_t nhits) { fNHits0 = nhits; }
   void SetNHits1(Int_t nhits) { fNHits1 = nhits; }
 
+  void SetSlope(Float_t slope) { fSlope = slope; }
+  void SetOffset(Float_t offset) { fOffset = offset; }
+  void SetError(Float_t error) { fError = error; }
+  void SetClusters(Float_t *res, Float_t *q, Int_t n);
+
+  Float_t GetSlope() const { return fSlope; }
+  Float_t GetOffset() const { return fOffset; }
+  Float_t GetError() const { return fError; }
+  Int_t   GetNClusters() const { return fNClusters; }
+  Float_t *GetResiduals() const { return fResiduals; }
+  Float_t *GetClsCharges() const { return fClsCharges; }
+
  protected:
   AliTRDgeometry *fGeo; //! TRD geometry
 
@@ -81,7 +98,14 @@ class AliTRDtrackletMCM : public AliTRDtrackletBase {
   Int_t fNHits0; // no. of contributing clusters in window 0
   Int_t fNHits1; // no. of contributing clusters in window 1
 
-  Int_t fLabel; // label for MC track
+  Int_t fLabel[3]; // up to 3 labels for MC track
+  
+  Float_t  fSlope;           // tracklet slope
+  Float_t  fOffset;          // tracklet offset
+  Float_t  fError;            // tracklet error
+  Int_t    fNClusters;       // no. of clusters
+  Float_t *fResiduals;       //[fNClusters] cluster to tracklet residuals
+  Float_t *fClsCharges;              //[fNClusters] cluster charge
 
  private:
   AliTRDtrackletMCM& operator=(const AliTRDtrackletMCM &rhs);   // not implemented