Fixing Oysteins old cross check histograms

[u/mrichter/AliRoot.git] / PWGLF / totEt / AliAnalysisTaskTotEt.cxx

diff --git a/PWGLF/totEt/AliAnalysisTaskTotEt.cxx b/PWGLF/totEt/AliAnalysisTaskTotEt.cxx
index 5b69b58fa658fd09528feeaaab5d76b67749e988..a64ee5c990674a418ab7c3ac5d2274bcc580afa4 100644 (file)
--- a/PWGLF/totEt/AliAnalysisTaskTotEt.cxx
+++ b/PWGLF/totEt/AliAnalysisTaskTotEt.cxx
@@ -124,8 +124,16 @@ void AliAnalysisTaskTotEt::UserCreateOutputObjects()
 {
   //input hander
   AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();
+  if (!man) {
+    AliFatal("Analysis manager needed");
+    return;
+  }
+
   AliInputEventHandler *inputHandler=dynamic_cast<AliInputEventHandler*>(man->GetInputEventHandler());
-  if (!inputHandler) AliFatal("Input handler needed");
+  if (!inputHandler) {
+    AliFatal("Input handler needed");
+    return;
+  }
 
   //pid response object
   fPIDResponse=inputHandler->GetPIDResponse();
@@ -149,11 +157,17 @@ void AliAnalysisTaskTotEt::UserCreateOutputObjects()
   fOutputList->Add(fHistDiffEtRecEtMCOverEtMC);
 
   Bool_t selectPrimaries=kTRUE;
-  if(fRecAnalysis->DataSet()==2010 || fRecAnalysis->DataSet()==20111||fRecAnalysis->DataSet()==2009){
+  if(fRecAnalysis->DataSet()==2010 || fRecAnalysis->DataSet()==20111||fRecAnalysis->DataSet()==2009 || fRecAnalysis->DataSet()==2012 || fRecAnalysis->DataSet()==2013){
     if(fRecAnalysis->DataSet()==2010)cout<<"Setting track cuts for the 2010 p+p collisions at 7 TeV"<<endl;
     else{
-      if(fRecAnalysis->DataSet()==2009){cout<<"Setting track cuts for the 2010 p+p collisions at 900 GeV"<<endl;}
-      else{cout<<"Setting track cuts for the 2011 p+p collisions at 2.76 TeV"<<endl;}
+      if(fRecAnalysis->DataSet()==2012)cout<<"Setting track cuts for the 2012 p+p collisions at 8 TeV"<<endl;
+      else{
+       if(fRecAnalysis->DataSet()==2013)cout<<"Setting track cuts for the 2013 p+Pb collisions at 5 TeV"<<endl;
+       else{
+         if(fRecAnalysis->DataSet()==2009){cout<<"Setting track cuts for the 2010 p+p collisions at 900 GeV"<<endl;}
+         else{cout<<"Setting track cuts for the 2011 p+p collisions at 2.76 TeV"<<endl;}
+       }
+      }
     }
     //cout<<"Warning:  Have not set 2010 track cuts yet!!"<<endl;
     fEsdtrackCutsITSTPC = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(selectPrimaries);
@@ -164,7 +178,7 @@ void AliAnalysisTaskTotEt::UserCreateOutputObjects()
     fEsdtrackCutsITS =  AliESDtrackCuts::GetStandardITSPureSATrackCuts2010(kTRUE,kFALSE);//we do want primaries but we do not want to require PID info
     fEsdtrackCutsITS->SetName("fEsdTrackCutsITS");
   }
-  if(fRecAnalysis->DataSet()==20100){
+  if(fRecAnalysis->DataSet()==20100 || fRecAnalysis->DataSet()==2011){
     cout<<"Setting track cuts for the 2010 Pb+Pb collisions at 2.76 TeV"<<endl;
     //cout<<"Warning:  Have not set 2010 track cuts yet!!"<<endl;
     fEsdtrackCutsITSTPC = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(selectPrimaries);
@@ -176,6 +190,26 @@ void AliAnalysisTaskTotEt::UserCreateOutputObjects()
     // fEsdtrackCutsITS =  AliESDtrackCuts::GetStandardITSPureSATrackCuts2010(kTRUE,kFALSE);//we do want primaries but we do not want to require PID info
    fEsdtrackCutsITS->SetName("fEsdTrackCutsITS");
   }
+  if(fRecAnalysis->DataSet()==2011){
+    cout<<"Using 2011 acceptance"<<endl;
+    if(fMCAnalysis) fMCAnalysis->ResetEventValues();//actually remakes AliAnalysisEtCuts if AliAnalysisEtCuts doesn't exist
+    fRecAnalysis->ResetEventValues();
+    if(fMCAnalysis && fMCAnalysis->GetCuts()){
+      fMCAnalysis->GetCuts()->SetGeometryEmcalPhiAccMinCut(80);
+      fMCAnalysis->GetCuts()->SetGeometryEmcalPhiAccMaxCut(80+100);
+      fMCAnalysis->GetSelector()->GetCuts()->SetGeometryEmcalPhiAccMinCut(80);
+      fMCAnalysis->GetSelector()->GetCuts()->SetGeometryEmcalPhiAccMaxCut(80+100);
+    }
+    else{if(fMCAnalysis)cerr<<"Error!  MC fCuts does not exist!"<<endl;}
+    if(fRecAnalysis->GetCuts()){
+      fRecAnalysis->GetCuts()->SetGeometryEmcalPhiAccMinCut(80);
+      fRecAnalysis->GetCuts()->SetGeometryEmcalPhiAccMaxCut(80+100);
+      fRecAnalysis->GetSelector()->GetCuts()->SetGeometryEmcalPhiAccMinCut(80);
+      fRecAnalysis->GetSelector()->GetCuts()->SetGeometryEmcalPhiAccMaxCut(80+100);
+    }
+    else{cerr<<"Error!  Reco fCuts does not exist!"<<endl;}
+  }
+
        
        
   fOutputList->Add(fEsdtrackCutsITSTPC);
@@ -229,6 +263,9 @@ void AliAnalysisTaskTotEt::UserExec(Option_t *)
     }
   if(fMCAnalysis)
     {
+      //set the number of tracks matched and the total number of hadrons calculated from the number of tracks matched so we can use them for calculating corrections
+      fMCAnalysis->SetNumberOfChargedHadronsMatched(fRecAnalysis->GetNumberOfChargedHadronsMatched());
+      fMCAnalysis->SetTotalNumberOfChargedHadrons(fRecAnalysis->GetTotalNumberOfChargedHadrons());
       fHistEtRecvsEtMC->Fill(fRecAnalysis->GetTotNeutralEt(), fMCAnalysis->GetTotNeutralEt());
       if(fMCAnalysis->GetTotNeutralEt()) fHistEtRecOverEtMC->Fill(fRecAnalysis->GetTotNeutralEt()/fMCAnalysis->GetTotNeutralEt(), cent->GetCentralityClass10("V0M"));
       if(fMCAnalysis->GetTotNeutralEt()) fHistDiffEtRecEtMCOverEtMC->Fill(fMCAnalysis->GetTotNeutralEt(), (fRecAnalysis->GetTotNeutralEt()-fMCAnalysis->GetTotNeutralEt())/fMCAnalysis->GetTotNeutralEt());