HW3DView.h

Go to the documentation of this file.

/*
 * HW3DView.h
 *
 *  Created on: 16 Sep 2021
 *      Author: Valerio Formato
 */

#ifndef HW3DView_H_
#define HW3DView_H_

#include "RQ_OBJECT.h"

// HERD headers
#include "algorithms/display/HWCaloAxesEventDisplay.h"
#include "algorithms/display/HWCaloEventDisplay.h"
#include "algorithms/display/HWHoughTrackEventDisplay.h"
#include "algorithms/display/HWMCTruthDisplay.h"
#include "algorithms/display/HWPSDEventDisplay.h"
#include "algorithms/display/HWTrackingDetectorEventDisplay.h"
#include "algorithms/display/HWTracks3DEventDisplay.h"

// EA headers
#include "analysis/EventLoopProxy.h"
#include "core/ObserverPtr.h"

// ROOT headers
#include "TApplication.h"
#include "TEveBoxSet.h"
#include "TEveGeoNode.h"
#include "TEveManager.h"
#include "TEvePointSet.h"
#include "TEveProjectionManager.h"
#include "TEveScene.h"
#include "TEveTrack.h"
#include "TEveTrackPropagator.h"
#include "TEveViewer.h"
#include "TEveWindow.h"
#include "TGNumberEntry.h"

namespace Herd {
class HERDward;

class HW3DView {
public:
  HW3DView();

  void Init();
  void AdjustCamera();
  void ClearScene();
  void DrawMCTruth(const HWMCTruthDisplay &mcTruthDisp);
  void DrawCaloHits(const HWCaloEventDisplay &caloDisp);
  void DrawPSDHits(const HWPSDEventDisplay &psdDisp);
  void DrawTrackingDetectorClusters(const HWTrackingDetectorEventDisplay &stkDisp);
  void DrawCaloAxes(const HWCaloAxesEventDisplay &axesDisp);
  void DrawTracks3D(const HWTracks3DEventDisplay &trackDisp);
  void LoadGeometryScene();

  char GetSceneTransparency() { return m_mainTransparency; }
  void SetGeoTransparency(char transp);

  static constexpr char startingTransparency = 95;

private:
  void FillScene(TEveElement *parent, const std::vector<std::string> &excluded, unsigned int depth);

  void SetSceneTransparency(TEveElement *scene, char transp);

  char m_mainTransparency = startingTransparency;

  std::unique_ptr<TEveWindowSlot> m_slot;
  std::unique_ptr<TEveWindowPack> m_pack;

  std::unique_ptr<TEveViewer> m_view;

  std::unique_ptr<TEveScene> m_geomScene;

  std::unique_ptr<TEveScene> m_eventScene;
};
} // namespace Herd

#endif