1 //--------------------------------------------------------------------------
2 // Base macro for submitting muon QA analysis.
4 // In case it is not run with full aliroot, it needs the following libraries:
9 // - libANALYSISalice.so
13 // The macro reads ESDs and store outputs in standard output file (AnalysisResults.root)
15 // Author: Philippe Pillot - SUBATECH Nantes
16 //--------------------------------------------------------------------------
18 enum {kLocal, kInteractif_xml, kInteractif_ESDList};
20 void RunMuonQA(TString inputFileName = "AliESDs.root", Bool_t selectPhysics = kTRUE,
21 Bool_t selectTrigger = kTRUE, Bool_t selectMatched = kFALSE, Short_t selectCharge = 0)
27 Int_t mode = GetMode(inputFileName);
29 Error("RunMuonQA","Please provide either an ESD root file or a collection of ESDs.");
33 // Load common libraries
34 gSystem->Load("libTree");
35 gSystem->Load("libGeom");
36 gSystem->Load("libVMC");
37 gSystem->Load("libPhysics");
38 gSystem->Load("libSTEERBase");
39 gSystem->Load("libESD");
40 gSystem->Load("libAOD");
41 gSystem->Load("libANALYSIS");
42 gSystem->Load("libANALYSISalice");
43 gSystem->Load("libCORRFW");
44 gSystem->Load("libPWG3muon");
47 TChain* chain = CreateChain(inputFileName);
50 // Create the analysis manager
51 AliAnalysisManager *mgr = new AliAnalysisManager("MuonQAAnalysis");
54 AliESDInputHandler* esdH = new AliESDInputHandler();
55 esdH->SetReadFriends(kFALSE);
56 mgr->SetInputEventHandler(esdH);
59 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
60 AliPhysicsSelectionTask* physicsSelection = AddTaskPhysicsSelection();
61 if(!physicsSelection) {
62 Error("RunMuonQA","AliPhysicsSelectionTask not created!");
67 gROOT->LoadMacro("$ALICE_ROOT/PWG3/muon/AddTaskMuonQA.C");
68 AliAnalysisTaskMuonQA* muonQA = AddTaskMuonQA(selectPhysics, selectTrigger, selectMatched, selectCharge);
70 Error("RunMuonQA","AliAnalysisTaskMuonQA not created!");
74 // Enable debug printouts
75 //mgr->SetDebugLevel(2);
77 // start local analysis
78 if (mgr->InitAnalysis()) {
80 mgr->StartAnalysis("local", chain);
87 //______________________________________________________________________________
88 Int_t GetMode(TString inputFileName)
90 if ( inputFileName.EndsWith(".xml") ) return kInteractif_xml;
91 else if ( inputFileName.EndsWith(".txt") ) return kInteractif_ESDList;
92 else if ( inputFileName.EndsWith(".root") ) return kLocal;
96 //______________________________________________________________________________
97 TChain* CreateChainFromCollection(const char *xmlfile)
99 // Create a chain from the collection of tags.
100 TAlienCollection* coll = TAlienCollection::Open(xmlfile);
102 ::Error("CreateChainFromTags", "Cannot create an AliEn collection from %s", xmlfile);
106 TGridResult* tagResult = coll->GetGridResult("",kFALSE,kFALSE);
107 AliTagAnalysis *tagAna = new AliTagAnalysis("ESD");
108 tagAna->ChainGridTags(tagResult);
110 AliRunTagCuts *runCuts = new AliRunTagCuts();
111 AliLHCTagCuts *lhcCuts = new AliLHCTagCuts();
112 AliDetectorTagCuts *detCuts = new AliDetectorTagCuts();
113 AliEventTagCuts *evCuts = new AliEventTagCuts();
115 // Check if the cuts configuration file was provided
116 if (!gSystem->AccessPathName("ConfigureCuts.C")) {
117 gROOT->LoadMacro("ConfigureCuts.C");
118 ConfigureCuts(runCuts, lhcCuts, detCuts, evCuts);
121 TChain *chain = tagAna->QueryTags(runCuts, lhcCuts, detCuts, evCuts);
122 if (!chain || !chain->GetNtrees()) return NULL;
127 //______________________________________________________________________________
128 TChain* CreateChainFromFile(const char *rootfile)
130 // Create a chain using the root file.
131 TChain* chain = new TChain("esdTree");
132 chain->Add(rootfile);
133 if (!chain->GetNtrees()) return NULL;
138 //______________________________________________________________________________
139 TChain* CreateChainFromESDList(const char *esdList)
141 // Create a chain using tags from the run list.
142 TChain* chain = new TChain("esdTree");
143 ifstream inFile(esdList);
145 if (inFile.is_open()) {
146 while (! inFile.eof() ) {
147 inFileName.ReadLine(inFile,kFALSE);
148 if(!inFileName.EndsWith(".root")) continue;
149 chain->Add(inFileName.Data());
153 if (!chain->GetNtrees()) return NULL;
158 //______________________________________________________________________________
159 TChain* CreateChain(TString inputFileName)
161 printf("*******************************\n");
162 printf("*** Getting the Chain ***\n");
163 printf("*******************************\n");
164 Int_t mode = GetMode(inputFileName);
165 if(mode == kInteractif_xml) return CreateChainFromCollection(inputFileName.Data());
166 else if (mode == kInteractif_ESDList) return CreateChainFromESDList(inputFileName.Data());
167 else if (mode == kLocal) return CreateChainFromFile(inputFileName.Data());