Code/Editor/EditorFramework/Assets/AssetBrowserFolderView.moc.h - dev - redgpu/ezEngine

Code/Editor/EditorFramework/Assets/AssetBrowserFolderView.moc.h
117 строк5 KB
Sanakan8472
Fix Editor Memory Leak (#1794)
27 янв 2026, 10:02
27 янв 2026, 10:02eb3a654
#pragma once

#include <EditorFramework/EditorFrameworkDLL.h>
#include <Foundation/Types/SharedPtr.h>
#include <ToolsFoundation/FileSystem/FileSystemModel.h>
#include <ToolsFoundation/Project/ToolsProject.h>

#include <QItemDelegate>
#include <QTreeWidget>
#include <QValidator>

class ezQtAssetBrowserFilter;

/// \brief Basic file name validator. Makes sure that under a given parent folder, the new file name is valid and not already in use by a different file.
class ezFileNameValidator : public QValidator
{
public:
  /// \brief Constructor. Validator requires the current location and name of the file.
  /// \param sParentFolder Absolute path to the location of the file.
  /// \param sCurrentName Current filename. If set, this name is marked as valid, even though it is already in use.
  ezFileNameValidator(QObject* pParent, ezStringView sParentFolder, ezStringView sCurrentName);
  virtual QValidator::State validate(QString& ref_sInput, int& ref_iPos) const override;

private:
  ezString m_sParentFolder;
  ezString m_sCurrentName;
};

/// \brief Custom delegate for the eqQtAssetBrowserFolderView to enable renaming folders. Does not do any model modifications. Instead, it fires editingFinished when the delegate editor closes.
class ezFolderNameDelegate : public QItemDelegate
{
  Q_OBJECT

public:
  ezFolderNameDelegate(QObject* pParent = nullptr);

  virtual QWidget* createEditor(QWidget* pParent, const QStyleOptionViewItem& option, const QModelIndex& index) const override;
  virtual void setModelData(QWidget* pEditor, QAbstractItemModel* pModel, const QModelIndex& index) const override;

signals:
  void editingFinished(const QString& sAbsPath, const QString& sNewName) const;
};

/// \brief Folder tree of the asset browser to allow filtering by folder.
///
/// This class keeps up to date with the folder structure in ezFileSystemModel. Events from ezFileSystemModel are cached ans flushed via OnFlushFileSystemEvents.
/// Folder movement, creation and deletion is supported and handled by this class. The context menu is implemented in ezQtAssetBrowserWidget as it requires a global context of the asset browser instance.
class eqQtAssetBrowserFolderView : public QTreeWidget
{
  Q_OBJECT
public:
  eqQtAssetBrowserFolderView(QWidget* pParent);
  ~eqQtAssetBrowserFolderView();

  /// \brief Required to be set right after the ctor. This class will call ezQtAssetBrowserFilter::SetPathFilter whenever the current selected item changes.
  void SetFilter(ezQtAssetBrowserFilter* pFilter);
  /// \brief In dialog mode, any modifications (folder movement, creation and deletion) are disabled.
  void SetDialogMode(bool bDialogMode);

  virtual void mouseDoubleClickEvent(QMouseEvent* e) override;
  virtual void mousePressEvent(QMouseEvent* e) override;

public Q_SLOTS:
  /// \brief Creates a new folder under the current selected item and enters edit mode to allow the user to rename it.
  void NewFolder();
  /// \brief Opens the current selected item in the windows explorer or OS equivalent.
  void TreeOpenExplorer();
  /// \brief Deletes the currently selected folder after confirmation.
  void DeleteFolder();

private Q_SLOTS:
  void OnFolderEditingFinished(const QString& sAbsPath, const QString& sNewName);
  void OnFlushFileSystemEvents();
  void OnItemSelectionChanged();
  void OnPathFilterChanged();

protected:
  virtual void dragMoveEvent(QDragMoveEvent* e) override;
  virtual void mouseMoveEvent(QMouseEvent* e) override;
  virtual void dropEvent(QDropEvent* event) override;
  virtual Qt::DropActions supportedDropActions() const override;
  ezStatus canDrop(QDropEvent* e, ezDynamicArray<ezString>& out_files, ezString& out_sTargetFolder);
  virtual QStringList mimeTypes() const override;
  virtual QMimeData* mimeData(const QList<QTreeWidgetItem*>& items) const override;
  virtual void keyPressEvent(QKeyEvent* e) override;

private:
  bool SelectPathFilter(QTreeWidgetItem* pParent, const QString& sPath);
  void UpdateDirectoryTree();
  void ClearDirectoryTree();
  void BuildDirectoryTree(const ezDataDirPath& path, ezStringView sCurPath, QTreeWidgetItem* pParent, ezStringView sCurPathToItem, bool bIsHidden);
  void RemoveDirectoryTreeItem(ezStringView sCurPath, QTreeWidgetItem* pParent, ezStringView sCurPathToItem);
  QTreeWidgetItem* FindDirectoryTreeItem(ezStringView sCurPath, QTreeWidgetItem* pParent, ezStringView sCurPathToItem);
  void ProjectEventHandler(const ezToolsProjectEvent& e);

private:
  // Shared object as the event handler can be called after this object is destroyed due to the use of ezCopyOnBroadcastEvent in ezFileSystemModel.
  // This is the only way to prevent race conditions and allow safe add/remove calls to multithreaded event broadcasters.
  class QueuedFolderEvents : public ezRefCounted
  {
  public:
    void FileSystemModelFolderEventHandler(const ezFolderChangedEvent& e);

    ezMutex m_FolderStructureMutex;
    eqQtAssetBrowserFolderView* m_pParent = nullptr;
    ezHybridArray<ezFolderChangedEvent, 2> m_Events;
  };

  bool m_bDialogMode = false;
  ezUInt32 m_uiKnownAssetFolderCount = 0;
  bool m_bTreeSelectionChangeInProgress = false;

  ezQtAssetBrowserFilter* m_pFilter = nullptr;
  ezSharedPtr<QueuedFolderEvents> m_pFolderEvents;

  ezEventSubscriptionID m_FolderChangedSubscription = 0;
};