PWG4/UserTasks/PHOS_PbPbQA/PHOSPbPb.C

   1 void PHOSPbPbQA(const char* dataset="collection.xml")
   2 {
   3   gSystem->Load("libTree.so");
   4   gSystem->Load("libGeom.so");
   5   gSystem->Load("libVMC.so");
   6   gSystem->Load("libPhysics.so");
   7
   8   //load analysis framework
   9   gSystem->Load("libANALYSIS");
  10   gSystem->Load("libANALYSISalice"); //AliAnalysisTaskSE
  11
  12   gSystem->Load("libPWG4UserTasks.so");
  13
  14   gSystem->AddIncludePath("-I$ALICE_ROOT/include -I$ALICE_ROOT/PHOS");
  15
  16   cout << "PHOSPbPbQA: processing collection " << dataset << endl;
  17
  18   TString data = dataset;
  19   TChain* chain = 0;
  20
  21   if(data.Contains(".xml")) {
  22
  23     TGrid::Connect("alien://");
  24
  25     chain = new TChain("esdTree");
  26     TGridCollection * collection = dynamic_cast<TGridCollection*>(TAlienCollection::Open(dataset));
  27
  28     TAlienResult* result = collection->GetGridResult("",0 ,0);
  29     TList* rawFileList = result->GetFileInfoList();
  30
  31     for (Int_t counter=0 ; counter < rawFileList->GetEntries() ; counter++) {
  32       TFileInfo * fi =  static_cast<TFileInfo*>(rawFileList->At(counter)) ;
  33       const char * rawFile = fi->GetCurrentUrl()->GetUrl() ;
  34       printf("Processing %s\n", rawFile) ;
  35       chain->Add(rawFile);
  36       printf("Chain: %d entries.\n",chain->GetEntries());
  37     }
  38   }
  39
  40   if(data.Contains(".txt")) {
  41     gROOT->LoadMacro("$ALICE_ROOT/PWG0/CreateESDChain.C");
  42     chain = CreateESDChain(dataset, 300);
  43   }
  44
  45   if(data.Contains(".root")) {
  46     chain = new TChain("esdTree");
  47     chain->Add(dataset);
  48   }
  49
  50   AliLog::SetGlobalLogLevel(AliLog::kError);
  51
  52   Int_t nentr = chain->GetEntries();
  53   printf("Number of events in the collection is %d\n",nentr);
  54
  55   // Make the analysis manager
  56   AliAnalysisManager *mgr  = new AliAnalysisManager("Manager", "Manager");
  57
  58     // Input handler
  59   AliESDInputHandler *esdHandler = new AliESDInputHandler();
  60   mgr->SetInputEventHandler(esdHandler);
  61
  62   // Output handler
  63   AliESDHandler* esdoutHandler   = new AliESDHandler();
  64
  65   // Debug level
  66   mgr->SetDebugLevel(0);
  67
  68   //Add centrality task!
  69   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");
  70   AliCentralitySelectionTask *taskCentrality = AddTaskCentrality() ;
  71   // taskCentrality->SetMCInput();
  72
  73   // Add my task
  74   AliAnalysisTaskPHOSPbPbQA *task1 = new AliAnalysisTaskPHOSPbPbQA("PbPbQA");
  75   mgr->AddTask(task1);
  76
  77   // Create containers for input/output
  78   AliAnalysisDataContainer *cinput   = mgr->GetCommonInputContainer();
  79   AliAnalysisDataContainer *coutput = mgr->CreateContainer("histESD",TList::Class(),
  80                                                             AliAnalysisManager::kOutputContainer,
  81                                                             "PHOSPbPbQA.root");
  82
  83   // Connect input/output for task1
  84   mgr->ConnectInput(task1 , 0, cinput);
  85   mgr->ConnectOutput(task1, 1, coutput);
  86
  87   if (mgr->InitAnalysis()) {
  88     mgr->PrintStatus();
  89     mgr->StartAnalysis("local", chain);
  90   }
  91
  92   cout <<" Analysis ended sucessfully "<< endl ;
  93
  94 }