Corrected define guard

[u/mrichter/AliRoot.git] / PWGPP / HMPID / AliHMPIDTaskQA.cxx

diff --git a/PWGPP/HMPID/AliHMPIDTaskQA.cxx b/PWGPP/HMPID/AliHMPIDTaskQA.cxx
index 4107460b7ce1339eeef3238d145b5622fe5149be..24f83409d7cbcc21e381633666d32cfa108664aa 100644 (file)
--- a/PWGPP/HMPID/AliHMPIDTaskQA.cxx
+++ b/PWGPP/HMPID/AliHMPIDTaskQA.cxx
@@ -222,9 +222,7 @@ void AliHMPIDTaskQA::ConnectInputData(Option_t *option)
     if (!mcH) {
       AliDebug(2,Form("ERROR: Could not get MCEventHandler"));
       fUseMC = kFALSE;
-    } else
-      fMC = mcH->MCEvent();
-      if (!fMC) AliDebug(2,Form("ERROR: Could not get MCEvent"));
+    }
   }
 }
 
@@ -240,7 +238,14 @@ void AliHMPIDTaskQA::UserExec(Option_t *)
   AliStack* pStack = 0;
   Int_t label = -1;
   if (fUseMC){
-    pStack = fMC->Stack();
+    AliMCEventHandler *mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
+    if (mcH) {
+      fMC = mcH->MCEvent();
+      pStack = fMC->Stack();
+    } else {
+      AliFatal("fMC flag set but MC handler not available");
+      return;
+    }  
   }
 
   //
@@ -380,10 +385,10 @@ void AliHMPIDTaskQA::UserCreateOutputObjects() {
    fHmpPesdPhmp = new TH2F("fHmpPesdPhmp","HMPID: ESD p - running p;HMP p (GeV/c);ESD p (Gev/c)",100,0,10,100,0,10);
    fHmpHistList->Add(fHmpPesdPhmp);
 
-   fHmpCkovPesd = new TH2F("fHmpCkovPesd","HMPID: ThetaCherenkov vs P;p_esd (GeV/c);#Theta_C;Entries",100,0,10,110,0,1.1);
+   fHmpCkovPesd = new TH2F("fHmpCkovPesd","HMPID: ThetaCherenkov vs P;p_esd (GeV/c);#Theta_C;Entries",500,0,10,500,0,1.0);
    fHmpHistList->Add(fHmpCkovPesd);
 
-   fHmpCkovPhmp = new TH2F("fHmpCkovPhmp","HMPID: ThetaCherenkov vs P;p_hmp (GeV/c);#Theta_C;Entries",100,0,10,110,0,1.1);
+   fHmpCkovPhmp = new TH2F("fHmpCkovPhmp","HMPID: ThetaCherenkov vs P;p_hmp (GeV/c);#Theta_C;Entries",500,0,10,500,0,1.0);
    fHmpHistList->Add(fHmpCkovPhmp);
 
    for (Int_t i=0; i<7; i++){