ecvDBRoot.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
// Qt
#include <QAbstractItemModel>
#include <QPoint>
#include <QTreeView>
 
// CV_DB_LIB
#include <ecvHObject.h>
 
// System
#include <unordered_set>
 
class QAction;
class QStandardItemModel;
 
class ccPropertiesTreeDelegate;
class ccHObject;
 
struct dbTreeSelectionInfo {
    size_t selCount;
    size_t sfCount;
    size_t colorCount;
    size_t normalsCount;
    size_t octreeCount;
    size_t cloudCount;
    size_t gridCound;
    size_t groupCount;
    size_t polylineCount;
    size_t planeCount;
    size_t circleCount;
    size_t meshCount;
    size_t imageCount;
    size_t sensorCount;
    size_t gblSensorCount;
    size_t cameraSensorCount;
    size_t kdTreeCount;
 
    void reset() { memset(this, 0, sizeof(dbTreeSelectionInfo)); }
};
 
class ccCustomQTreeView : public QTreeView {
    Q_OBJECT
 
public:
    explicit ccCustomQTreeView(QWidget* parent) : QTreeView(parent) {}
 
protected:
    // inherited from QTreeView
    virtual QItemSelectionModel::SelectionFlags selectionCommand(
            const QModelIndex& index, const QEvent* event = nullptr) const;
};
 
class ccDBRoot : public QAbstractItemModel {
    Q_OBJECT
 
public:
 
    ccDBRoot(ccCustomQTreeView* dbTreeWidget,
             QTreeView* propertiesTreeWidget,
             QObject* parent = nullptr);
 
    virtual ~ccDBRoot() override;
 
    void hide();
    void show();
 
    ccHObject* getRootEntity();
 
    void hidePropertiesView();
    void updatePropertiesView();
 
    ccPropertiesTreeDelegate* getPropertiesDelegate() {
        return m_ccPropDelegate;
    }
 
    void addElement(ccHObject* object, bool autoExpand = true);
 
 
    void removeElement(ccHObject* object);
 
 
    void removeElements(ccHObject::Container& objects);
 
    ccHObject* find(int uniqueID) const;
 
    int countSelectedEntities(CV_CLASS_ENUM filter = CV_TYPES::OBJECT);
 
    size_t getSelectedEntities(ccHObject::Container& selectedEntities,
                               CV_CLASS_ENUM filter = CV_TYPES::OBJECT,
                               dbTreeSelectionInfo* info = nullptr);
 
    void expandElement(ccHObject* object, bool state);
 
    void unselectEntity(ccHObject* obj);
 
    void unselectAllEntities();
 
    void unloadAll();
 
    // inherited from QAbstractItemModel
    virtual QVariant data(const QModelIndex& index, int role) const override;
    virtual QModelIndex index(
            int row,
            int column,
            const QModelIndex& parentIndex = QModelIndex()) const override;
    QModelIndex index(ccHObject* object);
    virtual QModelIndex parent(const QModelIndex& index) const override;
    virtual int rowCount(
            const QModelIndex& parent = QModelIndex()) const override;
    virtual int columnCount(
            const QModelIndex& parent = QModelIndex()) const override;
    virtual Qt::ItemFlags flags(const QModelIndex& index) const override;
    virtual bool setData(const QModelIndex& index,
                         const QVariant& value,
                         int role = Qt::EditRole) override;
    virtual Qt::DropActions supportedDropActions() const override;
    virtual bool dropMimeData(const QMimeData* data,
                              Qt::DropAction action,
                              int row,
                              int column,
                              const QModelIndex& parent) override;
    virtual QMap<int, QVariant> itemData(
            const QModelIndex& index) const override;
    virtual Qt::DropActions supportedDragActions() const override {
        return Qt::MoveAction;
    }
 
public slots:
    void changeSelection(const QItemSelection& selected,
                         const QItemSelection& deselected);
    void reflectObjectPropChange(ccHObject* obj);
    void redrawCCObject(ccHObject* object, bool forceRedraw = true);
    void redrawCCObjectAndChildren(ccHObject* object, bool forceRedraw = true);
    void updateCCObject(ccHObject* object);
    void deleteSelectedEntities();
 
 
    void selectEntity(ccHObject* obj, bool forceAdditiveSelection = false);
 
 
    void selectEntities(std::unordered_set<int> entIDs);
 
 
    void selectEntities(const ccHObject::Container& entities,
                        bool incremental = false);
 
protected:
    enum TOGGLE_PROPERTY {
        TG_ENABLE,
        TG_VISIBLE,
        TG_COLOR,
        TG_SF,
        TG_NORMAL,
        TG_MATERIAL,
        TG_3D_NAME
    };
 
    void toggleSelectedEntitiesProperty(TOGGLE_PROPERTY prop);
 
protected slots:
    void showContextMenu(const QPoint&);
 
    void expandBranch();
    void collapseBranch();
    void gatherRecursiveInformation();
    void sortChildrenAZ();
    void sortChildrenZA();
    void sortChildrenType();
    void selectByTypeAndName();
    void exportImages();
 
    inline void toggleSelectedEntities() {
        toggleSelectedEntitiesProperty(TG_ENABLE);
    }
    inline void toggleSelectedEntitiesVisibility() {
        toggleSelectedEntitiesProperty(TG_VISIBLE);
    }
    inline void toggleSelectedEntitiesColor() {
        toggleSelectedEntitiesProperty(TG_COLOR);
    }
    inline void toggleSelectedEntitiesNormals() {
        toggleSelectedEntitiesProperty(TG_NORMAL);
    }
    inline void toggleSelectedEntitiesSF() {
        toggleSelectedEntitiesProperty(TG_SF);
    }
    inline void toggleSelectedEntitiesMat() {
        toggleSelectedEntitiesProperty(TG_MATERIAL);
    }
    inline void toggleSelectedEntities3DName() {
        toggleSelectedEntitiesProperty(TG_3D_NAME);
    }
 
    void addEmptyGroup();
    void alignCameraWithEntityDirect() { alignCameraWithEntity(false); }
    void alignCameraWithEntityIndirect() { alignCameraWithEntity(true); }
    void enableBubbleViewMode();
    void editLabelScalarValue();
 
signals:
    void selectionChanged();
    void dbIsEmpty();
    void dbIsNotEmptyAnymore();
 
protected:
 
    void alignCameraWithEntity(bool reverse);
 
    void showPropertiesView(ccHObject* obj);
 
    enum SortRules { SORT_A2Z, SORT_Z2A, SORT_BY_TYPE };
 
    void sortSelectedEntitiesChildren(SortRules rule);
 
    void expandOrCollapseHoveredBranch(bool expand);
 
    void selectChildrenByTypeAndName(CV_CLASS_ENUM type,
                                     bool typeIsExclusive = true,
                                     QString name = QString(),
                                     bool nameIsRegex = false);
 
    ccHObject* m_treeRoot;
 
    QTreeView* m_dbTreeWidget;
 
    QTreeView* m_propertiesTreeWidget;
 
    QStandardItemModel* m_propertiesModel;
    ccPropertiesTreeDelegate* m_ccPropDelegate;
 
    QAction* m_expandBranch;
    QAction* m_collapseBranch;
    QAction* m_gatherInformation;
    QAction* m_sortChildrenAZ;
    QAction* m_sortChildrenZA;
    QAction* m_sortChildrenType;
    QAction* m_selectByTypeAndName;
    QAction* m_exportImages;
    QAction* m_deleteSelectedEntities;
    QAction* m_toggleSelectedEntities;
    QAction* m_toggleSelectedEntitiesVisibility;
    QAction* m_toggleSelectedEntitiesColor;
    QAction* m_toggleSelectedEntitiesNormals;
    QAction* m_toggleSelectedEntitiesMat;
    QAction* m_toggleSelectedEntitiesSF;
    QAction* m_toggleSelectedEntities3DName;
    QAction* m_addEmptyGroup;
    QAction* m_alignCameraWithEntity;
    QAction* m_alignCameraWithEntityReverse;
    QAction* m_enableBubbleViewMode;
    QAction* m_editLabelScalarValue;
 
    QPoint m_contextMenuPos;
};