KVDataAnalysisLauncher.cpp

#include "KVDataAnalysisLauncher.h"
#include "TGLabel.h"
#include "TGMsgBox.h"
#include "TColor.h"
#include "TGButtonGroup.h"
#include "TGFileDialog.h"
#include "TObjString.h"
#include "TDatime.h"
#include "TApplication.h"
#include "TSocket.h"
#include "TGToolTip.h"
#include "TGFileDialog.h"
#include "Riostream.h"
#include "TClass.h"
#include "KVDataRepositoryManager.h"
#include "KVDataSetRepository.h"
#include "KVDataSetManager.h"
#include "KVDataSet.h"
#include "KVDataAnalysisTask.h"
#include "KV2Body.h"
#include "KVNucleus.h"
#include "KVExpDB.h"
#include "KVBatchSystemManager.h"
#include "TSystemDirectory.h"
#include "KVInputDialog.h"
#include "KVBatchSystemGUI.h"
#include "KVBatchSystemParametersGUI.h"
#include <KVRunFile.h>
#include <optional>
 
#define TTDELAY 750
 
//maximum length of runlist shown next to "Selected Runs :"
#define MAX_LENGTH_SELECTED_RUNS 40
 
//#define KVDAL_DEBUG
 
using namespace std;
 
ClassImp(KVGFileList)
 
//__________________________________________
 
 
void KVGFileList::Init(const TString& fileList, const TString& title)
{
   // Init window
   TGLayoutHints* eXeY = new TGLayoutHints(kLHintsLeft | kLHintsTop |
                                           kLHintsExpandX | kLHintsExpandY,
                                           1, 1, 1, 1);
   TGLayoutHints* eX = new TGLayoutHints(kLHintsLeft | kLHintsTop |
                                         kLHintsExpandX,
                                         1, 1, 1, 1);
 
   // File list TGListBox
   lbFileList = new TGListBox(this, LB_Files);
   lbFileList->SetMultipleSelections(kTRUE);
   lbFileList->Resize(350, 96);
   this->AddFrame(lbFileList, eXeY);
 
   // File Name TGTextEntry
   teFileName = new TGTextEntry(this, "");
   this->AddFrame(teFileName, eX);
 
   // buttons
   TGCompositeFrame* cf = new TGCompositeFrame(this, 1200, 350, kHorizontalFrame);
   boutAdd = new TGTextButton(cf, "Add File");
   boutAdd->SetToolTipText("Add a file to the list", TTDELAY);
   boutAdd->Connect("Clicked()",
                    "KVGFileList",
                    this,
                    "AddFile()");
   cf->AddFrame(boutAdd, eX);
 
   boutRem = new TGTextButton(cf, "Remove File");
   boutRem->SetToolTipText("Remove the selected file from the list", TTDELAY);
   boutRem->Connect("Clicked()",
                    "KVGFileList",
                    this,
                    "RemoveFiles()");
   cf->AddFrame(boutRem, eX);
 
   boutAllRem = new TGTextButton(cf, "Remove All File");
   boutAllRem->SetToolTipText("Remove all files from the list", TTDELAY);
   boutAllRem->Connect("Clicked()",
                       "KVGFileList",
                       this,
                       "RemoveAllFiles()");
   cf->AddFrame(boutAllRem, eX);
 
   this->AddFrame(cf, eX);
 
   TGTextButton* bout = new TGTextButton(this, "Done");
   bout->SetToolTipText("Close window", TTDELAY);
   bout->Connect("Clicked()",
                 "KVGFileList",
                 this,
                 "Done()");
   this->AddFrame(bout, eX);
 
   // Map all widgets
   MapSubwindows();
   Resize(GetDefaultSize());
   SetWindowName(title);
   SetIconName(title);
 
   fileListString = fileList;
   fileDialogDir = gSystem->Getenv("PWD");
 
   entryMax = 0;
}
 
 
 
 
KVGFileList::KVGFileList(const TString& fileList, const TString& title,
                         const TGWindow* p, const TGWindow* main,
                         Bool_t ok):
   TGTransientFrame(p, main, 10, 10)
{
   // Createur
 
   if (ok) {
      Init(fileList, title);
      InitFileList();
      MapRaised();
      fClient->WaitFor(this);
   }
}
 
 
 
 
KVGFileList::~KVGFileList()
{
   // Destructeur
   delete lbFileList;
   delete teFileName;
   delete boutAdd;
   delete boutRem;
   delete boutAllRem;
}
 
 
 
 
void KVGFileList::InitFileList()
{
   // Init the file name list box from a TString. Add all fields separated by a
   // white space.
 
   fileListString.Begin(" ");
 
 
   while (!fileListString.End()) {
      auto fileName = fileListString.Next();
      if (fileName.Length() && !lbFileList->FindEntry(fileName)) {
         lbFileList->AddEntry(new TGString(fileName), entryMax);
         entryMax++;
      }
   }
 
   lbFileList->MapSubwindows();
   lbFileList->Layout();
}
 
 
 
 
void KVGFileList::AddFile(void)
{
   // Add a file to the listbox. If the TGTextEntry is ampty, a file open dialog
   // appears
   TString fileName = teFileName->GetText();
   if (!fileName.Length())
      fileName = GetFileFromDialog();
   if (fileName.Length() &&
         !lbFileList->FindEntry(fileName) &&
         CanAdd(fileName)) {
      lbFileList->AddEntry(new TGString(fileName), entryMax);
      entryMax++;
   }
   lbFileList->MapSubwindows();
   lbFileList->Layout();
}
 
 
 
 
void KVGFileList::RemoveFiles(void)
{
   // Remove all the selected files from the TGListBox
 
   TList sel;
   lbFileList->GetSelectedEntries(&sel);
   TIter next(&sel);
   TGLBEntry* e = nullptr;
   while ((e = (TGLBEntry*)next())) {
      gSystem->Unload(((TGTextLBEntry*)e)->GetText()->GetString());
      lbFileList->RemoveEntry(e->EntryId());
   }
   lbFileList->MapSubwindows();
   lbFileList->Layout();
}
 
 
 
 
void KVGFileList::RemoveAllFiles(void)
{
   // Remove all the files from the TGListBox
   lbFileList->RemoveEntries(0, entryMax);
   entryMax = 0;
   lbFileList->MapSubwindows();
   lbFileList->Layout();
}
 
 
 
 
void KVGFileList::Done(void)
{
   // build the file list string from the content of the TGListBox
   fileListString = "";
   for (Int_t i = 0; i < entryMax; i++) {
      if (TGTextLBEntry* e = (TGTextLBEntry*)lbFileList->GetEntry(i)) {
         if (fileListString.Length()) fileListString += " ";
         fileListString += e->GetText()->GetString();
      }
   }
   CloseWindow();
}
 
 
 
 
TString KVGFileList::GetFileFromDialog(void)
{
   // Gets the file name from a TGFileDialog
   TGFileInfo fil;
   const char* filTypes[] = {"Shared Object Files", "*.so",
                             0,                  0
                            };
   fil.fFileTypes = filTypes;
   fil.fIniDir = StrDup(fileDialogDir.Data());
   new TGFileDialog(gClient->GetRoot(), this, kFDOpen, &fil);
   fileDialogDir = fil.fIniDir;
   if (fil.fFilename)
      return fil.fFilename;
   return {};
}
 
 
 
 
Bool_t KVGFileList::CanAdd(const TString& fn)
{
   // tells whether the file in ths string fn can be added to the list box
   Bool_t ok = kTRUE;
 
   FileStat_t fs;
   auto tmp = fn;
   gSystem->ExpandPathName(tmp);
   if (gSystem->GetPathInfo(tmp, fs)) {
      ok = kFALSE;
      new TGMsgBox(gClient->GetRoot(), this,
                   "File does not exist",
                   Form("The file \"%s\" does not exist. Nothing added.",
                        fn.Data()),
                   kMBIconExclamation);
   }
   else {
      auto fname = fn;
      if (!fname.EndsWith(".so")) {
         ok = kFALSE;
         new TGMsgBox(gClient->GetRoot(), this,
                      "Not a share object",
                      Form("The file \"%s\" is not a shared object. Nothing added.",
                           fn.Data()),
                      kMBIconExclamation);
 
      }
   }
   return ok;
}
 
 
 
ClassImp(KVGDirectoryList)
 
 
 
void KVGDirectoryList::Init(const TString& fileList, const TString& title)
{
   // init window
   KVGFileList::Init(fileList, title);
 
   boutAdd->SetText("Add Directory");
   boutAdd->SetToolTipText("Add a directory to the list.", TTDELAY);
   boutAdd->Layout();
 
   boutRem->SetText("Remove Directory");
   boutRem->SetToolTipText("Remove the selected directories from the list.", TTDELAY);
   boutRem->Layout();
 
   boutAllRem->SetText("Remove All Directories");
   boutAllRem->SetToolTipText("Remove all directories from the list.", TTDELAY);
   boutAllRem->Layout();
}
 
 
 
 
KVGDirectoryList::KVGDirectoryList(const TString& fileList, const TString& title,
                                   const TGWindow* p, const TGWindow* main,
                                   Bool_t ok):
   KVGFileList(fileList, title, p, main, kFALSE)
{
   // Createur
 
   if (ok) {
      Init(fileList, title);
      InitFileList();
      MapRaised();
      fClient->WaitFor(this);
   }
}
 
 
 
 
TString KVGDirectoryList::GetFileFromDialog(void)
{
   // Gets the file name from a TGFileDialog
   TGFileInfo fil;
   const char* filTypes[] = {"Include Files", "*.[h,H]*",
                             0,                  0
                            };
   fil.fFileTypes = filTypes;
   fil.fIniDir = StrDup(fileDialogDir.Data());
   new TGFileDialog(gClient->GetRoot(), this, kFDOpen, &fil);
   fileDialogDir = fil.fIniDir;
   if (fil.fFilename)
      return fil.fIniDir;
   return {};
}
 
 
 
 
Bool_t KVGDirectoryList::CanAdd(const TString& fn)
{
   // tells whether the file in ths string fn can be added to the list box
   Bool_t ok = kTRUE;
 
   FileStat_t fs;
   TString tmp = fn;
   gSystem->ExpandPathName(tmp);
   if (gSystem->GetPathInfo(tmp.Data(), fs)) {
      ok = kFALSE;
      new TGMsgBox(gClient->GetRoot(), this,
                   "Directory does not exist",
                   Form("The directory \"%s\" does not exist. Nothing added.",
                        fn.Data()),
                   kMBIconExclamation);
   }
   else {
      if (!(fs.fMode & kS_IFDIR)) {
         ok = kFALSE;
         new TGMsgBox(gClient->GetRoot(), this,
                      "Not a directory",
                      Form("The path \"%s\" is not a directory. Nothing added.",
                           fn.Data()),
                      kMBIconExclamation);
 
      }
   }
   return ok;
}
 
 
ClassImp(KVDataAnalysisLauncher)
 
 
 
KVDataAnalysisLauncher::KVDataAnalysisLauncher(const TGWindow* p, UInt_t w, UInt_t h): TGMainFrame(p, w, h)
{
   // Main window width and height can be set using .kvrootrc variables:
   //     KaliVedaGUI.MainGUIWidth:       800
   //     KaliVedaGUI.MainGUIHeight:       600
   // Maximum column width of runlist can be set using:
   //     KaliVedaGUI.MaxColWidth:       500
 
   entryMax = -1;
   //Initialisation of resource file
   GUIenv = new TEnv(".KVDataAnalysisGUIrc");
   //initialisation of list used by Get/SetResource
   ResourceNames.Add(new TNamed("Repository", ""));
   ResourceNames.Add(new TNamed("DataSet", ""));
   ResourceNames.Add(new TNamed("Task", ""));
   NbResNames = ResourceNames.GetEntries();
 
   // list of user analysis classes present in working directory
   UserClassNames = new TList;
   UserClassNames->SetOwner(kTRUE);
 
   //initialise repositories, datasets, etc.
   if (!gDataRepositoryManager) {
      new KVDataRepositoryManager();
      gDataRepositoryManager->Init();
   }
   if (gDataRepositoryManager->GetListOfRepositories()->GetEntries() == 0) {
      exit(1);
   }
   ia = 0;
   GetDataAnalyser();//this will set up gBatchSystemManager
   gBatchSystemManager->GetDefaultBatchSystem()->cd();
 
   // Creation de l'environnement d'affichage et ajout des 2 boutons a cet
   // environnement
 
   TGLayoutHints* eXeY = new TGLayoutHints(kLHintsLeft | kLHintsTop |
                                           kLHintsExpandX | kLHintsExpandY,
                                           1, 1, 1, 1);
   TGLayoutHints* eX = new TGLayoutHints(kLHintsLeft | kLHintsTop |
                                         kLHintsExpandX,
                                         1, 1, 1, 1);
   TGLayoutHints* centerX = new TGLayoutHints(kLHintsCenterX | kLHintsTop,
         1, 1, 1, 1);
   TGLayoutHints* LHtopleft = new TGLayoutHints(kLHintsLeft | kLHintsTop,
         1, 1, 1, 1);
   fMainGuiWidth = gEnv->GetValue("KaliVedaGUI.MainGUIWidth", 400);
   fMainGuiHeight = gEnv->GetValue("KaliVedaGUI.MainGUIHeight", 600);
 
   // Creation des 3 frames dans la fenetre
   TGCompositeFrame* cfSelect = new TGCompositeFrame(this, fMainGuiWidth, 350, kVerticalFrame);
 
   Int_t justMode = kTextBottom | kTextRight;
   TGCompositeFrame* cf = new TGCompositeFrame(cfSelect, fMainGuiWidth, 350, kHorizontalFrame);
   // Label du Repository
   TGLabel* lab = new TGLabel(cf, "DATA REPOSITORY : ");
   lab->Resize(150, 20);
   lab->SetTextJustify(justMode);
   cf->AddFrame(lab);
   // ComboBox du Repository
   cbRepository = new TGComboBox(cf, CB_DataRepository);
   cbRepository->Select(-1);
   cbRepository->Resize(150, 20);
   cbRepository->Connect("Selected(Int_t)",
                         "KVDataAnalysisLauncher",
                         this,
                         "SelectedRepository(Int_t)");
   cf->AddFrame(cbRepository, LHtopleft);
 
   // Label du Data Set
   lab = new TGLabel(cf, "DATASET : ");
   lab->Resize(150, 20);
   lab->SetTextJustify(justMode);
   cf->AddFrame(lab, new TGLayoutHints(kLHintsLeft | kLHintsTop,
                                       20, 1, 1, 1));
   // ComboBox du Data Set
   cbDataSet = new TGComboBox(cf, CB_DataSet);
   cbDataSet->Select(-1);
   cbDataSet->Resize(150, 20);
   cbDataSet->Connect("Selected(Int_t)",
                      "KVDataAnalysisLauncher",
                      this,
                      "SelectedDataSet(Int_t)");
   cf->AddFrame(cbDataSet, LHtopleft);
   cfSelect->AddFrame(cf, centerX);
 
   cf = new TGCompositeFrame(cfSelect, fMainGuiWidth, 350, kHorizontalFrame);
   // Label du Task
   lab = new TGLabel(cf, "ANALYSIS TASK : ");
   lab->SetTextJustify(justMode);
   lab->Resize(150, 20);
   cf->AddFrame(lab, new TGLayoutHints(kLHintsLeft | kLHintsTop,
                                       1, 1, 1, 1));
   // ComboBox du Task
   cbTask = new TGComboBox(cf, CB_AnalysisTask);
   cbTask->Select(-1);
   cbTask->Resize(350, 20);
   cbTask->Connect("Selected(int)",
                   "KVDataAnalysisLauncher",
                   this,
                   "SetSystemList(int)");
   cf->AddFrame(cbTask, LHtopleft);
   cfSelect->AddFrame(cf, centerX);
 
   AddFrame(cfSelect, new TGLayoutHints(kLHintsLeft | kLHintsTop
                                        | kLHintsRight | kLHintsExpandX,
                                        10, 10, 1, 1));
   // Systems list
   lvSystems = new KVListView(KVDBSystem::Class(), this, fMainGuiWidth, 250);
   lvSystems->SetDataColumns(6);
   lvSystems->SetMaxColumnSize(gEnv->GetValue("KaliVedaGUI.MaxColWidth", 200));
   lvSystems->SetDataColumn(1, "Zproj");
   lvSystems->SetDataColumn(2, "Ztarget");
   lvSystems->SetDataColumn(3, "Ebeam");
   lvSystems->GetDataColumn(3)->SetDataFormat("%4.1lf");
   lvSystems->SetDataColumn(4, "Runs", "GetNumberRuns");
   lvSystems->SetDataColumn(5, "Events");
   lvSystems->SetDataColumn(0, "System", "GetName");
   lvSystems->ActivateSortButtons();
   // disable context menu, Browse & multi-select functions
   lvSystems->AllowBrowse(kFALSE);
   lvSystems->AllowContextMenu(kFALSE);
   lvSystems->AllowMultipleSelection(kTRUE);
   lvSystems->Connect("SelectionChanged()", "KVDataAnalysisLauncher", this, "SystemSelectionChanged()");
   AddFrame(lvSystems, new TGLayoutHints(kLHintsExpandX,
                                         10, 10, 15, 15));
 
   // Frame pour la liste des runs
 
   TGCompositeFrame* cfRuns = new TGCompositeFrame(this, fMainGuiWidth, 400, kVerticalFrame);
   lvRuns = new KVListView(KVRunFile::Class(), cfRuns, fMainGuiWidth, 300);
   lvRuns->SetDataColumns(8);
   lvRuns->SetMaxColumnSize(gEnv->GetValue("KaliVedaGUI.MaxColWidth", 200));
   int iicc = 0;
   lvRuns->SetDataColumn(iicc++, "Run/Index", "GetRunIndexString");
   lvRuns->SetDataColumn(iicc++, "Events", "", kTextRight);
   lvRuns->SetDataColumn(iicc++, "File", "GetName");
   lvRuns->SetDataColumn(iicc, "Date", "GetFileWrittenDatime");
   lvRuns->GetDataColumn(iicc++)->SetIsDateTime();
   lvRuns->SetDataColumn(iicc++, "Comments", "", kTextLeft);
   lvRuns->SetDataColumn(iicc++, "Trigger", "");
   lvRuns->SetDataColumn(iicc++, "Version");
   lvRuns->SetDataColumn(iicc++, "User");
   lvRuns->ActivateSortButtons();
   // disable context menu & Browse functions
   lvRuns->AllowBrowse(kFALSE);
   lvRuns->AllowContextMenu(kFALSE);
   lvRuns->Connect("SelectionChanged()", "KVDataAnalysisLauncher", this, "UpdateListOfSelectedRuns()");
   cfRuns->AddFrame(lvRuns, new TGLayoutHints(kLHintsLeft | kLHintsTop |
                    kLHintsExpandX | kLHintsExpandY,
                    10, 10, 15, 15));
 
   // Boutons de selection
   TGCompositeFrame* cfSelAll = new TGCompositeFrame(cfRuns, fMainGuiWidth, 20, kHorizontalFrame);
   auto BrefreshDir = new TGPictureButton(cfSelAll, gClient->GetPicture("refresh2.xpm"));
   BrefreshDir->Resize(20, 20);
   cfSelAll->AddFrame(BrefreshDir, eX);
   BrefreshDir->SetToolTipText("Update available runfiles for this datatype");
   BrefreshDir->Connect("Clicked()", "KVDataAnalysisLauncher", this, "UpdateAvailableRuns()");
   TGTextButton* bout = new TGTextButton(cfSelAll, "Select All");
   bout->SetToolTipText("Select all runs for the analysis.", TTDELAY);
   bout->Connect("Clicked()",
                 "KVDataAnalysisLauncher",
                 this,
                 "SelectAll()");
   cfSelAll->AddFrame(bout, eX);
   bout = new TGTextButton(cfSelAll, "Deselect All");
   bout->SetToolTipText("Deselect all runs.", TTDELAY);
   bout->Connect("Clicked()",
                 "KVDataAnalysisLauncher",
                 this,
                 "DeselectAll()");
   cfSelAll->AddFrame(bout, eX);
   bout = new TGTextButton(cfSelAll, "Runlist");
   bout->SetToolTipText("Enter list of runs to analyse.", TTDELAY);
   bout->Connect("Clicked()",
                 "KVDataAnalysisLauncher",
                 this,
                 "EnterRunlist()");
   cfSelAll->AddFrame(bout, eX);
   bStageSelect = new TGTextButton(cfSelAll, "Stage selection");
   bStageSelect->SetToolTipText("Select all unconverted runs", TTDELAY);
   // bStageSelect->Connect("Clicked()",
   //                       "KVDataAnalysisLauncher",
   //                       this,
   //                       "SelectAllUnconvertedRuns()");
   bStageSelect->Connect("Clicked()",
                         "KVDataAnalysisLauncher",
                         this,
                         "DoStageSelection()");
   bStageSelect->SetEnabled(false);
   cfSelAll->AddFrame(bStageSelect, eX);
   cfRuns->AddFrame(cfSelAll, eX);
 
   TGHorizontalFrame* runs_and_nbevents = new TGHorizontalFrame(cfRuns, fMainGuiWidth, 20);
   selectedRuns = new TGLabel(runs_and_nbevents, " ");
   runs_and_nbevents->AddFrame(selectedRuns, new TGLayoutHints(kLHintsExpandX | kLHintsCenterY, 2, 2, 0, 0));
   TGHorizontalFrame* bidule = new TGHorizontalFrame(runs_and_nbevents);
   TGLabel* nevents = new TGLabel(bidule, "Events : ");
   bidule->AddFrame(nevents, new TGLayoutHints(kLHintsLeft | kLHintsCenterY, 2, 2, 0, 0));
   teNbToRead = new TGNumberEntry(bidule, 0);
#ifdef __WITHOUT_TGNUMBERENTRY_SETNUMSTYLE
   teNbToRead->SetFormat(TGNumberFormat::kNESInteger, teNbToRead->GetNumAttr());
#else
   teNbToRead->SetNumStyle(TGNumberFormat::kNESInteger);
#endif
#ifdef __WITHOUT_TGNUMBERENTRY_SETNUMATTR
   teNbToRead->SetFormat(teNbToRead->GetNumStyle(), TGNumberFormat::kNEANonNegative);
#else
   teNbToRead->SetNumAttr(TGNumberFormat::kNEANonNegative);
#endif
   teNbToRead->GetNumberEntry()->SetToolTipText("Number of events to read [0 => all events]", TTDELAY);
   teNbToRead->Resize(150, 20);
   bidule->AddFrame(teNbToRead, new TGLayoutHints(kLHintsRight | kLHintsCenterY, 2, 2, 0, 0));
   runs_and_nbevents->AddFrame(bidule, new TGLayoutHints(kLHintsRight, 2, 2, 0, 0));
   cfRuns->AddFrame(runs_and_nbevents, eX);
 
   AddFrame(cfRuns, eXeY);
 
   // UserClass
   cfAnalysis = new TGCompositeFrame(this, fMainGuiWidth, 20, kVerticalFrame);
   cf = new TGCompositeFrame(cfAnalysis, fMainGuiWidth, 20, kHorizontalFrame);
   // Label for User Class name
   fUserClassLabel = new TGLabel(cf, "User Class");
   cf->AddFrame(fUserClassLabel, eX);
   // Label du Task
   lab = new TGLabel(cf, "User class options");
   cf->AddFrame(lab, eX);
 
   cfAnalysis->AddFrame(cf, eX);
 
   cf = new TGCompositeFrame(cfAnalysis, fMainGuiWidth, 20, kHorizontalFrame);
   cbUserClass = new TGComboBox(cf);
   cbUserClass->Select(-1);
   cbUserClass->Resize(150, 20);
   cf->AddFrame(cbUserClass, eX);
   cbUserClass->Connect("Selected(Int_t)", "KVDataAnalysisLauncher", this,
                        "SelectedUserClass(Int_t)");
   btEditClass = new TGPictureButton(cf, "query_new.xpm");
   btEditClass->SetEnabled(kFALSE);
   btEditClass->Connect("Clicked()", "KVDataAnalysisLauncher", this, "EditUserClassFiles()");
   btEditClass->SetToolTipText(Form("Open analysis class source files in %s", gSystem->Getenv("EDITOR")), TTDELAY);
   cf->AddFrame(btEditClass, new TGLayoutHints(kLHintsTop | kLHintsLeft, 2, 2, 2, 2));
 
   teUserOptions = new TGTextEntry(cf, "");
   teUserOptions->SetToolTipText("Comma-separated list of options for user analysis class: PAR1=VAL1,PAR2=VAL2,etc.", TTDELAY);
   cf->AddFrame(teUserOptions, eX);
 
   cfAnalysis->AddFrame(cf, eX);
 
   AddFrame(cfAnalysis, eX);
 
   cf = new TGCompositeFrame(this, fMainGuiWidth, 20, kHorizontalFrame);
   // Frame for the user's libraries
   bout = new TGTextButton(cf, "User's libraries", B_Libs);
   bout->Connect("Clicked()",
                 "KVDataAnalysisLauncher",
                 this,
                 "SetUserLibraries()");
   cf->AddFrame(bout, eX);
 
   bout = new TGTextButton(cf, "User's includes", B_Incs);
   bout->Connect("Clicked()",
                 "KVDataAnalysisLauncher",
                 this,
                 "SetUserIncludes()");
   cf->AddFrame(bout, eX);
 
   this->AddFrame(cf, eX);
 
   // Process et Quit
#ifdef KVDAL_DEBUG
   cout << "Creation Process/Quit" << endl;
#endif
   TGCompositeFrame* cfProcess = new TGCompositeFrame(this, fMainGuiWidth, 20, kHorizontalFrame);
   withBatch = new TGTextButton(cfProcess, "BatchMode");
   withBatch->SetToolTipText(gBatchSystem->GetTitle());
   withBatch->AllowStayDown(kTRUE);
   withBatch->Connect("Clicked()", "KVDataAnalysisLauncher", this, "SetBatch()");
   cfProcess->AddFrame(withBatch, eX);
   doBatchParams = new TGTextButton(cfProcess, "Batch Parameters");
   doBatchParams->SetToolTipText("Set parameters of batch jobs");
   doBatchParams->Connect("Clicked()", "KVDataAnalysisLauncher", this, "SetBatchParameters()");
   cfProcess->AddFrame(doBatchParams, eX);
   // Bouton de process
   bout = new TGTextButton(cfProcess, "&Process", B_Process);
   bout->SetToolTipText("Run the analysis.", TTDELAY);
   bout->Connect("Clicked()", "KVDataAnalysisLauncher", this, "Process()");
   // bout->Associate(this);
   cfProcess->AddFrame(bout, eX);
   // Bouton de sortie
   bout = new TGTextButton(cfProcess, "&Quit", B_Quit);
   bout->SetToolTipText("Close GUI and quit.", TTDELAY);
   bout->Connect("Clicked()", "KVDataAnalysisLauncher", this, "Exit()");
   // bout->Associate(this);
   cfProcess->AddFrame(bout, eX);
 
   this->AddFrame(cfProcess, eX);
   // On affiche tout le monde maintenant
   MapSubwindows();
 
   Resize(GetDefaultSize());
 
   SetWindowName(Form("KaliVeda Analysis Launcher on %s", gSystem->HostName()));
   SetIconName(Form("KaliVeda Analysis Launcher on %s", gSystem->HostName()));
 
   MapWindow();
   SetWMSize(fMainGuiWidth, fMainGuiHeight);
 
   SetRepositoryList();
 
   FillListOfUserClasses();
   //fill drop down list of user classes in working directory
   SetUserClassList();
   checkCompilation = kFALSE;
 
   // Reset last known state of interface
   TString tmp(GetResource("Repository", ""));
   SetRepository(tmp.Data());
 
   if (GUIenv->GetValue("KVDataAnalysisLauncher.Batch", kFALSE))
      withBatch->SetState(kButtonDown);
   else
      withBatch->SetState(kButtonUp);
   SetBatch();
 
   fUserLibraries = GUIenv->GetValue("KVDataAnalysisLauncher.UserLibraries", "");
   fUserIncludes = GUIenv->GetValue("KVDataAnalysisLauncher.UserIncludes", "");
 
}
 
 
 
 
KVDataAnalysisLauncher::~KVDataAnalysisLauncher()
{
   // Destructeur
   if (ia) delete ia;
   if (GUIenv) delete GUIenv;
   delete UserClassNames;
}
 
 
 
 
void KVDataAnalysisLauncher::SetRepositoryList(void)
{
   // Sets the list of all possible repositories in the repository combo box
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
   RemoveAll(cbRepository);
#else
   cbRepository->RemoveAll();
#endif
   cbRepository->Select(-1);
   TIter next((TList*)gDataRepositoryManager->GetListOfRepositories());
   TObject* o = 0;
   Int_t i = 0;
   while ((o = next())) {
      cbRepository->AddEntry(o->GetName(), i);
      i++;
   }
 
   cbRepository->Layout();
 
}
 
 
 
 
void KVDataAnalysisLauncher::SelectedRepository(Int_t)
{
   // Called when repository is selected in drop-down list
   //
   // Calls SetDataSetList() to fill list of available datasets for given repository
 
   auto sel = cbRepository->GetSelectedEntry();
   if (sel)
      SetDataSetList(sel->GetTitle());
}
 
 
 
 
void KVDataAnalysisLauncher::SetDataSetList(const TString& repository)
{
   // Sets the list of all available data sets in the data sets combo box
   if (!gDataRepositoryManager->GetRepository(repository)) {
      Warning("SetDataSetList", "Called for undefined repository '%s'", repository.Data());
      return;
   }
 
   SetResource("Repository", repository);
 
   TString ds = GetSavedResource("DataSet", "");
 
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
   RemoveAll(cbDataSet);
#else
   cbDataSet->RemoveAll();
#endif
   cbDataSet->Select(-1);
   gDataRepositoryManager->GetRepository(repository)->cd();
   Int_t nbds = gDataSetManager->GetNavailable();
   Int_t i = 0;
   while (i < nbds) {
      cbDataSet->AddEntry(gDataSetManager->GetAvailableDataSet(i + 1)->GetName(), i);
      i++;
   }
 
   cbDataSet->Layout();
 
   if (ds.Length()) {
      SetDataSet(ds.Data());
   }
   else {
      SetTask();
   }
 
}
 
 
 
 
void KVDataAnalysisLauncher::SelectedDataSet(Int_t)
{
   // Called when dataset is selected in drop-down list
   //
   // Calls SetTaskList() to fill list of available tasks for given dataset
 
   auto sel = cbDataSet->GetSelectedEntry();
   // allow list to close & GUI to relax before e.g. building database for new dataset
   gSystem->ProcessEvents();
   if (sel)
      SetTaskList(sel->GetTitle());
}
 
 
 
 
void KVDataAnalysisLauncher::SetTaskList(const TString& dataset)
{
   // Sets the list of all possible tasks in the tasks combo box
   // Called when a new dataset is selected in the dropdown list
 
   SetResource("DataSet", dataset);
 
   TString ds = GetSavedResource("Task", "");
 
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
   RemoveAll(cbTask);
#else
   cbTask->RemoveAll();
#endif
   cbTask->Select(-1);
#ifdef KVDAL_DEBUG
   cout << "DataSet : [" << dataset << "]" <<  endl;
#endif
   current_dataset = gDataSetManager->GetDataSet(dataset);
   // don't block GUI building full database, just need systems & runs
   current_database = current_dataset->GetDataBase("minimal");
   Int_t nbt = current_dataset->GetNtasks();
 
   GetDataAnalyser()->SetDataSet(current_dataset);
 
   noSystems = (!current_database || !current_database->GetSystems()->GetSize());
   if (noSystems) lvSystems->RemoveAll();
 
   Int_t i = 0;
   while (i < nbt) {
      cbTask->AddEntry(current_dataset->GetAnalysisTask(i + 1)->GetType(), i);
      i++;
   }
 
   cbTask->Layout();
 
   if (ds.Length()) {
      SetTask(ds.Data());
   }
   else {
      SetSystem();
   }
}
 
 
 
 
KVDataSetAnalyser* KVDataAnalysisLauncher::GetDataAnalyser(KVDataAnalysisTask* task)
{
   //Get analyser for task
   //If task = 0 we return the current analyser
 
   if (!task) {
      if (!ia) ia = new KVDataSetAnalyser;
   }
   else {
      if (ia) delete ia;
      ia = (KVDataSetAnalyser*)KVDataAnalyser::GetAnalyser(task->GetDataAnalyser());
      if (!ia) ia = new KVDataSetAnalyser;
      ia->SetAnalysisTask(task);
   }
   return ia;
}
 
 
 
 
void KVDataAnalysisLauncher::SetSystemList(Int_t itask)
{
   // Sets the list of all possible systems in the system list.
   // Called every time a task is selected.
   //
   // The choice of task may lead to a change of batch system:
   //  - if the current system is PROOFLite but the data to be read by the task is
   //    not in a TFile (i.e. not in a TTree) then we switch to Xterm batch system.
   //  - if the current system is Xterm but the data to be read by the task *is*
   //    in a TFile (i.e. in a TTree) then we switch to PROOFLite batch system
 
   // allow list to close & GUI to relax before e.g. building database for new dataset
   gSystem->ProcessEvents();
 
   KVDataAnalysisTask* task = current_dataset->GetAnalysisTask(itask + 1);
 
   TString current_batch = gBatchSystem->GetName();
   TString data_reader = current_dataset->GetDataSetEnv(Form("DataSet.RunFileClass.%s", task->GetPrereq()));
 
   if (current_batch == "PROOFLite" && data_reader != "TFile") {
      gBatchSystemManager->GetBatchSystem("Xterm")->cd();
      withBatch->SetToolTipText(gBatchSystem->GetTitle());
      Info("SetSystemList", "Switched batch system to %s: %s", gBatchSystem->GetName(), gBatchSystem->GetTitle());
   }
   else if (current_batch == "Xterm" && data_reader == "TFile") {
      gBatchSystemManager->GetBatchSystem("PROOFLite")->cd();
      withBatch->SetToolTipText(gBatchSystem->GetTitle());
      Info("SetSystemList", "Switched batch system to %s: %s", gBatchSystem->GetName(), gBatchSystem->GetTitle());
   }
 
   GetDataAnalyser(task);
 
   SetResource("Task", task->GetTitle());
 
   if (!task->WithUserClass()) {
      //no user class required
      DisableUserClassList();
   }
   else {
      //user class required
      EnableUserClassList();
   }
 
   source_type = task->GetPrereq();
   // conversion task? if so, enable stage selection
   if (!task->GetOutputDataType().IsNull()) {
      output_type = task->GetOutputDataType();
      bStageSelect->SetEnabled();
   }
   else {
      bStageSelect->SetEnabled(false);
   }
 
   //update display
   cfAnalysis->Layout();
 
   TString ds = GetSavedResource("System", "");
   lastSelectedSystem = 0;
   if (!noSystems) {
      auto sys_list = current_dataset->GetListOfAvailableSystems(task);
      if (!sys_list.IsEmpty()) {
         lvSystems->Display(&sys_list);
      }
   }
   if (ds.Length()) {
      SetSystem(ds.Data());
   }
   else {
      SetRunsList();
   }
}
 
 
 
 
void KVDataAnalysisLauncher::SetRunsList()
{
   // Sets the list of all available runs in the runs list box
 
//   SetResource("Trigger", "All");
   listOfRuns.Clear();
   listOfSystemRuns.Clear();
   if (entryMax > -1) {
      lvRuns->RemoveAll();
      list_of_runs.Clear();
      entryMax = -1;
   }
 
   KVDataAnalysisTask* task = current_dataset->GetAnalysisTask(cbTask->GetSelected() + 1);
   KVDBSystem* system = 0;
   if (!noSystems || !strcmp(task->GetPrereq(), "*")) {
      // case where systems are defined for dataset and user has
      // selected a specific system
      // OR for online analysis (prereq data type="*")
      system = lastSelectedSystem;
      GetDataAnalyser()->SetSystem(system);
      if (!system) {
         // no system selected
         // clear runs list
         SetRuns();
         return;
      }
   }
 
   //Setting name of system in ressources file
   if (!noSystems) {
      // dataset with defined systems
      if (system) {
         // user has chosen a system
         SetResource("System", system->GetName());
      }
      else {
         // user chose "All" for system
         SetResource("System", "All");
      }
   }
   else {
      // no systems defined for dataset
      SetResource("System", "Unknown");
   }
 
   // loop over selected systems, add all runs to run list
   lvRuns->RemoveAll();
   list_of_runs.Clear();
   run_index_list run_list;
   for (auto sys_p : listofSelectedSystems) {
      auto system = (KVDBSystem*)sys_p;
      auto rl = current_dataset->GetRunList(task->GetPrereq(), system);
      run_list.Add(rl);
      auto lor = current_dataset->GetListOfAvailableRunFilesForSystem(task, system);
      list_of_runs.AddAll(&lor);
      lor.Clear("nodelete");
   }
   entryMax = list_of_runs.GetEntries();
   listOfSystemRuns = run_list;
   lvRuns->Display(&list_of_runs);
 
   SetRuns("");
 
   // Set saved user class, number of events for current
   // repository, dataset, task, system, trigger & runs
   auto ds = GetSavedResource("UserClass", "");
   SetUserClass(ds);
   ds = GetSavedResource("NbEventsToRead", "");
   teNbToRead->SetIntNumber(ds.Atoi());
}
 
 
 
 
void KVDataAnalysisLauncher::SelectedUserClass(Int_t)
{
   // Called when user analysis class is selected in drop-down list
   //
   // Calls UserClassSelected() with name of class
 
   auto sel = cbUserClass->GetSelectedEntry();
   if (sel)
      UserClassSelected(sel->GetTitle());
}
 
 
 
 
 
void KVDataAnalysisLauncher::SelectAll(void)
{
   // Select all runs currently in the displayed list of runs
   lvRuns->SelectAll();
   UpdateListOfSelectedRuns();
}
 
 
 
 
void KVDataAnalysisLauncher::DeselectAll(void)
{
   // Deselect all runs currently in the displayed list of runs
   lvRuns->UnSelectAll();
   UpdateListOfSelectedRuns();
}
 
 
 
 
void KVDataAnalysisLauncher::Process(void)
{
   // Run the analysis task
 
   TString oriIncludePath = gSystem->GetIncludePath();
 
   KVDataAnalysisTask* task = current_dataset->GetAnalysisTask(cbTask->GetSelected() + 1);
   KVDataSetAnalyser* datan = GetDataAnalyser(task);
   bool online_analysis = !strcmp(task->GetPrereq(), "*");
 
   //set global pointer to analyser
   gDataAnalyser = datan;
 
   datan->SetDataSet(current_dataset);
   datan->SetAnalysisTask(task);
   if (listOfRuns.GetNValues()) {
      datan->SetRuns(listOfRuns, kFALSE);
      datan->SetFullRunList(listOfRuns);
   }
   else if (!online_analysis) {
      WarningBox("Empty Run List", "The list of runs to process is empty.");
      return;
   }
 
   if (fUserIncludes.Length()) {
      datan->SetUserIncludes(fUserIncludes.Data());
   }
   if (fUserLibraries.Length()) {
      datan->SetUserLibraries(fUserLibraries.Data());
   }
 
   //Need a user class for the analysis ?
   TString kvs(GetUserClass());
   if (task->WithUserClass()) {
      //read user's class name from input box
      if (kvs.Length()) {
         datan->SetUserClassOptions(teUserOptions->GetText());
         datan->SetUserClass({kvs}, checkCompilation);
         if (datan->IsUserClassValid())
            checkCompilation = kFALSE;
         else {
            // compilation failed. abort processing.
            delete ia;
            ia = 0;
            checkCompilation = kTRUE;
            if (WarningBox("Compilation failed", "Please correct mistakes in user analysis class", kTRUE)) EditUserClassFiles();
            return;
         }
      }
      else {
         delete ia;
         ia = 0;
         checkCompilation = kTRUE;
         WarningBox("No User Class", "Please enter the user analysis class name.");
         return;
      }
   }
   else if (strcmp(task->GetUserBaseClass(), "")) {
      //task with default "user" class (i.e. UserClass=no but UserClass.BaseClass!="")
      datan->SetUserClass({task->GetUserBaseClass()}, kFALSE);
   }
   Long64_t nbEventRead = (Long64_t)teNbToRead->GetIntNumber();
   // if in batch and nbEventRead>0, ask confirmation
   if (IsBatch() && nbEventRead) {
      if (!WarningBox("Read all events in batch mode?",
                      "This will submit batch jobs which will not read all events.\nAre you sure that is what you want?",
                      kTRUE)) {
         delete ia;
         ia = 0;
         return;
      }
   }
   // check batch parameters have been set
   if (IsBatch()
         && (!fBatchParameters.GetNpar() // never set
             // previously set with MultiJobsMode turned off (perhaps to read a single run), whereas we now have many runs to read
             || (fBatchParameters.HasBoolParameter("MultiJobsMode") && !fBatchParameters.GetBoolValue("MultiJobsMode") && listOfRuns.GetNValues() > 1))
      ) {
      if (!SetBatchParameters()) return; //abort analysis if user pressed cancel
   }
   datan->SetNbEventToRead((Long64_t)teNbToRead->GetIntNumber());
   SetResource("RunsList", listOfRuns.AsString());
   SetResource("UserClassOptions", teUserOptions->GetText());
   SetResource("NbEventsToRead", Form("%.0f", teNbToRead->GetNumber()));
   GUIenv->SaveLevel(kEnvUser);
 
   // before submitting the job(s), we make sure that the database for the current dataset
   // has been fully built, we don't want each job trying to do it (especially for batch/PROOFlite jobs)
   current_dataset->cd();
 
   // submit jobs for each selected system
   for (auto it : listofSelectedSystems) {
      auto sys = (KVDBSystem*)it;
 
      // build up runfile list for system
      run_index_list sys_run_list;
      for (auto it_r : sys->GetRuns()) {
         // keep all system runfiles in user's selection
         auto runfiles_for_run = dynamic_cast<KVDBRun*>(it_r)->GetRunIndexList();
         sys_run_list += runfiles_for_run && listOfRuns;
      }
      if (sys_run_list) {
         if (IsBatch()) {
            gBatchSystem->Clear();
            gBatchSystem->SetBatchSystemParameters(fBatchParameters);
            datan->SetBatchSystem(gBatchSystem);
         }
         else {
            datan->SetBatchSystem(nullptr);
         }
         datan->SetRuns(sys_run_list, false);
         datan->Run();
      }
   }
 
   gSystem->SetIncludePath(oriIncludePath.Data());
}
 
 
 
 
TString KVDataAnalysisLauncher::GetRepository(void)
{
   if (TGTextLBEntry* e = (TGTextLBEntry*)cbRepository->GetSelectedEntry()) {
      return e->GetText()->GetString();
   }
   else {
      return "";
   }
}
 
 
 
 
TString KVDataAnalysisLauncher::GetDataSet(void)
{
   if (TGTextLBEntry* e = (TGTextLBEntry*)cbDataSet->GetSelectedEntry()) {
      return e->GetText()->GetString();
   }
   else {
      return "";
   }
}
 
 
 
 
TString KVDataAnalysisLauncher::GetTask(void)
{
   if (TGTextLBEntry* e = (TGTextLBEntry*)cbTask->GetSelectedEntry()) {
      return e->GetText()->GetString();
   }
   else {
      return "";
   }
}
 
 
 
 
TString KVDataAnalysisLauncher::GetSystem(void)
{
   if (noSystems) return "";
   KVDBSystem* sys = (KVDBSystem*)lvSystems->GetLastSelectedObject();
   if (sys)
      return sys->GetName();
   else
      return "";
}
 
 
 
 
 
TString KVDataAnalysisLauncher::GetRuns(void)
{
   return listOfRuns.AsString();
}
 
 
 
 
void KVDataAnalysisLauncher::SetRepository(const TString& r)
{
   TGLBEntry* e = 0;
   if ((e = cbRepository->FindEntry(r))) {
      Int_t i = e->EntryId();
#ifdef __WITHOUT_TGCOMBOBOX_SELECT_BOOL_T
      cbRepository->Select(i);
#else
      cbRepository->Select(i, kFALSE);
#endif
      SetDataSetList(r);
   }
   else {
      SetDataSet();
   }
}
 
 
 
 
void KVDataAnalysisLauncher::SetDataSet(const TString& r)
{
   if (r.IsNull()) {
      //remove all datasets because no repository has been chosen yet
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
      RemoveAll(cbDataSet);
#else
      cbDataSet->RemoveAll();
#endif
      cbDataSet->Select(-1);
      SetResource("Repository", "");
      SetTask();
   }
   else {
      TGLBEntry* e = 0;
      if ((e = cbDataSet->FindEntry(r))) {
         Int_t i = e->EntryId();
#ifdef __WITHOUT_TGCOMBOBOX_SELECT_BOOL_T
         cbDataSet->Select(i);
#else
         cbDataSet->Select(i, kFALSE);
#endif
         SetTaskList(r);
      }
      else {
         SetTask();
      }
   }
}
 
 
 
 
void KVDataAnalysisLauncher::SetTask(const TString& r)
{
   if (r.IsNull()) {
      //remove all tasks from list because no dataset chosen yet
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
      RemoveAll(cbTask);
#else
      cbTask->RemoveAll();
#endif
      cbTask->Select(-1);
      SetResource("DataSet", "");
      SetSystem();
   }
   else {
      TGLBEntry* e = 0;
      if ((e = cbTask->FindEntry(r))) {
         Int_t i = e->EntryId();
#ifdef __WITHOUT_TGCOMBOBOX_SELECT_BOOL_T
         cbTask->Select(i);
#else
         cbTask->Select(i, kFALSE);
#endif
         SetSystemList(i);
      }
      else {
         SetSystem();
      }
   }
}
 
 
 
 
void KVDataAnalysisLauncher::SetSystem(const TString& r)
{
   if (r.IsNull()) {
      //remove all systems from list because no task chosen yet
      lvSystems->RemoveAll();
      SetResource("Task", "");
      //empty list of analysis classes and disable it
      DisableUserClassList();
      SetRuns();
      lvRuns->RemoveAll();
      list_of_runs.Clear();
   }
   else {
      lvSystems->ActivateItemWithColumnData("System", r);
      SystemSelectionChanged();
   }
}
 
 
 
 
void KVDataAnalysisLauncher::SystemSelectionChanged()
{
   //KVDBSystem* system = (KVDBSystem*)lvSystems->GetLastSelectedObject();
   //if (system == lastSelectedSystem) return;
   //lastSelectedSystem = system;
   //GetDataAnalyser()->SetSystem(system);
   listofSelectedSystems = lvSystems->GetSelectedObjects();
   if (listofSelectedSystems.IsEmpty())
      lastSelectedSystem = nullptr;
   else
      lastSelectedSystem = (KVDBSystem*)lvSystems->GetLastSelectedObject();
   SetRunsList();
}
 
 
 
 
void KVDataAnalysisLauncher::SetRuns(const TString& r)
{
   if (r.IsNull()) {
      ClearListOfSelectedRuns();
      SetResource("Trigger", "All");
      SetResource("RunsList", "");
   }
   else {
      listOfRuns = current_database->SetRunIndexListFromString(r);
      if (listOfRuns.GetNValues()) {
         lvRuns->ActivateItemsWithColumnData("Run/Index", listOfRuns);
         SetResource("RunsList", listOfRuns.AsString());
      }
      UpdateListOfSelectedRuns();
   }
}
 
 
 
 
void KVDataAnalysisLauncher::SetBatch()
{
   // Set the resource KVDataAnalysisLauncher.Batch according
   // to whether button 'Batch' is down or up
 
   if (IsBatch()) withBatch->SetText("BatchMode: On");
   else withBatch->SetText("BatchMode: Off");
   GUIenv->SetValue("KVDataAnalysisLauncher.Batch", IsBatch());
   doBatchParams->SetEnabled(IsBatch());
   GUIenv->SaveLevel(kEnvUser);
}
 
 
 
 
Bool_t KVDataAnalysisLauncher::SetBatchParameters()
{
   // Open dialog to set batch parameters for job
   // returns kFALSE if cancel is pressed
 
   gBatchSystem->GetBatchSystemParameterList(fBatchParameters);
   // use saved values of batch parameters
   fBatchParameters.SetFromEnv(GUIenv, "KVDataAnalysisLauncher");
   Bool_t cancel;
   // make sure runlist is set in analyser (controls multijobs mode)
   GetDataAnalyser()->SetDataSet(current_dataset);
   GetDataAnalyser()->SetRuns(listOfRuns, kFALSE);
   new KVBatchSystemParametersGUI(this, &fBatchParameters, GetDataAnalyser(), &cancel);
   if (!cancel) {
      // update saved batch parameter resources
      fBatchParameters.WriteToEnv(GUIenv, "KVDataAnalysisLauncher");
      GUIenv->SaveLevel(kEnvUser);
   }
   return !cancel;
}
 
 
 
 
 
void KVDataAnalysisLauncher::SetUserLibraries(void)
{
   // Set the User's libraries
   TString ori = fUserLibraries.Data();
   new KVGFileList(fUserLibraries, "User's Libraries",
                   gClient->GetRoot(), this);
   GUIenv->SetValue("KVDataAnalysisLauncher.UserLibraries", fUserLibraries.Data());
   GUIenv->SaveLevel(kEnvUser);
   checkCompilation = ori.CompareTo(fUserLibraries);
}
 
 
 
 
void KVDataAnalysisLauncher::SetUserIncludes(void)
{
   // Set the User's includes
   TString ori = fUserIncludes.Data();
   new KVGDirectoryList(fUserIncludes, "User's Includes",
                        gClient->GetRoot(), this);
   GUIenv->SetValue("KVDataAnalysisLauncher.UserIncludes", fUserIncludes.Data());
   GUIenv->SaveLevel(kEnvUser);
   checkCompilation = ori.CompareTo(fUserIncludes);
}
 
 
 
 
 
Bool_t KVDataAnalysisLauncher::WarningBox(const char* title, const char* msg, Bool_t confirm)
{
   // Warning box in case of problems
   // if confirm=kTRUE we ask for a yes/no answer from the user:
   //     if 'yes' is pressed, we return kTRUE, if 'no', kFALSE.
   // by default, only a 'dismiss' button is shown, and this method always returns kTRUE.
 
   Bool_t reply = kTRUE;
   if (!confirm)
      new TGMsgBox(gClient->GetRoot(), this, title, msg, kMBIconExclamation);
   else {
      Int_t ret_code = 0;
      new TGMsgBox(gClient->GetRoot(), this, title, msg, kMBIconExclamation, kMBYes | kMBNo, &ret_code);
      reply = (ret_code & kMBYes);
   }
   return reply;
}
 
 
 
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
 
 
void KVDataAnalysisLauncher::RemoveAll(TGComboBox* box)
{
   //replaces functionality of TGComboBox::RemoveAll for ROOT versions < 5.11/02
 
   Int_t n = box->GetListBox()->GetNumberOfEntries();
   if (n) box->RemoveEntries(0, n - 1);
   if (box->GetSelectedEntry()) {
      ((TGTextLBEntry*)box->GetSelectedEntry())->SetTitle("");
      fClient->NeedRedraw(box->GetSelectedEntry());
   }
   else {
      box->GetTextEntry()->SetTitle("");
      fClient->NeedRedraw(box->GetTextEntry());
   }
}
 
void KVDataAnalysisLauncher::RemoveAll(TGListBox* box)
 
 
{
   //replaces functionality of TGListBox::RemoveAll for ROOT versions < 5.11/02
 
   Int_t n = box->GetNumberOfEntries();
   if (n) box->RemoveEntries(0, n - 1);
   if (box->GetSelectedEntry()) {
      ((TGTextLBEntry*)box->GetSelectedEntry())->SetTitle("");
      fClient->NeedRedraw(box->GetSelectedEntry());
   }
}
 
#endif
 
 
 
 
void KVDataAnalysisLauncher::SetResource(const TString& name, const TString& value)
{
   // Handles resource file ".KVDataAnalysisGUIrc"
   // We store the current state of the interface using the following resource names:
   //
   // Repository              KVDataAnalysisLauncher.Repository
   // DataSet                   KVDataAnalysisLauncher.DataSet
   // Task                         KVDataAnalysisLauncher.Task
   // System                    KVDataAnalysisLauncher.System
   // Trigger                    KVDataAnalysisLauncher.Trigger
   // RunsList                  KVDataAnalysisLauncher.RunsList
   // UserClass               KVDataAnalysisLauncher.UserClass
   // KVDataSelector      KVDataAnalysisLauncher.KVDataSelector
   // NbEventsToRead    KVDataAnalysisLauncher.NbEventsToRead
   //
   // We also keep a "memory" of all selected configurations using the following
   // resource names:
   //
   // DataSet                   KVDataAnalysisLauncher.DataSet.[repository]
   // Task                         KVDataAnalysisLauncher.Task.[repository].[dataset]
   // System                    KVDataAnalysisLauncher.System.[repository].[dataset].[task]
   // Trigger                    KVDataAnalysisLauncher.Trigger.[repository].[dataset].[task].[system]
   // RunsList                  KVDataAnalysisLauncher.RunsList.[repository].[dataset].[task].[system].[trigger]
   // UserClass               KVDataAnalysisLauncher.UserClass.[repository].[dataset].[task].[system].[trigger]
   // KVDataSelector      KVDataAnalysisLauncher.KVDataSelector.[repository].[dataset].[task].[system].[trigger]
   // NbEventsToRead    KVDataAnalysisLauncher.NbEventsToRead.[repository].[dataset].[task].[system].[trigger]
   //
   // N.B. [task]:  it is the NAME of the task which is used in the resource name.
   // N.B.2. [system]:  it is the name given by SystemBatchName() which is used in the resource name.
   // N.B.3. [trigger]:  for the resource name we take "M > 4", "M>=8" "All" etc. and replace all ' ', '>' or '=' by ''
   //                          to give "M4", "M8", "All", etc.
 
   TString res, ful_res;
 
   BuildResourceName(name, res, ful_res);
 
   // save current value of resource
   GUIenv->SetValue(res.Data(), value);
 
   // save resource for future use if possible
   if (ful_res.Length()) GUIenv->SetValue(ful_res.Data(), value);
 
   GUIenv->SaveLevel(kEnvUser);
}
 
 
 
 
 
TString KVDataAnalysisLauncher::GetResource(const TString& name, const TString& defaultvalue)
{
   // Handles resource file ".KVDataAnalysisGUIrc"
   //
   // We return the current value of the resource "name"
 
   TString res = name;
   res.Prepend("KVDataAnalysisLauncher.");
   return GUIenv->GetValue(res.Data(), defaultvalue);
}
 
 
 
 
 
TString KVDataAnalysisLauncher::GetSavedResource(const TString& name, const TString& defaultvalue)
{
   // Handles resource file ".KVDataAnalysisGUIrc"
   //
   // We look for a stored value of the resource "name" corresponding to the current
   // values of all the resources which come before "name" in the list :
   //    Repository
   //    DataSet
   //    Task
   // These values are stored in resources with names like:
   //    KVDataAnalysisLauncher.[name].[repository].[dataset]...
   //
   // If no stored value is found, the defaultvalue is returned
 
   TString res, ful_res;
 
   BuildResourceName(name, res, ful_res);
 
   if (!ful_res.Length()) {
      return GUIenv->GetValue(res.Data(), defaultvalue);
   }
 
   return GUIenv->GetValue(ful_res.Data(), defaultvalue);
}
 
 
 
 
 
TString KVDataAnalysisLauncher::SystemBatchName(void)
{
   // Get the system name for the batch name
 
   TString tmp = GetSystem();
   if (tmp == "All") return GetSystem();
   return GetDataAnalyser()->SystemBatchName();
}
 
 
 
 
 
void KVDataAnalysisLauncher::BuildResourceName(const TString& name, TString& cur_res, TString& saved_res)
{
   // Build the full resource names for storing DataSet, Task, etc.
   //
   // We store the current state of the interface using the following resource names:
   //
   //  "name"                       "cur_res"
   // Repository              KVDataAnalysisLauncher.Repository
   // DataSet                   KVDataAnalysisLauncher.DataSet
   // Task                         KVDataAnalysisLauncher.Task
   // UserClass               KVDataAnalysisLauncher.UserClass
   // UserClassOptions        KVDataAnalysisLauncher.UserClassOptions
   // NbEventsToRead    KVDataAnalysisLauncher.NbEventsToRead
   //
   // We also keep a "memory" of all selected configurations using the following
   // resource names:
   //
   //  "name"                       "saved_res"
   // DataSet                   KVDataAnalysisLauncher.DataSet.[repository]
   // Task                         KVDataAnalysisLauncher.Task.[repository].[dataset]
   //
   // N.B. [task]:  it is the NAME of the task which is used in the resource name.
   //
   // If name = "Repository", saved_res="" as we do not save it.
   // If the resource cannot be saved because one of the resources that is needed to form the
   // full resource name has not been set, saved_res="".
 
   //Resource name for current value
   cur_res = name;
   cur_res.Prepend("KVDataAnalysisLauncher.");
 
   //Build full name for save
   Int_t index = -1;
   saved_res = "";
   // look for resource name in list ResourceNames
   TObject* resource = 0;
   if ((resource = ResourceNames.FindObject(name))) {
      //get index in resource table
      index = ResourceNames.IndexOf(resource);
   }
 
   if (index == 0) { // resource name = "Repository"; nothing more to do
      return;
   }
 
   if (index == -1)  index = NbResNames;   // force loop to end of list
 
   // resource name is written in format KVDataAnalysisLauncher.[name].[repository]....
   // where the suffixed resource values are all those in the list before the named resource
   // i.e. for name = "Task" we write the resource KVDataAnalysisLauncher.Task.[repository].[dataset]
 
   saved_res = cur_res;
   TIter next_res(&ResourceNames);
   Int_t i = 0;
   Bool_t ok = kTRUE;
 
   while ((resource = next_res()) && (i++ < index)) {
 
      TString tmp(GetResource(resource->GetName()));
      TString res;
      if (tmp == "") {
         // one of required resources is not set - none of following resources will be set either
         // we cannot save this resource
         ok = kFALSE;
         break;
      }
      if (!strcmp(resource->GetName(), "Task") && current_dataset) {
         // translate title to name for task
         KVDataAnalysisTask* tsk = current_dataset->GetAnalysisTask(tmp.Data());
         if (tsk) res.Form(".%s", tsk->GetName());
      }
      else {
         res = GetResource(resource->GetName()).Prepend(".");
      }
      saved_res += res;
   }
   if (name == "UserClassOptions") {
      if (GetResource("UserClass", "").Length())
         saved_res += GetResource("UserClass", "").Prepend(".");
      else ok = kFALSE;
   }
 
   if (!ok) saved_res = "";
}
 
 
 
 
 
void KVDataAnalysisLauncher::FillListOfUserClasses()
{
   // Look at files in working directory & deduce list of user analysis classes.
   // We look for any file ending in '.h'. If we can find a corresponding '.cpp' or '.C' or '.cxx',
   // we consider that it is a user analysis class. This list is used to fill the "User Class"
   // drop-down list.
   // We add "[NEW]" at the end of the list: if selected, this will generate a new user analysis
   // class for the currently selected data & analysis task
 
   TSystemDirectory dir("LocDir", ".");
   unique_ptr<TList> lf(dir.GetListOfFiles());
   if (!lf.get()) return;
   UserClassNames->Clear();
   //loop over file names
   TIter next(lf.get());
   while (TObject* file = next()) {
 
      // fill list with all '.h' files
      TString tmp(file->GetName());
      if (tmp.EndsWith(".h")) {
         //strip '.h' from filename
         tmp.Remove(tmp.Index(".h"));
         UserClassNames->Add(new TNamed(tmp.Data(), tmp.Data()));
      }
 
   }
 
   // now check that implementation files exist for all '.h' we found
   TIter next_cl(UserClassNames);
   KVString imp, dec;
   while (TNamed* clh = (TNamed*)next_cl()) {
      if (!KVBase::FindClassSourceFiles(clh->GetName(), imp, dec)) clh->SetName("_INVALID_");
   }
   // remove all invalid class names
   while (TObject* obj = UserClassNames->FindObject("_INVALID_")) {
      UserClassNames->Remove(obj);
      delete obj;
   }
   // add [NEW] to list
   UserClassNames->Add(new TNamed("[NEW]", "[NEW]"));
}
 
 
 
 
 
void KVDataAnalysisLauncher::SetUserClassList()
{
   // Sets the list of all available user classes in the drop down list
 
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
   RemoveAll(cbUserClass);
#else
   cbUserClass->RemoveAll();
#endif
   cbUserClass->Select(-1);
 
   Int_t nbcl = UserClassNames->GetEntries();
   Int_t i = 0;
   cbUserClass->AddEntry("", i++);
   while (i < nbcl + 1) {
      cbUserClass->AddEntry(UserClassNames->At(i - 1)->GetName(), i);
      i++;
   }
   cbUserClass->Layout();
}
 
 
 
 
 
void KVDataAnalysisLauncher::UserClassSelected(const TString& class_name)
{
   // Called when a user class is selected in the combo box.
   // Updates batch name if 'auto batch name' is selected.
 
   if (class_name == "[NEW]") {
      GenerateNewUserClass();
      return;
   }
 
   // save resource
   SetResource("UserClassOptions", teUserOptions->GetText());
   SetResource("UserClass", class_name);
   teUserOptions->SetText(GetSavedResource("UserClassOptions", ""));
   SetResource("UserClassOptions", teUserOptions->GetText());
   if (strcmp("", class_name)) {
      btEditClass->SetEnabled(kTRUE);
      GetDataAnalyser()->SetUserClass(KVDataAnalyser::UserClass{class_name}, kFALSE);
   }
   else btEditClass->SetEnabled(kFALSE);
   checkCompilation = kTRUE;
}
 
 
 
 
void KVDataAnalysisLauncher::GenerateNewUserClass()
{
   // called when user selects [NEW] in user class list
   // we generate a new analysis class for currently selected data & task
   // the source files are opened in the $EDITOR
   // the new class is selected for the analysis
 
   // Get name of new class
   TString classname;
   Bool_t ok;
   new KVInputDialog(this, "Enter name of new analysis class", &classname, &ok, "Enter name of new analysis class");
   // check new classname is not name of existing class
   KVString impfile, decfile;
   if (KVBase::FindClassSourceFiles(classname, impfile, decfile)) {
      ok = ok && WarningBox("Replacing existing class",
                            Form("%s is the name of an existing class defined in [%s,%s].\nDo you want to overwrite this class?\n(All existing code will be lost)",
                                 classname.Data(), decfile.Data(), impfile.Data()),
                            kTRUE);
   }
   if (ok) {
      // user class generation depends on gDataSet being set
      gDataSet=current_dataset;
      current_dataset->MakeAnalysisClass(GetTask(), classname);
      FillListOfUserClasses();
   }
   SetUserClassList();
   if (ok) {
      SetUserClass(classname);
      EditUserClassFiles();
   }
}
 
 
 
 
 
void KVDataAnalysisLauncher::SetUserClass(const Char_t* class_name)
{
   // Sets selected user class in combo box according to e.g. a previously stored resource value.
   // We update the resource corresponding to the current state of the interface.
 
   // look for user class in list
   TGLBEntry* e = cbUserClass->FindEntry(class_name);
 
   if (e) {
      Int_t i = e->EntryId();
#ifdef __WITHOUT_TGCOMBOBOX_SELECT_BOOL_T
      cbUserClass->Select(i);
#else
      cbUserClass->Select(i, kFALSE);
#endif
      // save current user class options
      // save current user class
      SetResource("UserClass", class_name);
      teUserOptions->SetText(GetSavedResource("UserClassOptions", ""));
      btEditClass->SetEnabled(kTRUE);
   }
 
   else
      // unknown user class
   {
      cbUserClass->Select(-1);
      SetResource("UserClass", "");
 
      btEditClass->SetEnabled(kFALSE);
   }
   checkCompilation = kTRUE;
}
 
 
 
 
 
TString KVDataAnalysisLauncher::GetUserClass()
{
   // Returns currently selected user class name
 
   if (TGTextLBEntry* e = (TGTextLBEntry*)cbUserClass->GetSelectedEntry()) {
      return e->GetText()->GetString();
   }
   else {
      return "";
   }
}
 
 
 
 
 
void KVDataAnalysisLauncher::DisableUserClassList()
{
   // Remove all entries from user class combo box & disable text entry
 
#ifdef __WITHOUT_TGCOMBOBOX_REMOVEALL
   RemoveAll(cbUserClass);
#else
   cbUserClass->RemoveAll();
#endif
   cbUserClass->Select(-1);
   cbUserClass->SetEnabled(kFALSE);
   teUserOptions->SetEnabled(kFALSE);
}
 
 
 
 
 
void KVDataAnalysisLauncher::EnableUserClassList()
{
   // Reenable user class combo box & text entry,
   // fill list with all known user classes & select the one corresponding
   // to the current environment
 
 
   SetUserClassList();
   SetUserClass(GetSavedResource("UserClass", ""));
   cbUserClass->SetEnabled(kTRUE);
   teUserOptions->SetEnabled(kTRUE);
}
 
 
 
 
void KVDataAnalysisLauncher::EditUserClassFiles()
{
   // If environment variable $EDITOR is set, and if the currently selected
   // user class has available source files, we open them in the
   // user's favourite editor
 
   TString editor = gSystem->Getenv("EDITOR");
   if (editor == "") return;
   TString uclass = GetUserClass();
   if (uclass == "") return;
   KVString imp, dec;
   if (!KVBase::FindClassSourceFiles(uclass, imp, dec)) return;
   gSystem->Exec(Form("%s %s %s &", editor.Data(), imp.Data(), dec.Data()));
}
 
 
 
 
void KVDataAnalysisLauncher::SelectAllUnconvertedRuns()
{
   // For analysis tasks which perform data conversions (raw -> recon, dst -> recon, ...)
   // clicking on this button will select all of the runs of the currently selected system(s)
   // which have not yet been converted.
 
   auto stage_sel = current_dataset->GetRunList_StageSelection(output_type, source_type, listofSelectedSystems);
   DeselectAll();
   if (stage_sel) {
      lvRuns->ActivateItemsWithColumnData("Run/Index", stage_sel);
      UpdateListOfSelectedRuns();
   }
}
 
 
 
 
void KVDataAnalysisLauncher::DoStageSelection()
{
   // Called when 'Stage selection' button is pressed, for conversion jobs
   // which generate 'target' files from 'source' files.
   //
   // Open small GUI for user to choose:
   //  - stage selection (i.e. select all 'source' files for which no 'target' file exists)
   //  - date selection (i.e. select all 'source ' files for which 'target' file is outside given date range)
 
   if (fStageSelections.IsEmpty()) {
      fStageSelections.SetValue("Missing", true);
      fStageSelections.SetValue("Date", false);
      fStageSelections.SetValue(" - earliest", KVDatime().AsSQLString());
      fStageSelections.SetValue(" -   latest", KVDatime().AsSQLString());
   }
   Bool_t cancel = false;
   auto param_gui = new KVNameValueListGUI(this, &fStageSelections, &cancel);
   param_gui->MutuallyExclusive({"Missing", "Date"});
   param_gui->EnableDependingOnBool(" - earliest", "Date");
   param_gui->EnableDependingOnBool(" -   latest", "Date");
   param_gui->DisplayDialog();
   if (!cancel) {
      if (fStageSelections.GetBoolValue("Date")) {
         std::optional<KVDatime> Dmin, Dmax;
         if (!fStageSelections.GetTStringValue(" - earliest").IsWhitespace())
            Dmin = KVDatime(fStageSelections.GetTStringValue(" - earliest"), KVDatime::kSQL);
         if (!fStageSelections.GetTStringValue(" -   latest").IsWhitespace())
            Dmax = KVDatime(fStageSelections.GetTStringValue(" -   latest"), KVDatime::kSQL);
         auto stage_sel = current_dataset->GetRunList_DateSelection(output_type, listofSelectedSystems, Dmin, Dmax);
         DeselectAll();
         if (stage_sel) {
            lvRuns->ActivateItemsWithColumnData("Run/Index", stage_sel);
            UpdateListOfSelectedRuns();
         }
      }
      else if (fStageSelections.GetBoolValue("Missing")) {
         SelectAllUnconvertedRuns();
      }
   }
}
 
 
 
 
void KVDataAnalysisLauncher::UpdateAvailableRuns()
{
   // Update available runs for current selected task (=datatype)
 
   current_dataset->UpdateAvailableRuns(source_type);
   // this will cause system list to be refilled in accordance with new available runfiles
   SetTask(GetTask());
}
 
 
 
 
 
void KVDataAnalysisLauncher::EnterRunlist()
{
   // Called when user presses "Runlist" button.
   // Open dialogue box in which a runlist can be entered.
   // The runs in the runlist will be selected.
 
   TString runs = listOfRuns.AsString();
   Bool_t ok = kFALSE;
   new KVInputDialog(this, "Enter list of runs", &runs, &ok);
   if (ok) {
      DeselectAll();
      SetRuns(runs.Data());
   }
}
 
 
 
 
 
void KVDataAnalysisLauncher::UpdateListOfSelectedRuns()
{
   // Called when the selected runs in TGListView lvRuns change.
   // We update the run_index_list listOfRuns according to the current selection
   // we modify the limits of the 'runs per job' widget
 
   listOfRuns.Clear();
   auto novolist = lvRuns->GetSelectedObjects();
   if (novolist.GetEntries() > 0) {
      TIter nxt(&novolist);
      KVRunFile* s = 0;
      while ((s = (KVRunFile*)nxt())) listOfRuns.Add(s->GetRunIndex());
   }
 
   SetResource("RunsList", listOfRuns.AsString());
   if (listOfRuns.GetNValues())
      selectedRuns->SetText(Form("%d Selected Run%s : %s",
                                 listOfRuns.GetNValues(),
                                 listOfRuns.GetNValues() > 1 ? "s" : "",
                                 listOfRuns.AsString(MAX_LENGTH_SELECTED_RUNS).Data()));
   else
      selectedRuns->SetText(" ");
   fClient->NeedRedraw(selectedRuns);
}
 
 
 
 
 
void KVDataAnalysisLauncher::ClearListOfSelectedRuns()
{
   // Empty displayed list of selected runs
   listOfRuns.Clear();
   SetResource("RunsList", "");
   selectedRuns->SetText(" ");
   fClient->NeedRedraw(selectedRuns);
}