lib/core: Allow comparison operators to work on any subclass of FlatText
[nit.git] / lib / core / text / flat.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # This file is free software, which comes along with NIT. This software is
4 # distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
5 # without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
6 # PARTICULAR PURPOSE. You can modify it is you want, provided this header
7 # is kept unaltered, and a notification of the changes is added.
8 # You are allowed to redistribute it and sell it, alone or is a part of
9 # another product.
10
11 # All the array-based text representations
12 module flat
13
14 intrude import abstract_text
15 intrude import native
16
17 `{
18 #include <stdio.h>
19 #include <string.h>
20 `}
21
22 private class FlatSubstringsIter
23 super Iterator[FlatText]
24
25 var tgt: nullable FlatText
26
27 redef fun item do
28 assert is_ok
29 return tgt.as(not null)
30 end
31
32 redef fun is_ok do return tgt != null
33
34 redef fun next do tgt = null
35 end
36
37 redef class FlatText
38
39 fun first_byte: Int do return 0
40
41 fun last_byte: Int do return _bytelen - 1
42
43 # Cache of the latest position (char) explored in the string
44 var position: Int = 0
45
46 # Cached position (bytes) in the NativeString underlying the String
47 var bytepos: Int = 0
48
49 # Index of the character `index` in `_items`
50 fun char_to_byte_index(index: Int): Int do
51 var ln = length
52 assert index >= 0
53 assert index < ln
54
55 var pos = _position
56 # Find best insertion point
57 var delta_begin = index
58 var delta_end = (ln - 1) - index
59 var delta_cache = (pos - index).abs
60 var min = delta_begin
61 var its = _items
62
63 if delta_cache < min then min = delta_cache
64 if delta_end < min then min = delta_end
65
66 var ns_i: Int
67 var my_i: Int
68
69 if min == delta_begin then
70 ns_i = first_byte
71 my_i = 0
72 else if min == delta_cache then
73 ns_i = _bytepos
74 my_i = pos
75 else
76 ns_i = its.find_beginning_of_char_at(last_byte)
77 my_i = length - 1
78 end
79
80 ns_i = its.char_to_byte_index_cached(index, my_i, ns_i)
81
82 _position = index
83 _bytepos = ns_i
84
85 return ns_i
86 end
87
88 # By escaping `self` to HTML, how many more bytes will be needed ?
89 fun chars_to_html_escape: Int do
90 var its = _items
91 var max = last_byte
92 var pos = first_byte
93 var endlen = 0
94 while pos <= max do
95 var c = its[pos]
96 if c == 0x3Cu8 then
97 endlen += 3
98 else if c == 0x3Eu8 then
99 endlen += 3
100 else if c == 0x26u8 then
101 endlen += 4
102 else if c == 0x22u8 then
103 endlen += 4
104 else if c == 0x27u8 then
105 endlen += 4
106 else if c == 0x2Fu8 then
107 endlen += 4
108 end
109 pos += 1
110 end
111 return endlen
112 end
113
114 redef fun html_escape
115 do
116 var extra = chars_to_html_escape
117 if extra == 0 then return to_s
118 var its = _items
119 var max = last_byte
120 var pos = first_byte
121 var nlen = extra + _bytelen
122 var nits = new NativeString(nlen)
123 var outpos = 0
124 while pos <= max do
125 var c = its[pos]
126 # Special codes:
127 # Some HTML characters are used as meta-data, they need
128 # to be replaced by an HTML-Escaped equivalent
129 #
130 # * 0x3C (<) => &lt;
131 # * 0x3E (>) => &gt;
132 # * 0x26 (&) => &amp;
133 # * 0x22 (") => &#34;
134 # * 0x27 (') => &#39;
135 # * 0x2F (/) => &#47;
136 if c == 0x3Cu8 then
137 nits[outpos] = 0x26u8
138 nits[outpos + 1] = 0x6Cu8
139 nits[outpos + 2] = 0x74u8
140 nits[outpos + 3] = 0x3Bu8
141 outpos += 4
142 else if c == 0x3Eu8 then
143 nits[outpos] = 0x26u8
144 nits[outpos + 1] = 0x67u8
145 nits[outpos + 2] = 0x74u8
146 nits[outpos + 3] = 0x3Bu8
147 outpos += 4
148 else if c == 0x26u8 then
149 nits[outpos] = 0x26u8
150 nits[outpos + 1] = 0x61u8
151 nits[outpos + 2] = 0x6Du8
152 nits[outpos + 3] = 0x70u8
153 nits[outpos + 4] = 0x3Bu8
154 outpos += 5
155 else if c == 0x22u8 then
156 nits[outpos] = 0x26u8
157 nits[outpos + 1] = 0x23u8
158 nits[outpos + 2] = 0x33u8
159 nits[outpos + 3] = 0x34u8
160 nits[outpos + 4] = 0x3Bu8
161 outpos += 5
162 else if c == 0x27u8 then
163 nits[outpos] = 0x26u8
164 nits[outpos + 1] = 0x23u8
165 nits[outpos + 2] = 0x33u8
166 nits[outpos + 3] = 0x39u8
167 nits[outpos + 4] = 0x3Bu8
168 outpos += 5
169 else if c == 0x2Fu8 then
170 nits[outpos] = 0x26u8
171 nits[outpos + 1] = 0x23u8
172 nits[outpos + 2] = 0x34u8
173 nits[outpos + 3] = 0x37u8
174 nits[outpos + 4] = 0x3Bu8
175 outpos += 5
176 else
177 nits[outpos] = c
178 outpos += 1
179 end
180 pos += 1
181 end
182 var s = new FlatString.with_infos(nits, nlen, 0, nlen - 1)
183 return s
184 end
185
186 # By escaping `self` to C, how many more bytes will be needed ?
187 #
188 # This enables a double-optimization in `escape_to_c` since if this
189 # method returns 0, then `self` does not need escaping and can be
190 # returned as-is
191 fun chars_to_escape_to_c: Int do
192 var its = _items
193 var max = last_byte
194 var pos = first_byte
195 var req_esc = 0
196 while pos <= max do
197 var c = its[pos]
198 if c == 0x0Au8 then
199 req_esc += 1
200 else if c == 0x09u8 then
201 req_esc += 1
202 else if c == 0x22u8 then
203 req_esc += 1
204 else if c == 0x27u8 then
205 req_esc += 1
206 else if c == 0x5Cu8 then
207 req_esc += 1
208 else if c < 32u8 then
209 req_esc += 3
210 end
211 pos += 1
212 end
213 return req_esc
214 end
215
216 redef fun escape_to_c do
217 var ln_extra = chars_to_escape_to_c
218 if ln_extra == 0 then return self.to_s
219 var its = _items
220 var max = last_byte
221 var nlen = _bytelen + ln_extra
222 var nns = new NativeString(nlen)
223 var pos = first_byte
224 var opos = 0
225 while pos <= max do
226 var c = its[pos]
227 # Special codes:
228 #
229 # Any byte with value < 32 is a control character
230 # All their uses will be replaced by their octal
231 # value in C.
232 #
233 # There are two exceptions however:
234 #
235 # * 0x09 => \t
236 # * 0x0A => \n
237 #
238 # Aside from the code points above, the following are:
239 #
240 # * 0x22 => \"
241 # * 0x27 => \'
242 # * 0x5C => \\
243 if c == 0x09u8 then
244 nns[opos] = 0x5Cu8
245 nns[opos + 1] = 0x74u8
246 opos += 2
247 else if c == 0x0Au8 then
248 nns[opos] = 0x5Cu8
249 nns[opos + 1] = 0x6Eu8
250 opos += 2
251 else if c == 0x22u8 then
252 nns[opos] = 0x5Cu8
253 nns[opos + 1] = 0x22u8
254 opos += 2
255 else if c == 0x27u8 then
256 nns[opos] = 0x5Cu8
257 nns[opos + 1] = 0x27u8
258 opos += 2
259 else if c == 0x5Cu8 then
260 nns[opos] = 0x5Cu8
261 nns[opos + 1] = 0x5Cu8
262 opos += 2
263 else if c < 32u8 then
264 nns[opos] = 0x5Cu8
265 nns[opos + 1] = 0x30u8
266 nns[opos + 2] = ((c & 0x38u8) >> 3) + 0x30u8
267 nns[opos + 3] = (c & 0x07u8) + 0x30u8
268 opos += 4
269 else
270 nns[opos] = c
271 opos += 1
272 end
273 pos += 1
274 end
275 return nns.to_s_with_length(nlen)
276 end
277
278 redef fun [](index) do return _items.char_at(char_to_byte_index(index))
279 end
280
281 # Immutable strings of characters.
282 class FlatString
283 super FlatText
284 super String
285
286 # Index at which `self` begins in `_items`, inclusively
287 redef var first_byte is noinit
288
289 # Index at which `self` ends in `_items`, inclusively
290 redef var last_byte is noinit
291
292 redef var chars = new FlatStringCharView(self) is lazy
293
294 redef var bytes = new FlatStringByteView(self) is lazy
295
296 redef var length is lazy do
297 if _bytelen == 0 then return 0
298 return _items.utf8_length(_first_byte, _last_byte)
299 end
300
301 redef var to_cstring is lazy do
302 var blen = _bytelen
303 var new_items = new NativeString(blen + 1)
304 _items.copy_to(new_items, blen, _first_byte, 0)
305 new_items[blen] = 0u8
306 return new_items
307 end
308
309 redef fun reversed
310 do
311 var b = new FlatBuffer.with_capacity(_bytelen + 1)
312 for i in [length - 1 .. 0].step(-1) do
313 b.add self[i]
314 end
315 var s = b.to_s.as(FlatString)
316 s.length = self.length
317 return s
318 end
319
320 redef fun fast_cstring do return _items.fast_cstring(_first_byte)
321
322 redef fun substring(from, count)
323 do
324 assert count >= 0
325
326 if from < 0 then
327 count += from
328 if count < 0 then count = 0
329 from = 0
330 end
331
332 if (count + from) > length then count = length - from
333 if count <= 0 then return ""
334 var end_index = from + count - 1
335
336 var bytefrom = char_to_byte_index(from)
337 var byteto = char_to_byte_index(end_index)
338 var its = _items
339 byteto += its.length_of_char_at(byteto) - 1
340
341 var s = new FlatString.full(its, byteto - bytefrom + 1, bytefrom, byteto, count)
342 return s
343 end
344
345 redef fun empty do return "".as(FlatString)
346
347 redef fun to_upper
348 do
349 var outstr = new FlatBuffer.with_capacity(self._bytelen + 1)
350
351 var mylen = length
352 var pos = 0
353
354 while pos < mylen do
355 outstr.add(chars[pos].to_upper)
356 pos += 1
357 end
358
359 return outstr.to_s
360 end
361
362 redef fun to_lower
363 do
364 var outstr = new FlatBuffer.with_capacity(self._bytelen + 1)
365
366 var mylen = length
367 var pos = 0
368
369 while pos < mylen do
370 outstr.add(chars[pos].to_lower)
371 pos += 1
372 end
373
374 return outstr.to_s
375 end
376
377 redef fun output
378 do
379 for i in chars do i.output
380 end
381
382 ##################################################
383 # String Specific Methods #
384 ##################################################
385
386 # Low-level creation of a new string with minimal data.
387 #
388 # `_items` will be used as is, without copy, to retrieve the characters of the string.
389 # Aliasing issues is the responsibility of the caller.
390 private init with_infos(items: NativeString, bytelen, from, to: Int)
391 do
392 self._items = items
393 self._bytelen = bytelen
394 _first_byte = from
395 _last_byte = to
396 _bytepos = from
397 end
398
399 # Low-level creation of a new string with all the data.
400 #
401 # `_items` will be used as is, without copy, to retrieve the characters of the string.
402 # Aliasing issues is the responsibility of the caller.
403 private init full(items: NativeString, bytelen, from, to, length: Int)
404 do
405 self._items = items
406 self.length = length
407 self._bytelen = bytelen
408 _first_byte = from
409 _last_byte = to
410 _bytepos = from
411 end
412
413 redef fun ==(other)
414 do
415 if not other isa FlatText then return super
416
417 if self.object_id == other.object_id then return true
418
419 var my_length = _bytelen
420
421 if other._bytelen != my_length then return false
422
423 var my_index = _first_byte
424 var its_index = other.first_byte
425
426 var last_iteration = my_index + my_length
427
428 var its_items = other._items
429 var my_items = self._items
430
431 while my_index < last_iteration do
432 if my_items[my_index] != its_items[its_index] then return false
433 my_index += 1
434 its_index += 1
435 end
436
437 return true
438 end
439
440 redef fun <(other)
441 do
442 if not other isa FlatText then return super
443
444 if self.object_id == other.object_id then return false
445
446 var myits = _items
447 var itsits = other._items
448
449 var mbt = _bytelen
450 var obt = other.bytelen
451
452 var minln = if mbt < obt then mbt else obt
453 var mst = _first_byte
454 var ost = other.first_byte
455
456 for i in [0 .. minln[ do
457 var my_curr_char = myits[mst]
458 var its_curr_char = itsits[ost]
459
460 if my_curr_char > its_curr_char then return false
461 if my_curr_char < its_curr_char then return true
462
463 mst += 1
464 ost += 1
465 end
466
467 return mbt < obt
468 end
469
470 redef fun +(o) do
471 var s = o.to_s
472 var slen = s.bytelen
473 var mlen = _bytelen
474 var nlen = mlen + slen
475 var mits = _items
476 var mifrom = _first_byte
477 if s isa FlatText then
478 var sits = s._items
479 var sifrom = s.first_byte
480 var ns = new NativeString(nlen + 1)
481 mits.copy_to(ns, mlen, mifrom, 0)
482 sits.copy_to(ns, slen, sifrom, mlen)
483 return new FlatString.full(ns, nlen, 0, nlen - 1, length + o.length)
484 else
485 abort
486 end
487 end
488
489 redef fun *(i) do
490 var mybtlen = _bytelen
491 var new_bytelen = mybtlen * i
492 var mylen = length
493 var newlen = mylen * i
494 var its = _items
495 var fb = _first_byte
496 var ns = new NativeString(new_bytelen + 1)
497 ns[new_bytelen] = 0u8
498 var offset = 0
499 while i > 0 do
500 its.copy_to(ns, mybtlen, fb, offset)
501 offset += mybtlen
502 i -= 1
503 end
504 return new FlatString.full(ns, new_bytelen, 0, new_bytelen - 1, newlen)
505 end
506
507
508 redef fun hash
509 do
510 if hash_cache == null then
511 # djb2 hash algorithm
512 var h = 5381
513 var i = _first_byte
514
515 var my_items = _items
516 var max = _last_byte
517
518 while i <= max do
519 h = (h << 5) + h + my_items[i].to_i
520 i += 1
521 end
522
523 hash_cache = h
524 end
525
526 return hash_cache.as(not null)
527 end
528
529 redef fun substrings do return new FlatSubstringsIter(self)
530 end
531
532 private class FlatStringCharReverseIterator
533 super IndexedIterator[Char]
534
535 var target: FlatString
536
537 var curr_pos: Int
538
539 redef fun is_ok do return curr_pos >= 0
540
541 redef fun item do return target[curr_pos]
542
543 redef fun next do curr_pos -= 1
544
545 redef fun index do return curr_pos
546
547 end
548
549 private class FlatStringCharIterator
550 super IndexedIterator[Char]
551
552 var target: FlatString
553
554 var max: Int is noautoinit
555
556 var curr_pos: Int
557
558 init do max = target.length - 1
559
560 redef fun is_ok do return curr_pos <= max
561
562 redef fun item do return target[curr_pos]
563
564 redef fun next do curr_pos += 1
565
566 redef fun index do return curr_pos
567
568 end
569
570 private class FlatStringCharView
571 super StringCharView
572
573 redef type SELFTYPE: FlatString
574
575 redef fun [](index) do return target[index]
576
577 redef fun iterator_from(start) do return new FlatStringCharIterator(target, start)
578
579 redef fun reverse_iterator_from(start) do return new FlatStringCharReverseIterator(target, start)
580
581 end
582
583 private class FlatStringByteReverseIterator
584 super IndexedIterator[Byte]
585
586 var target: FlatString
587
588 var target_items: NativeString is noautoinit
589
590 var curr_pos: Int
591
592 init
593 do
594 var tgt = target
595 target_items = tgt._items
596 curr_pos += tgt._first_byte
597 end
598
599 redef fun is_ok do return curr_pos >= target._first_byte
600
601 redef fun item do return target_items[curr_pos]
602
603 redef fun next do curr_pos -= 1
604
605 redef fun index do return curr_pos - target._first_byte
606
607 end
608
609 private class FlatStringByteIterator
610 super IndexedIterator[Byte]
611
612 var target: FlatString
613
614 var target_items: NativeString is noautoinit
615
616 var curr_pos: Int
617
618 init
619 do
620 var tgt = target
621 target_items = tgt._items
622 curr_pos += tgt._first_byte
623 end
624
625 redef fun is_ok do return curr_pos <= target._last_byte
626
627 redef fun item do return target_items[curr_pos]
628
629 redef fun next do curr_pos += 1
630
631 redef fun index do return curr_pos - target._first_byte
632
633 end
634
635 private class FlatStringByteView
636 super StringByteView
637
638 redef type SELFTYPE: FlatString
639
640 redef fun [](index)
641 do
642 # Check that the index (+ _first_byte) is not larger than _last_byte
643 # In other terms, if the index is valid
644 assert index >= 0
645 var target = self.target
646 var ind = index + target._first_byte
647 assert ind <= target._last_byte
648 return target._items[ind]
649 end
650
651 redef fun iterator_from(start) do return new FlatStringByteIterator(target, start)
652
653 redef fun reverse_iterator_from(start) do return new FlatStringByteReverseIterator(target, start)
654
655 end
656
657 redef class Buffer
658 redef new do return new FlatBuffer
659
660 redef new with_cap(i) do return new FlatBuffer.with_capacity(i)
661 end
662
663 # Mutable strings of characters.
664 class FlatBuffer
665 super FlatText
666 super Buffer
667
668 redef var chars: Sequence[Char] = new FlatBufferCharView(self) is lazy
669
670 redef var bytes = new FlatBufferByteView(self) is lazy
671
672 redef var length = 0
673
674 private var char_cache: Int = -1
675
676 private var byte_cache: Int = -1
677
678 private var capacity = 0
679
680 # Real items, used as cache for when to_cstring is called
681 private var real_items: NativeString is noinit
682
683 redef fun fast_cstring do return _items.fast_cstring(0)
684
685 redef fun substrings do return new FlatSubstringsIter(self)
686
687 # Re-copies the `NativeString` into a new one and sets it as the new `Buffer`
688 #
689 # This happens when an operation modifies the current `Buffer` and
690 # the Copy-On-Write flag `written` is set at true.
691 private fun reset do
692 var nns = new NativeString(capacity)
693 if _bytelen != 0 then _items.copy_to(nns, _bytelen, 0, 0)
694 _items = nns
695 written = false
696 end
697
698 # Shifts the content of the buffer by `len` bytes to the right, starting at byte `from`
699 #
700 # Internal only, does not modify _bytelen or length, this is the caller's responsability
701 private fun rshift_bytes(from: Int, len: Int) do
702 var oit = _items
703 var nit = _items
704 var bt = _bytelen
705 if bt + len > capacity then
706 capacity = capacity * 2 + 2
707 nit = new NativeString(capacity)
708 oit.copy_to(nit, 0, 0, from)
709 end
710 oit.copy_to(nit, bt - from, from, from + len)
711 end
712
713 # Shifts the content of the buffer by `len` bytes to the left, starting at `from`
714 #
715 # Internal only, does not modify _bytelen or length, this is the caller's responsability
716 private fun lshift_bytes(from: Int, len: Int) do
717 var it = _items
718 it.copy_to(it, _bytelen - from, from, from - len)
719 end
720
721 redef fun []=(index, item)
722 do
723 assert index >= 0 and index <= length
724 if written then reset
725 is_dirty = true
726 if index == length then
727 add item
728 return
729 end
730 var it = _items
731 var ip = it.char_to_byte_index(index)
732 var c = it.char_at(ip)
733 var clen = c.u8char_len
734 var itemlen = item.u8char_len
735 var size_diff = itemlen - clen
736 if size_diff > 0 then
737 rshift_bytes(ip + clen, size_diff)
738 else if size_diff < 0 then
739 lshift_bytes(ip + clen, -size_diff)
740 end
741 _bytelen += size_diff
742 bytepos += size_diff
743 it.set_char_at(ip, item)
744 end
745
746 redef fun add(c)
747 do
748 if written then reset
749 is_dirty = true
750 var clen = c.u8char_len
751 var bt = _bytelen
752 enlarge(bt + clen)
753 _items.set_char_at(bt, c)
754 _bytelen += clen
755 length += 1
756 end
757
758 redef fun clear do
759 is_dirty = true
760 if written then reset
761 _bytelen = 0
762 length = 0
763 end
764
765 redef fun empty do return new Buffer
766
767 redef fun enlarge(cap)
768 do
769 var c = capacity
770 if cap <= c then return
771 while c <= cap do c = c * 2 + 2
772 # The COW flag can be set at false here, since
773 # it does a copy of the current `Buffer`
774 written = false
775 var bln = _bytelen
776 var a = new NativeString(c+1)
777 if bln > 0 then
778 var it = _items
779 if bln > 0 then it.copy_to(a, bln, 0, 0)
780 end
781 _items = a
782 capacity = c
783 end
784
785 redef fun to_s
786 do
787 written = true
788 var bln = _bytelen
789 if bln == 0 then _items = new NativeString(1)
790 return new FlatString.full(_items, bln, 0, bln - 1, length)
791 end
792
793 redef fun to_cstring
794 do
795 if is_dirty then
796 var bln = _bytelen
797 var new_native = new NativeString(bln + 1)
798 new_native[bln] = 0u8
799 if length > 0 then _items.copy_to(new_native, bln, 0, 0)
800 real_items = new_native
801 is_dirty = false
802 end
803 return real_items
804 end
805
806 # Create a new empty string.
807 init do end
808
809 # Low-level creation a new buffer with given data.
810 #
811 # `_items` will be used as is, without copy, to store the characters of the buffer.
812 # Aliasing issues is the responsibility of the caller.
813 #
814 # If `_items` is shared, `written` should be set to true after the creation
815 # so that a modification will do a copy-on-write.
816 private init with_infos(items: NativeString, capacity, bytelen, length: Int)
817 do
818 self._items = items
819 self.capacity = capacity
820 self._bytelen = bytelen
821 self.length = length
822 end
823
824 # Create a new string copied from `s`.
825 init from(s: Text)
826 do
827 _items = new NativeString(s.bytelen)
828 if s isa FlatText then
829 _items = s._items
830 else
831 for i in substrings do i.as(FlatString)._items.copy_to(_items, i._bytelen, 0, 0)
832 end
833 _bytelen = s.bytelen
834 length = s.length
835 _capacity = _bytelen
836 written = true
837 end
838
839 # Create a new empty string with a given capacity.
840 init with_capacity(cap: Int)
841 do
842 assert cap >= 0
843 _items = new NativeString(cap + 1)
844 capacity = cap
845 _bytelen = 0
846 end
847
848 redef fun append(s)
849 do
850 if s.is_empty then return
851 is_dirty = true
852 var sl = s.bytelen
853 var nln = _bytelen + sl
854 enlarge(nln)
855 if s isa FlatText then
856 s._items.copy_to(_items, sl, s.first_byte, _bytelen)
857 else
858 for i in s.substrings do append i
859 return
860 end
861 _bytelen = nln
862 length += s.length
863 end
864
865 # Copies the content of self in `dest`
866 fun copy(start: Int, len: Int, dest: Buffer, new_start: Int)
867 do
868 var self_chars = self.chars
869 var dest_chars = dest.chars
870 for i in [0..len-1] do
871 dest_chars[new_start+i] = self_chars[start+i]
872 end
873 end
874
875 redef fun substring(from, count)
876 do
877 assert count >= 0
878 if from < 0 then from = 0
879 if (from + count) > length then count = length - from
880 if count <= 0 then return new Buffer
881 var its = _items
882 var bytefrom = its.char_to_byte_index(from)
883 var byteto = its.char_to_byte_index(count + from - 1)
884 byteto += its.char_at(byteto).u8char_len - 1
885 var byte_length = byteto - bytefrom + 1
886 var r_items = new NativeString(byte_length)
887 its.copy_to(r_items, byte_length, bytefrom, 0)
888 return new FlatBuffer.with_infos(r_items, byte_length, byte_length, count)
889 end
890
891 redef fun reverse
892 do
893 written = false
894 var ns = new FlatBuffer.with_capacity(capacity)
895 for i in chars.reverse_iterator do ns.add i
896 _items = ns._items
897 end
898
899 redef fun times(repeats)
900 do
901 var bln = _bytelen
902 var x = new FlatString.full(_items, bln, 0, bln - 1, length)
903 for i in [1 .. repeats[ do
904 append(x)
905 end
906 end
907
908 redef fun upper
909 do
910 if written then reset
911 for i in [0 .. length[ do self[i] = self[i].to_upper
912 end
913
914 redef fun lower
915 do
916 if written then reset
917 for i in [0 .. length[ do self[i] = self[i].to_lower
918 end
919 end
920
921 private class FlatBufferByteReverseIterator
922 super IndexedIterator[Byte]
923
924 var target: FlatBuffer
925
926 var target_items: NativeString is noautoinit
927
928 var curr_pos: Int
929
930 init do target_items = target._items
931
932 redef fun index do return curr_pos
933
934 redef fun is_ok do return curr_pos >= 0
935
936 redef fun item do return target_items[curr_pos]
937
938 redef fun next do curr_pos -= 1
939
940 end
941
942 private class FlatBufferByteView
943 super BufferByteView
944
945 redef type SELFTYPE: FlatBuffer
946
947 redef fun [](index) do return target._items[index]
948
949 redef fun iterator_from(pos) do return new FlatBufferByteIterator(target, pos)
950
951 redef fun reverse_iterator_from(pos) do return new FlatBufferByteReverseIterator(target, pos)
952
953 end
954
955 private class FlatBufferByteIterator
956 super IndexedIterator[Byte]
957
958 var target: FlatBuffer
959
960 var target_items: NativeString is noautoinit
961
962 var curr_pos: Int
963
964 init do target_items = target._items
965
966 redef fun index do return curr_pos
967
968 redef fun is_ok do return curr_pos < target._bytelen
969
970 redef fun item do return target_items[curr_pos]
971
972 redef fun next do curr_pos += 1
973
974 end
975
976 private class FlatBufferCharReverseIterator
977 super IndexedIterator[Char]
978
979 var target: FlatBuffer
980
981 var curr_pos: Int
982
983 redef fun index do return curr_pos
984
985 redef fun is_ok do return curr_pos >= 0
986
987 redef fun item do return target[curr_pos]
988
989 redef fun next do curr_pos -= 1
990
991 end
992
993 private class FlatBufferCharView
994 super BufferCharView
995
996 redef type SELFTYPE: FlatBuffer
997
998 redef fun [](index) do return target[index]
999
1000 redef fun []=(index, item)
1001 do
1002 assert index >= 0 and index <= length
1003 if index == length then
1004 add(item)
1005 return
1006 end
1007 target[index] = item
1008 end
1009
1010 redef fun push(c)
1011 do
1012 target.add(c)
1013 end
1014
1015 redef fun add(c)
1016 do
1017 target.add(c)
1018 end
1019
1020 fun enlarge(cap: Int)
1021 do
1022 target.enlarge(cap)
1023 end
1024
1025 redef fun append(s)
1026 do
1027 var s_length = s.length
1028 if target.capacity < s.length then enlarge(s_length + target.length)
1029 for i in s do target.add i
1030 end
1031
1032 redef fun iterator_from(pos) do return new FlatBufferCharIterator(target, pos)
1033
1034 redef fun reverse_iterator_from(pos) do return new FlatBufferCharReverseIterator(target, pos)
1035
1036 end
1037
1038 private class FlatBufferCharIterator
1039 super IndexedIterator[Char]
1040
1041 var target: FlatBuffer
1042
1043 var max: Int is noautoinit
1044
1045 var curr_pos: Int
1046
1047 init do max = target.length - 1
1048
1049 redef fun index do return curr_pos
1050
1051 redef fun is_ok do return curr_pos <= max
1052
1053 redef fun item do return target[curr_pos]
1054
1055 redef fun next do curr_pos += 1
1056
1057 end
1058
1059 redef class NativeString
1060 redef fun to_s
1061 do
1062 return to_s_with_length(cstring_length)
1063 end
1064
1065 # Returns `self` as a String of `length`.
1066 redef fun to_s_with_length(length): FlatString
1067 do
1068 assert length >= 0
1069 return clean_utf8(length)
1070 end
1071
1072 redef fun to_s_full(bytelen, unilen) do
1073 return new FlatString.full(self, bytelen, 0, bytelen - 1, unilen)
1074 end
1075
1076 # Returns `self` as a new String.
1077 redef fun to_s_with_copy: FlatString
1078 do
1079 var length = cstring_length
1080 var r = clean_utf8(length)
1081 if r.items != self then return r
1082 var new_self = new NativeString(length + 1)
1083 copy_to(new_self, length, 0, 0)
1084 var str = new FlatString.with_infos(new_self, length, 0, length - 1)
1085 new_self[length] = 0u8
1086 str.to_cstring = new_self
1087 return str
1088 end
1089
1090 # Cleans a NativeString if necessary
1091 fun clean_utf8(len: Int): FlatString do
1092 var replacements: nullable Array[Int] = null
1093 var end_length = len
1094 var pos = 0
1095 var chr_ln = 0
1096 while pos < len do
1097 var b = self[pos]
1098 var nxst = length_of_char_at(pos)
1099 var ok_st: Bool
1100 if nxst == 1 then
1101 ok_st = b & 0x80u8 == 0u8
1102 else if nxst == 2 then
1103 ok_st = b & 0xE0u8 == 0xC0u8
1104 else if nxst == 3 then
1105 ok_st = b & 0xF0u8 == 0xE0u8
1106 else
1107 ok_st = b & 0xF8u8 == 0xF0u8
1108 end
1109 if not ok_st then
1110 if replacements == null then replacements = new Array[Int]
1111 replacements.add pos
1112 end_length += 2
1113 pos += 1
1114 chr_ln += 1
1115 continue
1116 end
1117 var ok_c: Bool
1118 var c = char_at(pos)
1119 var cp = c.code_point
1120 if nxst == 1 then
1121 ok_c = cp >= 0 and cp <= 0x7F
1122 else if nxst == 2 then
1123 ok_c = cp >= 0x80 and cp <= 0x7FF
1124 else if nxst == 3 then
1125 ok_c = cp >= 0x800 and cp <= 0xFFFF
1126 ok_c = ok_c and not (cp >= 0xD800 and cp <= 0xDFFF) and cp != 0xFFFE and cp != 0xFFFF
1127 else
1128 ok_c = cp >= 0x10000 and cp <= 0x10FFFF
1129 end
1130 if not ok_c then
1131 if replacements == null then replacements = new Array[Int]
1132 replacements.add pos
1133 end_length += 2
1134 pos += 1
1135 chr_ln += 1
1136 continue
1137 end
1138 pos += c.u8char_len
1139 chr_ln += 1
1140 end
1141 var ret = self
1142 if end_length != len then
1143 ret = new NativeString(end_length)
1144 var old_repl = 0
1145 var off = 0
1146 var repls = replacements.as(not null)
1147 var r = repls.items.as(not null)
1148 var imax = repls.length
1149 for i in [0 .. imax[ do
1150 var repl_pos = r[i]
1151 var chkln = repl_pos - old_repl
1152 copy_to(ret, chkln, old_repl, off)
1153 off += chkln
1154 ret[off] = 0xEFu8
1155 ret[off + 1] = 0xBFu8
1156 ret[off + 2] = 0xBDu8
1157 old_repl = repl_pos + 1
1158 off += 3
1159 end
1160 copy_to(ret, len - old_repl, old_repl, off)
1161 end
1162 return new FlatString.full(ret, end_length, 0, end_length - 1, chr_ln)
1163 end
1164
1165 # Sets the next bytes at position `pos` to the value of `c`, encoded in UTF-8
1166 #
1167 # Very unsafe, make sure to have room for this char prior to calling this function.
1168 private fun set_char_at(pos: Int, c: Char) do
1169 var ln = c.u8char_len
1170 native_set_char(pos, c, ln)
1171 end
1172
1173 private fun native_set_char(pos: Int, c: Char, ln: Int) `{
1174 char* dst = self + pos;
1175 switch(ln){
1176 case 1:
1177 dst[0] = c;
1178 break;
1179 case 2:
1180 dst[0] = 0xC0 | ((c & 0x7C0) >> 6);
1181 dst[1] = 0x80 | (c & 0x3F);
1182 break;
1183 case 3:
1184 dst[0] = 0xE0 | ((c & 0xF000) >> 12);
1185 dst[1] = 0x80 | ((c & 0xFC0) >> 6);
1186 dst[2] = 0x80 | (c & 0x3F);
1187 break;
1188 case 4:
1189 dst[0] = 0xF0 | ((c & 0x1C0000) >> 18);
1190 dst[1] = 0x80 | ((c & 0x3F000) >> 12);
1191 dst[2] = 0x80 | ((c & 0xFC0) >> 6);
1192 dst[3] = 0x80 | (c & 0x3F);
1193 break;
1194 }
1195 `}
1196 end
1197
1198 redef class Int
1199 redef fun to_base(base, signed)
1200 do
1201 var l = digit_count(base)
1202 var s = new FlatBuffer.from(" " * l)
1203 fill_buffer(s, base, signed)
1204 return s.to_s
1205 end
1206
1207 # return displayable int in base 10 and signed
1208 #
1209 # assert 1.to_s == "1"
1210 # assert (-123).to_s == "-123"
1211 redef fun to_s do
1212 # Fast case for common numbers
1213 if self == 0 then return "0"
1214 if self == 1 then return "1"
1215
1216 var nslen = int_to_s_len
1217 var ns = new NativeString(nslen + 1)
1218 ns[nslen] = 0u8
1219 native_int_to_s(ns, nslen + 1)
1220 return new FlatString.full(ns, nslen, 0, nslen - 1, nslen)
1221 end
1222 end
1223
1224 redef class Array[E]
1225
1226 # Fast implementation
1227 redef fun plain_to_s
1228 do
1229 var l = length
1230 if l == 0 then return ""
1231 var its = _items.as(not null)
1232 var first = its[0]
1233 if l == 1 then if first == null then return "" else return first.to_s
1234 var na = new NativeArray[String](l)
1235 var i = 0
1236 var sl = 0
1237 var mypos = 0
1238 while i < l do
1239 var itsi = its[i]
1240 if itsi == null then
1241 i += 1
1242 continue
1243 end
1244 var tmp = itsi.to_s
1245 sl += tmp.bytelen
1246 na[mypos] = tmp
1247 i += 1
1248 mypos += 1
1249 end
1250 var ns = new NativeString(sl + 1)
1251 ns[sl] = 0u8
1252 i = 0
1253 var off = 0
1254 while i < mypos do
1255 var tmp = na[i]
1256 if tmp isa FlatString then
1257 var tpl = tmp._bytelen
1258 tmp._items.copy_to(ns, tpl, tmp._first_byte, off)
1259 off += tpl
1260 else
1261 for j in tmp.substrings do
1262 var s = j.as(FlatString)
1263 var slen = s._bytelen
1264 s._items.copy_to(ns, slen, s._first_byte, off)
1265 off += slen
1266 end
1267 end
1268 i += 1
1269 end
1270 return new FlatString.with_infos(ns, sl, 0, sl - 1)
1271 end
1272 end
1273
1274 redef class NativeArray[E]
1275 redef fun native_to_s do
1276 assert self isa NativeArray[String]
1277 var l = length
1278 var na = self
1279 var i = 0
1280 var sl = 0
1281 var mypos = 0
1282 while i < l do
1283 sl += na[i].bytelen
1284 i += 1
1285 mypos += 1
1286 end
1287 var ns = new NativeString(sl + 1)
1288 ns[sl] = 0u8
1289 i = 0
1290 var off = 0
1291 while i < mypos do
1292 var tmp = na[i]
1293 if tmp isa FlatString then
1294 var tpl = tmp._bytelen
1295 tmp._items.copy_to(ns, tpl, tmp._first_byte, off)
1296 off += tpl
1297 else
1298 for j in tmp.substrings do
1299 var s = j.as(FlatString)
1300 var slen = s._bytelen
1301 s._items.copy_to(ns, slen, s._first_byte, off)
1302 off += slen
1303 end
1304 end
1305 i += 1
1306 end
1307 return new FlatString.with_infos(ns, sl, 0, sl - 1)
1308 end
1309 end
1310
1311 redef class Map[K,V]
1312 redef fun join(sep, couple_sep)
1313 do
1314 if is_empty then return ""
1315
1316 var s = new Buffer # Result
1317
1318 # Concat first item
1319 var i = iterator
1320 var k = i.key
1321 var e = i.item
1322 s.append("{k or else "<null>"}{couple_sep}{e or else "<null>"}")
1323
1324 # Concat other _items
1325 i.next
1326 while i.is_ok do
1327 s.append(sep)
1328 k = i.key
1329 e = i.item
1330 s.append("{k or else "<null>"}{couple_sep}{e or else "<null>"}")
1331 i.next
1332 end
1333 return s.to_s
1334 end
1335 end