1 // © 2017 and later: Unicode, Inc. and others.
2 // License & terms of use: http://www.unicode.org/copyright.html
3
4 // edits.cpp
5 // created: 2017feb08 Markus W. Scherer
6
7 #include "unicode/edits.h"
8 #include "unicode/unistr.h"
9 #include "unicode/utypes.h"
10 #include "cmemory.h"
11 #include "uassert.h"
12 #include "util.h"
13
14 U_NAMESPACE_BEGIN
15
16 namespace {
17
18 // 0000uuuuuuuuuuuu records u+1 unchanged text units.
19 const int32_t MAX_UNCHANGED_LENGTH = 0x1000;
20 const int32_t MAX_UNCHANGED = MAX_UNCHANGED_LENGTH - 1;
21
22 // 0mmmnnnccccccccc with m=1..6 records ccc+1 replacements of m:n text units.
23 const int32_t MAX_SHORT_CHANGE_OLD_LENGTH = 6;
24 const int32_t MAX_SHORT_CHANGE_NEW_LENGTH = 7;
25 const int32_t SHORT_CHANGE_NUM_MASK = 0x1ff;
26 const int32_t MAX_SHORT_CHANGE = 0x6fff;
27
28 // 0111mmmmmmnnnnnn records a replacement of m text units with n.
29 // m or n = 61: actual length follows in the next edits array unit.
30 // m or n = 62..63: actual length follows in the next two edits array units.
31 // Bit 30 of the actual length is in the head unit.
32 // Trailing units have bit 15 set.
33 const int32_t LENGTH_IN_1TRAIL = 61;
34 const int32_t LENGTH_IN_2TRAIL = 62;
35
36 } // namespace
37
releaseArray()38 void Edits::releaseArray() U_NOEXCEPT {
39 if (array != stackArray) {
40 uprv_free(array);
41 }
42 }
43
copyArray(const Edits & other)44 Edits &Edits::copyArray(const Edits &other) {
45 if (U_FAILURE(errorCode_)) {
46 length = delta = numChanges = 0;
47 return *this;
48 }
49 if (length > capacity) {
50 uint16_t *newArray = (uint16_t *)uprv_malloc((size_t)length * 2);
51 if (newArray == nullptr) {
52 length = delta = numChanges = 0;
53 errorCode_ = U_MEMORY_ALLOCATION_ERROR;
54 return *this;
55 }
56 releaseArray();
57 array = newArray;
58 capacity = length;
59 }
60 if (length > 0) {
61 uprv_memcpy(array, other.array, (size_t)length * 2);
62 }
63 return *this;
64 }
65
moveArray(Edits & src)66 Edits &Edits::moveArray(Edits &src) U_NOEXCEPT {
67 if (U_FAILURE(errorCode_)) {
68 length = delta = numChanges = 0;
69 return *this;
70 }
71 releaseArray();
72 if (length > STACK_CAPACITY) {
73 array = src.array;
74 capacity = src.capacity;
75 src.array = src.stackArray;
76 src.capacity = STACK_CAPACITY;
77 src.reset();
78 return *this;
79 }
80 array = stackArray;
81 capacity = STACK_CAPACITY;
82 if (length > 0) {
83 uprv_memcpy(array, src.array, (size_t)length * 2);
84 }
85 return *this;
86 }
87
operator =(const Edits & other)88 Edits &Edits::operator=(const Edits &other) {
89 length = other.length;
90 delta = other.delta;
91 numChanges = other.numChanges;
92 errorCode_ = other.errorCode_;
93 return copyArray(other);
94 }
95
operator =(Edits && src)96 Edits &Edits::operator=(Edits &&src) U_NOEXCEPT {
97 length = src.length;
98 delta = src.delta;
99 numChanges = src.numChanges;
100 errorCode_ = src.errorCode_;
101 return moveArray(src);
102 }
103
~Edits()104 Edits::~Edits() {
105 releaseArray();
106 }
107
reset()108 void Edits::reset() U_NOEXCEPT {
109 length = delta = numChanges = 0;
110 errorCode_ = U_ZERO_ERROR;
111 }
112
addUnchanged(int32_t unchangedLength)113 void Edits::addUnchanged(int32_t unchangedLength) {
114 if(U_FAILURE(errorCode_) || unchangedLength == 0) { return; }
115 if(unchangedLength < 0) {
116 errorCode_ = U_ILLEGAL_ARGUMENT_ERROR;
117 return;
118 }
119 // Merge into previous unchanged-text record, if any.
120 int32_t last = lastUnit();
121 if(last < MAX_UNCHANGED) {
122 int32_t remaining = MAX_UNCHANGED - last;
123 if (remaining >= unchangedLength) {
124 setLastUnit(last + unchangedLength);
125 return;
126 }
127 setLastUnit(MAX_UNCHANGED);
128 unchangedLength -= remaining;
129 }
130 // Split large lengths into multiple units.
131 while(unchangedLength >= MAX_UNCHANGED_LENGTH) {
132 append(MAX_UNCHANGED);
133 unchangedLength -= MAX_UNCHANGED_LENGTH;
134 }
135 // Write a small (remaining) length.
136 if(unchangedLength > 0) {
137 append(unchangedLength - 1);
138 }
139 }
140
addReplace(int32_t oldLength,int32_t newLength)141 void Edits::addReplace(int32_t oldLength, int32_t newLength) {
142 if(U_FAILURE(errorCode_)) { return; }
143 if(oldLength < 0 || newLength < 0) {
144 errorCode_ = U_ILLEGAL_ARGUMENT_ERROR;
145 return;
146 }
147 if (oldLength == 0 && newLength == 0) {
148 return;
149 }
150 ++numChanges;
151 int32_t newDelta = newLength - oldLength;
152 if (newDelta != 0) {
153 if ((newDelta > 0 && delta >= 0 && newDelta > (INT32_MAX - delta)) ||
154 (newDelta < 0 && delta < 0 && newDelta < (INT32_MIN - delta))) {
155 // Integer overflow or underflow.
156 errorCode_ = U_INDEX_OUTOFBOUNDS_ERROR;
157 return;
158 }
159 delta += newDelta;
160 }
161
162 if(0 < oldLength && oldLength <= MAX_SHORT_CHANGE_OLD_LENGTH &&
163 newLength <= MAX_SHORT_CHANGE_NEW_LENGTH) {
164 // Merge into previous same-lengths short-replacement record, if any.
165 int32_t u = (oldLength << 12) | (newLength << 9);
166 int32_t last = lastUnit();
167 if(MAX_UNCHANGED < last && last < MAX_SHORT_CHANGE &&
168 (last & ~SHORT_CHANGE_NUM_MASK) == u &&
169 (last & SHORT_CHANGE_NUM_MASK) < SHORT_CHANGE_NUM_MASK) {
170 setLastUnit(last + 1);
171 return;
172 }
173 append(u);
174 return;
175 }
176
177 int32_t head = 0x7000;
178 if (oldLength < LENGTH_IN_1TRAIL && newLength < LENGTH_IN_1TRAIL) {
179 head |= oldLength << 6;
180 head |= newLength;
181 append(head);
182 } else if ((capacity - length) >= 5 || growArray()) {
183 int32_t limit = length + 1;
184 if(oldLength < LENGTH_IN_1TRAIL) {
185 head |= oldLength << 6;
186 } else if(oldLength <= 0x7fff) {
187 head |= LENGTH_IN_1TRAIL << 6;
188 array[limit++] = (uint16_t)(0x8000 | oldLength);
189 } else {
190 head |= (LENGTH_IN_2TRAIL + (oldLength >> 30)) << 6;
191 array[limit++] = (uint16_t)(0x8000 | (oldLength >> 15));
192 array[limit++] = (uint16_t)(0x8000 | oldLength);
193 }
194 if(newLength < LENGTH_IN_1TRAIL) {
195 head |= newLength;
196 } else if(newLength <= 0x7fff) {
197 head |= LENGTH_IN_1TRAIL;
198 array[limit++] = (uint16_t)(0x8000 | newLength);
199 } else {
200 head |= LENGTH_IN_2TRAIL + (newLength >> 30);
201 array[limit++] = (uint16_t)(0x8000 | (newLength >> 15));
202 array[limit++] = (uint16_t)(0x8000 | newLength);
203 }
204 array[length] = (uint16_t)head;
205 length = limit;
206 }
207 }
208
append(int32_t r)209 void Edits::append(int32_t r) {
210 if(length < capacity || growArray()) {
211 array[length++] = (uint16_t)r;
212 }
213 }
214
growArray()215 UBool Edits::growArray() {
216 int32_t newCapacity;
217 if (array == stackArray) {
218 newCapacity = 2000;
219 } else if (capacity == INT32_MAX) {
220 // Not U_BUFFER_OVERFLOW_ERROR because that could be confused on a string transform API
221 // with a result-string-buffer overflow.
222 errorCode_ = U_INDEX_OUTOFBOUNDS_ERROR;
223 return FALSE;
224 } else if (capacity >= (INT32_MAX / 2)) {
225 newCapacity = INT32_MAX;
226 } else {
227 newCapacity = 2 * capacity;
228 }
229 // Grow by at least 5 units so that a maximal change record will fit.
230 if ((newCapacity - capacity) < 5) {
231 errorCode_ = U_INDEX_OUTOFBOUNDS_ERROR;
232 return FALSE;
233 }
234 uint16_t *newArray = (uint16_t *)uprv_malloc((size_t)newCapacity * 2);
235 if (newArray == NULL) {
236 errorCode_ = U_MEMORY_ALLOCATION_ERROR;
237 return FALSE;
238 }
239 uprv_memcpy(newArray, array, (size_t)length * 2);
240 releaseArray();
241 array = newArray;
242 capacity = newCapacity;
243 return TRUE;
244 }
245
copyErrorTo(UErrorCode & outErrorCode)246 UBool Edits::copyErrorTo(UErrorCode &outErrorCode) {
247 if (U_FAILURE(outErrorCode)) { return TRUE; }
248 if (U_SUCCESS(errorCode_)) { return FALSE; }
249 outErrorCode = errorCode_;
250 return TRUE;
251 }
252
mergeAndAppend(const Edits & ab,const Edits & bc,UErrorCode & errorCode)253 Edits &Edits::mergeAndAppend(const Edits &ab, const Edits &bc, UErrorCode &errorCode) {
254 if (copyErrorTo(errorCode)) { return *this; }
255 // Picture string a --(Edits ab)--> string b --(Edits bc)--> string c.
256 // Parallel iteration over both Edits.
257 Iterator abIter = ab.getFineIterator();
258 Iterator bcIter = bc.getFineIterator();
259 UBool abHasNext = TRUE, bcHasNext = TRUE;
260 // Copy iterator state into local variables, so that we can modify and subdivide spans.
261 // ab old & new length, bc old & new length
262 int32_t aLength = 0, ab_bLength = 0, bc_bLength = 0, cLength = 0;
263 // When we have different-intermediate-length changes, we accumulate a larger change.
264 int32_t pending_aLength = 0, pending_cLength = 0;
265 for (;;) {
266 // At this point, for each of the two iterators:
267 // Either we are done with the locally cached current edit,
268 // and its intermediate-string length has been reset,
269 // or we will continue to work with a truncated remainder of this edit.
270 //
271 // If the current edit is done, and the iterator has not yet reached the end,
272 // then we fetch the next edit. This is true for at least one of the iterators.
273 //
274 // Normally it does not matter whether we fetch from ab and then bc or vice versa.
275 // However, the result is observably different when
276 // ab deletions meet bc insertions at the same intermediate-string index.
277 // Some users expect the bc insertions to come first, so we fetch from bc first.
278 if (bc_bLength == 0) {
279 if (bcHasNext && (bcHasNext = bcIter.next(errorCode)) != 0) {
280 bc_bLength = bcIter.oldLength();
281 cLength = bcIter.newLength();
282 if (bc_bLength == 0) {
283 // insertion
284 if (ab_bLength == 0 || !abIter.hasChange()) {
285 addReplace(pending_aLength, pending_cLength + cLength);
286 pending_aLength = pending_cLength = 0;
287 } else {
288 pending_cLength += cLength;
289 }
290 continue;
291 }
292 }
293 // else see if the other iterator is done, too.
294 }
295 if (ab_bLength == 0) {
296 if (abHasNext && (abHasNext = abIter.next(errorCode)) != 0) {
297 aLength = abIter.oldLength();
298 ab_bLength = abIter.newLength();
299 if (ab_bLength == 0) {
300 // deletion
301 if (bc_bLength == bcIter.oldLength() || !bcIter.hasChange()) {
302 addReplace(pending_aLength + aLength, pending_cLength);
303 pending_aLength = pending_cLength = 0;
304 } else {
305 pending_aLength += aLength;
306 }
307 continue;
308 }
309 } else if (bc_bLength == 0) {
310 // Both iterators are done at the same time:
311 // The intermediate-string lengths match.
312 break;
313 } else {
314 // The ab output string is shorter than the bc input string.
315 if (!copyErrorTo(errorCode)) {
316 errorCode = U_ILLEGAL_ARGUMENT_ERROR;
317 }
318 return *this;
319 }
320 }
321 if (bc_bLength == 0) {
322 // The bc input string is shorter than the ab output string.
323 if (!copyErrorTo(errorCode)) {
324 errorCode = U_ILLEGAL_ARGUMENT_ERROR;
325 }
326 return *this;
327 }
328 // Done fetching: ab_bLength > 0 && bc_bLength > 0
329
330 // The current state has two parts:
331 // - Past: We accumulate a longer ac edit in the "pending" variables.
332 // - Current: We have copies of the current ab/bc edits in local variables.
333 // At least one side is newly fetched.
334 // One side might be a truncated remainder of an edit we fetched earlier.
335
336 if (!abIter.hasChange() && !bcIter.hasChange()) {
337 // An unchanged span all the way from string a to string c.
338 if (pending_aLength != 0 || pending_cLength != 0) {
339 addReplace(pending_aLength, pending_cLength);
340 pending_aLength = pending_cLength = 0;
341 }
342 int32_t unchangedLength = aLength <= cLength ? aLength : cLength;
343 addUnchanged(unchangedLength);
344 ab_bLength = aLength -= unchangedLength;
345 bc_bLength = cLength -= unchangedLength;
346 // At least one of the unchanged spans is now empty.
347 continue;
348 }
349 if (!abIter.hasChange() && bcIter.hasChange()) {
350 // Unchanged a->b but changed b->c.
351 if (ab_bLength >= bc_bLength) {
352 // Split the longer unchanged span into change + remainder.
353 addReplace(pending_aLength + bc_bLength, pending_cLength + cLength);
354 pending_aLength = pending_cLength = 0;
355 aLength = ab_bLength -= bc_bLength;
356 bc_bLength = 0;
357 continue;
358 }
359 // Handle the shorter unchanged span below like a change.
360 } else if (abIter.hasChange() && !bcIter.hasChange()) {
361 // Changed a->b and then unchanged b->c.
362 if (ab_bLength <= bc_bLength) {
363 // Split the longer unchanged span into change + remainder.
364 addReplace(pending_aLength + aLength, pending_cLength + ab_bLength);
365 pending_aLength = pending_cLength = 0;
366 cLength = bc_bLength -= ab_bLength;
367 ab_bLength = 0;
368 continue;
369 }
370 // Handle the shorter unchanged span below like a change.
371 } else { // both abIter.hasChange() && bcIter.hasChange()
372 if (ab_bLength == bc_bLength) {
373 // Changes on both sides up to the same position. Emit & reset.
374 addReplace(pending_aLength + aLength, pending_cLength + cLength);
375 pending_aLength = pending_cLength = 0;
376 ab_bLength = bc_bLength = 0;
377 continue;
378 }
379 }
380 // Accumulate the a->c change, reset the shorter side,
381 // keep a remainder of the longer one.
382 pending_aLength += aLength;
383 pending_cLength += cLength;
384 if (ab_bLength < bc_bLength) {
385 bc_bLength -= ab_bLength;
386 cLength = ab_bLength = 0;
387 } else { // ab_bLength > bc_bLength
388 ab_bLength -= bc_bLength;
389 aLength = bc_bLength = 0;
390 }
391 }
392 if (pending_aLength != 0 || pending_cLength != 0) {
393 addReplace(pending_aLength, pending_cLength);
394 }
395 copyErrorTo(errorCode);
396 return *this;
397 }
398
Iterator(const uint16_t * a,int32_t len,UBool oc,UBool crs)399 Edits::Iterator::Iterator(const uint16_t *a, int32_t len, UBool oc, UBool crs) :
400 array(a), index(0), length(len), remaining(0),
401 onlyChanges_(oc), coarse(crs),
402 dir(0), changed(FALSE), oldLength_(0), newLength_(0),
403 srcIndex(0), replIndex(0), destIndex(0) {}
404
readLength(int32_t head)405 int32_t Edits::Iterator::readLength(int32_t head) {
406 if (head < LENGTH_IN_1TRAIL) {
407 return head;
408 } else if (head < LENGTH_IN_2TRAIL) {
409 U_ASSERT(index < length);
410 U_ASSERT(array[index] >= 0x8000);
411 return array[index++] & 0x7fff;
412 } else {
413 U_ASSERT((index + 2) <= length);
414 U_ASSERT(array[index] >= 0x8000);
415 U_ASSERT(array[index + 1] >= 0x8000);
416 int32_t len = ((head & 1) << 30) |
417 ((int32_t)(array[index] & 0x7fff) << 15) |
418 (array[index + 1] & 0x7fff);
419 index += 2;
420 return len;
421 }
422 }
423
updateNextIndexes()424 void Edits::Iterator::updateNextIndexes() {
425 srcIndex += oldLength_;
426 if (changed) {
427 replIndex += newLength_;
428 }
429 destIndex += newLength_;
430 }
431
updatePreviousIndexes()432 void Edits::Iterator::updatePreviousIndexes() {
433 srcIndex -= oldLength_;
434 if (changed) {
435 replIndex -= newLength_;
436 }
437 destIndex -= newLength_;
438 }
439
noNext()440 UBool Edits::Iterator::noNext() {
441 // No change before or beyond the string.
442 dir = 0;
443 changed = FALSE;
444 oldLength_ = newLength_ = 0;
445 return FALSE;
446 }
447
next(UBool onlyChanges,UErrorCode & errorCode)448 UBool Edits::Iterator::next(UBool onlyChanges, UErrorCode &errorCode) {
449 // Forward iteration: Update the string indexes to the limit of the current span,
450 // and post-increment-read array units to assemble a new span.
451 // Leaves the array index one after the last unit of that span.
452 if (U_FAILURE(errorCode)) { return FALSE; }
453 // We have an errorCode in case we need to start guarding against integer overflows.
454 // It is also convenient for caller loops if we bail out when an error was set elsewhere.
455 if (dir > 0) {
456 updateNextIndexes();
457 } else {
458 if (dir < 0) {
459 // Turn around from previous() to next().
460 // Post-increment-read the same span again.
461 if (remaining > 0) {
462 // Fine-grained iterator:
463 // Stay on the current one of a sequence of compressed changes.
464 ++index; // next() rests on the index after the sequence unit.
465 dir = 1;
466 return TRUE;
467 }
468 }
469 dir = 1;
470 }
471 if (remaining >= 1) {
472 // Fine-grained iterator: Continue a sequence of compressed changes.
473 if (remaining > 1) {
474 --remaining;
475 return TRUE;
476 }
477 remaining = 0;
478 }
479 if (index >= length) {
480 return noNext();
481 }
482 int32_t u = array[index++];
483 if (u <= MAX_UNCHANGED) {
484 // Combine adjacent unchanged ranges.
485 changed = FALSE;
486 oldLength_ = u + 1;
487 while (index < length && (u = array[index]) <= MAX_UNCHANGED) {
488 ++index;
489 oldLength_ += u + 1;
490 }
491 newLength_ = oldLength_;
492 if (onlyChanges) {
493 updateNextIndexes();
494 if (index >= length) {
495 return noNext();
496 }
497 // already fetched u > MAX_UNCHANGED at index
498 ++index;
499 } else {
500 return TRUE;
501 }
502 }
503 changed = TRUE;
504 if (u <= MAX_SHORT_CHANGE) {
505 int32_t oldLen = u >> 12;
506 int32_t newLen = (u >> 9) & MAX_SHORT_CHANGE_NEW_LENGTH;
507 int32_t num = (u & SHORT_CHANGE_NUM_MASK) + 1;
508 if (coarse) {
509 oldLength_ = num * oldLen;
510 newLength_ = num * newLen;
511 } else {
512 // Split a sequence of changes that was compressed into one unit.
513 oldLength_ = oldLen;
514 newLength_ = newLen;
515 if (num > 1) {
516 remaining = num; // This is the first of two or more changes.
517 }
518 return TRUE;
519 }
520 } else {
521 U_ASSERT(u <= 0x7fff);
522 oldLength_ = readLength((u >> 6) & 0x3f);
523 newLength_ = readLength(u & 0x3f);
524 if (!coarse) {
525 return TRUE;
526 }
527 }
528 // Combine adjacent changes.
529 while (index < length && (u = array[index]) > MAX_UNCHANGED) {
530 ++index;
531 if (u <= MAX_SHORT_CHANGE) {
532 int32_t num = (u & SHORT_CHANGE_NUM_MASK) + 1;
533 oldLength_ += (u >> 12) * num;
534 newLength_ += ((u >> 9) & MAX_SHORT_CHANGE_NEW_LENGTH) * num;
535 } else {
536 U_ASSERT(u <= 0x7fff);
537 oldLength_ += readLength((u >> 6) & 0x3f);
538 newLength_ += readLength(u & 0x3f);
539 }
540 }
541 return TRUE;
542 }
543
previous(UErrorCode & errorCode)544 UBool Edits::Iterator::previous(UErrorCode &errorCode) {
545 // Backward iteration: Pre-decrement-read array units to assemble a new span,
546 // then update the string indexes to the start of that span.
547 // Leaves the array index on the head unit of that span.
548 if (U_FAILURE(errorCode)) { return FALSE; }
549 // We have an errorCode in case we need to start guarding against integer overflows.
550 // It is also convenient for caller loops if we bail out when an error was set elsewhere.
551 if (dir >= 0) {
552 if (dir > 0) {
553 // Turn around from next() to previous().
554 // Set the string indexes to the span limit and
555 // pre-decrement-read the same span again.
556 if (remaining > 0) {
557 // Fine-grained iterator:
558 // Stay on the current one of a sequence of compressed changes.
559 --index; // previous() rests on the sequence unit.
560 dir = -1;
561 return TRUE;
562 }
563 updateNextIndexes();
564 }
565 dir = -1;
566 }
567 if (remaining > 0) {
568 // Fine-grained iterator: Continue a sequence of compressed changes.
569 int32_t u = array[index];
570 U_ASSERT(MAX_UNCHANGED < u && u <= MAX_SHORT_CHANGE);
571 if (remaining <= (u & SHORT_CHANGE_NUM_MASK)) {
572 ++remaining;
573 updatePreviousIndexes();
574 return TRUE;
575 }
576 remaining = 0;
577 }
578 if (index <= 0) {
579 return noNext();
580 }
581 int32_t u = array[--index];
582 if (u <= MAX_UNCHANGED) {
583 // Combine adjacent unchanged ranges.
584 changed = FALSE;
585 oldLength_ = u + 1;
586 while (index > 0 && (u = array[index - 1]) <= MAX_UNCHANGED) {
587 --index;
588 oldLength_ += u + 1;
589 }
590 newLength_ = oldLength_;
591 // No need to handle onlyChanges as long as previous() is called only from findIndex().
592 updatePreviousIndexes();
593 return TRUE;
594 }
595 changed = TRUE;
596 if (u <= MAX_SHORT_CHANGE) {
597 int32_t oldLen = u >> 12;
598 int32_t newLen = (u >> 9) & MAX_SHORT_CHANGE_NEW_LENGTH;
599 int32_t num = (u & SHORT_CHANGE_NUM_MASK) + 1;
600 if (coarse) {
601 oldLength_ = num * oldLen;
602 newLength_ = num * newLen;
603 } else {
604 // Split a sequence of changes that was compressed into one unit.
605 oldLength_ = oldLen;
606 newLength_ = newLen;
607 if (num > 1) {
608 remaining = 1; // This is the last of two or more changes.
609 }
610 updatePreviousIndexes();
611 return TRUE;
612 }
613 } else {
614 if (u <= 0x7fff) {
615 // The change is encoded in u alone.
616 oldLength_ = readLength((u >> 6) & 0x3f);
617 newLength_ = readLength(u & 0x3f);
618 } else {
619 // Back up to the head of the change, read the lengths,
620 // and reset the index to the head again.
621 U_ASSERT(index > 0);
622 while ((u = array[--index]) > 0x7fff) {}
623 U_ASSERT(u > MAX_SHORT_CHANGE);
624 int32_t headIndex = index++;
625 oldLength_ = readLength((u >> 6) & 0x3f);
626 newLength_ = readLength(u & 0x3f);
627 index = headIndex;
628 }
629 if (!coarse) {
630 updatePreviousIndexes();
631 return TRUE;
632 }
633 }
634 // Combine adjacent changes.
635 while (index > 0 && (u = array[index - 1]) > MAX_UNCHANGED) {
636 --index;
637 if (u <= MAX_SHORT_CHANGE) {
638 int32_t num = (u & SHORT_CHANGE_NUM_MASK) + 1;
639 oldLength_ += (u >> 12) * num;
640 newLength_ += ((u >> 9) & MAX_SHORT_CHANGE_NEW_LENGTH) * num;
641 } else if (u <= 0x7fff) {
642 // Read the lengths, and reset the index to the head again.
643 int32_t headIndex = index++;
644 oldLength_ += readLength((u >> 6) & 0x3f);
645 newLength_ += readLength(u & 0x3f);
646 index = headIndex;
647 }
648 }
649 updatePreviousIndexes();
650 return TRUE;
651 }
652
findIndex(int32_t i,UBool findSource,UErrorCode & errorCode)653 int32_t Edits::Iterator::findIndex(int32_t i, UBool findSource, UErrorCode &errorCode) {
654 if (U_FAILURE(errorCode) || i < 0) { return -1; }
655 int32_t spanStart, spanLength;
656 if (findSource) { // find source index
657 spanStart = srcIndex;
658 spanLength = oldLength_;
659 } else { // find destination index
660 spanStart = destIndex;
661 spanLength = newLength_;
662 }
663 if (i < spanStart) {
664 if (i >= (spanStart / 2)) {
665 // Search backwards.
666 for (;;) {
667 UBool hasPrevious = previous(errorCode);
668 U_ASSERT(hasPrevious); // because i>=0 and the first span starts at 0
669 (void)hasPrevious; // avoid unused-variable warning
670 spanStart = findSource ? srcIndex : destIndex;
671 if (i >= spanStart) {
672 // The index is in the current span.
673 return 0;
674 }
675 if (remaining > 0) {
676 // Is the index in one of the remaining compressed edits?
677 // spanStart is the start of the current span, first of the remaining ones.
678 spanLength = findSource ? oldLength_ : newLength_;
679 int32_t u = array[index];
680 U_ASSERT(MAX_UNCHANGED < u && u <= MAX_SHORT_CHANGE);
681 int32_t num = (u & SHORT_CHANGE_NUM_MASK) + 1 - remaining;
682 int32_t len = num * spanLength;
683 if (i >= (spanStart - len)) {
684 int32_t n = ((spanStart - i - 1) / spanLength) + 1;
685 // 1 <= n <= num
686 srcIndex -= n * oldLength_;
687 replIndex -= n * newLength_;
688 destIndex -= n * newLength_;
689 remaining += n;
690 return 0;
691 }
692 // Skip all of these edits at once.
693 srcIndex -= num * oldLength_;
694 replIndex -= num * newLength_;
695 destIndex -= num * newLength_;
696 remaining = 0;
697 }
698 }
699 }
700 // Reset the iterator to the start.
701 dir = 0;
702 index = remaining = oldLength_ = newLength_ = srcIndex = replIndex = destIndex = 0;
703 } else if (i < (spanStart + spanLength)) {
704 // The index is in the current span.
705 return 0;
706 }
707 while (next(FALSE, errorCode)) {
708 if (findSource) {
709 spanStart = srcIndex;
710 spanLength = oldLength_;
711 } else {
712 spanStart = destIndex;
713 spanLength = newLength_;
714 }
715 if (i < (spanStart + spanLength)) {
716 // The index is in the current span.
717 return 0;
718 }
719 if (remaining > 1) {
720 // Is the index in one of the remaining compressed edits?
721 // spanStart is the start of the current span, first of the remaining ones.
722 int32_t len = remaining * spanLength;
723 if (i < (spanStart + len)) {
724 int32_t n = (i - spanStart) / spanLength; // 1 <= n <= remaining - 1
725 srcIndex += n * oldLength_;
726 replIndex += n * newLength_;
727 destIndex += n * newLength_;
728 remaining -= n;
729 return 0;
730 }
731 // Make next() skip all of these edits at once.
732 oldLength_ *= remaining;
733 newLength_ *= remaining;
734 remaining = 0;
735 }
736 }
737 return 1;
738 }
739
destinationIndexFromSourceIndex(int32_t i,UErrorCode & errorCode)740 int32_t Edits::Iterator::destinationIndexFromSourceIndex(int32_t i, UErrorCode &errorCode) {
741 int32_t where = findIndex(i, TRUE, errorCode);
742 if (where < 0) {
743 // Error or before the string.
744 return 0;
745 }
746 if (where > 0 || i == srcIndex) {
747 // At or after string length, or at start of the found span.
748 return destIndex;
749 }
750 if (changed) {
751 // In a change span, map to its end.
752 return destIndex + newLength_;
753 } else {
754 // In an unchanged span, offset 1:1 within it.
755 return destIndex + (i - srcIndex);
756 }
757 }
758
sourceIndexFromDestinationIndex(int32_t i,UErrorCode & errorCode)759 int32_t Edits::Iterator::sourceIndexFromDestinationIndex(int32_t i, UErrorCode &errorCode) {
760 int32_t where = findIndex(i, FALSE, errorCode);
761 if (where < 0) {
762 // Error or before the string.
763 return 0;
764 }
765 if (where > 0 || i == destIndex) {
766 // At or after string length, or at start of the found span.
767 return srcIndex;
768 }
769 if (changed) {
770 // In a change span, map to its end.
771 return srcIndex + oldLength_;
772 } else {
773 // In an unchanged span, offset within it.
774 return srcIndex + (i - destIndex);
775 }
776 }
777
toString(UnicodeString & sb) const778 UnicodeString& Edits::Iterator::toString(UnicodeString& sb) const {
779 sb.append(u"{ src[", -1);
780 ICU_Utility::appendNumber(sb, srcIndex);
781 sb.append(u"..", -1);
782 ICU_Utility::appendNumber(sb, srcIndex + oldLength_);
783 if (changed) {
784 sb.append(u"] ⇝ dest[", -1);
785 } else {
786 sb.append(u"] ≡ dest[", -1);
787 }
788 ICU_Utility::appendNumber(sb, destIndex);
789 sb.append(u"..", -1);
790 ICU_Utility::appendNumber(sb, destIndex + newLength_);
791 if (changed) {
792 sb.append(u"], repl[", -1);
793 ICU_Utility::appendNumber(sb, replIndex);
794 sb.append(u"..", -1);
795 ICU_Utility::appendNumber(sb, replIndex + newLength_);
796 sb.append(u"] }", -1);
797 } else {
798 sb.append(u"] (no-change) }", -1);
799 }
800 return sb;
801 }
802
803 U_NAMESPACE_END
804