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 text_box_clusters_coll_name_;
    delete text_box_sim_thresh_;
    delete text_box_ecal_rec_hits_coll_name_;
    delete text_box_hcal_rec_hits_coll_name_;
    delete text_box_tracker_hits_coll_name_;
    delete text_box_ecal_score_plane_branch_;
 
    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 clusters_coll_name_ = "ecalClusters";
  std::string ecal_rec_hits_coll_name_ = "EcalRecHits";
  std::string hcal_rec_hits_coll_name_ = "HcalRecHits";
  std::string tracker_hits_coll_name_ = "RecoilSimHits";
  std::string ecal_sim_particles_coll_name_ = "EcalScoringPlaneHits";
 
  EveDetectorGeometry* the_detector_{nullptr};
 
  Objects objects_;
 
  TGTextEntry* text_box_clusters_coll_name_;
  TGTextEntry* text_box_sim_thresh_;
  TGTextEntry* text_box_ecal_rec_hits_coll_name_;
  TGTextEntry* text_box_hcal_rec_hits_coll_name_;
  TGTextEntry* text_box_tracker_hits_coll_name_;
  TGTextEntry* text_box_ecal_score_plane_branch_;
 
  TEveManager* manager_{nullptr};
 
  ClassDef(Display, 2);
};
}  // namespace eventdisplay
 
#endif