DtItemsListWidget.h

Go to the documentation of this file.

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


#pragma once

#include <vrvCoreQt/vrvCoreQt.h>
#include <vrvCoreQt/DtItemsListInterface.h>
#include <vrvCoreQt/DtTreeWidget.h>
#include <QtWidgets/QWidget>
#include <vrvUtil/DtSignalConnectionManager.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <boost/unordered_map.hpp>

class QVBoxLayout;
class QTreeWidgetItem;
class QLabel;
class QLineEdit;
class QPushButton;

namespace makVrv
{
   class DtDe;

   class DT_DLL_VRVCOREQT DtItemsListWidget : public QWidget
      , public DtItemsListInterface
   {
      Q_OBJECT

   public :

      friend class DtItemsListWidgetCreator;

      enum ItemsListWidgetItemDataRole
      {
         ItemElementId = Qt::UserRole,
         ItemDisplayName,
         ItemType,
         ItemIsSubType
      };

      virtual ~DtItemsListWidget();

      void setupGui();

      void connectSignals();

      virtual void addSubTypeTopLevelItem(const DtElementAttributes::ElementSubtypes& subType,
         const std::string& displayName, int index, const std::string& icon);

      virtual void addSubTypeTopLevelItem(const DtElementAttributes::ElementSubtypes& subType,
         const std::string& displayName, int index, double r, double g, double b);

      virtual void addSubTypeItem(const DtElementAttributes::ElementSubtypes& subType, DtElementID elementId, const std::string& displayName);

      virtual bool hasSubTypeTopLevelItem(const DtElementAttributes::ElementSubtypes& subType);

      virtual void addTypeTopLevelItem(const DtElementEntry::ElementType& type, 
         const std::string& displayName, int index, const std::string& icon);

      virtual void addTypeItem(const DtElementEntry::ElementType& type, DtElementID elementId, const std::string& displayName);

      virtual bool hasTypeTopLevelItem(const DtElementEntry::ElementType& type);

      virtual void sortItems();

      virtual void removeItem(DtElementID id);

      virtual void renameItem(DtElementID id, const std::string& newName);

      virtual void updateItemDamageState(DtElementID id, DtEntityStateListener::DamageState damageState);

      virtual void updateItemSubType(DtElementID id, int subType);

      virtual void setSelectedItems(const DtSelectionManager::IdSet& selected,
         const DtSelectionManager::IdSet& deselected);

      virtual bool comparator(const QTreeWidgetItem& lhs, const QTreeWidgetItem& rhs) const;

   protected:

      DtItemsListWidget(DtDe& de, QWidget* parent = NULL, const Qt::WindowFlags& f = 0);
      
      void setItemAttributes(QTreeWidgetItem* item, DtElementID elementId, const QString& displayName);

      DtElementID getItemElementId(QTreeWidgetItem* item);

      bool parseNumber(const QTreeWidgetItem& lhs, int& number, QString& name) const;

   protected slots:

      virtual void on_objectsTreeWidget_rightClickOnItem(QTreeWidgetItem* item, QMouseEvent*, int column);

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

      virtual void on_objectsTreeWidget_itemSelectionChanged();

      virtual void on_lookupLineEdit_textChanged(QString str);

      virtual void on_searchCleanButton_released();

   private:

      DtItemsListWidget() = delete;

      DtItemsListWidget(const DtItemsListWidget& orig) = delete;

      DtItemsListWidget& operator=(const DtItemsListWidget& orig) = delete;

   protected:
      DtDe & myDe;
      QVBoxLayout * myMainLayout;
      QLabel* myLookupLabel;
      QLineEdit* myLookupLineEdit;
      QPushButton* mySearchCleanButton;
      DtTreeWidget* myObjectsTreeWidget;
      boost::unordered_map<DtElementEntry::ElementType, QTreeWidgetItem*> myElementTypeItemsMap;
      boost::unordered_map<DtElementAttributes::ElementSubtypes, QTreeWidgetItem*> myElementSubtypeItemsMap;
      boost::unordered_map<DtElementID, QTreeWidgetItem*> myElementItemsMap;
     // Allow in const function
     mutable std::map<const QString, std::pair<QString, int> > myObjectListCache;
   };

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