RawDocumentTreeModel.moc.h

#pragma once
 
#include <EditorFramework/EditorFrameworkDLL.h>
 
#include <QAbstractItemModel>
#include <ToolsFoundation/Object/DocumentObjectManager.h>
 
class plDragDropInfo;
 
class PL_EDITORFRAMEWORK_DLL plQtDocumentTreeModelAdapter : public QObject
{
  Q_OBJECT;
 
public:
  plQtDocumentTreeModelAdapter(const plDocumentObjectManager* pTree, const plRTTI* pType, const char* szChildProperty);
  virtual const plRTTI* GetType() const;
  virtual const plString& GetChildProperty() const;
 
  virtual QVariant data(const plDocumentObject* pObject, int iRow, int iColumn, int iRole = Qt::DisplayRole) const = 0;
  virtual bool setData(const plDocumentObject* pObject, int iRow, int iColumn, const QVariant& value, int iRole) const;
  virtual Qt::ItemFlags flags(const plDocumentObject* pObject, int iRow, int iColumn) const;
 
Q_SIGNALS:
  void dataChanged(const plDocumentObject* pObject, QVector<int> roles);
 
protected:
  const plDocumentObjectManager* m_pTree = nullptr;
  const plRTTI* m_pType = nullptr;
  plString m_sChildProperty;
};
 
class PL_EDITORFRAMEWORK_DLL plQtDummyAdapter : public plQtDocumentTreeModelAdapter
{
  Q_OBJECT;
 
public:
  plQtDummyAdapter(const plDocumentObjectManager* pTree, const plRTTI* pType, const char* szChildProperty);
 
  virtual QVariant data(const plDocumentObject* pObject, int iRow, int iColumn, int iRole) const override;
};
 
class PL_EDITORFRAMEWORK_DLL plQtNamedAdapter : public plQtDocumentTreeModelAdapter
{
  Q_OBJECT;
 
public:
  plQtNamedAdapter(const plDocumentObjectManager* pTree, const plRTTI* pType, const char* szChildProperty, const char* szNameProperty);
  ~plQtNamedAdapter();
  virtual QVariant data(const plDocumentObject* pObject, int iRow, int iColumn, int iRole) const override;
 
protected:
  virtual void TreePropertyEventHandler(const plDocumentObjectPropertyEvent& e);
 
protected:
  plString m_sNameProperty;
};
 
class PL_EDITORFRAMEWORK_DLL plQtNameableAdapter : public plQtNamedAdapter
{
  Q_OBJECT;
 
public:
  plQtNameableAdapter(const plDocumentObjectManager* pTree, const plRTTI* pType, const char* szChildProperty, const char* szNameProperty);
  ~plQtNameableAdapter();
  virtual bool setData(const plDocumentObject* pObject, int iRow, int iColumn, const QVariant& value, int iRole) const override;
  virtual Qt::ItemFlags flags(const plDocumentObject* pObject, int iRow, int iColumn) const override;
};
 
class PL_EDITORFRAMEWORK_DLL plQtDocumentTreeModel : public QAbstractItemModel
{
public:
  plQtDocumentTreeModel(const plDocumentObjectManager* pTree, const plUuid& root = plUuid());
  ~plQtDocumentTreeModel();
 
  const plDocumentObjectManager* GetDocumentTree() const { return m_pDocumentTree; }
  void AddAdapter(plQtDocumentTreeModelAdapter* pAdapter);
  QModelIndex ComputeModelIndex(const plDocumentObject* pObject) const;
 
  void SetAllowDragDrop(bool bAllow);
 
  static bool MoveObjects(const plDragDropInfo& info);
 
public: // QAbstractItemModel
  virtual QModelIndex index(int iRow, int iColumn, const QModelIndex& parent = QModelIndex()) const override;
  virtual QModelIndex parent(const QModelIndex& child) 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 QVariant data(const QModelIndex& index, int iRole = Qt::DisplayRole) const override;
  virtual bool setData(const QModelIndex& index, const QVariant& value, int iRole) override;
 
  virtual Qt::DropActions supportedDropActions() const override;
 
  virtual bool canDropMimeData(const QMimeData* pData, Qt::DropAction action, int iRow, int iColumn, const QModelIndex& parent) const override;
  virtual bool dropMimeData(const QMimeData* pData, Qt::DropAction action, int iRow, int iColumn, const QModelIndex& parent) override;
  virtual QStringList mimeTypes() const override;
  virtual QMimeData* mimeData(const QModelIndexList& indexes) const override;
 
protected:
  virtual void TreeEventHandler(const plDocumentObjectStructureEvent& e);
 
private:
  QModelIndex ComputeParent(const plDocumentObject* pObject) const;
  plInt32 ComputeIndex(const plDocumentObject* pObject) const;
  const plDocumentObject* GetRoot() const;
  bool IsUnderRoot(const plDocumentObject* pObject) const;
 
  const plQtDocumentTreeModelAdapter* GetAdapter(const plRTTI* pType) const;
 
protected:
  const plDocumentObjectManager* m_pDocumentTree = nullptr;
  const plUuid m_Root;
  plHashTable<const plRTTI*, plQtDocumentTreeModelAdapter*> m_Adapters;
  bool m_bAllowDragDrop = false;
  plString m_sTargetContext = "scenetree";
};