1 | // © 2016 and later: Unicode, Inc. and others. |
2 | // License & terms of use: http://www.unicode.org/copyright.html |
3 | /* |
4 | ******************************************************************************* |
5 | * Copyright (C) 2013-2014, International Business Machines |
6 | * Corporation and others. All Rights Reserved. |
7 | ******************************************************************************* |
8 | * collationbuilder.cpp |
9 | * |
10 | * (replaced the former ucol_bld.cpp) |
11 | * |
12 | * created on: 2013may06 |
13 | * created by: Markus W. Scherer |
14 | */ |
15 | |
16 | #ifdef DEBUG_COLLATION_BUILDER |
17 | #include <stdio.h> |
18 | #endif |
19 | |
20 | #include "unicode/utypes.h" |
21 | |
22 | #if !UCONFIG_NO_COLLATION |
23 | |
24 | #include "unicode/caniter.h" |
25 | #include "unicode/normalizer2.h" |
26 | #include "unicode/tblcoll.h" |
27 | #include "unicode/parseerr.h" |
28 | #include "unicode/uchar.h" |
29 | #include "unicode/ucol.h" |
30 | #include "unicode/unistr.h" |
31 | #include "unicode/usetiter.h" |
32 | #include "unicode/utf16.h" |
33 | #include "unicode/uversion.h" |
34 | #include "cmemory.h" |
35 | #include "collation.h" |
36 | #include "collationbuilder.h" |
37 | #include "collationdata.h" |
38 | #include "collationdatabuilder.h" |
39 | #include "collationfastlatin.h" |
40 | #include "collationroot.h" |
41 | #include "collationrootelements.h" |
42 | #include "collationruleparser.h" |
43 | #include "collationsettings.h" |
44 | #include "collationtailoring.h" |
45 | #include "collationweights.h" |
46 | #include "normalizer2impl.h" |
47 | #include "uassert.h" |
48 | #include "ucol_imp.h" |
49 | #include "utf16collationiterator.h" |
50 | |
51 | U_NAMESPACE_BEGIN |
52 | |
53 | namespace { |
54 | |
55 | class BundleImporter : public CollationRuleParser::Importer { |
56 | public: |
57 | BundleImporter() {} |
58 | virtual ~BundleImporter(); |
59 | virtual void getRules( |
60 | const char *localeID, const char *collationType, |
61 | UnicodeString &rules, |
62 | const char *&errorReason, UErrorCode &errorCode); |
63 | }; |
64 | |
65 | BundleImporter::~BundleImporter() {} |
66 | |
67 | void |
68 | BundleImporter::getRules( |
69 | const char *localeID, const char *collationType, |
70 | UnicodeString &rules, |
71 | const char *& /*errorReason*/, UErrorCode &errorCode) { |
72 | CollationLoader::loadRules(localeID, collationType, rules, errorCode); |
73 | } |
74 | |
75 | } // namespace |
76 | |
77 | // RuleBasedCollator implementation ---------------------------------------- *** |
78 | |
79 | // These methods are here, rather than in rulebasedcollator.cpp, |
80 | // for modularization: |
81 | // Most code using Collator does not need to build a Collator from rules. |
82 | // By moving these constructors and helper methods to a separate file, |
83 | // most code will not have a static dependency on the builder code. |
84 | |
85 | RuleBasedCollator::RuleBasedCollator() |
86 | : data(NULL), |
87 | settings(NULL), |
88 | tailoring(NULL), |
89 | cacheEntry(NULL), |
90 | validLocale("" ), |
91 | explicitlySetAttributes(0), |
92 | actualLocaleIsSameAsValid(FALSE) { |
93 | } |
94 | |
95 | RuleBasedCollator::RuleBasedCollator(const UnicodeString &rules, UErrorCode &errorCode) |
96 | : data(NULL), |
97 | settings(NULL), |
98 | tailoring(NULL), |
99 | cacheEntry(NULL), |
100 | validLocale("" ), |
101 | explicitlySetAttributes(0), |
102 | actualLocaleIsSameAsValid(FALSE) { |
103 | internalBuildTailoring(rules, UCOL_DEFAULT, UCOL_DEFAULT, NULL, NULL, errorCode); |
104 | } |
105 | |
106 | RuleBasedCollator::RuleBasedCollator(const UnicodeString &rules, ECollationStrength strength, |
107 | UErrorCode &errorCode) |
108 | : data(NULL), |
109 | settings(NULL), |
110 | tailoring(NULL), |
111 | cacheEntry(NULL), |
112 | validLocale("" ), |
113 | explicitlySetAttributes(0), |
114 | actualLocaleIsSameAsValid(FALSE) { |
115 | internalBuildTailoring(rules, strength, UCOL_DEFAULT, NULL, NULL, errorCode); |
116 | } |
117 | |
118 | RuleBasedCollator::RuleBasedCollator(const UnicodeString &rules, |
119 | UColAttributeValue decompositionMode, |
120 | UErrorCode &errorCode) |
121 | : data(NULL), |
122 | settings(NULL), |
123 | tailoring(NULL), |
124 | cacheEntry(NULL), |
125 | validLocale("" ), |
126 | explicitlySetAttributes(0), |
127 | actualLocaleIsSameAsValid(FALSE) { |
128 | internalBuildTailoring(rules, UCOL_DEFAULT, decompositionMode, NULL, NULL, errorCode); |
129 | } |
130 | |
131 | RuleBasedCollator::RuleBasedCollator(const UnicodeString &rules, |
132 | ECollationStrength strength, |
133 | UColAttributeValue decompositionMode, |
134 | UErrorCode &errorCode) |
135 | : data(NULL), |
136 | settings(NULL), |
137 | tailoring(NULL), |
138 | cacheEntry(NULL), |
139 | validLocale("" ), |
140 | explicitlySetAttributes(0), |
141 | actualLocaleIsSameAsValid(FALSE) { |
142 | internalBuildTailoring(rules, strength, decompositionMode, NULL, NULL, errorCode); |
143 | } |
144 | |
145 | RuleBasedCollator::RuleBasedCollator(const UnicodeString &rules, |
146 | UParseError &parseError, UnicodeString &reason, |
147 | UErrorCode &errorCode) |
148 | : data(NULL), |
149 | settings(NULL), |
150 | tailoring(NULL), |
151 | cacheEntry(NULL), |
152 | validLocale("" ), |
153 | explicitlySetAttributes(0), |
154 | actualLocaleIsSameAsValid(FALSE) { |
155 | internalBuildTailoring(rules, UCOL_DEFAULT, UCOL_DEFAULT, &parseError, &reason, errorCode); |
156 | } |
157 | |
158 | void |
159 | RuleBasedCollator::internalBuildTailoring(const UnicodeString &rules, |
160 | int32_t strength, |
161 | UColAttributeValue decompositionMode, |
162 | UParseError *outParseError, UnicodeString *outReason, |
163 | UErrorCode &errorCode) { |
164 | const CollationTailoring *base = CollationRoot::getRoot(errorCode); |
165 | if(U_FAILURE(errorCode)) { return; } |
166 | if(outReason != NULL) { outReason->remove(); } |
167 | CollationBuilder builder(base, errorCode); |
168 | UVersionInfo noVersion = { 0, 0, 0, 0 }; |
169 | BundleImporter importer; |
170 | LocalPointer<CollationTailoring> t(builder.parseAndBuild(rules, noVersion, |
171 | &importer, |
172 | outParseError, errorCode)); |
173 | if(U_FAILURE(errorCode)) { |
174 | const char *reason = builder.getErrorReason(); |
175 | if(reason != NULL && outReason != NULL) { |
176 | *outReason = UnicodeString(reason, -1, US_INV); |
177 | } |
178 | return; |
179 | } |
180 | t->actualLocale.setToBogus(); |
181 | adoptTailoring(t.orphan(), errorCode); |
182 | // Set attributes after building the collator, |
183 | // to keep the default settings consistent with the rule string. |
184 | if(strength != UCOL_DEFAULT) { |
185 | setAttribute(UCOL_STRENGTH, (UColAttributeValue)strength, errorCode); |
186 | } |
187 | if(decompositionMode != UCOL_DEFAULT) { |
188 | setAttribute(UCOL_NORMALIZATION_MODE, decompositionMode, errorCode); |
189 | } |
190 | } |
191 | |
192 | // CollationBuilder implementation ----------------------------------------- *** |
193 | |
194 | // Some compilers don't care if constants are defined in the .cpp file. |
195 | // MS Visual C++ does not like it, but gcc requires it. clang does not care. |
196 | #ifndef _MSC_VER |
197 | const int32_t CollationBuilder::HAS_BEFORE2; |
198 | const int32_t CollationBuilder::HAS_BEFORE3; |
199 | #endif |
200 | |
201 | CollationBuilder::CollationBuilder(const CollationTailoring *b, UErrorCode &errorCode) |
202 | : nfd(*Normalizer2::getNFDInstance(errorCode)), |
203 | fcd(*Normalizer2Factory::getFCDInstance(errorCode)), |
204 | nfcImpl(*Normalizer2Factory::getNFCImpl(errorCode)), |
205 | base(b), |
206 | baseData(b->data), |
207 | rootElements(b->data->rootElements, b->data->rootElementsLength), |
208 | variableTop(0), |
209 | dataBuilder(new CollationDataBuilder(errorCode)), fastLatinEnabled(TRUE), |
210 | errorReason(NULL), |
211 | cesLength(0), |
212 | rootPrimaryIndexes(errorCode), nodes(errorCode) { |
213 | nfcImpl.ensureCanonIterData(errorCode); |
214 | if(U_FAILURE(errorCode)) { |
215 | errorReason = "CollationBuilder fields initialization failed" ; |
216 | return; |
217 | } |
218 | if(dataBuilder == NULL) { |
219 | errorCode = U_MEMORY_ALLOCATION_ERROR; |
220 | return; |
221 | } |
222 | dataBuilder->initForTailoring(baseData, errorCode); |
223 | if(U_FAILURE(errorCode)) { |
224 | errorReason = "CollationBuilder initialization failed" ; |
225 | } |
226 | } |
227 | |
228 | CollationBuilder::~CollationBuilder() { |
229 | delete dataBuilder; |
230 | } |
231 | |
232 | CollationTailoring * |
233 | CollationBuilder::parseAndBuild(const UnicodeString &ruleString, |
234 | const UVersionInfo rulesVersion, |
235 | CollationRuleParser::Importer *importer, |
236 | UParseError *outParseError, |
237 | UErrorCode &errorCode) { |
238 | if(U_FAILURE(errorCode)) { return NULL; } |
239 | if(baseData->rootElements == NULL) { |
240 | errorCode = U_MISSING_RESOURCE_ERROR; |
241 | errorReason = "missing root elements data, tailoring not supported" ; |
242 | return NULL; |
243 | } |
244 | LocalPointer<CollationTailoring> tailoring(new CollationTailoring(base->settings)); |
245 | if(tailoring.isNull() || tailoring->isBogus()) { |
246 | errorCode = U_MEMORY_ALLOCATION_ERROR; |
247 | return NULL; |
248 | } |
249 | CollationRuleParser parser(baseData, errorCode); |
250 | if(U_FAILURE(errorCode)) { return NULL; } |
251 | // Note: This always bases &[last variable] and &[first regular] |
252 | // on the root collator's maxVariable/variableTop. |
253 | // If we wanted this to change after [maxVariable x], then we would keep |
254 | // the tailoring.settings pointer here and read its variableTop when we need it. |
255 | // See http://unicode.org/cldr/trac/ticket/6070 |
256 | variableTop = base->settings->variableTop; |
257 | parser.setSink(this); |
258 | parser.setImporter(importer); |
259 | CollationSettings &ownedSettings = *SharedObject::copyOnWrite(tailoring->settings); |
260 | parser.parse(ruleString, ownedSettings, outParseError, errorCode); |
261 | errorReason = parser.getErrorReason(); |
262 | if(U_FAILURE(errorCode)) { return NULL; } |
263 | if(dataBuilder->hasMappings()) { |
264 | makeTailoredCEs(errorCode); |
265 | closeOverComposites(errorCode); |
266 | finalizeCEs(errorCode); |
267 | // Copy all of ASCII, and Latin-1 letters, into each tailoring. |
268 | optimizeSet.add(0, 0x7f); |
269 | optimizeSet.add(0xc0, 0xff); |
270 | // Hangul is decomposed on the fly during collation, |
271 | // and the tailoring data is always built with HANGUL_TAG specials. |
272 | optimizeSet.remove(Hangul::HANGUL_BASE, Hangul::HANGUL_END); |
273 | dataBuilder->optimize(optimizeSet, errorCode); |
274 | tailoring->ensureOwnedData(errorCode); |
275 | if(U_FAILURE(errorCode)) { return NULL; } |
276 | if(fastLatinEnabled) { dataBuilder->enableFastLatin(); } |
277 | dataBuilder->build(*tailoring->ownedData, errorCode); |
278 | tailoring->builder = dataBuilder; |
279 | dataBuilder = NULL; |
280 | } else { |
281 | tailoring->data = baseData; |
282 | } |
283 | if(U_FAILURE(errorCode)) { return NULL; } |
284 | ownedSettings.fastLatinOptions = CollationFastLatin::getOptions( |
285 | tailoring->data, ownedSettings, |
286 | ownedSettings.fastLatinPrimaries, UPRV_LENGTHOF(ownedSettings.fastLatinPrimaries)); |
287 | tailoring->rules = ruleString; |
288 | tailoring->rules.getTerminatedBuffer(); // ensure NUL-termination |
289 | tailoring->setVersion(base->version, rulesVersion); |
290 | return tailoring.orphan(); |
291 | } |
292 | |
293 | void |
294 | CollationBuilder::addReset(int32_t strength, const UnicodeString &str, |
295 | const char *&parserErrorReason, UErrorCode &errorCode) { |
296 | if(U_FAILURE(errorCode)) { return; } |
297 | U_ASSERT(!str.isEmpty()); |
298 | if(str.charAt(0) == CollationRuleParser::POS_LEAD) { |
299 | ces[0] = getSpecialResetPosition(str, parserErrorReason, errorCode); |
300 | cesLength = 1; |
301 | if(U_FAILURE(errorCode)) { return; } |
302 | U_ASSERT((ces[0] & Collation::CASE_AND_QUATERNARY_MASK) == 0); |
303 | } else { |
304 | // normal reset to a character or string |
305 | UnicodeString nfdString = nfd.normalize(str, errorCode); |
306 | if(U_FAILURE(errorCode)) { |
307 | parserErrorReason = "normalizing the reset position" ; |
308 | return; |
309 | } |
310 | cesLength = dataBuilder->getCEs(nfdString, ces, 0); |
311 | if(cesLength > Collation::MAX_EXPANSION_LENGTH) { |
312 | errorCode = U_ILLEGAL_ARGUMENT_ERROR; |
313 | parserErrorReason = "reset position maps to too many collation elements (more than 31)" ; |
314 | return; |
315 | } |
316 | } |
317 | if(strength == UCOL_IDENTICAL) { return; } // simple reset-at-position |
318 | |
319 | // &[before strength]position |
320 | U_ASSERT(UCOL_PRIMARY <= strength && strength <= UCOL_TERTIARY); |
321 | int32_t index = findOrInsertNodeForCEs(strength, parserErrorReason, errorCode); |
322 | if(U_FAILURE(errorCode)) { return; } |
323 | |
324 | int64_t node = nodes.elementAti(index); |
325 | // If the index is for a "weaker" node, |
326 | // then skip backwards over this and further "weaker" nodes. |
327 | while(strengthFromNode(node) > strength) { |
328 | index = previousIndexFromNode(node); |
329 | node = nodes.elementAti(index); |
330 | } |
331 | |
332 | // Find or insert a node whose index we will put into a temporary CE. |
333 | if(strengthFromNode(node) == strength && isTailoredNode(node)) { |
334 | // Reset to just before this same-strength tailored node. |
335 | index = previousIndexFromNode(node); |
336 | } else if(strength == UCOL_PRIMARY) { |
337 | // root primary node (has no previous index) |
338 | uint32_t p = weight32FromNode(node); |
339 | if(p == 0) { |
340 | errorCode = U_UNSUPPORTED_ERROR; |
341 | parserErrorReason = "reset primary-before ignorable not possible" ; |
342 | return; |
343 | } |
344 | if(p <= rootElements.getFirstPrimary()) { |
345 | // There is no primary gap between ignorables and the space-first-primary. |
346 | errorCode = U_UNSUPPORTED_ERROR; |
347 | parserErrorReason = "reset primary-before first non-ignorable not supported" ; |
348 | return; |
349 | } |
350 | if(p == Collation::FIRST_TRAILING_PRIMARY) { |
351 | // We do not support tailoring to an unassigned-implicit CE. |
352 | errorCode = U_UNSUPPORTED_ERROR; |
353 | parserErrorReason = "reset primary-before [first trailing] not supported" ; |
354 | return; |
355 | } |
356 | p = rootElements.getPrimaryBefore(p, baseData->isCompressiblePrimary(p)); |
357 | index = findOrInsertNodeForPrimary(p, errorCode); |
358 | // Go to the last node in this list: |
359 | // Tailor after the last node between adjacent root nodes. |
360 | for(;;) { |
361 | node = nodes.elementAti(index); |
362 | int32_t nextIndex = nextIndexFromNode(node); |
363 | if(nextIndex == 0) { break; } |
364 | index = nextIndex; |
365 | } |
366 | } else { |
367 | // &[before 2] or &[before 3] |
368 | index = findCommonNode(index, UCOL_SECONDARY); |
369 | if(strength >= UCOL_TERTIARY) { |
370 | index = findCommonNode(index, UCOL_TERTIARY); |
371 | } |
372 | // findCommonNode() stayed on the stronger node or moved to |
373 | // an explicit common-weight node of the reset-before strength. |
374 | node = nodes.elementAti(index); |
375 | if(strengthFromNode(node) == strength) { |
376 | // Found a same-strength node with an explicit weight. |
377 | uint32_t weight16 = weight16FromNode(node); |
378 | if(weight16 == 0) { |
379 | errorCode = U_UNSUPPORTED_ERROR; |
380 | if(strength == UCOL_SECONDARY) { |
381 | parserErrorReason = "reset secondary-before secondary ignorable not possible" ; |
382 | } else { |
383 | parserErrorReason = "reset tertiary-before completely ignorable not possible" ; |
384 | } |
385 | return; |
386 | } |
387 | U_ASSERT(weight16 > Collation::BEFORE_WEIGHT16); |
388 | // Reset to just before this node. |
389 | // Insert the preceding same-level explicit weight if it is not there already. |
390 | // Which explicit weight immediately precedes this one? |
391 | weight16 = getWeight16Before(index, node, strength); |
392 | // Does this preceding weight have a node? |
393 | uint32_t previousWeight16; |
394 | int32_t previousIndex = previousIndexFromNode(node); |
395 | for(int32_t i = previousIndex;; i = previousIndexFromNode(node)) { |
396 | node = nodes.elementAti(i); |
397 | int32_t previousStrength = strengthFromNode(node); |
398 | if(previousStrength < strength) { |
399 | U_ASSERT(weight16 >= Collation::COMMON_WEIGHT16 || i == previousIndex); |
400 | // Either the reset element has an above-common weight and |
401 | // the parent node provides the implied common weight, |
402 | // or the reset element has a weight<=common in the node |
403 | // right after the parent, and we need to insert the preceding weight. |
404 | previousWeight16 = Collation::COMMON_WEIGHT16; |
405 | break; |
406 | } else if(previousStrength == strength && !isTailoredNode(node)) { |
407 | previousWeight16 = weight16FromNode(node); |
408 | break; |
409 | } |
410 | // Skip weaker nodes and same-level tailored nodes. |
411 | } |
412 | if(previousWeight16 == weight16) { |
413 | // The preceding weight has a node, |
414 | // maybe with following weaker or tailored nodes. |
415 | // Reset to the last of them. |
416 | index = previousIndex; |
417 | } else { |
418 | // Insert a node with the preceding weight, reset to that. |
419 | node = nodeFromWeight16(weight16) | nodeFromStrength(strength); |
420 | index = insertNodeBetween(previousIndex, index, node, errorCode); |
421 | } |
422 | } else { |
423 | // Found a stronger node with implied strength-common weight. |
424 | uint32_t weight16 = getWeight16Before(index, node, strength); |
425 | index = findOrInsertWeakNode(index, weight16, strength, errorCode); |
426 | } |
427 | // Strength of the temporary CE = strength of its reset position. |
428 | // Code above raises an error if the before-strength is stronger. |
429 | strength = ceStrength(ces[cesLength - 1]); |
430 | } |
431 | if(U_FAILURE(errorCode)) { |
432 | parserErrorReason = "inserting reset position for &[before n]" ; |
433 | return; |
434 | } |
435 | ces[cesLength - 1] = tempCEFromIndexAndStrength(index, strength); |
436 | } |
437 | |
438 | uint32_t |
439 | CollationBuilder::getWeight16Before(int32_t index, int64_t node, int32_t level) { |
440 | U_ASSERT(strengthFromNode(node) < level || !isTailoredNode(node)); |
441 | // Collect the root CE weights if this node is for a root CE. |
442 | // If it is not, then return the low non-primary boundary for a tailored CE. |
443 | uint32_t t; |
444 | if(strengthFromNode(node) == UCOL_TERTIARY) { |
445 | t = weight16FromNode(node); |
446 | } else { |
447 | t = Collation::COMMON_WEIGHT16; // Stronger node with implied common weight. |
448 | } |
449 | while(strengthFromNode(node) > UCOL_SECONDARY) { |
450 | index = previousIndexFromNode(node); |
451 | node = nodes.elementAti(index); |
452 | } |
453 | if(isTailoredNode(node)) { |
454 | return Collation::BEFORE_WEIGHT16; |
455 | } |
456 | uint32_t s; |
457 | if(strengthFromNode(node) == UCOL_SECONDARY) { |
458 | s = weight16FromNode(node); |
459 | } else { |
460 | s = Collation::COMMON_WEIGHT16; // Stronger node with implied common weight. |
461 | } |
462 | while(strengthFromNode(node) > UCOL_PRIMARY) { |
463 | index = previousIndexFromNode(node); |
464 | node = nodes.elementAti(index); |
465 | } |
466 | if(isTailoredNode(node)) { |
467 | return Collation::BEFORE_WEIGHT16; |
468 | } |
469 | // [p, s, t] is a root CE. Return the preceding weight for the requested level. |
470 | uint32_t p = weight32FromNode(node); |
471 | uint32_t weight16; |
472 | if(level == UCOL_SECONDARY) { |
473 | weight16 = rootElements.getSecondaryBefore(p, s); |
474 | } else { |
475 | weight16 = rootElements.getTertiaryBefore(p, s, t); |
476 | U_ASSERT((weight16 & ~Collation::ONLY_TERTIARY_MASK) == 0); |
477 | } |
478 | return weight16; |
479 | } |
480 | |
481 | int64_t |
482 | CollationBuilder::getSpecialResetPosition(const UnicodeString &str, |
483 | const char *&parserErrorReason, UErrorCode &errorCode) { |
484 | U_ASSERT(str.length() == 2); |
485 | int64_t ce; |
486 | int32_t strength = UCOL_PRIMARY; |
487 | UBool isBoundary = FALSE; |
488 | UChar32 pos = str.charAt(1) - CollationRuleParser::POS_BASE; |
489 | U_ASSERT(0 <= pos && pos <= CollationRuleParser::LAST_TRAILING); |
490 | switch(pos) { |
491 | case CollationRuleParser::FIRST_TERTIARY_IGNORABLE: |
492 | // Quaternary CEs are not supported. |
493 | // Non-zero quaternary weights are possible only on tertiary or stronger CEs. |
494 | return 0; |
495 | case CollationRuleParser::LAST_TERTIARY_IGNORABLE: |
496 | return 0; |
497 | case CollationRuleParser::FIRST_SECONDARY_IGNORABLE: { |
498 | // Look for a tailored tertiary node after [0, 0, 0]. |
499 | int32_t index = findOrInsertNodeForRootCE(0, UCOL_TERTIARY, errorCode); |
500 | if(U_FAILURE(errorCode)) { return 0; } |
501 | int64_t node = nodes.elementAti(index); |
502 | if((index = nextIndexFromNode(node)) != 0) { |
503 | node = nodes.elementAti(index); |
504 | U_ASSERT(strengthFromNode(node) <= UCOL_TERTIARY); |
505 | if(isTailoredNode(node) && strengthFromNode(node) == UCOL_TERTIARY) { |
506 | return tempCEFromIndexAndStrength(index, UCOL_TERTIARY); |
507 | } |
508 | } |
509 | return rootElements.getFirstTertiaryCE(); |
510 | // No need to look for nodeHasAnyBefore() on a tertiary node. |
511 | } |
512 | case CollationRuleParser::LAST_SECONDARY_IGNORABLE: |
513 | ce = rootElements.getLastTertiaryCE(); |
514 | strength = UCOL_TERTIARY; |
515 | break; |
516 | case CollationRuleParser::FIRST_PRIMARY_IGNORABLE: { |
517 | // Look for a tailored secondary node after [0, 0, *]. |
518 | int32_t index = findOrInsertNodeForRootCE(0, UCOL_SECONDARY, errorCode); |
519 | if(U_FAILURE(errorCode)) { return 0; } |
520 | int64_t node = nodes.elementAti(index); |
521 | while((index = nextIndexFromNode(node)) != 0) { |
522 | node = nodes.elementAti(index); |
523 | strength = strengthFromNode(node); |
524 | if(strength < UCOL_SECONDARY) { break; } |
525 | if(strength == UCOL_SECONDARY) { |
526 | if(isTailoredNode(node)) { |
527 | if(nodeHasBefore3(node)) { |
528 | index = nextIndexFromNode(nodes.elementAti(nextIndexFromNode(node))); |
529 | U_ASSERT(isTailoredNode(nodes.elementAti(index))); |
530 | } |
531 | return tempCEFromIndexAndStrength(index, UCOL_SECONDARY); |
532 | } else { |
533 | break; |
534 | } |
535 | } |
536 | } |
537 | ce = rootElements.getFirstSecondaryCE(); |
538 | strength = UCOL_SECONDARY; |
539 | break; |
540 | } |
541 | case CollationRuleParser::LAST_PRIMARY_IGNORABLE: |
542 | ce = rootElements.getLastSecondaryCE(); |
543 | strength = UCOL_SECONDARY; |
544 | break; |
545 | case CollationRuleParser::FIRST_VARIABLE: |
546 | ce = rootElements.getFirstPrimaryCE(); |
547 | isBoundary = TRUE; // FractionalUCA.txt: FDD1 00A0, SPACE first primary |
548 | break; |
549 | case CollationRuleParser::LAST_VARIABLE: |
550 | ce = rootElements.lastCEWithPrimaryBefore(variableTop + 1); |
551 | break; |
552 | case CollationRuleParser::FIRST_REGULAR: |
553 | ce = rootElements.firstCEWithPrimaryAtLeast(variableTop + 1); |
554 | isBoundary = TRUE; // FractionalUCA.txt: FDD1 263A, SYMBOL first primary |
555 | break; |
556 | case CollationRuleParser::LAST_REGULAR: |
557 | // Use the Hani-first-primary rather than the actual last "regular" CE before it, |
558 | // for backward compatibility with behavior before the introduction of |
559 | // script-first-primary CEs in the root collator. |
560 | ce = rootElements.firstCEWithPrimaryAtLeast( |
561 | baseData->getFirstPrimaryForGroup(USCRIPT_HAN)); |
562 | break; |
563 | case CollationRuleParser::FIRST_IMPLICIT: |
564 | ce = baseData->getSingleCE(0x4e00, errorCode); |
565 | break; |
566 | case CollationRuleParser::LAST_IMPLICIT: |
567 | // We do not support tailoring to an unassigned-implicit CE. |
568 | errorCode = U_UNSUPPORTED_ERROR; |
569 | parserErrorReason = "reset to [last implicit] not supported" ; |
570 | return 0; |
571 | case CollationRuleParser::FIRST_TRAILING: |
572 | ce = Collation::makeCE(Collation::FIRST_TRAILING_PRIMARY); |
573 | isBoundary = TRUE; // trailing first primary (there is no mapping for it) |
574 | break; |
575 | case CollationRuleParser::LAST_TRAILING: |
576 | errorCode = U_ILLEGAL_ARGUMENT_ERROR; |
577 | parserErrorReason = "LDML forbids tailoring to U+FFFF" ; |
578 | return 0; |
579 | default: |
580 | UPRV_UNREACHABLE; |
581 | } |
582 | |
583 | int32_t index = findOrInsertNodeForRootCE(ce, strength, errorCode); |
584 | if(U_FAILURE(errorCode)) { return 0; } |
585 | int64_t node = nodes.elementAti(index); |
586 | if((pos & 1) == 0) { |
587 | // even pos = [first xyz] |
588 | if(!nodeHasAnyBefore(node) && isBoundary) { |
589 | // A <group> first primary boundary is artificially added to FractionalUCA.txt. |
590 | // It is reachable via its special contraction, but is not normally used. |
591 | // Find the first character tailored after the boundary CE, |
592 | // or the first real root CE after it. |
593 | if((index = nextIndexFromNode(node)) != 0) { |
594 | // If there is a following node, then it must be tailored |
595 | // because there are no root CEs with a boundary primary |
596 | // and non-common secondary/tertiary weights. |
597 | node = nodes.elementAti(index); |
598 | U_ASSERT(isTailoredNode(node)); |
599 | ce = tempCEFromIndexAndStrength(index, strength); |
600 | } else { |
601 | U_ASSERT(strength == UCOL_PRIMARY); |
602 | uint32_t p = (uint32_t)(ce >> 32); |
603 | int32_t pIndex = rootElements.findPrimary(p); |
604 | UBool isCompressible = baseData->isCompressiblePrimary(p); |
605 | p = rootElements.getPrimaryAfter(p, pIndex, isCompressible); |
606 | ce = Collation::makeCE(p); |
607 | index = findOrInsertNodeForRootCE(ce, UCOL_PRIMARY, errorCode); |
608 | if(U_FAILURE(errorCode)) { return 0; } |
609 | node = nodes.elementAti(index); |
610 | } |
611 | } |
612 | if(nodeHasAnyBefore(node)) { |
613 | // Get the first node that was tailored before this one at a weaker strength. |
614 | if(nodeHasBefore2(node)) { |
615 | index = nextIndexFromNode(nodes.elementAti(nextIndexFromNode(node))); |
616 | node = nodes.elementAti(index); |
617 | } |
618 | if(nodeHasBefore3(node)) { |
619 | index = nextIndexFromNode(nodes.elementAti(nextIndexFromNode(node))); |
620 | } |
621 | U_ASSERT(isTailoredNode(nodes.elementAti(index))); |
622 | ce = tempCEFromIndexAndStrength(index, strength); |
623 | } |
624 | } else { |
625 | // odd pos = [last xyz] |
626 | // Find the last node that was tailored after the [last xyz] |
627 | // at a strength no greater than the position's strength. |
628 | for(;;) { |
629 | int32_t nextIndex = nextIndexFromNode(node); |
630 | if(nextIndex == 0) { break; } |
631 | int64_t nextNode = nodes.elementAti(nextIndex); |
632 | if(strengthFromNode(nextNode) < strength) { break; } |
633 | index = nextIndex; |
634 | node = nextNode; |
635 | } |
636 | // Do not make a temporary CE for a root node. |
637 | // This last node might be the node for the root CE itself, |
638 | // or a node with a common secondary or tertiary weight. |
639 | if(isTailoredNode(node)) { |
640 | ce = tempCEFromIndexAndStrength(index, strength); |
641 | } |
642 | } |
643 | return ce; |
644 | } |
645 | |
646 | void |
647 | CollationBuilder::addRelation(int32_t strength, const UnicodeString &prefix, |
648 | const UnicodeString &str, const UnicodeString &extension, |
649 | const char *&parserErrorReason, UErrorCode &errorCode) { |
650 | if(U_FAILURE(errorCode)) { return; } |
651 | UnicodeString nfdPrefix; |
652 | if(!prefix.isEmpty()) { |
653 | nfd.normalize(prefix, nfdPrefix, errorCode); |
654 | if(U_FAILURE(errorCode)) { |
655 | parserErrorReason = "normalizing the relation prefix" ; |
656 | return; |
657 | } |
658 | } |
659 | UnicodeString nfdString = nfd.normalize(str, errorCode); |
660 | if(U_FAILURE(errorCode)) { |
661 | parserErrorReason = "normalizing the relation string" ; |
662 | return; |
663 | } |
664 | |
665 | // The runtime code decomposes Hangul syllables on the fly, |
666 | // with recursive processing but without making the Jamo pieces visible for matching. |
667 | // It does not work with certain types of contextual mappings. |
668 | int32_t nfdLength = nfdString.length(); |
669 | if(nfdLength >= 2) { |
670 | UChar c = nfdString.charAt(0); |
671 | if(Hangul::isJamoL(c) || Hangul::isJamoV(c)) { |
672 | // While handling a Hangul syllable, contractions starting with Jamo L or V |
673 | // would not see the following Jamo of that syllable. |
674 | errorCode = U_UNSUPPORTED_ERROR; |
675 | parserErrorReason = "contractions starting with conjoining Jamo L or V not supported" ; |
676 | return; |
677 | } |
678 | c = nfdString.charAt(nfdLength - 1); |
679 | if(Hangul::isJamoL(c) || |
680 | (Hangul::isJamoV(c) && Hangul::isJamoL(nfdString.charAt(nfdLength - 2)))) { |
681 | // A contraction ending with Jamo L or L+V would require |
682 | // generating Hangul syllables in addTailComposites() (588 for a Jamo L), |
683 | // or decomposing a following Hangul syllable on the fly, during contraction matching. |
684 | errorCode = U_UNSUPPORTED_ERROR; |
685 | parserErrorReason = "contractions ending with conjoining Jamo L or L+V not supported" ; |
686 | return; |
687 | } |
688 | // A Hangul syllable completely inside a contraction is ok. |
689 | } |
690 | // Note: If there is a prefix, then the parser checked that |
691 | // both the prefix and the string beging with NFC boundaries (not Jamo V or T). |
692 | // Therefore: prefix.isEmpty() || !isJamoVOrT(nfdString.charAt(0)) |
693 | // (While handling a Hangul syllable, prefixes on Jamo V or T |
694 | // would not see the previous Jamo of that syllable.) |
695 | |
696 | if(strength != UCOL_IDENTICAL) { |
697 | // Find the node index after which we insert the new tailored node. |
698 | int32_t index = findOrInsertNodeForCEs(strength, parserErrorReason, errorCode); |
699 | U_ASSERT(cesLength > 0); |
700 | int64_t ce = ces[cesLength - 1]; |
701 | if(strength == UCOL_PRIMARY && !isTempCE(ce) && (uint32_t)(ce >> 32) == 0) { |
702 | // There is no primary gap between ignorables and the space-first-primary. |
703 | errorCode = U_UNSUPPORTED_ERROR; |
704 | parserErrorReason = "tailoring primary after ignorables not supported" ; |
705 | return; |
706 | } |
707 | if(strength == UCOL_QUATERNARY && ce == 0) { |
708 | // The CE data structure does not support non-zero quaternary weights |
709 | // on tertiary ignorables. |
710 | errorCode = U_UNSUPPORTED_ERROR; |
711 | parserErrorReason = "tailoring quaternary after tertiary ignorables not supported" ; |
712 | return; |
713 | } |
714 | // Insert the new tailored node. |
715 | index = insertTailoredNodeAfter(index, strength, errorCode); |
716 | if(U_FAILURE(errorCode)) { |
717 | parserErrorReason = "modifying collation elements" ; |
718 | return; |
719 | } |
720 | // Strength of the temporary CE: |
721 | // The new relation may yield a stronger CE but not a weaker one. |
722 | int32_t tempStrength = ceStrength(ce); |
723 | if(strength < tempStrength) { tempStrength = strength; } |
724 | ces[cesLength - 1] = tempCEFromIndexAndStrength(index, tempStrength); |
725 | } |
726 | |
727 | setCaseBits(nfdString, parserErrorReason, errorCode); |
728 | if(U_FAILURE(errorCode)) { return; } |
729 | |
730 | int32_t cesLengthBeforeExtension = cesLength; |
731 | if(!extension.isEmpty()) { |
732 | UnicodeString nfdExtension = nfd.normalize(extension, errorCode); |
733 | if(U_FAILURE(errorCode)) { |
734 | parserErrorReason = "normalizing the relation extension" ; |
735 | return; |
736 | } |
737 | cesLength = dataBuilder->getCEs(nfdExtension, ces, cesLength); |
738 | if(cesLength > Collation::MAX_EXPANSION_LENGTH) { |
739 | errorCode = U_ILLEGAL_ARGUMENT_ERROR; |
740 | parserErrorReason = |
741 | "extension string adds too many collation elements (more than 31 total)" ; |
742 | return; |
743 | } |
744 | } |
745 | uint32_t ce32 = Collation::UNASSIGNED_CE32; |
746 | if((prefix != nfdPrefix || str != nfdString) && |
747 | !ignorePrefix(prefix, errorCode) && !ignoreString(str, errorCode)) { |
748 | // Map from the original input to the CEs. |
749 | // We do this in case the canonical closure is incomplete, |
750 | // so that it is possible to explicitly provide the missing mappings. |
751 | ce32 = addIfDifferent(prefix, str, ces, cesLength, ce32, errorCode); |
752 | } |
753 | addWithClosure(nfdPrefix, nfdString, ces, cesLength, ce32, errorCode); |
754 | if(U_FAILURE(errorCode)) { |
755 | parserErrorReason = "writing collation elements" ; |
756 | return; |
757 | } |
758 | cesLength = cesLengthBeforeExtension; |
759 | } |
760 | |
761 | int32_t |
762 | CollationBuilder::findOrInsertNodeForCEs(int32_t strength, const char *&parserErrorReason, |
763 | UErrorCode &errorCode) { |
764 | if(U_FAILURE(errorCode)) { return 0; } |
765 | U_ASSERT(UCOL_PRIMARY <= strength && strength <= UCOL_QUATERNARY); |
766 | |
767 | // Find the last CE that is at least as "strong" as the requested difference. |
768 | // Note: Stronger is smaller (UCOL_PRIMARY=0). |
769 | int64_t ce; |
770 | for(;; --cesLength) { |
771 | if(cesLength == 0) { |
772 | ce = ces[0] = 0; |
773 | cesLength = 1; |
774 | break; |
775 | } else { |
776 | ce = ces[cesLength - 1]; |
777 | } |
778 | if(ceStrength(ce) <= strength) { break; } |
779 | } |
780 | |
781 | if(isTempCE(ce)) { |
782 | // No need to findCommonNode() here for lower levels |
783 | // because insertTailoredNodeAfter() will do that anyway. |
784 | return indexFromTempCE(ce); |
785 | } |
786 | |
787 | // root CE |
788 | if((uint8_t)(ce >> 56) == Collation::UNASSIGNED_IMPLICIT_BYTE) { |
789 | errorCode = U_UNSUPPORTED_ERROR; |
790 | parserErrorReason = "tailoring relative to an unassigned code point not supported" ; |
791 | return 0; |
792 | } |
793 | return findOrInsertNodeForRootCE(ce, strength, errorCode); |
794 | } |
795 | |
796 | int32_t |
797 | CollationBuilder::findOrInsertNodeForRootCE(int64_t ce, int32_t strength, UErrorCode &errorCode) { |
798 | if(U_FAILURE(errorCode)) { return 0; } |
799 | U_ASSERT((uint8_t)(ce >> 56) != Collation::UNASSIGNED_IMPLICIT_BYTE); |
800 | |
801 | // Find or insert the node for each of the root CE's weights, |
802 | // down to the requested level/strength. |
803 | // Root CEs must have common=zero quaternary weights (for which we never insert any nodes). |
804 | U_ASSERT((ce & 0xc0) == 0); |
805 | int32_t index = findOrInsertNodeForPrimary((uint32_t)(ce >> 32), errorCode); |
806 | if(strength >= UCOL_SECONDARY) { |
807 | uint32_t lower32 = (uint32_t)ce; |
808 | index = findOrInsertWeakNode(index, lower32 >> 16, UCOL_SECONDARY, errorCode); |
809 | if(strength >= UCOL_TERTIARY) { |
810 | index = findOrInsertWeakNode(index, lower32 & Collation::ONLY_TERTIARY_MASK, |
811 | UCOL_TERTIARY, errorCode); |
812 | } |
813 | } |
814 | return index; |
815 | } |
816 | |
817 | namespace { |
818 | |
819 | /** |
820 | * Like Java Collections.binarySearch(List, key, Comparator). |
821 | * |
822 | * @return the index>=0 where the item was found, |
823 | * or the index<0 for inserting the string at ~index in sorted order |
824 | * (index into rootPrimaryIndexes) |
825 | */ |
826 | int32_t |
827 | binarySearchForRootPrimaryNode(const int32_t *rootPrimaryIndexes, int32_t length, |
828 | const int64_t *nodes, uint32_t p) { |
829 | if(length == 0) { return ~0; } |
830 | int32_t start = 0; |
831 | int32_t limit = length; |
832 | for (;;) { |
833 | int32_t i = (start + limit) / 2; |
834 | int64_t node = nodes[rootPrimaryIndexes[i]]; |
835 | uint32_t nodePrimary = (uint32_t)(node >> 32); // weight32FromNode(node) |
836 | if (p == nodePrimary) { |
837 | return i; |
838 | } else if (p < nodePrimary) { |
839 | if (i == start) { |
840 | return ~start; // insert s before i |
841 | } |
842 | limit = i; |
843 | } else { |
844 | if (i == start) { |
845 | return ~(start + 1); // insert s after i |
846 | } |
847 | start = i; |
848 | } |
849 | } |
850 | } |
851 | |
852 | } // namespace |
853 | |
854 | int32_t |
855 | CollationBuilder::findOrInsertNodeForPrimary(uint32_t p, UErrorCode &errorCode) { |
856 | if(U_FAILURE(errorCode)) { return 0; } |
857 | |
858 | int32_t rootIndex = binarySearchForRootPrimaryNode( |
859 | rootPrimaryIndexes.getBuffer(), rootPrimaryIndexes.size(), nodes.getBuffer(), p); |
860 | if(rootIndex >= 0) { |
861 | return rootPrimaryIndexes.elementAti(rootIndex); |
862 | } else { |
863 | // Start a new list of nodes with this primary. |
864 | int32_t index = nodes.size(); |
865 | nodes.addElement(nodeFromWeight32(p), errorCode); |
866 | rootPrimaryIndexes.insertElementAt(index, ~rootIndex, errorCode); |
867 | return index; |
868 | } |
869 | } |
870 | |
871 | int32_t |
872 | CollationBuilder::findOrInsertWeakNode(int32_t index, uint32_t weight16, int32_t level, UErrorCode &errorCode) { |
873 | if(U_FAILURE(errorCode)) { return 0; } |
874 | U_ASSERT(0 <= index && index < nodes.size()); |
875 | U_ASSERT(UCOL_SECONDARY <= level && level <= UCOL_TERTIARY); |
876 | |
877 | if(weight16 == Collation::COMMON_WEIGHT16) { |
878 | return findCommonNode(index, level); |
879 | } |
880 | |
881 | // If this will be the first below-common weight for the parent node, |
882 | // then we will also need to insert a common weight after it. |
883 | int64_t node = nodes.elementAti(index); |
884 | U_ASSERT(strengthFromNode(node) < level); // parent node is stronger |
885 | if(weight16 != 0 && weight16 < Collation::COMMON_WEIGHT16) { |
886 | int32_t hasThisLevelBefore = level == UCOL_SECONDARY ? HAS_BEFORE2 : HAS_BEFORE3; |
887 | if((node & hasThisLevelBefore) == 0) { |
888 | // The parent node has an implied level-common weight. |
889 | int64_t commonNode = |
890 | nodeFromWeight16(Collation::COMMON_WEIGHT16) | nodeFromStrength(level); |
891 | if(level == UCOL_SECONDARY) { |
892 | // Move the HAS_BEFORE3 flag from the parent node |
893 | // to the new secondary common node. |
894 | commonNode |= node & HAS_BEFORE3; |
895 | node &= ~(int64_t)HAS_BEFORE3; |
896 | } |
897 | nodes.setElementAt(node | hasThisLevelBefore, index); |
898 | // Insert below-common-weight node. |
899 | int32_t nextIndex = nextIndexFromNode(node); |
900 | node = nodeFromWeight16(weight16) | nodeFromStrength(level); |
901 | index = insertNodeBetween(index, nextIndex, node, errorCode); |
902 | // Insert common-weight node. |
903 | insertNodeBetween(index, nextIndex, commonNode, errorCode); |
904 | // Return index of below-common-weight node. |
905 | return index; |
906 | } |
907 | } |
908 | |
909 | // Find the root CE's weight for this level. |
910 | // Postpone insertion if not found: |
911 | // Insert the new root node before the next stronger node, |
912 | // or before the next root node with the same strength and a larger weight. |
913 | int32_t nextIndex; |
914 | while((nextIndex = nextIndexFromNode(node)) != 0) { |
915 | node = nodes.elementAti(nextIndex); |
916 | int32_t nextStrength = strengthFromNode(node); |
917 | if(nextStrength <= level) { |
918 | // Insert before a stronger node. |
919 | if(nextStrength < level) { break; } |
920 | // nextStrength == level |
921 | if(!isTailoredNode(node)) { |
922 | uint32_t nextWeight16 = weight16FromNode(node); |
923 | if(nextWeight16 == weight16) { |
924 | // Found the node for the root CE up to this level. |
925 | return nextIndex; |
926 | } |
927 | // Insert before a node with a larger same-strength weight. |
928 | if(nextWeight16 > weight16) { break; } |
929 | } |
930 | } |
931 | // Skip the next node. |
932 | index = nextIndex; |
933 | } |
934 | node = nodeFromWeight16(weight16) | nodeFromStrength(level); |
935 | return insertNodeBetween(index, nextIndex, node, errorCode); |
936 | } |
937 | |
938 | int32_t |
939 | CollationBuilder::insertTailoredNodeAfter(int32_t index, int32_t strength, UErrorCode &errorCode) { |
940 | if(U_FAILURE(errorCode)) { return 0; } |
941 | U_ASSERT(0 <= index && index < nodes.size()); |
942 | if(strength >= UCOL_SECONDARY) { |
943 | index = findCommonNode(index, UCOL_SECONDARY); |
944 | if(strength >= UCOL_TERTIARY) { |
945 | index = findCommonNode(index, UCOL_TERTIARY); |
946 | } |
947 | } |
948 | // Postpone insertion: |
949 | // Insert the new node before the next one with a strength at least as strong. |
950 | int64_t node = nodes.elementAti(index); |
951 | int32_t nextIndex; |
952 | while((nextIndex = nextIndexFromNode(node)) != 0) { |
953 | node = nodes.elementAti(nextIndex); |
954 | if(strengthFromNode(node) <= strength) { break; } |
955 | // Skip the next node which has a weaker (larger) strength than the new one. |
956 | index = nextIndex; |
957 | } |
958 | node = IS_TAILORED | nodeFromStrength(strength); |
959 | return insertNodeBetween(index, nextIndex, node, errorCode); |
960 | } |
961 | |
962 | int32_t |
963 | CollationBuilder::insertNodeBetween(int32_t index, int32_t nextIndex, int64_t node, |
964 | UErrorCode &errorCode) { |
965 | if(U_FAILURE(errorCode)) { return 0; } |
966 | U_ASSERT(previousIndexFromNode(node) == 0); |
967 | U_ASSERT(nextIndexFromNode(node) == 0); |
968 | U_ASSERT(nextIndexFromNode(nodes.elementAti(index)) == nextIndex); |
969 | // Append the new node and link it to the existing nodes. |
970 | int32_t newIndex = nodes.size(); |
971 | node |= nodeFromPreviousIndex(index) | nodeFromNextIndex(nextIndex); |
972 | nodes.addElement(node, errorCode); |
973 | if(U_FAILURE(errorCode)) { return 0; } |
974 | // nodes[index].nextIndex = newIndex |
975 | node = nodes.elementAti(index); |
976 | nodes.setElementAt(changeNodeNextIndex(node, newIndex), index); |
977 | // nodes[nextIndex].previousIndex = newIndex |
978 | if(nextIndex != 0) { |
979 | node = nodes.elementAti(nextIndex); |
980 | nodes.setElementAt(changeNodePreviousIndex(node, newIndex), nextIndex); |
981 | } |
982 | return newIndex; |
983 | } |
984 | |
985 | int32_t |
986 | CollationBuilder::findCommonNode(int32_t index, int32_t strength) const { |
987 | U_ASSERT(UCOL_SECONDARY <= strength && strength <= UCOL_TERTIARY); |
988 | int64_t node = nodes.elementAti(index); |
989 | if(strengthFromNode(node) >= strength) { |
990 | // The current node is no stronger. |
991 | return index; |
992 | } |
993 | if(strength == UCOL_SECONDARY ? !nodeHasBefore2(node) : !nodeHasBefore3(node)) { |
994 | // The current node implies the strength-common weight. |
995 | return index; |
996 | } |
997 | index = nextIndexFromNode(node); |
998 | node = nodes.elementAti(index); |
999 | U_ASSERT(!isTailoredNode(node) && strengthFromNode(node) == strength && |
1000 | weight16FromNode(node) < Collation::COMMON_WEIGHT16); |
1001 | // Skip to the explicit common node. |
1002 | do { |
1003 | index = nextIndexFromNode(node); |
1004 | node = nodes.elementAti(index); |
1005 | U_ASSERT(strengthFromNode(node) >= strength); |
1006 | } while(isTailoredNode(node) || strengthFromNode(node) > strength || |
1007 | weight16FromNode(node) < Collation::COMMON_WEIGHT16); |
1008 | U_ASSERT(weight16FromNode(node) == Collation::COMMON_WEIGHT16); |
1009 | return index; |
1010 | } |
1011 | |
1012 | void |
1013 | CollationBuilder::setCaseBits(const UnicodeString &nfdString, |
1014 | const char *&parserErrorReason, UErrorCode &errorCode) { |
1015 | if(U_FAILURE(errorCode)) { return; } |
1016 | int32_t numTailoredPrimaries = 0; |
1017 | for(int32_t i = 0; i < cesLength; ++i) { |
1018 | if(ceStrength(ces[i]) == UCOL_PRIMARY) { ++numTailoredPrimaries; } |
1019 | } |
1020 | // We should not be able to get too many case bits because |
1021 | // cesLength<=31==MAX_EXPANSION_LENGTH. |
1022 | // 31 pairs of case bits fit into an int64_t without setting its sign bit. |
1023 | U_ASSERT(numTailoredPrimaries <= 31); |
1024 | |
1025 | int64_t cases = 0; |
1026 | if(numTailoredPrimaries > 0) { |
1027 | const UChar *s = nfdString.getBuffer(); |
1028 | UTF16CollationIterator baseCEs(baseData, FALSE, s, s, s + nfdString.length()); |
1029 | int32_t baseCEsLength = baseCEs.fetchCEs(errorCode) - 1; |
1030 | if(U_FAILURE(errorCode)) { |
1031 | parserErrorReason = "fetching root CEs for tailored string" ; |
1032 | return; |
1033 | } |
1034 | U_ASSERT(baseCEsLength >= 0 && baseCEs.getCE(baseCEsLength) == Collation::NO_CE); |
1035 | |
1036 | uint32_t lastCase = 0; |
1037 | int32_t numBasePrimaries = 0; |
1038 | for(int32_t i = 0; i < baseCEsLength; ++i) { |
1039 | int64_t ce = baseCEs.getCE(i); |
1040 | if((ce >> 32) != 0) { |
1041 | ++numBasePrimaries; |
1042 | uint32_t c = ((uint32_t)ce >> 14) & 3; |
1043 | U_ASSERT(c == 0 || c == 2); // lowercase or uppercase, no mixed case in any base CE |
1044 | if(numBasePrimaries < numTailoredPrimaries) { |
1045 | cases |= (int64_t)c << ((numBasePrimaries - 1) * 2); |
1046 | } else if(numBasePrimaries == numTailoredPrimaries) { |
1047 | lastCase = c; |
1048 | } else if(c != lastCase) { |
1049 | // There are more base primary CEs than tailored primaries. |
1050 | // Set mixed case if the case bits of the remainder differ. |
1051 | lastCase = 1; |
1052 | // Nothing more can change. |
1053 | break; |
1054 | } |
1055 | } |
1056 | } |
1057 | if(numBasePrimaries >= numTailoredPrimaries) { |
1058 | cases |= (int64_t)lastCase << ((numTailoredPrimaries - 1) * 2); |
1059 | } |
1060 | } |
1061 | |
1062 | for(int32_t i = 0; i < cesLength; ++i) { |
1063 | int64_t ce = ces[i] & INT64_C(0xffffffffffff3fff); // clear old case bits |
1064 | int32_t strength = ceStrength(ce); |
1065 | if(strength == UCOL_PRIMARY) { |
1066 | ce |= (cases & 3) << 14; |
1067 | cases >>= 2; |
1068 | } else if(strength == UCOL_TERTIARY) { |
1069 | // Tertiary CEs must have uppercase bits. |
1070 | // See the LDML spec, and comments in class CollationCompare. |
1071 | ce |= 0x8000; |
1072 | } |
1073 | // Tertiary ignorable CEs must have 0 case bits. |
1074 | // We set 0 case bits for secondary CEs too |
1075 | // since currently only U+0345 is cased and maps to a secondary CE, |
1076 | // and it is lowercase. Other secondaries are uncased. |
1077 | // See [[:Cased:]&[:uca1=:]] where uca1 queries the root primary weight. |
1078 | ces[i] = ce; |
1079 | } |
1080 | } |
1081 | |
1082 | void |
1083 | CollationBuilder::suppressContractions(const UnicodeSet &set, const char *&parserErrorReason, |
1084 | UErrorCode &errorCode) { |
1085 | if(U_FAILURE(errorCode)) { return; } |
1086 | dataBuilder->suppressContractions(set, errorCode); |
1087 | if(U_FAILURE(errorCode)) { |
1088 | parserErrorReason = "application of [suppressContractions [set]] failed" ; |
1089 | } |
1090 | } |
1091 | |
1092 | void |
1093 | CollationBuilder::optimize(const UnicodeSet &set, const char *& /* parserErrorReason */, |
1094 | UErrorCode &errorCode) { |
1095 | if(U_FAILURE(errorCode)) { return; } |
1096 | optimizeSet.addAll(set); |
1097 | } |
1098 | |
1099 | uint32_t |
1100 | CollationBuilder::addWithClosure(const UnicodeString &nfdPrefix, const UnicodeString &nfdString, |
1101 | const int64_t newCEs[], int32_t newCEsLength, uint32_t ce32, |
1102 | UErrorCode &errorCode) { |
1103 | // Map from the NFD input to the CEs. |
1104 | ce32 = addIfDifferent(nfdPrefix, nfdString, newCEs, newCEsLength, ce32, errorCode); |
1105 | ce32 = addOnlyClosure(nfdPrefix, nfdString, newCEs, newCEsLength, ce32, errorCode); |
1106 | addTailComposites(nfdPrefix, nfdString, errorCode); |
1107 | return ce32; |
1108 | } |
1109 | |
1110 | uint32_t |
1111 | CollationBuilder::addOnlyClosure(const UnicodeString &nfdPrefix, const UnicodeString &nfdString, |
1112 | const int64_t newCEs[], int32_t newCEsLength, uint32_t ce32, |
1113 | UErrorCode &errorCode) { |
1114 | if(U_FAILURE(errorCode)) { return ce32; } |
1115 | |
1116 | // Map from canonically equivalent input to the CEs. (But not from the all-NFD input.) |
1117 | if(nfdPrefix.isEmpty()) { |
1118 | CanonicalIterator stringIter(nfdString, errorCode); |
1119 | if(U_FAILURE(errorCode)) { return ce32; } |
1120 | UnicodeString prefix; |
1121 | for(;;) { |
1122 | UnicodeString str = stringIter.next(); |
1123 | if(str.isBogus()) { break; } |
1124 | if(ignoreString(str, errorCode) || str == nfdString) { continue; } |
1125 | ce32 = addIfDifferent(prefix, str, newCEs, newCEsLength, ce32, errorCode); |
1126 | if(U_FAILURE(errorCode)) { return ce32; } |
1127 | } |
1128 | } else { |
1129 | CanonicalIterator prefixIter(nfdPrefix, errorCode); |
1130 | CanonicalIterator stringIter(nfdString, errorCode); |
1131 | if(U_FAILURE(errorCode)) { return ce32; } |
1132 | for(;;) { |
1133 | UnicodeString prefix = prefixIter.next(); |
1134 | if(prefix.isBogus()) { break; } |
1135 | if(ignorePrefix(prefix, errorCode)) { continue; } |
1136 | UBool samePrefix = prefix == nfdPrefix; |
1137 | for(;;) { |
1138 | UnicodeString str = stringIter.next(); |
1139 | if(str.isBogus()) { break; } |
1140 | if(ignoreString(str, errorCode) || (samePrefix && str == nfdString)) { continue; } |
1141 | ce32 = addIfDifferent(prefix, str, newCEs, newCEsLength, ce32, errorCode); |
1142 | if(U_FAILURE(errorCode)) { return ce32; } |
1143 | } |
1144 | stringIter.reset(); |
1145 | } |
1146 | } |
1147 | return ce32; |
1148 | } |
1149 | |
1150 | void |
1151 | CollationBuilder::addTailComposites(const UnicodeString &nfdPrefix, const UnicodeString &nfdString, |
1152 | UErrorCode &errorCode) { |
1153 | if(U_FAILURE(errorCode)) { return; } |
1154 | |
1155 | // Look for the last starter in the NFD string. |
1156 | UChar32 lastStarter; |
1157 | int32_t indexAfterLastStarter = nfdString.length(); |
1158 | for(;;) { |
1159 | if(indexAfterLastStarter == 0) { return; } // no starter at all |
1160 | lastStarter = nfdString.char32At(indexAfterLastStarter - 1); |
1161 | if(nfd.getCombiningClass(lastStarter) == 0) { break; } |
1162 | indexAfterLastStarter -= U16_LENGTH(lastStarter); |
1163 | } |
1164 | // No closure to Hangul syllables since we decompose them on the fly. |
1165 | if(Hangul::isJamoL(lastStarter)) { return; } |
1166 | |
1167 | // Are there any composites whose decomposition starts with the lastStarter? |
1168 | // Note: Normalizer2Impl does not currently return start sets for NFC_QC=Maybe characters. |
1169 | // We might find some more equivalent mappings here if it did. |
1170 | UnicodeSet composites; |
1171 | if(!nfcImpl.getCanonStartSet(lastStarter, composites)) { return; } |
1172 | |
1173 | UnicodeString decomp; |
1174 | UnicodeString newNFDString, newString; |
1175 | int64_t newCEs[Collation::MAX_EXPANSION_LENGTH]; |
1176 | UnicodeSetIterator iter(composites); |
1177 | while(iter.next()) { |
1178 | U_ASSERT(!iter.isString()); |
1179 | UChar32 composite = iter.getCodepoint(); |
1180 | nfd.getDecomposition(composite, decomp); |
1181 | if(!mergeCompositeIntoString(nfdString, indexAfterLastStarter, composite, decomp, |
1182 | newNFDString, newString, errorCode)) { |
1183 | continue; |
1184 | } |
1185 | int32_t newCEsLength = dataBuilder->getCEs(nfdPrefix, newNFDString, newCEs, 0); |
1186 | if(newCEsLength > Collation::MAX_EXPANSION_LENGTH) { |
1187 | // Ignore mappings that we cannot store. |
1188 | continue; |
1189 | } |
1190 | // Note: It is possible that the newCEs do not make use of the mapping |
1191 | // for which we are adding the tail composites, in which case we might be adding |
1192 | // unnecessary mappings. |
1193 | // For example, when we add tail composites for ae^ (^=combining circumflex), |
1194 | // UCA discontiguous-contraction matching does not find any matches |
1195 | // for ae_^ (_=any combining diacritic below) *unless* there is also |
1196 | // a contraction mapping for ae. |
1197 | // Thus, if there is no ae contraction, then the ae^ mapping is ignored |
1198 | // while fetching the newCEs for ae_^. |
1199 | // TODO: Try to detect this effectively. |
1200 | // (Alternatively, print a warning when prefix contractions are missing.) |
1201 | |
1202 | // We do not need an explicit mapping for the NFD strings. |
1203 | // It is fine if the NFD input collates like this via a sequence of mappings. |
1204 | // It also saves a little bit of space, and may reduce the set of characters with contractions. |
1205 | uint32_t ce32 = addIfDifferent(nfdPrefix, newString, |
1206 | newCEs, newCEsLength, Collation::UNASSIGNED_CE32, errorCode); |
1207 | if(ce32 != Collation::UNASSIGNED_CE32) { |
1208 | // was different, was added |
1209 | addOnlyClosure(nfdPrefix, newNFDString, newCEs, newCEsLength, ce32, errorCode); |
1210 | } |
1211 | } |
1212 | } |
1213 | |
1214 | UBool |
1215 | CollationBuilder::mergeCompositeIntoString(const UnicodeString &nfdString, |
1216 | int32_t indexAfterLastStarter, |
1217 | UChar32 composite, const UnicodeString &decomp, |
1218 | UnicodeString &newNFDString, UnicodeString &newString, |
1219 | UErrorCode &errorCode) const { |
1220 | if(U_FAILURE(errorCode)) { return FALSE; } |
1221 | U_ASSERT(nfdString.char32At(indexAfterLastStarter - 1) == decomp.char32At(0)); |
1222 | int32_t lastStarterLength = decomp.moveIndex32(0, 1); |
1223 | if(lastStarterLength == decomp.length()) { |
1224 | // Singleton decompositions should be found by addWithClosure() |
1225 | // and the CanonicalIterator, so we can ignore them here. |
1226 | return FALSE; |
1227 | } |
1228 | if(nfdString.compare(indexAfterLastStarter, 0x7fffffff, |
1229 | decomp, lastStarterLength, 0x7fffffff) == 0) { |
1230 | // same strings, nothing new to be found here |
1231 | return FALSE; |
1232 | } |
1233 | |
1234 | // Make new FCD strings that combine a composite, or its decomposition, |
1235 | // into the nfdString's last starter and the combining marks following it. |
1236 | // Make an NFD version, and a version with the composite. |
1237 | newNFDString.setTo(nfdString, 0, indexAfterLastStarter); |
1238 | newString.setTo(nfdString, 0, indexAfterLastStarter - lastStarterLength).append(composite); |
1239 | |
1240 | // The following is related to discontiguous contraction matching, |
1241 | // but builds only FCD strings (or else returns FALSE). |
1242 | int32_t sourceIndex = indexAfterLastStarter; |
1243 | int32_t decompIndex = lastStarterLength; |
1244 | // Small optimization: We keep the source character across loop iterations |
1245 | // because we do not always consume it, |
1246 | // and then need not fetch it again nor look up its combining class again. |
1247 | UChar32 sourceChar = U_SENTINEL; |
1248 | // The cc variables need to be declared before the loop so that at the end |
1249 | // they are set to the last combining classes seen. |
1250 | uint8_t sourceCC = 0; |
1251 | uint8_t decompCC = 0; |
1252 | for(;;) { |
1253 | if(sourceChar < 0) { |
1254 | if(sourceIndex >= nfdString.length()) { break; } |
1255 | sourceChar = nfdString.char32At(sourceIndex); |
1256 | sourceCC = nfd.getCombiningClass(sourceChar); |
1257 | U_ASSERT(sourceCC != 0); |
1258 | } |
1259 | // We consume a decomposition character in each iteration. |
1260 | if(decompIndex >= decomp.length()) { break; } |
1261 | UChar32 decompChar = decomp.char32At(decompIndex); |
1262 | decompCC = nfd.getCombiningClass(decompChar); |
1263 | // Compare the two characters and their combining classes. |
1264 | if(decompCC == 0) { |
1265 | // Unable to merge because the source contains a non-zero combining mark |
1266 | // but the composite's decomposition contains another starter. |
1267 | // The strings would not be equivalent. |
1268 | return FALSE; |
1269 | } else if(sourceCC < decompCC) { |
1270 | // Composite + sourceChar would not be FCD. |
1271 | return FALSE; |
1272 | } else if(decompCC < sourceCC) { |
1273 | newNFDString.append(decompChar); |
1274 | decompIndex += U16_LENGTH(decompChar); |
1275 | } else if(decompChar != sourceChar) { |
1276 | // Blocked because same combining class. |
1277 | return FALSE; |
1278 | } else { // match: decompChar == sourceChar |
1279 | newNFDString.append(decompChar); |
1280 | decompIndex += U16_LENGTH(decompChar); |
1281 | sourceIndex += U16_LENGTH(decompChar); |
1282 | sourceChar = U_SENTINEL; |
1283 | } |
1284 | } |
1285 | // We are at the end of at least one of the two inputs. |
1286 | if(sourceChar >= 0) { // more characters from nfdString but not from decomp |
1287 | if(sourceCC < decompCC) { |
1288 | // Appending the next source character to the composite would not be FCD. |
1289 | return FALSE; |
1290 | } |
1291 | newNFDString.append(nfdString, sourceIndex, 0x7fffffff); |
1292 | newString.append(nfdString, sourceIndex, 0x7fffffff); |
1293 | } else if(decompIndex < decomp.length()) { // more characters from decomp, not from nfdString |
1294 | newNFDString.append(decomp, decompIndex, 0x7fffffff); |
1295 | } |
1296 | U_ASSERT(nfd.isNormalized(newNFDString, errorCode)); |
1297 | U_ASSERT(fcd.isNormalized(newString, errorCode)); |
1298 | U_ASSERT(nfd.normalize(newString, errorCode) == newNFDString); // canonically equivalent |
1299 | return TRUE; |
1300 | } |
1301 | |
1302 | UBool |
1303 | CollationBuilder::ignorePrefix(const UnicodeString &s, UErrorCode &errorCode) const { |
1304 | // Do not map non-FCD prefixes. |
1305 | return !isFCD(s, errorCode); |
1306 | } |
1307 | |
1308 | UBool |
1309 | CollationBuilder::ignoreString(const UnicodeString &s, UErrorCode &errorCode) const { |
1310 | // Do not map non-FCD strings. |
1311 | // Do not map strings that start with Hangul syllables: We decompose those on the fly. |
1312 | return !isFCD(s, errorCode) || Hangul::isHangul(s.charAt(0)); |
1313 | } |
1314 | |
1315 | UBool |
1316 | CollationBuilder::isFCD(const UnicodeString &s, UErrorCode &errorCode) const { |
1317 | return U_SUCCESS(errorCode) && fcd.isNormalized(s, errorCode); |
1318 | } |
1319 | |
1320 | void |
1321 | CollationBuilder::closeOverComposites(UErrorCode &errorCode) { |
1322 | UnicodeSet composites(UNICODE_STRING_SIMPLE("[:NFD_QC=N:]" ), errorCode); // Java: static final |
1323 | if(U_FAILURE(errorCode)) { return; } |
1324 | // Hangul is decomposed on the fly during collation. |
1325 | composites.remove(Hangul::HANGUL_BASE, Hangul::HANGUL_END); |
1326 | UnicodeString prefix; // empty |
1327 | UnicodeString nfdString; |
1328 | UnicodeSetIterator iter(composites); |
1329 | while(iter.next()) { |
1330 | U_ASSERT(!iter.isString()); |
1331 | nfd.getDecomposition(iter.getCodepoint(), nfdString); |
1332 | cesLength = dataBuilder->getCEs(nfdString, ces, 0); |
1333 | if(cesLength > Collation::MAX_EXPANSION_LENGTH) { |
1334 | // Too many CEs from the decomposition (unusual), ignore this composite. |
1335 | // We could add a capacity parameter to getCEs() and reallocate if necessary. |
1336 | // However, this can only really happen in contrived cases. |
1337 | continue; |
1338 | } |
1339 | const UnicodeString &composite(iter.getString()); |
1340 | addIfDifferent(prefix, composite, ces, cesLength, Collation::UNASSIGNED_CE32, errorCode); |
1341 | } |
1342 | } |
1343 | |
1344 | uint32_t |
1345 | CollationBuilder::addIfDifferent(const UnicodeString &prefix, const UnicodeString &str, |
1346 | const int64_t newCEs[], int32_t newCEsLength, uint32_t ce32, |
1347 | UErrorCode &errorCode) { |
1348 | if(U_FAILURE(errorCode)) { return ce32; } |
1349 | int64_t oldCEs[Collation::MAX_EXPANSION_LENGTH]; |
1350 | int32_t oldCEsLength = dataBuilder->getCEs(prefix, str, oldCEs, 0); |
1351 | if(!sameCEs(newCEs, newCEsLength, oldCEs, oldCEsLength)) { |
1352 | if(ce32 == Collation::UNASSIGNED_CE32) { |
1353 | ce32 = dataBuilder->encodeCEs(newCEs, newCEsLength, errorCode); |
1354 | } |
1355 | dataBuilder->addCE32(prefix, str, ce32, errorCode); |
1356 | } |
1357 | return ce32; |
1358 | } |
1359 | |
1360 | UBool |
1361 | CollationBuilder::sameCEs(const int64_t ces1[], int32_t ces1Length, |
1362 | const int64_t ces2[], int32_t ces2Length) { |
1363 | if(ces1Length != ces2Length) { |
1364 | return FALSE; |
1365 | } |
1366 | U_ASSERT(ces1Length <= Collation::MAX_EXPANSION_LENGTH); |
1367 | for(int32_t i = 0; i < ces1Length; ++i) { |
1368 | if(ces1[i] != ces2[i]) { return FALSE; } |
1369 | } |
1370 | return TRUE; |
1371 | } |
1372 | |
1373 | #ifdef DEBUG_COLLATION_BUILDER |
1374 | |
1375 | uint32_t |
1376 | alignWeightRight(uint32_t w) { |
1377 | if(w != 0) { |
1378 | while((w & 0xff) == 0) { w >>= 8; } |
1379 | } |
1380 | return w; |
1381 | } |
1382 | |
1383 | #endif |
1384 | |
1385 | void |
1386 | CollationBuilder::makeTailoredCEs(UErrorCode &errorCode) { |
1387 | if(U_FAILURE(errorCode)) { return; } |
1388 | |
1389 | CollationWeights primaries, secondaries, tertiaries; |
1390 | int64_t *nodesArray = nodes.getBuffer(); |
1391 | #ifdef DEBUG_COLLATION_BUILDER |
1392 | puts("\nCollationBuilder::makeTailoredCEs()" ); |
1393 | #endif |
1394 | |
1395 | for(int32_t rpi = 0; rpi < rootPrimaryIndexes.size(); ++rpi) { |
1396 | int32_t i = rootPrimaryIndexes.elementAti(rpi); |
1397 | int64_t node = nodesArray[i]; |
1398 | uint32_t p = weight32FromNode(node); |
1399 | uint32_t s = p == 0 ? 0 : Collation::COMMON_WEIGHT16; |
1400 | uint32_t t = s; |
1401 | uint32_t q = 0; |
1402 | UBool pIsTailored = FALSE; |
1403 | UBool sIsTailored = FALSE; |
1404 | UBool tIsTailored = FALSE; |
1405 | #ifdef DEBUG_COLLATION_BUILDER |
1406 | printf("\nprimary %lx\n" , (long)alignWeightRight(p)); |
1407 | #endif |
1408 | int32_t pIndex = p == 0 ? 0 : rootElements.findPrimary(p); |
1409 | int32_t nextIndex = nextIndexFromNode(node); |
1410 | while(nextIndex != 0) { |
1411 | i = nextIndex; |
1412 | node = nodesArray[i]; |
1413 | nextIndex = nextIndexFromNode(node); |
1414 | int32_t strength = strengthFromNode(node); |
1415 | if(strength == UCOL_QUATERNARY) { |
1416 | U_ASSERT(isTailoredNode(node)); |
1417 | #ifdef DEBUG_COLLATION_BUILDER |
1418 | printf(" quat+ " ); |
1419 | #endif |
1420 | if(q == 3) { |
1421 | errorCode = U_BUFFER_OVERFLOW_ERROR; |
1422 | errorReason = "quaternary tailoring gap too small" ; |
1423 | return; |
1424 | } |
1425 | ++q; |
1426 | } else { |
1427 | if(strength == UCOL_TERTIARY) { |
1428 | if(isTailoredNode(node)) { |
1429 | #ifdef DEBUG_COLLATION_BUILDER |
1430 | printf(" ter+ " ); |
1431 | #endif |
1432 | if(!tIsTailored) { |
1433 | // First tailored tertiary node for [p, s]. |
1434 | int32_t tCount = countTailoredNodes(nodesArray, nextIndex, |
1435 | UCOL_TERTIARY) + 1; |
1436 | uint32_t tLimit; |
1437 | if(t == 0) { |
1438 | // Gap at the beginning of the tertiary CE range. |
1439 | t = rootElements.getTertiaryBoundary() - 0x100; |
1440 | tLimit = rootElements.getFirstTertiaryCE() & Collation::ONLY_TERTIARY_MASK; |
1441 | } else if(!pIsTailored && !sIsTailored) { |
1442 | // p and s are root weights. |
1443 | tLimit = rootElements.getTertiaryAfter(pIndex, s, t); |
1444 | } else if(t == Collation::BEFORE_WEIGHT16) { |
1445 | tLimit = Collation::COMMON_WEIGHT16; |
1446 | } else { |
1447 | // [p, s] is tailored. |
1448 | U_ASSERT(t == Collation::COMMON_WEIGHT16); |
1449 | tLimit = rootElements.getTertiaryBoundary(); |
1450 | } |
1451 | U_ASSERT(tLimit == 0x4000 || (tLimit & ~Collation::ONLY_TERTIARY_MASK) == 0); |
1452 | tertiaries.initForTertiary(); |
1453 | if(!tertiaries.allocWeights(t, tLimit, tCount)) { |
1454 | errorCode = U_BUFFER_OVERFLOW_ERROR; |
1455 | errorReason = "tertiary tailoring gap too small" ; |
1456 | return; |
1457 | } |
1458 | tIsTailored = TRUE; |
1459 | } |
1460 | t = tertiaries.nextWeight(); |
1461 | U_ASSERT(t != 0xffffffff); |
1462 | } else { |
1463 | t = weight16FromNode(node); |
1464 | tIsTailored = FALSE; |
1465 | #ifdef DEBUG_COLLATION_BUILDER |
1466 | printf(" ter %lx\n" , (long)alignWeightRight(t)); |
1467 | #endif |
1468 | } |
1469 | } else { |
1470 | if(strength == UCOL_SECONDARY) { |
1471 | if(isTailoredNode(node)) { |
1472 | #ifdef DEBUG_COLLATION_BUILDER |
1473 | printf(" sec+ " ); |
1474 | #endif |
1475 | if(!sIsTailored) { |
1476 | // First tailored secondary node for p. |
1477 | int32_t sCount = countTailoredNodes(nodesArray, nextIndex, |
1478 | UCOL_SECONDARY) + 1; |
1479 | uint32_t sLimit; |
1480 | if(s == 0) { |
1481 | // Gap at the beginning of the secondary CE range. |
1482 | s = rootElements.getSecondaryBoundary() - 0x100; |
1483 | sLimit = rootElements.getFirstSecondaryCE() >> 16; |
1484 | } else if(!pIsTailored) { |
1485 | // p is a root primary. |
1486 | sLimit = rootElements.getSecondaryAfter(pIndex, s); |
1487 | } else if(s == Collation::BEFORE_WEIGHT16) { |
1488 | sLimit = Collation::COMMON_WEIGHT16; |
1489 | } else { |
1490 | // p is a tailored primary. |
1491 | U_ASSERT(s == Collation::COMMON_WEIGHT16); |
1492 | sLimit = rootElements.getSecondaryBoundary(); |
1493 | } |
1494 | if(s == Collation::COMMON_WEIGHT16) { |
1495 | // Do not tailor into the getSortKey() range of |
1496 | // compressed common secondaries. |
1497 | s = rootElements.getLastCommonSecondary(); |
1498 | } |
1499 | secondaries.initForSecondary(); |
1500 | if(!secondaries.allocWeights(s, sLimit, sCount)) { |
1501 | errorCode = U_BUFFER_OVERFLOW_ERROR; |
1502 | errorReason = "secondary tailoring gap too small" ; |
1503 | #ifdef DEBUG_COLLATION_BUILDER |
1504 | printf("!secondaries.allocWeights(%lx, %lx, sCount=%ld)\n" , |
1505 | (long)alignWeightRight(s), (long)alignWeightRight(sLimit), |
1506 | (long)alignWeightRight(sCount)); |
1507 | #endif |
1508 | return; |
1509 | } |
1510 | sIsTailored = TRUE; |
1511 | } |
1512 | s = secondaries.nextWeight(); |
1513 | U_ASSERT(s != 0xffffffff); |
1514 | } else { |
1515 | s = weight16FromNode(node); |
1516 | sIsTailored = FALSE; |
1517 | #ifdef DEBUG_COLLATION_BUILDER |
1518 | printf(" sec %lx\n" , (long)alignWeightRight(s)); |
1519 | #endif |
1520 | } |
1521 | } else /* UCOL_PRIMARY */ { |
1522 | U_ASSERT(isTailoredNode(node)); |
1523 | #ifdef DEBUG_COLLATION_BUILDER |
1524 | printf("pri+ " ); |
1525 | #endif |
1526 | if(!pIsTailored) { |
1527 | // First tailored primary node in this list. |
1528 | int32_t pCount = countTailoredNodes(nodesArray, nextIndex, |
1529 | UCOL_PRIMARY) + 1; |
1530 | UBool isCompressible = baseData->isCompressiblePrimary(p); |
1531 | uint32_t pLimit = |
1532 | rootElements.getPrimaryAfter(p, pIndex, isCompressible); |
1533 | primaries.initForPrimary(isCompressible); |
1534 | if(!primaries.allocWeights(p, pLimit, pCount)) { |
1535 | errorCode = U_BUFFER_OVERFLOW_ERROR; // TODO: introduce a more specific UErrorCode? |
1536 | errorReason = "primary tailoring gap too small" ; |
1537 | return; |
1538 | } |
1539 | pIsTailored = TRUE; |
1540 | } |
1541 | p = primaries.nextWeight(); |
1542 | U_ASSERT(p != 0xffffffff); |
1543 | s = Collation::COMMON_WEIGHT16; |
1544 | sIsTailored = FALSE; |
1545 | } |
1546 | t = s == 0 ? 0 : Collation::COMMON_WEIGHT16; |
1547 | tIsTailored = FALSE; |
1548 | } |
1549 | q = 0; |
1550 | } |
1551 | if(isTailoredNode(node)) { |
1552 | nodesArray[i] = Collation::makeCE(p, s, t, q); |
1553 | #ifdef DEBUG_COLLATION_BUILDER |
1554 | printf("%016llx\n" , (long long)nodesArray[i]); |
1555 | #endif |
1556 | } |
1557 | } |
1558 | } |
1559 | } |
1560 | |
1561 | int32_t |
1562 | CollationBuilder::countTailoredNodes(const int64_t *nodesArray, int32_t i, int32_t strength) { |
1563 | int32_t count = 0; |
1564 | for(;;) { |
1565 | if(i == 0) { break; } |
1566 | int64_t node = nodesArray[i]; |
1567 | if(strengthFromNode(node) < strength) { break; } |
1568 | if(strengthFromNode(node) == strength) { |
1569 | if(isTailoredNode(node)) { |
1570 | ++count; |
1571 | } else { |
1572 | break; |
1573 | } |
1574 | } |
1575 | i = nextIndexFromNode(node); |
1576 | } |
1577 | return count; |
1578 | } |
1579 | |
1580 | class CEFinalizer : public CollationDataBuilder::CEModifier { |
1581 | public: |
1582 | CEFinalizer(const int64_t *ces) : finalCEs(ces) {} |
1583 | virtual ~CEFinalizer(); |
1584 | virtual int64_t modifyCE32(uint32_t ce32) const { |
1585 | U_ASSERT(!Collation::isSpecialCE32(ce32)); |
1586 | if(CollationBuilder::isTempCE32(ce32)) { |
1587 | // retain case bits |
1588 | return finalCEs[CollationBuilder::indexFromTempCE32(ce32)] | ((ce32 & 0xc0) << 8); |
1589 | } else { |
1590 | return Collation::NO_CE; |
1591 | } |
1592 | } |
1593 | virtual int64_t modifyCE(int64_t ce) const { |
1594 | if(CollationBuilder::isTempCE(ce)) { |
1595 | // retain case bits |
1596 | return finalCEs[CollationBuilder::indexFromTempCE(ce)] | (ce & 0xc000); |
1597 | } else { |
1598 | return Collation::NO_CE; |
1599 | } |
1600 | } |
1601 | |
1602 | private: |
1603 | const int64_t *finalCEs; |
1604 | }; |
1605 | |
1606 | CEFinalizer::~CEFinalizer() {} |
1607 | |
1608 | void |
1609 | CollationBuilder::finalizeCEs(UErrorCode &errorCode) { |
1610 | if(U_FAILURE(errorCode)) { return; } |
1611 | LocalPointer<CollationDataBuilder> newBuilder(new CollationDataBuilder(errorCode), errorCode); |
1612 | if(U_FAILURE(errorCode)) { |
1613 | return; |
1614 | } |
1615 | newBuilder->initForTailoring(baseData, errorCode); |
1616 | CEFinalizer finalizer(nodes.getBuffer()); |
1617 | newBuilder->copyFrom(*dataBuilder, finalizer, errorCode); |
1618 | if(U_FAILURE(errorCode)) { return; } |
1619 | delete dataBuilder; |
1620 | dataBuilder = newBuilder.orphan(); |
1621 | } |
1622 | |
1623 | int32_t |
1624 | CollationBuilder::ceStrength(int64_t ce) { |
1625 | return |
1626 | isTempCE(ce) ? strengthFromTempCE(ce) : |
1627 | (ce & INT64_C(0xff00000000000000)) != 0 ? UCOL_PRIMARY : |
1628 | ((uint32_t)ce & 0xff000000) != 0 ? UCOL_SECONDARY : |
1629 | ce != 0 ? UCOL_TERTIARY : |
1630 | UCOL_IDENTICAL; |
1631 | } |
1632 | |
1633 | U_NAMESPACE_END |
1634 | |
1635 | U_NAMESPACE_USE |
1636 | |
1637 | U_CAPI UCollator * U_EXPORT2 |
1638 | ucol_openRules(const UChar *rules, int32_t rulesLength, |
1639 | UColAttributeValue normalizationMode, UCollationStrength strength, |
1640 | UParseError *parseError, UErrorCode *pErrorCode) { |
1641 | if(U_FAILURE(*pErrorCode)) { return NULL; } |
1642 | if(rules == NULL && rulesLength != 0) { |
1643 | *pErrorCode = U_ILLEGAL_ARGUMENT_ERROR; |
1644 | return NULL; |
1645 | } |
1646 | RuleBasedCollator *coll = new RuleBasedCollator(); |
1647 | if(coll == NULL) { |
1648 | *pErrorCode = U_MEMORY_ALLOCATION_ERROR; |
1649 | return NULL; |
1650 | } |
1651 | UnicodeString r((UBool)(rulesLength < 0), rules, rulesLength); |
1652 | coll->internalBuildTailoring(r, strength, normalizationMode, parseError, NULL, *pErrorCode); |
1653 | if(U_FAILURE(*pErrorCode)) { |
1654 | delete coll; |
1655 | return NULL; |
1656 | } |
1657 | return coll->toUCollator(); |
1658 | } |
1659 | |
1660 | static const int32_t internalBufferSize = 512; |
1661 | |
1662 | // The @internal ucol_getUnsafeSet() was moved here from ucol_sit.cpp |
1663 | // because it calls UnicodeSet "builder" code that depends on all Unicode properties, |
1664 | // and the rest of the collation "runtime" code only depends on normalization. |
1665 | // This function is not related to the collation builder, |
1666 | // but it did not seem worth moving it into its own .cpp file, |
1667 | // nor rewriting it to use lower-level UnicodeSet and Normalizer2Impl methods. |
1668 | U_CAPI int32_t U_EXPORT2 |
1669 | ucol_getUnsafeSet( const UCollator *coll, |
1670 | USet *unsafe, |
1671 | UErrorCode *status) |
1672 | { |
1673 | UChar buffer[internalBufferSize]; |
1674 | int32_t len = 0; |
1675 | |
1676 | uset_clear(unsafe); |
1677 | |
1678 | // cccpattern = "[[:^tccc=0:][:^lccc=0:]]", unfortunately variant |
1679 | static const UChar cccpattern[25] = { 0x5b, 0x5b, 0x3a, 0x5e, 0x74, 0x63, 0x63, 0x63, 0x3d, 0x30, 0x3a, 0x5d, |
1680 | 0x5b, 0x3a, 0x5e, 0x6c, 0x63, 0x63, 0x63, 0x3d, 0x30, 0x3a, 0x5d, 0x5d, 0x00 }; |
1681 | |
1682 | // add chars that fail the fcd check |
1683 | uset_applyPattern(unsafe, cccpattern, 24, USET_IGNORE_SPACE, status); |
1684 | |
1685 | // add lead/trail surrogates |
1686 | // (trail surrogates should need to be unsafe only if the caller tests for UTF-16 code *units*, |
1687 | // not when testing code *points*) |
1688 | uset_addRange(unsafe, 0xd800, 0xdfff); |
1689 | |
1690 | USet *contractions = uset_open(0,0); |
1691 | |
1692 | int32_t i = 0, j = 0; |
1693 | ucol_getContractionsAndExpansions(coll, contractions, NULL, FALSE, status); |
1694 | int32_t contsSize = uset_size(contractions); |
1695 | UChar32 c = 0; |
1696 | // Contraction set consists only of strings |
1697 | // to get unsafe code points, we need to |
1698 | // break the strings apart and add them to the unsafe set |
1699 | for(i = 0; i < contsSize; i++) { |
1700 | len = uset_getItem(contractions, i, NULL, NULL, buffer, internalBufferSize, status); |
1701 | if(len > 0) { |
1702 | j = 0; |
1703 | while(j < len) { |
1704 | U16_NEXT(buffer, j, len, c); |
1705 | if(j < len) { |
1706 | uset_add(unsafe, c); |
1707 | } |
1708 | } |
1709 | } |
1710 | } |
1711 | |
1712 | uset_close(contractions); |
1713 | |
1714 | return uset_size(unsafe); |
1715 | } |
1716 | |
1717 | #endif // !UCONFIG_NO_COLLATION |
1718 | |