Display.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
#ifndef DDEVE_DISPLAY_H
#define DDEVE_DISPLAY_H
 
// Framework include files
#include "DD4hep/Detector.h"
#include "DD4hep/Printout.h"
#include "DDEve/PopupMenu.h"
#include "DDEve/EventHandler.h"
#include "DDEve/DisplayConfiguration.h"
 
// C/C++ include files
#include <set>
 
// Forward declarations
class TEveCalo3D;
class TEveCaloDataHist;
class TEveElementList;
class TEveManager;
class TEveElement;
class TEveCaloViz;
class TGMenuBar;
class TGClient;
class TFile;
 
namespace dd4hep {
 
  // Forward declarations
  class View;
  class ViewMenu;
  class DD4hepMenu;
  class EventHandler;
  class ViewConfiguration;
  class CalodataConfiguration;
  class GenericEventHandler;
  class DisplayConfiguration;
 
  /*
   * \author  M.Frank
   * \version 1.0
   * \ingroup DD4HEP_EVE
   */
  class Display : public EventConsumer {
  public:
    typedef DisplayConfiguration::ViewConfig         ViewConfig;
    typedef DisplayConfiguration::Config             DataConfig;
    typedef std::set<View*>                          Views;
    typedef std::set<DisplayConfiguration*>          Configurations;
    typedef std::set<PopupMenu*>                     Menus;
    typedef std::map<std::string, TEveElementList*>  Topics;
    typedef std::map<std::string, ViewConfig>        ViewConfigurations;
    typedef std::map<std::string, DataConfig>        DataConfigurations;
 
    struct CalodataContext {
      int slice = 0;
      TEveCalo3D* calo3D = 0;
      TEveCaloViz* caloViz = 0;
      TEveCaloDataHist* eveHist = 0;
      DisplayConfiguration::Config config;
      CalodataContext() = default;
      CalodataContext(const CalodataContext& c) = default;
      CalodataContext& operator=(const CalodataContext& c) = default;
    };
 
    typedef std::map<std::string, CalodataContext> Calodata;
   
  protected:
    TEveManager*         m_eve = 0;
    Detector*            m_detDesc = 0;
    GenericEventHandler* m_evtHandler = 0;
    TEveElementList*     m_geoGlobal = 0;
    TEveElementList*     m_eveGlobal = 0;
    ViewMenu*            m_viewMenu;
    DD4hepMenu*          m_dd4Menu;
    Topics               m_geoTopics;
    Topics               m_eveTopics;
    Views                m_eveViews;
    Menus                m_menus;
    ViewConfigurations   m_viewConfigs;
    DataConfigurations   m_calodataConfigs;
    DataConfigurations   m_collectionsConfigs;
    Calodata             m_calodata;
    int                  m_visLevel;
    int                  m_loadLevel;
    std::string          m_eventHandlerName;
 
  public:
    Display(TEveManager* eve);
    virtual ~Display();
 
    Detector& detectorDescription() const;
    TEveManager& manager() const                           { return *m_eve;             }
    const ViewConfigurations& viewConfigurations() const   { return m_viewConfigs;      }
    void setVisLevel(int new_level)                        { m_visLevel = new_level;    }
    void setLoadLevel(int new_level)                       { m_loadLevel = new_level;   }
    std::string getEventHandlerName()                      { return m_eventHandlerName; }
    void setEventHandlerName(const std::string& nam)       { m_eventHandlerName = nam;  }
 
    TGClient& client() const;
 
    void ChooseGeometry();
    void LoadXML(const char* xmlFile);
    void LoadGeometryRoot(const char* rootFile);
    TFile* Open(const char* rootFile) const;
 
    GenericEventHandler& eventHandler() const;
 
    void MessageBox(PrintLevel level, const std::string& text, const std::string& title="") const;
 
    std::string OpenXmlFileDialog(const std::string& default_dir)  const;
 
    std::string OpenEventFileDialog(const std::string& default_dir)  const;
 
    void LoadGeoChildren(TEveElement* start, int levels, bool redraw);
    void MakeNodesVisible(TEveElement* e, bool visible, int level);
 
    void ImportConfiguration(const DisplayConfiguration& config);
 
    const ViewConfig* GetViewConfiguration(const std::string& name)  const;
    const DataConfig* GetCalodataConfiguration(const std::string& name)  const;
    CalodataContext& GetCaloHistogram(const std::string& name);
 
    virtual void RegisterEvents(View* view);
    virtual void UnregisterEvents(View* view);
 
    TEveElementList& GetGeo();
    virtual TEveElementList& GetGeoTopic(const std::string& name);
    virtual TEveElementList& GetGeoTopic(const std::string& name)  const;
 
    void ImportGeo(TEveElement* el);
    void ImportGeo(const std::string& topic, TEveElement* el);
 
    virtual TEveElementList& GetEveTopic(const std::string& name);
    virtual TEveElementList& GetEveTopic(const std::string& name)  const;
 
    void ImportEvent(TEveElement* el);
    void ImportEvent(const std::string& topic, TEveElement* el);
 
    virtual void OnFileOpen(EventHandler& handler)  override;
    virtual void OnNewEvent(EventHandler& handler)  override;
 
    virtual void BuildMenus(TGMenuBar* menuBar=0);
    virtual void AddMenu(TGMenuBar* menuBar, PopupMenu* menu, int hints=kLHintsNormal);
 
    ClassDefOverride(Display,0);
  };
}      /* End namespace dd4hep   */
#endif // DDEVE_DISPLAY_H