PWG0/PWG0Helper.C

   1 /* $Id$ */
   2
   3 // Helper macros can be found in this file
   4 // A set of them can be used to connect to proof and execute selectors.
   5
   6 TVirtualProof* connectProof(const char* proofServer)
   7 {
   8   TVirtualProof* proof = TProof::Open(proofServer);
   9
  10   if (!proof)
  11   {
  12     printf("ERROR: PROOF connection not established.\n");
  13     return 0;
  14   }
  15
  16   // enable the new packetizer
  17   //proof->AddInput(new TNamed("PROOF_Packetizer", "TPacketizerProgressive"));
  18
  19   proof->ClearInput();
  20
  21   return proof;
  22 }
  23
  24 Bool_t prepareQuery(TString libraries, TString packages, Int_t useAliRoot)
  25 {
  26   // if not proof load libraries
  27   if (!gProof)
  28   {
  29     TObjArray* librariesList = libraries.Tokenize(";");
  30     for (Int_t i=0; i<librariesList->GetEntries(); ++i)
  31     {
  32       TObjString* str = dynamic_cast<TObjString*> (librariesList->At(i));
  33       if (!str)
  34         continue;
  35
  36       printf("Loading %s...", str->String().Data());
  37       Int_t result = CheckLoadLibrary(str->String());
  38       if (result < 0)
  39       {
  40         printf("failed\n");
  41         //return kFALSE;
  42       }
  43       else
  44         printf("succeeded\n");
  45     }
  46   }
  47   else
  48   {
  49     if (useAliRoot > 0)
  50       ProofEnableAliRoot(useAliRoot);
  51
  52     TObjArray* packagesList = packages.Tokenize(";");
  53     for (Int_t i=0; i<packagesList->GetEntries(); ++i)
  54     {
  55       TObjString* str = dynamic_cast<TObjString*> (packagesList->At(i));
  56       if (!str)
  57         continue;
  58
  59       /*if (!EnablePackageLocal(str->String()))
  60       {
  61         printf("Loading of package %s locally failed\n", str->String().Data());
  62         return kFALSE;
  63       }*/
  64
  65       if (gProof->UploadPackage(Form("%s.par", str->String().Data())))
  66       {
  67         printf("Uploading of package %s failed\n", str->String().Data());
  68         return kFALSE;
  69       }
  70
  71       if (gProof->EnablePackage(str->String()))
  72       {
  73         printf("Loading of package %s failed\n", str->String().Data());
  74         return kFALSE;
  75       }
  76     }
  77  }
  78
  79  return kTRUE;
  80 }
  81
  82 Int_t executeQuery(TChain* chain, TList* inputList, TString selectorName, const char* option = "", Long64_t entries = TChain::kBigNumber)
  83 {
  84   if (!gProof)
  85     chain->GetUserInfo()->AddAll(inputList);
  86   else
  87   {
  88     for (Int_t i=0; i<inputList->GetEntries(); ++i)
  89       gProof->AddInput(inputList->At(i));
  90   }
  91
  92   TStopwatch timer;
  93   timer.Start();
  94
  95   Long64_t result = -1;
  96
  97   if (gProof)
  98     result = chain->MakeTDSet()->Process(selectorName, option, entries);
  99   else
 100     result = chain->Process(selectorName, option, entries);
 101
 102   if (result < 0)
 103     printf("ERROR: Executing process failed with %d.\n", result);
 104
 105   timer.Stop();
 106   timer.Print();
 107
 108   return result;
 109 }
 110
 111 void ProofEnableAliRoot(Int_t aliroot)
 112 {
 113   // enables a locally deployed AliRoot in a PROOF cluster
 114
 115   /* executes the following commands on each node:
 116      gSystem->Setenv("ALICE_ROOT", "/home/alicecaf/ALICE/aliroot-head")
 117      gSystem->AddIncludePath("/home/alicecaf/ALICE/aliroot-head/include");
 118      gSystem->SetDynamicPath(Form("%s:%s", gSystem->GetDynamicPath(), "/home/alicecaf/ALICE/aliroot-head/lib/tgt_linux"))
 119      gSystem->Load("libMinuit");
 120      gROOT->Macro("$ALICE_ROOT/macros/loadlibs.C");
 121   */
 122
 123   const char* location = 0;
 124         const char* target = "tgt_linux";
 125
 126   switch (aliroot)
 127   {
 128     case 1: location = "/home/alicecaf/ALICE/aliroot-v4-04-Release"; break;
 129     case 2: location = "/home/alicecaf/ALICE/aliroot-head"; break;
 130                 case 11: location = "/data1/qfiete/aliroot-head"; target = "tgt_linuxx8664gcc"; break;
 131     default: return;
 132   }
 133
 134   gProof->Exec(Form("gSystem->Setenv(\"ALICE_ROOT\", \"%s\")", location), kTRUE);
 135   gProof->AddIncludePath(Form("%s/include", location));
 136   gProof->AddDynamicPath(Form("%s/lib/%s", location, target));
 137
 138   // load all libraries
 139   gProof->Exec("gSystem->Load(\"libMinuit\")");
 140   gProof->Exec("gROOT->Macro(\"$ALICE_ROOT/macros/loadlibs.C\")");
 141 }
 142
 143 Bool_t EnablePackageLocal(const char* package)
 144 {
 145   printf("Enabling package %s locally...\n", package);
 146
 147   TString currentDir(gSystem->pwd());
 148   if (!gSystem->cd(package))
 149     return kFALSE;
 150
 151   gROOT->ProcessLine(".x PROOF-INF/SETUP.C");
 152   gSystem->cd(currentDir);
 153
 154   return kTRUE;
 155 }
 156
 157 Int_t CheckLoadLibrary(const char* library)
 158 {
 159   // checks if a library is already loaded, if not loads the library
 160
 161   if (strlen(gSystem->GetLibraries(Form("%s.so", library), "", kFALSE)) > 0)
 162     return 1;
 163
 164   return gSystem->Load(library);
 165 }
 166
 167 void redeployPackages(const char* proofServer, Bool_t localAliRoot = kTRUE)
 168 {
 169   // deploys PWG0base and PWG0dep (the latter only when localAliRoot is true) that are expected in $ALICE_ROOT
 170   // when localAliRoot is false ESD.par is also deployed
 171
 172   TProof::Reset(proofServer);
 173   TVirtualProof* proof = TProof::Open(proofServer);
 174   proof->ClearPackages();
 175
 176   if (localAliRoot)
 177     ProofEnableAliRoot();
 178   else
 179   {
 180     proof->UploadPackage("$ALICE_ROOT/ESD.par");
 181     proof->EnablePackage("ESD");
 182   }
 183
 184   proof->UploadPackage("$ALICE_ROOT/PWG0base.par");
 185   proof->EnablePackage("PWG0base");
 186
 187   proof->UploadPackage("$ALICE_ROOT/PWG0dep.par");
 188   proof->EnablePackage("PWG0dep");
 189 }