TreeScopeAdopter.cpp [plain text]
#include "config.h"
#include "TreeScopeAdopter.h"
#include "Attr.h"
#include "Document.h"
#include "ElementRareData.h"
#include "ElementShadow.h"
#include "NodeRareData.h"
#include "NodeTraversal.h"
#include "RenderStyle.h"
#include "ShadowRoot.h"
namespace WebCore {
void TreeScopeAdopter::moveTreeToNewScope(Node* root) const
{
ASSERT(needsScopeChange());
m_oldScope->guardRef();
Document* oldDocument = m_oldScope->documentScope();
Document* newDocument = m_newScope->documentScope();
bool willMoveToNewDocument = oldDocument != newDocument;
if (oldDocument && willMoveToNewDocument)
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)
moveTreeToNewDocument(shadow, oldDocument, newDocument);
}
}
m_oldScope->guardDeref();
}
void TreeScopeAdopter::moveTreeToNewDocument(Node* root, Document* oldDocument, Document* newDocument) const
{
for (Node* node = root; node; node = NodeTraversal::next(node, root)) {
moveNodeToNewDocument(node, oldDocument, newDocument);
if (ShadowRoot* shadow = node->shadowRoot())
moveTreeToNewDocument(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);
m_newScope->guardRef();
m_oldScope->guardDeref();
node->setTreeScope(m_newScope);
}
inline void TreeScopeAdopter::moveNodeToNewDocument(Node* node, Document* oldDocument, Document* newDocument) const
{
ASSERT(!node->inDocument() || oldDocument != newDocument);
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);
}
}