#pragma once
#if ENABLE(ACCESSIBILITY_ISOLATED_TREE)
#include "AccessibilityObjectInterface.h"
#include "PageIdentifier.h"
#include <wtf/HashMap.h>
#include <wtf/RefPtr.h>
#include <wtf/ThreadSafeRefCounted.h>
namespace WTF {
class TextStream;
}
namespace WebCore {
class AXIsolatedObject;
class AXObjectCache;
class Page;
typedef unsigned AXIsolatedTreeID;
class AXIsolatedTree : public ThreadSafeRefCounted<AXIsolatedTree> {
WTF_MAKE_NONCOPYABLE(AXIsolatedTree); WTF_MAKE_FAST_ALLOCATED;
friend WTF::TextStream& operator<<(WTF::TextStream&, AXIsolatedTree&);
public:
static Ref<AXIsolatedTree> create();
virtual ~AXIsolatedTree();
static Ref<AXIsolatedTree> createTreeForPageID(PageIdentifier);
static void removeTreeForPageID(PageIdentifier);
static RefPtr<AXIsolatedTree> treeForPageID(PageIdentifier);
static RefPtr<AXIsolatedTree> treeForID(AXIsolatedTreeID);
AXObjectCache* axObjectCache() const { return m_axObjectCache; }
void setAXObjectCache(AXObjectCache* axObjectCache) { m_axObjectCache = axObjectCache; }
RefPtr<AXIsolatedObject> rootNode();
RefPtr<AXIsolatedObject> focusedNode();
RefPtr<AXIsolatedObject> nodeForID(AXID) const;
Vector<RefPtr<AXCoreObject>> objectsForIDs(Vector<AXID>) const;
struct NodeChange {
Ref<AXIsolatedObject> m_isolatedObject;
RetainPtr<AccessibilityObjectWrapper> m_wrapper;
NodeChange(AXIsolatedObject&, AccessibilityObjectWrapper*);
NodeChange(const NodeChange&);
};
void generateSubtree(AXCoreObject&, AXCoreObject*, bool attachWrapper);
void updateNode(AXCoreObject&);
void updateSubtree(AXCoreObject&);
void updateChildren(AXCoreObject&);
void removeNode(AXID);
void removeSubtree(AXID);
void setRootNode(AXIsolatedObject*);
void setFocusedNodeID(AXID);
void applyPendingChanges();
AXIsolatedTreeID treeID() const { return m_treeID; }
private:
AXIsolatedTree();
void clear();
static HashMap<AXIsolatedTreeID, Ref<AXIsolatedTree>>& treeIDCache();
static HashMap<PageIdentifier, Ref<AXIsolatedTree>>& treePageCache();
Ref<AXIsolatedObject> createSubtree(AXCoreObject&, AXID parentID, bool attachWrapper, Vector<NodeChange>&);
void appendNodeChanges(Vector<NodeChange>&&);
void updateChildrenIDs(AXID parentID, Vector<AXID>&& childrenIDs);
AXIsolatedTreeID m_treeID;
AXObjectCache* m_axObjectCache { nullptr };
HashMap<AXID, Vector<AXID>> m_nodeMap;
HashMap<AXID, Ref<AXIsolatedObject>> m_readerThreadNodeMap;
RefPtr<AXIsolatedObject> m_rootNode;
Vector<NodeChange> m_pendingAppends; Vector<AXID> m_pendingNodeRemovals; Vector<AXID> m_pendingSubtreeRemovals; Vector<std::pair<AXID, Vector<AXID>>> m_pendingChildrenUpdates;
AXID m_pendingFocusedNodeID { InvalidAXID };
AXID m_focusedNodeID { InvalidAXID };
Lock m_changeLogLock;
};
}
#endif