Add the connected pad info to the simulation (Raphaelle)

[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskPWG4HighPtQATPConly.C

diff --git a/PWG4/macros/AddTaskPWG4HighPtQATPConly.C b/PWG4/macros/AddTaskPWG4HighPtQATPConly.C
index 8287a31ac351969a33dfb2c90ba8944bdf5565d4..75540c23231588e5e8c23727a10b1c7c0df11016 100644 (file)
--- a/PWG4/macros/AddTaskPWG4HighPtQATPConly.C
+++ b/PWG4/macros/AddTaskPWG4HighPtQATPConly.C
@@ -43,9 +43,9 @@ AliPWG4HighPtQATPConly* AddTaskPWG4HighPtQATPConly(int cuts=1)//1: Standard Cuts
     trackCuts->SetDCAToVertex2D(kTRUE);
   }
   else if(cuts==2) {
+    trackCuts=trackCuts->GetStandardITSTPCTrackCuts2009(kTRUE);//Primary Track Selection
     trackCuts->SetEtaRange(-0.9,0.9);
     trackCuts->SetPtRange(0.15, 1e10);
-    trackCuts->GetStandardITSTPCTrackCuts2009(kTRUE);//Primary Track Selection
     trackCuts->SetRequireITSRefit(kFALSE);
   }
 
@@ -64,7 +64,7 @@ AliPWG4HighPtQATPConly* AddTaskPWG4HighPtQATPConly(int cuts=1)//1: Standard Cuts
     trackCutsITS->SetDCAToVertex2D(kTRUE); 
   }
  else if(cuts==2) {
-   trackCutsITS->GetStandardITSTPCTrackCuts2009(kTRUE);//Primary Track Selection
+   trackCutsITS=trackCutsITS->GetStandardITSTPCTrackCuts2009(kTRUE);//Primary Track Selection
    trackCutsITS->SetEtaRange(-0.9,0.9);
    trackCutsITS->SetPtRange(0.15, 1e10);
  }
@@ -73,15 +73,14 @@ AliPWG4HighPtQATPConly* AddTaskPWG4HighPtQATPConly(int cuts=1)//1: Standard Cuts
   AliPWG4HighPtQATPConly *taskPWG4QA = new AliPWG4HighPtQATPConly(Form("AliPWG4HighPtQATPConly%d",cuts));
   taskPWG4QA->SetCuts(trackCuts);
   taskPWG4QA->SetCutsITS(trackCutsITS);
-  taskPWG4QA->SetMaxCosmicAngle(0.008);
-  
+  taskPWG4QA->SetMaxCosmicAngle(1.);//0.008);
+  taskPWG4QA->SetCutType(cuts);
  
   // E. Create ONLY the output containers for the data produced by the task.
   // Get and connect other common input/output containers via the manager as below
   //==============================================================================
 
   //------ input data ------
-  //  AliAnalysisDataContainer *cinput0  = mgr->GetCommonInputContainer();
   TString outputfile = "";
   outputfile = AliAnalysisManager::GetCommonFileName();
   outputfile += Form(":PWG4_HighPtQATPConly%d",cuts);
@@ -90,27 +89,29 @@ AliPWG4HighPtQATPConly* AddTaskPWG4HighPtQATPConly(int cuts=1)//1: Standard Cuts
   AliAnalysisDataContainer *cout_hist1;
   AliAnalysisDataContainer *cout_hist2;
   AliAnalysisDataContainer *cout_hist3;
+  AliAnalysisDataContainer *cout_cuts0;
+  AliAnalysisDataContainer *cout_cuts1;
 
-  if(cuts==1) {
-    cout_hist0 = mgr->CreateContainer("qa_histsCuts1", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
-    cout_hist1 = mgr->CreateContainer("qa_histsTPCCuts1", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
-    cout_hist2 = mgr->CreateContainer("qa_histsITSCuts1", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
-    cout_hist3 = mgr->CreateContainer("qa_histsCosmicsCuts1", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
-  }
-  else if(cuts==2) {
-    cout_hist0 = mgr->CreateContainer("qa_histsCuts2", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
-    cout_hist1 = mgr->CreateContainer("qa_histsTPCCuts2", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
-    cout_hist2 = mgr->CreateContainer("qa_histsITSCuts2", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
-    cout_hist3 = mgr->CreateContainer("qa_histsCosmicsCuts2", TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
-  }
- 
+  cout_hist0 = mgr->CreateContainer(Form("qa_histsCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
+  cout_hist1 = mgr->CreateContainer(Form("qa_histsTPCCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
+  cout_hist2 = mgr->CreateContainer(Form("qa_histsITSCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
+  cout_hist3 = mgr->CreateContainer(Form("qa_histsCosmicsCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
+  cout_cuts0 = mgr->CreateContainer(Form("qa_trackCuts%d",cuts), AliESDtrackCuts::Class(), AliAnalysisManager::kParamContainer,outputfile);
+  cout_cuts1 = mgr->CreateContainer(Form("qa_trackCutsITS%d",cuts), AliESDtrackCuts::Class(), AliAnalysisManager::kParamContainer,outputfile);
+
+  //Add task to manager
   mgr->AddTask(taskPWG4QA);
 
+  //Connect input containter to manager
   mgr->ConnectInput(taskPWG4QA,0,mgr->GetCommonInputContainer());
+
+  //Connect output containers to manager
   mgr->ConnectOutput(taskPWG4QA,0,cout_hist0);
   mgr->ConnectOutput(taskPWG4QA,1,cout_hist1);
   mgr->ConnectOutput(taskPWG4QA,2,cout_hist2);
   mgr->ConnectOutput(taskPWG4QA,3,cout_hist3);
+  mgr->ConnectOutput(taskPWG4QA,4,cout_cuts0);
+  mgr->ConnectOutput(taskPWG4QA,5,cout_cuts1);
 
   // Return task pointer at the end
   return taskPWG4QA;