ANALYSIS/macros/TaskBuzzer.C

   1 #include "Riostream.h"
   2 #define BIT(n)       (1ULL << (n))
   3 /*
   4    This macro should be called by a script TaskBuzzer.sh like:
   5  export ARG="TaskBuzzer.C($1)"
   6  aliroot -b -q $ARG
   7    Call: TaskBuzzer.sh "kCar00" or: TaskBuzzer.sh "kTender|kCar01"
   8 */
   9
  10 TString config_name;
  11 const char *ConfigName(UInt_t);
  12 void LoadLibraries();
  13 void AddAnalysisTasks();
  14 TChain *CreateChain(const char *, const char *);
  15
  16 enum EBuzzerConfig {
  17    kBaseline         = 0x00,     // Baseline EVstat
  18    kCar00            = BIT(0),   // QAsym
  19    kCar01            = BIT(1),   // VZERO
  20    kCar02            = BIT(2),   // Vertex
  21    kCar03            = BIT(3),   // SPD
  22    kCar04            = BIT(4),   // SDD
  23    kCar05            = BIT(5),   // SSDdEdx
  24    kCar06            = BIT(6),   // FMD
  25    kCar07            = BIT(7),   // ITS
  26    kCar08            = BIT(8),   // TPC
  27    kCar09            = BIT(9),   // TRD
  28    kCar10            = BIT(10),  // CALO
  29    kCar11            = BIT(11),  // MUONTrig
  30    kTender           = BIT(30),  // TENDER
  31    kAOD              = BIT(31)   // AOD
  32 };
  33
  34 void TaskBuzzer(UInt_t config, const char *collection="wn.xml")
  35 {
  36   printf("Running buzzer with: %s\n", ConfigName(config));
  37   TGrid::Connect("alien://");
  38   if (!gGrid || !gGrid->IsConnected()) {
  39     ::Error("PilotAnalysis", "No grid connection");
  40     return;
  41   }
  42   TChain *chain = CreateChain(collection, "ESD");
  43   // Load libraries
  44   gSystem->SetIncludePath("-I. -I$ROOTSYS/include -I$ALICE_ROOT/include -I$ALICE_ROOT -I$ALICE_ROOT/ITS -I$ALICE_ROOT/TRD");
  45   LoadLibraries();
  46   // Create manager
  47   AliAnalysisManager *mgr  = new AliAnalysisManager("ESDfiltering", "Production train");
  48   mgr->SetNSysInfo(100);
  49   // Input handler
  50   AliESDInputHandlerRP *esdHandler = new AliESDInputHandlerRP();
  51 //  AliESDInputHandler *esdHandler = new AliESDInputHandler();
  52   esdHandler->SetReadFriends(kTRUE);
  53   esdHandler->SetActiveBranches("ESDfriend");
  54   mgr->SetInputEventHandler(esdHandler);
  55   if (config & kAOD) {
  56      // AOD output handler
  57      AliAODHandler* aodHandler   = new AliAODHandler();
  58      aodHandler->SetOutputFileName("AliAOD.root");
  59      if (!mergeExcludes.IsNull()) mergeExcludes += " ";
  60      mergeExcludes += "AliAOD.root";
  61      mgr->SetOutputEventHandler(aodHandler);
  62   }
  63   // AnalysisTasks
  64   AddAnalysisTasks(config);
  65
  66   mgr->SetDebugLevel(1);
  67   if (mgr->InitAnalysis()) {
  68     mgr->PrintStatus();
  69     mgr->StartAnalysis("local", chain);
  70   }
  71 }
  72
  73 //______________________________________________________________________________
  74 void AddAnalysisTasks(UInt_t config)
  75 {
  76   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  77   mgr->SetCommonFileName("QAresults.root");
  78   //
  79   // Event Statistics (Jan Fiete)
  80   //
  81
  82    gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
  83    AliPhysicsSelectionTask* physSelTask = AddTaskPhysicsSelection();
  84    mgr->RegisterExtraFile("event_stat.root");
  85   //
  86   // Tender and supplies
  87   //
  88   if (config & kTender) {
  89       gROOT->LoadMacro("$ALICE_ROOT/TENDER/TenderSupplies/AddTaskTender.C");
  90       AliAnalysisTaskSE *tender = AddTaskTender();
  91       // SelectCollisionCandidates ?? Tender has to derive from TaskSE
  92       tender->SelectCollisionCandidates();
  93       tender->SetDebugLevel(2);
  94   }
  95   // AOD creation with collision events
  96   if (config & kAOD) {
  97     gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskESDFilter.C");
  98     mgr->RegisterExtraFile("AliAOD.Muons.root");
  99     mgr->RegisterExtraFile("AliAOD.Dimuons.root");
 100     AliAnalysisTaskESDfilter *taskesdfilter = AddTaskESDFilter(kFALSE, kTRUE, kTRUE, doEventStat);
 101   }
 102   // TPC QA (E. Sicking)
 103   //
 104   if (config & kCar00) {
 105     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddTaskQAsym.C");
 106     AliAnalysisTaskSE * taskqasim = AddTaskQAsym(0);
 107     taskqasim->SelectCollisionCandidates();
 108   }
 109   //
 110   // VZERO QA  (C. Cheshkov)
 111   //
 112   if (config & kCar01) {
 113     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddTaskVZEROQA.C");
 114     AliAnalysisTaskSE * taskv0qa = AddTaskVZEROQA(0);
 115 //  taskv0qa->SelectCollisionCandidates();
 116   }
 117   // Vertexing (A. Dainese)
 118   //
 119   if (config & kCar02) {
 120     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/macros/AddTaskVertexESD.C");
 121     AliAnalysisTaskVertexESD* taskvertexesd =  AddTaskVertexESD();
 122     taskvertexesd->SelectCollisionCandidates();
 123   }
 124   //
 125   // SPD (A. Mastroserio)
 126   //
 127   if (config & kCar03) {
 128     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddTaskSPDQA.C");
 129     AliAnalysisTaskSE* taskspdqa = AddTaskSPDQA();
 130     taskspdqa->SelectCollisionCandidates();
 131   }
 132   //
 133   // SDD (F. Prino)
 134   //
 135   if (config & kCar04) {
 136     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddSDDPoints.C");
 137     AliAnalysisTaskSE* tasksdd = AddSDDPoints();
 138     tasksdd->SelectCollisionCandidates();
 139   }
 140   //
 141   // SSD dEdx (Marek Chojnacki)
 142   //
 143   if (config & kCar05) {
 144     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddTaskdEdxSSDQA.C");
 145     AliAnalysisTaskSE* taskssddedx = AddTaskdEdxSSDQA();
 146     taskssddedx->SelectCollisionCandidates();
 147   }
 148
 149   // FMD (Hans Hjersing Dalsgaard)
 150   //
 151   if (config & kCar06) {
 152     gROOT->LoadMacro("$ALICE_ROOT/PWGPP/PilotTrain/AddTaskFMD.C");
 153     AliAnalysisTaskSE* taskfmd = AddTaskFMD();
 154     taskfmd->SelectCollisionCandidates();
 155   }
 156   //
 157   // ITS
 158   //
 159   if (config & kCar07) {
 160       gROOT->LoadMacro("$ALICE_ROOT/PWGPP/macros/AddTaskPerformanceITS.C");
 161       AliAnalysisTaskITSTrackingCheck *itsQA = AddTaskPerformanceITS(kFALSE);
 162   }
 163   //
 164   // TPC (Jacek Otwinowski)
 165   //
 166   if (config & kCar08) {
 167     gROOT->LoadMacro("$(ALICE_ROOT)/PWGPP/TPC/macros/AddTaskPerformanceTPCQA.C");
 168     AliPerformanceTask *tpcQA = AddTaskPerformanceTPCQA(kFALSE, kTRUE);
 169   }
 170   //
 171   // TRD (Alex Bercuci, M. Fasel)
 172   //
 173   if(config & kCar09) {
 174       gROOT->LoadMacro("$ALICE_ROOT/PWGPP/macros/AddTrainPerformanceTRD.C");
 175       AddTrainPerformanceTRD("ALL");
 176   }
 177
 178   //
 179   // Calorimetry (Gustavo Conesa)
 180   //
 181
 182   if(config & kCar10) {
 183       gROOT->LoadMacro("$ALICE_ROOT/PWG4/macros/QA/AddTaskCalorimeterQA.C");
 184       AliAnalysisTaskParticleCorrelation *taskCaloQA = AddTaskCalorimeterQA("ESD", kTRUE, kFALSE);
 185       taskCaloQA->SetDebugLevel(0);
 186   }
 187
 188   //
 189   // Muon Trigger
 190   //
 191
 192   if(config & kCar11) {
 193       gROOT->LoadMacro("$ALICE_ROOT/PWGPP/macros/AddTaskMTRchamberEfficiency.C");
 194       AliAnalysisTaskTrigChEff *taskMuonTrig = AddTaskMTRchamberEfficiency();
 195   }
 196
 197   //
 198   // Muon Efficiency
 199   //
 200 /*
 201   if(doMUONEff) {
 202       gROOT->LoadMacro("$ALICE_ROOT/PWG3/muondep/AddTaskMUONTrackingEfficiency.C");
 203       AliAnalysisTaskMuonTrackingEff *taskMuonTrackEff = AddTaskMUONTrackingEfficiency();
 204   }
 205
 206   //
 207   // V0-Decay Reconstruction (Ana Marin)
 208   //
 209
 210   if (doV0) {
 211       gROOT->LoadMacro("$ALICE_ROOT/PWGPP/macros/AddTaskV0QA.C");
 212       AliAnalysisTaskV0QA *taskv0QA = AddTaskV0QA(kFALSE);
 213   }
 214 */
 215 }
 216
 217 //______________________________________________________________________________
 218 const char *ConfigName(UInt_t config)
 219 {
 220    config_name = "EVstat";
 221    if (config & kTender) config_name += "+TENDER";
 222    if (config & kAOD)    config_name   += "+AOD";
 223    if (config & kCar00)  config_name += "+QAsym";
 224    if (config & kCar01)  config_name += "+VZERO";
 225    if (config & kCar02)  config_name += "+Vertex";
 226    if (config & kCar03)  config_name += "+SPD";
 227    if (config & kCar04)  config_name += "+SDD";
 228    if (config & kCar05)  config_name += "+SSDdEdx";
 229    if (config & kCar06)  config_name += "+FMD";
 230    if (config & kCar07)  config_name += "+ITS";
 231    if (config & kCar08)  config_name += "+TPC";
 232    if (config & kCar09)  config_name += "+TRD";
 233    if (config & kCar10)  config_name += "+CALO";
 234    if (config & kCar11)  config_name += "+MUONTrig";
 235    return config_name.Data();
 236 }
 237
 238 //______________________________________________________________________________
 239 void LoadLibraries()
 240 {
 241   gSystem->Load("libANALYSIS");
 242   gSystem->Load("libANALYSISalice");
 243   gSystem->Load("libCORRFW");
 244   gSystem->Load("libTender");
 245   gSystem->Load("libTenderSupplies");
 246   gSystem->Load("libPWG0base");
 247   gSystem->Load("libPWG0dep");
 248   gSystem->Load("libPWG0selectors");
 249   gSystem->Load("libPWGPP");
 250   gSystem->Load("libPWG2");
 251   gSystem->Load("libPWG2forward");
 252   gSystem->Load("libEMCALUtils");
 253   gSystem->Load("libPWG4PartCorrBase");
 254   gSystem->Load("libPWG4PartCorrDep");
 255   gSystem->Load("libPWGHFbase");
 256   gSystem->Load("libPWGmuon");
 257   gSystem->Load("libPWGmuondep");
 258 // fast xrootd reading enabled
 259    printf("!!! You requested FastRead option. Using xrootd flags to reduce timeouts. Note that this may skip some files that could be accessed !!!");
 260    gEnv->SetValue("XNet.ConnectTimeout",10);
 261    gEnv->SetValue("XNet.RequestTimeout",10);
 262    gEnv->SetValue("XNet.MaxRedirectCount",2);
 263    gEnv->SetValue("XNet.ReconnectTimeout",10);
 264    gEnv->SetValue("XNet.FirstConnectMaxCnt",1);
 265 }
 266
 267 //________________________________________________________________________________
 268 TChain* CreateChain(const char *xmlfile, const char *type)
 269 {
 270 // Create a chain using url's from xml file
 271    TString treename = type;
 272    treename.ToLower();
 273    treename += "Tree";
 274    printf("***************************************\n");
 275    printf("    Getting chain of trees %s\n", treename.Data());
 276    printf("***************************************\n");
 277    TAlienCollection *coll = TAlienCollection::Open(xmlfile);
 278    if (!coll) {
 279       ::Error("CreateChain", "Cannot create an AliEn collection from %s", xmlfile);
 280       return NULL;
 281    }
 282    TChain *chain = new TChain(treename);
 283    coll->Reset();
 284    while (coll->Next()) {
 285       chain->Add(coll->GetTURL(""));
 286    }
 287    if (!chain->GetNtrees()) {
 288       ::Error("CreateChain", "No tree found from collection %s", xmlfile);
 289       return NULL;
 290    }
 291    return chain;
 292 }