-V0 finder for MC same as in data

diff --git a/PWGDQ/dielectron/AliDielectron.cxx b/PWGDQ/dielectron/AliDielectron.cxx
index 6c90c05d3ccf264ce07bcda3461ff6bafc756b68..86946203d73873bfbab0f742c14fe8e068dd88c9 100644 (file)
--- a/PWGDQ/dielectron/AliDielectron.cxx
+++ b/PWGDQ/dielectron/AliDielectron.cxx
@@ -65,6 +65,7 @@ The names are available via the function PairClassName(Int_t i)
 #include "AliDielectronDebugTree.h"
 #include "AliDielectronSignalMC.h"
 #include "AliDielectronMixingHandler.h"
+#include "AliDielectronV0Cuts.h"
 
 #include "AliDielectron.h"
 
@@ -233,6 +234,14 @@ void AliDielectron::Process(AliVEvent *ev1, AliVEvent *ev2)
     return;
   }
 
+  // modify event numbers in MC so that we can identify new events 
+  // in AliDielectronV0Cuts (not neeeded for collision data)
+  if(GetHasMC()) {
+    ev1->SetBunchCrossNumber(1);
+    ev1->SetOrbitNumber(1);
+    ev1->SetPeriodNumber(1);
+  }
+
   // set event
   AliDielectronVarManager::SetEvent(ev1);
   if (fMixing){
@@ -308,7 +317,16 @@ void AliDielectron::Process(AliVEvent *ev1, AliVEvent *ev2)
 
   // clear arrays
   if (!fDontClearArrays) ClearArrays();
+
+  // reset TPC EP and unique identifiers for v0 cut class 
   AliDielectronVarManager::SetTPCEventPlane(0x0);
+  if(GetHasMC()) { // only for MC needed
+    for (Int_t iCut=0; iCut<fTrackFilter.GetCuts()->GetEntries();++iCut) {
+      if ( fTrackFilter.GetCuts()->At(iCut)->IsA() == AliDielectronV0Cuts::Class() )
+       ((AliDielectronV0Cuts*)fTrackFilter.GetCuts()->At(iCut))->ResetUniqueEventNumbers();
+    }
+  }
+
 }
 
 //________________________________________________________________

diff --git a/PWGDQ/dielectron/AliDielectronV0Cuts.cxx b/PWGDQ/dielectron/AliDielectronV0Cuts.cxx
index 9b785e8cba4d332f4fc7a56b4a11e4c849fd1286..b115b812a9029c137a2858a552d4c6047a19826e 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronV0Cuts.cxx
+++ b/PWGDQ/dielectron/AliDielectronV0Cuts.cxx
@@ -104,9 +104,6 @@ void AliDielectronV0Cuts::InitEvent(AliVTrack *trk)
   if(!IsNewEvent(ev)) return;
   //  printf(" Build V0 candidates according to the applied cuts \n");
 
-  // TODO think about MCevent
-  //  Print();
-
   // rest booleans
   fV0TrackArr.ResetAllBits();
 
@@ -122,7 +119,8 @@ void AliDielectronV0Cuts::InitEvent(AliVTrack *trk)
   AliDielectronPID dauPIDcuts;
   if(fPID>=0) dauPIDcuts.SetDefaults(fPID);
 
-  Int_t nV0s = 0;
+  Int_t nV0s      = 0;
+  Int_t nV0stored = 0;
   AliDielectronPair candidate;
   candidate.SetPdgCode(fMotherPdg);
 
@@ -177,7 +175,6 @@ void AliDielectronV0Cuts::InitEvent(AliVTrack *trk)
   else if(ev->IsA() == AliAODEvent::Class()) {
     const AliAODEvent *aodEv = static_cast<const AliAODEvent*>(ev);
 
-    //    Int_t nV0stored = 0;
     // loop over vertices
     for (Int_t ivertex=0; ivertex<aodEv->GetNumberOfVertices(); ++ivertex){
       AliAODVertex *v=aodEv->GetVertex(ivertex);
@@ -190,7 +187,7 @@ void AliDielectronV0Cuts::InitEvent(AliVTrack *trk)
        printf("Error: Couldn't get V0 daughter: %p - %p\n",trNeg,trPos);
        continue;
       }
-      //nV0stored++;
+      nV0stored++;
 
       // PID default cuts
       if(fPID>=0) {
@@ -221,7 +218,7 @@ void AliDielectronV0Cuts::InitEvent(AliVTrack *trk)
   else
     return;
 
-  //  printf(" Number of V0s candiates found %d \n",nV0s);
+  //  printf(" Number of V0s candiates found %d/%d \n",nV0s,nV0stored);
 
 }
 //________________________________________________________________________

diff --git a/PWGDQ/dielectron/AliDielectronV0Cuts.h b/PWGDQ/dielectron/AliDielectronV0Cuts.h
index c296065bf77bea07c7bb600978035b15ad902fa8..6891bc355a6048feb895ef724c2577c3a342ef38 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronV0Cuts.h
+++ b/PWGDQ/dielectron/AliDielectronV0Cuts.h
@@ -43,6 +43,7 @@ public:
   void SetPdgCodes(Int_t mother, Int_t negDaughter, Int_t posDaughter) {fMotherPdg=mother; fNegPdg=negDaughter; fPosPdg=posDaughter;}
   void SetExcludeTracks(Bool_t exclude) {fExcludeTracks=exclude;}
   void SetDefaultPID(Int_t def) {fPID=def;}
+  void ResetUniqueEventNumbers() { fOrbit=0; fPeriod=0; fBunchCross=0; }
 
 private: