In case of merging AOD inputs, recover vertex from both inputs, not from first.

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

diff --git a/PWG4/PartCorrBase/AliCaloTrackAODReader.cxx b/PWG4/PartCorrBase/AliCaloTrackAODReader.cxx
index 7e3c3e3cbab7c9c4fa35aedcbf7f9f381cd48da6..b5ca134047b2157185bcc48f3cd48f15f8513f3f 100755 (executable)
--- a/PWG4/PartCorrBase/AliCaloTrackAODReader.cxx
+++ b/PWG4/PartCorrBase/AliCaloTrackAODReader.cxx
@@ -175,6 +175,7 @@ void AliCaloTrackAODReader::FillInputEMCAL() {
 
   //If second input event available, add the clusters.
   if(fSecondInputAODTree && fSecondInputAODEvent){
+         GetSecondInputAODVertex(v);
          nclusters = ((AliAODEvent*)fSecondInputAODEvent)->GetNCaloClusters();
          if(fDebug > 1) printf("AliCaloTrackAODReader::FillInputEMCAL() - Add second input clusters, entries %d\n", nclusters) ;
                for (Int_t iclus =  0; iclus < nclusters; iclus++) {
@@ -240,7 +241,8 @@ void AliCaloTrackAODReader::FillInputPHOS() {
   if(fDebug > 1) printf("AliCaloTrackAODReader::FillInputPHOS()  - aod entries %d\n", fAODPHOSNormalInputEntries);
 
   //If second input event available, add the clusters.
-  if(fSecondInputAODTree && fSecondInputAODEvent){
+  if(fSecondInputAODTree && fSecondInputAODEvent){  
+         GetSecondInputAODVertex(v);
          nclusters = ((AliAODEvent*)fSecondInputAODEvent)->GetNCaloClusters();
          if(fDebug > 1) printf("AliCaloTrackAODReader::FillInputPHOS()  - Add second input clusters, entries %d\n", nclusters);
                for (Int_t iclus =  0; iclus < nclusters; iclus++) {
@@ -293,6 +295,15 @@ void AliCaloTrackAODReader::GetVertex(Double_t  v[3]) const {
   v[2] = ((AliAODEvent*)fInputEvent)->GetVertex(0)->GetZ() ;//CHECK!!!
 }
 
+//____________________________________________________________________________
+void AliCaloTrackAODReader::GetSecondInputAODVertex(Double_t  v[3]) const {
+       //Return vertex position of second AOD input
+       
+       v[0] = ((AliAODEvent*)fSecondInputAODEvent)->GetVertex(0)->GetX() ;//CHECK!!!
+       v[1] = ((AliAODEvent*)fSecondInputAODEvent)->GetVertex(0)->GetY() ;//CHECK!!!
+       v[2] = ((AliAODEvent*)fSecondInputAODEvent)->GetVertex(0)->GetZ() ;//CHECK!!!
+}
+
 //____________________________________________________________________________
 Double_t AliCaloTrackAODReader::GetBField() const {
   //Return magnetic field