Attempt to monitor what file is read and merged by what job

[u/mrichter/AliRoot.git] / PWGCF / EBYE / macros / runBalanceFunctionInpp.C

enum analysisModes {mLocal,mLocalPAR,mPROOF,mGrid,mGridPAR};\r
enum analysisTypes {mESD,mAOD,mMC,mMCESD};\r
\r
//\r
class AliAnalysisGrid;\r
class AliAnalysisTaskBF;\r
class AliBalance;\r
\r
//________________________________________________________________________//\r
void runBalanceFunctionInpp(Int_t mode = mLocal, \r
			    Int_t type = mAOD,\r
			    Bool_t DATA = kFALSE) {\r
  // Time:\r
  TStopwatch timer;\r
  timer.Start();\r
  \r
  //Check analysis mode\r
  if((mode < 0) || (mode > 4)) {\r
    Printf("Analysis mode not recognized!");\r
    Printf("You can select out of 0: local, 1: local with par files, 2: proof, 3: grid, 4: grid with par files");\r
    return;\r
  }\r
  \r
  //Check analysis type\r
  if((type < 0) || (type > 3)) {\r
    Printf("Analysis type not recognized!");\r
    Printf("You can select out of 0: ESD, 1: AOD, 2: MC (stack), 3: MC (from the ESD)");\r
    return;\r
  }\r
  \r
  // Load needed libraries:\r
  LoadLibraries(mode);\r
  \r
 // Create and configure the AliEn plug-in:\r
  if(mode == mGrid || mode == mGridPAR) {\r
    gROOT->LoadMacro("CreateAlienHandler.C");\r
    AliAnalysisGrid *alienHandler = CreateAlienHandler(runListFileName);  \r
    if (!alienHandler) return;\r
  }\r
  // Chains:   \r
  if(mode == mLocal || mode == mLocalPAR) {\r
    TChain* chain = 0x0;\r
    if((type == mESD)||(type == mMCESD))  \r
      chain = new TChain("esdTree");\r
    else if(type == mAOD)\r
      chain = new TChain("aodTree");\r
    else if(type == mMC)\r
      chain = new TChain("TE");\r
\r
    TString filename;\r
    for(Int_t i = 1; i < 20; i++) {\r
      filename = "/data/alice2/pchrist/pp/LHC10c/0.9TeV/Data/";\r
      filename += "Set"; filename += i; \r
      if((type == mESD)||(type == mMCESD))  \r
	filename += "/AliESDs.root";\r
      else if(type == mAOD)\r
	filename += "/AliAOD.root";\r
      else if(type == mMC)\r
     	filename += "/galice.root";\r
\r
      chain->Add(filename.Data());\r
    }\r
  }\r
  //Proof\r
  if(mode == mPROOF) {\r
    gROOT->ProcessLine(Form(".include %s/include", gSystem->ExpandPathName("$ALICE_ROOT")));\r
  }\r
  \r
  // analysis manager\r
  AliAnalysisManager* mgr = new AliAnalysisManager("balanceFunctionManager");\r
  if(mode == mGrid || mode == mGridPAR)\r
    mgr->SetGridHandler(alienHandler);\r
    \r
  // input handler (ESD or AOD)\r
  AliVEventHandler* inputH = NULL;\r
  if((type == mESD)||(type == mMCESD))  \r
    inputH = new AliESDInputHandler();\r
  else if(type == mAOD)\r
    inputH = new AliAODInputHandler();\r
  mgr->SetInputEventHandler(inputH);\r
    \r
  // mc event handler\r
  if((type == mMC) || (type == mMCESD)) {\r
    AliMCEventHandler* mchandler = new AliMCEventHandler();\r
    // Not reading track references\r
    mchandler->SetReadTR(kFALSE);\r
    mgr->SetMCtruthEventHandler(mchandler);\r
  }   \r
\r
  if(type != mAOD){\r
    gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");\r
    AliPhysicsSelectionTask* physSelTask = AddTaskPhysicsSelection();\r
  }\r
\r
  //Add the BF task (all centralities)\r
  gROOT->LoadMacro("AddTaskBalanceFunctionInpp.C"); \r
  AddTaskBalanceFunctionInpp();\r
\r
  // enable debug printouts\r
  mgr->SetDebugLevel(2);\r
  mgr->SetUseProgressBar(1,100);\r
  if (!mgr->InitAnalysis()) return;\r
  mgr->PrintStatus();\r
  \r
  // start analysis\r
  if(mode == mLocal || mode == mLocalPAR) \r
    mgr->StartAnalysis("local",chain);\r
  else if(mode == mPROOF) \r
    mgr->StartAnalysis("proof",dataDir,nRuns,offset);\r
  else if(mode == mGrid || mode == mGridPAR) \r
    mgr->StartAnalysis("grid");\r
\r
  // Print real and CPU time used for analysis:  \r
  timer.Stop();\r
  timer.Print();\r
}\r
\r
//=============================================================//\r
void LoadLibraries(const analysisModes mode) {  \r
  //--------------------------------------\r
  // Load the needed libraries most of them already loaded by aliroot\r
  //--------------------------------------\r
  gSystem->Load("libCore.so");        \r
  gSystem->Load("libGeom.so");\r
  gSystem->Load("libVMC.so");\r
  gSystem->Load("libPhysics.so");\r
  gSystem->Load("libTree.so");\r
\r
  //----------------------------------------------------------\r
  // >>>>>>>>>>> Local mode <<<<<<<<<<<<<< \r
  //----------------------------------------------------------\r
  if (mode==mLocal || mode==mGrid || mode == mGridPAR) {\r
    //--------------------------------------------------------\r
    // If you want to use already compiled libraries \r
    // in the aliroot distribution\r
    //--------------------------------------------------------\r
    gSystem->Load("libSTEERBase.so");\r
    gSystem->Load("libESD.so");\r
    gSystem->Load("libAOD.so");\r
    gSystem->Load("libANALYSIS.so");\r
    gSystem->Load("libANALYSISalice.so");\r
    gSystem->Load("libEventMixing.so");\r
    gSystem->Load("libPWGCFebye.so");\r
    // Use AliRoot includes to compile our task\r
    gROOT->ProcessLine(".include $ALICE_ROOT/include");\r
  }\r
  \r
  else if (mode == mLocalPAR) {\r
    //--------------------------------------------------------\r
    //If you want to use root and par files from aliroot\r
    //--------------------------------------------------------  \r
    SetupPar("STEERBase");\r
    SetupPar("ESD");\r
    SetupPar("AOD");\r
    SetupPar("ANALYSIS");\r
    SetupPar("ANALYSISalice");\r
    SetupPar("PWGCFebye");\r
}\r
  \r
  //---------------------------------------------------------\r
  // <<<<<<<<<< PROOF mode >>>>>>>>>>>>\r
  //---------------------------------------------------------\r
  else if (mode==mPROOF) {\r
    // Connect to proof\r
    printf("*** Connect to PROOF ***\n");\r
    gEnv->SetValue("XSec.GSI.DelegProxy","2");\r
    // Put appropriate username here\r
    TProof::Open("alice-caf.cern.ch");\r
    //TProof::Open("skaf.saske.sk");\r
    //TProof::Open("prf000-iep-grid.saske.sk");\r
\r
    gProof->EnablePackage("VO_ALICE@AliRoot::v4-21-12-AN");\r
  }  \r
  \r
} // end of void LoadLibraries(const anaModes mode)\r
\r
//======================================================================//\r
void SetupPar(char* pararchivename) {\r
  //Load par files, create analysis libraries\r
  //For testing, if par file already decompressed and modified\r
  //classes then do not decompress.\r
  \r
  TString cdir(Form("%s", gSystem->WorkingDirectory() )) ; \r
  TString parpar(Form("%s.par", pararchivename)) ; \r
  if ( gSystem->AccessPathName(parpar.Data()) ) {\r
    gSystem->ChangeDirectory(gSystem->Getenv("ALICE_ROOT")) ;\r
    TString processline(Form(".! make %s", parpar.Data())) ; \r
    gROOT->ProcessLine(processline.Data()) ;\r
    gSystem->ChangeDirectory(cdir) ; \r
    processline = Form(".! mv /tmp/%s .", parpar.Data()) ;\r
    gROOT->ProcessLine(processline.Data()) ;\r
  } \r
  if ( gSystem->AccessPathName(pararchivename) ) {  \r
    TString processline = Form(".! tar xvzf %s",parpar.Data()) ;\r
    gROOT->ProcessLine(processline.Data());\r
  }\r
  \r
  TString ocwd = gSystem->WorkingDirectory();\r
  gSystem->ChangeDirectory(pararchivename);\r
  \r
  // check for BUILD.sh and execute\r
  if (!gSystem->AccessPathName("PROOF-INF/BUILD.sh")) {\r
    printf("*******************************\n");\r
    printf("*** Building PAR archive    ***\n");\r
    cout<<pararchivename<<endl;\r
    printf("*******************************\n");\r
    if (gSystem->Exec("PROOF-INF/BUILD.sh")) {\r
      Error("runProcess","Cannot Build the PAR Archive! - Abort!");\r
      return -1;\r
    }\r
  }\r
  // check for SETUP.C and execute\r
  if (!gSystem->AccessPathName("PROOF-INF/SETUP.C")) {\r
    printf("*******************************\n");\r
    printf("*** Setup PAR archive       ***\n");\r
    cout<<pararchivename<<endl;\r
    printf("*******************************\n");\r
    gROOT->Macro("PROOF-INF/SETUP.C");\r
  }\r
  \r
  gSystem->ChangeDirectory(ocwd.Data());\r
  printf("Current dir: %s\n", ocwd.Data());\r
\r
} // end of void SetupPar(char* pararchivename) \r