Display.h

Go to the documentation of this file.

// $Id: $
//==========================================================================
//  AIDA Detector description implementation for LCD
//--------------------------------------------------------------------------
// 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 DD4HEP_DDEVE_DISPLAY_H
#define DD4HEP_DDEVE_DISPLAY_H

// Framework include files
#include "DD4hep/LCDD.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;
      TEveCalo3D* calo3D;
      TEveCaloViz* caloViz;
      TEveCaloDataHist* eveHist;
      DisplayConfiguration::Config config;
      CalodataContext();
      CalodataContext(const CalodataContext& c);
      CalodataContext& operator=(const CalodataContext& c);
    };
    typedef std::map<std::string, CalodataContext> Calodata;
   
  protected:
    TEveManager* m_eve;
    Geometry::LCDD* m_lcdd;
    GenericEventHandler* m_evtHandler;
    TEveElementList* m_geoGlobal;
    TEveElementList* m_eveGlobal;
    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();

    Geometry::LCDD& lcdd() 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;   }
    void setEventHandlerName(std::string eventHandlerName) {m_eventHandlerName = eventHandlerName;}
    std::string getEventHandlerName() {return m_eventHandlerName;}

    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 OnNewEvent(EventHandler* handler);

    virtual void BuildMenus(TGMenuBar* bar=0);
    virtual void AddMenu(TGMenuBar* bar, PopupMenu* menu, int hints=kLHintsNormal);

    ClassDef(Display,0);
  };


} /* End namespace DD4hep   */


#endif /* DD4HEP_DDEVE_DISPLAY_H */