Changes for request #60532 Port to the release: Include number of ITS cluster per...

[u/mrichter/AliRoot.git] / STEER / AliMultiplicity.cxx

diff --git a/STEER/AliMultiplicity.cxx b/STEER/AliMultiplicity.cxx
index a88fbc7f691b7024f3e733a695687f1c9a81b9ae..161563336e7c3651bfef74bcabc43984d609605d 100644 (file)
--- a/STEER/AliMultiplicity.cxx
+++ b/STEER/AliMultiplicity.cxx
@@ -1,5 +1,6 @@
 #include <string.h>
 #include "AliMultiplicity.h"
+#include "AliLog.h"
 
 ClassImp(AliMultiplicity)
 
@@ -21,6 +22,7 @@ AliMultiplicity::AliMultiplicity():
   // Default Constructor
   fFiredChips[0] = 0;
   fFiredChips[1] = 0;
+  for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0;
 }
 
 //______________________________________________________________________
@@ -66,6 +68,7 @@ AliMultiplicity::AliMultiplicity(Int_t ntr, Float_t *th,  Float_t *ph, Float_t *
   fFiredChips[0] = nfcL1;
   fFiredChips[1] = nfcL2;
   fFastOrFiredChips = fFastOr;
+  for(Int_t ilayer = 0; ilayer < 6; ilayer++)fITSClusters[ilayer] = 0;
 }
 
 //______________________________________________________________________
@@ -160,6 +163,11 @@ void AliMultiplicity::Duplicate(const AliMultiplicity& m){
 
   fFiredChips[0] = m.fFiredChips[0];
   fFiredChips[1] = m.fFiredChips[1];
+  for(Int_t ilayer = 0; ilayer < 6; ilayer++){
+    fITSClusters[ilayer] = m.fITSClusters[ilayer];
+  }
+
+  
 
   fFastOrFiredChips = m.fFastOrFiredChips;
 }
@@ -194,3 +202,25 @@ void AliMultiplicity::SetLabel(Int_t i, Int_t layer, Int_t label)
     }
     Error("SetLabel","Invalid track number %d or layer %d",i,layer);
 }
+
+//______________________________________________________________________
+UInt_t AliMultiplicity::GetNumberOfITSClusters(Int_t layMin, Int_t layMax) const {
+
+  if(layMax < layMin) {
+    AliError("layer min > layer max");
+    return 0;
+  }
+  if(layMin < 0) {
+    AliError("layer min < 0");
+    return 0;
+  }
+  if(layMax < 0) {
+    AliError("layer max > 0");
+    return 0;
+  }
+
+  Int_t sum=0; 
+  for (Int_t i=layMin; i<=layMax; i++) sum+=fITSClusters[i]; 
+  return sum; 
+
+}