Mior fixes

[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis / AliFMDAnalysisTaskDndeta.cxx

diff --git a/PWG2/FORWARD/analysis/AliFMDAnalysisTaskDndeta.cxx b/PWG2/FORWARD/analysis/AliFMDAnalysisTaskDndeta.cxx
index e6cb5ff2cb781f21c86963753da18c37f91af719..e9998a4b7fef1583bc2d9b084ebe3d88888a3ba1 100644 (file)
--- a/PWG2/FORWARD/analysis/AliFMDAnalysisTaskDndeta.cxx
+++ b/PWG2/FORWARD/analysis/AliFMDAnalysisTaskDndeta.cxx
@@ -29,6 +29,7 @@
 #include "AliFMDStripIndex.h"
 #include "AliESDInputHandler.h"
 #include "AliGenDPMjetEventHeader.h"
+#include "AliLog.h"
 ClassImp(AliFMDAnalysisTaskDndeta)
 
 
@@ -310,12 +311,16 @@ void AliFMDAnalysisTaskDndeta::Terminate(Option_t */*option*/) {
       for(Int_t i =0; i<nVtxbins; i++) {
        
        TH2F* hMultTotal = (TH2F*)fOutputList->FindObject(Form("dNdeta_FMD%d%c_vtxbin%d",det,ringChar,i));
-       if(fVtxEff)
+       fOutputList->Add(hMultTotal->Clone(Form("%s_orig", hMultTotal->GetName())));
+       if(fVtxEff) { 
          hMultTotal->Scale(fVtxEff);
+       }
        
        TH2F* hMultTotalNSD = (TH2F*)fOutputList->FindObject(Form("dNdetaNSD_FMD%d%c_vtxbin%d",det,ringChar,i));
-       if(fVtxEffNSD)
+       fOutputList->Add(hMultTotalNSD->Clone(Form("%s_orig", hMultTotalNSD->GetName())));
+       if(fVtxEffNSD) { 
          hMultTotalNSD->Scale(fVtxEffNSD);
+       }
        
        //TH2F* hMultTrVtx = (TH2F*)hMultTotal->Clone(Form("dNdeta_FMD%d%c_TrVtx_vtxbin%d",det,ringChar,i));
        TH2F* hMultTrVtx = (TH2F*)fOutputList->FindObject(Form("dNdetaTrVtx_FMD%d%c_vtxbin%d",det,ringChar,i));
@@ -358,10 +363,13 @@ void AliFMDAnalysisTaskDndeta::Terminate(Option_t */*option*/) {
 //_____________________________________________________________________
 void AliFMDAnalysisTaskDndeta::ProcessPrimary() {
   
-  AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
+  AliMCEventHandler* eventHandler = 
+    dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()
+                                     ->GetMCtruthEventHandler());
+  if (!eventHandler) return;
+
   AliMCEvent* mcEvent = eventHandler->MCEvent();
-  if(!mcEvent)
-    return;
+  if(!mcEvent) return;
   
   fLastTrackByStrip.Reset(-1);
   
@@ -487,3 +495,4 @@ void AliFMDAnalysisTaskDndeta::ProcessPrimary() {
 //
 //
 // EOF
+// EOF