add multiplicity/centrality monitoring

diff --git a/PWG1/TRD/info/AliTRDeventInfo.cxx b/PWG1/TRD/info/AliTRDeventInfo.cxx
index 2acc2249b9ae3c7a94e5b7cf0bd53d875115676c..0c4165b71ac81a2a0f2bd56d39f8967ac8d4d314 100644 (file)
--- a/PWG1/TRD/info/AliTRDeventInfo.cxx
+++ b/PWG1/TRD/info/AliTRDeventInfo.cxx
@@ -16,13 +16,19 @@
 #include "AliTRDeventInfo.h"
 
 ClassImp(AliTRDeventInfo)
-
+Int_t const AliTRDeventInfo::fgkMultBin[AliTRDeventInfo::kCentralityClasses] = {
+  700, 1400, 2100, 2800, 3500
+};
+Float_t const AliTRDeventInfo::fgkCentBin[AliTRDeventInfo::kCentralityClasses] = {
+  0.1, 0.2, 0.5, 0.8, 1.
+};
 //____________________________________________________________________
 AliTRDeventInfo::AliTRDeventInfo():
   TObject()
-  ,fHeader(0x0)
-  ,fRun(0x0)
+  ,fHeader(NULL)
+  ,fRun(NULL)
   ,fCentrality(-1)
+  ,fMult(-1)
 {
   //
   // Default Constructor
@@ -36,6 +42,7 @@ AliTRDeventInfo::AliTRDeventInfo(AliESDHeader *header, AliESDRun *run):
   ,fHeader(header)
   ,fRun(run)
   ,fCentrality(-1)
+  ,fMult(-1)
 {
   //
   // Constructor with Arguments
@@ -59,6 +66,7 @@ AliTRDeventInfo::AliTRDeventInfo(const AliTRDeventInfo &info):
   ,fHeader(info.fHeader)
   ,fRun(info.fRun)
   ,fCentrality(info.fCentrality)
+  ,fMult(info.fMult)
 {
   //
   // Copy Constructor
@@ -77,6 +85,7 @@ AliTRDeventInfo& AliTRDeventInfo::operator=(const AliTRDeventInfo& info){
   fHeader     = info.fHeader;
   fRun        = info.fRun;
   fCentrality = info.fCentrality;
+  fMult       = info.fMult;
   SetBit(kOwner, 0);
   return *this;
 }
@@ -100,8 +109,8 @@ void AliTRDeventInfo::Delete(const Option_t *){
     if(fHeader) delete fHeader;
     if(fRun) delete fRun;
   };
-  fHeader = 0x0;
-  fRun = 0x0;
+  fHeader = NULL;
+  fRun = NULL;
 }
 
 //____________________________________________________________________
@@ -121,6 +130,26 @@ UShort_t  AliTRDeventInfo::GetBunchFill() const
   return fHeader->GetBunchCrossNumber();
 }
 
+//____________________________________________________________________
+Int_t  AliTRDeventInfo::GetCentralityBin(Float_t cenPer)
+{
+// calculate centrality bin
+  for(Int_t icen(0); icen<kCentralityClasses; icen++){
+    if(cenPer<fgkCentBin[icen]) return icen;
+  }
+  return -1;
+}
+
+//____________________________________________________________________
+Int_t  AliTRDeventInfo::GetMultiplicityBin(Int_t n)
+{
+// calculate centrality bin
+  for(Int_t im(0); im<kCentralityClasses; im++){
+    if(n<fgkMultBin[im]) return kCentralityClasses-im-1;
+  }
+  return 0;
+}
+
 //____________________________________________________________________
 void AliTRDeventInfo::GetListOfIsolatedBunches(TH1D* hbc, Int_t bunchSpacing) {
   //

diff --git a/PWG1/TRD/info/AliTRDeventInfo.h b/PWG1/TRD/info/AliTRDeventInfo.h
index cedb88e415e53f9f93c8ced52de732714ce7ae09..b142b52475279066ba691b055d44f2f78e1f3962 100644 (file)
--- a/PWG1/TRD/info/AliTRDeventInfo.h
+++ b/PWG1/TRD/info/AliTRDeventInfo.h
@@ -33,22 +33,29 @@ public:
   AliESDHeader* GetEventHeader() const                 { return fHeader; }
   AliESDRun*    GetRunInfo() const                     { return fRun; }
   Int_t         GetCentrality() const                  { return fCentrality; }
+  static Int_t  GetCentralityBin(Float_t cenPer);
+  Int_t         GetMultiplicity() const                { return fMult; }
+  static Int_t  GetMultiplicityBin(Int_t n);
   UShort_t      GetBunchFill() const;
   static void   GetListOfIsolatedBunches(TH1D *hbc, Int_t bunchSpacing=10);
   Bool_t        IsOwner() const                        { return TestBit(kOwner); }
   void          SetEventHeader(AliESDHeader *evHeader) { fHeader = evHeader; }
   void          SetRunInfo(AliESDRun *evRun)           { fRun = evRun; }
-  void          SetCentrality(Int_t centrality)        { fCentrality = centrality; }
+  void          SetCentrality(Float_t cent)            { fCentrality = cent>=0.?GetCentralityBin(cent):-1;}
+  void          SetMultiplicity(Int_t n)               { fMult = n>=0?GetMultiplicityBin(n):-1;}
   void          SetOwner();
 
 private:
   enum{
     kOwner = BIT(14)
   };
+  static Int_t const   fgkMultBin[kCentralityClasses]; // multiplicity bins
+  static Float_t const fgkCentBin[kCentralityClasses]; // centrality bins
   AliESDHeader* fHeader;      //! The ESD Header
   AliESDRun*    fRun;         //! The ESD Run Info
-  Int_t         fCentrality;  //! Centrality class
+  Int_t         fCentrality;  //! Centrality class based on AliCentrality
+  Int_t         fMult;        //! Centrality class based on AliMultiplicity
 
-  ClassDef(AliTRDeventInfo, 1) // Event info  relevant for TRD analysis
+  ClassDef(AliTRDeventInfo, 2) // Event info  relevant for TRD analysis
 };
 #endif