Application.cpp

#include "TEveBrowser.h"
#include "TEveViewer.h"
#include "TGLLightSet.h"

#include "display/application/Application.h"

EDApplication::EDApplication(int argc, char **argv)
    : TApplication("GGSLeonard", &argc, argv, 0, -1), _MainWindow(0), _FileManager(0) {
  static const std::string routineName("EDApplication::EDApplication");

  gApplication = this;
  _fileLoaded = _fileToBeLoaded = "";

  _FileManager = new GGSFileManager();

  HandleArgs();
  if (args.flag_debug)
    GGSSmartLog::verboseLevel = GGSSmartLog::DEBUG; // Print only DEBUG messages or more important

  if (args.geofilename == nullptr && args.infilename == nullptr) {
    GGSCOUT(ERROR) << R"(Must provide at least one between input file (-i) and geometry file (-g))" << GGSENDL;
  }

  InitGUI();

  if (args.infilename) {
    _fileToBeLoaded = args.infilename;
    LoadRootFile();
  }

  if (_FileManager->FileIsLoaded()) {
    _MainWindow->SetFileName(_fileLoaded);
    _FileManager->ReadEvent(0);
  }
}

EDApplication::~EDApplication() {
  delete _FileManager;
  delete _MainWindow;
  // delete _GeoManager;
}

void EDApplication::NextEvent() {
  static const std::string routineName("EDApplication::NextEvent");

  _FileManager->NextEvent();
  _evtLoaded = _FileManager->GetNevent();
  _MainWindow->SetEventNumber(_evtLoaded + 1);
}

void EDApplication::PrevEvent() {
  static const std::string routineName("EDApplication::PrevEvent");

  _FileManager->PrevEvent();
  _evtLoaded = _FileManager->GetNevent();
  _MainWindow->SetEventNumber(_evtLoaded + 1);
}

void EDApplication::LoadEvent() {
  static const std::string routineName("EDApplication::LoadEvent");

  if (_evtToBeLoaded == _evtLoaded)
    return;

  _FileManager->ReadEvent(_evtToBeLoaded);
  _evtLoaded = _evtToBeLoaded;
  _MainWindow->SetEventNumber(_evtLoaded + 1);
}

void EDApplication::SetFileToBeLoaded(char *text) { _fileToBeLoaded = text; }

void EDApplication::SetEventToBeLoaded(char *text) {
  _evtToBeLoaded = atoi(text) - 1;
  if (_evtToBeLoaded < 0)
    _evtToBeLoaded = 0;
}

void EDApplication::SetDisplayDetector(char *det) {
  static const std::string routineName("EDApplication::SetDisplayDetector");

  Bool_t s = kTRUE;
  if (gTQSender) { // check that this is coming from a signal
    TGCheckButton *but = (TGCheckButton *)gTQSender;
    s = but->IsDown();
  }

  _MainWindow->SetDisplayDetector(det, s);
}

void EDApplication::SetDisplayPartHits(Bool_t s) { _MainWindow->SetDisplayPartHits(s); }

void EDApplication::LoadRootFile() {
  static const std::string routineName("EDApplication::LoadRootFile");

  if (!_fileToBeLoaded.CompareTo(_fileLoaded))
    return;

  _FileManager->Load(_fileToBeLoaded);
  _fileLoaded = _fileToBeLoaded;

  if (_MainWindow && _FileManager->FileIsLoaded()) {
    _MainWindow->SetFileName(_fileLoaded);
    _MainWindow->UpdateFileEntries();
    _MainWindow->CreateMaps();
    _FileManager->ReadEvent(0, true);
    _MainWindow->SetEventNumber(0);
  }
}

void EDApplication::HandleArgs() {

  int parse_status = argp_parse(&argp, Argc(), Argv(), 0, 0, &args);
  if (parse_status)
    exit(42);
}

void EDApplication::InitGUI() {

  TEveUtil::SetupEnvironment();
  TEveUtil::SetupGUI();
  _MainWindow = new MainWindow();
  if (!args.flag_debug) {
    _MainWindow->GetBrowser()->HideBottomTab();
  }

  LoadGeometry();
}

void EDApplication::LoadGeometry() {
  static const std::string routineName("EDApplication::EDApplication");

  const std::string geoFileName = args.geofilename ? args.geofilename : args.infilename;

#ifdef USE_TGEOGDML
  if (geoFileName.substr(geoFileName.size() - 5) == ".gdml") {
    TGeoManager::LockDefaultUnits(kFALSE);
    TGeoManager::SetDefaultUnits(TGeoManager::EDefaultUnits::kRootUnits);
  }
#endif

  if (GGSSmartLog::verboseLevel < GGSSmartLog::DEBUG) {
    GGSSmartLog::MuteOutput();
  }
  _MainWindow->GetGeometry(geoFileName);
  GGSSmartLog::UnmuteOutput();
  _GeoManager = new LeonardGeoManager();

  gGeoManager->DefaultColors();

  _MainWindow->AddGlobalElement(_GeoManager->GetEveGeoTopNode());
  _GeoManager->GetEveGeoTopNode()->ExpandIntoListTreesRecursively();

  _MainWindow->FullRedraw3D(kTRUE);

  _MainWindow->GetDefaultGLViewer()->GetClipSet()->SetClipType(TGLClip::EType(2));

  _MainWindow->GetDefaultGLViewer()->ResetCameras();
  _MainWindow->GetDefaultGLViewer()->CurrentCamera().RotateRad(-TMath::Pi() / 6, -9 * TMath::Pi() / 12);
  _MainWindow->GetDefaultGLViewer()->GetLightSet()->SetUseSpecular(false);
}

int EDApplication::ParseArgs(int key, char *arg, struct argp_state *state) {

  struct arguments *a = (struct arguments *)state->input;

  switch (key) {
  case 'd': {
    a->flag_debug = kTRUE;
    break;
  }
  case 'i': {
    a->infilename = arg;
    break;
  }
  case 'g': {
    a->geofilename = arg;
    break;
  }
  case ARGP_KEY_ARG: {
    argp_failure(state, 1, 0, "No arguments requested");
    break;
  }
  case ARGP_KEY_INIT: {
    a->infilename = NULL;
    a->geofilename = NULL;
    a->flag_debug = false;
    break;
  }
  case ARGP_KEY_END: {
    if (a->geofilename == nullptr && a->infilename == nullptr) {
      argp_failure(state, 1, 0, R"(Must provide at least one between input file (-i) and geometry file (-g))");
    }
    // if (!a->geofilename)
    //   argp_failure(
    //       state, 1, 0,
    //       "No geometry file specified \nUse -G flag to provide VGM geometry");
    // std::cout << std::endl;
    break;
  }
  }

  return 0;
}

// -- Static for cli parsing
struct argp_option EDApplication::options[] = {{"debug", 'd', 0, 0, "Enable Debug Info", 0},
                                               {"input-file", 'i', "FILE", 0, "Input file", 0},
                                               {"geometry", 'g', "FILE", 0, "Geometry file", 0},
                                               {0, 0, 0, 0, 0, 0}};
struct argp EDApplication::argp = {options, EDApplication::ParseArgs, 0, 0, 0, 0, 0};