#include "config.h"
#include "TextIterator.h"
#include "CharacterNames.h"
#include "Document.h"
#include "Element.h"
#include "HTMLNames.h"
#include "htmlediting.h"
#include "InlineTextBox.h"
#include "Position.h"
#include "Range.h"
#include "RenderTableCell.h"
#include "RenderTableRow.h"
#include "visible_units.h"
using namespace std;
using namespace WTF::Unicode;
namespace WebCore {
using namespace HTMLNames;
class CircularSearchBuffer : Noncopyable {
public:
CircularSearchBuffer(const String& target, bool isCaseSensitive);
void clear() { m_cursor = 0; m_isBufferFull = false; }
void append(UChar);
bool isMatch() const;
unsigned length() const;
private:
void append(UChar, bool isCharacterStart);
String m_target;
bool m_isCaseSensitive;
Vector<UChar> m_characterBuffer;
Vector<bool> m_isCharacterStartBuffer;
bool m_isBufferFull;
unsigned m_cursor;
};
TextIterator::TextIterator() : m_startContainer(0), m_startOffset(0), m_endContainer(0), m_endOffset(0), m_positionNode(0), m_lastCharacter(0)
{
}
TextIterator::TextIterator(const Range* r, bool emitCharactersBetweenAllVisiblePositions)
: m_startContainer(0)
, m_startOffset(0)
, m_endContainer(0)
, m_endOffset(0)
, m_positionNode(0)
, m_emitCharactersBetweenAllVisiblePositions(emitCharactersBetweenAllVisiblePositions)
{
if (!r)
return;
ExceptionCode ec = 0;
Node *startContainer = r->startContainer(ec);
int startOffset = r->startOffset(ec);
Node *endContainer = r->endContainer(ec);
int endOffset = r->endOffset(ec);
if (ec)
return;
ASSERT(r->boundaryPointsValid());
m_startContainer = startContainer;
m_startOffset = startOffset;
m_endContainer = endContainer;
m_endOffset = endOffset;
m_node = r->startNode();
if (m_node == 0)
return;
m_offset = m_node == m_startContainer ? m_startOffset : 0;
m_handledNode = false;
m_handledChildren = false;
m_pastEndNode = r->pastEndNode();
m_needAnotherNewline = false;
m_textBox = 0;
m_haveEmitted = false;
m_lastTextNode = 0;
m_lastTextNodeEndedWithCollapsedSpace = false;
m_lastCharacter = 0;
#ifndef NDEBUG
m_positionNode = m_node;
#endif
advance();
}
void TextIterator::advance()
{
m_positionNode = 0;
m_textLength = 0;
if (m_needAnotherNewline) {
emitCharacter('\n', m_node->parentNode(), m_node, 1, 1);
m_needAnotherNewline = false;
return;
}
if (m_textBox) {
handleTextBox();
if (m_positionNode)
return;
}
while (m_node && m_node != m_pastEndNode) {
if (m_node == m_endContainer && m_endOffset == 0) {
representNodeOffsetZero();
m_node = 0;
return;
}
RenderObject *renderer = m_node->renderer();
if (!renderer) {
m_handledNode = true;
m_handledChildren = true;
} else {
if (!m_handledNode) {
if (renderer->isText() && m_node->nodeType() == Node::TEXT_NODE) m_handledNode = handleTextNode();
else if (renderer && (renderer->isImage() || renderer->isWidget() || (renderer->element() && renderer->element()->isControl())))
m_handledNode = handleReplacedElement();
else
m_handledNode = handleNonTextNode();
if (m_positionNode)
return;
}
}
Node *next = m_handledChildren ? 0 : m_node->firstChild();
m_offset = 0;
if (!next) {
next = m_node->nextSibling();
if (!next) {
bool pastEnd = m_node->traverseNextNode() == m_pastEndNode;
while (!next && m_node->parentNode()) {
if (pastEnd && m_node->parentNode() == m_endContainer || m_endContainer->isDescendantOf(m_node->parentNode()))
return;
bool haveRenderer = m_node->renderer();
m_node = m_node->parentNode();
if (haveRenderer)
exitNode();
if (m_positionNode) {
m_handledNode = true;
m_handledChildren = true;
return;
}
next = m_node->nextSibling();
}
}
}
m_node = next;
m_handledNode = false;
m_handledChildren = false;
if (m_positionNode)
return;
}
}
static inline bool compareBoxStart(const InlineTextBox *first, const InlineTextBox *second)
{
return first->start() < second->start();
}
bool TextIterator::handleTextNode()
{
RenderText* renderer = static_cast<RenderText*>(m_node->renderer());
if (renderer->style()->visibility() != VISIBLE)
return false;
m_lastTextNode = m_node;
String str = renderer->text();
if (!renderer->style()->collapseWhiteSpace()) {
int runStart = m_offset;
if (m_lastTextNodeEndedWithCollapsedSpace) {
emitCharacter(' ', m_node, 0, runStart, runStart);
return false;
}
int strLength = str.length();
int end = (m_node == m_endContainer) ? m_endOffset : INT_MAX;
int runEnd = min(strLength, end);
if (runStart >= runEnd)
return true;
emitText(m_node, runStart, runEnd);
return true;
}
if (!renderer->firstTextBox() && str.length() > 0) {
m_lastTextNodeEndedWithCollapsedSpace = true; return true;
}
if (renderer->containsReversedText()) {
m_sortedTextBoxes.clear();
for (InlineTextBox * textBox = renderer->firstTextBox(); textBox; textBox = textBox->nextTextBox()) {
m_sortedTextBoxes.append(textBox);
}
std::sort(m_sortedTextBoxes.begin(), m_sortedTextBoxes.end(), compareBoxStart);
m_sortedTextBoxesPosition = 0;
}
m_textBox = renderer->containsReversedText() ? m_sortedTextBoxes[0] : renderer->firstTextBox();
handleTextBox();
return true;
}
void TextIterator::handleTextBox()
{
RenderText *renderer = static_cast<RenderText *>(m_node->renderer());
String str = renderer->text();
int start = m_offset;
int end = (m_node == m_endContainer) ? m_endOffset : INT_MAX;
while (m_textBox) {
int textBoxStart = m_textBox->m_start;
int runStart = max(textBoxStart, start);
InlineTextBox *firstTextBox = renderer->containsReversedText() ? m_sortedTextBoxes[0] : renderer->firstTextBox();
bool needSpace = m_lastTextNodeEndedWithCollapsedSpace
|| (m_textBox == firstTextBox && textBoxStart == runStart && runStart > 0);
if (needSpace && !isCollapsibleWhitespace(m_lastCharacter) && m_lastCharacter) {
emitCharacter(' ', m_node, 0, runStart, runStart);
return;
}
int textBoxEnd = textBoxStart + m_textBox->m_len;
int runEnd = min(textBoxEnd, end);
InlineTextBox *nextTextBox = 0;
if (renderer->containsReversedText()) {
if (m_sortedTextBoxesPosition + 1 < m_sortedTextBoxes.size())
nextTextBox = m_sortedTextBoxes[m_sortedTextBoxesPosition + 1];
} else
nextTextBox = m_textBox->nextTextBox();
if (runStart < runEnd) {
if (str[runStart] == '\n') {
emitCharacter(' ', m_node, 0, runStart, runStart + 1);
m_offset = runStart + 1;
} else {
int subrunEnd = str.find('\n', runStart);
if (subrunEnd == -1 || subrunEnd > runEnd)
subrunEnd = runEnd;
m_offset = subrunEnd;
emitText(m_node, runStart, subrunEnd);
}
if (m_positionEndOffset < textBoxEnd)
return;
int nextRunStart = nextTextBox ? nextTextBox->m_start : str.length();
if (nextRunStart > runEnd)
m_lastTextNodeEndedWithCollapsedSpace = true; m_textBox = nextTextBox;
if (renderer->containsReversedText())
++m_sortedTextBoxesPosition;
return;
}
m_textBox = nextTextBox;
if (renderer->containsReversedText())
++m_sortedTextBoxesPosition;
}
}
bool TextIterator::handleReplacedElement()
{
if (m_node->renderer()->style()->visibility() != VISIBLE)
return false;
if (m_lastTextNodeEndedWithCollapsedSpace) {
emitCharacter(' ', m_lastTextNode->parentNode(), m_lastTextNode, 1, 1);
return false;
}
m_haveEmitted = true;
if (m_emitCharactersBetweenAllVisiblePositions) {
emitCharacter(',', m_node->parentNode(), m_node, 0, 1);
return true;
}
m_positionNode = m_node->parentNode();
m_positionOffsetBaseNode = m_node;
m_positionStartOffset = 0;
m_positionEndOffset = 1;
m_textCharacters = 0;
m_textLength = 0;
m_lastCharacter = 0;
return true;
}
static bool shouldEmitTabBeforeNode(Node* node)
{
RenderObject* r = node->renderer();
if (!r || !isTableCell(node))
return false;
RenderTableCell* rc = static_cast<RenderTableCell*>(r);
RenderTable* t = rc->table();
return t && (t->cellBefore(rc) || t->cellAbove(rc));
}
static bool shouldEmitNewlineForNode(Node* node)
{
RenderObject* r = node->renderer();
if (!r)
return node->hasTagName(brTag);
return r->isBR();
}
static bool shouldEmitNewlinesBeforeAndAfterNode(Node* node)
{
RenderObject* r = node->renderer();
if (!r) {
return (node->hasTagName(blockquoteTag)
|| node->hasTagName(ddTag)
|| node->hasTagName(divTag)
|| node->hasTagName(dlTag)
|| node->hasTagName(dtTag)
|| node->hasTagName(h1Tag)
|| node->hasTagName(h2Tag)
|| node->hasTagName(h3Tag)
|| node->hasTagName(h4Tag)
|| node->hasTagName(h5Tag)
|| node->hasTagName(h6Tag)
|| node->hasTagName(hrTag)
|| node->hasTagName(liTag)
|| node->hasTagName(listingTag)
|| node->hasTagName(olTag)
|| node->hasTagName(pTag)
|| node->hasTagName(preTag)
|| node->hasTagName(trTag)
|| node->hasTagName(ulTag));
}
if (isTableCell(node))
return false;
if (r->isTableRow()) {
RenderTable* t = static_cast<RenderTableRow*>(r)->table();
if (t && !t->isInline())
return true;
}
return !r->isInline() && r->isRenderBlock() && !r->isFloatingOrPositioned() && !r->isBody();
}
static bool shouldEmitNewlineAfterNode(Node* node)
{
if (!shouldEmitNewlinesBeforeAndAfterNode(node))
return false;
while ((node = node->traverseNextSibling()))
if (node->renderer())
return true;
return false;
}
static bool shouldEmitNewlineBeforeNode(Node* node)
{
return shouldEmitNewlinesBeforeAndAfterNode(node);
}
static bool shouldEmitExtraNewlineForNode(Node* node)
{
RenderObject* r = node->renderer();
if (!r)
return false;
if (node->hasTagName(h1Tag)
|| node->hasTagName(h2Tag)
|| node->hasTagName(h3Tag)
|| node->hasTagName(h4Tag)
|| node->hasTagName(h5Tag)
|| node->hasTagName(h6Tag)
|| node->hasTagName(pTag)) {
RenderStyle* style = r->style();
if (style) {
int bottomMargin = r->collapsedMarginBottom();
int fontSize = style->fontDescription().computedPixelSize();
if (bottomMargin * 2 >= fontSize)
return true;
}
}
return false;
}
bool TextIterator::shouldRepresentNodeOffsetZero()
{
if (m_emitCharactersBetweenAllVisiblePositions && m_node->renderer() && m_node->renderer()->isTable())
return true;
if (m_lastCharacter == '\n')
return false;
if (m_haveEmitted)
return true;
if (m_node == m_startContainer)
return false;
if (!m_node->isDescendantOf(m_startContainer))
return true;
if (m_startOffset == 0)
return false;
VisiblePosition startPos = VisiblePosition(m_startContainer, m_startOffset, DOWNSTREAM);
VisiblePosition currPos = VisiblePosition(m_node, 0, DOWNSTREAM);
return currPos.isNotNull() && !inSameLine(startPos, currPos);
}
bool TextIterator::shouldEmitSpaceBeforeAndAfterNode(Node* node)
{
return node->renderer() && node->renderer()->isTable() && (node->renderer()->isInline() || m_emitCharactersBetweenAllVisiblePositions);
}
void TextIterator::representNodeOffsetZero()
{
if (shouldEmitTabBeforeNode(m_node)) {
if (shouldRepresentNodeOffsetZero())
emitCharacter('\t', m_node->parentNode(), m_node, 0, 0);
} else if (shouldEmitNewlineBeforeNode(m_node)) {
if (shouldRepresentNodeOffsetZero())
emitCharacter('\n', m_node->parentNode(), m_node, 0, 0);
} else if (shouldEmitSpaceBeforeAndAfterNode(m_node)) {
if (shouldRepresentNodeOffsetZero())
emitCharacter(' ', m_node->parentNode(), m_node, 0, 0);
}
}
bool TextIterator::handleNonTextNode()
{
if (shouldEmitNewlineForNode(m_node))
emitCharacter('\n', m_node->parentNode(), m_node, 0, 1);
else if (m_emitCharactersBetweenAllVisiblePositions && m_node->renderer() && m_node->renderer()->isHR())
emitCharacter(' ', m_node->parentNode(), m_node, 0, 1);
else
representNodeOffsetZero();
return true;
}
void TextIterator::exitNode()
{
if (!m_haveEmitted)
return;
Node* baseNode = m_node->lastChild() ? m_node->lastChild() : m_node;
if (m_lastTextNode && shouldEmitNewlineAfterNode(m_node)) {
bool addNewline = shouldEmitExtraNewlineForNode(m_node);
if (m_lastCharacter != '\n') {
emitCharacter('\n', baseNode->parentNode(), baseNode, 1, 1);
ASSERT(!m_needAnotherNewline);
m_needAnotherNewline = addNewline;
} else if (addNewline)
emitCharacter('\n', baseNode->parentNode(), baseNode, 1, 1);
}
if (!m_positionNode && shouldEmitSpaceBeforeAndAfterNode(m_node))
emitCharacter(' ', baseNode->parentNode(), baseNode, 1, 1);
}
void TextIterator::emitCharacter(UChar c, Node *textNode, Node *offsetBaseNode, int textStartOffset, int textEndOffset)
{
m_haveEmitted = true;
m_positionNode = textNode;
m_positionOffsetBaseNode = offsetBaseNode;
m_positionStartOffset = textStartOffset;
m_positionEndOffset = textEndOffset;
m_singleCharacterBuffer = c;
m_textCharacters = &m_singleCharacterBuffer;
m_textLength = 1;
m_lastTextNodeEndedWithCollapsedSpace = false;
m_lastCharacter = c;
}
void TextIterator::emitText(Node *textNode, int textStartOffset, int textEndOffset)
{
RenderText* renderer = static_cast<RenderText*>(m_node->renderer());
String str = renderer->text();
m_positionNode = textNode;
m_positionOffsetBaseNode = 0;
m_positionStartOffset = textStartOffset;
m_positionEndOffset = textEndOffset;
m_textCharacters = str.characters() + textStartOffset;
m_textLength = textEndOffset - textStartOffset;
m_lastCharacter = str[textEndOffset - 1];
m_lastTextNodeEndedWithCollapsedSpace = false;
m_haveEmitted = true;
}
PassRefPtr<Range> TextIterator::range() const
{
if (m_positionNode) {
if (m_positionOffsetBaseNode) {
int index = m_positionOffsetBaseNode->nodeIndex();
m_positionStartOffset += index;
m_positionEndOffset += index;
m_positionOffsetBaseNode = 0;
}
return new Range(m_positionNode->document(), m_positionNode, m_positionStartOffset, m_positionNode, m_positionEndOffset);
}
if (m_endContainer)
return new Range(m_endContainer->document(), m_endContainer, m_endOffset, m_endContainer, m_endOffset);
return 0;
}
SimplifiedBackwardsTextIterator::SimplifiedBackwardsTextIterator() : m_positionNode(0)
{
}
SimplifiedBackwardsTextIterator::SimplifiedBackwardsTextIterator(const Range *r)
{
m_positionNode = 0;
if (!r)
return;
int exception = 0;
Node *startNode = r->startContainer(exception);
if (exception)
return;
Node *endNode = r->endContainer(exception);
if (exception)
return;
int startOffset = r->startOffset(exception);
if (exception)
return;
int endOffset = r->endOffset(exception);
if (exception)
return;
if (!startNode->offsetInCharacters()) {
if (startOffset >= 0 && startOffset < static_cast<int>(startNode->childNodeCount())) {
startNode = startNode->childNode(startOffset);
startOffset = 0;
}
}
if (!endNode->offsetInCharacters()) {
if (endOffset > 0 && endOffset <= static_cast<int>(endNode->childNodeCount())) {
endNode = endNode->childNode(endOffset - 1);
endOffset = endNode->offsetInCharacters() ? endNode->maxCharacterOffset() : endNode->childNodeCount();
}
}
m_node = endNode;
m_offset = endOffset;
m_handledNode = false;
m_handledChildren = endOffset == 0;
m_startNode = startNode;
m_startOffset = startOffset;
m_endNode = endNode;
m_endOffset = endOffset;
#ifndef NDEBUG
m_positionNode = endNode;
#endif
m_lastTextNode = 0;
m_lastCharacter = '\n';
if (startOffset == 0 || !startNode->firstChild()) {
m_pastStartNode = startNode->previousSibling();
while (!m_pastStartNode && startNode->parentNode()) {
startNode = startNode->parentNode();
m_pastStartNode = startNode->previousSibling();
}
} else
m_pastStartNode = startNode->childNode(startOffset - 1);
advance();
}
void SimplifiedBackwardsTextIterator::advance()
{
ASSERT(m_positionNode);
m_positionNode = 0;
m_textLength = 0;
while (m_node && m_node != m_pastStartNode) {
if (!m_handledNode && !(m_node == m_endNode && m_endOffset == 0)) {
RenderObject *renderer = m_node->renderer();
if (renderer && renderer->isText() && m_node->nodeType() == Node::TEXT_NODE) {
if (renderer->style()->visibility() == VISIBLE && m_offset > 0)
m_handledNode = handleTextNode();
} else if (renderer && (renderer->isImage() || renderer->isWidget())) {
if (renderer->style()->visibility() == VISIBLE && m_offset > 0)
m_handledNode = handleReplacedElement();
} else
m_handledNode = handleNonTextNode();
if (m_positionNode)
return;
}
Node* next = m_handledChildren ? 0 : m_node->lastChild();
if (!next) {
if (!m_handledNode &&
canHaveChildrenForEditing(m_node) &&
m_node->parentNode() &&
(!m_node->lastChild() || m_node == m_endNode && m_endOffset == 0)) {
exitNode();
if (m_positionNode) {
m_handledNode = true;
m_handledChildren = true;
return;
}
}
next = m_node->previousSibling();
while (!next) {
if (!m_node->parentNode())
break;
m_node = m_node->parentNode();
exitNode();
if (m_positionNode) {
m_handledNode = true;
m_handledChildren = true;
return;
}
next = m_node->previousSibling();
}
}
m_node = next;
m_offset = m_node ? caretMaxOffset(m_node) : 0;
m_handledNode = false;
m_handledChildren = false;
if (m_positionNode)
return;
}
}
bool SimplifiedBackwardsTextIterator::handleTextNode()
{
m_lastTextNode = m_node;
RenderText *renderer = static_cast<RenderText *>(m_node->renderer());
String str = renderer->text();
if (!renderer->firstTextBox() && str.length() > 0)
return true;
m_positionEndOffset = m_offset;
m_offset = (m_node == m_startNode) ? m_startOffset : 0;
m_positionNode = m_node;
m_positionStartOffset = m_offset;
m_textLength = m_positionEndOffset - m_positionStartOffset;
m_textCharacters = str.characters() + m_positionStartOffset;
m_lastCharacter = str[m_positionEndOffset - 1];
return true;
}
bool SimplifiedBackwardsTextIterator::handleReplacedElement()
{
unsigned index = m_node->nodeIndex();
emitCharacter(',', m_node->parentNode(), index, index + 1);
return true;
}
bool SimplifiedBackwardsTextIterator::handleNonTextNode()
{
if (shouldEmitNewlineForNode(m_node) ||
shouldEmitNewlineAfterNode(m_node) ||
shouldEmitTabBeforeNode(m_node)) {
unsigned index = m_node->nodeIndex();
emitCharacter('\n', m_node->parentNode(), index + 1, index + 1);
}
return true;
}
void SimplifiedBackwardsTextIterator::exitNode()
{
if (shouldEmitNewlineForNode(m_node) ||
shouldEmitNewlineBeforeNode(m_node) ||
shouldEmitTabBeforeNode(m_node))
emitCharacter('\n', m_node, 0, 0);
}
void SimplifiedBackwardsTextIterator::emitCharacter(UChar c, Node *node, int startOffset, int endOffset)
{
m_singleCharacterBuffer = c;
m_positionNode = node;
m_positionStartOffset = startOffset;
m_positionEndOffset = endOffset;
m_textCharacters = &m_singleCharacterBuffer;
m_textLength = 1;
m_lastCharacter = c;
}
PassRefPtr<Range> SimplifiedBackwardsTextIterator::range() const
{
if (m_positionNode)
return new Range(m_positionNode->document(), m_positionNode, m_positionStartOffset, m_positionNode, m_positionEndOffset);
return new Range(m_startNode->document(), m_startNode, m_startOffset, m_startNode, m_startOffset);
}
CharacterIterator::CharacterIterator()
: m_offset(0), m_runOffset(0), m_atBreak(true)
{
}
CharacterIterator::CharacterIterator(const Range *r, bool emitCharactersBetweenAllVisiblePositions)
: m_offset(0), m_runOffset(0), m_atBreak(true), m_textIterator(r, emitCharactersBetweenAllVisiblePositions)
{
while (!atEnd() && m_textIterator.length() == 0)
m_textIterator.advance();
}
PassRefPtr<Range> CharacterIterator::range() const
{
RefPtr<Range> r = m_textIterator.range();
if (!m_textIterator.atEnd()) {
if (m_textIterator.length() <= 1) {
ASSERT(m_runOffset == 0);
} else {
int exception = 0;
Node *n = r->startContainer(exception);
ASSERT(n == r->endContainer(exception));
int offset = r->startOffset(exception) + m_runOffset;
r->setStart(n, offset, exception);
r->setEnd(n, offset + 1, exception);
}
}
return r.release();
}
void CharacterIterator::advance(int count)
{
if (count <= 0) {
ASSERT(count == 0);
return;
}
m_atBreak = false;
int remaining = m_textIterator.length() - m_runOffset;
if (count < remaining) {
m_runOffset += count;
m_offset += count;
return;
}
count -= remaining;
m_offset += remaining;
for (m_textIterator.advance(); !atEnd(); m_textIterator.advance()) {
int runLength = m_textIterator.length();
if (runLength == 0)
m_atBreak = true;
else {
if (count < runLength) {
m_runOffset = count;
m_offset += count;
return;
}
count -= runLength;
m_offset += runLength;
}
}
m_atBreak = true;
m_runOffset = 0;
}
DeprecatedString CharacterIterator::string(int numChars)
{
DeprecatedString result;
result.reserve(numChars);
while (numChars > 0 && !atEnd()) {
int runSize = min(numChars, length());
result.append(reinterpret_cast<const DeprecatedChar*>(characters()), runSize);
numChars -= runSize;
advance(runSize);
}
return result;
}
WordAwareIterator::WordAwareIterator()
: m_previousText(0), m_didLookAhead(false)
{
}
WordAwareIterator::WordAwareIterator(const Range *r)
: m_previousText(0), m_didLookAhead(false), m_textIterator(r)
{
m_didLookAhead = true; advance(); }
void WordAwareIterator::advance()
{
m_previousText = 0;
m_buffer = "";
if (!m_didLookAhead) {
ASSERT(!m_textIterator.atEnd());
m_textIterator.advance();
}
m_didLookAhead = false;
while (!m_textIterator.atEnd() && m_textIterator.length() == 0)
m_textIterator.advance();
m_range = m_textIterator.range();
if (m_textIterator.atEnd())
return;
while (1) {
if (isSpaceOrNewline(m_textIterator.characters()[m_textIterator.length() - 1]))
return;
if (m_buffer.isEmpty()) {
m_previousText = m_textIterator.characters();
m_previousLength = m_textIterator.length();
}
m_textIterator.advance();
if (m_textIterator.atEnd() || m_textIterator.length() == 0 || isSpaceOrNewline(m_textIterator.characters()[0])) {
m_didLookAhead = true;
return;
}
if (m_buffer.isEmpty()) {
m_buffer.append(reinterpret_cast<const DeprecatedChar*>(m_previousText), m_previousLength);
m_previousText = 0;
}
m_buffer.append(reinterpret_cast<const DeprecatedChar*>(m_textIterator.characters()), m_textIterator.length());
int exception = 0;
m_range->setEnd(m_textIterator.range()->endContainer(exception), m_textIterator.range()->endOffset(exception), exception);
}
}
int WordAwareIterator::length() const
{
if (!m_buffer.isEmpty())
return m_buffer.length();
if (m_previousText)
return m_previousLength;
return m_textIterator.length();
}
const UChar* WordAwareIterator::characters() const
{
if (!m_buffer.isEmpty())
return reinterpret_cast<const UChar*>(m_buffer.unicode());
if (m_previousText)
return m_previousText;
return m_textIterator.characters();
}
CircularSearchBuffer::CircularSearchBuffer(const String& s, bool isCaseSensitive)
: m_target(isCaseSensitive ? s : s.foldCase())
, m_isCaseSensitive(isCaseSensitive)
, m_characterBuffer(m_target.length())
, m_isCharacterStartBuffer(m_target.length())
, m_isBufferFull(false)
, m_cursor(0)
{
ASSERT(!m_target.isEmpty());
m_target.replace(noBreakSpace, ' ');
}
inline void CircularSearchBuffer::append(UChar c, bool isStart)
{
m_characterBuffer[m_cursor] = c == noBreakSpace ? ' ' : c;
m_isCharacterStartBuffer[m_cursor] = isStart;
if (++m_cursor == m_target.length()) {
m_cursor = 0;
m_isBufferFull = true;
}
}
inline void CircularSearchBuffer::append(UChar c)
{
if (m_isCaseSensitive) {
append(c, true);
return;
}
const int maxFoldedCharacters = 16; UChar foldedCharacters[maxFoldedCharacters];
bool error;
int numFoldedCharacters = foldCase(foldedCharacters, maxFoldedCharacters, &c, 1, &error);
ASSERT(!error);
ASSERT(numFoldedCharacters);
ASSERT(numFoldedCharacters <= maxFoldedCharacters);
if (!error && numFoldedCharacters) {
numFoldedCharacters = min(numFoldedCharacters, maxFoldedCharacters);
append(foldedCharacters[0], true);
for (int i = 1; i < numFoldedCharacters; ++i)
append(foldedCharacters[i], false);
}
}
inline bool CircularSearchBuffer::isMatch() const
{
if (!m_isBufferFull)
return false;
if (!m_isCharacterStartBuffer[m_cursor])
return false;
unsigned tailSpace = m_target.length() - m_cursor;
return memcmp(&m_characterBuffer[m_cursor], m_target.characters(), tailSpace * sizeof(UChar)) == 0
&& memcmp(&m_characterBuffer[0], m_target.characters() + tailSpace, m_cursor * sizeof(UChar)) == 0;
}
unsigned CircularSearchBuffer::length() const
{
ASSERT(isMatch());
unsigned bufferSize = m_target.length();
unsigned length = 0;
for (unsigned i = 0; i < bufferSize; ++i)
length += m_isCharacterStartBuffer[i];
return length;
}
int TextIterator::rangeLength(const Range *r, bool forSelectionPreservation)
{
int length = 0;
for (TextIterator it(r, forSelectionPreservation); !it.atEnd(); it.advance())
length += it.length();
return length;
}
PassRefPtr<Range> TextIterator::subrange(Range* entireRange, int characterOffset, int characterCount)
{
CharacterIterator chars(entireRange);
chars.advance(characterOffset);
RefPtr<Range> start = chars.range();
chars.advance(characterCount);
RefPtr<Range> end = chars.range();
ExceptionCode ec = 0;
RefPtr<Range> result(new Range(entireRange->ownerDocument(),
start->startContainer(ec),
start->startOffset(ec),
end->startContainer(ec),
end->startOffset(ec)));
ASSERT(!ec);
return result.release();
}
PassRefPtr<Range> TextIterator::rangeFromLocationAndLength(Element *scope, int rangeLocation, int rangeLength, bool forSelectionPreservation)
{
RefPtr<Range> resultRange = scope->document()->createRange();
int docTextPosition = 0;
int rangeEnd = rangeLocation + rangeLength;
bool startRangeFound = false;
RefPtr<Range> textRunRange;
TextIterator it(rangeOfContents(scope).get(), forSelectionPreservation);
if (rangeLocation == 0 && rangeLength == 0 && it.atEnd()) {
int exception = 0;
textRunRange = it.range();
resultRange->setStart(textRunRange->startContainer(exception), 0, exception);
ASSERT(exception == 0);
resultRange->setEnd(textRunRange->startContainer(exception), 0, exception);
ASSERT(exception == 0);
return resultRange.release();
}
for (; !it.atEnd(); it.advance()) {
int len = it.length();
textRunRange = it.range();
bool foundStart = rangeLocation >= docTextPosition && rangeLocation <= docTextPosition + len;
bool foundEnd = rangeEnd >= docTextPosition && rangeEnd <= docTextPosition + len;
if (foundStart || foundEnd) {
if (len == 1 && it.characters()[0] == UChar('\n')) {
Position runStart = textRunRange->startPosition();
Position runEnd = VisiblePosition(runStart).next().deepEquivalent();
if (runEnd.isNotNull()) {
ExceptionCode ec = 0;
textRunRange->setEnd(runEnd.node(), runEnd.offset(), ec);
}
}
}
if (foundStart) {
startRangeFound = true;
int exception = 0;
if (textRunRange->startContainer(exception)->isTextNode()) {
int offset = rangeLocation - docTextPosition;
resultRange->setStart(textRunRange->startContainer(exception), offset + textRunRange->startOffset(exception), exception);
} else {
if (rangeLocation == docTextPosition)
resultRange->setStart(textRunRange->startContainer(exception), textRunRange->startOffset(exception), exception);
else
resultRange->setStart(textRunRange->endContainer(exception), textRunRange->endOffset(exception), exception);
}
}
if (foundEnd) {
int exception = 0;
if (textRunRange->startContainer(exception)->isTextNode()) {
int offset = rangeEnd - docTextPosition;
resultRange->setEnd(textRunRange->startContainer(exception), offset + textRunRange->startOffset(exception), exception);
} else {
if (rangeEnd == docTextPosition)
resultRange->setEnd(textRunRange->startContainer(exception), textRunRange->startOffset(exception), exception);
else
resultRange->setEnd(textRunRange->endContainer(exception), textRunRange->endOffset(exception), exception);
}
docTextPosition += len;
break;
}
docTextPosition += len;
}
if (!startRangeFound)
return 0;
if (rangeLength != 0 && rangeEnd > docTextPosition) { int exception = 0;
resultRange->setEnd(textRunRange->endContainer(exception), textRunRange->endOffset(exception), exception);
}
return resultRange.release();
}
UChar* plainTextToMallocAllocatedBuffer(const Range* r, unsigned& bufferLength)
{
UChar* result = 0;
static const unsigned cMaxSegmentSize = 1 << 16;
bufferLength = 0;
typedef pair<UChar*, unsigned> TextSegment;
Vector<TextSegment>* textSegments = 0;
Vector<UChar> textBuffer;
textBuffer.reserveCapacity(cMaxSegmentSize);
for (TextIterator it(r); !it.atEnd(); it.advance()) {
if (textBuffer.size() && textBuffer.size() + it.length() > cMaxSegmentSize) {
UChar* newSegmentBuffer = static_cast<UChar*>(malloc(textBuffer.size() * sizeof(UChar)));
if (!newSegmentBuffer)
goto exit;
memcpy(newSegmentBuffer, textBuffer.data(), textBuffer.size() * sizeof(UChar));
if (!textSegments)
textSegments = new Vector<TextSegment>;
textSegments->append(make_pair(newSegmentBuffer, textBuffer.size()));
textBuffer.clear();
}
textBuffer.append(it.characters(), it.length());
bufferLength += it.length();
}
if (!bufferLength)
return 0;
result = static_cast<UChar*>(malloc(bufferLength * sizeof(UChar)));
if (!result)
goto exit;
{
UChar* resultPos = result;
if (textSegments) {
unsigned size = textSegments->size();
for (unsigned i = 0; i < size; ++i) {
const TextSegment& segment = textSegments->at(i);
memcpy(resultPos, segment.first, segment.second * sizeof(UChar));
resultPos += segment.second;
}
}
memcpy(resultPos, textBuffer.data(), textBuffer.size() * sizeof(UChar));
}
exit:
if (textSegments) {
unsigned size = textSegments->size();
for (unsigned i = 0; i < size; ++i)
free(textSegments->at(i).first);
delete textSegments;
}
return result;
}
String plainText(const Range* r)
{
unsigned length;
UChar* buf = plainTextToMallocAllocatedBuffer(r, length);
if (!buf)
return "";
String result(buf, length);
free(buf);
return result;
}
PassRefPtr<Range> findPlainText(const Range* range, const String& target, bool forward, bool caseSensitive)
{
ExceptionCode ec = 0;
RefPtr<Range> result = range->cloneRange(ec);
result->collapse(!forward, ec);
if (target.isEmpty() || target.find('\n') != -1)
return result.release();
unsigned matchStart = 0;
unsigned matchLength = 0;
{
CircularSearchBuffer searchBuffer(target, caseSensitive);
CharacterIterator it(range);
for (;;) {
if (searchBuffer.isMatch()) {
unsigned matchEnd = it.characterOffset();
matchLength = searchBuffer.length();
ASSERT(matchLength);
ASSERT(matchEnd >= matchLength);
matchStart = matchEnd - matchLength;
if (forward)
break;
}
if (it.atBreak()) {
if (it.atEnd())
break;
searchBuffer.clear();
}
searchBuffer.append(it.characters()[0]);
it.advance(1);
}
}
if (matchLength) {
CharacterIterator it(range);
it.advance(matchStart);
result->setStart(it.range()->startContainer(ec), it.range()->startOffset(ec), ec);
it.advance(matchLength - 1);
result->setEnd(it.range()->endContainer(ec), it.range()->endOffset(ec), ec);
}
return result.release();
}
}