X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliMultiplicity.h;h=4d5880684343a2a45d5bf9043e1adeb530018661;hb=d97f1dbeec08289789d6b55b4c984c22d3f46b6b;hp=2e8e474dd8462fdbc4655604923756aa69934b1f;hpb=32e449bea4d0a4cd7107d5dbd162566cfa61fe33;p=u%2Fmrichter%2FAliRoot.git

diff --git a/STEER/AliMultiplicity.h b/STEER/AliMultiplicity.h
index 2e8e474dd84..4d588068434 100644
--- a/STEER/AliMultiplicity.h
+++ b/STEER/AliMultiplicity.h
@@ -2,6 +2,8 @@
 #define ALIMULTIPLICITY_H
 
 #include<TObject.h>
+#include <TBits.h>
+#include<TMath.h>
 
 ////////////////////////////////////////////////////////
 ////   Class containing multiplicity information      //
@@ -14,26 +16,100 @@ class AliMultiplicity : public TObject {
 
   AliMultiplicity();               // default constructor
   // standard constructor
-  AliMultiplicity(Int_t ntr,Float_t *t=NULL, Float_t *ph=NULL, Float_t *df=NULL); 
+  AliMultiplicity(Int_t ntr,Float_t *th, Float_t *ph, Float_t *dth, Float_t *dph, Int_t *labels,
+         Int_t* labelsL2, Int_t ns, Float_t *ts, Float_t *ps, Short_t nfcL1, Short_t nfcL2, TBits fFastOrFiredChips);
   AliMultiplicity(const AliMultiplicity& m);
   AliMultiplicity& operator=(const AliMultiplicity& m);
+  virtual void Copy(TObject &obj) const;
   virtual ~AliMultiplicity();
+// methods to access tracklet information
   Int_t GetNumberOfTracklets() const {return fNtracks;}
-  Float_t GetTheta(Int_t i) const { if(i>=0 && i<fNtracks) {return fTh[i];}
-  else {Error("GetTheta","Invalid track number %d",i); return -9999.;}}
-  Float_t GetPhi(Int_t i) const { if(i>=0 && i<fNtracks) {return fPhi[i];}
-  else {Error("GetTheta","Invalid track number %d",i); return -9999.;}}
-  Float_t GetDeltaPhi(Int_t i) const {if(i>=0 && i<fNtracks) {return fDeltPhi[i];}
-  else {Error("GetDeltaPhi","Invalid track number %d",i); return -9999.;}}
+  Double_t GetTheta(Int_t i) const { 
+    if(i>=0 && i<fNtracks) return fTh[i];
+    Error("GetTheta","Invalid track number %d",i); return -9999.;
+  }
+  Double_t GetEta(Int_t i) const { 
+    if(i>=0 && i<fNtracks) return -TMath::Log(TMath::Tan(fTh[i]/2.));
+    Error("GetEta","Invalid track number %d",i); return -9999.;
+  }
+  Double_t GetPhi(Int_t i) const { 
+    if(i>=0 && i<fNtracks) return fPhi[i];
+    Error("GetPhi","Invalid track number %d",i); return -9999.;
+  }
+  Double_t GetDeltaTheta(Int_t i) const {
+    if(fDeltTh && i>=0 && i<fNtracks) return fDeltTh[i];
+    Error("GetDeltaTheta","DeltaTheta not available in data or Invalid track number %d(max %d)",i, fNtracks); return -9999.;
+  }
+  Double_t GetDeltaPhi(Int_t i) const {
+    if(i>=0 && i<fNtracks) return fDeltPhi[i];
+    Error("GetDeltaPhi","Invalid track number %d",i); return -9999.;
+  }
+
+  Int_t GetLabel(Int_t i, Int_t layer) const;
+  void  SetLabel(Int_t i, Int_t layer, Int_t label);
+  
+// methods to access single cluster information
+  Int_t GetNumberOfSingleClusters() const {return fNsingle;}
+  Double_t GetThetaSingle(Int_t i) const { 
+    if(i>=0 && i<fNsingle) return fThsingle[i];
+    Error("GetThetaSingle","Invalid cluster number %d",i); return -9999.;
+  }
+
+  Double_t GetPhiSingle(Int_t i) const { 
+    if(i>=0 && i<fNsingle) return fPhisingle[i];
+    Error("GetPhisingle","Invalid cluster number %d",i); return -9999.;
+  }
+
+  Short_t GetNumberOfFiredChips(Int_t layer) const { return fFiredChips[layer]; }
+  void SetFiredChips(Int_t layer, Short_t firedChips) { fFiredChips[layer] = firedChips; }
+
+  UInt_t GetNumberOfITSClusters(Int_t layer) const { return layer<6 ? fITSClusters[layer] : 0; }
+  UInt_t GetNumberOfITSClusters(Int_t layMin, Int_t layMax) const ;
+  void SetITSClusters(Int_t layer, UInt_t clusters) { fITSClusters[layer] = clusters; }
+
+  void   SetFastOrFiredChips(UInt_t chipKey){fFastOrFiredChips.SetBitNumber(chipKey);}
+  TBits  GetFastOrFiredChips() const {return fFastOrFiredChips;}
+  Bool_t TestFastOrFiredChips(UInt_t chipKey) const {return fFastOrFiredChips.TestBitNumber(chipKey);}
 
   protected:
   void Duplicate(const AliMultiplicity &m);  // used by copy ctr.
-  Int_t fNtracks;         // Number of tracklets (=-1 when mult is not determined)
-  Float_t *fTh;           //[fNtracks] array with theta values
-  Float_t *fPhi;          //[fNtracks] array with phi values
-  Float_t *fDeltPhi;      //[fNtracks] array with delta phi values
 
-  ClassDef(AliMultiplicity,1);
+  Int_t fNtracks;            // Number of tracklets
+  Int_t fNsingle;            // Number of clusters on SPD layer 1, not associated
+                             // with a tracklet on SPD layer 2
+  Int_t *fLabels;            //[fNtracks] array with labels of cluster in L1 used for tracklet
+  Int_t *fLabelsL2;          //[fNtracks] array with labels of cluster in L2 used for tracklet
+  Double32_t *fTh;           //[fNtracks] array with theta values
+  Double32_t *fPhi;          //[fNtracks] array with phi values
+  Double32_t *fDeltTh;       //[fNtracks] array with delta theta values
+  Double32_t *fDeltPhi;      //[fNtracks] array with delta phi values
+  Double32_t *fThsingle;     //[fNsingle] array with theta values of L1 clusters
+  Double32_t *fPhisingle;    //[fNsingle] array with phi values of L2 clusters
+  Short_t fFiredChips[2];    // Number of fired chips in the two SPD layers
+  UInt_t fITSClusters[6];   // Number of ITS cluster per layer
+  TBits fFastOrFiredChips;   // Map of FastOr fired chips
+
+  ClassDef(AliMultiplicity,10);
 };
 
+inline Int_t AliMultiplicity::GetLabel(Int_t i, Int_t layer) const
+{
+    if(i>=0 && i<fNtracks) {
+	if (layer == 0) {
+	    return fLabels[i];
+	} else if (layer == 1) {
+	    if (fLabelsL2) {
+		return fLabelsL2[i];
+	    } else {
+		Warning("GetLabel", "No information for layer 2 available !");
+		return -9999;
+	    }
+	} else {
+	    Error("GetLabel","Invalid layer number %d",layer); return -9999;
+	}
+    } else {
+	Error("GetLabel","Invalid track number %d",i); return -9999;
+    }
+    return -9999;
+}
 #endif