scuffed-code/icu4c/source/i18n/ucol.cpp
2001-01-15 19:02:30 +00:00

2559 lines
95 KiB
C++

/*
*******************************************************************************
* Copyright (C) 1996-1999, International Business Machines
* Corporation and others. All Rights Reserved.
*******************************************************************************
*/
#include "ucolimp.h"
#include "unicode/uloc.h"
#include "unicode/coll.h"
#include "unicode/tblcoll.h"
#include "unicode/coleitr.h"
#include "unicode/ustring.h"
#include "unicode/normlzr.h"
#include "unicode/unorm.h"
#include "cpputils.h"
#include "cstring.h"
#include <stdio.h>
#include "ucmp32.h"
#include "tcoldata.h"
#include "tables.h"
#include "unicode/udata.h"
#include "umutex.h"
static UCollator* UCA = NULL;
static UBool
isAcceptable(void *context,
const char *type, const char *name,
const UDataInfo *pInfo){
if( pInfo->size>=20 &&
pInfo->isBigEndian==U_IS_BIG_ENDIAN &&
pInfo->charsetFamily==U_CHARSET_FAMILY &&
pInfo->dataFormat[0]==0x55 && /* dataFormat="UCol" */
pInfo->dataFormat[1]==0x43 &&
pInfo->dataFormat[2]==0x6f &&
pInfo->dataFormat[3]==0x6c &&
pInfo->formatVersion[0]==1 &&
pInfo->dataVersion[0]==3 &&
pInfo->dataVersion[1]==0 &&
pInfo->dataVersion[2]==0 &&
pInfo->dataVersion[3]==0) {
return TRUE;
} else {
return FALSE;
}
}
UCollator* ucol_initCollator(const UCATableHeader *image, UCollator *fillIn, UErrorCode *status) {
UCollator *result = fillIn;
if(U_FAILURE(*status)) {
return NULL;
}
if(result == NULL) {
result = (UCollator *)uprv_malloc(sizeof(UCollator));
if(result == NULL) {
*status = U_MEMORY_ALLOCATION_ERROR;
return result;
}
result->freeOnClose = TRUE;
} else {
result->freeOnClose = FALSE;
}
result->image = image;
const uint8_t *mapping = (uint8_t*)result->image+result->image->mappingPosition;
CompactIntArray *newUCAmapping = ucmp32_openFromData(&mapping, status);
if(U_SUCCESS(*status)) {
result->mapping = newUCAmapping;
} else {
if(result->freeOnClose == TRUE) {
uprv_free(result);
result = NULL;
}
return result;
}
result->latinOneMapping = (uint32_t*)((uint8_t*)result->image+result->image->latinOneMapping);
result->contractionCEs = (uint32_t*)((uint8_t*)result->image+result->image->contractionCEs);
result->contractionIndex = (UChar*)((uint8_t*)result->image+result->image->contractionIndex);
result->expansion = (uint32_t*)((uint8_t*)result->image+result->image->expansion);
/* set attributes */
result->caseFirst = result->image->caseFirst;
result->caseLevel = result->image->caseLevel;
result->frenchCollation = result->image->frenchCollation;
result->normalizationMode = result->image->normalizationMode;
result->strength = result->image->strength;
result->variableTopValue = result->image->variableTopValue;
result->caseFirstisDefault = TRUE;
result->caseLevelisDefault = TRUE;
result->frenchCollationisDefault = TRUE;
result->normalizationModeisDefault = TRUE;
result->strengthisDefault = TRUE;
result->variableTopValueisDefault = TRUE;
uint32_t variableMaxCE = ucmp32_get(result->mapping, result->variableTopValue);
result->variableMax = (variableMaxCE & 0xFF000000) >> 24;
return result;
}
void ucol_initUCA(UErrorCode *status) {
if(U_FAILURE(*status)) return;
if(UCA == NULL) {
UCollator *newUCA = (UCollator *)uprv_malloc(sizeof(UCollator));
UDataMemory *result = udata_openChoice(NULL, UCA_DATA_TYPE, UCA_DATA_NAME, isAcceptable, NULL, status);
newUCA = ucol_initCollator((const UCATableHeader *)udata_getMemory(result), newUCA, status);
newUCA->rb = NULL;
if(U_FAILURE(*status)) {
udata_close(result);
uprv_free(newUCA);
}
umtx_lock(NULL);
if(UCA == NULL) {
UCA = newUCA;
newUCA = NULL;
}
umtx_unlock(NULL);
if(newUCA != NULL) {
udata_close(result);
uprv_free(newUCA);
}
}
}
U_CAPI UCollator*
ucol_open( const char *loc,
UErrorCode *status)
{
/* New version */
if(U_FAILURE(*status)) return 0;
ucol_initUCA(status);
UCollator *result = NULL;
UResourceBundle *b = ures_open(NULL, loc, status);
UResourceBundle *binary = ures_getByKey(b, "%%CollationNew", NULL, status);
if(*status = U_MISSING_RESOURCE_ERROR) { /* if we don't find tailoring, we'll fallback to UCA */
result = UCA;
*status = U_USING_DEFAULT_ERROR;
} else if(U_SUCCESS(*status)) { /* otherwise, we'll pick a collation data that exists */
int32_t len = 0;
const uint8_t *inData = ures_getBinary(binary, &len, status);
result = ucol_initCollator((const UCATableHeader *)inData, result, status);
result->rb = b;
}
ures_close(binary);
return result;
}
U_CAPI UCollator*
ucol_openRules( const UChar *rules,
int32_t rulesLength,
UNormalizationMode mode,
UCollationStrength strength,
UErrorCode *status)
{
if(U_FAILURE(*status)) return 0;
Normalizer::EMode normMode;
switch(mode) {
case UCOL_NO_NORMALIZATION:
normMode = Normalizer::NO_OP;
break;
case UCOL_DECOMP_CAN:
normMode = Normalizer::DECOMP;
break;
case UCOL_DECOMP_COMPAT:
normMode = Normalizer::DECOMP_COMPAT;
break;
case UCOL_DECOMP_CAN_COMP_COMPAT:
normMode = Normalizer::COMPOSE;
break;
case UCOL_DECOMP_COMPAT_COMP_CAN:
normMode = Normalizer::COMPOSE_COMPAT;
break;
default:
*status = U_ILLEGAL_ARGUMENT_ERROR;
return 0;
}
return NULL;
}
/* This one is currently used by genrb & tests. After constructing from rules (tailoring),*/
/* you should be able to get the binary chunk to write out... Doesn't look very full now */
U_CAPI uint8_t *
ucol_cloneRuleData(UCollator *coll, int32_t *length, UErrorCode *status)
{
*length = 0;
return NULL;
}
U_CAPI void
ucol_close(UCollator *coll)
{
/* Here, it would be advisable to close: */
/* - UData for UCA (unless we stuff it in the root resb */
/* Again, do we need additional housekeeping... HMMM! */
if(coll->rb != NULL) {
ures_close(coll->rb);
}
if(coll->freeOnClose == TRUE) {
uprv_free(coll);
}
}
uint32_t ucol_getNextCE(const UCollator *coll, collIterate *collationSource, UErrorCode *status) {
uint32_t order;
if (U_FAILURE(*status) || (collationSource->pos>=collationSource->len
&& collationSource->CEpos <= collationSource->toReturn)) { /* have we run out of string and CEs?? */
order = UCOL_NULLORDER; /* if so, we won't play any more */
} else if (collationSource->CEpos > collationSource->toReturn) { /* Are there any CEs from previous expansions? */
order = *(collationSource->toReturn++); /* if so, return them */
collationSource->pos--;
} else { /* This is the real business now */
UChar ch = *collationSource->pos;
collationSource->CEpos = collationSource->toReturn = collationSource->CEs;
if(ch < 0xFF) { /* if it's Latin One, we'll try to fast track it */
order = coll->latinOneMapping[ch]; /* by looking in up in an array */
} else { /* otherwise, */
order = ucmp32_get(coll->mapping, ch); /* we'll go for slightly slower trie */
}
if(order >= UCOL_NOT_FOUND) { /* if a CE is special */
*(collationSource->CEpos) = order; /* prepare the buffer */
order = getSpecialCE(coll, collationSource, status); /* and try to get the special CE */
if(order == UCOL_NOT_FOUND) { /* We couldn't find a good CE in the tailoring */
order = ucol_getNextUCA(ch, collationSource, status);
}
}
}
collationSource->pos++; /* we're advancing to the next codepoint */
/* This means that contraction should spit back the last codepoint eaten! */
return order; /* return the CE */
}
uint32_t ucol_getNextUCA(UChar ch, collIterate *collationSource, UErrorCode *status) {
uint32_t order;
if(ch < 0xFF) { /* so we'll try to find it in the UCA */
order = UCA->latinOneMapping[ch];
} else {
order = ucmp32_get(UCA->mapping, ch);
}
if(order >= UCOL_NOT_FOUND) { /* UCA also gives us a special CE */
order = getSpecialCE(UCA, collationSource, status);
}
if(order == UCOL_NOT_FOUND) { /* This is where we have to resort to algorithmical generation */
/* We have to check if ch is possibly a first surrogate - then we need to take the next code unit */
/* and make a bigger CE */
UChar nextChar;
if(UTF_IS_FIRST_SURROGATE(ch) && (collationSource->pos<collationSource->len) &&
UTF_IS_SECOND_SURROGATE((nextChar=*(collationSource->pos+1)))) {
uint32_t cp = (((ch)<<10UL)+(nextChar)-((0xd800<<10UL)+0xdc00));
collationSource->pos++;
/* This is a code point minus 0x10000, that's what algorithm requires */
order = 0xE0800303 | (cp & 0xF0000) << 8 | (cp & 0xFE00) << 7;
*(collationSource->CEpos++) = 0xF0040000 | (cp & 0x1FF) << 19;
} else {
/* otherwise */
/* Make up an artifical CE from code point as per UCA */
order = 0xD08004F1;
/*order = 0xD01004F1;*/
order |= ((uint32_t)ch & 0xF000)<<12;
order |= ((uint32_t)ch & 0x0FFF)<<11;
}
}
return order; /* return the CE */
}
uint32_t getSpecialCE(const UCollator *coll, collIterate *source, UErrorCode *status) {
int32_t i = 0; /* general counter */
uint32_t CE = *source->CEpos;
while (true) {
const uint32_t *CEOffset = NULL;
const UChar *UCharOffset = NULL;
UChar schar, tchar;
uint32_t size = 0;
switch(getCETag(CE)) {
case NOT_FOUND_TAG:
/* This one is not found, and we'll let somebody else bother about it... no more games */
return CE;
break;
case SURROGATE_TAG:
/* pending surrogate discussion with Markus and Mark */
return UCOL_NOT_FOUND;
break;
case THAI_TAG:
/* Thai/Lao reordering */
if(source->isThai == TRUE) { /* if we encountered Thai prevowel & the string is not yet touched */
source->isThai = FALSE; /* We will touch the string */
if((source->len - source->pos) > UCOL_WRITABLE_BUFFER_SIZE) {
/* Problematic part - if the stack buffer is too small, we need to allocate */
/* However, somebody needs to keep track of that allocated space */
/* And context structure is not good for that */
/* allocate a new buffer - This is unfortunate and should be way smarter */
/*source->writableBuffer = (UChar *)ucol_getABuffer(coll, (source->len - source->pos)*sizeof(UChar));*/
}
UChar *sourceCopy = source->pos;
UChar *targetCopy = source->writableBuffer;
while(sourceCopy < source->len) {
if(UCOL_ISTHAIPREVOWEL(*(sourceCopy)) && /* This is the combination that needs to be swapped */
UCOL_ISTHAIBASECONSONANT(*(sourceCopy+1))) {
*(targetCopy) = *(sourceCopy+1);
*(targetCopy+1) = *(sourceCopy);
targetCopy+=2;
sourceCopy+=2;
} else {
*(targetCopy++) = *(sourceCopy++);
}
}
source->pos = source->writableBuffer;
source->len = targetCopy;
source->CEpos = source->toReturn = source->CEs;
CE = UCOL_IGNORABLE;
} else { /* we have already played with the string, so treat Thai as a length one expansion */
CEOffset = coll->expansion+getExpansionOffset(CE); /* find the offset to expansion table */
CE = *CEOffset++;
}
break;
case CONTRACTION_TAG:
/* This should handle contractions */
while(true) {
/* First we position ourselves at the begining of contraction sequence */
const UChar *ContractionStart = UCharOffset = (UChar *)coll->image+getContractOffset(CE);
/* we need to convey the notion of having a backward search - most probably through the context object */
/* if (backwardsSearch) offset += contractionUChars[(int16_t)offset]; else UCharOffset++; */
UCharOffset++; /* skip the backward offset, see above */
if (source->pos>=source->len) { /* this is the end of string */
CE = *(coll->contractionCEs + (UCharOffset - coll->contractionIndex)); /* So we'll pick whatever we have at the point... */
source->pos--; /* I think, since we'll advance in the getCE */
break;
}
schar = *(++source->pos);
while(schar > (tchar = *UCharOffset)) { /* since the contraction codepoints should be ordered, we skip all that are smaller */
UCharOffset++;
}
if(schar != tchar) { /* we didn't find the correct codepoint. We can use either the first or the last CE */
if(tchar != 0xFFFF) {
UCharOffset = ContractionStart; /* We're not at the end, bailed out in the middle. Better use starting CE */
}
source->pos--; /* Spit out the last char of the string, wasn't tasty enough */
}
CE = *(coll->contractionCEs + (UCharOffset - coll->contractionIndex));
if(!isContraction(CE)) {
/* Maybe not */
/*source->pos--;*/ /* I think, since we'll advance in the getCE */
break;
}
}
break;
case EXPANSION_TAG:
/* This should handle expansion. */
/* NOTE: we can encounter both continuations and expansions in an expansion! */
/* I have to decide where continuations are going to be dealt with */
CEOffset = (uint32_t *)coll->image+getExpansionOffset(CE); /* find the offset to expansion table */
size = getExpansionCount(CE);
CE = *CEOffset++;
if(size != 0) { /* if there are less than 16 elements in expansion, we don't terminate */
for(i = 1; i<size; i++) {
*(source->CEpos++) = *CEOffset++;
}
} else { /* else, we do */
while(*CEOffset != 0) {
*(source->CEpos++) = *CEOffset++;
}
}
/*source->toReturn++;*/
return CE;
break;
case CHARSET_TAG:
/* probably after 1.8 */
return UCOL_NOT_FOUND;
break;
default:
*status = U_INTERNAL_PROGRAM_ERROR;
CE=0;
break;
}
if (CE <= UCOL_NOT_FOUND) break;
}
return CE;
}
uint8_t *reallocateBuffer(uint8_t **secondaries, uint8_t *secStart, uint8_t *second, int32_t *secSize, UErrorCode *status) {
uint8_t *newStart = NULL;
if(secStart==second) {
newStart=(uint8_t*)uprv_malloc(*secSize*2);
if(newStart==NULL) {
*status = U_MEMORY_ALLOCATION_ERROR;
return NULL;
}
uprv_memcpy(newStart, secStart, *secondaries-secStart);
} else {
newStart=(uint8_t*)uprv_realloc(secStart, *secSize*2);
if(newStart==NULL) {
*status = U_MEMORY_ALLOCATION_ERROR;
return NULL;
}
}
*secondaries=newStart+(*secondaries-secStart);
*secSize*=2;
return newStart;
}
void uprv_ucol_reverse_buffer(uint8_t *start, uint8_t *end) {
uint8_t temp;
while(start<end) {
temp = *start;
*start++ = *end;
*end-- = temp;
}
}
U_CAPI UBool
ucol_greater( const UCollator *coll,
const UChar *source,
int32_t sourceLength,
const UChar *target,
int32_t targetLength)
{
return (ucol_strcoll(coll, source, sourceLength, target, targetLength)
== UCOL_GREATER);
}
U_CAPI UBool
ucol_greaterOrEqual( const UCollator *coll,
const UChar *source,
int32_t sourceLength,
const UChar *target,
int32_t targetLength)
{
return (ucol_strcoll(coll, source, sourceLength, target, targetLength)
!= UCOL_LESS);
}
U_CAPI UBool
ucol_equal( const UCollator *coll,
const UChar *source,
int32_t sourceLength,
const UChar *target,
int32_t targetLength)
{
return (ucol_strcoll(coll, source, sourceLength, target, targetLength)
== UCOL_EQUAL);
}
#define MIN_VALUE 0x02
#define UCOL_VARIABLE_MAX 0x20
#define UCOL_NEW_IGNORABLE 0
U_CFUNC uint8_t *ucol_getSortKeyWithAllocation(const UCollator *coll,
const UChar *source,
int32_t sourceLength,
int32_t *resultLen) {
uint8_t *result = NULL;
UErrorCode status = U_ZERO_ERROR;
*resultLen = ucol_calcSortKey(coll, source, sourceLength, &result, 0, TRUE, &status);
return result;
}
int32_t ucol_getSortKeySize(const UCollator *coll, collIterate *s, int32_t currentSize, UColAttributeValue strength, int32_t len) {
UErrorCode status = U_ZERO_ERROR;
uint8_t compareSec = (strength >= UCOL_SECONDARY)?0:0xFF;
uint8_t compareTer = (strength >= UCOL_TERTIARY)?0:0xFF;
uint8_t compareQuad = (strength >= UCOL_QUATERNARY)?0:0xFF;
UBool compareIdent = (strength == UCOL_IDENTICAL);
UBool doCase = (coll->caseLevel == UCOL_ON);
UBool shifted = (coll->alternateHandling == UCOL_SHIFTED);
uint8_t variableMax = coll->variableMax;
int32_t order = UCOL_NULLORDER;
uint16_t primary = 0;
uint8_t primary1 = 0;
uint8_t primary2 = 0;
uint8_t primary3 = 0;
uint32_t ce = 0;
uint8_t secondary = 0;
uint8_t tertiary = 0;
int32_t caseShift = 0;
for(;;) {
order = ucol_getNextCE(coll, s, &status);
/*UCOL_GETNEXTCE(order, coll, *s, &status);*/
if(order == UCOL_NULLORDER) {
break;
}
/* We're saving order in ce, since we will destroy order in order to get primary, secondary, tertiary in order ;)*/
ce = order;
tertiary = (order & UCOL_TERTIARYORDERMASK);
secondary = (order >>= 8) & 0xFF;
primary3 = 0; /* the third primary */
primary2 = (order >>= 8) & 0xFF;;
primary1 = order >>= 8;
if(isLongPrimary(ce)) {
/* if we have a long primary, we'll mark secondary unmarked & add min value to tertiary */
primary3 = secondary;
secondary = UCOL_UNMARKED;
tertiary += MIN_VALUE;
}
if(shifted && primary1 < variableMax && primary1 != 0) {
currentSize++;
if(primary2 != 0) {
currentSize++;
}
} else {
/* Note: This code assumes that the table is well built i.e. not having 0 bytes where they are not supposed to be. */
/* Usually, we'll have non-zero primary1 & primary2, except in cases of LatinOne and friends, when primary2 will */
/* be zero with non zero primary1. primary3 is different than 0 only for long primaries - see above. */
if(primary1 != UCOL_NEW_IGNORABLE) {
currentSize++;
if(primary2 != UCOL_NEW_IGNORABLE) {
currentSize++;
if(primary3 != UCOL_NEW_IGNORABLE) {
currentSize++;
}
}
}
if(secondary > compareSec) { /* I think that != 0 test should be != IGNORABLE */
/* This thing should also contain the compression logic, as in: */
/*
if (ws == COMMON2 && COMMON2 <= secondary[-1] && secondary[-1] < COMMON_MAX2)
++secondary[-1]; // simply increment!!
else *secondary++ = ws;
*/
currentSize++;
}
if(doCase) {
if (caseShift == 0) {
currentSize++;
caseShift = 7;
}
caseShift--;
}
if(tertiary > compareTer) { /* I think that != 0 test should be != IGNORABLE */
/* This thing should also contain the compression logic, as in: */
/*
if (ws == COMMON2 && COMMON2 <= secondary[-1] && secondary[-1] < COMMON_MAX2)
++secondary[-1]; // simply increment!!
else *secondary++ = ws;
*/
currentSize++;
}
if(shifted && primary1 > compareQuad) {
currentSize++;
}
}
}
if(compareIdent) {
currentSize += len*sizeof(UChar);
UChar *ident = s->string;
while(ident<s->len) {
if((*(ident) >> 8) + utf16fixup[*(ident) >> 11]<0x02) {
currentSize++;
}
if((*(ident) & 0xFF)<0x02) {
currentSize++;
}
}
}
return currentSize;
}
int32_t
ucol_calcSortKey(const UCollator *coll,
const UChar *source,
int32_t sourceLength,
uint8_t **result,
int32_t resultLength,
UBool allocatePrimary,
UErrorCode *status)
{
uint32_t i = 0; /* general purpose counter */
/* Stack allocated buffers for buffers we use */
uint8_t second[UCOL_MAX_BUFFER], tert[UCOL_MAX_BUFFER], caseB[UCOL_MAX_BUFFER], quad[UCOL_MAX_BUFFER];
uint8_t *primaries = *result, *secondaries = second, *tertiaries = tert, *cases = caseB, *quads = quad;
if(U_FAILURE(*status)) {
return 0;
}
if(primaries == NULL && allocatePrimary == TRUE) {
primaries = *result = (uint8_t *)uprv_malloc(2*UCOL_MAX_BUFFER);
resultLength = 2*UCOL_MAX_BUFFER;
}
int32_t primSize = resultLength, secSize = UCOL_MAX_BUFFER, terSize = UCOL_MAX_BUFFER,
caseSize = UCOL_MAX_BUFFER, quadSize = UCOL_MAX_BUFFER;
int32_t sortKeySize = 1; // it is always \0 terminated
UChar normBuffer[UCOL_NORMALIZATION_GROWTH*UCOL_MAX_BUFFER];
UChar *normSource = normBuffer;
int32_t normSourceLen = UCOL_NORMALIZATION_GROWTH*UCOL_MAX_BUFFER;
int32_t len = (sourceLength == -1 ? u_strlen(source) : sourceLength);
uint8_t variableMax = coll->variableMax;
UColAttributeValue strength = coll->strength;
uint8_t compareSec = (strength >= UCOL_SECONDARY)?0:0xFF;
uint8_t compareTer = (strength >= UCOL_TERTIARY)?0:0xFF;
uint8_t compareQuad = (strength >= UCOL_QUATERNARY)?0:0xFF;
UBool compareIdent = (strength == UCOL_IDENTICAL);
UBool doCase = (coll->caseLevel == UCOL_ON);
UBool upperFirst = (coll->caseFirst == UCOL_UPPER_FIRST);
UBool shifted = (coll->alternateHandling == UCOL_SHIFTED);
UBool isFrenchSec = (coll->frenchCollation == UCOL_ON);
/* support for special features like caselevel and funky secondaries */
uint8_t *frenchStartPtr = NULL;
uint8_t *frenchEndPtr = NULL;
uint32_t caseShift = 0;
sortKeySize += ((compareSec?0:1) + (compareTer?0:1) + (doCase?1:0) + (compareQuad?0:1) + (compareIdent?1:0));
collIterate s;
init_collIterate((UChar *)source, len, &s, FALSE);
// If we need to normalize, we'll do it all at once at the beggining!
UColAttributeValue normMode = coll->normalizationMode;
if(normMode != UCOL_OFF) {
normSourceLen = u_normalize(source, sourceLength, UNORM_NFD, 0, normSource, normSourceLen, status);
if(U_FAILURE(*status)) {
*status=U_ZERO_ERROR;
normSource = (UChar *) uprv_malloc((normSourceLen+1)*sizeof(UChar));
normSourceLen = u_normalize(source, sourceLength, UNORM_NFD, 0, normSource, (normSourceLen+1), status);
}
normSource[normSourceLen] = 0;
s.string = normSource;
s.pos = normSource;
s.len = normSource+normSourceLen;
}
len = s.len-s.pos;
if(resultLength == 0) {
return ucol_getSortKeySize(coll, &s, sortKeySize, strength, len);
}
int32_t minBufferSize = UCOL_MAX_BUFFER;
uint8_t *primStart = primaries;
uint8_t *secStart = secondaries;
uint8_t *terStart = tertiaries;
uint8_t *caseStart = cases;
uint8_t *quadStart = quads;
*(secondaries++) = UCOL_LEVELTERMINATOR;
*(tertiaries++) = UCOL_LEVELTERMINATOR;
*(cases++) = UCOL_LEVELTERMINATOR;
*(quads++) = UCOL_LEVELTERMINATOR;
uint32_t order = 0;
uint32_t ce = 0;
uint8_t carry = 0;
uint8_t primary1 = 0;
uint8_t primary2 = 0;
uint8_t primary3 = 0;
uint8_t secondary = 0;
uint8_t tertiary = 0;
UBool finished = FALSE;
UBool resultOverflow = FALSE;
int32_t prevBuffSize = 0;
int32_t compressedSecs = 0;
for(;;) {
for(i=prevBuffSize; i<minBufferSize; ++i) {
order = ucol_getNextCE(coll, &s, status);
/*UCOL_GETNEXTCENEW(order, coll, s, status);*/
if(order == UCOL_NULLORDER) {
finished = TRUE;
break;
}
/* We're saving order in ce, since we will destroy order in order to get primary, secondary, tertiary in order ;)*/
ce = order;
tertiary = (order & UCOL_NEW_TERTIARYORDERMASK);
secondary = (order >>= 8) & 0xFF;
primary3 = 0; /* the third primary */
primary2 = (order >>= 8) & 0xFF;
primary1 = order >>= 8;
if(upperFirst && ((ce & 0x80) == 0)) { /* if there is a case bit */
/* Upper cases have this bit turned on, so that they always come after the lower cases */
/* if we want to reverse this situation, we'll flip this bit */
tertiary ^= UCOL_CASE_BIT_MASK;
}
if(isLongPrimary(ce)) {
/* if we have a long primary, we'll mark secondary unmarked & add min value to tertiary */
primary3 = secondary;
secondary = UCOL_UNMARKED;
tertiary += MIN_VALUE;
}
/* In the code below, every increase in any of buffers is followed by the increase to */
/* sortKeySize - this might look tedious, but it is needed so that we can find out if */
/* we're using too much space and need to reallocate the primary buffer or easily bail */
/* out to ucol_getSortKeySizeNew. */
if(shifted && primary1 < variableMax && primary1 != 0) {
/* We are dealing with a variable and we're treating them as shifted */
/* This is a shifted ignorable */
*quads++ = primary1;
sortKeySize++;
if(primary2 != 0) {
*quads++ = primary2;
sortKeySize++;
}
} else {
/* Note: This code assumes that the table is well built i.e. not having 0 bytes where they are not supposed to be. */
/* Usually, we'll have non-zero primary1 & primary2, except in cases of LatinOne and friends, when primary2 will */
/* be zero with non zero primary1. primary3 is different than 0 only for long primaries - see above. */
if(primary1 != UCOL_NEW_IGNORABLE) {
*primaries++ = primary1; /* scriptOrder[primary1]; */ /* This is the script ordering thingie */
sortKeySize++;
if(primary2 != UCOL_NEW_IGNORABLE) {
*primaries++ = primary2; /* second part */
sortKeySize++;
if(primary3 != UCOL_NEW_IGNORABLE) {
*primaries++ = primary3; /* third part */
sortKeySize++;
}
}
}
if(secondary > compareSec) {
/* This thing should also contain the compression logic, as in: */
if (secondary == UCOL_COMMON2 && *(secondaries-1) == UCOL_COMMON2 && compressedSecs < UCOL_COMMON_MAX2 - UCOL_COMMON2) {
compressedSecs++;
} else {
if(compressedSecs != 0) {
*(secondaries-1) = UCOL_COMMON_MAX2 - compressedSecs;
compressedSecs = 0;
}
*secondaries++ = secondary;
sortKeySize++;
if(isFrenchSec) {
/* Do the special handling for French secondaries */
/* We need to get continuation elements and do intermediate restore */
/* abc1c2c3de with french secondaries need to be edc1c2c3ba NOT edc3c2c1ba */
if(isContinuation(ce)) {
if (frenchStartPtr == NULL) {
frenchStartPtr = secondaries - 2;
}
frenchEndPtr = secondaries-1;
} else if (frenchStartPtr != NULL) {
/* reverse secondaries from frenchStartPtr up to frenchEndPtr */
uprv_ucol_reverse_buffer(frenchStartPtr, frenchEndPtr);
frenchStartPtr = NULL;
}
}
}
}
if(doCase) {
if (caseShift == 0) {
*cases++ = 0x80;
sortKeySize++;
caseShift = 7;
}
*(cases-1) |= (tertiary & 0x80) >> (8-caseShift--);
}
if(tertiary > compareTer) {
/* This thing should also contain the compression logic, as in: */
if (tertiary == UCOL_COMMON3 && UCOL_COMMON3 <= *(tertiaries-1) && *(tertiaries-1) < UCOL_COMMON_MAX3) {
++*(tertiaries-1); /* simply increment!! */
} else {
*tertiaries++ = tertiary;
sortKeySize++;
}
}
if(shifted && primary1 > compareQuad) {
/* Some compression should go here also */
//if (tertiary == UCOL_COMMON3 && UCOL_COMMON3 <= *(tertiaries-1) && *(tertiaries-1) < UCOL_COMMON_MAX3) {
// ++*(tertiaries-1); /* simply increment!! */
//} else {
*quads++ = 0xFF;
sortKeySize++;
}
}
if(sortKeySize>resultLength) { /* We have stepped over the primary buffer */
if(allocatePrimary == FALSE) { /* need to save our butts if we cannot reallocate */
resultOverflow = TRUE;
sortKeySize = ucol_getSortKeySize(coll, &s, sortKeySize, strength, len);
*status = U_MEMORY_ALLOCATION_ERROR;
finished = TRUE;
break;
} else { /* It's much nicer if we can actually reallocate */
uint8_t *newStart;
newStart = (uint8_t *)uprv_realloc(primStart, 2*sortKeySize);
if(primStart == NULL) {
*status = U_MEMORY_ALLOCATION_ERROR;
finished = TRUE;
break;
}
primaries=newStart+(primaries-primStart);
resultLength = 2*sortKeySize;
primStart = *result = newStart;
}
}
}
if(finished) {
break;
} else {
prevBuffSize = minBufferSize;
secStart = reallocateBuffer(&secondaries, secStart, second, &secSize, status);
terStart = reallocateBuffer(&tertiaries, terStart, tert, &terSize, status);
caseStart = reallocateBuffer(&cases, caseStart, cases, &caseSize, status);
quadStart = reallocateBuffer(&quads, quadStart, quads, &quadSize, status);
minBufferSize *= 2;
}
}
if(U_SUCCESS(*status)) {
/* we have done all the CE's, now let's put them together to form a key */
if(compareSec == 0) {
*(primaries++) = UCOL_LEVELTERMINATOR;
uint32_t secsize = secondaries-secStart;
if(isFrenchSec) { /* do the reverse copy */
/* If there are any unresolved continuation secondaries, reverse them here so that we can reverse the whole secondary thing */
if(frenchStartPtr != NULL) {
uprv_ucol_reverse_buffer(frenchStartPtr, frenchEndPtr);
}
for(i = 1; i<secsize; i++) {
*(primaries++) = *(secondaries-i-1);
}
} else {
uprv_memcpy(primaries, secStart+1, secsize);
primaries += secsize;
}
}
if(doCase) {
uint32_t casesize = cases - caseStart;
uprv_memcpy(primaries, caseStart, casesize);
primaries += casesize;
}
if(compareTer == 0) {
uint32_t tersize = tertiaries - terStart;
uprv_memcpy(primaries, terStart, tersize);
primaries += tersize;
}
if(compareQuad == 0) {
uint32_t quadsize = quads - quadStart;
uprv_memcpy(primaries, quadStart, quadsize);
primaries += quadsize;
}
if(compareIdent) {
UChar *ident = s.string;
uint8_t idByte = 0;
sortKeySize += len * sizeof(UChar);
*(primaries++) = UCOL_LEVELTERMINATOR;
if(sortKeySize <= resultLength) {
while(ident < s.len) {
idByte = (*(ident) >> 8) + utf16fixup[*(ident) >> 11];
if(idByte < 0x02) {
if(sortKeySize < resultLength) {
*(primaries++) = 0x01;
sortKeySize++;
*(primaries++) = idByte + 1;
}
} else {
*(primaries++) = idByte;
}
idByte = (*(ident) & 0xFF);
if(idByte < 0x02) {
if(sortKeySize < resultLength) {
*(primaries++) = 0x01;
sortKeySize++;
*(primaries++) = idByte + 1;
}
} else {
*(primaries++) = idByte;
}
ident++;
}
} else {
while(ident < s.len) {
idByte = (*(ident) >> 8) + utf16fixup[*(ident) >> 11];
if(idByte < 0x02) {
sortKeySize++;
}
idByte = (*(ident) & 0xFF);
if(idByte < 0x02) {
sortKeySize++;
}
ident++;
}
}
}
*(primaries++) = '\0';
} else {
sortKeySize = 0;
}
if(terStart != tert) {
uprv_free(terStart);
uprv_free(secStart);
uprv_free(caseStart);
uprv_free(quadStart);
}
if(normSource != normBuffer) {
uprv_free(normSource);
}
return sortKeySize;
}
U_CAPI int32_t
ucol_getSortKey(const UCollator *coll,
const UChar *source,
int32_t sourceLength,
uint8_t *result,
int32_t resultLength)
{
UErrorCode status = U_ZERO_ERROR;
return ucol_calcSortKey(coll, source, sourceLength, &result, resultLength, FALSE, &status);
}
U_CAPI void ucol_setAttribute(UCollator *coll, UColAttribute attr, UColAttributeValue value, UErrorCode *status) {
switch(attr) {
case UCOL_FRENCH_COLLATION: /* attribute for direction of secondary weights*/
if(value == UCOL_ON) {
coll->frenchCollation = UCOL_ON;
coll->frenchCollationisDefault = FALSE;
} else if (value == UCOL_OFF) {
coll->frenchCollation = UCOL_OFF;
coll->frenchCollationisDefault = FALSE;
} else if (value == UCOL_DEFAULT) {
coll->frenchCollationisDefault = TRUE;
coll->frenchCollation = coll->image->frenchCollation;
} else {
*status = U_ILLEGAL_ARGUMENT_ERROR ;
}
break;
case UCOL_ALTERNATE_HANDLING: /* attribute for handling variable elements*/
if(value == UCOL_SHIFTED) {
coll->alternateHandling = UCOL_SHIFTED;
coll->alternateHandlingisDefault = FALSE;
} else if (value == UCOL_NON_IGNORABLE) {
coll->alternateHandling = UCOL_NON_IGNORABLE;
coll->alternateHandlingisDefault = FALSE;
} else if (value == UCOL_DEFAULT) {
coll->alternateHandlingisDefault = TRUE;
coll->alternateHandling = coll->image->alternateHandling ;
} else {
*status = U_ILLEGAL_ARGUMENT_ERROR ;
}
break;
case UCOL_CASE_FIRST: /* who goes first, lower case or uppercase */
if(value == UCOL_LOWER_FIRST) {
coll->caseFirst = UCOL_LOWER_FIRST;
coll->caseFirstisDefault = FALSE;
} else if (value == UCOL_UPPER_FIRST) {
coll->caseFirst = UCOL_UPPER_FIRST;
coll->caseFirstisDefault = FALSE;
} else if (value == UCOL_DEFAULT) {
coll->caseFirst = coll->image->caseFirst;
coll->caseFirstisDefault = TRUE;
} else {
*status = U_ILLEGAL_ARGUMENT_ERROR ;
}
break;
case UCOL_CASE_LEVEL: /* do we have an extra case level */
if(value == UCOL_ON) {
coll->caseLevel = UCOL_ON;
coll->caseLevelisDefault = FALSE;
} else if (value == UCOL_OFF) {
coll->caseLevel = UCOL_OFF;
coll->caseLevelisDefault = FALSE;
} else if (value == UCOL_DEFAULT) {
coll->caseLevel = coll->image->caseLevel;
coll->caseLevelisDefault = TRUE;
} else {
*status = U_ILLEGAL_ARGUMENT_ERROR ;
}
break;
case UCOL_NORMALIZATION_MODE: /* attribute for normalization */
if(value == UCOL_ON) {
coll->normalizationMode = UCOL_ON;
coll->normalizationModeisDefault = FALSE;
} else if (value == UCOL_OFF) {
coll->normalizationMode = UCOL_OFF;
coll->normalizationModeisDefault = FALSE;
} else if (value == UCOL_ON_WITHOUT_HANGUL) {
coll->normalizationMode = UCOL_ON_WITHOUT_HANGUL ;
coll->normalizationModeisDefault = FALSE;
} else if (value == UCOL_DEFAULT) {
coll->normalizationModeisDefault = TRUE;
coll->normalizationMode = coll->image->normalizationMode;
} else {
*status = U_ILLEGAL_ARGUMENT_ERROR ;
}
break;
case UCOL_STRENGTH: /* attribute for strength */
if (value == UCOL_DEFAULT) {
coll->strengthisDefault = TRUE;
coll->strength = coll->image->strength;
} else if (value <= UCOL_IDENTICAL) {
coll->strengthisDefault = FALSE;
coll->strength = value;
} else {
*status = U_ILLEGAL_ARGUMENT_ERROR ;
}
break;
case UCOL_ATTRIBUTE_COUNT:
default:
*status = U_ILLEGAL_ARGUMENT_ERROR;
break;
}
}
U_CAPI UColAttributeValue ucol_getAttribute(const UCollator *coll, UColAttribute attr, UErrorCode *status) {
switch(attr) {
case UCOL_FRENCH_COLLATION: /* attribute for direction of secondary weights*/
if(coll->frenchCollationisDefault) {
return coll->image->frenchCollation;
} else {
return coll->frenchCollation;
}
break;
case UCOL_ALTERNATE_HANDLING: /* attribute for handling variable elements*/
if(coll->alternateHandlingisDefault) {
return coll->image->alternateHandling;
} else {
return coll->alternateHandling;
}
break;
case UCOL_CASE_FIRST: /* who goes first, lower case or uppercase */
if(coll->caseFirstisDefault) {
return coll->image->caseFirst;
} else {
return coll->caseFirst;
}
break;
case UCOL_CASE_LEVEL: /* do we have an extra case level */
if(coll->caseLevelisDefault) {
return coll->image->caseLevel;
} else {
return coll->caseLevel;
}
break;
case UCOL_NORMALIZATION_MODE: /* attribute for normalization */
if(coll->normalizationModeisDefault) {
return coll->image->normalizationMode;
} else {
return coll->normalizationMode;
}
break;
case UCOL_STRENGTH: /* attribute for strength */
if(coll->strengthisDefault) {
return coll->image->strength;
} else {
return coll->strength;
}
break;
case UCOL_ATTRIBUTE_COUNT:
default:
*status = U_ILLEGAL_ARGUMENT_ERROR;
break;
}
return UCOL_DEFAULT;
}
U_CAPI void
ucol_setNormalization( UCollator *coll,
UNormalizationMode mode)
{
UErrorCode status = U_ZERO_ERROR;
switch(mode) {
case UCOL_NO_NORMALIZATION:
ucol_setAttribute(coll, UCOL_NORMALIZATION_MODE, UCOL_OFF, &status);
break;
case UCOL_DECOMP_CAN:
ucol_setAttribute(coll, UCOL_NORMALIZATION_MODE, UCOL_ON, &status);
break;
default:
/* Shouldn't get here. */
/* This is quite a bad API */
/* deprecate */
/* *status = U_ILLEGAL_ARGUMENT_ERROR; */
return;
}
}
/* This is the new function */
/* This is the incremental function */
U_CAPI UCollationResult ucol_strcollinc(const UCollator *coll,
UCharForwardIterator *source, void *sourceContext,
UCharForwardIterator *target, void *targetContext)
{
Collator *cppColl = (Collator*)coll;
UCollationResult result = UCOL_EQUAL;
UErrorCode status = U_ZERO_ERROR;
incrementalContext sColl, tColl;
init_incrementalContext(source, sourceContext, &sColl);
init_incrementalContext(target, targetContext, &tColl);
if(cppColl->getDecomposition() != Normalizer::NO_OP) { // run away screaming!!!!
return alternateIncrementalProcessing(coll, &sColl, &tColl);
}
if (U_FAILURE(status))
{
return UCOL_EQUAL;
}
UColAttributeValue strength = ucol_getAttribute(coll, UCOL_STRENGTH, &status);
uint32_t sOrder=UCOL_NULLORDER, tOrder=UCOL_NULLORDER;
uint32_t pSOrder, pTOrder;
UBool gets = TRUE, gett = TRUE;
UBool initialCheckSecTer = strength >= UCOL_SECONDARY;
UBool checkSecTer = initialCheckSecTer;
UBool checkTertiary = strength >= UCOL_TERTIARY;
UBool checkQuad = strength >= UCOL_QUATERNARY;
UBool isFrenchSec = (cppColl->getAttribute(UCOL_FRENCH_COLLATION, status) == UCOL_ON) && checkSecTer;
if(!isFrenchSec) {
for(;;)
{
// Get the next collation element in each of the strings, unless
// we've been requested to skip it.
if (gets)
{
sOrder = ucol_getIncrementalCE(coll, &sColl, &status);
}
gets = TRUE;
if (gett)
{
tOrder = ucol_getIncrementalCE(coll, &tColl, &status);
}
gett = TRUE;
// If we've hit the end of one of the strings, jump out of the loop
if ((sOrder == UCOL_NULLORDER)||
(tOrder == UCOL_NULLORDER)) {
if(sColl.panic == TRUE || tColl.panic == TRUE) {
return alternateIncrementalProcessing(coll, &sColl, &tColl);
}
break;
}
// If there's no difference at this position, we can skip to the
// next one.
if (sOrder == tOrder)
{
continue;
}
// Compare primary differences first.
pSOrder = UCOL_PRIMARYORDER(sOrder);
pTOrder = UCOL_PRIMARYORDER(tOrder);
if (pSOrder != pTOrder)
{
if (sOrder == UCOL_IGNORABLE)
{
// The entire source element is ignorable.
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
continue;
}
if (tOrder == UCOL_IGNORABLE)
{
gets = FALSE;
continue;
}
// The source and target elements aren't ignorable, but it's still possible
// for the primary component of one of the elements to be ignorable....
if (pSOrder == UCOL_PRIMIGNORABLE) // primary order in source is ignorable
{
// The source's primary is ignorable, but the target's isn't. We treat ignorables
// as a secondary difference, so remember that we found one.
if (checkSecTer)
{
result = UCOL_GREATER; // (strength is SECONDARY) - still need to check for tertiary or quad
checkSecTer = FALSE;
}
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
}
else if (pTOrder == UCOL_PRIMIGNORABLE)
{
// record differences - see the comment above.
if (checkSecTer)
{
result = UCOL_LESS; // (strength is SECONDARY) - still need to check for tertiary or quad
checkSecTer = FALSE;
}
// Skip to the next target element, but don't fetch another source element.
gets = FALSE;
}
else
{
// Neither of the orders is ignorable, and we already know that the primary
// orders are different because of the (pSOrder != pTOrder) test above.
// Record the difference and stop the comparison.
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
if (pSOrder < pTOrder)
{
return UCOL_LESS; // (strength is PRIMARY)
}
return UCOL_GREATER; // (strength is PRIMARY)
}
}
else
{ // else of if ( pSOrder != pTOrder )
// primary order is the same, but complete order is different. So there
// are no base elements at this point, only ignorables (Since the strings are
// normalized)
if (checkSecTer)
{
// a secondary or tertiary difference may still matter
uint32_t secSOrder = UCOL_SECONDARYORDER(sOrder);
uint32_t secTOrder = UCOL_SECONDARYORDER(tOrder);
if (secSOrder != secTOrder)
{
// there is a secondary difference
result = (secSOrder < secTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is SECONDARY)
checkSecTer = FALSE;
}
else
{
if (checkTertiary)
{
// a tertiary difference may still matter
uint32_t terSOrder = UCOL_TERTIARYORDER(sOrder);
uint32_t terTOrder = UCOL_TERTIARYORDER(tOrder);
if (terSOrder != terTOrder)
{
// there is a tertiary difference
result = (terSOrder < terTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is TERTIARY)
checkTertiary = FALSE;
}
}
}
} // if (checkSecTer)
} // if ( pSOrder != pTOrder )
} // while()
if (sOrder != UCOL_NULLORDER)
{
// (tOrder must be CollationElementIterator::NULLORDER,
// since this point is only reached when sOrder or tOrder is NULLORDER.)
// The source string has more elements, but the target string hasn't.
do
{
if (UCOL_PRIMARYORDER(sOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the source string.
// This is a primary difference, so the source is greater
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
return UCOL_GREATER; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(sOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements mean the source string is greater
if (checkSecTer)
{
result = UCOL_GREATER; // (strength is SECONDARY)
checkSecTer = FALSE;
}
}
sOrder = ucol_getIncrementalCE(coll, &sColl, &status);
}
//while ((sOrder = ucol_getIncrementalCE(coll, &sColl, &status)) != CollationElementIterator::NULLORDER);
while (sOrder != UCOL_NULLORDER);
}
else if (tOrder != UCOL_NULLORDER)
{
// The target string has more elements, but the source string hasn't.
do
{
if (UCOL_PRIMARYORDER(tOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the target string.
// This is a primary difference, so the source is less
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
return UCOL_LESS; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(tOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements in the target mean the source string is less
if (checkSecTer)
{
result = UCOL_LESS; // (strength is SECONDARY)
checkSecTer = FALSE;
}
}
tOrder = ucol_getIncrementalCE(coll, &tColl, &status);
}
while ( tOrder != UCOL_NULLORDER);
//while ((tOrder = ucol_getIncrementalCE(coll, &tColl, &status)) != CollationElementIterator::NULLORDER);
}
} else { //French
// there is a bad situation with French when there is a different number of secondaries...
// If that situation arises (when one primary is ignorable with nonignorable secondary and the other primary is not
// ignorable
// TODO: if the buffer is not big enough, we should use sortkeys
UBool bufferFrenchSec = FALSE;
uint32_t sourceFrenchSec[UCOL_MAX_BUFFER], targetFrenchSec[UCOL_MAX_BUFFER];
uint32_t *sFSBEnd = sourceFrenchSec+UCOL_MAX_BUFFER;
uint32_t *tFSBEnd = targetFrenchSec+UCOL_MAX_BUFFER;
for(;;)
{
// Get the next collation element in each of the strings, unless
// we've been requested to skip it.
if (gets)
{
sOrder = ucol_getIncrementalCE(coll, &sColl, &status);
*(--sFSBEnd) = UCOL_SECONDARYORDER(sOrder);
if(sFSBEnd == sourceFrenchSec) { /* overflowing the buffer, bail out */
return alternateIncrementalProcessing(coll, &sColl, &tColl);
}
}
gets = TRUE;
if (gett)
{
tOrder = ucol_getIncrementalCE(coll, &tColl, &status);
*(--tFSBEnd) = UCOL_SECONDARYORDER(tOrder);
if(tFSBEnd == targetFrenchSec) { /* overflowing the buffer, bail out */
return alternateIncrementalProcessing(coll, &sColl, &tColl);
}
}
gett = TRUE;
// If we've hit the end of one of the strings, jump out of the loop
if ((sOrder == UCOL_NULLORDER)||
(tOrder == UCOL_NULLORDER)) {
break;
}
// If there's no difference at this position, we can skip to the
// next one.
if (sOrder == tOrder)
{
continue;
}
// Compare primary differences first.
pSOrder = UCOL_PRIMARYORDER(sOrder);
pTOrder = UCOL_PRIMARYORDER(tOrder);
if (pSOrder != pTOrder)
{
if (sOrder == UCOL_IGNORABLE)
{
// The entire source element is ignorable.
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
continue;
}
if (tOrder == UCOL_IGNORABLE)
{
gets = FALSE;
continue;
}
// The source and target elements aren't ignorable, but it's still possible
// for the primary component of one of the elements to be ignorable....
if (pSOrder == UCOL_PRIMIGNORABLE) // primary order in source is ignorable
{
// The source's primary is ignorable, but the target's isn't. We treat ignorables
// as a secondary difference, so remember that we found one.
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
}
else if (pTOrder == UCOL_PRIMIGNORABLE)
{
// record differences - see the comment above.
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
// Skip to the next target element, but don't fetch another source element.
gets = FALSE;
}
else
{
// Neither of the orders is ignorable, and we already know that the primary
// orders are different because of the (pSOrder != pTOrder) test above.
// Record the difference and stop the comparison.
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
if (pSOrder < pTOrder)
{
return UCOL_LESS; // (strength is PRIMARY)
}
return UCOL_GREATER; // (strength is PRIMARY)
}
}
else
{ // else of if ( pSOrder != pTOrder )
// primary order is the same, but complete order is different. So there
// are no base elements at this point, only ignorables (Since the strings are
// normalized)
if (checkSecTer)
{
// a secondary or tertiary difference may still matter
uint32_t secSOrder = UCOL_SECONDARYORDER(sOrder);
uint32_t secTOrder = UCOL_SECONDARYORDER(tOrder);
if (secSOrder != secTOrder)
{
// there is a secondary difference
result = (secSOrder < secTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is SECONDARY)
checkSecTer = isFrenchSec; // We still want to track the French secondaries
//checkSecTer = FALSE;
}
else
{
if (checkTertiary)
{
// a tertiary difference may still matter
uint32_t terSOrder = UCOL_TERTIARYORDER(sOrder);
uint32_t terTOrder = UCOL_TERTIARYORDER(tOrder);
if (terSOrder != terTOrder)
{
// there is a tertiary difference
result = (terSOrder < terTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is TERTIARY)
checkTertiary = FALSE;
}
}
}
} // if (checkSecTer)
} // if ( pSOrder != pTOrder )
} // while()
if (sOrder != UCOL_NULLORDER)
{
// (tOrder must be CollationElementIterator::NULLORDER,
// since this point is only reached when sOrder or tOrder is NULLORDER.)
// The source string has more elements, but the target string hasn't.
do
{
if (UCOL_PRIMARYORDER(sOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the source string.
// This is a primary difference, so the source is greater
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
return UCOL_GREATER; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(sOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements mean the source string is greater
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
}
sOrder = ucol_getIncrementalCE(coll, &sColl, &status);
*(--sFSBEnd) = UCOL_SECONDARYORDER(sOrder);
if(sFSBEnd == sourceFrenchSec) { /* overflowing the buffer, bail out */
return alternateIncrementalProcessing(coll, &sColl, &tColl);
}
}
//while ((sOrder = ucol_getIncrementalCE(coll, &sColl, &status)) != CollationElementIterator::NULLORDER);
while (sOrder != UCOL_NULLORDER);
}
else if (tOrder != UCOL_NULLORDER)
{
// The target string has more elements, but the source string hasn't.
do
{
if (UCOL_PRIMARYORDER(tOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the target string.
// This is a primary difference, so the source is less
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
return UCOL_LESS; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(tOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements in the target mean the source string is less
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
}
tOrder = ucol_getIncrementalCE(coll, &tColl, &status);
*(--tFSBEnd) = UCOL_SECONDARYORDER(tOrder);
if(tFSBEnd == targetFrenchSec) { /* overflowing the buffer, bail out */
return alternateIncrementalProcessing(coll, &sColl, &tColl);
}
}
while ( tOrder != UCOL_NULLORDER);
}
if(bufferFrenchSec) {
while(sFSBEnd < sourceFrenchSec+UCOL_MAX_BUFFER && tFSBEnd < targetFrenchSec+UCOL_MAX_BUFFER) {
if(*sFSBEnd == *tFSBEnd) {
sFSBEnd++;
tFSBEnd++;
} else if(*sFSBEnd < *tFSBEnd) {
result = UCOL_LESS;
break;
} else {
result = UCOL_GREATER;
break;
}
}
}
}
// For IDENTICAL comparisons, we use a bitwise character comparison
// as a tiebreaker if all else is equal
// NOTE: The java code compares result with 0, and
// puts the result of the string comparison directly into result
if (result == UCOL_EQUAL && strength == UCOL_IDENTICAL)
{
UnicodeString sourceDecomp, targetDecomp;
int8_t comparison;
Normalizer::normalize(UnicodeString(sColl.stringP, sColl.len-sColl.stringP-1), ((RuleBasedCollator *)coll)->getDecomposition(),
0, sourceDecomp, status);
Normalizer::normalize(UnicodeString(tColl.stringP, tColl.len-tColl.stringP-1), ((RuleBasedCollator *)coll)->getDecomposition(),
0, targetDecomp, status);
comparison = sourceDecomp.compare(targetDecomp);
if (comparison < 0)
{
result = UCOL_LESS;
}
else if (comparison == 0)
{
result = UCOL_EQUAL;
}
else
{
result = UCOL_GREATER;
}
}
incctx_cleanUpContext(&sColl);
incctx_cleanUpContext(&tColl);
return result;
}
/* This is the new function */
U_CAPI UCollationResult
ucol_strcoll( const UCollator *coll,
const UChar *source,
int32_t sourceLength,
const UChar *target,
int32_t targetLength)
{
#if 0
Collator *cppColl = (Collator*)coll;
// check if source and target are valid strings
if (((source == 0) && (target == 0)) ||
((sourceLength == 0) && (targetLength == 0)))
{
return UCOL_EQUAL;
}
UCollationResult result = UCOL_EQUAL;
UErrorCode status = U_ZERO_ERROR;
UChar normSource[UCOL_MAX_BUFFER], normTarget[UCOL_MAX_BUFFER];
UChar *normSourceP = normSource;
UChar *normTargetP = normTarget;
uint32_t normSourceLength = UCOL_MAX_BUFFER, normTargetLength = UCOL_MAX_BUFFER;
collIterate sColl, tColl;
if(cppColl->getDecomposition() == Normalizer::NO_OP) {
init_collIterate(source, sourceLength == -1 ? u_strlen(source) : sourceLength, &sColl, FALSE);
init_collIterate(target, targetLength == -1 ? u_strlen(target) : targetLength, &tColl, FALSE);
} else { /* TODO: This is bad behaved if we're working with small buffers */
/* We really need the normalization quick check here*/
UNormalizationMode normMode = ucol_getNormalization(coll);
normSourceLength = u_normalize(source, sourceLength, normMode, 0, normSource, normSourceLength, &status);
if(U_FAILURE(status)) { /* This would be buffer overflow */
normSourceP = (UChar *)uprv_malloc((normSourceLength+1)*sizeof(UChar));
status = U_ZERO_ERROR;
normSourceLength = u_normalize(source, sourceLength, normMode, 0, normSourceP, normSourceLength+1, &status);
normTargetLength = u_normalize(target, targetLength, normMode, 0, normTargetP, normTargetLength, &status);
if(U_FAILURE(status)) { /* This would be buffer overflow */
normTargetP = (UChar *)uprv_malloc((normTargetLength+1)*sizeof(UChar));
status = U_ZERO_ERROR;
normTargetLength = u_normalize(target, targetLength, normMode, 0, normTargetP, normTargetLength+1, &status);
}
Normalizer::EMode mode = cppColl->getDecomposition();
cppColl->setDecomposition(Normalizer::NO_OP);
UCollationResult result = ucol_strcoll(coll, normSourceP, normSourceLength, normTargetP, normTargetLength);
cppColl->setDecomposition(mode);
uprv_free(normSourceP);
if(normTargetP != normTarget) {
uprv_free(normTargetP);
}
return result;
}
normTargetLength = u_normalize(target, targetLength, normMode, 0, normTarget, normTargetLength, &status);
if(U_FAILURE(status)) { /* This would be buffer overflow */
normTargetP = (UChar *)uprv_malloc((normTargetLength+1)*sizeof(UChar));
status = U_ZERO_ERROR;
normTargetLength = u_normalize(target, targetLength, normMode, 0, normTargetP, normTargetLength+1, &status);
Normalizer::EMode mode = cppColl->getDecomposition();
cppColl->setDecomposition(Normalizer::NO_OP);
UCollationResult result = ucol_strcoll(coll, normSourceP, normSourceLength, normTargetP, normTargetLength);
cppColl->setDecomposition(mode);
uprv_free(normTargetP);
return result;
}
init_collIterate(normSource, normSourceLength, &sColl, TRUE);
init_collIterate(normTarget, normTargetLength, &tColl, TRUE);
}
if (U_FAILURE(status))
{
return UCOL_EQUAL;
}
UColAttributeValue strength = ucol_getAttribute(coll, UCOL_STRENGTH, &status);
uint32_t sOrder=UCOL_NULLORDER, tOrder=UCOL_NULLORDER;
uint32_t pSOrder, pTOrder;
UBool gets = TRUE, gett = TRUE;
UBool initialCheckSecTer = strength >= UCOL_SECONDARY;
UBool checkSecTer = initialCheckSecTer;
UBool checkTertiary = strength >= UCOL_TERTIARY;
UBool checkQuad = strength >= UCOL_QUATERNARY;
UBool isFrenchSec = (cppColl->getAttribute(UCOL_FRENCH_COLLATION, status) == UCOL_ON) && checkSecTer;
if(!isFrenchSec) {
for(;;)
{
// Get the next collation element in each of the strings, unless
// we've been requested to skip it.
if (gets)
{
UCOL_GETNEXTCE(sOrder, coll, sColl, status);
}
gets = TRUE;
if (gett)
{
UCOL_GETNEXTCE(tOrder, coll, tColl, status);
}
gett = TRUE;
// If we've hit the end of one of the strings, jump out of the loop
if ((sOrder == UCOL_NULLORDER)||
(tOrder == UCOL_NULLORDER)) {
break;
}
// If there's no difference at this position, we can skip to the
// next one.
if (sOrder == tOrder)
{
continue;
}
// Compare primary differences first.
pSOrder = UCOL_PRIMARYORDER(sOrder);
pTOrder = UCOL_PRIMARYORDER(tOrder);
if (pSOrder != pTOrder)
{
if (sOrder == UCOL_IGNORABLE)
{
// The entire source element is ignorable.
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
continue;
}
if (tOrder == UCOL_IGNORABLE)
{
gets = FALSE;
continue;
}
// The source and target elements aren't ignorable, but it's still possible
// for the primary component of one of the elements to be ignorable....
if (pSOrder == UCOL_PRIMIGNORABLE) // primary order in source is ignorable
{
// The source's primary is ignorable, but the target's isn't. We treat ignorables
// as a secondary difference, so remember that we found one.
if (checkSecTer)
{
result = UCOL_GREATER; // (strength is SECONDARY) - still need to check for tertiary or quad
checkSecTer = FALSE;
}
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
}
else if (pTOrder == UCOL_PRIMIGNORABLE)
{
// record differences - see the comment above.
if (checkSecTer)
{
result = UCOL_LESS; // (strength is SECONDARY) - still need to check for tertiary or quad
checkSecTer = FALSE;
}
// Skip to the next target element, but don't fetch another source element.
gets = FALSE;
}
else
{
// Neither of the orders is ignorable, and we already know that the primary
// orders are different because of the (pSOrder != pTOrder) test above.
// Record the difference and stop the comparison.
if (pSOrder < pTOrder)
{
return UCOL_LESS; // (strength is PRIMARY)
}
return UCOL_GREATER; // (strength is PRIMARY)
}
}
else
{ // else of if ( pSOrder != pTOrder )
// primary order is the same, but complete order is different. So there
// are no base elements at this point, only ignorables (Since the strings are
// normalized)
if (checkSecTer)
{
// a secondary or tertiary difference may still matter
uint32_t secSOrder = UCOL_SECONDARYORDER(sOrder);
uint32_t secTOrder = UCOL_SECONDARYORDER(tOrder);
if (secSOrder != secTOrder)
{
// there is a secondary difference
result = (secSOrder < secTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is SECONDARY)
checkSecTer = FALSE;
}
else
{
if (checkTertiary)
{
// a tertiary difference may still matter
uint32_t terSOrder = UCOL_TERTIARYORDER(sOrder);
uint32_t terTOrder = UCOL_TERTIARYORDER(tOrder);
if (terSOrder != terTOrder)
{
// there is a tertiary difference
result = (terSOrder < terTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is TERTIARY)
checkTertiary = FALSE;
}
}
}
} // if (checkSecTer)
} // if ( pSOrder != pTOrder )
} // while()
if (sOrder != UCOL_NULLORDER)
{
// (tOrder must be CollationElementIterator::NULLORDER,
// since this point is only reached when sOrder or tOrder is NULLORDER.)
// The source string has more elements, but the target string hasn't.
do
{
if (UCOL_PRIMARYORDER(sOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the source string.
// This is a primary difference, so the source is greater
return UCOL_GREATER; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(sOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements mean the source string is greater
if (checkSecTer)
{
result = UCOL_GREATER; // (strength is SECONDARY)
checkSecTer = FALSE;
}
}
UCOL_GETNEXTCE(sOrder, coll, sColl, status);
}
//while ((sOrder = ucol_getNextCE(coll, &sColl, &status)) != CollationElementIterator::NULLORDER);
while (sOrder != UCOL_NULLORDER);
}
else if (tOrder != UCOL_NULLORDER)
{
// The target string has more elements, but the source string hasn't.
do
{
if (UCOL_PRIMARYORDER(tOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the target string.
// This is a primary difference, so the source is less
return UCOL_LESS; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(tOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements in the target mean the source string is less
if (checkSecTer)
{
result = UCOL_LESS; // (strength is SECONDARY)
checkSecTer = FALSE;
}
}
UCOL_GETNEXTCE(tOrder, coll, tColl, status);
}
while ( tOrder != UCOL_NULLORDER);
//while ((tOrder = ucol_getNextCE(coll, &tColl, &status)) != CollationElementIterator::NULLORDER);
}
} else { //French
// there is a bad situation with French when there is a different number of secondaries...
// If that situation arises (when one primary is ignorable with nonignorable secondary and the other primary is not
// ignorable
// TODO: if the buffer is not big enough, we should use sortkeys
UBool bufferFrenchSec = FALSE;
uint32_t sourceFrenchSec[UCOL_MAX_BUFFER], targetFrenchSec[UCOL_MAX_BUFFER];
uint32_t *sFSBEnd = sourceFrenchSec+UCOL_MAX_BUFFER;
uint32_t *tFSBEnd = targetFrenchSec+UCOL_MAX_BUFFER;
for(;;)
{
// Get the next collation element in each of the strings, unless
// we've been requested to skip it.
if (gets)
{
UCOL_GETNEXTCE(sOrder, coll, sColl, status);
*(--sFSBEnd) = UCOL_SECONDARYORDER(sOrder);
if(sFSBEnd == sourceFrenchSec) { /* overflowing the buffer, bail out */
return ucol_compareUsingSortKeys(coll, source, sourceLength, target, targetLength);
}
}
gets = TRUE;
if (gett)
{
UCOL_GETNEXTCE(tOrder, coll, tColl, status);
*(--tFSBEnd) = UCOL_SECONDARYORDER(tOrder);
if(tFSBEnd == targetFrenchSec) { /* overflowing the buffer, bail out */
return ucol_compareUsingSortKeys(coll, source, sourceLength, target, targetLength);
}
}
gett = TRUE;
// If we've hit the end of one of the strings, jump out of the loop
if ((sOrder == UCOL_NULLORDER)||
(tOrder == UCOL_NULLORDER)) {
break;
}
// If there's no difference at this position, we can skip to the
// next one.
if (sOrder == tOrder)
{
continue;
}
// Compare primary differences first.
pSOrder = UCOL_PRIMARYORDER(sOrder);
pTOrder = UCOL_PRIMARYORDER(tOrder);
if (pSOrder != pTOrder)
{
if (sOrder == UCOL_IGNORABLE)
{
// The entire source element is ignorable.
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
continue;
}
if (tOrder == UCOL_IGNORABLE)
{
gets = FALSE;
continue;
}
// The source and target elements aren't ignorable, but it's still possible
// for the primary component of one of the elements to be ignorable....
if (pSOrder == UCOL_PRIMIGNORABLE) // primary order in source is ignorable
{
// The source's primary is ignorable, but the target's isn't. We treat ignorables
// as a secondary difference, so remember that we found one.
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
// Skip to the next source element, but don't fetch another target element.
gett = FALSE;
}
else if (pTOrder == UCOL_PRIMIGNORABLE)
{
// record differences - see the comment above.
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
// Skip to the next target element, but don't fetch another source element.
gets = FALSE;
}
else
{
// Neither of the orders is ignorable, and we already know that the primary
// orders are different because of the (pSOrder != pTOrder) test above.
// Record the difference and stop the comparison.
if (pSOrder < pTOrder)
{
return UCOL_LESS; // (strength is PRIMARY)
}
return UCOL_GREATER; // (strength is PRIMARY)
}
}
else
{ // else of if ( pSOrder != pTOrder )
// primary order is the same, but complete order is different. So there
// are no base elements at this point, only ignorables (Since the strings are
// normalized)
if (checkSecTer)
{
// a secondary or tertiary difference may still matter
uint32_t secSOrder = UCOL_SECONDARYORDER(sOrder);
uint32_t secTOrder = UCOL_SECONDARYORDER(tOrder);
if (secSOrder != secTOrder)
{
// there is a secondary difference
result = (secSOrder < secTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is SECONDARY)
checkSecTer = isFrenchSec; // We still want to track the French secondaries
//checkSecTer = FALSE;
}
else
{
if (checkTertiary)
{
// a tertiary difference may still matter
uint32_t terSOrder = UCOL_TERTIARYORDER(sOrder);
uint32_t terTOrder = UCOL_TERTIARYORDER(tOrder);
if (terSOrder != terTOrder)
{
// there is a tertiary difference
result = (terSOrder < terTOrder) ? UCOL_LESS : UCOL_GREATER;
// (strength is TERTIARY)
checkTertiary = FALSE;
}
}
}
} // if (checkSecTer)
} // if ( pSOrder != pTOrder )
} // while()
if (sOrder != UCOL_NULLORDER)
{
// (tOrder must be CollationElementIterator::NULLORDER,
// since this point is only reached when sOrder or tOrder is NULLORDER.)
// The source string has more elements, but the target string hasn't.
do
{
if (UCOL_PRIMARYORDER(sOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the source string.
// This is a primary difference, so the source is greater
return UCOL_GREATER; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(sOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements mean the source string is greater
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
}
UCOL_GETNEXTCE(sOrder, coll, sColl, status);
*(--sFSBEnd) = UCOL_SECONDARYORDER(sOrder);
if(sFSBEnd == sourceFrenchSec) { /* overflowing the buffer, bail out */
return ucol_compareUsingSortKeys(coll, source, sourceLength, target, targetLength);
}
}
//while ((sOrder = ucol_getNextCE(coll, &sColl, &status)) != CollationElementIterator::NULLORDER);
while (sOrder != UCOL_NULLORDER);
}
else if (tOrder != UCOL_NULLORDER)
{
// The target string has more elements, but the source string hasn't.
do
{
if (UCOL_PRIMARYORDER(tOrder) != UCOL_PRIMIGNORABLE)
{
// We found an additional non-ignorable base character in the target string.
// This is a primary difference, so the source is less
return UCOL_LESS; // (strength is PRIMARY)
}
if (UCOL_SECONDARYORDER(tOrder) != UCOL_SECIGNORABLE)
{
// Additional secondary elements in the target mean the source string is less
if (checkSecTer)
{
bufferFrenchSec = TRUE;
}
}
UCOL_GETNEXTCE(tOrder, coll, tColl, status);
*(--tFSBEnd) = UCOL_SECONDARYORDER(tOrder);
if(tFSBEnd == targetFrenchSec) { /* overflowing the buffer, bail out */
return ucol_compareUsingSortKeys(coll, source, sourceLength, target, targetLength);
}
}
while ( tOrder != UCOL_NULLORDER);
}
if(bufferFrenchSec) {
while(sFSBEnd < sourceFrenchSec+UCOL_MAX_BUFFER && tFSBEnd < targetFrenchSec+UCOL_MAX_BUFFER) {
if(*sFSBEnd == *tFSBEnd) {
sFSBEnd++;
tFSBEnd++;
} else if(*sFSBEnd < *tFSBEnd) {
result = UCOL_LESS;
break;
} else {
result = UCOL_GREATER;
break;
}
}
}
}
// For IDENTICAL comparisons, we use a bitwise character comparison
// as a tiebreaker if all else is equal
// NOTE: The java code compares result with 0, and
// puts the result of the string comparison directly into result
if (result == UCOL_EQUAL && strength == UCOL_IDENTICAL)
{
UnicodeString sourceDecomp, targetDecomp;
int8_t comparison;
Normalizer::normalize(UnicodeString(source, sourceLength), ((RuleBasedCollator *)coll)->getDecomposition(),
0, sourceDecomp, status);
Normalizer::normalize(UnicodeString(target, targetLength), ((RuleBasedCollator *)coll)->getDecomposition(),
0, targetDecomp, status);
comparison = sourceDecomp.compare(targetDecomp);
if (comparison < 0)
{
result = UCOL_LESS;
}
else if (comparison == 0)
{
result = UCOL_EQUAL;
}
else
{
result = UCOL_GREATER;
}
}
return result;
#endif
return UCOL_EQUAL;
}
void init_incrementalContext(UCharForwardIterator *source, void *sourceContext, incrementalContext *s) {
s->len = s->stringP = s->stackString ;
s->capacity = s->stackString+UCOL_MAX_BUFFER;
s->CEpos = s->toReturn = s->CEs;
s->source = source;
s->sourceContext = sourceContext;
s->currentChar = 0xFFFF;
s->lastChar = 0xFFFF;
s->panic = FALSE;
}
int32_t ucol_getIncrementalCE(const UCollator *coll, incrementalContext *ctx, UErrorCode *status) {
#if 0
uint32_t order;
if (U_FAILURE(*status) /*|| (ctx->CEpos <= ctx->toReturn)*/) {
return UCOL_NULLORDER;
}
if (ctx->CEpos > ctx->toReturn) {
return(*(ctx->toReturn++));
}
ctx->CEpos = ctx->toReturn = ctx->CEs;
if(ctx->lastChar == 0xFFFF) {
ctx->currentChar = ctx->source(ctx->sourceContext);
incctx_appendChar(ctx, ctx->currentChar);
//*(ctx->len++) = ctx->currentChar;
if(ctx->currentChar == 0xFFFF) {
return UCOL_NULLORDER;
}
} else {
ctx->currentChar = ctx->lastChar;
ctx->lastChar = 0xFFFF;
}
order = ucmp32_get(((RuleBasedCollator *)coll)->data->mapping, ctx->currentChar);
// this should benefit from reordering of the clauses, so that the cleanest case is returned the first.
if(order < UCOL_EXPANDCHARINDEX && !(UCOL_ISTHAIPREVOWEL(ctx->currentChar))) {
return (order);
}
if (order == UCOL_UNMAPPED) {
// Returned an "unmapped" flag and save the character so it can be
// returned next time this method is called.
if (ctx->currentChar == 0x0000) return ctx->currentChar; // \u0000 is not valid in C++'s UnicodeString
//*(ctx->CEpos++) = UCOL_UNMAPPEDCHARVALUE;
order = UCOL_UNMAPPEDCHARVALUE;
*(ctx->CEpos++) = ctx->currentChar<<16;
} else {
// Contraction sequence start...
if (order >= UCOL_CONTRACTCHARINDEX) {
UChar key[1024];
uint32_t posKey = 0;
VectorOfPToContractElement* list = ((RuleBasedCollator *)coll)->data->contractTable->at(order-UCOL_CONTRACTCHARINDEX);
// The upper line obtained a list of contracting sequences.
if (list != NULL) {
EntryPair *pair = (EntryPair *)list->at(0); // Taking out the first one.
order = pair->value; // This got us mapping for just the first element - the one that signalled a contraction.
key[posKey++] = ctx->currentChar;
// This tries to find the longes common match for the data in contraction table...
// and needs to be rewritten, especially the test down there!
int32_t i;
int32_t listSize = list->size();
UBool foundSmaller = TRUE;
UBool endOfString = FALSE;
//*(ctx->len++) = ctx->lastChar;
incctx_appendChar(ctx, ctx->lastChar);
while(!endOfString && foundSmaller) {
endOfString = ((ctx->lastChar = ctx->source(ctx->sourceContext)) == 0xFFFF);
key[posKey++] = ctx->lastChar;
foundSmaller = FALSE;
i = 0;
while(i<listSize && !foundSmaller) {
pair = list->at(i);
if ((pair != NULL) && (pair->fwd == TRUE /*fwd*/) && (pair->equalTo(key, posKey))) {
order = pair->value;
//*(ctx->len++) = ctx->lastChar;
incctx_appendChar(ctx, ctx->lastChar);
foundSmaller = TRUE;
}
i++;
}
}
}
}
// Expansion sequence start...
if (order >= UCOL_EXPANDCHARINDEX) {
VectorOfInt *v = ((RuleBasedCollator *)coll)->data->expandTable->at(order-UCOL_EXPANDCHARINDEX);
if(v != NULL) {
int32_t expandindex=0;
int32_t vSize = v->size();
order = v->at(expandindex++); // first character....
while(expandindex < vSize) {
*(ctx->CEpos++) = v->at(expandindex++);
}
}
}
// Thai/Lao reordering
// This is gonna be way too goofy - so we're gonna bail out and let others do the work...
if (UCOL_ISTHAIPREVOWEL(ctx->currentChar)) {
ctx->panic = TRUE;
return UCOL_NULLORDER;
}
}
return order;
#endif
return 0;
}
void incctx_cleanUpContext(incrementalContext *ctx) {
if(ctx->stringP != ctx->stackString) {
uprv_free(ctx->stringP);
}
}
UChar incctx_appendChar(incrementalContext *ctx, UChar c) {
if(ctx->len == ctx->capacity) { /* bother, said Pooh, we need to reallocate */
UChar *newStuff;
if(ctx->stringP == ctx->stackString) { /* we haven't allocated before, need to allocate */
newStuff = (UChar *)uprv_malloc(2*(ctx->capacity - ctx->stringP)*sizeof(UChar));
if(newStuff == NULL) {
/*freak out*/
}
uprv_memcpy(newStuff, ctx->stringP, (ctx->capacity - ctx->stringP)*sizeof(UChar));
} else { /* we have already allocated, need to reallocate */
newStuff = (UChar *)uprv_realloc(ctx->stringP, 2*(ctx->capacity - ctx->stringP)*sizeof(UChar));
if(newStuff == NULL) {
/*freak out*/
}
}
ctx->len=newStuff+(ctx->len - ctx->stringP);
ctx->capacity = newStuff+2*(ctx->capacity - ctx->stringP);
ctx->stringP = newStuff;
}
*(ctx->len++) = c;
return c;
}
UCollationResult alternateIncrementalProcessing(const UCollator *coll, incrementalContext *srcCtx, incrementalContext *trgCtx) {
if(srcCtx->stringP == srcCtx->len || *(srcCtx->len-1) != 0xFFFF) {
while(incctx_appendChar(srcCtx, srcCtx->source(srcCtx->sourceContext)) != 0xFFFF);
}
if(trgCtx->stringP == trgCtx->len || *(trgCtx->len-1) != 0xFFFF) {
while(incctx_appendChar(trgCtx, trgCtx->source(trgCtx->sourceContext)) != 0xFFFF);
}
UCollationResult result = ucol_strcoll(coll, srcCtx->stringP, srcCtx->len-srcCtx->stringP-1, trgCtx->stringP, trgCtx->len-trgCtx->stringP-1);
incctx_cleanUpContext(srcCtx);
incctx_cleanUpContext(trgCtx);
return result;
}
U_CAPI void
ucol_getVersion(const UCollator* coll,
UVersionInfo versionInfo)
{
((Collator*)coll)->getVersion(versionInfo);
}
U_CAPI UCollationStrength
ucol_getStrength(const UCollator *coll)
{
UErrorCode status = U_ZERO_ERROR;
return ucol_getAttribute(coll, UCOL_STRENGTH, &status);
}
U_CAPI void
ucol_setStrength( UCollator *coll,
UCollationStrength strength)
{
UErrorCode status = U_ZERO_ERROR;
ucol_setAttribute(coll, UCOL_STRENGTH, strength, &status);
}
U_CAPI const UChar*
ucol_getRules( const UCollator *coll,
int32_t *length)
{
*length = 0;
return NULL;
/*
const UnicodeString& rules = ((RuleBasedCollator*)coll)->getRules();
*length = rules.length();
return rules.getUChars();
*/
}
U_CAPI int32_t
ucol_getDisplayName( const char *objLoc,
const char *dispLoc,
UChar *result,
int32_t resultLength,
UErrorCode *status)
{
if(U_FAILURE(*status)) return -1;
UnicodeString dst(result, resultLength, resultLength);
Collator::getDisplayName(Locale(objLoc), Locale(dispLoc), dst);
int32_t actLen;
T_fillOutputParams(&dst, result, resultLength, &actLen, status);
return actLen;
}
U_CAPI const char*
ucol_getAvailable(int32_t index)
{
return uloc_getAvailable(index);
}
U_CAPI int32_t
ucol_countAvailable()
{
return uloc_countAvailable();
}
U_CAPI UNormalizationMode
ucol_getNormalization(const UCollator* coll)
{
UErrorCode status = U_ZERO_ERROR;
if(ucol_getAttribute(coll, UCOL_NORMALIZATION_MODE, &status) == UCOL_ON) {
return UCOL_DECOMP_CAN;
} else {
return UCOL_NO_NORMALIZATION;
}
}
#if 0
/* This is the old implementation, which should be removed... */
inline void *ucol_getABuffer(const UCollatorOld *coll, uint32_t size) {
return ((RuleBasedCollator *)coll)->getSomeMemory(size);
}
UCollationResult ucol_compareUsingSortKeys(const UCollatorOld *coll,
const UChar *source,
int32_t sourceLength,
const UChar *target,
int32_t targetLength)
{
uint8_t sourceKey[UCOL_MAX_BUFFER], targetKey[UCOL_MAX_BUFFER];
uint8_t *sourceKeyP = sourceKey;
uint8_t *targetKeyP = targetKey;
int32_t sourceKeyLen = UCOL_MAX_BUFFER, targetKeyLen = UCOL_MAX_BUFFER;
sourceKeyLen = ucol_getSortKey(coll, source, sourceLength, sourceKeyP, sourceKeyLen);
if(sourceKeyLen > UCOL_MAX_BUFFER) {
sourceKeyP = (uint8_t*)uprv_malloc(sourceKeyLen*sizeof(uint8_t));
sourceKeyLen = ucol_getSortKey(coll, source, sourceLength, sourceKeyP, sourceKeyLen);
}
targetKeyLen = ucol_getSortKey(coll, target, targetLength, targetKeyP, targetKeyLen);
if(targetKeyLen > UCOL_MAX_BUFFER) {
targetKeyP = (uint8_t*)uprv_malloc(targetKeyLen*sizeof(uint8_t));
targetKeyLen = ucol_getSortKey(coll, target, targetLength, targetKeyP, targetKeyLen);
}
int32_t result = uprv_strcmp((const char*)sourceKeyP, (const char*)targetKeyP);
if(sourceKeyP != sourceKey) {
uprv_free(sourceKeyP);
}
if(targetKeyP != targetKey) {
uprv_free(targetKeyP);
}
if(result<0) {
return UCOL_LESS;
} else if(result>0) {
return UCOL_GREATER;
} else {
return UCOL_EQUAL;
}
}
int32_t getComplicatedCE(const UCollatorOld *coll, collIterate *source, UErrorCode *status) {
if (*(source->CEpos) == UCOL_UNMAPPED) {
// Returned an "unmapped" flag and save the character so it can be
// returned next time this method is called.
if (*(source->pos) == 0x0000) return *(source->pos++); // \u0000 is not valid in C++'s UnicodeString
*(source->CEpos++) = UCOL_UNMAPPEDCHARVALUE;
*(source->CEpos++) = *(source->pos)<<16;
} else {
// Contraction sequence start...
if (*(source->CEpos) >= UCOL_CONTRACTCHARINDEX) {
UChar key[1024];
uint32_t posKey = 0;
VectorOfPToContractElement* list = ((RuleBasedCollator *)coll)->data->contractTable->at(*(source->CEpos)-UCOL_CONTRACTCHARINDEX);
// The upper line obtained a list of contracting sequences.
if (list != NULL) {
EntryPair *pair = (EntryPair *)list->at(0); // Taking out the first one.
int32_t order = pair->value; // This got us mapping for just the first element - the one that signalled a contraction.
key[posKey++] = *(source->pos++);
// This tries to find the longes common match for the data in contraction table...
// and needs to be rewritten, especially the test down there!
int32_t i;
int32_t listSize = list->size();
UBool foundSmaller = TRUE;
while(source->pos<source->len && foundSmaller) {
key[posKey++] = *source->pos;
foundSmaller = FALSE;
i = 0;
while(i<listSize && !foundSmaller) {
pair = list->at(i);
if ((pair != NULL) && (pair->fwd == TRUE /*fwd*/) && (pair->equalTo(key, posKey))) {
/* Found a matching contraction sequence */
order = pair->value; /* change the CE value */
source->pos++; /* consume another char from the source */
foundSmaller = TRUE;
}
i++;
}
}
source->pos--; /* spit back the last char - it wasn't part of the sequence */
*(source->CEpos) = order;
}
}
// Expansion sequence start...
if (*(source->CEpos) >= UCOL_EXPANDCHARINDEX) {
VectorOfInt *v = ((RuleBasedCollator *)coll)->data->expandTable->at(*(source->CEpos)-UCOL_EXPANDCHARINDEX);
if(v != NULL) {
int32_t expandindex=0;
int32_t vSize = v->size();
while(expandindex < vSize) {
*(source->CEpos++) = v->at(expandindex++);
}
}
}
// Thai/Lao reordering
if (UCOL_ISTHAIPREVOWEL(*(source->pos)) &&
UCOL_ISTHAIBASECONSONANT(*(source->pos+1))) {
if(source->isThai == TRUE) {
source->isThai = FALSE;
if((source->len - source->pos) > UCOL_WRITABLE_BUFFER_SIZE) {
// allocate a new buffer
source->writableBuffer = (UChar *)ucol_getABuffer(coll, (source->len - source->pos)*sizeof(UChar));
}
UChar *sourceCopy = source->pos;
UChar *targetCopy = source->writableBuffer;
while(sourceCopy < source->len) {
if(UCOL_ISTHAIPREVOWEL(*(sourceCopy)) &&
UCOL_ISTHAIBASECONSONANT(*(sourceCopy+1))) {
*(targetCopy) = *(sourceCopy+1);
*(targetCopy+1) = *(sourceCopy);
targetCopy+=2;
sourceCopy+=2;
} else {
*(targetCopy++) = *(sourceCopy++);
}
}
source->pos = source->writableBuffer;
source->len = targetCopy;
source->CEpos = source->toReturn = source->CEs;
return UCOL_IGNORABLE;
}
}
}
source->pos++;
return (*(source->toReturn++));
}
U_CAPI UCollatorOld *ucol_safeClone(const UCollatorOld *coll, void *stackBuffer, uint32_t bufferSize, UErrorCode *status) {
return (UCollatorOld *)(((RuleBasedCollator *)coll)->safeClone());
}
U_CAPI int32_t ucol_getRulesEx(const UCollatorOld *coll, UColRuleOption delta, UChar *buffer, int32_t bufferLen) {
return 0;
}
#endif