adding support for ZNA and CL1 cent. estimators in HBT code

[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemto / AliFemtoEventReaderESDChain.cxx

diff --git a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChain.cxx b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChain.cxx
index 8548453d55a5a1b473c0dd1bde0796c4529bcb6e..495ed24c89e0175917e93affb55678b8bf74e7c1 100644 (file)
--- a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChain.cxx
+++ b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoEventReaderESDChain.cxx
@@ -687,6 +687,8 @@ AliFemtoEvent* AliFemtoEventReaderESDChain::ReturnHbtEvent()
   AliCentrality *cent = fEvent->GetCentrality();
   if (cent) {
     hbtEvent->SetCentralityV0(cent->GetCentralityPercentile("V0M"));
+    hbtEvent->SetCentralityZNA(cent->GetCentralityPercentile("ZNA"));
+    hbtEvent->SetCentralityCL1(cent->GetCentralityPercentile("CL1"));
     //    hbtEvent->SetCentralityFMD(cent->GetCentralityPercentile("FMD"));
     hbtEvent->SetCentralitySPD1(cent->GetCentralityPercentile("CL1"));
     //    hbtEvent->SetCentralityTrk(cent->GetCentralityPercentile("TRK"));
@@ -723,6 +725,30 @@ AliFemtoEvent* AliFemtoEventReaderESDChain::ReturnHbtEvent()
                             10.0*cent->GetCentralityPercentile("V0M"), lrint(10.0*cent->GetCentralityPercentile("V0M")));
     }
   }
+  else if (fEstEventMult == kCentralityZNA) {
+    // centrality between 0 (central) and 1 (very peripheral)
+
+    if (cent) {
+      if (cent->GetCentralityPercentile("ZNA") < 0.00001)
+       hbtEvent->SetNormalizedMult(-1);
+      else
+       hbtEvent->SetNormalizedMult(lrint(10.0*cent->GetCentralityPercentile("ZNA")));
+      if (Debug()>1) printf ("Set Centrality %i %f %li\n", hbtEvent->UncorrectedNumberOfPrimaries(), 
+                            10.0*cent->GetCentralityPercentile("ZNA"), lrint(10.0*cent->GetCentralityPercentile("ZNA")));
+    }
+  }
+  else if (fEstEventMult == kCentralityCL1) {
+    // centrality between 0 (central) and 1 (very peripheral)
+
+    if (cent) {
+      if (cent->GetCentralityPercentile("CL1") < 0.00001)
+       hbtEvent->SetNormalizedMult(-1);
+      else
+       hbtEvent->SetNormalizedMult(lrint(10.0*cent->GetCentralityPercentile("CL1")));
+      if (Debug()>1) printf ("Set Centrality %i %f %li\n", hbtEvent->UncorrectedNumberOfPrimaries(), 
+                            10.0*cent->GetCentralityPercentile("CL1"), lrint(10.0*cent->GetCentralityPercentile("CL1")));
+    }
+  }
 
   if (tNormMultPos > tNormMultNeg)
     hbtEvent->SetZDCParticipants(tNormMultPos);