HWProjections.h

Go to the documentation of this file.

/*
 * HWProjections.h
 *
 *  Created on: 02 Jan 2019
 *      Author: Valerio Formato
 */

#ifndef HWProjections_H_
#define HWProjections_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 HWProjections {
public:
  HWProjections();

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

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

  static constexpr char startingTransparency = 95;

private:
  void FillScene(RefFrame::View view, 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_XZView;
  std::unique_ptr<TEveViewer> m_YZView;
  std::unique_ptr<TEveViewer> m_XYView;

  std::unique_ptr<TEveScene> m_XZGeomScene;
  std::unique_ptr<TEveScene> m_YZGeomScene;
  std::unique_ptr<TEveScene> m_XYGeomScene;

  std::unique_ptr<TEveScene> m_XZEventScene;
  std::unique_ptr<TEveScene> m_YZEventScene;
  std::unique_ptr<TEveScene> m_XYEventScene;
};
} // namespace Herd

#endif