#include "config.h"
#include "NodeList.h"
#include "Document.h"
#include "Element.h"
namespace WebCore {
NodeList::NodeList(PassRefPtr<Node> _rootNode)
: rootNode(_rootNode),
isLengthCacheValid(false),
isItemCacheValid(false)
{
rootNode->registerNodeList(this);
}
NodeList::~NodeList()
{
rootNode->unregisterNodeList(this);
}
unsigned NodeList::recursiveLength(Node* start) const
{
if (!start)
start = rootNode.get();
if (isLengthCacheValid && start == rootNode)
return cachedLength;
unsigned len = 0;
for (Node* n = start->firstChild(); n; n = n->nextSibling())
if (n->isElementNode()) {
if (nodeMatches(n))
len++;
len += recursiveLength(n);
}
if (start == rootNode) {
cachedLength = len;
isLengthCacheValid = true;
}
return len;
}
Node* NodeList::itemForwardsFromCurrent(Node* start, unsigned offset, int remainingOffset) const
{
ASSERT(remainingOffset >= 0);
for (Node *n = start; n; n = n->traverseNextNode(rootNode.get())) {
if (n->isElementNode()) {
if (nodeMatches(n)) {
if (!remainingOffset) {
lastItem = n;
lastItemOffset = offset;
isItemCacheValid = true;
return n;
}
remainingOffset--;
}
}
}
return 0; }
Node* NodeList::itemBackwardsFromCurrent(Node* start, unsigned offset, int remainingOffset) const
{
ASSERT(remainingOffset < 0);
for (Node *n = start; n; n = n->traversePreviousNode(rootNode.get())) {
if (n->isElementNode()) {
if (nodeMatches(n)) {
if (!remainingOffset) {
lastItem = n;
lastItemOffset = offset;
isItemCacheValid = true;
return n;
}
remainingOffset++;
}
}
}
return 0; }
Node* NodeList::recursiveItem(unsigned offset, Node* start) const
{
int remainingOffset = offset;
if (!start) {
start = rootNode->firstChild();
if (isItemCacheValid) {
if (offset == lastItemOffset) {
return lastItem;
} else if (offset > lastItemOffset || lastItemOffset - offset < offset) {
start = lastItem;
remainingOffset -= lastItemOffset;
}
}
}
if (remainingOffset < 0)
return itemBackwardsFromCurrent(start, offset, remainingOffset);
else
return itemForwardsFromCurrent(start, offset, remainingOffset);
}
Node* NodeList::itemWithName(const AtomicString& elementId) const
{
if (rootNode->isDocumentNode() || rootNode->inDocument()) {
Node* node = rootNode->document()->getElementById(elementId);
if (!node || !nodeMatches(node))
return 0;
for (Node* p = node->parentNode(); p; p = p->parentNode())
if (p == rootNode)
return node;
return 0;
}
unsigned l = length();
for (unsigned i = 0; i < l; i++) {
Node* node = item(i);
if (node->isElementNode() && static_cast<Element*>(node)->getIDAttribute() == elementId)
return node;
}
return 0;
}
void NodeList::rootNodeChildrenChanged()
{
isLengthCacheValid = false;
isItemCacheValid = false;
lastItem = 0;
}
}