Revision e8d22123e00d22791239b4381619714ad3d598db authored by Axel Naumann on 02 July 2014, 15:52:29 UTC, committed by Axel Naumann on 02 July 2014, 15:52:29 UTC
1 parent 2f69925
finalizeProof.C
//
// Macro to finalize queries run with the macro tutorials/runProof .
// This macro uses an existing PROOF session or starts one at the indicated URL.
// In the case non existing PROOF session is found and no URL is given, the macro
// tries to start a local PROOF session.
//
// To run the macro:
//
// root[] .L proof/finalizeProof.C+
// root[] finalizeProof("<analysis>")
//
// See runProof.C for the analysis currently available.
//
// The macro looks for the last completed queries for the chosen analysis and
// asks which one to finalize. If there is only available, it finalizes it
// without asking.
// All queries are considered for this, both those run synchronously and those
// run asynchronously, e.g. runProof("h1(asyn)").
//
#include "Getline.h"
#include "TChain.h"
#include "TEnv.h"
#include "TProof.h"
#include "TString.h"
#include "TDrawFeedback.h"
#include "TList.h"
#include "TQueryResult.h"
#include "TObjArray.h"
#include "getProof.C"
void finalizeProof(const char *what = "simple",
const char *url = "proof://localhost:11093",
Int_t nwrks = -1)
{
// Temp dir for PROOF tutorials
TString tutdir = Form("%s/.proof-tutorial", gSystem->TempDirectory());
if (gSystem->AccessPathName(tutdir)) {
Printf("runProof: creating the temporary directory"
" for the tutorial (%s) ... ", tutdir.Data());
if (gSystem->mkdir(tutdir, kTRUE) != 0) {
Printf("runProof: could not assert / create the temporary directory"
" for the tutorial (%s)", tutdir.Data());
return;
}
}
// Get / Attach-to the PROOF Session
TProof *proof = getProof(url, nwrks, tutdir.Data(), "");
if (!proof) {
Printf("runProof: could not start/attach a PROOF session");
return;
}
// Get the last session run for the tutorial
TObjArray *qt = new TObjArray();
TString lasttag;
TString proofsessions(Form("%s/sessions",tutdir.Data()));
// Save tag of the used session
FILE *fs = fopen(proofsessions.Data(), "r");
if (!fs) {
Printf("runProof: could not create files for sessions tags");
} else {
char line[1024];
while (fgets(line, sizeof(line), fs)) {
int l = strlen(line);
if (l <= 0) continue;
if (strncmp(line,"session-",strlen("session-"))) continue;
if (line[l-1] == '\n') line[l-1] = 0;
lasttag = line;
qt->Add(new TObjString(lasttag.Data()));
}
fclose(fs);
}
// Retrieve the list of available query results
TList *ql = proof->GetListOfQueries("A");
if (!ql || ql->GetSize() <= 0) {
Printf("runProof: no queries to be finalized");
return;
}
ql->Print();
// Where is the code to run
char *rootbin = gSystem->Which(gSystem->Getenv("PATH"), "root.exe", kExecutePermission);
if (!rootbin) {
Printf("runProof: root.exe not found: please check the environment!");
return;
}
TString rootsys(gSystem->DirName(rootbin));
rootsys = gSystem->DirName(rootsys);
TString tutorials(Form("%s/tutorials", rootsys.Data()));
delete[] rootbin;
// Create feedback displayer
TDrawFeedback fb(proof);
// Parse 'what'; it is in the form 'analysis(arg1,arg2,...)'
TString args(what);
args.ReplaceAll("("," ");
args.ReplaceAll(")"," ");
args.ReplaceAll(","," ");
Ssiz_t from = 0;
TString act, tok;
if (!args.Tokenize(act, from, " ")) {
// Cannot continue
Printf("runProof: action not found: check your arguments (%s)", what);
return;
}
TObjArray *qa = new TObjArray();
TString sel;
// Action
if (act == "simple") {
sel = "ProofSimple";
} else if (act == "h1") {
sel = "h1analysis";
} else if (act == "pythia8") {
sel = "ProofPythia";
} else {
// Do not know what to run
Printf("runProof: unknown tutorial: %s", what);
}
// Get last completed queries for the chosen analysis
TString ref;
Int_t nt = qt->GetEntriesFast();
while (ref.IsNull() && nt--) {
lasttag = ((TObjString *)(qt->At(nt)))->GetName();
if (!lasttag.IsNull())
Printf("runProof: checking session: %s", lasttag.Data());
TIter nxq(ql);
TQueryResult *qr = 0;
while ((qr = (TQueryResult *)nxq())) {
if (qr->IsDone() && !lasttag.CompareTo(qr->GetTitle()) &&
!sel.CompareTo(qr->GetSelecImp()->GetTitle())) {
TString r = Form("%s:%s",qr->GetTitle(),qr->GetName());
qa->Add(new TObjString(r.Data()));
}
}
if (qa->GetEntriesFast() > 0) {
Int_t qn = 0;
if (qa->GetEntriesFast() > 1) {
// Query the client which query to finalize
Printf("finalizeProof: queries completed for analysis '%s'", act.Data());
for (Int_t k = 0; k < qa->GetEntriesFast(); k++) {
Printf(" [%d] %s", k, ((TObjString *)(qa->At(k)))->GetName());
}
Bool_t ask = kTRUE;
while (ask) {
char *answer = Getline("finalizeProof: enter the one you would like to finalize? [0] ");
if (answer) {
if (answer[0] == 'Q' || answer[0] == 'q') {
ask = kFALSE;
return;
}
TString sn(answer);
sn.Remove(sn.Length()-1);
if (sn.IsDigit()) {
qn = sn.Atoi();
if (qn >= 0 && qn < qa->GetEntriesFast()) {
break;
} else {
Printf("finalizeProof: choice must be in [0,%d] ('Q' to quit)",
qa->GetEntriesFast()-1);
}
} else {
if (sn.IsNull()) {
qn = 0;
break;
} else {
Printf("finalizeProof: choice must be a number in [0,%d] ('Q' to quit) (%s)",
qa->GetEntriesFast()-1, sn.Data());
}
}
}
}
}
ref = ((TObjString *)(qa->At(qn)))->GetName();
}
}
if (!ref.IsNull()) {
// Retrieve
proof->Retrieve(ref);
// Finalize
proof->Finalize(ref);
} else {
Printf("runProof: no queries to be finalized for analysis '%s'", act.Data());
return;
}
}
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...