]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG1/TRD/run.C
bug corrected
[u/mrichter/AliRoot.git] / PWG1 / TRD / run.C
index 8c7ed312082435a747fc65d999ab7c2767375e48..801c68dfe4f07365d5172ba89d4b37ca72728c68 100644 (file)
@@ -1,8 +1,9 @@
 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
 // 
 // Usage:
-//   run.C(tasks, files)
-//   tasks : "ALL" or one/more of the following:
+//   run.C(optList, files, nev, first, runNo, ocdb_uri, grp_uri)
+//
+//   optList : "ALL" [default] or one/more of the following:
 //     "EFF"  : TRD Tracking Efficiency 
 //     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations
 //     "MULT"  : TRD single track selection
 //     "ALGN" : TRD alignment
 //     "PID"  : TRD PID - pion efficiency 
 //     "PIDR" : TRD PID - reference data
+//     "V0"   : monitor V0 performance for use in TRD PID calibration
 //     "DET"  : Basic TRD Detector checks
+//      ****** SPECIAL OPTIONS **********
 //     "NOFR" : Data set does not have AliESDfriends.root 
-//     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
-//              on MC information are switched off
+//     "NOMC" : Data set does not have Monte Carlo Informations (default have MC), 
 //
+//     files : the list of ESD files to be processed [default AliESds.root from cwd]
+//     nev   : number of events to be processed [default all]
+//     first : first event to process [default 0]
+//     runNo : run number [default 0]
+//     ocdb_uri : OCDB location [default local, $ALICE_ROOT]. In case of AliEn the syntax should be of the form
+//                 alien://folder=/alice/data/2010/OCDB?user=username?cacheF=yourDirectory?cacheS=yourSize
+//     grp_uri  : GRP/GRP/Data location [default cwd]. In case of AliEn the syntax should be of the form
+//                 alien://folder=/alice/data/2010/OCDB?user=username?cacheF=yourDirectory?cacheS=yourSize
 // In compiled mode : 
 // Don't forget to load first the libraries
 // gSystem->Load("libMemStat.so")
 // gSystem->Load("libMemStatGui.so")
 // gSystem->Load("libANALYSIS.so")
 // gSystem->Load("libANALYSISalice.so")
+// gSystem->Load("libTENDER.so");
 // gSystem->Load("libPWG1.so");
 // gSystem->Load("libNetx.so") ;
 // gSystem->Load("libRAliEn.so");
 #include "TRD/AliTRDtrackerV1.h"
 #include "TRD/AliTRDcalibDB.h"
 
-#include "PWG1/TRD/macros/AliTRDperformanceTrain.h"
 #include "PWG1/TRD/macros/AddTRDcheckESD.C"
 #include "PWG1/TRD/macros/AddTRDinfoGen.C"
 #include "PWG1/TRD/macros/AddTRDcheckDET.C"
 #include "PWG1/TRD/macros/AddTRDefficiency.C"
 #include "PWG1/TRD/macros/AddTRDresolution.C"
 #include "PWG1/TRD/macros/AddTRDcheckPID.C"
-
 #endif
 
-#include "macros/AliTRDperformanceTrain.h"
-
-
 Bool_t MEM = kFALSE;
 
-TChain* MakeChainLST(const char* filename = 0x0);
-TChain* MakeChainXML(const char* filename = 0x0);
-void run(Char_t *optList="ALL", const Char_t *files=0x0, Long64_t nev=1234567890, Long64_t first = 0, Int_t runNo=0, const Char_t *ocdb_uri="local://$ALICE_ROOT/OCDB", const Char_t *grp_uri=Form("local://%s", gSystem->pwd()))
+TChain* MakeChainLST(const char* filename = NULL);
+TChain* MakeChainXML(const char* filename = NULL);
+Bool_t UseMC(Char_t *opt);
+Bool_t UseFriends(Char_t *opt);
+void run(Char_t *optList="ALL", Int_t run, const Char_t *files=NULL, Long64_t nev=1234567890, Long64_t first = 0)
 {
-  TMemStat *mem = 0x0;
+  TMemStat *mem = NULL;
   if(MEM){ 
     if(gSystem->Load("libMemStat.so")<0) return;
     if(gSystem->Load("libMemStatGui.so")<0) return;
@@ -94,35 +102,18 @@ void run(Char_t *optList="ALL", const Char_t *files=0x0, Long64_t nev=1234567890
   timer.Start();
 
   // VERY GENERAL SETTINGS
-  AliLog::SetGlobalLogLevel(AliLog::kError);
+  //AliLog::SetGlobalLogLevel(AliLog::kError);
   if(gSystem->Load("libANALYSIS.so")<0) return;
   if(gSystem->Load("libANALYSISalice.so")<0) return;
   if(gSystem->Load("libTENDER.so")<0) return;
   if(gSystem->Load("libPWG1.so")<0) return;
+  if(gSystem->Load("libCORRFW.so")<0) return;
 
-  Bool_t fHasMCdata =  HasReadMCData(optList);
-  Bool_t fHasFriends = HasReadFriendData(optList);
+  Bool_t fHasMCdata = UseMC(optList);
+  Bool_t fHasFriends = UseFriends(optList);
   
-  // INITIALIZATION OF RUNNING ENVIRONMENT
-  // initialize OCDB manager
-  AliCDBManager *cdbManager = AliCDBManager::Instance();
-  cdbManager->SetDefaultStorage(ocdb_uri);
-  if(!cdbManager->IsDefaultStorageSet()){
-    Error("run.C", "Error setting OCDB.");
-    return;
-  }
-  cdbManager->SetRun(runNo);
-  cdbManager->SetSpecificStorage("GRP/GRP/Data", grp_uri);
-  cdbManager->SetCacheFlag(kFALSE);
-  // initialize magnetic field from the GRP manager.
-  AliGRPManager grpMan;
-  if(!grpMan.ReadGRPEntry()) return;
-  if(!grpMan.SetMagField()) return;
-  //AliRunInfo *runInfo = grpMan.GetRunInfo();
-  AliGeomManager::LoadGeometry();
-
   // DEFINE DATA CHAIN
-  TChain *chain = 0x0;
+  TChain *chain = NULL;
   if(!files) chain = MakeChainLST();
   else{
     TString fn(files);
@@ -140,11 +131,20 @@ void run(Char_t *optList="ALL", const Char_t *files=0x0, Long64_t nev=1234567890
   AliAnalysisManager *mgr = new AliAnalysisManager("TRD Reconstruction Performance & Calibration");
   AliESDInputHandlerRP *esdH(NULL);
   mgr->SetInputEventHandler(esdH = new AliESDInputHandlerRP);
-  esdH->SetReadFriends(kTRUE);
-  esdH->SetActiveBranches("ESDfriend");
+  if(fHasFriends){
+    esdH->SetReadFriends(kTRUE);
+    esdH->SetActiveBranches("ESDfriend");
+  }
   AliMCEventHandler *mcH(NULL);
   if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
   //mgr->SetDebugLevel(10);
+  mgr->SetSkipTerminate(kTRUE);
+
+  // add CDB task
+  gROOT->LoadMacro("$ALICE_ROOT/PWG1/PilotTrain/AddTaskCDBconnect.C");
+  AliTaskCDBconnect *taskCDB = AddTaskCDBconnect();
+  if (!taskCDB) return;
+  taskCDB->SetRunNumber(run);
 
   gROOT->LoadMacro("$ALICE_ROOT/PWG1/macros/AddTrainPerformanceTRD.C");
   if(!AddTrainPerformanceTRD(optList)) {
@@ -162,12 +162,9 @@ void run(Char_t *optList="ALL", const Char_t *files=0x0, Long64_t nev=1234567890
   }
   //mgr->PrintStatus();
   mgr->StartAnalysis("local", chain, nev, first);
-
   timer.Stop();
   timer.Print();  
 
-  delete cdbManager;
-
   // verbosity
   printf("\tCLEANING TASK LIST:\n");
   mgr->GetTasks()->Delete();
@@ -211,24 +208,24 @@ TChain* MakeChainXML(const char* xmlfile)
 {
   if (!TFile::Open(xmlfile)) {
     Error("MakeChainXML", Form("No file %s was found", xmlfile));
-    return 0x0;
+    return NULL;
   }
 
-  if(gSystem->Load("libNetx.so")<0) return 0x0;
-  if(gSystem->Load("libRAliEn.so")<0) return 0x0;
+  if(gSystem->Load("libNetx.so")<0) return NULL;
+  if(gSystem->Load("libRAliEn.so")<0) return NULL;
   TGrid::Connect("alien://") ;
 
   TGridCollection *collection = (TGridCollection*) TAlienCollection::Open(xmlfile);
   if (!collection) {
     Error("MakeChainXML", Form("No collection found in %s", xmlfile)) ; 
-    return 0x0; 
+    return NULL;
   }
   //collection->CheckIfOnline();
 
   TGridResult* result = collection->GetGridResult("",0 ,0);
   if(!result->GetEntries()){
     Error("MakeChainXML", Form("No entries found in %s", xmlfile)) ; 
-    return 0x0; 
+    return NULL;
   }
   // Makes the ESD chain 
   TChain* chain = new TChain("esdTree");
@@ -237,3 +234,13 @@ TChain* MakeChainXML(const char* xmlfile)
   }
   return chain;
 }
+
+//______________________________________________________
+Bool_t UseMC(Char_t *opt){
+  return !(Bool_t)strstr(opt, "NOMC");
+}
+
+//____________________________________________
+Bool_t UseFriends(Char_t *opt){
+  return !(Bool_t)strstr(opt, "NOFR");
+}