SelectionManager.h

#pragma once
 
#include <ToolsFoundation/Object/DocumentObjectBase.h>
#include <ToolsFoundation/ToolsFoundationDLL.h>
 
class plDocument;
struct plDocumentObjectStructureEvent;
 
struct plSelectionManagerEvent
{
  enum class Type
  {
    SelectionCleared,
    SelectionSet,
    ObjectAdded,
    ObjectRemoved,
  };
 
  Type m_Type;
  const plDocument* m_pDocument;
  const plDocumentObject* m_pObject;
};
 
class PL_TOOLSFOUNDATION_DLL plSelectionManager
{
public:
  plCopyOnBroadcastEvent<const plSelectionManagerEvent&> m_Events;
 
  // \brief Storage for the selection so it can be swapped when using multiple sub documents.
  class Storage : public plRefCounted
  {
  public:
    plDeque<const plDocumentObject*> m_SelectionList;
    plSet<plUuid> m_SelectionSet;
    const plDocumentObjectManager* m_pObjectManager = nullptr;
    plCopyOnBroadcastEvent<const plSelectionManagerEvent&> m_Events;
  };
 
public:
  plSelectionManager(const plDocumentObjectManager* pObjectManager);
  ~plSelectionManager();
 
  void Clear();
  void AddObject(const plDocumentObject* pObject);
  void RemoveObject(const plDocumentObject* pObject, bool bRecurseChildren = false);
  void SetSelection(const plDocumentObject* pSingleObject);
  void SetSelection(const plDeque<const plDocumentObject*>& selection);
  void ToggleObject(const plDocumentObject* pObject);
 
  const plDocumentObject* GetCurrentObject() const;
 
  const plDeque<const plDocumentObject*>& GetSelection() const { return m_pSelectionStorage->m_SelectionList; }
 
  bool IsSelectionEmpty() const { return m_pSelectionStorage->m_SelectionList.IsEmpty(); }
 
  const plDeque<const plDocumentObject*> GetTopLevelSelection() const;
 
  const plDeque<const plDocumentObject*> GetTopLevelSelection(const plRTTI* pBase) const;
 
  bool IsSelected(const plDocumentObject* pObject) const;
  bool IsParentSelected(const plDocumentObject* pObject) const;
 
  const plDocument* GetDocument() const;
 
  plSharedPtr<plSelectionManager::Storage> SwapStorage(plSharedPtr<plSelectionManager::Storage> pNewStorage);
  plSharedPtr<plSelectionManager::Storage> GetStorage() { return m_pSelectionStorage; }
 
private:
  void TreeEventHandler(const plDocumentObjectStructureEvent& e);
  bool RecursiveRemoveFromSelection(const plDocumentObject* pObject);
 
  friend class plDocument;
 
  plSharedPtr<plSelectionManager::Storage> m_pSelectionStorage;
 
  plCopyOnBroadcastEvent<const plDocumentObjectStructureEvent&>::Unsubscriber m_ObjectStructureUnsubscriber;
  plCopyOnBroadcastEvent<const plSelectionManagerEvent&>::Unsubscriber m_EventsUnsubscriber;
};