PWG0/trigger/run.C

   1 void Load(const char* taskName, Bool_t debug)
   2 {
   3   TString compileTaskName;
   4   compileTaskName.Form("%s.cxx++", taskName);
   5   if (debug)
   6     compileTaskName += "g";
   7
   8   if (gProof) {
   9     gProof->Load(compileTaskName);
  10   } else
  11     gROOT->Macro(compileTaskName);
  12
  13   // Enable debug printouts
  14   if (debug)
  15   {
  16     AliLog::SetClassDebugLevel(taskName, AliLog::kDebug+2);
  17   }
  18   else
  19     AliLog::SetClassDebugLevel(taskName, AliLog::kWarning);
  20 }
  21
  22 void GetTimes(UInt_t run, UInt_t* startTime = 0, UInt_t* endTime = 0)
  23 {
  24   gSystem->Load("libXMLParser");
  25   gSystem->Load("libXMLIO");
  26   gSystem->Load("libCDB");
  27   gSystem->Load("libSTEER");
  28
  29   AliCDBManager * man = AliCDBManager::Instance();
  30   man->SetDefaultStorage("raw://");
  31   man->SetRun(run);
  32   AliCDBPath cdb("GRP", "GRP", "Data");
  33   obj = man->Get(cdb);
  34   grp = (AliGRPObject*) obj->GetObject();
  35
  36   if (startTime)
  37     *startTime = grp->GetTimeStart();
  38   if (endTime)
  39     *endTime = grp->GetTimeEnd();
  40
  41   Printf("Got start and endtime from OCDB: %d, %d", grp->GetTimeStart(), grp->GetTimeEnd());
  42 }
  43
  44 void run(const Char_t* data, Int_t nRuns=20, Int_t offset=0, Bool_t aDebug = kFALSE, Int_t aProof = kFALSE, UInt_t startTime = 0, UInt_t endTime = 0, const char* option = "")
  45 {
  46   // aProof option: 0 no proof
  47   //                1 proof with chain
  48   //                2 proof with dataset
  49   //
  50   // option is passed to the task(s)
  51
  52 /*
  53  .x run.C("/PWG0/jgrosseo/ERP_run98097", -1, 0, kFALSE, 2, 1258045012, 1258045458)
  54  .x run.C("/PWG0/jgrosseo/ERP_run98576", -1, 0, kFALSE, 2, 1258123911, 1258124103)
  55  .x run.C("/PWG0/jgrosseo/ERP_run98569", -1, 0, kFALSE, 2, 1258122187, 1258122524)
  56  .x run.C("/PWG0/jgrosseo/run101235", -1, 0, kFALSE, 2, 1258821541, 1258822595)
  57  .x run.C("/PWG0/jgrosseo/bgRuns_101424to102689_allITS_2", -1, 0, kFALSE, 2, 1258950519, 1259567587)
  58  .x run.C("/PWG0/jgrosseo/run000102556", -1, 0, kFALSE, 2, 1259481473, 1259494364)
  59
  60   timestamps:
  61  .x run.C("/ALIREC/aliprod/run101498", -1, 0, kFALSE, 2, 1258990726, 1258993311)
  62   orbits:
  63  .x run.C("/ALIREC/aliprod/run101498", -1, 0, kFALSE, 2, 13587, 16749493)
  64
  65  .x run.C("/ALIREC/aliprod/run104044", -1, 0, 1, 2, 1260078397, 1260082975)
  66
  67  .x run.C("/ALIREC/aliprod/run104065", -1, 0, 1, 2, 1260084071, 1260084250)
  68  .x run.C("/ALIREC/aliprod/run104065", -1, 0, 1, 2, 1353, 1504)
  69
  70  .x run.C("/ALIREC/aliprod/run104321", -1, 0, 1, 2, 1260237771, 1260249377)
  71
  72 */
  73
  74   if (nRuns < 0)
  75     nRuns = 1234567890;
  76
  77   if (aProof)
  78   {
  79     TProof::Mgr("alicecaf")->SetROOTVersion("v5-24-00a");
  80     TProof::Open("alicecaf");
  81     //gProof->SetParallel(1);
  82
  83     // Enable the needed package
  84     if (0)
  85     {
  86       gProof->UploadPackage("$ALICE_ROOT/STEERBase");
  87       gProof->EnablePackage("$ALICE_ROOT/STEERBase");
  88       gProof->UploadPackage("$ALICE_ROOT/ESD");
  89       gProof->EnablePackage("$ALICE_ROOT/ESD");
  90       gProof->UploadPackage("$ALICE_ROOT/AOD");
  91       gProof->EnablePackage("$ALICE_ROOT/AOD");
  92       gProof->UploadPackage("$ALICE_ROOT/ANALYSIS");
  93       gProof->EnablePackage("$ALICE_ROOT/ANALYSIS");
  94       gProof->UploadPackage("$ALICE_ROOT/ANALYSISalice");
  95       gProof->EnablePackage("$ALICE_ROOT/ANALYSISalice");
  96     }
  97     else
  98     {
  99       gProof->UploadPackage("$ALICE_ROOT/AF-v4-18-12-AN.par");
 100       gProof->EnablePackage("AF-v4-18-12-AN");
 101     }
 102
 103     gProof->UploadPackage("$ALICE_ROOT/PWG0base");
 104     gProof->EnablePackage("$ALICE_ROOT/PWG0base");
 105   }
 106   else
 107   {
 108     gSystem->AddIncludePath("-I${ALICE_ROOT}/include/ -I${ALICE_ROOT}/PWG0/ -I${ALICE_ROOT}/PWG0/dNdEta/");
 109     gSystem->Load("libVMC");
 110     gSystem->Load("libTree");
 111     gSystem->Load("libProof");
 112     gSystem->Load("libSTEERBase");
 113     gSystem->Load("libESD");
 114     gSystem->Load("libAOD");
 115     gSystem->Load("libANALYSIS");
 116     gSystem->Load("libANALYSISalice");
 117     gSystem->Load("libPWG0base");
 118   }
 119
 120   if (startTime == endTime && startTime > 0)
 121   {
 122     // get times from OCDB, startTime must be run number
 123
 124     // WARNING only works if your par files loaded above are compatible with the libraries loaded here...
 125     GetTimes(startTime, &startTime, &endTime);
 126   }
 127
 128   // Create the analysis manager
 129   mgr = new AliAnalysisManager;
 130
 131   // Add ESD handler
 132   AliESDInputHandler* esdH = new AliESDInputHandler;
 133   esdH->SetInactiveBranches("AliESDACORDE ALIESDTZERO AliRawDataErrorLogs CaloClusters Cascades EMCALCells EMCALTrigger ESDfriend Kinks Kinks Cascades AliESDTZERO ALIESDACORDE MuonTracks TrdTracks CaloClusters");
 134   mgr->SetInputEventHandler(esdH);
 135
 136   cInput = mgr->GetCommonInputContainer();
 137
 138   Load("AliTriggerTask", aDebug);
 139   TString optStr(option);
 140
 141   // remove SAVE option if set
 142   Bool_t save = kFALSE;
 143   if (optStr.Contains("SAVE"))
 144   {
 145     optStr = optStr(0,optStr.Index("SAVE")) + optStr(optStr.Index("SAVE")+4, optStr.Length());
 146     save = kTRUE;
 147   }
 148
 149   task = new AliTriggerTask(optStr);
 150   task->SetTimes(startTime, endTime);
 151   //task->SetUseOrbits(kTRUE);
 152
 153   mgr->AddTask(task);
 154
 155   // Attach input
 156   mgr->ConnectInput(task, 0, cInput);
 157
 158   // Attach output
 159   cOutput = mgr->CreateContainer("cOutput", TList::Class(), AliAnalysisManager::kOutputContainer);
 160   mgr->ConnectOutput(task, 0, cOutput);
 161
 162   // Enable debug printouts
 163   if (aDebug)
 164     mgr->SetDebugLevel(2);
 165
 166   // Run analysis
 167   mgr->InitAnalysis();
 168   mgr->PrintStatus();
 169
 170   if (aProof == 2)
 171   {
 172     // process dataset
 173
 174     mgr->StartAnalysis("proof", data, nRuns, offset);
 175
 176     if (save)
 177     {
 178       TString path("maps/");
 179       path += TString(data).Tokenize("/")->Last()->GetName();
 180
 181       gSystem->mkdir(path, kTRUE);
 182       gSystem->Rename("trigger.root", path + "/trigger.root");
 183
 184       Printf(">>>>> Moved files to %s", path.Data());
 185     }
 186   }
 187   else if (aProof == 3)
 188   {
 189     gROOT->ProcessLine(".L CreateChainFromDataSet.C");
 190     ds = gProof->GetDataSet(data)->GetStagedSubset();
 191     chain = CreateChainFromDataSet(ds);
 192     mgr->StartAnalysis("local", chain, nRuns, offset);
 193   }
 194   else
 195   {
 196     // Create chain of input files
 197     gROOT->LoadMacro("../CreateESDChain.C");
 198
 199     chain = CreateESDChain(data, nRuns, offset);
 200     //chain = CreateChain("TE", data, nRuns, offset);
 201
 202     mgr->StartAnalysis((aProof > 0) ? "proof" : "local", chain);
 203   }
 204
 205 }