DtTerrainEditorWidget.h

Go to the documentation of this file.

/*****************************************************************************
* Copyright (c) 2021 MAK Technologies, Inc.
* All rights reserved.
*****************************************************************************/


#pragma once

#include <vrvTerrainEditor/vrvTerrainEditor.h>
#include <QtWidgets/QWidget>
#include <vrvTerrainEditor/DtTerrainEditorInterface.h>
#include <vrvTerrainEditor/DtTerrainEditorPropertyTreeWidgetItem.h>
#include <vrvUtil/DtSignalConnectionManager.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrvOsgEarth/DtOsgEarthPropertyItemKey.h>

// Class Forward Declarations
class QHBoxLayout;
class QVBoxLayout;
class QGridLayout;
class QTreeWidget;
class QTabWidget;
class QPushButton;
class QToolButton;
class QMenu;
class QTreeWidgetItem;

namespace makVrv
{
   class DtTerrainEditorProperties;
   class DtTerrainEditorPropertyTreeWidgetItem;
   class DtOsgEarthAddTerrainLayerWidget;
   class DtOsgEarthAddModelLayerWidget;
   class DtTdbAddFeatureLayerWidget;
   class DtOsgEarthCacheDialog;
   class DtOsgEarthGenerateBoundaryWidget;
   class DtTerrainEditorLogic;

   class DT_DLL_VRVTERRAINEDITOR DtTerrainEditorWidget : public QWidget,
      public DtTerrainEditorInterface
   {
      Q_OBJECT

   public:
      friend class DtTerrainEditorWidgetCreator;

      enum TerrainContentItemDataRole
      {
         ItemDataType = Qt::UserRole,
         ItemDataId,
         ItemDataName,
         ItemDataParentType,
         ItemDataParentId,
         ItemDataDisplayName
      };

      enum OsgEarthFragmentItemDataRole
      {
         ItemDataUrl = Qt::UserRole
      };

      enum TreeWidgetType
      {
         TerrainContent,
         TerrainFragment,
         TerrainProperty,
         UnknownTree
      };
      
      typedef std::map<TerrainContentItemKey, QTreeWidgetItem*> TerrainContentTreeWidgetItemsMap;

      typedef std::map<TerrainContentItemKey, DtTerrainEditorPropertyTreeWidgetItem*> TerrainPropertyTreeWidgetItemsMap;

      typedef std::map<std::string, QTreeWidgetItem*> OsgEarthFragmentContentItemsMap;

   public:
      virtual ~DtTerrainEditorWidget();

   private:
      DtTerrainEditorWidget();
      DtTerrainEditorWidget(const DtTerrainEditorWidget&);
      DtTerrainEditorWidget &operator=(const DtTerrainEditorWidget&);

   public:
      virtual void initialize();

      virtual bool isInitialized();

      virtual void cleanup();


      virtual void addTerrainContentTopLevelItem(const TerrainContentItemKey& key, const std::string& displayName);

      virtual void addTerrainContentItem(const TerrainContentItemKey& key, const TerrainContentItemKey& parentKey, const std::string& displayName, int index = -1, bool editable = false,
         bool checked = true);

      virtual void addOsgEarthFragmentContentTopLevelItem(const std::string& key, const std::string& displayName);

      virtual void addOsgEarthFragmentContentItem(const std::string& key, const std::string& parentKey, const std::string& displayName);

      virtual void clearTreeWidgets();

      virtual void cleanOsgEarthFragmentTreeWidget();

      virtual void setSelectedOsgEarthFragmentKeys(const std::vector<TerrainContentItemKey>& keys);

      virtual void setFocusOnTerrainContentTreeWidgetItem(const TerrainContentItemKey& key, bool expanded = true);

      virtual void removeTerrainContentTreeWidgetItem(const TerrainContentItemKey& key);

      virtual bool hasTerrainContentTreeWidgetItem(const TerrainContentItemKey& key);

      virtual bool hasTerrainPropertyTreeWidgetItem(const TerrainContentItemKey& key);

      virtual void resizeTerrainPropertyTreeWidget();

      virtual void resizeTerrainFragmentTreeWidget();

      virtual void moveTerrainContentTreeWidgetItem( TerrainContentItemKey key, bool up);

      virtual void setGenerateCacheButtonEnabled(bool enabled);

      virtual void setGenerateBoundaryButtonEnabled(bool enabled);

      virtual void setSaveTerrainButtonEnabled(bool enabled);

      virtual void setCommitChangeButtonEnabled(bool enabled);

      virtual void setRevertChangeButtonEnabled(bool enabled);

      virtual void enableOsgEarthFragmentTabWidget(bool enabled);

      virtual void showTerrainPropertyTreeWidgetItem(const TerrainContentItemKey& key);

      virtual void showTerrainPropertyTreeWidgetItems(const std::vector<TerrainContentItemKey>& key);

      int getIndexOfTerrainContentItem(const TerrainContentItemKey& key);

      virtual void setProperties(const TerrainContentItemKey& key, const DtTerrainEditorProperties* props);

      virtual bool properties(const TerrainContentItemKey& key, DtTerrainEditorProperties* props);

      virtual void showCacheDialog(const std::string& osgEarthTerrainDefinition);

      virtual DtUnicode showSaveMtfFileDialog(const DtUnicode& directory);

      virtual DtUnicode showSaveOsgEarthFileDialog(const DtUnicode& directory);

      virtual void showSaveFileConfirmationMessageBox(const DtUnicode& fileName);

      virtual bool showWarnReloadTerrainMessageBox(bool& checked);

      virtual bool showReloadTerrainAfterGenerateMaskMessageBox(const std::string& terrainPatch);

      virtual bool showChooseLayerTypeMessageBox(const DtUnicode& url, int& type);

      virtual void showFileExtensionNotSupportedMessageBox(const DtUnicode& fileName);

      virtual void setGenerateBoundaryModels(const std::vector<std::string>& models);

      virtual void recursivelyExpandOrCollapseTerrainContentTreeItemByKey(const TerrainContentItemKey& key, bool isExpanded);

      virtual void recursivelyExpandOrCollapseQTreeWidgetItem(QTreeWidgetItem* item, bool isExpanded);

      virtual void expandTerrainContentTreeWidget();

      virtual void expandOsgEarthFragmentTreeWidget();

      virtual bool checkIfNameDuplicate(const TerrainContentItemKey& key, const std::string& name);

      virtual void setTerrainContentTreeItemName(const TerrainContentItemKey& key, const std::string& name);

      virtual void setTerrainContentTreeItemVisibility(const TerrainContentItemKey& key, bool visible);

      virtual void setTerrainPropertyTreeItemName(const TerrainContentItemKey& key, const std::string& name);

      virtual void setTerrainPropertyTreeItemVisibility(const TerrainContentItemKey& key, bool visible);

   protected:

      DtTerrainEditorWidget(DtDe& de, QWidget* parent);

      void setupGui();

      virtual void connectSignals();

      virtual void setStyleSheetForTopPanelButtons();

      virtual void addTerrainPropertyTreeWidgetItem(const TerrainContentItemKey& key, DtTerrainEditorPropertyTreeWidgetItem* item);

      virtual QTreeWidgetItem* getTerrainContentItem(const TerrainContentItemKey& key);

      virtual void setTerrainContentItemAttributes(QTreeWidgetItem* item, const TerrainContentItemKey& key, const std::string& displayName, bool checked = true);

      virtual void removeTerrainContentTreeWidgetItem(QTreeWidgetItem* item);

      virtual void removeTerrainContentTreeWidgetItemFromMap(QTreeWidgetItem* item);

      int getTerrainContentItemType(QTreeWidgetItem* item);

      int getTerrainContentItemId(QTreeWidgetItem* item);

      std::string getTerrainContentItemName(QTreeWidgetItem* item);

      std::string getTerrainContentItemDisplayName(QTreeWidgetItem* item);

      int getTerrainContentItemParentType(QTreeWidgetItem* item);

      int getTerrainContentItemParentId(QTreeWidgetItem* item);

      TerrainContentItemKey getTerrainContentItemKey(QTreeWidgetItem* item);

      virtual DtTerrainEditorPropertyTreeWidgetItem* getTerrainEditorPropertyTreeWidgetItem(const TerrainContentItemKey& key) const;

      virtual DtTerrainEditorPropertyTreeWidgetItem* createTerrainEditorPropertyTreeWidgetItem(const TerrainContentItemKey& key);

      virtual std::string getTerrainEditorPropertyTreeWidgetItemCreatorName(TerrainContentItemType type);

      std::string getOsgEarthFragmentItemKey(QTreeWidgetItem* item);

      bool hasTerrainContentItem(const TerrainContentItemKey& key);

      virtual void populateTerrainContentTreeContextMenu(QContextMenuEvent* e);

      virtual void populateTerrainFragmentTreeContextMenu(QContextMenuEvent* e);

      virtual void populateTerrainPropertyTreeContextMenu(QContextMenuEvent* e);

      virtual void renameTerrainContentTreeWidgetItem();

      virtual void saveTerrainContentTreeWidgetItem();

      virtual void deleteTerrainContentTreeWidgetItem();

      virtual void hideAllTerrainContentTreeTopLevelItems();

      virtual void hideAllTerrainPropertyTreeTopLevelItems();

      virtual bool isTerrainContentTabEnabled();

      virtual bool isTerrainPropertyTabEnabled();

   protected:

      virtual void slot_osgEarthTerrainLayerRequestAdded(int type, const std::string& name, const std::string& driver, const DtUnicode& url);

      virtual void slot_osgEarthModelLayerDataRequestAdded(const std::string& name, const std::string& driver, const DtUnicode& url, const std::string& wfsType);

      virtual void slot_MAKFeatureLayerDataRequestAdded(const DtUnicode& filepath, bool toFrontEnd, bool toSimEngine);

      virtual void slot_generateBoundary(const std::string& definition, const std::string& name, bool, bool);

   protected slots:

      virtual void on_terrainContentTreeWidget_currentItemChanged(QTreeWidgetItem* current, QTreeWidgetItem* previous);

      virtual void on_terrainContentTreeWidget_itemPressed(QTreeWidgetItem* item, int column);

      virtual void on_terrainContentTreeWidget_itemDoubleClicked(QTreeWidgetItem* item, int column);

      virtual void on_terrainContentTreeWidget_itemChanged(QTreeWidgetItem* item, int column);

      virtual void on_osgEarthFragmentTreeWidget_currentItemChanged(QTreeWidgetItem* current, QTreeWidgetItem* previous);

      virtual void on_terrainPropertyTreeWidget_currentItemChanged(QTreeWidgetItem* current, QTreeWidgetItem* previous);

      virtual void on_terrainPropertyTreeWidget_itemPressed(QTreeWidgetItem*, int column);

      virtual void on_terrainContentTabWidget_currentChanged(int index);

      virtual void on_terrainPropertiesTabWidget_currentChanged(int index);

      virtual void on_terrainContentTreeMenu_triggered(QAction* action);

      virtual void on_terrainPropertyTreeMenu_triggered(QAction* action);

      virtual void on_terrainContentTree_checkBox_stateChanged(const QTreeWidgetItem *, int checked);

      virtual void on_addTerrainPatchButton_released();

      virtual void on_addImageLayerButton_released();

      virtual void on_addElevationLayerButton_released();

      virtual void on_addFeatureButton_released();

      virtual void on_addOceanLayerButton_released();

      virtual void on_commitChangesButton_released();

      virtual void on_revertChangesButton_released();

      virtual void on_saveTerrainButton_released();

      virtual void on_generateCacheButton_released();

      virtual void on_generateBoundaryButton_released();

      virtual void contextMenuEvent(QContextMenuEvent* e);

      virtual void dragEnterEvent(QDragEnterEvent *e);

      virtual void dropEvent(QDropEvent* e);


   protected:

      DtDe & myDe;
      bool myIsInitialized;
      DtTerrainEditorLogic* myLogic;
      DtSignalConnectionManager mySignals;

      DtOsgEarthAddTerrainLayerWidget* myOsgEarthAddTerrainLayerWidget;
      DtOsgEarthAddModelLayerWidget* myOsgEarthAddModelLayerWidget;
      DtTdbAddFeatureLayerWidget* myMAKAddFeatureLayerWidget;
      DtOsgEarthCacheDialog* myOsgEarthCacheDialog;
      DtOsgEarthGenerateBoundaryWidget* myOsgEarthGenerateBoundaryWidget;

      QVBoxLayout* myMainLayout;
      QTabWidget* myTopLevelTabWidget;

      QWidget* myAddDataWidget;
      QGridLayout* myAddDataButtonsLayout;
      QToolButton* myAddTerrainPatchButton;
      QToolButton* myAddImageLayerButton;
      QToolButton* myAddElevationLayerButton;
      QToolButton* myAddFeatureButton;
      QToolButton* myAddFeatureLayerButton;
      QToolButton* myAddOceanLayerButton;

      QWidget* myCacheWidget;
      QHBoxLayout* myCacheButtonsLayout;
      QToolButton* myGenerateCacheButton;
      QToolButton* myGenerateBoundaryButton;

      QWidget* mySaveWidget;
      QHBoxLayout* mySaveButtonsLayout;
      QToolButton* mySaveOsgEarthTerrainButton;
      QToolButton* mySaveTerrainButton;

      QTabWidget* myTerrainPropertiesTabWidget;

      QHBoxLayout* myBottomButtonsLayout;
      QPushButton* myCommitChangesButton;
      QPushButton* myRevertChangesButton;

      QTabWidget*  myTerrainContentTabWidget;
      QTreeWidget* myTerrainContentTreeWidget;
      QTreeWidget* myOsgEarthFragmentTreeWidget;
      QTreeWidget* myTerrainPropertyTreeWidget;

      TerrainContentItemKey mySelectedTerrainContentItemKey;
      TerrainContentTreeWidgetItemsMap myTerrainContentTreeWidgetItemsMap;
      TreeWidgetType mySelectedTreeWidget;
      DtOsgEarthPropertyItemKey mySelectedDtOsgEarthPropertyItemKey;
      std::vector<TerrainContentItemKey> mySelectedOsgEarthFragmentKeys;

      OsgEarthFragmentContentItemsMap myOsgEarthFragmentContentItemsMap;

      TerrainPropertyTreeWidgetItemsMap myTerrainPropertyTreeWidgetItemsMap;

      QMenu* myTerrainContentTreeMenu;
      QMenu* myTerrainPropertyTreeMenu;
   };

   class DT_DLL_VRVTERRAINEDITOR DtTerrainEditorWidgetCreator
      : public DtVirtualBaseClass
   {
   public:
      static DtTerrainEditorWidgetCreator& instance(DtDe& de);
      static void registerInstance(DtDe& de, DtTerrainEditorWidgetCreator* creator);
      virtual DtTerrainEditorWidget* create(DtDe& de, QWidget* parent) const;
   };
}