CustomModelLogic.h

#pragma once
#include "DLLInfo.h"
#include <NDEVR/ContextMenuManager.h>
#include <NDEVR/QTModelManager.h>
#include <NDEVR/DesignObjectDialog.h>
#include <NDEVR/Pointer.h>
#include <NDEVR/UUID.h>
#include <NDEVR/String.h>
#include <QMenu>
#include <QAction>
namespace NDEVR
{
    class CustomTreeWidgets;
    class SceneTree;
    class CustomModelSorter;
    class QTWindowManager;
    class NDVSetupCallback;
    class CustomDesignObjectDialogSetupCallback;
    class CustomModelLogic 
    {
    //Override the below to add custom program behavior
    protected:
        virtual void setupUI(WindowManager*) {}
        virtual void addMenuItems(Model&, QTModelManager*, QMenu&, const Buffer<UUID>&, const SelectionInfo&) {}
        virtual void executeMenuItems(Model&, QTModelManager*, QAction*, const Buffer<UUID>&, const SelectionInfo&, PopupInfo) {}
        virtual void setupPropertiesDialog(DesignObjectDialog*, const DynamicPointer<QTDesignObject>&) {}
        virtual void setupModel(Model&) {}
        virtual void setTreeWidgetsForSelection(const Model&, Buffer<QWidget*>&, SceneTree*) {};
        NDEVR_API void setSorter(const std::function<bool(const Model& a, const Model& b)>& sort_function);
        virtual bool canBeHandledBySelectionWidget() const { return true; };
    protected:
        friend class CustomModelContextMenu;
        friend class CustomModelTreeWidgets;
        friend class CustomModelSorter;
        friend class CustomDesignObjectDialogSetupCallback;
        NDEVR_API CustomModelLogic(QTModelManager* manager, const StringView& type_name);
        NDEVR_API CustomModelLogic(QTModelManager* manager, const Buffer<String>& type_names);
        NDEVR_API virtual ~CustomModelLogic();
        NDEVR_API CustomTreeWidgets* treeWidgetCallback();
        NDEVR_API ContextMenuCallback* contextMenuCallback();
        NDEVR_API CustomDesignObjectDialogSetupCallback* dialogSetupCallback();
        NDEVR_API NDVSetupCallback* ndvFileReadLogic();
    public:
        NDEVR_API virtual void registerAll();
        NDEVR_API virtual const UUID& id() const;
        NDEVR_API const Buffer<String>& typeNames() const;
    protected:
        QTModelManager* m_manager; 
        CustomTreeWidgets* m_tree_button_menu = nullptr; 
        ContextMenuCallback* m_custom_context_menu = nullptr; 
        CustomModelSorter* m_sorter = nullptr; 
        CustomDesignObjectDialogSetupCallback* m_dialog_callback = nullptr; 
        NDVSetupCallback* m_ndv_setup_callback = nullptr; 
        Buffer<String> m_type_names; 
        UUID m_id; 
    };
}