Updating SDD task for calibration (Ruben) - see comment #27 in https://savannah.cern...

[u/mrichter/AliRoot.git] / PWGPP / macros / RecoQAtrain.C

diff --git a/PWGPP/macros/RecoQAtrain.C b/PWGPP/macros/RecoQAtrain.C
index 89e94bddbc51d03a5f018559ddc8e579f004509e..4873dd9467f7705d6acdb332bf665ba327beef23 100644 (file)
--- a/PWGPP/macros/RecoQAtrain.C
+++ b/PWGPP/macros/RecoQAtrain.C
@@ -54,13 +54,6 @@ void AddAnalysisTasks(Bool_t writeITSTP, Bool_t useTPCcrv)
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   mgr->SetCommonFileName("RecoQAresults.root");
   //
-  if (writeITSTP) {
-    AliAODHandler* aodHandler = new AliAODHandler();
-    aodHandler->SetOutputFileName( "AODtpITS.root" );
-    aodHandler->SetCreateNonStandardAOD();
-    mgr->SetOutputEventHandler(aodHandler);
-  }
-  //
   // Event Statistics (Jan Fiete)
   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
   AliPhysicsSelectionTask* physSelTask = AddTaskPhysicsSelection(kFALSE /*MC*/);
@@ -75,6 +68,14 @@ void AddAnalysisTasks(Bool_t writeITSTP, Bool_t useTPCcrv)
   itsAlign->SetDoSDDDriftTime(kFALSE);
   itsAlign->SetMinMaxMult(20.,1070.);
   itsAlign->SetUseTPCMomentum(useTPCcrv);
-  //
   itsAlign->SetDoFillTPTree(writeITSTP);
+  //
+  if (writeITSTP) {
+    AliAnalysisDataContainer *coutputTP = mgr->CreateContainer("ITSSumTP",
+                                                              TTree::Class(),
+                                                              AliAnalysisManager::kOutputContainer,
+                                                              "AODtpITS.root" );
+    mgr->ConnectOutput(itsAlign, 2, coutputTP);
+  }
+  //
 }