// Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
//
// Usage:
-// run.C(tasks, files, entries)
+// run.C(tasks, files)
// tasks : "ALL" 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
// "RES" : TRD tracking Resolution
+// "CLRES": clusters Resolution
// "CAL" : TRD calibration
+// "ALGN" : TRD alignment
// "PID" : TRD PID - pion efficiency
// "PIDR" : TRD PID - reference data
+// "DET" : Basic TRD Detector checks
+// "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
//
// gSystem->Load("libMemStat.so")
// gSystem->Load("libMemStatGui.so")
// gSystem->Load("libANALYSIS.so")
+// gSystem->Load("libANALYSISalice.so")
// gSystem->Load("libTRDqaRec.so")
+// gSystem->Load("libPWG1.so");
+// gSystem->Load("libNetx.so") ;
+// gSystem->Load("libRAliEn.so");
//
// Authors:
// Alex Bercuci (A.Bercuci@gsi.de)
#include "TMemStatViewerGUI.h"
#include "TROOT.h"
+#include "TClass.h"
#include "TSystem.h"
#include "TError.h"
#include "TChain.h"
+#include "TGrid.h"
+#include "TAlienCollection.h"
+#include "TGridCollection.h"
+#include "TGridResult.h"
+#include "TGeoGlobalMagField.h"
-#include "AliMagFMaps.h"
+#include "AliMagF.h"
#include "AliTracker.h"
#include "AliLog.h"
#include "AliCDBManager.h"
+#include "AliGeomManager.h"
#include "AliAnalysisManager.h"
#include "AliAnalysisDataContainer.h"
#include "AliMCEventHandler.h"
#include "TRD/AliTRDtrackerV1.h"
#include "TRD/AliTRDcalibDB.h"
-#include "TRD/qaRec/AliTRDtrackInfoGen.h"
-#include "TRD/qaRec/AliTRDtrackingEfficiency.h"
-#include "TRD/qaRec/AliTRDtrackingEfficiencyCombined.h"
-#include "TRD/qaRec/AliTRDtrackingResolution.h"
-#include "TRD/qaRec/AliTRDcalibration.h"
+
+#include "TRD/qaRec/macros/AddTRDcheckESD.C"
+#include "TRD/qaRec/macros/AddTRDinfoGen.C"
+#include "TRD/qaRec/macros/AddTRDcheckDET.C"
+#include "TRD/qaRec/macros/AddTRDefficiency.C"
+#include "TRD/qaRec/macros/AddTRDresolution.C"
+#include "TRD/qaRec/macros/AddTRDcheckPID.C"
+
+#include "../../PWG1/macros/AddPerformanceTask.C"
#endif
-#define BIT(n) (1 << (n))
-#define SETBIT(n,i) ((n) |= BIT(i))
-#define TESTBIT(n,i) ((Bool_t)(((n) & BIT(i)) != 0))
-#define CLEARBIT(n,i) ((n) &= ~BIT(i))
+#include "macros/AliTRDperformanceTrain.h"
+#include "../../PWG1/macros/AddPerformanceTask.h"
+
Bool_t MEM = kFALSE;
-const Int_t fknTasks = 5;
-Char_t *fTaskName[fknTasks] = {"Barrel Tracking Effiency", "Combined Tracking Efficiency", "Tracking Resolution", "Calibration", "PID Checker"};
-enum AliTRDrecoTasks{
- kTrackingEfficiency = 0
- ,kTrackingCombinedEfficiency = 1
- ,kTrackingResolution = 2
- ,kCalibration = 3
- ,kPIDChecker = 4
-};
-
-TChain* CreateESDChain(const char* filename = 0x0, Int_t nfiles=-1 );
-void run(const Char_t *files=0x0, Char_t *tasks="ALL", Int_t nmax=-1)
+Bool_t fHasMCdata = kTRUE;
+Bool_t fHasFriends = kTRUE;
+
+TChain* MakeChainLST(const char* filename = 0x0);
+TChain* MakeChainXML(const char* filename = 0x0);
+void run(Char_t *trd="ALL", Char_t *tpc="ALL", const Char_t *files=0x0, Long64_t nev=1234567890, Long64_t first = 0)
{
TMemStat *mem = 0x0;
if(MEM){
mem = new TMemStat("new, gnubuildin");
mem->AddStamp("Start");
}
-
TStopwatch timer;
timer.Start();
- gSystem->Load("libANALYSIS.so");
- gSystem->Load("libTRDqaRec.so");
-
- Int_t fSteerTask = 0;
- Bool_t fHasMCdata = kTRUE;
- TObjArray *tasksArray = TString(tasks).Tokenize(" ");
- for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){
- TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String();
- if(s.CompareTo("ALL") == 0){
- for(Int_t itask = 0; itask < fknTasks; itask++) SETBIT(fSteerTask, itask);
- continue;
- } else if(s.CompareTo("EFF") == 0){
- SETBIT(fSteerTask, kTrackingEfficiency);
- continue;
- } else if(s.CompareTo("EFFC") == 0){
- SETBIT(fSteerTask, kTrackingCombinedEfficiency);
- continue;
- } else if(s.CompareTo("RES") == 0){
- SETBIT(fSteerTask, kTrackingResolution);
- continue;
- } else if(s.CompareTo("CAL" ) == 0){
- SETBIT(fSteerTask, kCalibration);
- continue;
- } else if(s.CompareTo("PID" ) == 0){
- SETBIT(fSteerTask, kCalibration);
- continue;
- } else if(s.CompareTo("NOMC") == 0){
- CLEARBIT(fSteerTask, kTrackingEfficiency);
- CLEARBIT(fSteerTask, kTrackingCombinedEfficiency);
+
+
+ // VERY GENERAL SETTINGS
+ AliLog::SetGlobalLogLevel(AliLog::kError);
+ if(gSystem->Load("libANALYSIS.so")<0) return;
+ if(gSystem->Load("libANALYSISalice.so")<0) return;
+
+/* } else if(s.CompareTo("NOFR") == 0){
+ fHasFriends = kFALSE;
+ } else if(s.CompareTo("NOMC") == 0){
fHasMCdata = kFALSE;
- } else{
- Info("run.C", Form("Task %s not implemented (yet).", s.Data()));
- continue;
- }
- }
- printf("\n\tRUNNING TRAIN FOR TASKS:\n");
- for(Int_t itask = 0; itask < fknTasks; itask++){
- if(TESTBIT(fSteerTask, itask)) printf("\t%s\n", fTaskName[itask]);
+*/
+
+ // INITIALIZATION OF RUNNING ENVIRONMENT
+ //TODO We should use the GRP if available similar to AliReconstruction::InitGRP()!
+ // initialize OCDB manager
+ AliCDBManager *cdbManager = AliCDBManager::Instance();
+ cdbManager->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
+ cdbManager->SetRun(0);
+ cdbManager->SetCacheFlag(kFALSE);
+ // initialize magnetic field.
+ if(!TGeoGlobalMagField::Instance()->GetField()){
+ TGeoGlobalMagField::Instance()->SetField(
+ new AliMagF("Maps","Maps", 2, 1., 10., AliMagF::k5kG)
+ //AliMagF("Maps","Maps", 2, 0., 10., AliMagF::k2kG);
+ );
+ AliGeomManager::LoadGeometry();
}
- // define task list pointers;
- AliAnalysisTask *taskPtr[fknTasks], *task = 0x0;
- memset(taskPtr, 0, fknTasks*sizeof(AliAnalysisTask*));
- Int_t jtask = 0;
- //____________________________________________//
- //gROOT->LoadMacro(Form("%s/TRD/qaRec/CreateESDChain.C", gSystem->ExpandPathName("$ALICE_ROOT")));
- TChain *chain = CreateESDChain(files, nmax);
- //chain->SetBranchStatus("*", 0);
+ // DEFINE DATA CHAIN
+ TChain *chain = 0x0;
+ if(!files) chain = MakeChainLST();
+ else{
+ TString fn(files);
+ if(fn.EndsWith("xml")) chain = MakeChainXML(files);
+ else chain = MakeChainLST(files);
+ }
+ if(!chain) return;
chain->SetBranchStatus("*FMD*",0);
chain->SetBranchStatus("*Calo*",0);
chain->SetBranchStatus("Tracks", 1);
chain->Lookup();
chain->GetListOfFiles()->Print();
printf("\n ----> CHAIN HAS %d ENTRIES <----\n\n", (Int_t)chain->GetEntries());
-
- AliLog::SetGlobalLogLevel(AliLog::kError);
- //____________________________________________
- // Make the analysis manager
- AliAnalysisManager *mgr = new AliAnalysisManager("TRD Track Info Manager");
- //mgr->SetSpecialOutputLocation(source); // To Be Changed
+
+ // BUILD ANALYSIS MANAGER
+ AliAnalysisManager *mgr = new AliAnalysisManager("Post Reconstruction Calibration/QA");
AliVEventHandler *esdH = 0x0, *mcH = 0x0;
mgr->SetInputEventHandler(esdH = new AliESDInputHandler);
if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
//mgr->SetDebugLevel(10);
- //____________________________________________
- // TRD track summary generator
- mgr->AddTask(task = new AliTRDtrackInfoGen());
- taskPtr[jtask++] = task;
- ((AliTRDtrackInfoGen*)task)->SetDebugLevel(1);
- ((AliTRDtrackInfoGen*)task)->SetMCdata(fHasMCdata);
- // Create containers for input/output
- AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("data", TChain::Class(), AliAnalysisManager::kInputContainer);
- AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("TrackInfoList", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
- mgr->ConnectInput( task, 0, cinput1);
- mgr->ConnectOutput(task, 0, coutput1);
-
- //____________________________________________
- // TRD barrel tracking efficiency
- if(TESTBIT(fSteerTask, kTrackingEfficiency)){
- mgr->AddTask(task = new AliTRDtrackingEfficiency());
- ((AliTRDtrackingEfficiency*)task)->SetDebugLevel(1);
- //Create containers for input/output
- AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("TrackingEff", TList::Class(), AliAnalysisManager::kOutputContainer, "TRD.TaskTrackingEff.root");
- mgr->ConnectInput( task, 0, coutput1);
- mgr->ConnectOutput(task, 0, coutput2);
- taskPtr[jtask++] = task;
- }
-
- //____________________________________________
- // TRD combined tracking efficiency
- if(TESTBIT(fSteerTask, kTrackingCombinedEfficiency)){
- mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined());
- ((AliTRDtrackingEfficiencyCombined*)task)->SetDebugLevel(0);
-
- // Create containers for input/output
- AliAnalysisDataContainer *coutput3 = mgr->CreateContainer("TrackingEffMC", TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.TaskTrackingEffMC.root");
- mgr->ConnectInput( task, 0, coutput1);
- mgr->ConnectOutput(task, 0, coutput3);
- taskPtr[jtask++] = task;
- }
- //____________________________________________
- // TRD tracking resolution
- if(TESTBIT(fSteerTask, kTrackingResolution)){
- mgr->AddTask(task = new AliTRDtrackingResolution());
- ((AliTRDtrackingResolution*)task)->SetMCdata(fHasMCdata);
- ((AliTRDtrackingResolution*)task)->SetDebugLevel(1);
- // Create containers for input/output
- AliAnalysisDataContainer *coutput4 = mgr->CreateContainer("Resolution", TList::Class(), AliAnalysisManager::kOutputContainer, "TRD.TaskResolution.root");
- mgr->ConnectInput( task, 0, coutput1);
- mgr->ConnectOutput(task, 0, coutput4);
- taskPtr[jtask++] = task;
- }
- //____________________________________________
- // TRD calibration
- if(TESTBIT(fSteerTask, kCalibration)){
- mgr->AddTask(task = new AliTRDcalibration());
- ((AliTRDcalibration*)task)->SetLow(0);
- ((AliTRDcalibration*)task)->SetHigh(30);
- ((AliTRDcalibration*)task)->SetDebugLevel(0);
- ((AliTRDcalibration*)task)->SetFillZero(kFALSE);
- // Create containers for input/output
- AliAnalysisDataContainer *coutput5 = mgr->CreateContainer("Calibration", TList::Class(), AliAnalysisManager::kOutputContainer, "TRD.TaskCalibration.root");
- mgr->ConnectInput(task, 0, coutput1);
- mgr->ConnectOutput(task, 0, coutput5);
- taskPtr[jtask++] = task;
- }
-
- //____________________________________________
- // TRD pid checker
- if(TESTBIT(fSteerTask, kPIDChecker)){
- mgr->AddTask(task = new AliTRDpidChecker());
- taskPtr[jtask++] = task;
- //task->SetDebugLevel(1);
- // Create containers for input/output
- AliAnalysisDataContainer *coutput6 = mgr->CreateContainer("PID", TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.TaskPID.root");
- mgr->ConnectInput( task, 0, coutput1);
- mgr->ConnectOutput(task, 0, coutput6);
- }
+///////////////////////////////////////////////////////////
+/////////////// TRD ///////////
+///////////////////////////////////////////////////////////
+ // TRD specific library
+ if(gSystem->Load("libTRDqaRec.so")<0) return;
+ // TRD data containers
+ AliAnalysisDataContainer *ci[] = {0x0, 0x0};
- if (!mgr->InitAnalysis()) return;
- mgr->PrintStatus();
-
- AliCDBManager *cdbManager = AliCDBManager::Instance();
- cdbManager->SetDefaultStorage("local://$ALICE_ROOT");
- //cdbManager->SetSpecificStorage("TRD/Calib/FEE","local:///u/bailhach/aliroot/database30head/database");
- cdbManager->SetRun(0);
- cdbManager->SetCacheFlag(kFALSE);
-
// initialize TRD settings
- AliMagFMaps *field = new AliMagFMaps("Maps","Maps", 2, 1., 10., AliMagFMaps::k5kG);
- AliTracker::SetFieldMap(field, kTRUE);
AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
AliTRDtrackerV1::SetNTimeBins(cal->GetNumberOfTimeBins());
+ // plug (set of) TRD wagons in the train
+ if(trd){
+ for(Int_t it=0; it<NTRDQATASKS; it++){
+ if(gROOT->LoadMacro(Form("$ALICE_ROOT/TRD/qaRec/macros/Add%s.C+", TString(fgkTRDtaskClassName[it])(3,20).Data()))) {
+ Error("run.C", Form("Error loading %s task.", fgkTRDtaskClassName[it]));
+ return;
+ }
+
+ switch(it){
+ case kCheckESD:
+ AddTRDcheckESD(mgr); break;
+ case kInfoGen:
+ AddTRDinfoGen(mgr, trd, 0x0, ci); break;
+ case kCheckDET:
+ AddTRDcheckDET(mgr, trd, ci); break;
+ case kEfficiency:
+ AddTRDefficiency(mgr, trd, ci); break;
+ case kResolution:
+ AddTRDresolution(mgr, trd, ci); break;
+ case kCheckPID:
+ AddTRDcheckPID(mgr, trd, ci); break;
+ default:
+ Warning("run.C", Form("No performance task registered at slot %d.", it));
+ }
+ }
+ }
+
+///////////////////////////////////////////////////////////
+/////////////// TPC ///////////
+///////////////////////////////////////////////////////////
+ if(gSystem->Load("libPWG1.so")<0) return;
+
+ // BUILD STEERING TASK FOR TPC
+ if(tpc){
+ if(gROOT->LoadMacro("$ALICE_ROOT/PWG1/macros/AddPerformanceTask.C+")) {
+ Error("run.C", "Error loading AliPerformanceTask task.");
+ return;
+ }
+ AddPerformanceTask(mgr, tpc);
+ }
- mgr->StartAnalysis("local",chain);
+ if (!mgr->InitAnalysis()) return;
+ // verbosity
+ printf("\n\tRUNNING TRAIN FOR TASKS:\n");
+ mgr->GetTasks()->ls();
+ //mgr->PrintStatus();
+
+ mgr->StartAnalysis("local", chain, nev, first);
timer.Stop();
timer.Print();
cal->Terminate();
- delete field;
+ TGeoGlobalMagField::Instance()->SetField(NULL);
delete cdbManager;
- for(Int_t it=jtask-1; it>=0; it--) delete taskPtr[it];
+
+ // verbosity
+ printf("\n\tCLEANING UP TRAIN:\n");
+ mgr->GetTasks()->Delete();
+
if(mcH) delete mcH;
delete esdH;
delete mgr;
if(MEM) TMemStatViewerGUI::ShowGUI();
}
-
-TChain* CreateESDChain(const char* filename, Int_t nfiles)
+//____________________________________________
+TChain* MakeChainLST(const char* filename)
{
// Create the chain
TChain* chain = new TChain("esdTree");
ifstream in;
in.open(filename);
TString esdfile;
- while(in.good() && (nfiles--) ) {
+ while(in.good()) {
in >> esdfile;
if (!esdfile.Contains("root")) continue; // protection
chain->Add(esdfile.Data());
return chain;
}
+
+//____________________________________________
+TChain* MakeChainXML(const char* xmlfile)
+{
+ if (!TFile::Open(xmlfile)) {
+ Error("MakeChainXML", Form("No file %s was found", xmlfile));
+ return 0x0;
+ }
+
+ if(gSystem->Load("libNetx.so")<0) return 0x0;
+ if(gSystem->Load("libRAliEn.so")<0) return 0x0;
+ TGrid::Connect("alien://") ;
+
+ TGridCollection *collection = (TGridCollection*) TAlienCollection::Open(xmlfile);
+ if (!collection) {
+ Error("MakeChainXML", Form("No collection found in %s", xmlfile)) ;
+ return 0x0;
+ }
+ //collection->CheckIfOnline();
+
+ TGridResult* result = collection->GetGridResult("",0 ,0);
+ if(!result->GetEntries()){
+ Error("MakeChainXML", Form("No entries found in %s", xmlfile)) ;
+ return 0x0;
+ }
+ // Makes the ESD chain
+ TChain* chain = new TChain("esdTree");
+ for (Int_t idx = 0; idx < result->GetEntries(); idx++) {
+ chain->Add(result->GetKey(idx, "turl"));
+ }
+ return chain;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff