Display.h

Go to the documentation of this file.

 
#ifndef EVENTDISPLAY_EVENTDISPLAY_H_
#define EVENTDISPLAY_EVENTDISPLAY_H_
 
#include <iostream>
 
#include "EventDisplay/EveDetectorGeometry.h"
#include "EventDisplay/Objects.h"
#include "Framework/Event.h"
#include "Framework/EventFile.h"
#include "TBranchElement.h"
#include "TEveBrowser.h"
#include "TEveElement.h"
#include "TEveEventManager.h"
#include "TEveManager.h"
#include "TEveViewer.h"
#include "TFile.h"
#include "TGButton.h"
#include "TGFrame.h"
#include "TGLViewer.h"
#include "TGTextEntry.h"
#include "TRint.h"
#include "TString.h"
#include "TTree.h"
 
namespace eventdisplay {
 
class Display : public TGMainFrame {
 public:
  Display(TEveManager* manager, bool verbose);
 
  virtual ~Display() {
    // need to delete EventFile so that it is deleted before Event
    the_file_.reset(nullptr);
 
    delete theDetector_;
 
    delete textBoxClustersCollName_;
    delete textBoxSimThresh_;
    delete textBoxEcalRecHitsCollName_;
    delete textBoxHcalRecHitsCollName_;
    delete textBoxTrackerHitsCollName_;
    delete textBoxEcalScorePlaneBranch_;
 
    delete manager_;
  }
 
  bool SetFile(const TString file);
 
  void NextEvent();
 
  void GetECALRecHitsCollInput();
 
  void GetClustersCollInput();
 
  void GetTrackerHitsCollInput();
 
  void GetHCALRecHitsCollInput();
 
  void GetEcalSimParticlesCollInput();
 
  bool SetSimThresh();
 
  void ColorClusters();
 
 private:
  std::string getText(TGTextEntry* box) const { return box->GetText(); }
 
  template <typename EventObjectType>
  void draw(const std::string& name) {
    try {
      auto event_object{the_event_.getObject<EventObjectType>(name)};
      objects_.draw(event_object);
      if (verbose_) {
        std::cout << "[ Display ] : Loaded '" << name
                  << "' into memory as a EVE object." << std::endl;
      }
    } catch (const framework::exception::Exception& e) {
      std::cerr << "[ Display ] : Unable to draw an event object." << std::endl;
      std::cerr << "[" << e.name() << "] : " << e.message() << "\n"
                << "  at " << e.module() << ":" << e.line() << " in "
                << e.function() << std::endl;
    }
  }
 
 private:
  bool verbose_;  //* verbosity flag
 
  framework::Event the_event_{"display"};
 
  std::unique_ptr<framework::EventFile> the_file_;
 
  std::string clustersCollName_ = "ecalClusters";
  std::string ecalRecHitsCollName_ = "EcalRecHits";
  std::string hcalRecHitsCollName_ = "HcalRecHits";
  std::string trackerHitsCollName_ = "RecoilSimHits";
  std::string ecalSimParticlesCollName_ = "EcalScoringPlaneHits";
 
  EveDetectorGeometry* theDetector_{nullptr};
 
  Objects objects_;
 
  TGTextEntry* textBoxClustersCollName_;
  TGTextEntry* textBoxSimThresh_;
  TGTextEntry* textBoxEcalRecHitsCollName_;
  TGTextEntry* textBoxHcalRecHitsCollName_;
  TGTextEntry* textBoxTrackerHitsCollName_;
  TGTextEntry* textBoxEcalScorePlaneBranch_;
 
  TEveManager* manager_{nullptr};
 
  ClassDef(Display, 2);
};
}  // namespace eventdisplay
 
#endif