444 lines
14 KiB
C++
444 lines
14 KiB
C++
/*
|
|
**********************************************************************
|
|
* Copyright (C) 2001-2008,2010 IBM and others. All rights reserved.
|
|
**********************************************************************
|
|
* Date Name Description
|
|
* 03/22/2000 helena Creation.
|
|
**********************************************************************
|
|
*/
|
|
|
|
#include "unicode/utypes.h"
|
|
|
|
#if !UCONFIG_NO_COLLATION && !UCONFIG_NO_BREAK_ITERATION
|
|
|
|
#include "unicode/brkiter.h"
|
|
#include "unicode/schriter.h"
|
|
#include "unicode/search.h"
|
|
#include "usrchimp.h"
|
|
#include "cmemory.h"
|
|
|
|
// public constructors and destructors -----------------------------------
|
|
U_NAMESPACE_BEGIN
|
|
|
|
SearchIterator::SearchIterator(const SearchIterator &other)
|
|
: UObject(other)
|
|
{
|
|
m_breakiterator_ = other.m_breakiterator_;
|
|
m_text_ = other.m_text_;
|
|
m_search_ = (USearch *)uprv_malloc(sizeof(USearch));
|
|
m_search_->breakIter = other.m_search_->breakIter;
|
|
m_search_->isCanonicalMatch = other.m_search_->isCanonicalMatch;
|
|
m_search_->isOverlap = other.m_search_->isOverlap;
|
|
m_search_->elementComparisonType = other.m_search_->elementComparisonType;
|
|
m_search_->matchedIndex = other.m_search_->matchedIndex;
|
|
m_search_->matchedLength = other.m_search_->matchedLength;
|
|
m_search_->text = other.m_search_->text;
|
|
m_search_->textLength = other.m_search_->textLength;
|
|
}
|
|
|
|
SearchIterator::~SearchIterator()
|
|
{
|
|
if (m_search_ != NULL) {
|
|
uprv_free(m_search_);
|
|
}
|
|
}
|
|
|
|
// public get and set methods ----------------------------------------
|
|
|
|
void SearchIterator::setAttribute(USearchAttribute attribute,
|
|
USearchAttributeValue value,
|
|
UErrorCode &status)
|
|
{
|
|
if (U_SUCCESS(status)) {
|
|
switch (attribute)
|
|
{
|
|
case USEARCH_OVERLAP :
|
|
m_search_->isOverlap = (value == USEARCH_ON ? TRUE : FALSE);
|
|
break;
|
|
case USEARCH_CANONICAL_MATCH :
|
|
m_search_->isCanonicalMatch = (value == USEARCH_ON ? TRUE : FALSE);
|
|
break;
|
|
case USEARCH_ELEMENT_COMPARISON :
|
|
if (value == USEARCH_PATTERN_BASE_WEIGHT_IS_WILDCARD || value == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD) {
|
|
m_search_->elementComparisonType = (int16_t)value;
|
|
} else {
|
|
m_search_->elementComparisonType = 0;
|
|
}
|
|
break;
|
|
default:
|
|
status = U_ILLEGAL_ARGUMENT_ERROR;
|
|
}
|
|
}
|
|
if (value == USEARCH_ATTRIBUTE_VALUE_COUNT) {
|
|
status = U_ILLEGAL_ARGUMENT_ERROR;
|
|
}
|
|
}
|
|
|
|
USearchAttributeValue SearchIterator::getAttribute(
|
|
USearchAttribute attribute) const
|
|
{
|
|
switch (attribute) {
|
|
case USEARCH_OVERLAP :
|
|
return (m_search_->isOverlap == TRUE ? USEARCH_ON : USEARCH_OFF);
|
|
case USEARCH_CANONICAL_MATCH :
|
|
return (m_search_->isCanonicalMatch == TRUE ? USEARCH_ON :
|
|
USEARCH_OFF);
|
|
case USEARCH_ELEMENT_COMPARISON :
|
|
{
|
|
int16_t value = m_search_->elementComparisonType;
|
|
if (value == USEARCH_PATTERN_BASE_WEIGHT_IS_WILDCARD || value == USEARCH_ANY_BASE_WEIGHT_IS_WILDCARD) {
|
|
return (USearchAttributeValue)value;
|
|
} else {
|
|
return USEARCH_STANDARD_ELEMENT_COMPARISON;
|
|
}
|
|
}
|
|
default :
|
|
return USEARCH_DEFAULT;
|
|
}
|
|
}
|
|
|
|
int32_t SearchIterator::getMatchedStart() const
|
|
{
|
|
return m_search_->matchedIndex;
|
|
}
|
|
|
|
int32_t SearchIterator::getMatchedLength() const
|
|
{
|
|
return m_search_->matchedLength;
|
|
}
|
|
|
|
void SearchIterator::getMatchedText(UnicodeString &result) const
|
|
{
|
|
int32_t matchedindex = m_search_->matchedIndex;
|
|
int32_t matchedlength = m_search_->matchedLength;
|
|
if (matchedindex != USEARCH_DONE && matchedlength != 0) {
|
|
result.setTo(m_search_->text + matchedindex, matchedlength);
|
|
}
|
|
else {
|
|
result.remove();
|
|
}
|
|
}
|
|
|
|
void SearchIterator::setBreakIterator(BreakIterator *breakiter,
|
|
UErrorCode &status)
|
|
{
|
|
if (U_SUCCESS(status)) {
|
|
#if 0
|
|
m_search_->breakIter = NULL;
|
|
// the c++ breakiterator may not make use of ubreakiterator.
|
|
// so we'll have to keep track of it ourselves.
|
|
#else
|
|
// Well, gee... the Constructors that take a BreakIterator
|
|
// all cast the BreakIterator to a UBreakIterator and
|
|
// pass it to the corresponding usearch_openFromXXX
|
|
// routine, so there's no reason not to do this.
|
|
//
|
|
// Besides, a UBreakIterator is a BreakIterator, so
|
|
// any subclass of BreakIterator should work fine here...
|
|
m_search_->breakIter = (UBreakIterator *) breakiter;
|
|
#endif
|
|
|
|
m_breakiterator_ = breakiter;
|
|
}
|
|
}
|
|
|
|
const BreakIterator * SearchIterator::getBreakIterator(void) const
|
|
{
|
|
return m_breakiterator_;
|
|
}
|
|
|
|
void SearchIterator::setText(const UnicodeString &text, UErrorCode &status)
|
|
{
|
|
if (U_SUCCESS(status)) {
|
|
if (text.length() == 0) {
|
|
status = U_ILLEGAL_ARGUMENT_ERROR;
|
|
}
|
|
else {
|
|
m_text_ = text;
|
|
m_search_->text = m_text_.getBuffer();
|
|
m_search_->textLength = m_text_.length();
|
|
}
|
|
}
|
|
}
|
|
|
|
void SearchIterator::setText(CharacterIterator &text, UErrorCode &status)
|
|
{
|
|
if (U_SUCCESS(status)) {
|
|
text.getText(m_text_);
|
|
setText(m_text_, status);
|
|
}
|
|
}
|
|
|
|
const UnicodeString & SearchIterator::getText(void) const
|
|
{
|
|
return m_text_;
|
|
}
|
|
|
|
// operator overloading ----------------------------------------------
|
|
|
|
UBool SearchIterator::operator==(const SearchIterator &that) const
|
|
{
|
|
if (this == &that) {
|
|
return TRUE;
|
|
}
|
|
return (m_breakiterator_ == that.m_breakiterator_ &&
|
|
m_search_->isCanonicalMatch == that.m_search_->isCanonicalMatch &&
|
|
m_search_->isOverlap == that.m_search_->isOverlap &&
|
|
m_search_->elementComparisonType == that.m_search_->elementComparisonType &&
|
|
m_search_->matchedIndex == that.m_search_->matchedIndex &&
|
|
m_search_->matchedLength == that.m_search_->matchedLength &&
|
|
m_search_->textLength == that.m_search_->textLength &&
|
|
getOffset() == that.getOffset() &&
|
|
(uprv_memcmp(m_search_->text, that.m_search_->text,
|
|
m_search_->textLength * sizeof(UChar)) == 0));
|
|
}
|
|
|
|
// public methods ----------------------------------------------------
|
|
|
|
int32_t SearchIterator::first(UErrorCode &status)
|
|
{
|
|
if (U_FAILURE(status)) {
|
|
return USEARCH_DONE;
|
|
}
|
|
setOffset(0, status);
|
|
return handleNext(0, status);
|
|
}
|
|
|
|
int32_t SearchIterator::following(int32_t position,
|
|
UErrorCode &status)
|
|
{
|
|
if (U_FAILURE(status)) {
|
|
return USEARCH_DONE;
|
|
}
|
|
setOffset(position, status);
|
|
return handleNext(position, status);
|
|
}
|
|
|
|
int32_t SearchIterator::last(UErrorCode &status)
|
|
{
|
|
if (U_FAILURE(status)) {
|
|
return USEARCH_DONE;
|
|
}
|
|
setOffset(m_search_->textLength, status);
|
|
return handlePrev(m_search_->textLength, status);
|
|
}
|
|
|
|
int32_t SearchIterator::preceding(int32_t position,
|
|
UErrorCode &status)
|
|
{
|
|
if (U_FAILURE(status)) {
|
|
return USEARCH_DONE;
|
|
}
|
|
setOffset(position, status);
|
|
return handlePrev(position, status);
|
|
}
|
|
|
|
int32_t SearchIterator::next(UErrorCode &status)
|
|
{
|
|
if (U_SUCCESS(status)) {
|
|
int32_t offset = getOffset();
|
|
int32_t matchindex = m_search_->matchedIndex;
|
|
int32_t matchlength = m_search_->matchedLength;
|
|
m_search_->reset = FALSE;
|
|
if (m_search_->isForwardSearching == TRUE) {
|
|
int32_t textlength = m_search_->textLength;
|
|
if (offset == textlength || matchindex == textlength ||
|
|
(matchindex != USEARCH_DONE &&
|
|
matchindex + matchlength >= textlength)) {
|
|
// not enough characters to match
|
|
setMatchNotFound();
|
|
return USEARCH_DONE;
|
|
}
|
|
}
|
|
else {
|
|
// switching direction.
|
|
// if matchedIndex == USEARCH_DONE, it means that either a
|
|
// setOffset has been called or that previous ran off the text
|
|
// string. the iterator would have been set to offset 0 if a
|
|
// match is not found.
|
|
m_search_->isForwardSearching = TRUE;
|
|
if (m_search_->matchedIndex != USEARCH_DONE) {
|
|
// there's no need to set the collation element iterator
|
|
// the next call to next will set the offset.
|
|
return matchindex;
|
|
}
|
|
}
|
|
|
|
if (matchlength > 0) {
|
|
// if matchlength is 0 we are at the start of the iteration
|
|
if (m_search_->isOverlap) {
|
|
offset ++;
|
|
}
|
|
else {
|
|
offset += matchlength;
|
|
}
|
|
}
|
|
return handleNext(offset, status);
|
|
}
|
|
return USEARCH_DONE;
|
|
}
|
|
|
|
int32_t SearchIterator::previous(UErrorCode &status)
|
|
{
|
|
if (U_SUCCESS(status)) {
|
|
int32_t offset;
|
|
if (m_search_->reset) {
|
|
offset = m_search_->textLength;
|
|
m_search_->isForwardSearching = FALSE;
|
|
m_search_->reset = FALSE;
|
|
setOffset(offset, status);
|
|
}
|
|
else {
|
|
offset = getOffset();
|
|
}
|
|
|
|
int32_t matchindex = m_search_->matchedIndex;
|
|
if (m_search_->isForwardSearching == TRUE) {
|
|
// switching direction.
|
|
// if matchedIndex == USEARCH_DONE, it means that either a
|
|
// setOffset has been called or that next ran off the text
|
|
// string. the iterator would have been set to offset textLength if
|
|
// a match is not found.
|
|
m_search_->isForwardSearching = FALSE;
|
|
if (matchindex != USEARCH_DONE) {
|
|
return matchindex;
|
|
}
|
|
}
|
|
else {
|
|
if (offset == 0 || matchindex == 0) {
|
|
// not enough characters to match
|
|
setMatchNotFound();
|
|
return USEARCH_DONE;
|
|
}
|
|
}
|
|
|
|
if (matchindex != USEARCH_DONE) {
|
|
if (m_search_->isOverlap) {
|
|
matchindex += m_search_->matchedLength - 2;
|
|
}
|
|
|
|
return handlePrev(matchindex, status);
|
|
}
|
|
|
|
return handlePrev(offset, status);
|
|
}
|
|
|
|
return USEARCH_DONE;
|
|
}
|
|
|
|
void SearchIterator::reset()
|
|
{
|
|
UErrorCode status = U_ZERO_ERROR;
|
|
setMatchNotFound();
|
|
setOffset(0, status);
|
|
m_search_->isOverlap = FALSE;
|
|
m_search_->isCanonicalMatch = FALSE;
|
|
m_search_->elementComparisonType = 0;
|
|
m_search_->isForwardSearching = TRUE;
|
|
m_search_->reset = TRUE;
|
|
}
|
|
|
|
// protected constructors and destructors -----------------------------
|
|
|
|
SearchIterator::SearchIterator()
|
|
{
|
|
m_search_ = (USearch *)uprv_malloc(sizeof(USearch));
|
|
m_search_->breakIter = NULL;
|
|
m_search_->isOverlap = FALSE;
|
|
m_search_->isCanonicalMatch = FALSE;
|
|
m_search_->elementComparisonType = 0;
|
|
m_search_->isForwardSearching = TRUE;
|
|
m_search_->reset = TRUE;
|
|
m_search_->matchedIndex = USEARCH_DONE;
|
|
m_search_->matchedLength = 0;
|
|
m_search_->text = NULL;
|
|
m_search_->textLength = 0;
|
|
m_breakiterator_ = NULL;
|
|
}
|
|
|
|
SearchIterator::SearchIterator(const UnicodeString &text,
|
|
BreakIterator *breakiter) :
|
|
m_breakiterator_(breakiter),
|
|
m_text_(text)
|
|
{
|
|
m_search_ = (USearch *)uprv_malloc(sizeof(USearch));
|
|
m_search_->breakIter = NULL;
|
|
m_search_->isOverlap = FALSE;
|
|
m_search_->isCanonicalMatch = FALSE;
|
|
m_search_->elementComparisonType = 0;
|
|
m_search_->isForwardSearching = TRUE;
|
|
m_search_->reset = TRUE;
|
|
m_search_->matchedIndex = USEARCH_DONE;
|
|
m_search_->matchedLength = 0;
|
|
m_search_->text = m_text_.getBuffer();
|
|
m_search_->textLength = text.length();
|
|
}
|
|
|
|
SearchIterator::SearchIterator(CharacterIterator &text,
|
|
BreakIterator *breakiter) :
|
|
m_breakiterator_(breakiter)
|
|
{
|
|
m_search_ = (USearch *)uprv_malloc(sizeof(USearch));
|
|
m_search_->breakIter = NULL;
|
|
m_search_->isOverlap = FALSE;
|
|
m_search_->isCanonicalMatch = FALSE;
|
|
m_search_->elementComparisonType = 0;
|
|
m_search_->isForwardSearching = TRUE;
|
|
m_search_->reset = TRUE;
|
|
m_search_->matchedIndex = USEARCH_DONE;
|
|
m_search_->matchedLength = 0;
|
|
text.getText(m_text_);
|
|
m_search_->text = m_text_.getBuffer();
|
|
m_search_->textLength = m_text_.length();
|
|
m_breakiterator_ = breakiter;
|
|
}
|
|
|
|
// protected methods ------------------------------------------------------
|
|
|
|
SearchIterator & SearchIterator::operator=(const SearchIterator &that)
|
|
{
|
|
if (this != &that) {
|
|
m_breakiterator_ = that.m_breakiterator_;
|
|
m_text_ = that.m_text_;
|
|
m_search_->breakIter = that.m_search_->breakIter;
|
|
m_search_->isCanonicalMatch = that.m_search_->isCanonicalMatch;
|
|
m_search_->isOverlap = that.m_search_->isOverlap;
|
|
m_search_->elementComparisonType = that.m_search_->elementComparisonType;
|
|
m_search_->matchedIndex = that.m_search_->matchedIndex;
|
|
m_search_->matchedLength = that.m_search_->matchedLength;
|
|
m_search_->text = that.m_search_->text;
|
|
m_search_->textLength = that.m_search_->textLength;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
void SearchIterator::setMatchLength(int32_t length)
|
|
{
|
|
m_search_->matchedLength = length;
|
|
}
|
|
|
|
void SearchIterator::setMatchStart(int32_t position)
|
|
{
|
|
m_search_->matchedIndex = position;
|
|
}
|
|
|
|
void SearchIterator::setMatchNotFound()
|
|
{
|
|
setMatchStart(USEARCH_DONE);
|
|
setMatchLength(0);
|
|
UErrorCode status = U_ZERO_ERROR;
|
|
// by default no errors should be returned here since offsets are within
|
|
// range.
|
|
if (m_search_->isForwardSearching) {
|
|
setOffset(m_search_->textLength, status);
|
|
}
|
|
else {
|
|
setOffset(0, status);
|
|
}
|
|
}
|
|
|
|
|
|
U_NAMESPACE_END
|
|
|
|
#endif /* #if !UCONFIG_NO_COLLATION */
|