Objects.h

Go to the documentation of this file.

 
#ifndef EVENTDISPLAY_EVENTOBJECTS_H_
#define EVENTDISPLAY_EVENTOBJECTS_H_
 
#include "DetDescr/HcalID.h"
#include "Ecal/Event/EcalCluster.h"
#include "Ecal/Event/EcalHit.h"
#include "EventDisplay/DetectorGeometry.h"
#include "EventDisplay/EveDetectorGeometry.h"
#include "EventDisplay/EveShapeDrawer.h"
#include "Hcal/Event/HcalHit.h"
#include "SimCore/Event/SimCalorimeterHit.h"
#include "SimCore/Event/SimParticle.h"
#include "SimCore/Event/SimTrackerHit.h"
#include "TColor.h"
#include "TEveArrow.h"
#include "TEveBox.h"
#include "TEveRGBAPalette.h"
#include "TRandom.h"
 
namespace eventdisplay {
 
class Objects {
 public:
  Objects();
 
  ~Objects() {
    delete sim_objects_;
    delete rec_objects_;
  }
 
  void initialize();
 
  template <typename T>
  void draw(T o) {
    EXCEPTION_RAISE("NotImp", "Drawing not implemented for the input type.");
  }
 
  void draw(std::vector<ldmx::EcalHit> hits);
 
  void draw(std::vector<ldmx::HcalHit> hits);
 
  void draw(std::vector<ldmx::EcalCluster> clusters);
 
  void draw(std::vector<ldmx::SimTrackerHit> hits);
 
  void draw(std::vector<ldmx::SimCalorimeterHit> hits);
 
  void draw(std::map<int, ldmx::SimParticle> particles);
 
  void setSimThresh(double simThresh);
 
  void colorClusters();
 
  TEveElement* getSimObjects() { return sim_objects_; }
 
  TEveElement* getRecObjects() { return rec_objects_; }
 
 private:
  TEveElement* sim_objects_;
  TEveElement* rec_objects_;
 
  double sim_thresh_ = 0;
 
  std::vector<Color_t> colors_ = {kRed,     kBlue,  kGreen,  kYellow,
                                  kMagenta, kBlack, kOrange, kPink};
 
  TRandom r_;
};
}  // namespace eventdisplay
 
#endif