AXIsolatedTree.cpp [plain text]
#include "config.h"
#if ENABLE(ACCESSIBILITY_ISOLATED_TREE)
#include "AXIsolatedTree.h"
#include "AXIsolatedTreeNode.h"
#include "Page.h"
#include <wtf/NeverDestroyed.h>
namespace WebCore {
static Lock s_cacheLock;
static unsigned newTreeID()
{
static unsigned s_currentTreeID = 0;
return ++s_currentTreeID;
}
HashMap<PageIdentifier, Ref<AXIsolatedTree>>& AXIsolatedTree::treePageCache()
{
static NeverDestroyed<HashMap<PageIdentifier, Ref<AXIsolatedTree>>> map;
return map;
}
HashMap<AXIsolatedTreeID, Ref<AXIsolatedTree>>& AXIsolatedTree::treeIDCache()
{
static NeverDestroyed<HashMap<AXIsolatedTreeID, Ref<AXIsolatedTree>>> map;
return map;
}
AXIsolatedTree::AXIsolatedTree()
: m_treeID(newTreeID())
{
}
AXIsolatedTree::~AXIsolatedTree() = default;
Ref<AXIsolatedTree> AXIsolatedTree::create()
{
ASSERT(isMainThread());
return adoptRef(*new AXIsolatedTree());
}
RefPtr<AXIsolatedObject> AXIsolatedTree::nodeInTreeForID(AXIsolatedTreeID treeID, AXID axID)
{
return treeForID(treeID)->nodeForID(axID);
}
RefPtr<AXIsolatedTree> AXIsolatedTree::treeForID(AXIsolatedTreeID treeID)
{
return treeIDCache().get(treeID);
}
Ref<AXIsolatedTree> AXIsolatedTree::createTreeForPageID(PageIdentifier pageID)
{
LockHolder locker(s_cacheLock);
auto newTree = AXIsolatedTree::create();
treePageCache().set(pageID, newTree.copyRef());
treeIDCache().set(newTree->treeIdentifier(), newTree.copyRef());
return newTree;
}
RefPtr<AXIsolatedTree> AXIsolatedTree::treeForPageID(PageIdentifier pageID)
{
LockHolder locker(s_cacheLock);
if (auto tree = treePageCache().get(pageID))
return makeRefPtr(tree);
return nullptr;
}
RefPtr<AXIsolatedObject> AXIsolatedTree::nodeForID(AXID axID) const
{
if (!axID)
return nullptr;
return m_readerThreadNodeMap.get(axID);
}
RefPtr<AXIsolatedObject> AXIsolatedTree::focusedUIElement()
{
m_focusedNodeID = m_pendingFocusedNodeID;
return nodeForID(m_focusedNodeID);
}
RefPtr<AXIsolatedObject> AXIsolatedTree::rootNode()
{
return nodeForID(m_rootNodeID);
}
void AXIsolatedTree::setRootNode(Ref<AXIsolatedObject>& root)
{
LockHolder locker { m_changeLogLock };
m_rootNodeID = root->objectID();
m_readerThreadNodeMap.add(root->objectID(), WTFMove(root));
}
void AXIsolatedTree::setFocusedNodeID(AXID axID)
{
LockHolder locker { m_changeLogLock };
m_pendingFocusedNodeID = axID;
}
void AXIsolatedTree::removeNode(AXID axID)
{
LockHolder locker { m_changeLogLock };
m_pendingRemovals.append(axID);
}
void AXIsolatedTree::appendNodeChanges(Vector<Ref<AXIsolatedObject>>& log)
{
LockHolder locker { m_changeLogLock };
for (auto& node : log)
m_pendingAppends.append(node.copyRef());
}
void AXIsolatedTree::applyPendingChanges()
{
RELEASE_ASSERT(!isMainThread());
LockHolder locker { m_changeLogLock };
Vector<Ref<AXIsolatedObject>> appendCopy;
std::swap(appendCopy, m_pendingAppends);
Vector<AXID> removeCopy({ WTFMove(m_pendingRemovals) });
locker.unlockEarly();
m_focusedNodeID = m_pendingFocusedNodeID;
for (auto& item : appendCopy)
m_readerThreadNodeMap.add(item->objectID(), WTFMove(item));
for (auto item : removeCopy)
m_readerThreadNodeMap.remove(item);
}
}
#endif // ENABLE(ACCESSIBILITY_ISOLATED_TREE)