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 "herd/analysis/algorithms/display/HW3DView.h"
#include "herd/analysis/algorithms/display/HWCaloEventDisplay.h"
#include "herd/analysis/algorithms/display/HWHoughTrackEventDisplay.h"
#include "herd/analysis/algorithms/display/HWMCTruthDisplay.h"
#include "herd/analysis/algorithms/display/HWPSDEventDisplay.h"
#include "herd/analysis/algorithms/display/HWProjections.h"
#include "herd/analysis/algorithms/display/HWTrackingDetectorEventDisplay.h"
#include "herd/analysis/algorithms/display/HWTracks3DEventDisplay.h"
#include "herd/analysis/common/RefFrame.h"
#include "herd/analysis/dataobjects/CaloHits.h"
#include "herd/analysis/dataobjects/MCTruth.h"
#include "herd/analysis/dataobjects/PsdGeoParamsColl.h"
#include "herd/analysis/dataobjects/PsdHitsColl.h"
#include "herd/analysis/dataobjects/Track2DColl.h"
#include "herd/analysis/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 "herd/analysis/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, 2)
};
 
} // namespace Herd
 
#endif