get the v0 signal and multiplicity, accessible for different analysis, add histograms...

[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliCaloTrackESDReader.cxx

diff --git a/PWG4/PartCorrBase/AliCaloTrackESDReader.cxx b/PWG4/PartCorrBase/AliCaloTrackESDReader.cxx
index ddcb48928febec2a6409e5130358247dff1da0bc..0f189d0e7ce4bb31a83a85541d8abf5dafed361e 100755 (executable)
--- a/PWG4/PartCorrBase/AliCaloTrackESDReader.cxx
+++ b/PWG4/PartCorrBase/AliCaloTrackESDReader.cxx
@@ -35,6 +35,7 @@
 #include "AliMultiEventInputHandler.h"
 #include "AliAnalysisManager.h"
 #include "AliMixedEvent.h"
+#include "AliESDVZERO.h"
 
 
 ClassImp(AliCaloTrackESDReader)
@@ -93,3 +94,27 @@ void AliCaloTrackESDReader::SetInputOutputMCEvent(AliVEvent* esd, AliAODEvent* a
   SetMC(mc);
   
 }
+
+//____________________________________________________________________________
+void AliCaloTrackESDReader::FillInputVZERO(){
+  //Fill VZERO information in data member, add all the channels information.
+  AliESDVZERO* esdV0 = ((AliESDEvent*)fInputEvent)->GetVZEROData();
+  //printf("Init V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
+
+  if (esdV0) 
+  {
+    for (Int_t i = 0; i < 32; i++)
+    {
+      fV0ADC[0] += esdV0->GetAdcV0C(i);
+      fV0ADC[1] += esdV0->GetAdcV0A(i);
+      fV0Mul[0] += esdV0->GetMultiplicityV0C(i);
+      fV0Mul[1] += esdV0->GetMultiplicityV0A(i);
+    }
+    if(fDebug > 0)
+      printf("V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
+  }
+  else
+  {
+    printf("Cannot retrieve V0 ESD! Run w/ null V0 charges");
+  }
+}