Fix Coverity

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

diff --git a/TRD/AliTRDtrackletMCM.h b/TRD/AliTRDtrackletMCM.h
index 3541f672ef6d696aed31cbc0812bed0ab5789177..3b4251a26a08fc4f53a53146391106adcbe75209 100644 (file)
--- a/TRD/AliTRDtrackletMCM.h
+++ b/TRD/AliTRDtrackletMCM.h
@@ -20,29 +20,41 @@ class AliTRDtrackletMCM : public AliTRDtrackletBase {
  public:
   AliTRDtrackletMCM(UInt_t trackletWord = 0);
   AliTRDtrackletMCM(UInt_t trackletWword, Int_t hcid);
+  AliTRDtrackletMCM(UInt_t trackletWword, Int_t hcid, Int_t rob, Int_t mcm);
   AliTRDtrackletMCM(const AliTRDtrackletMCM &rhs);
   ~AliTRDtrackletMCM();
 
   // ----- Getters for contents of tracklet word -----
-  Int_t GetYbin() const; 
-  Int_t GetdY() const; 
+  Int_t GetYbin() const;
+  Int_t GetdY() const;
   Int_t GetZbin() const { return ((fTrackletWord >> 20) & 0xf); }
   Int_t GetPID() const { return ((fTrackletWord >> 24) & 0xff); }
 
   // ----- 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; }
-  Double_t GetPID(Int_t /* is */) const { return 0; }
+  Double_t GetPID(Int_t /* is */) const { return GetPID()/255.; }
   Int_t GetDetector() const { return fHCId / 2; }
   Int_t GetHCId() const { return fHCId; }
   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; }
+  Int_t GetNHits() const { return fNHits; }
+  Int_t GetNHits0() const { return fNHits0; }
+  Int_t GetNHits1() const { return fNHits1; }
 
   UInt_t GetTrackletWord() const { return fTrackletWord; }
   void SetTrackletWord(UInt_t trackletWord) { fTrackletWord = trackletWord; }
@@ -51,23 +63,54 @@ 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
 
   Int_t fHCId;                  // half-chamber ID (only transient)
-  UInt_t fTrackletWord;                // tracklet word: PID | Z | deflection length | Y 
+  UInt_t fTrackletWord;                // tracklet word: PID | Z | deflection length | Y
                                //          bits:  12   4            7          13
   Int_t fMCM; // MCM no. in which the tracklet was found
   Int_t fROB; // ROB no. on which the tracklet was found
 
-  Int_t fLabel; // label for MC track
+  Int_t fQ0; // accumulated charge in the first time window
+  Int_t fQ1; // accumulated charge in the second time window
+
+  Int_t fNHits;  // no. of contributing clusters
+  Int_t fNHits0; // no. of contributing clusters in window 0
+  Int_t fNHits1; // no. of contributing clusters in window 1
+
+  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
 
-  ClassDef(AliTRDtrackletMCM, 1);
+  ClassDef(AliTRDtrackletMCM, 2);
 };
 
 #endif