Use new profiling tools in local mode, set proper name for tasks when running with...

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

diff --git a/PWG4/macros/AnalysisTrainPWG4Jets.C b/PWG4/macros/AnalysisTrainPWG4Jets.C
index b34e54cdf1a88945c027ef2267d401a76c0e2ce0..3017aeb53be8fa79c86c82ae10e10b1fda5d35f7 100644 (file)
--- a/PWG4/macros/AnalysisTrainPWG4Jets.C
+++ b/PWG4/macros/AnalysisTrainPWG4Jets.C
@@ -226,8 +226,11 @@ void AnalysisTrainPWG4Jets(const char *analysis_mode="local",
    AliAnalysisManager *mgr  = new AliAnalysisManager("PWG4Train", "pwg4 mini train");
    if (kCommonOutputFileName.Length()>0)mgr->SetCommonFileName(kCommonOutputFileName.Data());
    if (kProofSaveToAlien) mgr->SetSpecialOutputLocation(kProofOutdir);
+   mgr->SetNSysInfo(0);
    if (!strcmp(plugin_mode, "test")) mgr->SetNSysInfo(1);
    if (kUseSysInfo)mgr->SetNSysInfo(kUseSysInfo);
+   
+
    // Load analysis specific libraries
    if (!LoadAnalysisLibraries(smode)) {
       ::Error("AnalysisTrain", "Could not load analysis libraries");
@@ -246,7 +249,7 @@ void AnalysisTrainPWG4Jets(const char *analysis_mode="local",
       AliESDInputHandler *esdHandler = new AliESDInputHandler();
       if (kUseESDTags) esdHandler->SetReadTags();
       mgr->SetInputEventHandler(esdHandler);       
-      if(iPWG4PtQATPC)esdHandler->SetActiveBranches("ESDfriend");
+      //      if(iPWG4PtQATPC&& !kTrainName.Contains("pass5"))esdHandler->SetActiveBranches("ESDfriend");
    }
 
    // Monte Carlo handler
@@ -535,6 +538,11 @@ void AnalysisTrainPWG4Jets(const char *analysis_mode="local",
      }
      AliLog::SetGlobalLogLevel(AliLog::kError);
      StartAnalysis(smode, chain);
+     if((kUseSysInfo>0&&smode=="LOCAL")||!strcmp(plugin_mode, "test")){
+       for(int i = 0;i < mgr->GetTopTasks()->GetEntries();i++){
+        mgr->ProfileTask(i);
+       }
+     }
      if (!strcmp(plugin_mode, "offline")&&smode=="GRID"){
        // Offline mode path files
        //      PatchJDL();