GGSFileManager.cpp

#include "TEveBrowser.h"
#include "TEveManager.h"
#include "TGLViewer.h"
#include "TObject.h"

#include "application/Application.h"
#include "application/event/GGSFileManager.h"

GGSFileManager::GGSFileManager() : GGSTRootReader() {

  static const std::string routineName("GGSFileManager::GGSFileManager");

  _currEvent = -1;
  _fileLoaded = kFALSE;
}
GGSFileManager::~GGSFileManager() {}

void GGSFileManager::Load(TString fileName) {

  static const std::string routineName("GGSFileManager::Load");

  if (gEve)
    gEve->GetDefaultGLViewer()->GetClipSet()->SetClipType(TGLClip::EType(0));

  COUT(DEBUG) << "Opening " << fileName << ENDL;

  GGSTFilesHandler *hand = Open(fileName.Data(), NULL);
  if (hand == NULL) {
    COUT(ERROR) << "There was an error opening file " << fileName << ENDL;
    return;
  }

  COUT(DEBUG) << hand->GetNFiles() << "  files in the handler" << ENDL;
  COUT(DEBUG) << hand->GetFileName(0) << ENDL;

  _Reader = GetReader<GGSTHitsReader>(hand);
  _detList = _Reader->GetListOfDetectors();
  for (TString _det : _detList) {
    _Reader->SetDetector(_det, _Reader->HasPartHits(_det));
    COUT(DEBUG) << "Adding " << _det << " to detlist" << ENDL;
  }

  _MCTruthReader = GetReader<GGSTMCTruthReader>(hand);

  _pTree = new InteractionTree<GGSTPartHit *>;

  COUT(DEBUG) << "_Reader at " << _Reader << ENDL;
  COUT(DEBUG) << "_pTree at " << _pTree << ENDL;

  _fileLoaded = kTRUE;

  // ReadEvent( 0 );
}

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

  COUT(DEBUG) << "Click!" << ENDL;
  if (!_fileLoaded) {
    COUT(ERROR) << "You must open a file before clicking!" << ENDL;
    return;
  }
  ReadEvent(_currEvent - 1);
  ((MainWindow *)gEve)->LoadEvent();
}

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

  COUT(DEBUG) << "Click!" << ENDL;
  if (!_fileLoaded) {
    COUT(ERROR) << "You must open a file before clicking!" << ENDL;
    return;
  }
  ReadEvent(_currEvent + 1);
  ((MainWindow *)gEve)->LoadEvent();
}

void GGSFileManager::ReadEvent(int iev, bool force) {
  static const std::string routineName("GGSFileManager::ReadEvent");

  COUT(DEBUG) << "Click! " << iev << ENDL;
  if (!_fileLoaded) {
    COUT(ERROR) << "You must open a file before clicking!" << ENDL;
    return;
  }
  if (!force && (iev < 0 || _currEvent == iev))
    return;

  GetEntry(iev);
  BuildInteractionTree();
  ((MainWindow *)gEve)->LoadEvent();

  _currEvent = iev;
}

void GGSFileManager::BuildInteractionTree() {
  static const std::string routineName("GGSFileManager::BuildInteractionTree");

  _pTree->Clear();

  COUT(DEBUG) << _pTree->hashMap.size() << ENDL;

  for (TString _det : _detList) {
    if (_Reader->HasPartHits(_det)) {

      int nHits = _Reader->GetNHits(_det);
      GGSTIntHitBase *thisHit = nullptr;
      GGSTPartHit *thisPHit = nullptr;

      bool partHitClassChecked = false;
      for (int iHit = 0; iHit < nHits; iHit++) {
        thisHit = _Reader->GetHit(_det, iHit);
        if (!partHitClassChecked) {
          if (thisHit->GetPartHitsClassName() != "GGSTPartHit") {
            COUT(WARNING) << "Particle hits for detector " << _det
                          << " are not of class GGSTPartHit. Interaction teree for them will not be built." << ENDL;
            return;
          } else {
            partHitClassChecked = true;
          }
        }
        int nPHits = thisHit->GetNPartHits();
        for (int iPHit = 0; iPHit < nPHits; iPHit++) {
          thisPHit = (GGSTPartHit *)(_Reader->GetHit(_det, iHit)->GetPartHit(iPHit));
          _pTree->AddHit(thisPHit->trackID, thisPHit);
        }
      }
    }
  }

  _pTree->SortHits([](GGSTPartHit *h1, GGSTPartHit *h2) -> bool { return h1->time < h2->time; });
}