]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - HMPID/AliHMPIDAnalysisTask.C
change eve redraw while processing event
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDAnalysisTask.C
index f6bbc476f097e11df1401a468b5e05f290e18759..b2006085ec0831d57184074786cebfc807688bfc 100644 (file)
-Bool_t AliHMPIDAnalysisTask()
+void AliHMPIDAnalysisTask()
 {
-  
-  TBenchmark benchmark;
-  benchmark.Start("AliHMPIDAnalysisTask");
 
-  AliLog::SetGlobalDebugLevel(0);
-
-  Load() ; //load the required libraries
-
-  TChain * analysisChain ;
-   analysisChain = new TChain("esdTree");
-    //here put your input data path 
-   analysisChain->Add("AliESDs.root");
-
-
-  Info("AliHMPIDAnalysisTask",Form("N events %d",(Int_t)analysisChain->GetEntries()));
-
-  // create the task
-  AliHMPIDAnalysisTask *task = new AliHMPIDAnalysisTask("CosmicAnalysisTask");
-  task->SetTrigger(2);
-
-  // Make the analysis manager
-  AliAnalysisManager *mgr = new AliAnalysisManager("TestManager");
-
-  AliESDInputHandler* esdHandler = new AliESDInputHandler();
-  mgr->SetInputEventHandler(esdHandler);
-
-  // Create and connect containers for input/output
+  //
+  // Macro to run the AliHMPIDAnalysisTask
+  // To use: copy this macro to your work directory, modify the global
+  // part to match your needs, then run root.
+  // root[0] .x AliHMPIDAnalysisTask.C
+  //
+
+  //______ Input settings
+  Int_t nEvents = 650000;           //on Proof
+  Int_t nOffset = 0;               //on Proof
+  Int_t nFile2Xml = 10;         //on Grid - how many ESD files to chain in xml
+  TString xmlName="HmpidCheckRun122080_test.xml";//new.xml";       //name of the xml collection created
+  Char_t *indataset = "/alice/sim/LHC10b4_114932";
+//   char *indataset = "$HOME/HMPID/data/AliESDs.root";
+
+  //______ Select Local, Proof or Intercative Grid Analysis (but just only one :-))
+  Bool_t bLOCAL        = kFALSE;
+  Bool_t bPROOF        = kTRUE;
+  Bool_t bGRIDINT      = kFALSE;
+
+//  AliLog::SetGlobalDebugLevel(2);
+//  AliLog::SetGlobalLogLevel(0);
+
+
+  //______ Init data set chain
+  TString dataset(indataset);
+  TChain *chain = new TChain("esdTree");
+
+  //______ Define settings for PROOF
+  const char* proofNode = "alice-caf";
+  gEnv->SetValue("XSec.GSI.DelegProxy","2");
+  gSystem->SetIncludePath("-I. -I$HOME/HMPID -I$ALICE_ROOT/include -I$ROOTSYS/include");
+  gROOT->LoadMacro("$ALICE_ROOT/PWG0/CreateESDChain.C");
+  gSystem->Load("libTree.so");
+  gSystem->Load("libGeom.so");
+  gSystem->Load("libVMC.so");
+  gSystem->Load("libPhysics.so");
+  gSystem->Load("libSTEERBase");
+  gSystem->Load("libESD");
+  gSystem->Load("libAOD");
+  gSystem->Load("libANALYSIS");
+  gSystem->Load("libANALYSISalice");
+
+ //______ Create analysis manager
+   AliAnalysisManager *mgr  = new AliAnalysisManager("HMPID Analysis Train", "HMPID Analysis Train");
+   mgr->SetDebugLevel(0);
+
+   //______ Create input handler, default ESD
+   AliESDInputHandler *esdHandler = new AliESDInputHandler();
+   mgr->SetInputEventHandler(esdHandler);
+
+   //______ Create MC input handler
+   AliMCEventHandler* mcHandler = new AliMCEventHandler();
+   mgr->SetMCtruthEventHandler(mcHandler);
+
+   //______ Create default input container
+   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
+   AliAnalysisDataContainer *hmpoutput= mgr->CreateContainer("hmpoutput", TList::Class(),AliAnalysisManager::kOutputContainer,"HmpidOutput.root");
+
+   if(bLOCAL)
+  {
+
+   Printf("========> Running LOCAL HMPID Analysis <=========");
+   gROOT->LoadMacro("$ALICE_ROOT/PWG0/CreateESDChain.C");
+   gSystem->Load("libSTEERBase");
+   gSystem->Load("libESD");
+   gSystem->Load("libAOD");
+   gSystem->Load("libANALYSIS");
+   gSystem->Load("libANALYSISalice");
+   //chain = CreateESDChain(dataset.Data());//,1,0,kFALSE,kFALSE,0);
+   chain->Add(indataset);
+   gROOT->LoadMacro("AliHMPIDAnalysisTask.cxx++g");//"AliHMPIDAnalysisTask.cxx++");
+   AliHMPIDAnalysisTask *hmpTask = new AliHMPIDAnalysisTask("HMPIDAnalysisTask");
+   mgr->AddTask(hmpTask);
+  }
+  else if(bPROOF)
+  {
+    Printf("========> Running PROOF HMPID Analysis <=========");
+    TProof::Open(proofNode);
+//    gProof->SetParallel(2);
+    gProof->EnablePackage("VO_ALICE@AliRoot::v4-19-14-AN");   
+    gProof->Load(Form("%s/HMPID/AliHMPIDAnalysisTask.cxx++g",gSystem->Getenv("ALICE_ROOT")));
+//    gProof->Load("AliHMPIDAnalysisTask.cxx++g");//"AliHMPIDAnalysisTask.cxx++");
+    AliHMPIDAnalysisTask *hmpTask = new AliHMPIDAnalysisTask("HMPIDAnalysisTask");
+    mgr->AddTask(hmpTask);
+  }
+  else if(bGRIDINT)
+  {
+    Printf("========> Running INTERCATIVE GRID HMPID Analysis <=========");
+    gSystem->Load("libSTEERBase");
+    gSystem->Load("libESD");
+    gSystem->Load("libANALYSIS");
+    gSystem->Load("libANALYSISalice");
+    //______ Add HMPID task
+    gSystem->SetIncludePath("-I. -I$HOME/HMPID -I$ALICE_ROOT/include -I$ROOTSYS/include");
+    gROOT->LoadMacro("$ALICE_ROOT/HMPID/AliHMPIDAnalysisTask.cxx++");
+
+
+    TGrid::Connect("alien://");
+    chain = CreateChainFromCollection(xmlName.Data(),"esdTree",nFile2Xml);
+    //gROOT->LoadMacro(Form("%s/HMPID/AliHMPIDAnalysisTask.cxx++g",gSystem->Getenv("ALICE_ROOT")));//"AliHMPIDAnalysisTask.cxx++");
+    //gROOT->LoadMacro("$ALICE_ROOT/HMPID/AliHMPIDAnalysisTask.cxx++");
+    AliHMPIDAnalysisTask *hmpTask = new AliHMPIDAnalysisTask("HMPIDAnalysisTask");
+    mgr->AddTask(hmpTask);
+     
+  }
 
-  //------ input data ------
-  AliAnalysisDataContainer *cinput0  = mgr->GetCommonInputContainer();
 
-  // ----- output data -----
-  
-  //slot 0 : default output tree (by default handled by AliAnalysisTaskSE)
-//  AliAnalysisDataContainer *coutput0 = mgr->CreateContainer("ctree0", TTree::Class(),AliAnalysisManager::kOutputContainer,"Houtput.root");
+  if(bLOCAL)
+  {
+    //______ Add HMPID task
+    gROOT->LoadMacro("$ALICE_ROOT/PWG0/CreateESDChain.C");
+    gSystem->Load("libSTEERBase");
+    gSystem->Load("libESD");
+    gSystem->Load("libAOD");
+    gSystem->Load("libANALYSIS");
+    gSystem->Load("libANALYSISalice");
+  }
+  if ( bPROOF ) 
+  {
+    
+  }
 
-  //now comes user's output objects :
-  
-  // output TH1I for event counting
-  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("chist0", TH1I::Class(),AliAnalysisManager::kOutputContainer,"Houtput.root");
-  // output list of histos
-  AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("clist0", TList::Class(),AliAnalysisManager::kOutputContainer,"Houtput.root");
-  cinput0->SetData(analysisChain);
+  mgr->ConnectInput(hmpTask,0,cinput);
+  mgr->ConnectOutput(hmpTask,0,hmpoutput);
 
-  mgr->AddTask(task);
-  mgr->ConnectInput(task,0,cinput0);
-//  mgr->ConnectOutput(task,0,coutput0);
-  mgr->ConnectOutput(task,1,coutput1);
-  mgr->ConnectOutput(task,2,coutput2);
 
-  //RUN !!!
   if (mgr->InitAnalysis()) {
     mgr->PrintStatus();
-    mgr->StartAnalysis("local",analysisChain);
+    if(bLOCAL || bGRIDINT) mgr->StartAnalysis("local",chain);
+    else if(bPROOF) mgr->StartAnalysis("proof",dataset.Data(), nEvents,nOffset);
   }
+}   
 
-  benchmark.Stop("AliHMPIDAnalysisTask");
-  benchmark.Show("AliHMPIDAnalysisTask");
+//__________________________________________________________________________________
+TChain *CreateChainFromCollection(const char* xmlfile, const char *treeName="esdTree",Int_t nFiles = 0)
+{
+// Create a chain from an alien collection.
+   TAlienCollection * myCollection  = TAlienCollection::Open(xmlfile);
+
+   if (!myCollection) {
+      ::Error("CreateChainSingle", "Cannot create an AliEn collection from %s", xmlfile);
+     return NULL ;
+   }
+
+  TChain* chain = new TChain(treeName);
+  myCollection->Reset();
+  Int_t iCount = 0;
+  while ( myCollection->Next() ){
+    if(nFiles!=0)iCount++;
+    if(iCount > nFiles)break;
+    chain->Add(myCollection->GetTURL("")) ;
+    Printf("Adding %s",myCollection->GetTURL(""));
+  }
+  chain->ls();
+  return chain;
+}
 
-  return kTRUE ;
+//__________________________________________________________________________________
+void SetupPar(char* pararchivename)
+{
+  //Load par files, create analysis libraries
+  //For testing, if par file already decompressed and modified
+  //classes then do not decompress.
 
+  TString cdir(Form("%s", gSystem->WorkingDirectory() ));
+  TString parpar(Form("%s.par", pararchivename));
 
-}
+  if (!gSystem->AccessPathName(pararchivename) ) {
+    TString processline = Form(".! tar xvzf %s",parpar.Data());
+    gROOT->ProcessLine(processline.Data());
+  }
 
-void Load() {
+  TString ocwd = gSystem->WorkingDirectory();
+  gSystem->ChangeDirectory(pararchivename);
 
-  //load the required aliroot libraries
-  gSystem->Load("libANALYSIS") ;
-  gSystem->Load("libANALYSISalice") ;
+  // check for BUILD.sh and execute
+  if (!gSystem->AccessPathName("PROOF-INF/BUILD.sh")) {
+    printf("*******************************\n");
+    printf("*** Building PAR archive    ***\n");
+    cout<<pararchivename<<endl;
+    printf("*******************************\n");
 
-  //compile online the task class
-  gSystem->SetIncludePath("-I. -I$HOME/HMPID -I$ALICE_ROOT/include -I$ROOTSYS/include");
-  gROOT->LoadMacro("./AliHMPIDAnalysisTask.cxx+");
-}
+    if (gSystem->Exec("PROOF-INF/BUILD.sh")) {
+      Error("runProcess","Cannot Build the PAR Archive! - Abort!");
+      return -1;
+    }
+  }
+  // check for SETUP.C and execute
+  if (!gSystem->AccessPathName("PROOF-INF/SETUP.C")) {
+    printf("*******************************\n");
+    printf("*** Setup PAR archive       ***\n");
+    cout<<pararchivename<<endl;
+    printf("*******************************\n");
+    gROOT->Macro("PROOF-INF/SETUP.C");
+  }
+
+  gSystem->ChangeDirectory(ocwd.Data());
+  printf("Current dir: %s\n", ocwd.Data());
+}
\ No newline at end of file