| /* |
| ******************************************************************************* |
| * Copyright (C) 1996-2001, International Business Machines |
| * Corporation and others. All Rights Reserved. |
| ******************************************************************************* |
| * file name: ucol.cpp |
| * encoding: US-ASCII |
| * tab size: 8 (not used) |
| * indentation:4 |
| * |
| * Modification history |
| * Date Name Comments |
| * 1996-1999 various members of ICU team maintained C API for collation framework |
| * 02/16/2001 synwee Added internal method getPrevSpecialCE |
| * 03/01/2001 synwee Added maxexpansion functionality. |
| * 03/16/2001 weiv Collation framework is rewritten in C and made UCA compliant |
| */ |
| |
| #include "ucol_bld.h" |
| #include "ucol_imp.h" |
| #include "ucol_tok.h" |
| #include "ucol_elm.h" |
| |
| #include "unicode/uloc.h" |
| #include "unicode/coll.h" |
| #include "unicode/tblcoll.h" |
| #include "unicode/coleitr.h" |
| #include "unicode/unorm.h" |
| #include "unicode/udata.h" |
| |
| #include "cpputils.h" |
| #include "cstring.h" |
| #include "ucmp32.h" |
| #include "umutex.h" |
| #include "uhash.h" |
| |
| #include <stdio.h> |
| |
| |
| |
| static UCollator* UCA = NULL; |
| |
| extern "C" UBool checkFCD(const UChar*, int32_t, UErrorCode*); |
| |
| /* Fixup table a la Markus */ |
| /* see http://www.ibm.com/software/developer/library/utf16.html for further explanation */ |
| static uint8_t utf16fixup[32] = { |
| 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0x20, 0xf8, 0xf8, 0xf8, 0xf8 |
| }; |
| |
| static UBool U_CALLCONV |
| isAcceptableUCA(void *context, |
| const char *type, const char *name, |
| const UDataInfo *pInfo){ |
| /* context, type & name are intentionally not used */ |
| 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; |
| } |
| } |
| /****************************************************************************/ |
| /* Following are the open/close functions */ |
| /* */ |
| /****************************************************************************/ |
| 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); |
| /* first take on tailoring version: */ |
| /* get CollationElements -> Version */ |
| 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 */ |
| *status = U_USING_DEFAULT_ERROR; |
| result = ucol_initCollator(UCA->image, result, status); |
| /*result = UCA;*/ |
| result->hasRealData = FALSE; |
| } 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); |
| if((uint32_t)len > sizeof(UCATableHeader)) { |
| result = ucol_initCollator((const UCATableHeader *)inData, result, status); |
| result->hasRealData = TRUE; |
| } else { |
| result = ucol_initCollator(UCA->image, result, status); |
| ucol_setOptionsFromHeader(result, (const UCATableHeader *)inData, status); |
| result->hasRealData = FALSE; |
| } |
| } else { /* There is another error, and we're just gonna clean up */ |
| ures_close(b); |
| return NULL; |
| } |
| |
| result->rb = b; |
| ures_close(binary); |
| |
| return result; |
| } |
| |
| U_CAPI UCollator * U_EXPORT2 |
| ucol_openVersion(const char *loc, |
| UVersionInfo version, |
| UErrorCode *status) { |
| UCollator *collator; |
| UVersionInfo info; |
| |
| collator=ucol_open(loc, status); |
| if(U_SUCCESS(*status)) { |
| ucol_getVersion(collator, info); |
| if(0!=uprv_memcmp(version, info, sizeof(UVersionInfo))) { |
| ucol_close(collator); |
| *status=U_MISSING_RESOURCE_ERROR; |
| return NULL; |
| } |
| } |
| return collator; |
| } |
| |
| 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->freeOnClose == FALSE){ |
| return; /* for safeClone, if freeOnClose is FALSE, |
| don't free the other instance data */ |
| } |
| if(coll->mapping != NULL) { |
| ucmp32_close(coll->mapping); |
| } |
| if(coll->rules != NULL && coll->freeRulesOnClose) { |
| uprv_free((UChar *)coll->rules); |
| } |
| if(coll->rb != NULL) { /* pointing to read-only memory */ |
| ures_close(coll->rb); |
| } else if(coll->hasRealData == TRUE) { |
| uprv_free((UCATableHeader *)coll->image); |
| } |
| uprv_free(coll); |
| } |
| |
| U_CAPI UCollator* |
| ucol_openRules( const UChar *rules, |
| int32_t rulesLength, |
| UNormalizationMode mode, |
| UCollationStrength strength, |
| UErrorCode *status) |
| { |
| uint32_t listLen = 0; |
| UColTokenParser src; |
| |
| ucol_initUCA(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; |
| } |
| |
| /*src.source = rules;*/ |
| src.source = (UChar *)uprv_malloc((rulesLength+UCOL_TOK_EXTRA_RULE_SPACE_SIZE)*sizeof(UChar)); |
| uprv_memcpy(src.source, rules, rulesLength*sizeof(UChar)); |
| src.current = src.source; |
| src.end = src.source+rulesLength; |
| src.sourceCurrent = src.source; |
| src.extraCurrent = src.end; |
| src.extraEnd = src.end+UCOL_TOK_EXTRA_RULE_SPACE_SIZE; |
| src.UCA = UCA; |
| src.invUCA = ucol_initInverseUCA(status); |
| src.resultLen = 0; |
| src.lh = 0; |
| |
| src.image = (UCATableHeader *)uprv_malloc(sizeof(UCATableHeader)); |
| |
| |
| uprv_memcpy(src.image, UCA->image, sizeof(UCATableHeader)); |
| |
| listLen = ucol_tok_assembleTokenList(&src, status); |
| if(U_FAILURE(*status)) { |
| /* if status is U_ILLEGAL_ARGUMENT_ERROR, src->current points at the offending option */ |
| /* if status is U_INVALID_FORMAT_ERROR, src->current points after the problematic part of the rules */ |
| /* so something might be done here... or on lower level */ |
| #ifdef UCOL_DEBUG |
| if(*status == U_ILLEGAL_ARGUMENT_ERROR) { |
| fprintf(stderr, "bad option starting at offset %i\n", src.current-src.source); |
| } else { |
| fprintf(stderr, "invalid rule just before offset %i\n", src.current-src.source); |
| } |
| #endif |
| uprv_free(src.image); |
| ucol_tok_closeTokenList(&src); |
| return NULL; |
| } |
| UCollator *result = NULL; |
| UCATableHeader *table = NULL; |
| |
| if(src.resultLen > 0) { /* we have a set of rules, let's make something of it */ |
| table = ucol_assembleTailoringTable(&src, status); |
| result = ucol_initCollator(table,0,status); |
| result->hasRealData = TRUE; |
| } else { /* no rules, but no error either */ |
| /* must be only options */ |
| result = ucol_initCollator(UCA->image,0,status); |
| ucol_setOptionsFromHeader(result, src.image, status); |
| result->hasRealData = FALSE; |
| } |
| if(U_SUCCESS(*status)) { |
| result->dataInfo.dataVersion[0] = UCOL_BUILDER_VERSION; |
| result->rules = (UChar *)uprv_malloc((u_strlen(rules)+1)*sizeof(UChar)); |
| u_strcpy((UChar *)result->rules, rules); |
| result->freeRulesOnClose = TRUE; |
| result->rb = 0; |
| } else { |
| if(table != NULL) { |
| uprv_free(table); |
| ucol_close(result); |
| } |
| uprv_free(src.image); |
| ucol_tok_closeTokenList(&src); |
| return NULL; |
| } |
| |
| uprv_free(src.image); |
| ucol_tok_closeTokenList(&src); |
| |
| ucol_setAttribute(result, UCOL_STRENGTH, strength, status); |
| |
| return result; |
| } |
| |
| /* 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) |
| { |
| uint8_t *result = NULL; |
| if(coll->hasRealData == TRUE) { |
| *length = coll->image->size; |
| result = (uint8_t *)uprv_malloc(*length); |
| uprv_memcpy(result, coll->image, *length); |
| } else { |
| *length = sizeof(UCATableHeader); |
| result = (uint8_t *)uprv_malloc(sizeof(UCATableHeader)); |
| UCATableHeader *head = (UCATableHeader *)result; |
| ucol_putOptionsToHeader(coll, head, status); |
| } |
| return result; |
| } |
| |
| void ucol_setOptionsFromHeader(UCollator* result, const UCATableHeader * image, UErrorCode *status) { |
| if(U_FAILURE(*status)) { |
| return; |
| } |
| result->caseFirst = image->caseFirst; |
| result->caseLevel = image->caseLevel; |
| result->frenchCollation = image->frenchCollation; |
| result->normalizationMode = image->normalizationMode; |
| result->strength = image->strength; |
| result->variableTopValue = image->variableTopValue; |
| result->alternateHandling = image->alternateHandling; |
| |
| result->caseFirstisDefault = TRUE; |
| result->caseLevelisDefault = TRUE; |
| result->frenchCollationisDefault = TRUE; |
| result->normalizationModeisDefault = TRUE; |
| result->strengthisDefault = TRUE; |
| result->variableTopValueisDefault = TRUE; |
| |
| ucol_updateInternalState(result); |
| } |
| |
| void ucol_putOptionsToHeader(UCollator* result, UCATableHeader * image, UErrorCode *status) { |
| if(U_FAILURE(*status)) { |
| return; |
| } |
| image->caseFirst = result->caseFirst; |
| image->caseLevel = result->caseLevel; |
| image->frenchCollation = result->frenchCollation; |
| image->normalizationMode = result->normalizationMode; |
| image->strength = result->strength; |
| image->variableTopValue = result->variableTopValue; |
| image->alternateHandling = result->alternateHandling; |
| } |
| |
| |
| |
| UCollator* ucol_initCollator(const UCATableHeader *image, UCollator *fillIn, UErrorCode *status) { |
| UCollator *result = fillIn; |
| if(U_FAILURE(*status) || image == NULL) { |
| 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->alternateHandling = result->image->alternateHandling; |
| |
| result->caseFirstisDefault = TRUE; |
| result->caseLevelisDefault = TRUE; |
| result->frenchCollationisDefault = TRUE; |
| result->normalizationModeisDefault = TRUE; |
| result->strengthisDefault = TRUE; |
| result->variableTopValueisDefault = TRUE; |
| result->alternateHandlingisDefault = TRUE; |
| |
| uint32_t variableMaxCE = ucmp32_get(result->mapping, result->variableTopValue); |
| result->variableMax1 = (uint8_t)((variableMaxCE & 0xFF000000) >> 24); |
| result->variableMax2 = (uint8_t)((variableMaxCE & 0x00FF0000) >> 16); |
| |
| result->scriptOrder = NULL; |
| |
| result->zero = 0; |
| result->rules = NULL; |
| /* get the version info form UCATableHeader and populate the Collator struct*/ |
| result->dataInfo.dataVersion[0] = result->image->version[0]; /* UCA Builder version*/ |
| result->dataInfo.dataVersion[1] = result->image->version[1]; /* UCA Tailoring rules version*/ |
| |
| #if 0 |
| /* Build the unsafe chars hash table */ |
| uint8_t *t; /*non-const, unlike result->unsafeCP */ |
| /* result->unsafeCP = */ t = (uint8_t *)uprv_malloc(UCOL_UNSAFECP_TABLE_SIZE); |
| buildUnsafeCPTable(t, result); |
| #endif |
| result->unsafeCP = (uint8_t *)result->image + result->image->unsafeCP; |
| |
| /* max expansion tables */ |
| result->endExpansionCE = (uint32_t*)((uint8_t*)result->image + |
| result->image->endExpansionCE); |
| result->lastEndExpansionCE = result->endExpansionCE + |
| result->image->endExpansionCECount - 1; |
| result->expansionCESize = (uint8_t*)result->image + |
| result->image->expansionCESize; |
| result->errorCode = *status; |
| |
| ucol_updateInternalState(result); |
| 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, isAcceptableUCA, NULL, status); |
| |
| if(U_FAILURE(*status)) { |
| udata_close(result); |
| uprv_free(newUCA); |
| } |
| |
| if(result != NULL) { /* It looks like sometimes we can fail to find the data file */ |
| newUCA = ucol_initCollator((const UCATableHeader *)udata_getMemory(result), newUCA, status); |
| newUCA->rb = NULL; |
| umtx_lock(NULL); |
| if(UCA == NULL) { |
| UCA = newUCA; |
| newUCA = NULL; |
| } |
| umtx_unlock(NULL); |
| |
| if(newUCA != NULL) { |
| udata_close(result); |
| uprv_free(newUCA); |
| } |
| } |
| |
| } |
| } |
| |
| |
| /****************************************************************************/ |
| /* Following are the CE retrieval functions */ |
| /* */ |
| /****************************************************************************/ |
| |
| /* there should be a macro version of this function in the header file */ |
| /* This is the first function that tries to fetch a collation element */ |
| /* If it's not succesfull or it encounters a more difficult situation */ |
| /* some more sofisticated and slower functions are invoked */ |
| uint32_t ucol_getNextCE(const UCollator *coll, collIterate *collationSource, UErrorCode *status) { |
| uint32_t order; |
| if (collationSource->CEpos > collationSource->toReturn) { /* Are there any CEs from previous expansions? */ |
| order = *(collationSource->toReturn++); /* if so, return them */ |
| if(collationSource->CEpos == collationSource->toReturn) { |
| collationSource->CEpos = collationSource->toReturn = collationSource->CEs; |
| } |
| } else if(collationSource->pos < collationSource->len) { /* This is the real business now */ |
| UChar ch = *collationSource->pos++; |
| 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, order, 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 */ |
| } else { |
| order = UCOL_NO_MORE_CES; /* if so, we won't play any more */ |
| } |
| /* This means that contraction should spit back the last codepoint eaten! */ |
| return order; /* return the CE */ |
| } |
| |
| /* this should be connected to special Jamo handling */ |
| uint32_t ucol_getFirstCE(const UCollator *coll, UChar u, UErrorCode *status) { |
| collIterate colIt; |
| uint32_t order; |
| init_collIterate(coll, &u, 1, &colIt, FALSE); |
| order = ucol_getNextCE(coll, &colIt, status); |
| /*UCOL_GETNEXTCE(order, coll, colIt, status);*/ |
| return order; |
| } |
| |
| #if 0 |
| /* bogus code, based on the wrong assumption */ |
| void getSpecialJamo(const UCollator *coll, uint32_t CE, uint32_t **buffer) { |
| for(;;) { |
| uint32_t tag = getCETag(CE); |
| if(tag == THAI_TAG || tag == EXPANSION_TAG) { |
| uint32_t i = 0; |
| uint32_t *CEOffset = (uint32_t *)coll->image+getExpansionOffset(CE); /* find the offset to expansion table */ |
| uint32_t size = getExpansionCount(CE); |
| if(size != 0) { /* if there are less than 16 elements in expansion, we don't terminate */ |
| for(i = 1; i<size; i++) { |
| *(*buffer++) = *CEOffset++; |
| } |
| } else { /* else, we do */ |
| while(*CEOffset != 0) { |
| *(*buffer++) = *CEOffset++; |
| } |
| } |
| break; |
| } else if(tag == CONTRACTION_TAG) { |
| const UChar *ContractionStart = (UChar *)coll->image+getContractOffset(CE); |
| *(*buffer++) = *(coll->contractionCEs + (ContractionStart- coll->contractionIndex)); |
| } |
| } |
| } |
| |
| void ucol_getJamoCEs(const UCollator *coll, UChar ch, uint32_t **buffer) { |
| uint32_t order; |
| 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 */ |
| getSpecialJamo(coll, order, buffer); /* and try to get the special CE */ |
| } else if(order == UCOL_NOT_FOUND) { /* consult the UCA */ |
| if(ch <= 0xFF) { /* if it's Latin One, we'll try to fast track it */ |
| order = UCA->latinOneMapping[ch]; /* by looking in up in an array */ |
| } else { /* otherwise, */ |
| order = ucmp32_get(UCA->mapping, ch); /* we'll go for slightly slower trie */ |
| } |
| if(order > UCOL_NOT_FOUND) { |
| getSpecialJamo(UCA, order, buffer); /* and try to get the special CE */ |
| } |
| } |
| *(*buffer++) = order; |
| } |
| |
| #endif |
| |
| /* This function tries to get a CE from UCA, which should be always around */ |
| /* UChar is passed in in order to speed things up */ |
| /* here is also the generation of implicit CEs */ |
| 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, order, 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; |
| const uint32_t |
| SBase = 0xAC00, LBase = 0x1100, VBase = 0x1161, TBase = 0x11A7, |
| LCount = 19, VCount = 21, TCount = 28, |
| NCount = VCount * TCount, // 588 |
| SCount = LCount * NCount; // 11172 |
| //LLimit = LBase + LCount, // 1113 |
| //VLimit = VBase + VCount, // 1176 |
| //TLimit = TBase + TCount, // 11C3 |
| //SLimit = SBase + SCount; // D7A4 |
| |
| // once we have failed to find a match for codepoint cp, and are in the implicit code. |
| |
| uint32_t L = ch - SBase; |
| //if (ch < SLimit) { // since it is unsigned, catchs zero case too |
| if (L < SCount) { // since it is unsigned, catchs zero case too |
| |
| // divide into pieces |
| |
| uint32_t T = L % TCount; // we do it in this order since some compilers can do % and / in one operation |
| L /= TCount; |
| uint32_t V = L % VCount; |
| L /= VCount; |
| |
| // offset them |
| |
| L += LBase; |
| V += VBase; |
| T += TBase; |
| |
| // return the first CE, but first put the rest into the expansion buffer |
| if (!collationSource->coll->image->jamoSpecial) { // FAST PATH |
| |
| *(collationSource->CEpos++) = ucmp32_get(UCA->mapping, V); |
| if (T != TBase) { |
| *(collationSource->CEpos++) = ucmp32_get(UCA->mapping, T); |
| } |
| |
| return ucmp32_get(UCA->mapping, L); // return first one |
| |
| } else { // Jamo is Special |
| collIterate jamos; |
| UChar jamoString[3]; |
| uint32_t CE = UCOL_NOT_FOUND; |
| const UCollator *collator = collationSource->coll; |
| jamoString[0] = L; |
| jamoString[1] = V; |
| if (T != TBase) { |
| jamoString[2] = T; |
| init_collIterate(collator, jamoString, 3, &jamos, TRUE); |
| } else { |
| init_collIterate(collator, jamoString, 2, &jamos, TRUE); |
| } |
| |
| CE = ucol_getNextCE(collator, &jamos, status); |
| |
| while(CE != UCOL_NO_MORE_CES) { |
| *(collationSource->CEpos++) = CE; |
| CE = ucol_getNextCE(collator, &jamos, status); |
| } |
| return *(collationSource->toReturn++); |
| |
| /* Code and pseudocode below is bogus - we didn't take into */ |
| /* account that any combo of L,V,T could be */ |
| /* in fact a contraction - we cannot look at them separately */ |
| |
| /* |
| ucol_getJamoCEs(collationSource->coll, L, &collationSource->CEpos); |
| ucol_getJamoCEs(collationSource->coll, V, &collationSource->CEpos); |
| if (T != TBase) { |
| ucol_getJamoCEs(collationSource->coll, T, &collationSource->CEpos); |
| } |
| return *(collationSource->toReturn++); |
| */ |
| /* |
| // do recursive processing of L, V, and T with fetchCE (but T only if not equal to TBase!!) |
| // Since fetchCE returns a CE, and (potentially) stuffs items into the ce buffer, |
| // this is how it is done. |
| int firstCE = fetchCE(L, ...); |
| int* lastExpansion = expansionBufferEnd++; // set pointer, leave gap! |
| *lastExpansion = fetchCE(V,...); |
| if (T != TBase) { |
| lastExpansion = expansionBufferEnd++; // set pointer, leave gap! |
| *lastExpansion = fetchCE(T,...); |
| } |
| */ |
| } |
| } |
| |
| if(UTF_IS_FIRST_SURROGATE(ch)) { |
| if( (collationSource->pos<collationSource->len) && |
| UTF_IS_SECOND_SURROGATE((nextChar=*collationSource->pos))) { |
| uint32_t cp = (((ch)<<10UL)+(nextChar)-((0xd800<<10UL)+0xdc00)); |
| collationSource->pos++; |
| if ((cp & 0xFFFE) == 0xFFFE || (0xD800 <= cp && cp <= 0xDC00)) { |
| return 0; /* illegal code value, use completely ignoreable! */ |
| } |
| /* This is a code point minus 0x10000, that's what algorithm requires */ |
| order = 0xE0010303 | (cp & 0xFFE00) << 8; |
| |
| *(collationSource->CEpos++) = 0x80200080 | (cp & 0x001FF) << 22; |
| } else { |
| return 0; /* completely ignorable */ |
| } |
| } else { |
| /* otherwise */ |
| if(UTF_IS_SECOND_SURROGATE((ch)) || (ch & 0xFFFE) == 0xFFFE) { |
| return 0; /* completely ignorable */ |
| } |
| /* Make up an artifical CE from code point as per UCA */ |
| order = 0xD0800303 | (ch & 0xF000) << 12 | (ch & 0x0FE0) << 11; |
| *(collationSource->CEpos++) = 0x04000080 | (ch & 0x001F) << 27; |
| } |
| } |
| return order; /* return the CE */ |
| } |
| |
| /* |
| * This function tries to get a CE from UCA, which should be always around |
| * UChar is passed in in order to speed things up here is also the generation |
| * of implicit CEs |
| */ |
| uint32_t ucol_getPrevUCA(UChar ch, collIterate *collationSource, |
| UErrorCode *status) |
| { |
| uint32_t order; |
| if (ch < 0xFF) { |
| order = UCA->latinOneMapping[ch]; |
| } |
| else { |
| order = ucmp32_get(UCA->mapping, ch); |
| } |
| |
| if (order >= UCOL_NOT_FOUND) { |
| order = getSpecialPrevCE(UCA, order, 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 prevChar; |
| uint32_t |
| SBase = 0xAC00, LBase = 0x1100, VBase = 0x1161, TBase = 0x11A7, |
| LCount = 19, VCount = 21, TCount = 28, |
| NCount = VCount * TCount, /* 588 */ |
| SCount = LCount * NCount; /* 11172 */ |
| /* |
| LLimit = LBase + LCount, // 1113 |
| VLimit = VBase + VCount, // 1176 |
| TLimit = TBase + TCount, // 11C3 |
| SLimit = SBase + SCount; // D7A4 |
| */ |
| |
| /* |
| once we have failed to find a match for codepoint cp, and are in the |
| implicit code. |
| */ |
| |
| uint32_t L = ch - SBase; |
| if (L < SCount) |
| { /* since it is unsigned, catchs zero case too */ |
| |
| /* |
| divide into pieces. |
| we do it in this order since some compilers can do % and / in one |
| operation |
| */ |
| uint32_t T = L % TCount; |
| L /= TCount; |
| uint32_t V = L % VCount; |
| L /= VCount; |
| |
| /* offset them */ |
| L += LBase; |
| V += VBase; |
| T += TBase; |
| |
| /* |
| return the first CE, but first put the rest into the expansion buffer |
| */ |
| if (!collationSource->coll->image->jamoSpecial) |
| { |
| *(collationSource->CEpos ++) = ucmp32_get(UCA->mapping, L); |
| *(collationSource->CEpos ++) = ucmp32_get(UCA->mapping, V); |
| if (T != TBase) |
| *(collationSource->CEpos ++) = ucmp32_get(UCA->mapping, T); |
| |
| collationSource->toReturn = collationSource->CEpos - 1; |
| return *(collationSource->toReturn); |
| } else { |
| collIterate jamos; |
| UChar jamoString[3]; |
| uint32_t CE = UCOL_NOT_FOUND; |
| const UCollator *collator = collationSource->coll; |
| jamoString[0] = L; |
| jamoString[1] = V; |
| if (T != TBase) { |
| jamoString[2] = T; |
| init_collIterate(collator, jamoString, 3, &jamos, TRUE); |
| } else { |
| init_collIterate(collator, jamoString, 2, &jamos, TRUE); |
| } |
| |
| CE = ucol_getNextCE(collator, &jamos, status); |
| |
| while(CE != UCOL_NO_MORE_CES) { |
| *(collationSource->CEpos++) = CE; |
| CE = ucol_getNextCE(collator, &jamos, status); |
| } |
| collationSource->toReturn = collationSource->CEpos - 1; |
| return *(collationSource->toReturn); |
| |
| /*return *(collationSource->toReturn++);*/ |
| /* |
| ucol_getJamoCEs(collationSource->coll, L, &collationSource->CEpos); |
| ucol_getJamoCEs(collationSource->coll, V, &collationSource->CEpos); |
| if (T != TBase) { |
| ucol_getJamoCEs(collationSource->coll, T, &collationSource->CEpos); |
| } |
| collationSource->toReturn = collationSource->CEpos - 1; |
| return *(collationSource->toReturn); |
| */ |
| /* |
| Jamo is Special |
| do recursive processing of L, V, and T with fetchCE (but T only if not |
| equal to TBase!!) |
| Since fetchCE returns a CE, and (potentially) stuffs items into the ce |
| buffer, |
| this is how it is done. |
| */ |
| /* |
| int firstCE = fetchCE(L, ...); |
| // set pointer, leave gap! |
| int* lastExpansion = expansionBufferEnd++; |
| *lastExpansion = fetchCE(V,...); |
| if (T != TBase) { |
| lastExpansion = expansionBufferEnd++; // set pointer, leave gap! |
| *lastExpansion = fetchCE(T,...); |
| } |
| */ |
| } |
| } |
| |
| if (UTF_IS_SECOND_SURROGATE(ch)) |
| { |
| /* This is where the s***t hits the fan */ |
| /* it turns out, the first part of the if can be satisfied even if we're */ |
| /* at the beggining of the string */ |
| /* we have to make sure we know what is the situation we're in */ |
| /* quick fix is by using isUsingWritable, as shown below */ |
| if ((collationSource->start < collationSource->pos) && |
| (UTF_IS_FIRST_SURROGATE(prevChar = *(collationSource->pos - 1)))) |
| { |
| uint32_t cp = ((prevChar << 10UL) + ch - ((0xd800 << 10UL) + 0xdc00)); |
| collationSource->pos --; |
| if ((cp & 0xFFFE) == 0xFFFE || (0xD800 <= cp && cp <= 0xDC00)) { |
| return 0; /* illegal code value, use completely ignoreable! */ |
| } |
| |
| /* |
| This is a code point minus 0x10000, that's what algorithm requires |
| */ |
| *(collationSource->CEpos ++) = 0xE0010303 | (cp & 0xFFE00) << 8; |
| order = 0x80200080 | (cp & 0x001FF) << 22; |
| collationSource->toReturn = collationSource->CEpos; |
| *(collationSource->CEpos ++) = order; |
| } |
| else { |
| return 0; /* completely ignorable */ |
| } |
| } |
| else |
| { |
| /* otherwise */ |
| if (UTF_IS_FIRST_SURROGATE(ch) || (ch & 0xFFFE) == 0xFFFE) { |
| return 0; /* completely ignorable */ |
| } |
| |
| /* Make up an artifical CE from code point as per UCA */ |
| *(collationSource->CEpos ++) = 0xD0800303 | (ch & 0xF000) << 12 | |
| (ch & 0x0FE0) << 11; |
| collationSource->toReturn = collationSource->CEpos; |
| order = 0x04000080 | (ch & 0x001F) << 27; |
| *(collationSource->CEpos ++) = order; |
| } |
| } |
| return order; /* return the CE */ |
| } |
| |
| /* This function handles the special CEs like contractions, expansions, surrogates, Thai */ |
| /* It is called by both getNextCE and getNextUCA */ |
| uint32_t getSpecialCE(const UCollator *coll, uint32_t CE, collIterate *source, UErrorCode *status) { |
| uint32_t i = 0; /* general counter */ |
| uint32_t firstCE = UCOL_NOT_FOUND; |
| UChar *firstUChar = source->pos; |
| //uint32_t CE = *source->CEpos; |
| for (;;) { |
| 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; |
| case SURROGATE_TAG: |
| /* pending surrogate discussion with Markus and Mark */ |
| return UCOL_NOT_FOUND; |
| 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 */ |
| --source->pos; |
| 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->start = 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 = (uint32_t *)coll->image+getExpansionOffset(CE); /* find the offset to expansion table */ |
| CE = *CEOffset++; |
| } |
| break; |
| case CONTRACTION_TAG: |
| /* This should handle contractions */ |
| for (;;) { |
| /* First we position ourselves at the begining of contraction sequence */ |
| const UChar *ContractionStart = UCharOffset = (UChar *)coll->image+getContractOffset(CE); |
| |
| 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... */ |
| if (CE == UCOL_NOT_FOUND) { |
| source->pos = firstUChar; /* spit all the not found chars, which led us in this contraction */ |
| if(firstCE != UCOL_NOT_FOUND) { |
| CE = firstCE; |
| } |
| } |
| } |
| break; |
| } |
| |
| /* 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 */ |
| |
| |
| 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 */ |
| UCharOffset = ContractionStart; /* We're not at the end, bailed out in the middle. Better use starting CE */ |
| /*source->pos = firstUChar; *//* spit all the not found chars, which led us in this contraction */ |
| source->pos--; /* Spit out the last char of the string, wasn't tasty enough */ |
| } |
| CE = *(coll->contractionCEs + (UCharOffset - coll->contractionIndex)); |
| |
| if(CE == UCOL_NOT_FOUND) { |
| source->pos = firstUChar; /* spit all the not found chars, which led us in this contraction */ |
| if(firstCE != UCOL_NOT_FOUND) { |
| CE = firstCE; |
| } |
| break; |
| } else if(isContraction(CE)) { /* fix for the bug. Other places need to be checked */ |
| /* this is contraction, and we will continue. However, we can fail along the */ |
| /* th road, which means that we have part of contraction correct */ |
| uint32_t tempCE = *(coll->contractionCEs + (ContractionStart - coll->contractionIndex)); |
| if(tempCE != UCOL_NOT_FOUND) { |
| firstCE = *(coll->contractionCEs + (ContractionStart - coll->contractionIndex)); |
| firstUChar = source->pos-1; |
| } |
| } else { |
| 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++; |
| } |
| } |
| return CE; |
| case CHARSET_TAG: |
| /* probably after 1.8 */ |
| return UCOL_NOT_FOUND; |
| default: |
| *status = U_INTERNAL_PROGRAM_ERROR; |
| CE=0; |
| break; |
| } |
| if (CE <= UCOL_NOT_FOUND) break; |
| } |
| return CE; |
| } |
| |
| /** |
| * This function handles the special CEs like contractions, expansions, |
| * surrogates, Thai. |
| * It is called by both getPrevCE and getPrevUCA |
| */ |
| uint32_t getSpecialPrevCE(const UCollator *coll, uint32_t CE, |
| collIterate *source, |
| UErrorCode *status) |
| { |
| uint32_t count = 0; |
| const uint32_t *CEOffset = NULL; |
| const UChar *UCharOffset = NULL; |
| UChar schar, |
| tchar; |
| const UChar *strend = NULL; |
| const UChar *constart = NULL; |
| uint32_t size; |
| uint32_t firstCE = UCOL_NOT_FOUND; |
| UChar *firstUChar = source->pos; |
| for(;;) |
| { |
| /* the only ces that loops are thai and contractions */ |
| switch (getCETag(CE)) |
| { |
| case NOT_FOUND_TAG: /* this tag always returns */ |
| return CE; |
| case SURROGATE_TAG: /* this tag always returns */ |
| /* pending surrogate discussion with Markus and Mark */ |
| return UCOL_NOT_FOUND; |
| case THAI_TAG: |
| if (source->isThai == TRUE) |
| { /* if we encountered Thai prevowel & the string is not yet touched */ |
| source->isThai = FALSE; |
| strend = source->pos; |
| size = strend - source->string; |
| if (size > UCOL_WRITABLE_BUFFER_SIZE) |
| { |
| /* |
| someone else has already allocated something |
| */ |
| if (source->writableBuffer != source->stackWritableBuffer) { |
| uprv_free(source->writableBuffer); |
| } |
| source->writableBuffer = |
| (UChar *)uprv_malloc(size * sizeof(UChar)); |
| source->isThai = FALSE; |
| } |
| UChar *sourceCopy = source->string; |
| UChar *targetCopy = source->writableBuffer; |
| while (sourceCopy < strend) { |
| 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 = targetCopy; |
| source->start = 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 |
| */ |
| /* find the offset to expansion table */ |
| CEOffset = (uint32_t *)coll->image + getExpansionOffset(CE); |
| CE = *CEOffset ++; |
| } |
| break; |
| case CONTRACTION_TAG: |
| /* This should handle contractions */ |
| for(;;) |
| { |
| /* |
| First we position ourselves at the begining of contraction sequence |
| */ |
| constart = UCharOffset = (UChar *)coll->image + getContractOffset(CE); |
| strend = source->len; |
| |
| if (firstCE == UCOL_NOT_FOUND) { |
| firstCE = *(coll->contractionCEs + |
| (UCharOffset - coll->contractionIndex)); |
| } |
| |
| if (source->pos <= source->start) { |
| /* this is the start of string */ |
| CE = *(coll->contractionCEs + |
| (UCharOffset - coll->contractionIndex)); |
| if (CE == UCOL_NOT_FOUND && firstCE != UCOL_NOT_FOUND) { |
| CE = firstCE; |
| /* firstCE = UCOL_NOT_FOUND; */ |
| source->pos = firstUChar; |
| } |
| |
| break; |
| } |
| |
| /* |
| Progressing to backwards block |
| */ |
| UCharOffset += *UCharOffset; |
| |
| schar = *(source->pos - 1); |
| while (schar > (tchar = *UCharOffset)) { |
| UCharOffset ++; |
| } |
| |
| if (schar != tchar) { |
| UCharOffset = constart; |
| } |
| else { |
| source->pos --; |
| } |
| |
| CE = *(coll->contractionCEs + (UCharOffset - coll->contractionIndex)); |
| if (!isContraction(CE)) { |
| if (CE == UCOL_NOT_FOUND) { |
| CE = firstCE; |
| source->pos = firstUChar; |
| } |
| firstCE = UCOL_NOT_FOUND; |
| |
| break; |
| } |
| } |
| break; |
| case EXPANSION_TAG: /* this tag always returns */ |
| /* |
| 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 |
| */ |
| /* find the offset to expansion table */ |
| CEOffset = (uint32_t *)coll->image + getExpansionOffset(CE); |
| size = getExpansionCount(CE); |
| if (size != 0) { |
| /* |
| if there are less than 16 elements in expansion, we don't terminate |
| */ |
| for (count = 0; count < size; count++) { |
| *(source->CEpos ++) = *CEOffset++; |
| } |
| } |
| else { |
| /* else, we do */ |
| while (*CEOffset != 0) { |
| *(source->CEpos ++) = *CEOffset ++; |
| } |
| } |
| source->toReturn = source->CEpos - 1; |
| return *(source->toReturn); |
| case CHARSET_TAG: /* this tag always returns */ |
| /* probably after 1.8 */ |
| return UCOL_NOT_FOUND; |
| default: /* this tag always returns */ |
| *status = U_INTERNAL_PROGRAM_ERROR; |
| CE=0; |
| break; |
| } |
| if (CE <= UCOL_NOT_FOUND) { |
| break; |
| } |
| } |
| return CE; |
| } |
| |
| /* This should really be a macro */ |
| /* However, it is used only when stack buffers are not sufficiently big, and then we're messed up performance wise */ |
| /* anyway */ |
| uint8_t *reallocateBuffer(uint8_t **secondaries, uint8_t *secStart, uint8_t *second, uint32_t *secSize, uint32_t newSize, UErrorCode *status) { |
| fprintf(stderr, "."); |
| uint8_t *newStart = NULL; |
| |
| if(secStart==second) { |
| newStart=(uint8_t*)uprv_malloc(newSize); |
| if(newStart==NULL) { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| return NULL; |
| } |
| uprv_memcpy(newStart, secStart, *secondaries-secStart); |
| } else { |
| newStart=(uint8_t*)uprv_realloc(secStart, newSize); |
| if(newStart==NULL) { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| return NULL; |
| } |
| } |
| *secondaries=newStart+(*secondaries-secStart); |
| *secSize=newSize; |
| return newStart; |
| } |
| |
| |
| /* This should really be a macro */ |
| /* This function is used to reverse parts of a buffer. We need this operation when doing continuation */ |
| /* secondaries in French */ |
| /* |
| void uprv_ucol_reverse_buffer(uint8_t *start, uint8_t *end) { |
| uint8_t temp; |
| while(start<end) { |
| temp = *start; |
| *start++ = *end; |
| *end-- = temp; |
| } |
| } |
| */ |
| |
| #define uprv_ucol_reverse_buffer(TYPE, start, end) { \ |
| TYPE tempA; \ |
| while((start)<(end)) { \ |
| tempA = *(start); \ |
| *(start)++ = *(end); \ |
| *(end)-- = tempA; \ |
| } \ |
| } |
| |
| /****************************************************************************/ |
| /* Following are the sortkey generation functions */ |
| /* */ |
| /****************************************************************************/ |
| |
| /* sortkey API */ |
| 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; |
| /* this uses the function pointer that is set in updateinternalstate */ |
| /* currently, there are two funcs: */ |
| /*ucol_calcSortKey(...);*/ |
| /*ucol_calcSortKeySimpleTertiary(...);*/ |
| |
| int32_t keySize = coll->sortKeyGen(coll, source, sourceLength, &result, resultLength, FALSE, &status); |
| ((UCollator *)coll)->errorCode = status; /*semantically const */ |
| return keySize; |
| } |
| |
| /* this function is called by the C++ API for sortkey generation */ |
| 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 = coll->sortKeyGen(coll, source, sourceLength, &result, 0, TRUE, &status); |
| return result; |
| } |
| |
| |
| /* This function tries to get the size of a sortkey. It will be invoked if the size of resulting buffer is 0 */ |
| /* or if we run out of space while making a sortkey and want to return ASAP */ |
| 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 = (uint8_t)((strength >= UCOL_SECONDARY)?0:0xFF); |
| uint8_t compareTer = (uint8_t)((strength >= UCOL_TERTIARY)?0:0xFF); |
| uint8_t compareQuad = (uint8_t)((strength >= UCOL_QUATERNARY)?0:0xFF); |
| UBool compareIdent = (strength == UCOL_IDENTICAL); |
| UBool doCase = (coll->caseLevel == UCOL_ON); |
| UBool shifted = (coll->alternateHandling == UCOL_SHIFTED); |
| UBool qShifted = shifted && (compareQuad == 0); |
| UBool isFrenchSec = (coll->frenchCollation == UCOL_ON) && (compareSec == 0); |
| |
| uint8_t variableMax1 = coll->variableMax1; |
| uint8_t variableMax2 = coll->variableMax2; |
| uint8_t UCOL_COMMON_BOT4 = (uint8_t)(variableMax1+1); |
| uint8_t UCOL_BOT_COUNT4 = (uint8_t)(0xFF - UCOL_COMMON_BOT4); |
| |
| uint32_t order = UCOL_NO_MORE_CES; |
| uint8_t primary1 = 0; |
| uint8_t primary2 = 0; |
| uint32_t ce = 0; |
| uint8_t secondary = 0; |
| uint8_t tertiary = 0; |
| int32_t caseShift = 0; |
| uint32_t c2 = 0, c3 = 0, c4 = 0; /* variables for compression */ |
| |
| uint8_t caseSwitch = coll->caseSwitch; |
| uint8_t tertiaryMask = coll->tertiaryMask; |
| |
| UBool wasShifted = FALSE; |
| UBool notIsContinuation = FALSE; |
| |
| |
| for(;;) { |
| /*order = ucol_getNextCE(coll, s, &status);*/ |
| UCOL_GETNEXTCE(order, coll, *s, &status); |
| |
| if(order == UCOL_NO_MORE_CES) { |
| break; |
| } |
| /* fix me... we should check if we're in continuation first */ |
| if(isCEIgnorable(order)) { |
| continue; |
| } |
| |
| /* We're saving order in ce, since we will destroy order in order to get primary, secondary, tertiary in order ;)*/ |
| ce = order; |
| notIsContinuation = !isContinuation(ce); |
| |
| |
| order ^= caseSwitch; |
| if(notIsContinuation) { |
| tertiary = (uint8_t)((order & tertiaryMask)); |
| } else { |
| tertiary = (uint8_t)((order & UCOL_REMOVE_CASE)); |
| } |
| secondary = (uint8_t)((order >>= 8) & 0xFF); |
| primary2 = (uint8_t)((order >>= 8) & 0xFF); |
| primary1 = (uint8_t)(order >>= 8); |
| |
| |
| if(shifted && ((notIsContinuation && primary1 <= variableMax1 && primary1 > 0 |
| && (primary1 < variableMax1 || primary1 == variableMax1 && primary2 < variableMax2)) |
| || (!notIsContinuation && wasShifted))) { |
| if(compareQuad == 0) { |
| if(c4 > 0) { |
| currentSize += (c2/UCOL_BOT_COUNT4)+1; |
| c4 = 0; |
| } |
| currentSize++; |
| if(primary2 != 0) { |
| currentSize++; |
| } |
| } |
| wasShifted = TRUE; |
| } else { |
| wasShifted = FALSE; |
| /* 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_IGNORABLE) { |
| currentSize++; |
| if(primary2 != UCOL_IGNORABLE) { |
| currentSize++; |
| } |
| } |
| |
| if(secondary > compareSec) { /* I think that != 0 test should be != IGNORABLE */ |
| if(!isFrenchSec){ |
| if (secondary == UCOL_COMMON2 && notIsContinuation) { |
| c2++; |
| } else { |
| if(c2 > 0) { |
| if (secondary > UCOL_COMMON2) { // not necessary for 4th level. |
| currentSize += (c2/UCOL_TOP_COUNT2)+1; |
| } else { |
| currentSize += (c2/UCOL_BOT_COUNT2)+1; |
| } |
| c2 = 0; |
| } |
| currentSize++; |
| } |
| } else { |
| currentSize++; |
| } |
| } |
| |
| if(doCase) { |
| if (caseShift == 0) { |
| currentSize++; |
| caseShift = UCOL_CASE_SHIFT_START; |
| } |
| if(tertiary > 0 && notIsContinuation) { |
| caseShift--; |
| } |
| } |
| |
| if(tertiary > compareTer) { /* I think that != 0 test should be != IGNORABLE */ |
| if (tertiary == UCOL_COMMON3 && notIsContinuation) { |
| c3++; |
| } else { |
| if(c3 > 0) { |
| if (tertiary > UCOL_COMMON3) { // not necessary for 4th level. |
| currentSize += (c3/UCOL_TOP_COUNT3)+1; |
| } else { |
| currentSize += (c3/UCOL_BOT_COUNT3)+1; |
| } |
| c3 = 0; |
| } |
| currentSize++; |
| } |
| } |
| |
| if(qShifted && notIsContinuation) { |
| c4++; |
| } |
| |
| } |
| } |
| |
| if(c2 > 0) { |
| currentSize += (c2/UCOL_BOT_COUNT2)+1; |
| } |
| |
| if(c3 > 0) { |
| currentSize += (c3/UCOL_BOT_COUNT3)+1; |
| } |
| |
| if(c4 > 0 && compareQuad == 0) { |
| currentSize += (c4/UCOL_BOT_COUNT4)+1; |
| } |
| |
| 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++; |
| } |
| ident++; |
| } |
| |
| } |
| |
| return currentSize; |
| |
| } |
| |
| /* This is the sortkey work horse function */ |
| int32_t |
| ucol_calcSortKey(const UCollator *coll, |
| const UChar *source, |
| int32_t sourceLength, |
| uint8_t **result, |
| uint32_t resultLength, |
| UBool allocatePrimary, |
| UErrorCode *status) |
| { |
| uint32_t i = 0; /* general purpose counter */ |
| |
| /* Stack allocated buffers for buffers we use */ |
| uint8_t prim[UCOL_PRIMARY_MAX_BUFFER], second[UCOL_SECONDARY_MAX_BUFFER], tert[UCOL_TERTIARY_MAX_BUFFER], caseB[UCOL_CASE_MAX_BUFFER], quad[UCOL_QUAD_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 = prim; |
| resultLength = UCOL_PRIMARY_MAX_BUFFER; |
| } |
| |
| uint32_t secSize = UCOL_SECONDARY_MAX_BUFFER, terSize = UCOL_TERTIARY_MAX_BUFFER, |
| caseSize = UCOL_CASE_MAX_BUFFER, quadSize = UCOL_QUAD_MAX_BUFFER; |
| |
| uint32_t sortKeySize = 1; /* it is always \0 terminated */ |
| |
| UChar normBuffer[UCOL_NORMALIZATION_MAX_BUFFER]; |
| UChar *normSource = normBuffer; |
| int32_t normSourceLen = UCOL_NORMALIZATION_MAX_BUFFER; |
| |
| int32_t len = (sourceLength == -1 ? u_strlen(source) : sourceLength); |
| |
| uint8_t variableMax1 = coll->variableMax1; |
| uint8_t variableMax2 = coll->variableMax2; |
| uint8_t UCOL_COMMON_BOT4 = (uint8_t)(variableMax1+1); |
| uint8_t UCOL_BOT_COUNT4 = (uint8_t)(0xFF - UCOL_COMMON_BOT4); |
| |
| UColAttributeValue strength = coll->strength; |
| |
| uint8_t compareSec = (uint8_t)((strength >= UCOL_SECONDARY)?0:0xFF); |
| uint8_t compareTer = (uint8_t)((strength >= UCOL_TERTIARY)?0:0xFF); |
| uint8_t compareQuad = (uint8_t)((strength >= UCOL_QUATERNARY)?0:0xFF); |
| UBool compareIdent = (strength == UCOL_IDENTICAL); |
| UBool doCase = (coll->caseLevel == UCOL_ON); |
| UBool isFrenchSec = (coll->frenchCollation == UCOL_ON) && (compareSec == 0); |
| UBool shifted = (coll->alternateHandling == UCOL_SHIFTED); |
| UBool qShifted = shifted && (compareQuad == 0); |
| const uint8_t *scriptOrder = coll->scriptOrder; |
| |
| /* 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) + (qShifted?1:0)/*(compareQuad?0:1)*/ + (compareIdent?1:0)); |
| |
| collIterate s; |
| init_collIterate(coll, (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(compareIdent) { |
| if(unorm_quickCheck(source, len, UNORM_NFD, status) != UNORM_YES) { |
| normSourceLen = unorm_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 = unorm_normalize(source, sourceLength, UNORM_NFD, 0, normSource, (normSourceLen+1), status); |
| } |
| normSource[normSourceLen] = 0; |
| s.string = normSource; |
| s.pos = normSource; |
| s.len = normSource+normSourceLen; |
| } |
| } else if((normMode != UCOL_OFF) |
| /* changed by synwee */ |
| && !checkFCD(source, len, status)) |
| { |
| normSourceLen = unorm_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 = unorm_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 || primaries == NULL) { |
| return ucol_getSortKeySize(coll, &s, sortKeySize, strength, len); |
| } |
| uint8_t *primarySafeEnd = primaries + resultLength - 2; |
| |
| uint32_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; |
| |
| uint32_t order = 0; |
| uint32_t ce = 0; |
| |
| uint8_t primary1 = 0; |
| uint8_t primary2 = 0; |
| uint8_t secondary = 0; |
| uint8_t tertiary = 0; |
| uint8_t caseSwitch = coll->caseSwitch; |
| uint8_t tertiaryMask = coll->tertiaryMask; |
| UBool caseBit = FALSE; |
| |
| UBool finished = FALSE; |
| UBool resultOverflow = FALSE; |
| UBool wasShifted = FALSE; |
| UBool notIsContinuation = FALSE; |
| |
| uint32_t prevBuffSize = 0; |
| |
| uint32_t count2 = 0, count3 = 0, count4 = 0; |
| |
| for(;;) { |
| for(i=prevBuffSize; i<minBufferSize; ++i) { |
| |
| /*order = ucol_getNextCE(coll, &s, status);*/ |
| UCOL_GETNEXTCE(order, coll, s, status); |
| |
| if(order == UCOL_NO_MORE_CES) { |
| finished = TRUE; |
| break; |
| } |
| |
| /* fix me... we should check if we're in continuation first */ |
| if(isCEIgnorable(order)) { |
| continue; |
| } |
| |
| /* We're saving order in ce, since we will destroy order in order to get primary, secondary, tertiary in order ;)*/ |
| ce = order; |
| notIsContinuation = !isContinuation(ce); |
| |
| |
| order ^= caseSwitch; |
| caseBit = (order & UCOL_CASE_BIT_MASK); |
| if(notIsContinuation) { |
| tertiary = (uint8_t)((order & tertiaryMask)); |
| } else { |
| tertiary = (uint8_t)((order & UCOL_REMOVE_CASE)); |
| } |
| |
| secondary = (uint8_t)((order >>= 8) & UCOL_BYTE_SIZE_MASK); |
| primary2 = (uint8_t)((order >>= 8) & UCOL_BYTE_SIZE_MASK); |
| primary1 = (uint8_t)(order >>= 8); |
| |
| if(notIsContinuation) { |
| if(scriptOrder != NULL) { |
| primary1 = scriptOrder[primary1]; |
| } |
| } |
| |
| |
| /* 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 && ((notIsContinuation && primary1 <= variableMax1 && primary1 > 0 |
| && (primary1 < variableMax1 || primary1 == variableMax1 && primary2 < variableMax2)) |
| || (!notIsContinuation && wasShifted))) { |
| if(count4 > 0) { |
| while (count4 >= UCOL_BOT_COUNT4) { |
| *quads++ = (uint8_t)(UCOL_COMMON_BOT4 + UCOL_BOT_COUNT4); |
| count4 -= UCOL_BOT_COUNT4; |
| } |
| *quads++ = (uint8_t)(UCOL_COMMON_BOT4 + count4); |
| count4 = 0; |
| } |
| /* We are dealing with a variable and we're treating them as shifted */ |
| /* This is a shifted ignorable */ |
| if(primary1 != 0) { |
| *quads++ = primary1; |
| } |
| if(primary2 != 0) { |
| *quads++ = primary2; |
| } |
| wasShifted = TRUE; |
| } else { |
| wasShifted = FALSE; |
| /* 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_IGNORABLE) { |
| *primaries++ = primary1; /* scriptOrder[primary1]; */ /* This is the script ordering thingie */ |
| if(primary2 != UCOL_IGNORABLE) { |
| *primaries++ = primary2; /* second part */ |
| } |
| } |
| |
| if(secondary > compareSec) { |
| if(!isFrenchSec) { |
| /* This is compression code. */ |
| if (secondary == UCOL_COMMON2 && notIsContinuation) { |
| ++count2; |
| } else { |
| if (count2 > 0) { |
| if (secondary > UCOL_COMMON2) { // not necessary for 4th level. |
| while (count2 >= UCOL_TOP_COUNT2) { |
| *secondaries++ = UCOL_COMMON_TOP2 - UCOL_TOP_COUNT2; |
| count2 -= UCOL_TOP_COUNT2; |
| } |
| *secondaries++ = (uint8_t)(UCOL_COMMON_TOP2 - count2); |
| } else { |
| while (count2 >= UCOL_BOT_COUNT2) { |
| *secondaries++ = UCOL_COMMON_BOT2 + UCOL_BOT_COUNT2; |
| count2 -= UCOL_BOT_COUNT2; |
| } |
| *secondaries++ = (uint8_t)(UCOL_COMMON_BOT2 + count2); |
| } |
| count2 = 0; |
| } |
| *secondaries++ = secondary; |
| } |
| } else { |
| *secondaries++ = secondary; |
| /* 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(!notIsContinuation) { |
| if (frenchStartPtr == NULL) { |
| frenchStartPtr = secondaries - 2; |
| } |
| frenchEndPtr = secondaries-1; |
| } else if (frenchStartPtr != NULL) { |
| /* reverse secondaries from frenchStartPtr up to frenchEndPtr */ |
| uprv_ucol_reverse_buffer(uint8_t, frenchStartPtr, frenchEndPtr); |
| frenchStartPtr = NULL; |
| } |
| } |
| } |
| |
| if(doCase) { |
| if (caseShift == 0) { |
| *cases++ = UCOL_CASE_BYTE_START; |
| caseShift = UCOL_CASE_SHIFT_START; |
| } |
| if(notIsContinuation) { |
| if(tertiary != 0) { |
| *(cases-1) |= (caseBit!=0) << (--caseShift); |
| } else { |
| caseShift--; |
| } |
| } |
| } |
| |
| if(tertiary > compareTer) { |
| /* This is compression code. */ |
| /* sequence size check is included in the if clause */ |
| if (tertiary == UCOL_COMMON3 && notIsContinuation) { |
| ++count3; |
| } else { |
| if(tertiary > UCOL_COMMON3) { |
| tertiary |= UCOL_FLAG_BIT_MASK; |
| } |
| if (count3 > 0) { |
| if (tertiary > UCOL_COMMON3) { |
| while (count3 >= UCOL_TOP_COUNT3) { |
| *tertiaries++ = UCOL_COMMON_TOP3 - UCOL_TOP_COUNT3; |
| count3 -= UCOL_TOP_COUNT3; |
| } |
| *tertiaries++ = (uint8_t)(UCOL_COMMON_TOP3 - count3); |
| } else { |
| while (count3 >= UCOL_BOT_COUNT3) { |
| *tertiaries++ = UCOL_COMMON_BOT3 + UCOL_BOT_COUNT3; |
| count3 -= UCOL_BOT_COUNT3; |
| } |
| *tertiaries++ = (uint8_t)(UCOL_COMMON_BOT3 + count3); |
| } |
| count3 = 0; |
| } |
| *tertiaries++ = tertiary; |
| } |
| } |
| |
| if(qShifted && notIsContinuation) { |
| count4++; |
| } |
| } |
| |
| if(primaries > primarySafeEnd) { /* We have stepped over the primary buffer */ |
| int32_t sks = sortKeySize+(primaries - primStart)+(secondaries - secStart)+(tertiaries - terStart)+(cases-caseStart)+(quads-quadStart); |
| if(allocatePrimary == FALSE) { /* need to save our butts if we cannot reallocate */ |
| resultOverflow = TRUE; |
| sortKeySize = ucol_getSortKeySize(coll, &s, sks, strength, len); |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| finished = TRUE; |
| break; |
| } else { /* It's much nicer if we can actually reallocate */ |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sks, status); |
| *result = primStart; |
| primarySafeEnd = primStart + resultLength - 2; |
| } |
| } |
| } |
| if(finished) { |
| break; |
| } else { |
| prevBuffSize = minBufferSize; |
| secStart = reallocateBuffer(&secondaries, secStart, second, &secSize, 2*secSize, status); |
| terStart = reallocateBuffer(&tertiaries, terStart, tert, &terSize, 2*terSize, status); |
| caseStart = reallocateBuffer(&cases, caseStart, caseB, &caseSize, 2*caseSize, status); |
| quadStart = reallocateBuffer(&quads, quadStart, quad, &quadSize, 2*quadSize, status); |
| minBufferSize *= 2; |
| } |
| } |
| |
| /* Here, we are generally done with processing */ |
| /* bailing out would not be too productive */ |
| |
| |
| if(U_SUCCESS(*status)) { |
| sortKeySize += (primaries - primStart); |
| /* we have done all the CE's, now let's put them together to form a key */ |
| if(compareSec == 0) { |
| if (count2 > 0) { |
| while (count2 >= UCOL_BOT_COUNT2) { |
| *secondaries++ = UCOL_COMMON_BOT2 + UCOL_BOT_COUNT2; |
| count2 -= UCOL_BOT_COUNT2; |
| } |
| *secondaries++ = (uint8_t)(UCOL_COMMON_BOT2 + count2); |
| } |
| *(primaries++) = UCOL_LEVELTERMINATOR; |
| uint32_t secsize = secondaries-secStart; |
| sortKeySize += secsize; |
| if(sortKeySize <= resultLength) { |
| 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(uint8_t, frenchStartPtr, frenchEndPtr); |
| } |
| for(i = 0; i<secsize; i++) { |
| *(primaries++) = *(secondaries-i-1); |
| } |
| } else { |
| uprv_memcpy(primaries, secStart, secsize); |
| primaries += secsize; |
| } |
| } else { |
| if(allocatePrimary == TRUE) { /* need to save our butts if we cannot reallocate */ |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| 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(uint8_t, frenchStartPtr, frenchEndPtr); |
| } |
| for(i = 0; i<secsize; i++) { |
| *(primaries++) = *(secondaries-i-1); |
| } |
| } else { |
| uprv_memcpy(primaries, secStart, secsize); |
| primaries += secsize; |
| } |
| } else { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| } |
| |
| if(doCase) { |
| uint32_t casesize = cases - caseStart; |
| sortKeySize += casesize; |
| *(primaries++) = UCOL_LEVELTERMINATOR; |
| if(sortKeySize <= resultLength) { |
| uprv_memcpy(primaries, caseStart, casesize); |
| primaries += casesize; |
| } else { |
| if(allocatePrimary == TRUE) { |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| uprv_memcpy(primaries, caseStart, casesize); |
| } else { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| } |
| |
| if(compareTer == 0) { |
| if (count3 > 0) { |
| while (count3 >= UCOL_BOT_COUNT3) { |
| *tertiaries++ = UCOL_COMMON_BOT3 + UCOL_BOT_COUNT3; |
| count3 -= UCOL_BOT_COUNT3; |
| } |
| *tertiaries++ = (uint8_t)(UCOL_COMMON_BOT3 + count3); |
| } |
| uint32_t tersize = tertiaries - terStart; |
| sortKeySize += tersize; |
| *(primaries++) = UCOL_LEVELTERMINATOR; |
| if(sortKeySize <= resultLength) { |
| uprv_memcpy(primaries, terStart, tersize); |
| primaries += tersize; |
| if(/*compareQuad == 0*/qShifted == TRUE) { |
| if(count4 > 0) { |
| while (count4 >= UCOL_BOT_COUNT4) { |
| *quads++ = (uint8_t)(UCOL_COMMON_BOT4 + UCOL_BOT_COUNT4); |
| count4 -= UCOL_BOT_COUNT4; |
| } |
| *quads++ = (uint8_t)(UCOL_COMMON_BOT4 + count4); |
| } |
| *(primaries++) = UCOL_LEVELTERMINATOR; |
| uint32_t quadsize = quads - quadStart; |
| sortKeySize += quadsize; |
| if(sortKeySize <= resultLength) { |
| uprv_memcpy(primaries, quadStart, quadsize); |
| primaries += quadsize; |
| } else { |
| if(allocatePrimary == TRUE) { |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| uprv_memcpy(primaries, quadStart, quadsize); |
| } else { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| } |
| } else { |
| if(allocatePrimary == TRUE) { |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| uprv_memcpy(primaries, terStart, tersize); |
| } else { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| |
| 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 = (uint8_t)((*(ident) >> 8) + utf16fixup[*(ident) >> 11]); |
| if(idByte < 0x02) { |
| if(sortKeySize < resultLength) { |
| *(primaries++) = 0x01; |
| sortKeySize++; |
| *(primaries++) = (uint8_t)(idByte + 1); |
| } |
| } else { |
| *(primaries++) = idByte; |
| } |
| idByte = (uint8_t)((*(ident) & UCOL_BYTE_SIZE_MASK)); |
| if(idByte < 0x02) { |
| if(sortKeySize < resultLength) { |
| *(primaries++) = 0x01; |
| sortKeySize++; |
| *(primaries++) = (uint8_t)(idByte + 1); |
| } |
| } else { |
| *(primaries++) = idByte; |
| } |
| |
| ident++; |
| } |
| } else { |
| if(allocatePrimary == TRUE) { |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| while(ident < s.len) { |
| idByte = (uint8_t)((*(ident) >> 8) + utf16fixup[*(ident) >> 11]); |
| if(idByte < 0x02) { |
| *(primaries++) = 0x01; |
| sortKeySize++; |
| *(primaries++) = (uint8_t)(idByte + 1); |
| } else { |
| *(primaries++) = idByte; |
| } |
| idByte = (uint8_t)((*(ident) & UCOL_BYTE_SIZE_MASK)); |
| if(idByte < 0x02) { |
| *(primaries++) = 0x01; |
| sortKeySize++; |
| *(primaries++) = (uint8_t)(idByte + 1); |
| } else { |
| *(primaries++) = idByte; |
| } |
| ident++; |
| } |
| } else { |
| while(ident < s.len) { |
| idByte = (uint8_t)((*(ident) >> 8) + utf16fixup[*(ident) >> 11]); |
| if(idByte < 0x02) { |
| sortKeySize++; |
| } |
| idByte = (uint8_t)((*(ident) & UCOL_BYTE_SIZE_MASK)); |
| if(idByte < 0x02) { |
| sortKeySize++; |
| } |
| ident++; |
| } |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| } |
| } |
| |
| *(primaries++) = '\0'; |
| } |
| |
| if(terStart != tert) { |
| uprv_free(terStart); |
| uprv_free(secStart); |
| uprv_free(caseStart); |
| uprv_free(quadStart); |
| } |
| |
| if(normSource != normBuffer) { |
| uprv_free(normSource); |
| } |
| |
| if(allocatePrimary == TRUE) { |
| *result = (uint8_t*)uprv_malloc(sortKeySize); |
| uprv_memcpy(*result, primStart, sortKeySize); |
| if(primStart != prim) { |
| uprv_free(primStart); |
| } |
| } |
| |
| return sortKeySize; |
| } |
| |
| int32_t |
| ucol_calcSortKeySimpleTertiary(const UCollator *coll, |
| const UChar *source, |
| int32_t sourceLength, |
| uint8_t **result, |
| uint32_t resultLength, |
| UBool allocatePrimary, |
| UErrorCode *status) |
| { |
| uint32_t i = 0; /* general purpose counter */ |
| |
| /* Stack allocated buffers for buffers we use */ |
| uint8_t prim[UCOL_PRIMARY_MAX_BUFFER], second[UCOL_SECONDARY_MAX_BUFFER], tert[UCOL_TERTIARY_MAX_BUFFER]; |
| |
| uint8_t *primaries = *result, *secondaries = second, *tertiaries = tert; |
| |
| if(U_FAILURE(*status)) { |
| return 0; |
| } |
| |
| if(primaries == NULL && allocatePrimary == TRUE) { |
| primaries = *result = prim; |
| resultLength = UCOL_PRIMARY_MAX_BUFFER; |
| } |
| |
| uint32_t secSize = UCOL_SECONDARY_MAX_BUFFER, terSize = UCOL_TERTIARY_MAX_BUFFER; |
| |
| uint32_t sortKeySize = 3; /* it is always \0 terminated plus separators for secondary and tertiary */ |
| |
| UChar normBuffer[UCOL_NORMALIZATION_MAX_BUFFER]; |
| UChar *normSource = normBuffer; |
| int32_t normSourceLen = UCOL_NORMALIZATION_MAX_BUFFER; |
| |
| int32_t len = (sourceLength == -1 ? u_strlen(source) : sourceLength); |
| |
| |
| collIterate s; |
| init_collIterate(coll, (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) |
| /* && (unorm_quickCheck(source, len, UNORM_NFD, status) != UNORM_YES) |
| && (unorm_quickCheck(source, len, UNORM_NFC, status) != UNORM_YES)) */ |
| /* changed by synwee */ |
| && !checkFCD(source, len, status)) |
| { |
| |
| normSourceLen = unorm_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 = unorm_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 || primaries == NULL) { |
| return ucol_getSortKeySize(coll, &s, sortKeySize, coll->strength, len); |
| } |
| uint8_t *primarySafeEnd = primaries + resultLength - 2; |
| |
| uint32_t minBufferSize = UCOL_MAX_BUFFER; |
| |
| uint8_t *primStart = primaries; |
| uint8_t *secStart = secondaries; |
| uint8_t *terStart = tertiaries; |
| |
| uint32_t order = 0; |
| uint32_t ce = 0; |
| |
| uint8_t primary1 = 0; |
| uint8_t primary2 = 0; |
| uint8_t secondary = 0; |
| uint8_t tertiary = 0; |
| uint8_t caseSwitch = coll->caseSwitch; |
| uint8_t tertiaryMask = coll->tertiaryMask; |
| |
| |
| uint32_t prevBuffSize = 0; |
| |
| UBool finished = FALSE; |
| UBool resultOverflow = FALSE; |
| UBool notIsContinuation = FALSE; |
| |
| uint32_t count2 = 0, count3 = 0; |
| |
| for(;;) { |
| for(i=prevBuffSize; i<minBufferSize; ++i) { |
| |
| /*order = ucol_getNextCE(coll, &s, status);*/ |
| UCOL_GETNEXTCE(order, coll, s, status); |
| |
| if(isCEIgnorable(order)) { |
| continue; |
| } |
| |
| if(order == UCOL_NO_MORE_CES) { |
| 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; |
| notIsContinuation = !isContinuation(ce); |
| |
| order ^= caseSwitch; |
| if(notIsContinuation) { |
| tertiary = (uint8_t)((order & tertiaryMask)); |
| } else { |
| tertiary = (uint8_t)((order & UCOL_REMOVE_CASE)); |
| } |
| secondary = (uint8_t)((order >>= 8) & UCOL_BYTE_SIZE_MASK); |
| primary2 = (uint8_t)((order >>= 8) & UCOL_BYTE_SIZE_MASK); |
| primary1 = (uint8_t)(order >>= 8); |
| |
| /* 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. */ |
| |
| /* 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_IGNORABLE) { |
| *primaries++ = primary1; /* scriptOrder[primary1]; */ /* This is the script ordering thingie */ |
| if(primary2 != UCOL_IGNORABLE) { |
| *primaries++ = primary2; /* second part */ |
| } |
| } |
| |
| if(secondary > 0) { /* I think that != 0 test should be != IGNORABLE */ |
| /* This is compression code. */ |
| if (secondary == UCOL_COMMON2 && notIsContinuation) { |
| ++count2; |
| } else { |
| if (count2 > 0) { |
| if (secondary > UCOL_COMMON2) { // not necessary for 4th level. |
| while (count2 >= UCOL_TOP_COUNT2) { |
| *secondaries++ = UCOL_COMMON_TOP2 - UCOL_TOP_COUNT2; |
| count2 -= UCOL_TOP_COUNT2; |
| } |
| *secondaries++ = (uint8_t)(UCOL_COMMON_TOP2 - count2); |
| } else { |
| while (count2 >= UCOL_BOT_COUNT2) { |
| *secondaries++ = UCOL_COMMON_BOT2 + UCOL_BOT_COUNT2; |
| count2 -= UCOL_BOT_COUNT2; |
| } |
| *secondaries++ = (uint8_t)(UCOL_COMMON_BOT2 + count2); |
| } |
| count2 = 0; |
| } |
| *secondaries++ = secondary; |
| } |
| } |
| |
| |
| if(tertiary > 0) { |
| /* This is compression code. */ |
| /* sequence size check is included in the if clause */ |
| if (tertiary == UCOL_COMMON3 && notIsContinuation) { |
| ++count3; |
| } else { |
| if(tertiary > UCOL_COMMON3) { |
| tertiary |= UCOL_FLAG_BIT_MASK; |
| } |
| if (count3 > 0) { |
| if (tertiary > UCOL_COMMON3) { |
| while (count3 >= UCOL_TOP_COUNT3) { |
| *tertiaries++ = UCOL_COMMON_TOP3 - UCOL_TOP_COUNT3; |
| count3 -= UCOL_TOP_COUNT3; |
| } |
| *tertiaries++ = (uint8_t)(UCOL_COMMON_TOP3 - count3); |
| } else { |
| while (count3 >= UCOL_BOT_COUNT3) { |
| *tertiaries++ = UCOL_COMMON_BOT3 + UCOL_BOT_COUNT3; |
| count3 -= UCOL_BOT_COUNT3; |
| } |
| *tertiaries++ = (uint8_t)(UCOL_COMMON_BOT3 + count3); |
| } |
| count3 = 0; |
| } |
| *tertiaries++ = tertiary; |
| } |
| } |
| |
| if(primaries > primarySafeEnd) { /* We have stepped over the primary buffer */ |
| int32_t sks = sortKeySize+(primaries - primStart)+(secondaries - secStart)+(tertiaries - terStart); |
| if(allocatePrimary == FALSE) { /* need to save our butts if we cannot reallocate */ |
| resultOverflow = TRUE; |
| sortKeySize = ucol_getSortKeySize(coll, &s, sks, coll->strength, len); |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| finished = TRUE; |
| break; |
| } else { /* It's much nicer if we can actually reallocate */ |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sks, status); |
| *result = primStart; |
| primarySafeEnd = primStart + resultLength - 2; |
| } |
| } |
| } |
| if(finished) { |
| break; |
| } else { |
| prevBuffSize = minBufferSize; |
| secStart = reallocateBuffer(&secondaries, secStart, second, &secSize, 2*secSize, status); |
| terStart = reallocateBuffer(&tertiaries, terStart, tert, &terSize, 2*terSize, status); |
| minBufferSize *= 2; |
| } |
| } |
| |
| if(U_SUCCESS(*status)) { |
| sortKeySize += (primaries - primStart); |
| /* we have done all the CE's, now let's put them together to form a key */ |
| if (count2 > 0) { |
| while (count2 >= UCOL_BOT_COUNT2) { |
| *secondaries++ = UCOL_COMMON_BOT2 + UCOL_BOT_COUNT2; |
| count2 -= UCOL_BOT_COUNT2; |
| } |
| *secondaries++ = (uint8_t)(UCOL_COMMON_BOT2 + count2); |
| } |
| uint32_t secsize = secondaries-secStart; |
| sortKeySize += secsize; |
| if(sortKeySize <= resultLength) { |
| *(primaries++) = UCOL_LEVELTERMINATOR; |
| uprv_memcpy(primaries, secStart, secsize); |
| primaries += secsize; |
| } else { |
| if(allocatePrimary == TRUE) { |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| uprv_memcpy(primaries, secStart, secsize); |
| } else { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| |
| if (count3 > 0) { |
| while (count3 >= UCOL_BOT_COUNT3) { |
| *tertiaries++ = UCOL_COMMON_BOT3 + UCOL_BOT_COUNT3; |
| count3 -= UCOL_BOT_COUNT3; |
| } |
| *tertiaries++ = (uint8_t)(UCOL_COMMON_BOT3 + count3); |
| } |
| *(primaries++) = UCOL_LEVELTERMINATOR; |
| uint32_t tersize = tertiaries - terStart; |
| sortKeySize += tersize; |
| if(sortKeySize <= resultLength) { |
| uprv_memcpy(primaries, terStart, tersize); |
| primaries += tersize; |
| } else { |
| if(allocatePrimary == TRUE) { |
| primStart = reallocateBuffer(&primaries, *result, prim, &resultLength, 2*sortKeySize, status); |
| *result = primStart; |
| uprv_memcpy(primaries, terStart, tersize); |
| } else { |
| *status = U_MEMORY_ALLOCATION_ERROR; |
| } |
| } |
| |
| *(primaries++) = '\0'; |
| } |
| |
| if(terStart != tert) { |
| uprv_free(terStart); |
| uprv_free(secStart); |
| } |
| |
| if(normSource != normBuffer) { |
| uprv_free(normSource); |
| } |
| |
| if(allocatePrimary == TRUE) { |
| *result = (uint8_t*)uprv_malloc(sortKeySize); |
| uprv_memcpy(*result, primStart, sortKeySize); |
| if(primStart != prim) { |
| uprv_free(primStart); |
| } |
| } |
| |
| return sortKeySize; |
| } |
| |
| /* this function makes a string with representation of a sortkey */ |
| U_CAPI char U_EXPORT2 *ucol_sortKeyToString(const UCollator *coll, const uint8_t *sortkey, char *buffer, uint32_t *len) { |
| int32_t strength = UCOL_PRIMARY; |
| uint32_t res_size = 0; |
| UBool doneCase = FALSE; |
| |
| char *current = buffer; |
| const uint8_t *currentSk = sortkey; |
| |
| sprintf(current, "["); |
| current++; |
| |
| while(strength <= UCOL_QUATERNARY && strength <= coll->strength) { |
| if(strength > UCOL_PRIMARY) { |
| sprintf(current, " . "); |
| current += 3; |
| } |
| while(*currentSk != 0x01 && *currentSk != 0x00) { /* print a level */ |
| sprintf(current, "%02X ", *currentSk++); |
| current+=3; |
| } |
| if(coll->caseLevel == UCOL_ON && strength == UCOL_SECONDARY && doneCase == FALSE) { |
| doneCase = TRUE; |
| } else if(coll->caseLevel == UCOL_OFF || doneCase == TRUE || strength != UCOL_SECONDARY) { |
| strength ++; |
| } |
| sprintf(current, "%02X", *(currentSk++)); /* This should print '01' */ |
| current +=2; |
| if(strength == UCOL_QUATERNARY && coll->alternateHandling == UCOL_NON_IGNORABLE) { |
| break; |
| } |
| } |
| |
| if(coll->strength == UCOL_IDENTICAL) { |
| sprintf(current, " . "); |
| current += 3; |
| while(*currentSk != 0) { |
| if(*currentSk == 0x01) { |
| sprintf(current, "%02X", *(currentSk++)); |
| current +=2; |
| } |
| |
| sprintf(current, "%02X%02X ", *currentSk, *(currentSk+1)); |
| current +=5; |
| currentSk+=2; |
| } |
| |
| sprintf(current, "%02X", *(currentSk++)); /* This should print '00' */ |
| current += 2; |
| |
| } |
| sprintf(current, "]"); |
| current += 3; |
| |
| if(res_size > *len) { |
| return NULL; |
| } |
| |
| return buffer; |
| |
| |
| } |
| |
| /* This is a trick string compare function that goes in and uses sortkeys to compare */ |
| /* It is used when compare gets in trouble and needs to bail out */ |
| UCollationResult ucol_compareUsingSortKeys(const UCollator *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; |
| } |
| } |
| |
| |
| /****************************************************************************/ |
| /* Following are the functions that deal with the properties of a collator */ |
| /* there are new APIs and some compatibility APIs */ |
| /****************************************************************************/ |
| void ucol_updateInternalState(UCollator *coll) { |
| if(coll->caseFirst == UCOL_UPPER_FIRST) { |
| coll->caseSwitch = UCOL_CASE_SWITCH; |
| } else { |
| coll->caseSwitch = UCOL_NO_CASE_SWITCH; |
| } |
| if(coll->caseLevel == UCOL_ON || coll->caseFirst == UCOL_OFF) { |
| coll->tertiaryMask = UCOL_REMOVE_CASE; |
| } else { |
| coll->tertiaryMask = UCOL_KEEP_CASE; |
| } |
| if(coll->caseLevel == UCOL_OFF && coll->strength == UCOL_TERTIARY |
| && coll->frenchCollation == UCOL_OFF && coll->alternateHandling == UCOL_NON_IGNORABLE) { |
| coll->sortKeyGen = ucol_calcSortKeySimpleTertiary; |
| } else { |
| coll->sortKeyGen = ucol_calcSortKey; |
| } |
| |
| } |
| |
| /* Attribute setter API */ |
| 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_OFF) { |
| coll->caseFirst = UCOL_OFF; |
| 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; |
| } |
| ucol_updateInternalState(coll); |
| } |
| |
| 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; |
| } |
| } |
| |
| 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; |
| } |
| } |
| |
| U_CAPI void |
| ucol_setStrength( UCollator *coll, |
| UCollationStrength strength) |
| { |
| UErrorCode status = U_ZERO_ERROR; |
| ucol_setAttribute(coll, UCOL_STRENGTH, strength, &status); |
| } |
| |
| U_CAPI UCollationStrength |
| ucol_getStrength(const UCollator *coll) |
| { |
| UErrorCode status = U_ZERO_ERROR; |
| return ucol_getAttribute(coll, UCOL_STRENGTH, &status); |
| } |
| |
| /****************************************************************************/ |
| /* Following are misc functions */ |
| /* there are new APIs and some compatibility APIs */ |
| /****************************************************************************/ |
| |
| U_CAPI UCollator * |
| ucol_safeClone(const UCollator *coll, void *stackBuffer, int32_t * pBufferSize, UErrorCode *status) |
| { |
| UCollator * localCollator; |
| int32_t bufferSizeNeeded = sizeof(UCollator); |
| |
| if (status == NULL || U_FAILURE(*status)){ |
| return 0; |
| } |
| if (!pBufferSize || !coll){ |
| *status = U_ILLEGAL_ARGUMENT_ERROR; |
| return 0; |
| } |
| if (*pBufferSize == 0){ /* 'preflighting' request - set needed size into *pBufferSize */ |
| *pBufferSize = bufferSizeNeeded; |
| return 0; |
| } |
| if (*pBufferSize < bufferSizeNeeded || stackBuffer == NULL) { |
| /* allocate one here...*/ |
| int32_t length; |
| const UChar * rules = ucol_getRules(coll, &length); |
| |
| localCollator = ucol_openRules(rules, |
| length, |
| ucol_getNormalization(coll), |
| ucol_getStrength(coll), |
| status); |
| if (U_SUCCESS(*status)) |
| { |
| *status = U_SAFECLONE_ALLOCATED_ERROR; |
| } |
| } else { |
| localCollator = (UCollator *)stackBuffer; |
| memcpy(localCollator, coll, sizeof(UCollator)); |
| localCollator->freeOnClose = FALSE; |
| } |
| return localCollator; |
| } |
| |
| U_CAPI int32_t |
| ucol_getRulesEx(const UCollator *coll, UColRuleOption delta, UChar *buffer, int32_t bufferLen) { |
| int32_t len = 0; |
| int32_t UCAlen = 0; |
| const UChar* ucaRules = 0; |
| const UChar *rules = ucol_getRules(coll, &len); |
| if(delta == UCOL_FULL_RULES) { |
| UErrorCode status = U_ZERO_ERROR; |
| /* take the UCA rules and append real rules at the end */ |
| /* UCA rules will be probably coming from the root RB */ |
| ucaRules = ures_getStringByKey(coll->rb,"%%UCARULES",&UCAlen,&status); |
| } |
| if(buffer){ |
| *buffer=0; |
| if(bufferLen >= len + UCAlen) { |
| u_strcat(buffer, rules); |
| if(UCAlen >0) |
| u_strcat(buffer,ucaRules); |
| } else { |
| u_strncat(buffer, rules, (bufferLen-UCAlen)*sizeof(UChar)); |
| } |
| } |
| return len+UCAlen; |
| } |
| |
| U_CAPI const UChar* |
| ucol_getRules( const UCollator *coll, |
| int32_t *length) |
| { |
| if(coll->rules != NULL) { |
| *length = u_strlen(coll->rules); |
| return coll->rules; |
| } else { |
| UErrorCode status = U_ZERO_ERROR; |
| if(coll->rb != NULL) { |
| UResourceBundle *collElem = ures_getByKey(coll->rb, "CollationElements", NULL, &status); |
| if(U_SUCCESS(status)) { |
| /*Semantic const */ |
| ((UCollator *)coll)->rules = ures_getStringByKey(collElem, "Sequence", length, &status); |
| ((UCollator *)coll)->freeRulesOnClose = FALSE; |
| ures_close(collElem); |
| return coll->rules; |
| } |
| } |
| *length = 0; |
| return &coll->zero; |
| } |
| } |
| |
| 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); |
| return uprv_fillOutputString(dst, result, resultLength, status); |
| } |
| |
| U_CAPI const char* |
| ucol_getAvailable(int32_t index) |
| { |
| return uloc_getAvailable(index); |
| } |
| |
| U_CAPI int32_t |
| ucol_countAvailable() |
| { |
| return uloc_countAvailable(); |
| } |
| |
| U_CAPI void |
| ucol_getVersion(const UCollator* coll, |
| UVersionInfo versionInfo) |
| { |
| /* RunTime version */ |
| uint8_t rtVersion = UCOL_RUNTIME_VERSION; |
| /* Builder version*/ |
| uint8_t bdVersion = coll->dataInfo.dataVersion[0]; |
| |
| /* Charset Version. Need to get the version from cnv files |
| * makeconv should populate cnv files with version and |
| * an api has to be provided in ucnv.h to obtain this version |
| */ |
| uint8_t csVersion = 0; |
| |
| /* combine the version info */ |
| uint16_t cmbVersion = (uint16_t)((rtVersion<<11) | (bdVersion<<6) | (csVersion)); |
| |
| /* Tailoring rules */ |
| versionInfo[0] = (uint8_t)(cmbVersion>>8); |
| versionInfo[1] = (uint8_t)cmbVersion; |
| versionInfo[2] = coll->dataInfo.dataVersion[1]; |
| versionInfo[3] = UCA->dataInfo.dataVersion[1]; |
| } |
| |
| |
| static UBool ucol_unsafeCP(UChar c, const UCollator *coll) { |
| int32_t hash = c; |
| uint8_t htbyte; |
| |
| if (hash >= UCOL_UNSAFECP_TABLE_SIZE*8) { |
| if (hash >= 0xd800 && hash <= 0xf8ff) { |
| /* Part of a surrogate, or in private use area. */ |
| /* These are always considered unsafe. */ |
| return TRUE; |
| } |
| hash = (hash & UCOL_UNSAFECP_TABLE_MASK) + 256; |
| } |
| htbyte = coll->unsafeCP[hash>>3]; |
| if (((htbyte >> (hash & 7)) & 1) == 1) |
| return TRUE; |
| |
| /* TODO: main UCA table data needs to be merged into tailoring tables, */ |
| /* and this second level of test removed from here. */ |
| if (coll == UCA) |
| return FALSE; |
| |
| htbyte = UCA->unsafeCP[hash>>3]; |
| return ((htbyte >> (hash & 7)) & 1) == 1; |
| } |
| |
| /* This internal API checks whether a character is tailored or not */ |
| U_CAPI UBool isTailored(const UCollator *coll, const UChar u, UErrorCode *status) { |
| uint32_t CE = UCOL_NOT_FOUND; |
| const UChar *ContractionStart = NULL; |
| if(U_SUCCESS(*status) && coll != NULL) { |
| if(coll == UCA) { |
| return FALSE; |
| } else if(u < 0x100) { /* latin-1 */ |
| CE = coll->latinOneMapping[u]; |
| if(CE == UCA->latinOneMapping[u]) { |
| return FALSE; |
| } |
| } else { /* regular */ |
| CE = ucmp32_get(coll->mapping, u); |
| } |
| |
| if(isContraction(CE)) { |
| ContractionStart = (UChar *)coll->image+getContractOffset(CE); |
| CE = *(coll->contractionCEs + (ContractionStart- coll->contractionIndex)); |
| } |
| |
| if(CE == UCOL_NOT_FOUND) { |
| return FALSE; |
| } else { |
| return TRUE; |
| } |
| } else { |
| return FALSE; |
| } |
| } |
| |
| |
| /****************************************************************************/ |
| /* Following are the string compare functions */ |
| /* */ |
| /****************************************************************************/ |
| |
| /* compare two strings... Can get interesting */ |
| U_CAPI UCollationResult |
| ucol_strcoll( const UCollator *coll, |
| const UChar *source, |
| int32_t sourceLength, |
| const UChar *target, |
| int32_t targetLength) |
| { |
| /* check if source and target are valid strings */ |
| if (source==target && sourceLength==targetLength) |
| { |
| return UCOL_EQUAL; |
| } |
| |
| /* |
| sourceLength = sourceLength == -1 ? u_strlen(source) : sourceLength; |
| targetLength = targetLength == -1 ? u_strlen(target) : targetLength; |
| |
| if(sourceLength == targetLength && uprv_memcmp(source, target, sizeof(UChar)*sourceLength) == 0) { |
| return UCOL_EQUAL; |
| } |
| */ |
| |
| /* Scan the strings. Find: */ |
| /* their length, if not given by caller */ |
| /* The length of any leading portion that is equal */ |
| /* Whether they are exactly equal. (in which case we just return */ |
| const UChar *pSrc = source; |
| const UChar *pTarg = target; |
| |
| const UChar *pSrcEnd = source + sourceLength; |
| const UChar *pTargEnd = target + targetLength; |
| |
| int32_t equalLength; |
| |
| // Scan while the strings are bitwise ==, or until one is exhausted. |
| for (;;) { |
| if (pSrc == pSrcEnd || pTarg == pTargEnd) |
| break; |
| if (*pSrc == 0 && (sourceLength == -1 || targetLength == -1)) |
| break; |
| if (*pSrc != *pTarg) |
| break; |
| pSrc++; |
| pTarg++; |
| } |
| equalLength = pSrc - source; |
| |
| // If we made it all the way through both strings, we are done. They are == |
| if ((pSrc ==pSrcEnd || (pSrcEnd <pSrc && *pSrc==0)) && /* At end of src string, however it was specified. */ |
| (pTarg==pTargEnd || (pTargEnd<pTarg && *pTarg==0))) /* and also at end of dest string */ |
| return UCOL_EQUAL; |
| |
| // If we don't know the length of the src string, continue scanning it to get the length.. |
| if (sourceLength == -1) { |
| while (*pSrc != 0 ) { |
| pSrc++; |
| } |
| sourceLength = pSrc - source; |
| } |
| |
| // If we don't know the length of the targ string, continue scanning it to get the length.. |
| if (targetLength == -1) { |
| while (*pTarg != 0 ) { |
| pTarg++; |
| } |
| targetLength = pTarg - target; |
| } |
| |
| |
| if (equalLength > 2) { |
| /* There is an identical portion at the beginning of the two strings. */ |
| /* If the identical portion ends within a contraction or a comibining */ |
| /* character sequence, back up to the start of that sequence. */ |
| pSrc = source + equalLength; /* point to the first differing chars */ |
| pTarg = target + equalLength; |
| if (pSrc != source+sourceLength && ucol_unsafeCP(*pSrc, coll) || |
| pTarg != target+targetLength && ucol_unsafeCP(*pTarg, coll)) |
| { |
| // We are stopped in the middle of a contraction. |
| // Scan backwards through the == part of the string looking for the start of the contraction. |
| // It doesn't matter which string we scan, since they are the same in this region. |
| do |
| { |
| equalLength--; |
| pSrc--; |
| } |
| while (equalLength>0 && ucol_unsafeCP(*pSrc, coll)); |
| } |
| |
| source += equalLength; |
| target += equalLength; |
| sourceLength -= equalLength; |
| targetLength -= equalLength; |
| } |
| |
| |
| |
| UColAttributeValue strength = coll->strength; |
| UBool initialCheckSecTer = (strength >= UCOL_SECONDARY); |
| |
| UBool checkSecTer = initialCheckSecTer; |
| UBool checkTertiary = (strength >= UCOL_TERTIARY); |
| UBool checkQuad = (strength >= UCOL_QUATERNARY); |
| UBool checkIdent = (strength == UCOL_IDENTICAL); |
| UBool checkCase = (coll->caseLevel == UCOL_ON); |
| UBool isFrenchSec = (coll->frenchCollation == UCOL_ON) && checkSecTer; |
| UBool shifted = (coll->alternateHandling == UCOL_SHIFTED); |
| UBool qShifted = shifted && checkQuad; |
| |
| 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; |
| |
| |
| init_collIterate(coll, source, sourceLength, &sColl, FALSE); |
| if(checkIdent) { |
| if(unorm_quickCheck(sColl.string, sColl.len - sColl.string, UNORM_NFD, &status) != UNORM_YES) { |
| normSourceLength = unorm_normalize(source, sourceLength, UNORM_NFD, 0, normSource, normSourceLength, &status); |
| /* if we don't have enough space in buffers, we'll recursively call strcoll, so that we have single point */ |
| /* of exit - to free buffers we allocated. Otherwise, returns from strcoll are in various places and it */ |
| /* would be hard to track all the exit points. */ |
| if(U_FAILURE(status)) { /* This would be buffer overflow */ |
| UColAttributeValue mode = coll->normalizationMode; |
| normSourceP = (UChar *)uprv_malloc((normSourceLength+1)*sizeof(UChar)); |
| status = U_ZERO_ERROR; |
| normSourceLength = unorm_normalize(source, sourceLength, UNORM_NFD, 0, normSourceP, normSourceLength+1, &status); |
| normTargetLength = unorm_normalize(target, targetLength, UNORM_NFD, 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 = unorm_normalize(target, targetLength, UNORM_NFD, 0, normTargetP, normTargetLength+1, &status); |
| } |
| ((UCollator *)coll)->normalizationMode = UCOL_OFF; |
| UCollationResult result = ucol_strcoll(coll, normSourceP, normSourceLength, normTargetP, normTargetLength); |
| ((UCollator *)coll)->normalizationMode = mode; |
| uprv_free(normSourceP); |
| if(normTargetP != normTarget) { |
| uprv_free(normTargetP); |
| } |
| return result; |
| } |
| init_collIterate(coll, normSource, normSourceLength, &sColl, TRUE); |
| } |
| } else if((coll->normalizationMode == UCOL_ON) |
| /* && (unorm_quickCheck( sColl.string, sColl.len - sColl.string, UNORM_NFD, &status) != UNORM_YES) |
| && (unorm_quickCheck( sColl.string, sColl.len - sColl.string, UNORM_NFC, &status) != UNORM_YES)) */ |
| /* changed by synwee */ |
| && !checkFCD(sColl.string, sColl.len - sColl.string, &status)) |
| { |
| normSourceLength = unorm_normalize(source, sourceLength, UNORM_NFD, 0, normSource, normSourceLength, &status); |
| /* if we don't have enough space in buffers, we'll recursively call strcoll, so that we have single point */ |
| /* of exit - to free buffers we allocated. Otherwise, returns from strcoll are in various places and it */ |
| /* would be hard to track all the exit points. */ |
| if(U_FAILURE(status)) { /* This would be buffer overflow */ |
| UColAttributeValue mode = coll->normalizationMode; |
| normSourceP = (UChar *)uprv_malloc((normSourceLength+1)*sizeof(UChar)); |
| status = U_ZERO_ERROR; |
| normSourceLength = unorm_normalize(source, sourceLength, UNORM_NFD, 0, normSourceP, normSourceLength+1, &status); |
| normTargetLength = unorm_normalize(target, targetLength, UNORM_NFD, 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 = unorm_normalize(target, targetLength, UNORM_NFD, 0, normTargetP, normTargetLength+1, &status); |
| } |
| ((UCollator *)coll)->normalizationMode = UCOL_OFF; |
| UCollationResult result = ucol_strcoll(coll, normSourceP, normSourceLength, normTargetP, normTargetLength); |
| ((UCollator *)coll)->normalizationMode = mode; |
| uprv_free(normSourceP); |
| if(normTargetP != normTarget) { |
| uprv_free(normTargetP); |
| } |
| return result; |
| } |
| init_collIterate(coll, normSource, normSourceLength, &sColl, TRUE); |
| } |
| |
| init_collIterate(coll, target, targetLength, &tColl, FALSE); |
| if(checkIdent) { |
| if(unorm_quickCheck(tColl.string, tColl.len - tColl.string, UNORM_NFD, &status) != UNORM_YES) { |
| normTargetLength = unorm_normalize(target, targetLength, UNORM_NFD, 0, normTarget, normTargetLength, &status); |
| if(U_FAILURE(status)) { /* This would be buffer overflow */ |
| UColAttributeValue mode = coll->normalizationMode; |
| normTargetP = (UChar *)uprv_malloc((normTargetLength+1)*sizeof(UChar)); |
| status = U_ZERO_ERROR; |
| normTargetLength = unorm_normalize(target, targetLength, UNORM_NFD, 0, normTargetP, normTargetLength+1, &status); |
| ((UCollator *)coll)->normalizationMode = UCOL_OFF; |
| UCollationResult result = ucol_strcoll(coll, normSourceP, normSourceLength, normTargetP, normTargetLength); |
| ((UCollator *)coll)->normalizationMode = mode; |
| uprv_free(normTargetP); |
| return result; |
| } |
| init_collIterate(coll, normTarget, normTargetLength, &tColl, TRUE); |
| } |
| } else if((coll->normalizationMode == UCOL_ON) |
| /* && (unorm_quickCheck(tColl.string, tColl.len - tColl.string, UNORM_NFD, &status) != UNORM_YES) |
| && (unorm_quickCheck(tColl.string, tColl.len - tColl.string, UNORM_NFC, &status) != UNORM_YES)) */ |
| /* changed by synwee */ |
| && !checkFCD(tColl.string, tColl.len - tColl.string, &status)) |
| { |
| normTargetLength = unorm_normalize(target, targetLength, UNORM_NFD, 0, normTarget, normTargetLength, &status); |
| if(U_FAILURE(status)) { /* This would be buffer overflow */ |
| UColAttributeValue mode = coll->normalizationMode; |
| normTargetP = (UChar *)uprv_malloc((normTargetLength+1)*sizeof(UChar)); |
| status = U_ZERO_ERROR; |
| normTargetLength = unorm_normalize(target, targetLength, UNORM_NFD, 0, normTargetP, normTargetLength+1, &status); |
| ((UCollator *)coll)->normalizationMode = UCOL_OFF; |
| UCollationResult result = ucol_strcoll(coll, normSourceP, normSourceLength, normTargetP, normTargetLength); |
| ((UCollator *)coll)->normalizationMode = mode; |
| uprv_free(normTargetP); |
| return result; |
| } |
| init_collIterate(coll, normTarget, normTargetLength, &tColl, TRUE); |
| } |
| |
| if (U_FAILURE(status)) |
| { |
| return UCOL_EQUAL; |
| } |
| |
| uint32_t sCEsArray[512], tCEsArray[512]; |
| uint32_t *sCEs = sCEsArray, *tCEs = tCEsArray; |
| uint32_t *sCEend = sCEs+512, *tCEend = tCEs+512; |
| uint8_t caseSwitch = coll->caseSwitch; |
| uint8_t tertiaryMask = coll->tertiaryMask; |
| |
| uint32_t LVT = (shifted)?((coll->variableMax1)<<24 | (coll->variableMax2)<<16):0; |
| |
| uint32_t secS = 0, secT = 0; |
| |
| uint32_t sOrder=0, tOrder=0; |
| if(!shifted) { |
| for(;;) { |
| if(sCEs == sCEend || tCEs == tCEend) { |
| return ucol_compareUsingSortKeys(coll, source, sourceLength, target, targetLength); |
| } |
| |
| /* Get the next collation element in each of the strings, unless */ |
| /* we've been requested to skip it. */ |
| while(sOrder == 0) { |
| UCOL_GETNEXTCE(sOrder, coll, sColl, &status); |
| /*sOrder = ucol_getNextCE(coll, &sColl, &status);*/ |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| sOrder &= 0xFFFF0000; |
| } |
| |
| while(tOrder == 0) { |
| UCOL_GETNEXTCE(tOrder, coll, tColl, &status); |
| /*tOrder = ucol_getNextCE(coll, &tColl, &status);*/ |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| tOrder &= 0xFFFF0000; |
| } |
| |
| if(sOrder == tOrder) { |
| if(sOrder == 0x00010000) { |
| |
| break; |
| } else { |
| sOrder = 0; tOrder = 0; |
| continue; |
| } |
| } else if(sOrder < tOrder) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } /* no primary difference... do the rest from the buffers */ |
| } else { /* shifted - do a slightly more complicated processing */ |
| for(;;) { |
| UBool sInShifted = FALSE; |
| UBool tInShifted = FALSE; |
| |
| if(sCEs == sCEend || tCEs == tCEend) { |
| return ucol_compareUsingSortKeys(coll, source, sourceLength, target, targetLength); |
| } |
| |
| /* This is where abridged version for shifted should go */ |
| for(;;) { |
| UCOL_GETNEXTCE(sOrder, coll, sColl, &status); |
| /*sOrder = ucol_getNextCE(coll, &sColl, &status);*/ |
| if(sOrder == UCOL_NO_MORE_CES) { |
| *(sCEs++) = sOrder; |
| break; |
| } else if((sOrder & 0xFFFFFFBF) == 0) { |
| continue; |
| } else if(isContinuation(sOrder)) { |
| if((sOrder & 0xFFFF0000) > 0) { /* There is primary value */ |
| if(sInShifted) { |
| sOrder &= 0xFFFF0000; |
| *(sCEs++) = sOrder; |
| continue; |
| } else { |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| break; |
| } |
| } else { /* Just lower level values */ |
| if(sInShifted) { |
| continue; |
| } else { |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| continue; |
| } |
| } |
| } else { /* regular */ |
| if(sOrder > LVT) { |
| *(sCEs++) = sOrder; |
| break; |
| } else { |
| if((sOrder & 0xFFFF0000) > 0) { |
| sInShifted = TRUE; |
| sOrder &= 0xFFFF0000; |
| *(sCEs++) = sOrder; |
| continue; |
| } else { |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| continue; |
| } |
| } |
| } |
| } |
| sOrder &= 0xFFFF0000; |
| sInShifted = FALSE; |
| |
| for(;;) { |
| UCOL_GETNEXTCE(tOrder, coll, tColl, &status); |
| /*tOrder = ucol_getNextCE(coll, &tColl, &status);*/ |
| if(tOrder == UCOL_NO_MORE_CES) { |
| *(tCEs++) = tOrder; |
| break; |
| } else if((tOrder & 0xFFFFFFBF) == 0) { |
| continue; |
| } else if(isContinuation(tOrder)) { |
| if((tOrder & 0xFFFF0000) > 0) { /* There is primary value */ |
| if(tInShifted) { |
| tOrder &= 0xFFFF0000; |
| *(tCEs++) = tOrder; |
| continue; |
| } else { |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| break; |
| } |
| } else { /* Just lower level values */ |
| if(tInShifted) { |
| continue; |
| } else { |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| continue; |
| } |
| } |
| } else { /* regular */ |
| if(tOrder > LVT) { |
| *(tCEs++) = tOrder; |
| break; |
| } else { |
| if((tOrder & 0xFFFF0000) > 0) { |
| tInShifted = TRUE; |
| tOrder &= 0xFFFF0000; |
| *(tCEs++) = tOrder; |
| continue; |
| } else { |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| continue; |
| } |
| } |
| } |
| } |
| tOrder &= 0xFFFF0000; |
| tInShifted = FALSE; |
| |
| if(sOrder == tOrder) { |
| if(sOrder == 0x00010000) { |
| break; |
| } else { |
| sOrder = 0; tOrder = 0; |
| continue; |
| } |
| } else if(sOrder < tOrder) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } /* no primary difference... do the rest from the buffers */ |
| } |
| |
| /* now, we're gonna reexamine collected CEs */ |
| sCEend = sCEs; |
| tCEend = tCEs; |
| |
| /* This is the secondary level of comparison */ |
| if(checkSecTer) { |
| if(!isFrenchSec) { /* normal */ |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while (secS == 0) { |
| secS = *(sCEs++) & 0xFF00; |
| } |
| |
| while(secT == 0) { |
| secT = *(tCEs++) & 0xFF00; |
| } |
| |
| if(secS == secT) { |
| if(secS == 0x0100) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } else { /* do the French */ |
| uint32_t *sCESave = NULL; |
| uint32_t *tCESave = NULL; |
| sCEs = sCEend-2; /* this could also be sCEs-- if needs to be optimized */ |
| tCEs = tCEend-2; |
| for(;;) { |
| while (secS == 0 && sCEs >= sCEsArray) { |
| if(sCESave == 0) { |
| secS = *(sCEs--) & 0xFF80; |
| if(isContinuation(secS)) { |
| while(isContinuation(secS = *(sCEs--) & 0xFF80)); |
| /* after this, secS has the start of continuation, and sCEs points before that */ |
| sCESave = sCEs; /* we save it, so that we know where to come back AND that we need to go forward */ |
| sCEs+=2; /* need to point to the first continuation CP */ |
| /* However, now you can just continue doing stuff */ |
| } |
| } else { |
| secS = *(sCEs++) & 0xFF80; |
| if(!isContinuation(secS)) { /* This means we have finished with this cont */ |
| sCEs = sCESave; /* reset the pointer to before continuation */ |
| sCESave = 0; |
| continue; |
| } |
| } |
| secS &= 0xFF00; /* remove the continuation bit */ |
| } |
| |
| while(secT == 0 && tCEs >= tCEsArray) { |
| if(tCESave == 0) { |
| secT = *(tCEs--) & 0xFF80; |
| if(isContinuation(secT)) { |
| while(isContinuation(secT = *(tCEs--) & 0xFF80)); |
| /* after this, secS has the start of continuation, and sCEs points before that */ |
| tCESave = tCEs; /* we save it, so that we know where to come back AND that we need to go forward */ |
| tCEs+=2; /* need to point to the first continuation CP */ |
| /* However, now you can just continue doing stuff */ |
| } |
| } else { |
| secT = *(tCEs++) & 0xFF80; |
| if(!isContinuation(secT)) { /* This means we have finished with this cont */ |
| tCEs = tCESave; /* reset the pointer to before continuation */ |
| tCESave = 0; |
| continue; |
| } |
| } |
| secT &= 0xFF00; /* remove the continuation bit */ |
| } |
| |
| if(secS == secT) { |
| if(secS == 0x0100 || (sCEs < sCEsArray && tCEs < tCEsArray)) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } |
| } |
| |
| /* doing the case bit */ |
| if(checkCase) { |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while((secS & UCOL_REMOVE_CASE) == 0) { |
| if(!isContinuation(*sCEs++)) { |
| secS =*(sCEs-1) & UCOL_TERT_CASE_MASK; |
| } |
| } |
| |
| while((secT & UCOL_REMOVE_CASE) == 0) { |
| if(!isContinuation(*tCEs++)) { |
| secT = *(tCEs-1) & UCOL_TERT_CASE_MASK; |
| } |
| } |
| |
| if((secS & UCOL_CASE_BIT_MASK) < (secT & UCOL_CASE_BIT_MASK)) { |
| return UCOL_LESS; |
| } else if((secS & UCOL_CASE_BIT_MASK) > (secT & UCOL_CASE_BIT_MASK)) { |
| return UCOL_GREATER; |
| } |
| |
| if((secS & UCOL_REMOVE_CASE) == 0x01 || (secT & UCOL_REMOVE_CASE) == 0x01 ) { |
| break; |
| } else { |
| secS = 0; |
| secT = 0; |
| } |
| } |
| } |
| |
| /* Tertiary level */ |
| if(checkTertiary) { |
| secS = 0; |
| secT = 0; |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while((secS & UCOL_REMOVE_CASE) == 0) { |
| secS = *(sCEs++) & tertiaryMask; |
| } |
| |
| while((secT & UCOL_REMOVE_CASE) == 0) { |
| secT = *(tCEs++) & tertiaryMask; |
| } |
| |
| if(secS == secT) { |
| if((secS & UCOL_REMOVE_CASE) == 1) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } |
| |
| |
| if(qShifted) { |
| UBool sInShifted = TRUE; |
| UBool tInShifted = TRUE; |
| secS = 0; |
| secT = 0; |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while(secS == 0 && secS != 0x00010101 || (isContinuation(secS) && !sInShifted)) { |
| secS = *(sCEs++); |
| if(isContinuation(secS) && !sInShifted) { |
| continue; |
| } |
| if(secS > LVT || (secS & 0xFFFF0000) == 0) { |
| secS = 0xFFFF0000; |
| sInShifted = FALSE; |
| } else { |
| sInShifted = TRUE; |
| } |
| } |
| secS &= 0xFFFF0000; |
| |
| |
| while(secT == 0 && secT != 0x00010101 || (isContinuation(secT) && !tInShifted)) { |
| secT = *(tCEs++); |
| if(isContinuation(secT) && !tInShifted) { |
| continue; |
| } |
| if(secT > LVT || (secT & 0xFFFF0000) == 0) { |
| secT = 0xFFFF0000; |
| tInShifted = FALSE; |
| } else { |
| tInShifted = TRUE; |
| } |
| } |
| secT &= 0xFFFF0000; |
| |
| if(secS == secT) { |
| if(secS == 0x00010000) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } |
| |
| /* 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) */ |
| if(checkIdent) |
| { |
| int8_t comparison; |
| uint32_t sLen = sColl.len-sColl.string; |
| uint32_t tLen = tColl.len-tColl.string; |
| uint32_t compLen = 0; |
| |
| if(sLen > tLen) { |
| compLen = tLen; |
| } else { |
| compLen = sLen; |
| } |
| |
| comparison = u_strncmp(sColl.string, tColl.string, compLen); |
| |
| if (comparison < 0) |
| { |
| result = UCOL_LESS; |
| } |
| else if (comparison == 0) |
| { |
| if(sLen > tLen) { |
| result = UCOL_GREATER; |
| } else if(sLen < tLen) { |
| result = UCOL_LESS; |
| } else { |
| result = UCOL_EQUAL; |
| } |
| } |
| else |
| { |
| result = UCOL_GREATER; |
| } |
| } |
| |
| return result; |
| } |
| |
| |
| void init_incrementalContext(const UCollator *coll, 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; |
| s->coll = coll; |
| } |
| |
| /* This is the incremental function */ |
| U_CAPI UCollationResult ucol_strcollinc(const UCollator *coll, |
| UCharForwardIterator *source, void *sourceContext, |
| UCharForwardIterator *target, void *targetContext) |
| { |
| incrementalContext sColl, tColl; |
| |
| init_incrementalContext(coll, source, sourceContext, &sColl); |
| init_incrementalContext(coll, target, targetContext, &tColl); |
| |
| #if 0 |
| /* This is Andy's fast preparatory scan */ |
| /* It's good to have it - once the regular function is working */ |
| /* Scan the strings. Find: */ |
| /* their length, if not given by caller */ |
| /* The length of any leading portion that is equal */ |
| /* Whether they are exactly equal. (in which case we just return */ |
| const UChar *pSrc = source; |
| const UChar *pTarg = target; |
| |
| const UChar *pSrcEnd = source + sourceLength; |
| const UChar *pTargEnd = target + targetLength; |
| |
| int32_t equalLength = 0; |
| |
| // Scan while the strings are bitwise ==, or until one is exhausted. |
| for (;;) { |
| if (pSrc == pSrcEnd || pTarg == pTargEnd) |
| break; |
| if (*pSrc != *pTarg) |
| break; |
| if (*pSrc == 0 && (sourceLength == -1 || targetLength == -1)) |
| break; |
| equalLength++; |
| pSrc++; |
| pTarg++; |
| } |
| |
| // If we made it all the way through both strings, we are done. They are == |
| if ((pSrc ==pSrcEnd || (pSrcEnd <pSrc && *pSrc==0)) && /* At end of src string, however it was specified. */ |
| (pTarg==pTargEnd || (pTargEnd<pTarg && *pTarg==0))) /* and also at end of dest string */ |
| return UCOL_EQUAL; |
| |
| // If we don't know the length of the src string, continue scanning it to get the length.. |
| if (sourceLength == -1) { |
| while (*pSrc != 0 ) { |
| pSrc++; |
| } |
| sourceLength = pSrc - source; |
| } |
| |
| // If we don't know the length of the targ string, continue scanning it to get the length.. |
| if (targetLength == -1) { |
| while (*pTarg != 0 ) { |
| pTarg++; |
| } |
| targetLength = pTarg - target; |
| } |
| |
| |
| if (equalLength > 2) { |
| /* There is an identical portion at the beginning of the two strings. */ |
| /* If the identical portion ends within a contraction or a comibining */ |
| /* character sequence, back up to the start of that sequence. */ |
| pSrc = source + equalLength; /* point to the first differing chars */ |
| pTarg = target + equalLength; |
| if (pSrc != source+sourceLength && ucol_unsafeCP(*pSrc, coll) || |
| pTarg != target+targetLength && ucol_unsafeCP(*pTarg, coll)) |
| { |
| // We are stopped in the middle of a contraction. |
| // Scan backwards through the == part of the string looking for the start of the contraction. |
| // It doesn't matter which string we scan, since they are the same in this region. |
| do |
| { |
| equalLength--; |
| pSrc--; |
| } |
| while (equalLength>0 && ucol_unsafeCP(*pSrc, coll)); |
| } |
| |
| source += equalLength; |
| target += equalLength; |
| sourceLength -= equalLength; |
| targetLength -= equalLength; |
| } |
| |
| #endif |
| |
| UCollationResult result = UCOL_EQUAL; |
| UErrorCode status = U_ZERO_ERROR; |
| |
| if(coll->normalizationMode != UCOL_OFF) { /* run away screaming!!!! */ |
| return alternateIncrementalProcessing(coll, &sColl, &tColl); |
| } |
| |
| UColAttributeValue strength = coll->strength; |
| UBool initialCheckSecTer = (strength >= UCOL_SECONDARY); |
| |
| UBool checkSecTer = initialCheckSecTer; |
| UBool checkTertiary = (strength >= UCOL_TERTIARY); |
| UBool checkQuad = (strength >= UCOL_QUATERNARY); |
| UBool checkIdent = (strength == UCOL_IDENTICAL); |
| UBool checkCase = (coll->caseLevel == UCOL_ON); |
| UBool isFrenchSec = (coll->frenchCollation == UCOL_ON) && checkSecTer; |
| UBool shifted = (coll->alternateHandling == UCOL_SHIFTED); |
| UBool qShifted = shifted && checkQuad; |
| |
| uint32_t sCEsArray[512], tCEsArray[512]; |
| uint32_t *sCEs = sCEsArray, *tCEs = tCEsArray; |
| uint32_t *sCEend = sCEs+512, *tCEend = tCEs+512; |
| uint8_t caseSwitch = coll->caseSwitch; |
| uint8_t tertiaryMask = coll->tertiaryMask; |
| |
| uint32_t LVT = (shifted)?((coll->variableMax1)<<24 | (coll->variableMax2)<<16):0; |
| |
| uint32_t secS = 0, secT = 0; |
| |
| uint32_t sOrder=0, tOrder=0; |
| if(!shifted) { |
| for(;;) { |
| if(sCEs == sCEend || tCEs == tCEend) { |
| return alternateIncrementalProcessing(coll, &sColl, &tColl); |
| } |
| |
| /* Get the next collation element in each of the strings, unless */ |
| /* we've been requested to skip it. */ |
| while(sOrder == 0) { |
| sOrder = ucol_getIncrementalCE(coll, &sColl, &status); |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| sOrder &= 0xFFFF0000; |
| } |
| |
| while(tOrder == 0) { |
| tOrder = ucol_getIncrementalCE(coll, &tColl, &status); |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| tOrder &= 0xFFFF0000; |
| } |
| |
| if((sOrder == (UCOL_NO_MORE_CES & UCOL_PRIMARYORDERMASK) && sColl.panic == TRUE) || |
| (tOrder == (UCOL_NO_MORE_CES & UCOL_PRIMARYORDERMASK) && tColl.panic == TRUE)) { |
| return alternateIncrementalProcessing(coll, &sColl, &tColl); |
| } |
| |
| if(sOrder == tOrder) { |
| if(sOrder == (UCOL_NO_MORE_CES & UCOL_PRIMARYORDERMASK)) { |
| |
| break; |
| } else { |
| sOrder = 0; tOrder = 0; |
| continue; |
| } |
| } else if(sOrder < tOrder) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } /* no primary difference... do the rest from the buffers */ |
| } else { /* shifted - do a slightly more complicated processing */ |
| for(;;) { |
| UBool sInShifted = FALSE; |
| UBool tInShifted = FALSE; |
| |
| if(sCEs == sCEend || tCEs == tCEend) { |
| return alternateIncrementalProcessing(coll, &sColl, &tColl); |
| } |
| |
| /* This is where abridged version for shifted should go */ |
| for(;;) { |
| sOrder = ucol_getIncrementalCE(coll, &sColl, &status); |
| if(sOrder == UCOL_NO_MORE_CES) { |
| if(sColl.panic == TRUE) { |
| return alternateIncrementalProcessing(coll, &sColl, &tColl); |
| } |
| *(sCEs++) = sOrder; |
| break; |
| } else if((sOrder & 0xFFFFFFBF) == 0) { |
| continue; |
| } else if(isContinuation(sOrder)) { |
| if((sOrder & 0xFFFF0000) > 0) { /* There is primary value */ |
| if(sInShifted) { |
| sOrder &= 0xFFFF0000; |
| *(sCEs++) = sOrder; |
| continue; |
| } else { |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| break; |
| } |
| } else { /* Just lower level values */ |
| if(sInShifted) { |
| continue; |
| } else { |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| continue; |
| } |
| } |
| } else { /* regular */ |
| if(sOrder > LVT) { |
| *(sCEs++) = sOrder; |
| break; |
| } else { |
| if((sOrder & 0xFFFF0000) > 0) { |
| sInShifted = TRUE; |
| sOrder &= 0xFFFF0000; |
| *(sCEs++) = sOrder; |
| continue; |
| } else { |
| sOrder ^= caseSwitch; |
| *(sCEs++) = sOrder; |
| continue; |
| } |
| } |
| } |
| } |
| sOrder &= 0xFFFF0000; |
| sInShifted = FALSE; |
| |
| for(;;) { |
| tOrder = ucol_getIncrementalCE(coll, &tColl, &status); |
| if(tOrder == UCOL_NO_MORE_CES) { |
| if(tColl.panic == TRUE) { |
| return alternateIncrementalProcessing(coll, &sColl, &tColl); |
| } |
| *(tCEs++) = tOrder; |
| break; |
| } else if((tOrder & 0xFFFFFFBF) == 0) { |
| continue; |
| } else if(isContinuation(tOrder)) { |
| if((tOrder & 0xFFFF0000) > 0) { /* There is primary value */ |
| if(tInShifted) { |
| tOrder &= 0xFFFF0000; |
| *(tCEs++) = tOrder; |
| continue; |
| } else { |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| break; |
| } |
| } else { /* Just lower level values */ |
| if(tInShifted) { |
| continue; |
| } else { |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| continue; |
| } |
| } |
| } else { /* regular */ |
| if(tOrder > LVT) { |
| *(tCEs++) = tOrder; |
| break; |
| } else { |
| if((tOrder & 0xFFFF0000) > 0) { |
| tInShifted = TRUE; |
| tOrder &= 0xFFFF0000; |
| *(tCEs++) = tOrder; |
| continue; |
| } else { |
| tOrder ^= caseSwitch; |
| *(tCEs++) = tOrder; |
| continue; |
| } |
| } |
| } |
| } |
| tOrder &= 0xFFFF0000; |
| tInShifted = FALSE; |
| |
| if(sOrder == tOrder) { |
| if(sOrder == 0x00010000) { |
| break; |
| } else { |
| sOrder = 0; tOrder = 0; |
| continue; |
| } |
| } else if(sOrder < tOrder) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } /* no primary difference... do the rest from the buffers */ |
| } |
| |
| /* now, we're gonna reexamine collected CEs */ |
| sCEend = sCEs; |
| tCEend = tCEs; |
| |
| /* This is the secondary level of comparison */ |
| if(checkSecTer) { |
| if(!isFrenchSec) { /* normal */ |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while (secS == 0) { |
| secS = *(sCEs++) & 0xFF00; |
| } |
| |
| while(secT == 0) { |
| secT = *(tCEs++) & 0xFF00; |
| } |
| |
| if(secS == secT) { |
| if(secS == 0x0100) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } else { /* do the French */ |
| uint32_t *sCESave = NULL; |
| uint32_t *tCESave = NULL; |
| sCEs = sCEend-2; /* this could also be sCEs-- if needs to be optimized */ |
| tCEs = tCEend-2; |
| for(;;) { |
| while (secS == 0 && sCEs >= sCEsArray) { |
| if(sCESave == 0) { |
| secS = *(sCEs--) & 0xFF80; |
| if(isContinuation(secS)) { |
| while(isContinuation(secS = *(sCEs--) & 0xFF80)); |
| /* after this, secS has the start of continuation, and sCEs points before that */ |
| sCESave = sCEs; /* we save it, so that we know where to come back AND that we need to go forward */ |
| sCEs+=2; /* need to point to the first continuation CP */ |
| /* However, now you can just continue doing stuff */ |
| } |
| } else { |
| secS = *(sCEs++) & 0xFF80; |
| if(!isContinuation(secS)) { /* This means we have finished with this cont */ |
| sCEs = sCESave; /* reset the pointer to before continuation */ |
| sCESave = 0; |
| continue; |
| } |
| } |
| secS &= 0xFF00; /* remove the continuation bit */ |
| } |
| |
| while(secT == 0 && tCEs >= tCEsArray) { |
| if(tCESave == 0) { |
| secT = *(tCEs--) & 0xFF80; |
| if(isContinuation(secT)) { |
| while(isContinuation(secT = *(tCEs--) & 0xFF80)); |
| /* after this, secS has the start of continuation, and sCEs points before that */ |
| tCESave = tCEs; /* we save it, so that we know where to come back AND that we need to go forward */ |
| tCEs+=2; /* need to point to the first continuation CP */ |
| /* However, now you can just continue doing stuff */ |
| } |
| } else { |
| secT = *(tCEs++) & 0xFF80; |
| if(!isContinuation(secT)) { /* This means we have finished with this cont */ |
| tCEs = tCESave; /* reset the pointer to before continuation */ |
| tCESave = 0; |
| continue; |
| } |
| } |
| secT &= 0xFF00; /* remove the continuation bit */ |
| } |
| |
| if(secS == secT) { |
| if(secS == 0x0100 || (sCEs < sCEsArray && tCEs < tCEsArray)) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } |
| } |
| |
| /* doing the case bit */ |
| if(checkCase) { |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while((secS & UCOL_REMOVE_CASE) == 0) { |
| if(!isContinuation(*sCEs++)) { |
| secS =*(sCEs-1) & UCOL_TERT_CASE_MASK; |
| } |
| } |
| |
| while((secT & UCOL_REMOVE_CASE) == 0) { |
| if(!isContinuation(*tCEs++)) { |
| secT = *(tCEs-1) & UCOL_TERT_CASE_MASK; |
| } |
| } |
| |
| if((secS & UCOL_CASE_BIT_MASK) < (secT & UCOL_CASE_BIT_MASK)) { |
| return UCOL_LESS; |
| } else if((secS & UCOL_CASE_BIT_MASK) > (secT & UCOL_CASE_BIT_MASK)) { |
| return UCOL_GREATER; |
| } |
| |
| if((secS & UCOL_REMOVE_CASE) == 0x01 || (secT & UCOL_REMOVE_CASE) == 0x01 ) { |
| break; |
| } else { |
| secS = 0; |
| secT = 0; |
| } |
| } |
| } |
| |
| /* Tertiary level */ |
| if(checkTertiary) { |
| secS = 0; |
| secT = 0; |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while((secS & UCOL_REMOVE_CASE) == 0) { |
| secS = *(sCEs++) & tertiaryMask; |
| } |
| |
| while((secT & UCOL_REMOVE_CASE) == 0) { |
| secT = *(tCEs++) & tertiaryMask; |
| } |
| |
| if(secS == secT) { |
| if((secS & UCOL_REMOVE_CASE) == 1) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } |
| |
| |
| if(qShifted) { |
| UBool sInShifted = TRUE; |
| UBool tInShifted = TRUE; |
| secS = 0; |
| secT = 0; |
| sCEs = sCEsArray; |
| tCEs = tCEsArray; |
| for(;;) { |
| while(secS == 0 && secS != 0x00010101 || (isContinuation(secS) && !sInShifted)) { |
| secS = *(sCEs++); |
| if(isContinuation(secS) && !sInShifted) { |
| continue; |
| } |
| if(secS > LVT || (secS & 0xFFFF0000) == 0) { |
| secS = 0xFFFF0000; |
| sInShifted = FALSE; |
| } else { |
| sInShifted = TRUE; |
| } |
| } |
| secS &= 0xFFFF0000; |
| |
| |
| while(secT == 0 && secT != 0x00010101 || (isContinuation(secT) && !tInShifted)) { |
| secT = *(tCEs++); |
| if(isContinuation(secT) && !tInShifted) { |
| continue; |
| } |
| if(secT > LVT || (secT & 0xFFFF0000) == 0) { |
| secT = 0xFFFF0000; |
| tInShifted = FALSE; |
| } else { |
| tInShifted = TRUE; |
| } |
| } |
| secT &= 0xFFFF0000; |
| |
| if(secS == secT) { |
| if(secS == 0x00010000) { |
| break; |
| } else { |
| secS = 0; secT = 0; |
| continue; |
| } |
| } else if(secS < secT) { |
| return UCOL_LESS; |
| } else { |
| return UCOL_GREATER; |
| } |
| } |
| } |
| |
| /* 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) */ |
| if(checkIdent) |
| { |
| UnicodeString sourceDecomp, targetDecomp; |
| |
| int8_t comparison; |
| |
| /* synwee : implemented in c++ since normalizer is implemented there */ |
| Normalizer::EMode mode = Normalizer::getNormalizerEMode( |
| ucol_getNormalization(coll), status); |
| |
| Normalizer::normalize(UnicodeString(sColl.stringP, sColl.len-sColl.stringP-1), |
| mode, 0, sourceDecomp, status); |
| |
| Normalizer::normalize(UnicodeString(tColl.stringP, tColl.len-tColl.stringP-1), |
| mode, 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; |
| } |
| |
| /* convenience function for comparing strings */ |
| 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); |
| } |
| |
| /* convenience function for comparing strings */ |
| 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); |
| } |
| |
| /* convenience function for comparing strings */ |
| 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); |
| } |
| |
| |
| int32_t ucol_getIncrementalCE(const UCollator *coll, incrementalContext *ctx, UErrorCode *status) { |
| uint32_t order; |
| if (ctx->CEpos > ctx->toReturn) { /* Are there any CEs from previous expansions? */ |
| order = *(ctx->toReturn++); /* if so, return them */ |
| if(ctx->CEpos == ctx->toReturn) { |
| ctx->CEpos = ctx->toReturn = ctx->CEs; |
| } |
| } else { /* This is the real business now */ |
| if(ctx->lastChar == 0xFFFF) { |
| ctx->currentChar = ctx->source(ctx->sourceContext); |
| incctx_appendChar(ctx, ctx->currentChar); |
| if(ctx->currentChar == 0xFFFF) { |
| return UCOL_NO_MORE_CES; |
| } |
| } else { |
| ctx->currentChar = ctx->lastChar; |
| ctx->lastChar = 0xFFFF; |
| } |
| |
| UChar ch = ctx->currentChar; |
| 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 */ |
| order = ucol_getIncrementalSpecialCE(coll, order, ctx, 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_getIncrementalUCA(ch, ctx, status); |
| } |
| } |
| } |
| /* This means that contraction should spit back the last codepoint eaten! */ |
| return order; /* return the CE */ |
| } |
| |
| /* This function tries to get a CE from UCA, which should be always around */ |
| /* UChar is passed in in order to speed things up */ |
| /* here is also the generation of implicit CEs */ |
| uint32_t ucol_getIncrementalUCA(UChar ch, incrementalContext *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 = ucol_getIncrementalSpecialCE(UCA, order, 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 */ |
| const uint32_t |
| SBase = 0xAC00, LBase = 0x1100, VBase = 0x1161, TBase = 0x11A7, |
| LCount = 19, VCount = 21, TCount = 28, |
| NCount = VCount * TCount, // 588 |
| SCount = LCount * NCount; // 11172 |
| //LLimit = LBase + LCount, // 1113 |
| //VLimit = VBase + VCount, // 1176 |
| //TLimit = TBase + TCount, // 11C3 |
| //SLimit = SBase + SCount; // D7A4 |
| |
| // once we have failed to find a match for codepoint cp, and are in the implicit code. |
| |
| uint32_t L = ch - SBase; |
| //if (ch < SLimit) { // since it is unsigned, catchs zero case too |
| if (L < SCount) { // since it is unsigned, catchs zero case too |
| |
| // divide into pieces |
| |
| uint32_t T = L % TCount; // we do it in this order since some compilers can do % and / in one operation |
| L /= TCount; |
| uint32_t V = L % VCount; |
| L /= VCount; |
| |
| // offset them |
| |
| L += LBase; |
| V += VBase; |
| T += TBase; |
| |
| // return the first CE, but first put the rest into the expansion buffer |
| if (!collationSource->coll->image->jamoSpecial) { // FAST PATH |
| |
| *(collationSource->CEpos++) = ucmp32_get(UCA->mapping, V); |
| if (T != TBase) { |
| *(collationSource->CEpos++) = ucmp32_get(UCA->mapping, T); |
| } |
| |
| return ucmp32_get(UCA->mapping, L); // return first one |
| |
| } else { // Jamo is Special |
| collIterate jamos; |
| UChar jamoString[3]; |
| uint32_t CE = UCOL_NOT_FOUND; |
| const UCollator *collator = collationSource->coll; |
| jamoString[0] = L; |
| jamoString[1] = V; |
| if (T != TBase) { |
| jamoString[2] = T; |
| init_collIterate(collator, jamoString, 3, &jamos, TRUE); |
| } else { |
| init_collIterate(collator, jamoString, 2, &jamos, TRUE); |
| } |
| |
| CE = ucol_getNextCE(collator, &jamos, status); |
| |
| while(CE != UCOL_NO_MORE_CES) { |
| *(collationSource->CEpos++) = CE; |
| CE = ucol_getNextCE(collator, &jamos, status); |
| } |
| return *(collationSource->toReturn++); |
| |
| /* |
| ucol_getJamoCEs(collationSource->coll, L, &collationSource->CEpos); |
| ucol_getJamoCEs(collationSource->coll, V, &collationSource->CEpos); |
| if (T != TBase) { |
| ucol_getJamoCEs(collationSource->coll, T, &collationSource->CEpos); |
| } |
| return *(collationSource->toReturn++); |
| */ |
| |
| /* |
| // do recursive processing of L, V, and T with fetchCE (but T only if not equal to TBase!!) |
| // Since fetchCE returns a CE, and (potentially) stuffs items into the ce buffer, |
| // this is how it is done. |
| |
| int firstCE = fetchCE(L, ...); |
| int* lastExpansion = expansionBufferEnd++; // set pointer, leave gap! |
| *lastExpansion = fetchCE(V,...); |
| if (T != TBase) { |
| lastExpansion = expansionBufferEnd++; // set pointer, leave gap! |
| *lastExpansion = fetchCE(T,...); |
| } |
| */ |
| } |
| } |
| |
| collationSource->lastChar = collationSource->source(collationSource->sourceContext); |
| incctx_appendChar(collationSource, collationSource->lastChar); |
| |
| if(UTF_IS_FIRST_SURROGATE(ch)) { |
| if( (collationSource->lastChar != 0xFFFF) && |
| UTF_IS_SECOND_SURROGATE((collationSource->lastChar))) { |
| uint32_t cp = (((ch)<<10UL)+(collationSource->lastChar)-((0xd800<<10UL)+0xdc00)); |
| collationSource->lastChar = 0xFFFF; /*used up*/ |
| if ((cp & 0xFFFE) == 0xFFFE || (0xD800 <= cp && cp <= 0xDC00)) { |
| return 0; /* illegal code value, use completely ignoreable! */ |
| } |
| /* This is a code point minus 0x10000, that's what algorithm requires */ |
| order = 0xE0010303 | (cp & 0xFFE00) << 8; |
| |
| *(collationSource->CEpos++) = 0x80200080 | (cp & 0x001FF) << 22; |
| } else { |
| return 0; /* completely ignorable */ |
| } |
| } else { |
| /* otherwise */ |
| if(UTF_IS_SECOND_SURROGATE((ch)) || (ch & 0xFFFE) == 0xFFFE) { |
| return 0; /* completely ignorable */ |
| } |
| /* Make up an artifical CE from code point as per UCA */ |
| order = 0xD0800303 | (ch & 0xF000) << 12 | (ch & 0x0FE0) << 11; |
| *(collationSource->CEpos++) = 0x04000080 | (ch & 0x001F) << 27; |
| } |
| } |
| return order; /* return the CE */ |
| } |
| |
| |
| int32_t ucol_getIncrementalSpecialCE(const UCollator *coll, uint32_t CE, incrementalContext *source, UErrorCode *status) { |
| uint32_t i = 0; /* general counter */ |
| |
| if(U_FAILURE(*status)) return -1; |
| |
| for(;;) { |
| 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 */ |
| source->panic = TRUE; |
| return UCOL_NO_MORE_CES; |
| break; |
| case CONTRACTION_TAG: |
| /* This should handle contractions */ |
| for(;;) { |
| /* 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++; */ |
| schar = source->lastChar = source->source(source->sourceContext); |
| incctx_appendChar(source, source->lastChar); |
| if (schar == 0xFFFF) { /* 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; |
| } |
| UCharOffset++; /* skip the backward offset, see above */ |
| //! 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 */ |
| } else { |
| source->lastChar = 0xFFFF; |
| } |
| CE = *(coll->contractionCEs + (UCharOffset - coll->contractionIndex)); |
| /* |
| if(!isContraction(CE)) { |
| break; |
| } |
| */ |
| if(isContraction(CE)) { /* fix for the bug. Other places need to be checked */ |
| /* this is contraction, and we will continue. However, we can fail along the */ |
| /* th road, which means that we have part of contraction correct */ |
| source->panic = TRUE; |
| return UCOL_NO_MORE_CES; |
| } else { |
| 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; |
| |
| } |
| |
| 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; |
| } |