TreeScopeAdopter.cpp [plain text]
#include "config.h"
#include "TreeScopeAdopter.h"
#include "Attr.h"
#include "ElementRareData.h"
#include "NodeTraversal.h"
namespace WebCore {
void TreeScopeAdopter::moveTreeToNewScope(Node* root) const
{
ASSERT(needsScopeChange());
Document& oldDocument = m_oldScope.documentScope();
Document& newDocument = m_newScope.documentScope();
bool willMoveToNewDocument = &oldDocument != &newDocument;
if (willMoveToNewDocument) {
oldDocument.incrementReferencingNodeCount();
oldDocument.incDOMTreeVersion();
}
for (Node* node = root; node; node = NodeTraversal::next(node, root)) {
updateTreeScope(node);
if (willMoveToNewDocument)
moveNodeToNewDocument(node, &oldDocument, &newDocument);
else if (node->hasRareData()) {
NodeRareData* rareData = node->rareData();
if (rareData->nodeLists())
rareData->nodeLists()->adoptTreeScope();
}
if (!node->isElementNode())
continue;
if (node->hasSyntheticAttrChildNodes()) {
const Vector<RefPtr<Attr>>& attrs = toElement(node)->attrNodeList();
for (unsigned i = 0; i < attrs.size(); ++i)
moveTreeToNewScope(attrs[i].get());
}
if (ShadowRoot* shadow = node->shadowRoot()) {
shadow->setParentTreeScope(&m_newScope);
if (willMoveToNewDocument)
moveShadowTreeToNewDocument(shadow, &oldDocument, &newDocument);
}
}
if (willMoveToNewDocument)
oldDocument.decrementReferencingNodeCount();
}
void TreeScopeAdopter::moveShadowTreeToNewDocument(ShadowRoot* shadowRoot, Document* oldDocument, Document* newDocument) const
{
for (Node* node = shadowRoot; node; node = NodeTraversal::next(node, shadowRoot)) {
moveNodeToNewDocument(node, oldDocument, newDocument);
if (ShadowRoot* shadow = node->shadowRoot())
moveShadowTreeToNewDocument(shadow, oldDocument, newDocument);
}
}
#ifndef NDEBUG
static bool didMoveToNewDocumentWasCalled = false;
static Document* oldDocumentDidMoveToNewDocumentWasCalledWith = 0;
void TreeScopeAdopter::ensureDidMoveToNewDocumentWasCalled(Document* oldDocument)
{
ASSERT(!didMoveToNewDocumentWasCalled);
ASSERT_UNUSED(oldDocument, oldDocument == oldDocumentDidMoveToNewDocumentWasCalledWith);
didMoveToNewDocumentWasCalled = true;
}
#endif
inline void TreeScopeAdopter::updateTreeScope(Node* node) const
{
ASSERT(!node->isTreeScope());
ASSERT(&node->treeScope() == &m_oldScope);
node->setTreeScope(m_newScope);
}
inline void TreeScopeAdopter::moveNodeToNewDocument(Node* node, Document* oldDocument, Document* newDocument) const
{
ASSERT(!node->inDocument() || oldDocument != newDocument);
newDocument->incrementReferencingNodeCount();
oldDocument->decrementReferencingNodeCount();
if (node->hasRareData()) {
NodeRareData* rareData = node->rareData();
if (rareData->nodeLists())
rareData->nodeLists()->adoptDocument(oldDocument, newDocument);
}
if (oldDocument)
oldDocument->moveNodeIteratorsToNewDocument(node, newDocument);
if (node->isShadowRoot())
toShadowRoot(node)->setDocumentScope(newDocument);
#ifndef NDEBUG
didMoveToNewDocumentWasCalled = false;
oldDocumentDidMoveToNewDocumentWasCalledWith = oldDocument;
#endif
node->didMoveToNewDocument(oldDocument);
ASSERT(didMoveToNewDocumentWasCalled);
}
}