-//===================== ANALYSIS TRAIN ==========================================
+//===================== ANALYSIS TRAIN =========================================
// To use: copy this macro to your work directory, modify the global part to match
// your needs, then run root.
// root[0] .L AnalysisTrain.C
// root[1] AnalysisTrainNew("grid", "full")
// CAF mode (requires root v5-23-02 + aliroot v4-16-Rev08)
// root[2] AnalysisTrainNew("proof")
-// Local mode requires AliESds.root or AliAOD.root in the work directory
+// Local mode requires AliESds.root or AliAOD.root in ./data directory
// root[3] AnalysisTrainNew("local")
+// In proof and grid modes, a token is needed and sourcing the produced environment file.
+//
+// If 'saveTrain' flag is set, the train will generate a directory name and run
+// in this directory. A configuration file 'ConfigTrain.C' will be generated.
+// One can replay at any time the train via:
+// root[1] AnalysisTrainNew(ana_mode, plugin_mode, "train_default_<date>/ConfigTrain.C")
-const char *root_version = "v5-23-02";
-const char *aliroot_version = "v4-16-Rev-08";
-const char *cluster = "alicecaf.cern.ch";
-//const char *AFversion = "AF-v4-16";
-const char *AFversion = "";
-// Dataset name. If empty assumes local ESD/AOD. In CAF it is a dataset name.
-// In grid has to point to an xml file or be empty if using the plugin
-// === PROOF
-const char *proof_dataset = "/COMMON/COMMON/LHC09a4_run8100X#/esdTree";
-
-// === ALIEN
-const char *dataset = "";
-const char *alien_datadir = "/alice/sim/PDC_09/LHC09a4/";
-Int_t run_numbers[10] = {81071, 0, 0, 0, 0,
- 0, 0, 0, 0, 0};
-
-Bool_t useDBG = kTRUE;
-Bool_t useMC = kFALSE;
-Bool_t usePAR = kTRUE;
-Bool_t useTR = kFALSE;
-Bool_t usePLUGIN = kTRUE;
-Bool_t useCORRFW = kFALSE; // do not change
-
-Int_t iAODanalysis = 0;
-Int_t iAODhandler = 1;
-Int_t iESDfilter = 1;
-Int_t iMUONfilter = 0;
-Int_t iMUONcopyAOD = 0;
-Int_t iJETAN = 1;
-Int_t iPWG4partcorr = 1;
-Int_t iPWG4pi0 = 1;
-Int_t iPWG3vertexing = 0;
-Int_t iPWG2femto = 0;
-Int_t iPWG2spectra = 1;
-Int_t iPWG2flow = 0;
-Int_t iPWG2res = 0;
+//================== TRAIN NAME ============================================
+TString train_name = "default"; // enters file names, so no blancs or special characters
+//==============================================================================
+
+// ### Settings that make sense in PROOF only
+//==============================================================================
+TString proof_cluster = "alicecaf.cern.ch";
+Bool_t useAFPAR = kFALSE; // use AF special par file
+TString AFversion = "AF-v4-16";
+// Change CAF dataset here
+TString proof_dataset = "/COMMON/COMMON/LHC09a4_run8100X#/esdTree";
+TString proof_outdir = "";
+
+// ### Settings that make sense when using the Alien plugin
+//==============================================================================
+Bool_t usePLUGIN = kTRUE; // do not change
+// Usage of par files ONLY in grid mode and ONLY if the code is not available
+// in the deployed AliRoot versions. Par file search path: local dir, if not there $ALICE_ROOT.
+// To refresh par files, remove the ones in the workdir, then do "make <target.par>" in
+// AliRoot.
+Bool_t usePAR = kFALSE; // use par files for extra libs
+Bool_t useCPAR = kFALSE; // use par files for common libs
+TString root_version = "v5-23-04";
+TString aliroot_version = "v4-17-02";
+// Change production base directory here
+TString alien_datadir = "/alice/sim/PDC_09/LHC09a4/";
+// Use up to 10 non-zero run numbers
+//Int_t run_numbers[10] = {81272, 81273 , 81274, 0, 0,
+// 0, 0, 0, 0, 0};
+Int_t run_range[2] = {81270, 81275};
+// ### Settings that make sense only for local analysis
+//==============================================================================
+// Change local xml dataset for local interactive analysis
+TString local_xmldataset = "";
+
+// ### Other flags to steer the analysis
+//==============================================================================
+Bool_t useDBG = kTRUE; // activate debugging
+Bool_t useMC = kTRUE; // use MC info
+Bool_t useTAGS = kFALSE; // use ESD tags for selection
+Bool_t useKFILTER = kTRUE; // use Kinematics filter
+Bool_t useTR = kFALSE; // use track references
+Bool_t useCORRFW = kFALSE; // do not change
+Bool_t useAODTAGS = kTRUE; // use AOD tags
+Bool_t saveTrain = kTRUE; // save train configuration as:
+Bool_t saveProofToAlien = kFALSE; // save proof outputs in AliEn
+ // train_[trainName]_ddMonthyyyy_time.C
+// ### Analysis modules to be included. Some may not be yet fully implemented.
+//==============================================================================
+Int_t iAODanalysis = 0; // Analysis on input AOD's
+Int_t iAODhandler = 1; // Analysis produces an AOD or dAOD's
+Int_t iESDfilter = 1; // ESD to AOD filter (barrel + muon tracks)
+Int_t iMUONcopyAOD = 0; // Task that copies only muon events in a separate AOD (PWG3)
+Int_t iJETAN = 1; // Jet analysis (PWG4) - needs ESD filter
+Int_t iPWG4partcorr = 1; // Gamma-hadron correlations task (PWG4)
+Int_t iPWG3vertexing = 1; // Vertexing HF task (PWG2)
+Int_t iPWG2femto = 1; // Femtoscopy task (PWG2)
+Int_t iPWG2spectra = 1; // Spectra PWG2 tasks (protons, cascades, V0 check, strange)
+Int_t iPWG2flow = 1; // Flow analysis task (PWG2)
+Int_t iPWG2res = 0; // Resonances task (PWG2)
+Int_t iPWG2kink = 1; // Kink analysis task (PWG2)
+// Temporaries.
TString anaPars = "";
TString anaLibs = "";
// Function signatures
class AliAnalysisGrid;
//______________________________________________________________________________
-void AnalysisTrainNew(const char *analysis_mode="grid", const char *plugin_mode="test")
+void AnalysisTrainNew(const char *analysis_mode="grid",
+ const char *plugin_mode="full",
+ const char *config_file="")
{
-// Example of running analysis train
+// Main analysis train macro. If a configuration file is provided, all parameters
+// are taken from there but may be altered by CheckModuleFlags.
+ if (strlen(config_file) && !LoadConfig(config_file)) return;
TString smode(analysis_mode);
- smode.ToUpper();
+ smode.ToUpper();
+ if (saveTrain) WriteConfig();
// Check compatibility of selected modules
CheckModuleFlags(smode);
printf("==================================================================\n");
- printf("=========== RUNNING ANALYSIS TRAIN IN %s MODE ==========\n", smode.Data());
+ printf("=========== RUNNING ANALYSIS TRAIN %s IN %s MODE ==========\n", train_name.Data(),smode.Data());
printf("==================================================================\n");
printf("= Configuring analysis train for: =\n");
if (iAODanalysis) printf("= AOD analysis =\n");
else printf("= ESD analysis =\n");
if (iESDfilter) printf("= ESD filter =\n");
- if (iMUONfilter) printf("= MUON filter =\n");
if (iMUONcopyAOD) printf("= MUON copy AOD =\n");
if (iJETAN) printf("= Jet analysis =\n");
if (iPWG2spectra) printf("= PWG2 proton, checkCascade, checkV0, strange =\n");
if (iPWG2femto) printf("= PWG2 femtoscopy =\n");
if (iPWG2flow) printf("= PWG2 flow =\n");
if (iPWG2res) printf("= PWG2 resonances =\n");
+ if (iPWG2kink) printf("= PWG2 kink analysis =\n");
if (iPWG3vertexing) printf("= PWG3 vertexing =\n");
- if (iPWG4partcorr) printf("= PWG4 gamma-hadron correlations =\n");
- if (iPWG4pi0) printf("= PWG4 pi0 analysis =\n");
+ if (iPWG4partcorr) printf("= PWG4 gamma-hadron, pi0 and gamma-jet correlations =\n");
printf("==================================================================\n");
printf(":: use MC truth %d\n", (UInt_t)useMC);
+ printf(":: use KINE filter %d\n", (UInt_t)useKFILTER);
printf(":: use track refs %d\n", (UInt_t)useTR);
+ printf(":: use tags %d\n", (UInt_t)useTAGS);
+ printf(":: use AOD tags %d\n", (UInt_t)useAODTAGS);
printf(":: use debugging %d\n", (UInt_t)useDBG);
printf(":: use PAR files %d\n", (UInt_t)usePAR);
printf(":: use AliEn plugin %d\n", (UInt_t)usePLUGIN);
+ //==========================================================================
// Connect to back-end system
if (!Connect(smode)) {
::Error("AnalysisTrain", "Could not connect to %s back-end", analysis_mode);
if (!LoadCommonLibraries(smode)) {
::Error("AnalysisTrain", "Could not load common libraries");
return;
- }
+ }
+ // Make the analysis manager and connect event handlers
+ AliAnalysisManager *mgr = new AliAnalysisManager("Analysis Train", "Production train");
+ if (saveProofToAlien) mgr->SetSpecialOutputLocation(proof_outdir);
+
// Load analysis specific libraries
if (!LoadAnalysisLibraries(smode)) {
::Error("AnalysisTrain", "Could not load analysis libraries");
return;
}
-
- //==========================================================================
- // Make the analysis manager and connect event handlers
- AliAnalysisManager *mgr = new AliAnalysisManager("Analysis Train", "Production train");
-
// Create input handler (input container created automatically)
if (iAODanalysis) {
// AOD input handler
} else {
// ESD input handler
AliESDInputHandler *esdHandler = new AliESDInputHandler();
+ if (useTAGS) esdHandler->SetReadTags();
mgr->SetInputEventHandler(esdHandler);
}
// Monte Carlo handler
if (iAODhandler) {
// AOD output handler
AliAODHandler* aodHandler = new AliAODHandler();
- mgr->SetOutputEventHandler(aodHandler);
aodHandler->SetOutputFileName("AliAOD.root");
+ mgr->SetOutputEventHandler(aodHandler);
+ if (iAODanalysis) {
+ aodHandler->SetCreateNonStandardAOD();
+ if (iPWG3vertexing) aodHandler->SetOutputFileName("AliAOD.VertexingHF.root");
+ }
}
// Debugging if needed
if (useDBG) mgr->SetDebugLevel(3);
if (iESDfilter && !iAODanalysis) {
// ESD filter task configuration.
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskESDFilter.C");
- AliAnalysisTaskESDfilter *taskesdfilter = AddTaskESDFilter();
+ AliAnalysisTaskESDfilter *taskesdfilter = AddTaskESDFilter(useKFILTER);
}
- // Muon filter
- if (iMUONfilter && !iAODanalysis) {
- // There is no configuration from outside for this task yet.
- AliAnalysisTaskESDMuonFilter *taskesdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
- mgr->AddTask(taskesdmuonfilter);
- // Connect to data containers
- mgr->ConnectInput (taskesdmuonfilter, 0, mgr->GetCommonInputContainer());
+ // AOD tags
+ if (useAODTAGS) {
+ AliAnalysisTaskTagCreator* tagTask = new AliAnalysisTaskTagCreator("AOD Tag Creator");
+ mgr->AddTask(tagTask);
+ AliAnalysisDataContainer *coutTags = mgr->CreateContainer("cTag", TTree::Class(),
+ AliAnalysisManager::kOutputContainer, "AOD.tag.root");
+ mgr->ConnectInput (tagTask, 0, mgr->GetCommonInputContainer());
+ mgr->ConnectOutput(tagTask, 1, coutTags);
}
// Jet analysis
if (iJETAN) {
gROOT->LoadMacro("$ALICE_ROOT/PWG4/macros/AddTaskJets.C");
- AliAnalysisTaskJets *taskjets = AddTaskJets("ESD", "UA1");
+ AliAnalysisTaskJets *taskjets = AddTaskJets("AOD", "UA1");
+ if (!taskjets) ::Warning("AnalysisTrainNew", "AliAnalysisTaskJets cannot run for this train conditions - EXCLUDED");
}
// Proton analysis
// protons
gROOT->LoadMacro("$ALICE_ROOT/PWG2/SPECTRA/macros/AddTaskProtons.C");
AliAnalysisTaskProtons *taskprotons = AddTaskProtons();
+ if (!taskprotons) ::Warning("AnalysisTrainNew", "AliAnalysisTaskProtons cannot run for this train conditions - EXCLUDED");
// cascades
- gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCheckCascade.C");
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/SPECTRA/macros/AddTaskCheckCascade.C");
AliAnalysisTaskCheckCascade *taskcheckcascade = AddTaskCheckCascade();
+ if (!taskcheckcascade) ::Warning("AnalysisTrainNew", "AliAnalysisTaskCheckCascade cannot run for this train conditions - EXCLUDED");
// v0's
- gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCheckV0.C");
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/SPECTRA/macros/AddTaskCheckV0.C");
AliAnalysisTaskCheckV0 *taskcheckV0 = AddTaskCheckV0();
+ if (!taskcheckV0) ::Warning("AnalysisTrainNew", "AliAnalysisTaskCheckV0 cannot run for this train conditions - EXCLUDED");
// strangeness
- gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskStrange.C");
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/SPECTRA/macros/AddTaskStrange.C");
AliAnalysisTaskStrange *taskstrange = AddTaskStrange();
+ if (!taskstrange) ::Warning("AnalysisTrainNew", "AliAnalysisTaskStrange cannot run for this train conditions - EXCLUDED");
}
+ // Femtoscopy analysis modules
+ if (iPWG2femto) {
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/FEMTOSCOPY/macros/AddTaskFemto.C");
+ AliAnalysisTaskFemto *taskfemto = AddTaskFemto();
+ if (!taskfemto) ::Warning("AnalysisTrainNew", "AliAnalysisTaskFemto cannot run for this train conditions - EXCLUDED");
+ }
+
+ // Kink analysis
+ if (iPWG2kink) {
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/KINK/macros/AddTaskKink.C");
+ AliAnalysisKinkESDMC *taskkink = AddTaskKink();
+ if (!taskkink) ::Warning("AnalysisTrainNew", "AliAnalysisKinkESDMC cannot run for this train conditions - EXCLUDED");
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/KINK/macros/AddTaskKinkResonance.C");
+ AliResonanceKinkPID *taskkinkres = AddTaskKinkResonance();
+ if (!taskkinkres) ::Warning("AnalysisTrainNew", "AliResonanceKinkPID cannot run for this train conditions - EXCLUDED");
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/KINK/macros/AddTaskKinkResonanceLikeSign.C");
+ AliResonanceKinkLikeSign *taskkinklikesign = AddTaskKinkResonanceLikeSign();
+ if (!taskkinklikesign) ::Warning("AnalysisTrainNew", "AliResonanceKinkLikeSign cannot run for this train conditions - EXCLUDED");
+ }
+
+ // Flow analysis
+ if (iPWG2flow) {
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/FLOW/macros/AddTaskFlow.C");
+ Bool_t SP = kTRUE;
+ Bool_t LYZ1 = kTRUE;
+ Bool_t LYZ2 = kFALSE;
+ Bool_t LYZEP = kFALSE;
+ Bool_t GFC = kTRUE;
+ Bool_t QC = kTRUE;
+ Bool_t FQD = kTRUE;
+ Bool_t MCEP = kFALSE; //not for pp
+ Bool_t kineFromESD = kTRUE;
+ Bool_t METHODS[] = {SP,LYZ1,LYZ2,LYZEP,GFC,QC,FQD,MCEP};
+ // Analysis type can be ESD, AOD, MC, ESDMC0, ESDMC1
+ TString type = "AOD";
+ if (!iAODanalysis) type = "ESD";
+ if (useMC) {
+ type += "MC";
+ if (!kineFromESD) type += "1";
+ else type += "0";
+ }
+ // Boolean to fill/not fill the QA histograms
+ Bool_t QA = kTRUE;
+ // Boolean to use/not use weights for the Q vector
+ Bool_t WEIGHTS[] = {kFALSE,kFALSE,kFALSE}; //Phi, v'(pt), v'(eta)
+ AliAnalysisTaskFlowEvent* taskFE = AddTaskFlow(type,METHODS,QA,WEIGHTS);
+ if (!taskFE) ::Warning("AnalysisTrainNew", "AliAnalysisTaskFlowEvent cannot run for this train conditions - EXCLUDED");
+ }
+
+ // PWG2 resonances
+ if (iPWG2res) {
+ gROOT->LoadMacro("$ALICE_ROOT/PWG2/RESONANCES/macros/AddAnalysisTaskRsn.C");
+ AddAnalysisTaskRsn("rsn.root", useMC);
+ }
+
+ // PWG3 vertexing
+ if (iPWG3vertexing) {
+ gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/AddTaskVertexingHF.C");
+ AliAnalysisTaskSEVertexingHF *taskvertexingHF = AddTaskVertexingHF();
+ if (!taskvertexingHF) ::Warning("AnalysisTrainNew", "AliAnalysisTaskSEVertexingHF cannot run for this train conditions - EXCLUDED");
+ }
+
// PWG4 hadron correlations
if (iPWG4partcorr) {
- gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskGammaHadronCorr.C");
- AliAnalysisTaskParticleCorrelation *taskgammahadron = AddTaskGammaHadronCorr();
- }
-
- // PWG4 pi0
- if (iPWG4pi0) {
- gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPi0.C");
- AliAnalysisTaskParticleCorrelation *taskpi0 = AddTaskPi0();
+ gROOT->LoadMacro("$ALICE_ROOT/PWG4/macros/AddTaskPartCorr.C");
+ AliAnalysisTaskParticleCorrelation *taskpartcorrPHOS = AddTaskPartCorr("AOD", "PHOS");
+ if (!taskpartcorrPHOS) ::Warning("AnalysisTrainNew", "AliAnalysisTaskParticleCorrelation PHOS cannot run for this train conditions - EXCLUDED");
+ AliAnalysisTaskParticleCorrelation *taskpartcorrEMCAL = AddTaskPartCorr("AOD", "EMCAL");
+ if (!taskpartcorrEMCAL) ::Warning("AnalysisTrainNew", "AliAnalysisTaskParticleCorrelation EMCAL cannot run for this train conditions - EXCLUDED");
}
-
//==========================================================================
// FOR THE REST OF THE TASKS THE MACRO AddTaskXXX() is not yet implemented/
// Run the analysis
//
if (mgr->InitAnalysis()) {
mgr->PrintStatus();
+ if (saveTrain) gSystem->ChangeDirectory(train_name);
StartAnalysis(smode, chain);
}
}
switch (imode) {
case 0:
if (!chain) {
- ::Error("StartAnalysis", "Cannot create the chain");
+ ::Error("AnalysisTrainNew.C::StartAnalysis", "Cannot create the chain");
return;
}
mgr->StartAnalysis(mode, chain);
return;
case 1:
- if (!strlen(proof_dataset)) {
- ::Error("StartAnalysis", "proof_dataset is empty");
+ if (!proof_dataset.Length()) {
+ ::Error("AnalysisTrainNew.C::StartAnalysis", "proof_dataset is empty");
return;
}
mgr->StartAnalysis(mode, proof_dataset, 1000);
case 2:
if (usePLUGIN) {
if (!mgr->GetGridHandler()) {
- ::Error("StartAnalysis", "Grid plugin not initialized");
+ ::Error("AnalysisTrainNew.C::StartAnalysis", "Grid plugin not initialized");
return;
}
mgr->StartAnalysis("grid");
} else {
if (!chain) {
- ::Error("StartAnalysis", "Cannot create the chain");
+ ::Error("AnalysisTrainNew.C::StartAnalysis", "Cannot create the chain");
return;
}
mgr->StartAnalysis(mode, chain);
if (!strcmp(mode, "PROOF")) imode = 1;
if (!strcmp(mode, "GRID")) imode = 2;
if (imode==1) {
- usePAR = kTRUE;
+ if (!usePAR) {
+ ::Info("AnalysisTrainNew.C::CheckModuleFlags", "PAR files enabled due to PROOF analysis");
+ usePAR = kTRUE;
+ }
+ }
+ if (imode != 2) {
+ ::Info("AnalysisTrainNew.C::CheckModuleFlags", "AliEn plugin disabled since not in GRID mode");
+ usePLUGIN = kFALSE;
}
if (iAODanalysis) {
// AOD analysis
+ if (useMC)
+ ::Info("AnalysisTrainNew.C::CheckModuleFlags", "MC usage disabled in analysis on AOD's");
useMC = kFALSE;
useTR = kFALSE;
+ if (iESDfilter)
+ ::Info("AnalysisTrainNew.C::CheckModuleFlags", "ESD filter disabled in analysis on AOD's");
iESDfilter = 0;
- iMUONfilter = 0;
+ if (!iAODhandler) {
+ if (iJETAN)
+ ::Info("AnalysisTrainNew.C::CheckModuleFlags", "JETAN disabled in analysis on AOD's without AOD handler");
+ iJETAN = 0;
+ }
// Disable tasks that do not work yet on AOD data
+ if (iPWG2kink)
+ ::Info("AnalysisTrainNew.C::CheckModuleFlags", "PWG2kink disabled in analysis on AOD's");
+ iPWG2kink = 0;
} else {
// ESD analysis
iMUONcopyAOD = 0;
- iPWG3vertexing = 0;
}
- if (iMUONfilter || iJETAN) iESDfilter=1;
+ if (iJETAN) iESDfilter=1;
if (iESDfilter) iAODhandler=1;
- if (iAODanalysis || !iAODhandler) {
- iPWG4partcorr=0;
- iPWG4pi0=0;
- }
- if (iPWG2spectra || iPWG2flow) useCORRFW = kTRUE;
+ if (iPWG2spectra || iPWG2flow || iPWG3vertexing) useCORRFW = kTRUE;
+ if (useKFILTER && !useMC) useKFILTER = kFALSE;
+ if (useAODTAGS && !iAODhandler) useAODTAGS = kFALSE;
}
//______________________________________________________________________________
break;
case 1:
if (!username.Length()) {
- ::Error(Form("Connect <%s>", mode), "Make sure you:\n \
+ ::Error(Form("AnalysisTrainNew.C::Connect <%s>", mode), "Make sure you:\n \
1. Have called: alien-token-init <username>\n \
2. Have called: >source /tmp/gclient_env_$UID");
return kFALSE;
}
- ::Info("Connect", "Connecting user <%s> to PROOF cluster <%s>",
- username.Data(), cluster);
- TProof::Open(Form("%s@%s", username.Data(), cluster));
+ ::Info("AnalysisTrainNew.C::Connect", "Connecting user <%s> to PROOF cluster <%s>",
+ username.Data(), proof_cluster.Data());
+ gEnv->SetValue("XSec.GSI.DelegProxy", "2");
+// TProof::Open(Form("%s@%s:31093", username.Data(), proof_cluster.Data()));
+ TProof::Open(Form("%s@%s", username.Data(), proof_cluster.Data()));
if (!gProof) {
if (strcmp(gSystem->Getenv("XrdSecGSISRVNAMES"), "lxfsrd0506.cern.ch"))
- ::Error(Form("Connect <%s>", mode), "Environment XrdSecGSISRVNAMES different from lxfsrd0506.cern.ch");
+ ::Error(Form("AnalysisTrainNew.C::Connect <%s>", mode), "Environment XrdSecGSISRVNAMES different from lxfsrd0506.cern.ch");
return kFALSE;
}
+ TGrid::Connect("alien://");
+ if (gGrid) {
+ TString homedir = gGrid->GetHomeDirectory();
+ TString workdir = homedir + train_name;
+ if (!gGrid->Cd(workdir)) {
+ gGrid->Cd(homedir);
+ if (gGrid->Mkdir(workdir)) {
+ gGrid->Cd(train_name);
+ ::Info("AnalysisTrainNew::Connect()", "Directory %s created", gGrid->Pwd());
+ }
+ }
+ gGrid->Mkdir("proof_output");
+ gGrid->Cd("proof_output");
+ proof_outdir = Form("alien://%s", gGrid->Pwd());
+ }
break;
case 2:
if (!username.Length()) {
- ::Error(Form("Connect <%s>", mode), "Make sure you:\n \
+ ::Error(Form("AnalysisTrainNew.C::Connect <%s>", mode), "Make sure you:\n \
1. Have called: alien-token-init <username>\n \
2. Have called: >source /tmp/gclient_env_$UID");
return kFALSE;
}
if (usePLUGIN && !gSystem->Getenv("alien_CLOSE_SE")) {
- ::Error(Form("Connect <%s>", mode),
+ ::Error(Form("AnalysisTrainNew.C::Connect <%s>", mode),
"When using the AliEn plugin it is preferable to define the \
variable alien_CLOSE_SE in your environment.");
return kFALSE;
}
- ::Info("Connect", "Connecting user <%s> to AliEn ...",
+ ::Info("AnalysisTrainNew.C::Connect", "Connecting user <%s> to AliEn ...",
username.Data());
TGrid::Connect("alien://");
if (!gGrid || !gGrid->IsConnected()) return kFALSE;
break;
default:
- ::Error("Connect", "Unknown run mode: %s", mode);
+ ::Error("AnalysisTrainNew.C::Connect", "Unknown run mode: %s", mode);
return kFALSE;
}
- ::Info("Connect","Connected in %s mode", mode);
+ ::Info("AnalysisTrainNew.C::Connect","Connected in %s mode", mode);
return kTRUE;
}
if (!strcmp(mode, "PROOF")) imode = 1;
if (!strcmp(mode, "GRID")) imode = 2;
if (!gSystem->Getenv("ALICE_ROOT")) {
- ::Error("LoadCommonLibraries", "Analysis train requires that analysis libraries are compiled with a local AliRoot");
+ ::Error("AnalysisTrainNew.C::LoadCommonLibraries", "Analysis train requires that analysis libraries are compiled with a local AliRoot");
return kFALSE;
}
Bool_t success = kTRUE;
switch (imode) {
case 0:
case 2:
- success &= LoadLibrary("libSTEERBase.so", mode);
- success &= LoadLibrary("libESD.so", mode);
- success &= LoadLibrary("libAOD.so", mode);
- success &= LoadLibrary("libANALYSIS.so", mode);
- success &= LoadLibrary("libANALYSISalice.so", mode);
- if (useCORRFW) success &= LoadLibrary("libCORRFW.so", mode);
- gROOT->ProcessLine(".include $ALICE_ROOT/include");
+ if (useCPAR) {
+ success &= LoadLibrary("STEERBase", mode, kTRUE);
+ success &= LoadLibrary("ESD", mode, kTRUE);
+ success &= LoadLibrary("AOD", mode, kTRUE);
+ success &= LoadLibrary("ANALYSIS", mode, kTRUE);
+ success &= LoadLibrary("ANALYSISalice", mode, kTRUE);
+ if (useCORRFW) success &= LoadLibrary("CORRFW", mode, kTRUE);
+ } else {
+ success &= LoadLibrary("libSTEERBase.so", mode);
+ success &= LoadLibrary("libESD.so", mode);
+ success &= LoadLibrary("libAOD.so", mode);
+ success &= LoadLibrary("libANALYSIS.so", mode);
+ success &= LoadLibrary("libANALYSISalice.so", mode);
+ if (useCORRFW) success &= LoadLibrary("libCORRFW.so", mode);
+ gROOT->ProcessLine(".include $ALICE_ROOT/include");
+ }
break;
case 1:
Int_t ires = -1;
- if (!gSystem->AccessPathName(AFversion)) ires = gProof->UploadPackage(AFversion);
+ if (useAFPAR && !gSystem->AccessPathName(AFversion)) ires = gProof->UploadPackage(AFversion);
if (ires < 0) {
success &= LoadLibrary("STEERBase", mode);
success &= LoadLibrary("ESD", mode);
if (useCORRFW) success &= LoadLibrary("CORRFW", mode);
} else {
ires = gProof->EnablePackage(AFversion);
+ if (ires<0) success = kFALSE;
if (useCORRFW) success &= LoadLibrary("CORRFW", mode);
}
- if (ires<0) success = kFALSE;
break;
default:
- ::Error("LoadCommonLibraries", "Unknown run mode: %s", mode);
+ ::Error("AnalysisTrainNew.C::LoadCommonLibraries", "Unknown run mode: %s", mode);
return kFALSE;
}
if (success) {
- ::Info("LoadCommodLibraries", "Load common libraries: SUCCESS");
- ::Info("LoadCommodLibraries", "Include path for Aclic compilation:\n%s",
+ ::Info("AnalysisTrainNew.C::LoadCommodLibraries", "Load common libraries: SUCCESS");
+ ::Info("AnalysisTrainNew.C::LoadCommodLibraries", "Include path for Aclic compilation:\n%s",
gSystem->GetIncludePath());
} else {
- ::Info("LoadCommodLibraries", "Load common libraries: FAILED");
+ ::Info("AnalysisTrainNew.C::LoadCommodLibraries", "Load common libraries: FAILED");
}
return success;
{
// Load common analysis libraries.
Bool_t success = kTRUE;
- if (iMUONfilter) {
+ if (iESDfilter) {
if (!LoadLibrary("PWG3base", mode, kTRUE) ||
!LoadLibrary("PWG3muon", mode, kTRUE)) return kFALSE;
}
}
// PWG4 particle correlations
- if (iPWG4partcorr || iPWG4pi0) {
+ if (iPWG4partcorr) {
if (!LoadLibrary("PWG4PartCorrBase", mode, kTRUE) ||
!LoadLibrary("PWG4PartCorrDep", mode, kTRUE)) return kFALSE;
- TFile::Cp(gSystem->ExpandPathName("$(ALICE_ROOT)/PWG4/macros/ConfigAnalysisGammaHadronCorrelation.C"), "ConfigAnalysisGammaHadronCorrelation.C");
- TFile::Cp(gSystem->ExpandPathName("$(ALICE_ROOT)/PWG4/macros/ConfigAnalysisPi0.C"), "ConfigAnalysisPi0.C");
}
// PWG2 task protons
if (iPWG2spectra) {
}
// PWG2 flow
if (iPWG2flow) {
- if (!LoadLibrary("PWG2AOD", mode, kTRUE) ||
- !LoadLibrary("PWG2flow", mode, kTRUE)) return kFALSE;
+ if (!LoadLibrary("PWG2flowCommon", mode, kTRUE) ||
+ !LoadLibrary("PWG2flowTasks", mode, kTRUE)) return kFALSE;
}
// PWG2 resonances
if (iPWG2res) {
if (!LoadLibrary("PWG2resonances", mode, kTRUE)) return kFALSE;
}
+ // PWG2 kink
+ if (iPWG2kink) {
+ if (!LoadLibrary("PWG2kink", mode, kTRUE)) return kFALSE;
+ }
// PWG2 femtoscopy
if (iPWG2femto) {
if (!LoadLibrary("PWG2AOD", mode, kTRUE) ||
!LoadLibrary("PWG2femtoscopy", mode, kTRUE) ||
!LoadLibrary("PWG2femtoscopyUser", mode, kTRUE)) return kFALSE;
+ TFile::Cp(gSystem->ExpandPathName("$(ALICE_ROOT)/PWG2/FEMTOSCOPY/macros/ConfigFemtoAnalysis.C"), Form("%s/ConfigFemtoAnalysis.C", train_name.Data()));
+ anaLibs += "ConfigFemtoAnalysis.C ";
}
// Vertexing HF
if (iPWG3vertexing) {
- if (!LoadLibrary("PWG3vertexingHF", mode, kTRUE)) return kFALSE;
+ if (!LoadLibrary("PWG3base", mode, kTRUE) ||
+ !LoadLibrary("PWG3vertexingHF", mode, kTRUE)) return kFALSE;
}
- ::Info("LoadAnalysisLibraries", "Load other libraries: SUCCESS");
+ ::Info("AnalysisTrainNew.C::LoadAnalysisLibraries", "Load other libraries: SUCCESS");
return kTRUE;
}
if (!strcmp(mode, "GRID")) imode = 2;
TString mod(module);
if (!mod.Length()) {
- ::Error("LoadLibrary", "Empty module name");
+ ::Error("AnalysisTrainNew.C::LoadLibrary", "Empty module name");
return kFALSE;
}
// If a library is specified, just load it
mod.Remove(mod.Index(".so"));
result = gSystem->Load(mod);
if (result < 0) {
- ::Error("LoadLibrary", "Could not load library %s", module);
+ ::Error("AnalysisTrainNew.C::LoadLibrary", "Could not load library %s", module);
return kFALSE;
}
if (rec) anaLibs += Form("%s.so ",mod.Data());
if (result<0) {
result = gProof->UploadPackage(gSystem->ExpandPathName(Form("$ALICE_ROOT/%s.par", module)));
if (result<0) {
- ::Error("LoadLibrary", "Could not find module %s.par in current directory nor in $ALICE_ROOT", module);
+ ::Error("AnalysisTrainNew.C::LoadLibrary", "Could not find module %s.par in current directory nor in $ALICE_ROOT", module);
return kFALSE;
}
}
return kFALSE;
}
if (result < 0) {
- ::Error("LoadLibrary", "Could not load module %s", module);
+ ::Error("AnalysisTrainNew.C::LoadLibrary", "Could not load module %s", module);
return kFALSE;
}
return kTRUE;
switch (imode) {
case 0:
if (iAODanalysis) {
- if (!strlen(dataset)) {
+ if (!local_xmldataset.Length()) {
// Local AOD
chain = new TChain("aodTree");
- if (gSystem->AccessPathName("AliAOD.root"))
- ::Error("CreateChain", "File: AliAOD.root not in current dir");
- else chain->Add("AliAOD.root");
+ if (gSystem->AccessPathName("data/AliAOD.root"))
+ ::Error("AnalysisTrainNew.C::CreateChain", "File: AliAOD.root not in ./data dir");
+ else {
+ if (!saveTrain) chain->Add("data/AliAOD.root");
+ else chain->Add("../data/AliAOD.root");
+ }
} else {
// Interactive AOD
- chain = CreateChainSingle(dataset, "aodTree");
+ chain = CreateChainSingle(local_xmldataset, "aodTree");
}
} else {
- if (!strlen(dataset)) {
+ if (!local_xmldataset.Length()) {
// Local ESD
chain = new TChain("esdTree");
- if (gSystem->AccessPathName("AliESDs.root"))
- ::Error("CreateChain", "File: AliESDs.root not in current dir");
- else chain->Add("AliESDs.root");
+ if (gSystem->AccessPathName("data/AliESDs.root"))
+ ::Error("AnalysisTrainNew.C::CreateChain", "File: AliESDs.root not in ./data dir");
+ else {
+ if (!saveTrain) chain->Add("data/AliESDs.root");
+ else chain->Add("../data/AliESDs.root");
+ }
} else {
// Interactive ESD
- chain = CreateChainSingle(dataset, "esdTree");
+ chain = CreateChainSingle(local_xmldataset, "esdTree");
}
}
break;
TAlienCollection * myCollection = TAlienCollection::Open(xmlfile);
if (!myCollection) {
- ::Error("CreateChainSingle", "Cannot create an AliEn collection from %s", xmlfile) ;
+ ::Error("AnalysisTrainNew.C::CreateChainSingle", "Cannot create an AliEn collection from %s", xmlfile) ;
return NULL ;
}
char processline[1024];
if (gSystem->AccessPathName(Form("%s.par", pararchivename))) {
if (!gSystem->AccessPathName(Form("%s/%s.par", gSystem->Getenv("ALICE_ROOT"),pararchivename))) {
- ::Info("SetupPar", "Getting %s.par from $ALICE_ROOT", pararchivename);
+ ::Info("AnalysisTrainNew.C::SetupPar", "Getting %s.par from $ALICE_ROOT", pararchivename);
TFile::Cp(gSystem->ExpandPathName(Form("$ALICE_ROOT/%s.par", pararchivename)),
Form("%s.par",pararchivename));
} else {
- ::Error("SetupPar", "Cannot find %s.par", pararchivename);
+ ::Error("AnalysisTrainNew.C::SetupPar", "Cannot find %s.par", pararchivename);
return -1;
}
}
+ if (usePLUGIN && saveTrain) gSystem->Exec(Form("ln -s ../%s.par %s",pararchivename, train_name.Data()));
gSystem->Exec(Form("tar xvzf %s.par", pararchivename));
TString ocwd = gSystem->WorkingDirectory();
// Define production directory LFN
plugin->SetGridDataDir(alien_datadir);
// Set data search pattern
- plugin->SetDataPattern("*ESDs.root");
+ if (iAODanalysis) plugin->SetDataPattern("*AliAOD.root");
+ else plugin->SetDataPattern("*AliESDs.root");
// ...then add run numbers to be considered
- for (Int_t i=0; i<10; i++) {
- if (run_numbers[i]==0) break;
- plugin->AddRunNumber(run_numbers[i]);
- }
+ plugin->SetRunRange(run_range[0], run_range[1]);
+// for (Int_t i=0; i<10; i++) {
+// if (run_numbers[i]==0) break;
+// plugin->AddRunNumber(run_numbers[i]);
+// }
// Method 2: Declare existing data files (raw collections, xml collections, root file)
// If no path mentioned data is supposed to be in the work directory (see SetGridWorkingDir())
// XML collections added via this method can be combined with the first method if
// plugin->AddDataFile("tag.xml");
// plugin->AddDataFile("/alice/data/2008/LHC08c/000057657/raw/Run57657.Merged.RAW.tag.root");
// Define alien work directory where all files will be copied. Relative to alien $HOME.
- plugin->SetGridWorkingDir("analysisESD");
+ if (iAODanalysis) plugin->SetGridWorkingDir("analysisAOD");
+ else plugin->SetGridWorkingDir("analysisESD");
// Declare alien output directory. Relative to working directory.
- plugin->SetGridOutputDir("output"); // In this case will be $HOME/work/output
+ plugin->SetGridOutputDir(Form("output_%s",train_name.Data())); // In this case will be $HOME/work/output
TString ana_sources = "";
TString ana_add = "";
if (usePAR && anaPars.Length()) {
+ printf("%s\n", anaPars.Data());
TObjArray *arr;
TObjString *objstr;
arr = anaPars.Tokenize(" ");
// (can be like: file.root or file.root@ALICE::Niham::File)
plugin->SetDefaultOutputs();
plugin->SetMergeExcludes("AliAOD.root");
-// plugin->SetOutputFiles("AliAOD.root");
// Optionally define the files to be archived.
// plugin->SetOutputArchive("log_archive.zip:stdout,stderr@ALICE::NIHAM::File root_archive.zip:*.root@ALICE::NIHAM::File");
plugin->SetOutputArchive("log_archive.zip:stdout,stderr");
// Optionally set a name for the generated analysis macro (default MyAnalysis.C)
- plugin->SetAnalysisMacro("testAnalysis.C");
+ plugin->SetAnalysisMacro(Form("%s.C", train_name.Data()));
// Optionally set maximum number of input files/subjob (default 100, put 0 to ignore)
- plugin->SetSplitMaxInputFileNumber(50);
+ plugin->SetSplitMaxInputFileNumber(100);
// Optionally set number of failed jobs that will trigger killing waiting sub-jobs.
// plugin->SetMaxInitFailed(5);
// Optionally resubmit threshold.
// Optionally set input format (default xml-single)
plugin->SetInputFormat("xml-single");
// Optionally modify the name of the generated JDL (default analysis.jdl)
- plugin->SetJDLName("testAnalysis.jdl");
+ plugin->SetJDLName(Form("%s.jdl", train_name.Data()));
+// Optionally modify the executable name (default analysis.sh)
+ plugin->SetExecutable(Form("%s.sh", train_name.Data()));
// Optionally modify job price (default 1)
plugin->SetPrice(1);
// Optionally modify split mode (default 'se')
plugin->SetSplitMode("se");
return plugin;
}
+
+//______________________________________________________________________________
+void WriteConfig()
+{
+// Write train configuration in a file. The file name has the format:
+// train_[trainName]_ddMonthyyyy_time.C
+ gSystem->Exec("date +%d%b%Y_%Hh%M > date.tmp");
+ ifstream fdate("date.tmp");
+ if (!fdate.is_open()) {
+ ::Error("AnalysisTrainNew.C::Export","Could not generate file name");
+ return;
+ }
+ const char date[64];
+ fdate.getline(date,64);
+ fdate.close();
+ gSystem->Exec("rm date.tmp");
+ train_name = Form("train_%s_%s", train_name.Data(), date);
+ TString cdir = gSystem->WorkingDirectory();
+ gSystem->MakeDirectory(train_name);
+ gSystem->ChangeDirectory(train_name);
+ ofstream out;
+ out.open("ConfigTrain.C", ios::out);
+ if (out.bad()) {
+ ::Error("AnalysisTrainNew.C::Export", "Cannot open ConfigTrain.C for writing");
+ return;
+ }
+ out << "{" << endl;
+ out << " train_name = " << "\"" << train_name.Data() << "\";" << endl;
+ out << " proof_cluster = " << "\"" << proof_cluster.Data() << "\";" << endl;
+ out << " useAFPAR = " << useAFPAR << ";" << endl;
+ if (useAFPAR)
+ out << " AFversion = " << AFversion.Data() << ";" << endl;
+ out << " proof_dataset = " << "\"" << proof_dataset.Data() << "\";" << endl;
+ out << " usePLUGIN = " << usePLUGIN << ";" << endl;
+ out << " usePAR = " << usePAR << ";" << endl;
+ out << " useCPAR = " << useCPAR << ";" << endl;
+ out << " root_version = " << "\"" << root_version.Data() << "\";" << endl;
+ out << " aliroot_version = " << "\"" << aliroot_version.Data() << "\";" << endl;
+ out << " alien_datadir = " << "\"" << alien_datadir.Data() << "\";" << endl;
+// for (Int_t i=0; i<10; i++) {
+// if (run_numbers[i])
+// out << " run_numbers[" << i << "] = " << run_numbers[i] << ";" << endl;
+// }
+ out << " run_range[0] = " << run_range[0] << ";" << endl;
+ out << " run_range[1] = " << run_range[1] << ";" << endl;
+ out << " useDBG = " << useDBG << ";" << endl;
+ out << " useMC = " << useMC << ";" << endl;
+ out << " useTAGS = " << useTAGS << ";" << endl;
+ out << " useKFILTER = " << useKFILTER << ";" << endl;
+ out << " useTR = " << useTR << ";" << endl;
+ out << " useCORRFW = " << useCORRFW << ";" << endl;
+ out << " useAODTAGS = " << useAODTAGS << ";" << endl;
+ out << " saveTrain = " << "kFALSE;" << endl << endl;
+ out << " // Analysis modules" << endl;
+ out << " iAODanalysis = " << iAODanalysis << ";" << endl;
+ out << " iAODhandler = " << iAODhandler << ";" << endl;
+ out << " iESDfilter = " << iESDfilter << ";" << endl;
+ out << " iMUONcopyAOD = " << iMUONcopyAOD << ";" << endl;
+ out << " iJETAN = " << iJETAN << ";" << endl;
+ out << " iPWG4partcorr = " << iPWG4partcorr << ";" << endl;
+ out << " iPWG2femto = " << iPWG2femto << ";" << endl;
+ out << " iPWG2spectra = " << iPWG2spectra << ";" << endl;
+ out << " iPWG2flow = " << iPWG2flow << ";" << endl;
+ out << " iPWG2res = " << iPWG2res << ";" << endl;
+ out << " iPWG2kink = " << iPWG2kink << ";" << endl;
+ out << "}" << endl;
+ ::Info("AnalysisTrainNew.C::WriteConfig", "Train configuration wrote to file %s", Form("config_%s.C", train_name.Data()));
+ gSystem->ChangeDirectory(cdir);
+}
+
+//______________________________________________________________________________
+Bool_t LoadConfig(const char *filename)
+{
+// Read train configuration from file
+ if (gSystem->AccessPathName(filename)) {
+ ::Error("AnalysisTrainNew.C::LoadConfig", "Config file name not found");
+ return kFALSE;
+ }
+ gROOT->ProcessLine(Form(".x %s", filename));
+ ::Info("AnalysisTrainNew.C::LoadConfig", "Train configuration loaded from file %s", filename);
+ return kTRUE;
+}