HWMainWindow.h

Go to the documentation of this file.

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

#ifndef HWMainWindow_H_
#define HWMainWindow_H_

#include "RQ_OBJECT.h"

// HERD headers
#include "algorithms/display/HW3DView.h"
#include "algorithms/display/HWCaloEventDisplay.h"
#include "algorithms/display/HWHoughTrackEventDisplay.h"
#include "algorithms/display/HWMCTruthDisplay.h"
#include "algorithms/display/HWPSDEventDisplay.h"
#include "algorithms/display/HWProjections.h"
#include "algorithms/display/HWTrackingDetectorEventDisplay.h"
#include "algorithms/display/HWTracks3DEventDisplay.h"
#include "common/RefFrame.h"
#include "dataobjects/CaloHits.h"
#include "dataobjects/MCTruth.h"
#include "dataobjects/PsdGeoParamsColl.h"
#include "dataobjects/PsdHitsColl.h"
#include "dataobjects/StkClustersColl.h"
#include "dataobjects/StkGeoParamsColl.h"
#include "dataobjects/Track2DColl.h"
#include "dataobjects/Tracks.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"
#include "TGSlider.h"
#include "dataobjects/Track2DColl.h"

namespace Herd {
class HERDward;

class HWMainWindow : public TApplication, public TEveManager {
  RQ_OBJECT("HWMainWindow")

  friend class HERDward;

public:
  HWMainWindow();
  virtual ~HWMainWindow() { Terminate(); }; // NOLINT

  void SetReturnFromRun(bool ret) { TApplication::SetReturnFromRun(ret); };
  bool ReturnFromRun() { return TApplication::ReturnFromRun(); };
  void ExitFromLoop();
  void PrevEvent();
  void NextEvent();
  void LoadEvent();
  void SetEventNumber(const char *text);
  void SetDisplayDetector(const char *text);
  void SetActiveTab(int tab);
  void SetGeoTransparency(int transp);
  virtual void Terminate(int status = 0); // NOLINT

private:
  int m_evNo;
  int m_totEvents;

  int m_activeTab = 0;

  TGNumberEntryField *m_evField;

  std::unique_ptr<TEveWindowSlot> m_3DSlot;

  std::unique_ptr<HWProjections> m_projections;
  std::unique_ptr<HW3DView> m_3DView;

  void LoadGeometry();
  void SetSceneTransparency(TEveGeoNode *node, int transp);
  void MakeTransparentScene(TEveGeoNode *node, int transp);

  HWMCTruthDisplay m_mcTruthDisp;
  HWCaloEventDisplay m_caloDisp;
  HWTrackingDetectorEventDisplay m_trackerDisp;
  HWTrackingDetectorEventDisplay m_scdDisp{"SCD"};
  HWHoughTrackEventDisplay m_houghDisp;
  HWTracks3DEventDisplay m_tracks3DDisp;
  HWCaloAxesEventDisplay m_caloAxesDisp;
  HWPSDEventDisplay m_psdDisp;

  template <class TrackingGeoParams>
  void SetTrackingClusters(const ClustersColl &clColl, const SidesArray<std::vector<TrackingGeoParams>> &geoParamsColl,
                           HWTrackingDetectorEventDisplay &_disp);

  void SetCaloHits(const CaloHits &caloHits, const CaloGeoParams &caloParams);
  void SetPSDHits(const PsdHitsColl &psdColl, const PsdGeoParamsColl &geoParamsColl);
  void SetHoughTracks(const Track2DColl &tracks);
  void SetTracks(const Tracks &tracks);
  void SetCaloAxes(const Tracks &axes);
  void SetMCTruth(const MCTruth &mcTruth);
  void SetTrackerName(std::string name) { m_trackerDisp.SetName(std::move(name)); }

  std::unordered_map<std::string, bool> m_detEnableMap;
  TGMainFrame *m_frmMain;
  TGVerticalFrame *m_frmMainVertical;
  TGGroupFrame *m_hitGroup;
  TGHSlider *m_transpSlider;

  void ClearScene();
  void MakeGui();
  void AddGUIHitButtons();
  void AddGUITranspSlide();

  // Event loop stuff
  EA::observer_ptr<EA::EventLoopProxy> m_evLoopProxy;
  void SetEventNumber(int evNo);
  void SetNumberOfEvents(int totEvents) { m_totEvents = totEvents; };
  void UpdateDetectorStatus();

  ClassDef(HWMainWindow, 1)
};

} // namespace Herd

#endif