core/flat: remove named init `with_pos` and use auto-constructor
[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 FlatString 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 FlatString then return super
443
444 if self.object_id == other.object_id then return false
445
446 var my_length = self._bytelen
447 var its_length = other._bytelen
448
449 var max = if my_length < its_length then my_length else its_length
450
451 var myits = self.bytes
452 var itsits = other.bytes
453
454 for i in [0 .. max[ do
455 var my_curr_char = myits[i]
456 var its_curr_char = itsits[i]
457
458 if my_curr_char != its_curr_char then
459 if my_curr_char < its_curr_char then return true
460 return false
461 end
462 end
463
464 return my_length < its_length
465 end
466
467 redef fun +(o) do
468 var s = o.to_s
469 var slen = s.bytelen
470 var mlen = _bytelen
471 var nlen = mlen + slen
472 var mits = _items
473 var mifrom = _first_byte
474 if s isa FlatText then
475 var sits = s._items
476 var sifrom = s.first_byte
477 var ns = new NativeString(nlen + 1)
478 mits.copy_to(ns, mlen, mifrom, 0)
479 sits.copy_to(ns, slen, sifrom, mlen)
480 return new FlatString.full(ns, nlen, 0, nlen - 1, length + o.length)
481 else
482 abort
483 end
484 end
485
486 redef fun *(i) do
487 var mybtlen = _bytelen
488 var new_bytelen = mybtlen * i
489 var mylen = length
490 var newlen = mylen * i
491 var its = _items
492 var fb = _first_byte
493 var ns = new NativeString(new_bytelen + 1)
494 ns[new_bytelen] = 0u8
495 var offset = 0
496 while i > 0 do
497 its.copy_to(ns, mybtlen, fb, offset)
498 offset += mybtlen
499 i -= 1
500 end
501 return new FlatString.full(ns, new_bytelen, 0, new_bytelen - 1, newlen)
502 end
503
504
505 redef fun hash
506 do
507 if hash_cache == null then
508 # djb2 hash algorithm
509 var h = 5381
510 var i = _first_byte
511
512 var my_items = _items
513 var max = _last_byte
514
515 while i <= max do
516 h = (h << 5) + h + my_items[i].to_i
517 i += 1
518 end
519
520 hash_cache = h
521 end
522
523 return hash_cache.as(not null)
524 end
525
526 redef fun substrings do return new FlatSubstringsIter(self)
527 end
528
529 private class FlatStringCharReverseIterator
530 super IndexedIterator[Char]
531
532 var target: FlatString
533
534 var curr_pos: Int
535
536 redef fun is_ok do return curr_pos >= 0
537
538 redef fun item do return target[curr_pos]
539
540 redef fun next do curr_pos -= 1
541
542 redef fun index do return curr_pos
543
544 end
545
546 private class FlatStringCharIterator
547 super IndexedIterator[Char]
548
549 var target: FlatString
550
551 var max: Int is noautoinit
552
553 var curr_pos: Int
554
555 init do max = target.length - 1
556
557 redef fun is_ok do return curr_pos <= max
558
559 redef fun item do return target[curr_pos]
560
561 redef fun next do curr_pos += 1
562
563 redef fun index do return curr_pos
564
565 end
566
567 private class FlatStringCharView
568 super StringCharView
569
570 redef type SELFTYPE: FlatString
571
572 redef fun [](index) do return target[index]
573
574 redef fun iterator_from(start) do return new FlatStringCharIterator(target, start)
575
576 redef fun reverse_iterator_from(start) do return new FlatStringCharReverseIterator(target, start)
577
578 end
579
580 private class FlatStringByteReverseIterator
581 super IndexedIterator[Byte]
582
583 var target: FlatString
584
585 var target_items: NativeString is noautoinit
586
587 var curr_pos: Int
588
589 init
590 do
591 var tgt = target
592 target_items = tgt._items
593 curr_pos += tgt._first_byte
594 end
595
596 redef fun is_ok do return curr_pos >= target._first_byte
597
598 redef fun item do return target_items[curr_pos]
599
600 redef fun next do curr_pos -= 1
601
602 redef fun index do return curr_pos - target._first_byte
603
604 end
605
606 private class FlatStringByteIterator
607 super IndexedIterator[Byte]
608
609 var target: FlatString
610
611 var target_items: NativeString is noautoinit
612
613 var curr_pos: Int
614
615 init
616 do
617 var tgt = target
618 target_items = tgt._items
619 curr_pos += tgt._first_byte
620 end
621
622 redef fun is_ok do return curr_pos <= target._last_byte
623
624 redef fun item do return target_items[curr_pos]
625
626 redef fun next do curr_pos += 1
627
628 redef fun index do return curr_pos - target._first_byte
629
630 end
631
632 private class FlatStringByteView
633 super StringByteView
634
635 redef type SELFTYPE: FlatString
636
637 redef fun [](index)
638 do
639 # Check that the index (+ _first_byte) is not larger than _last_byte
640 # In other terms, if the index is valid
641 assert index >= 0
642 var target = self.target
643 var ind = index + target._first_byte
644 assert ind <= target._last_byte
645 return target._items[ind]
646 end
647
648 redef fun iterator_from(start) do return new FlatStringByteIterator(target, start)
649
650 redef fun reverse_iterator_from(start) do return new FlatStringByteReverseIterator(target, start)
651
652 end
653
654 redef class Buffer
655 redef new do return new FlatBuffer
656
657 redef new with_cap(i) do return new FlatBuffer.with_capacity(i)
658 end
659
660 # Mutable strings of characters.
661 class FlatBuffer
662 super FlatText
663 super Buffer
664
665 redef var chars: Sequence[Char] = new FlatBufferCharView(self) is lazy
666
667 redef var bytes = new FlatBufferByteView(self) is lazy
668
669 redef var length = 0
670
671 private var char_cache: Int = -1
672
673 private var byte_cache: Int = -1
674
675 private var capacity = 0
676
677 # Real items, used as cache for when to_cstring is called
678 private var real_items: NativeString is noinit
679
680 redef fun fast_cstring do return _items.fast_cstring(0)
681
682 redef fun substrings do return new FlatSubstringsIter(self)
683
684 # Re-copies the `NativeString` into a new one and sets it as the new `Buffer`
685 #
686 # This happens when an operation modifies the current `Buffer` and
687 # the Copy-On-Write flag `written` is set at true.
688 private fun reset do
689 var nns = new NativeString(capacity)
690 if _bytelen != 0 then _items.copy_to(nns, _bytelen, 0, 0)
691 _items = nns
692 written = false
693 end
694
695 # Shifts the content of the buffer by `len` bytes to the right, starting at byte `from`
696 #
697 # Internal only, does not modify _bytelen or length, this is the caller's responsability
698 private fun rshift_bytes(from: Int, len: Int) do
699 var oit = _items
700 var nit = _items
701 var bt = _bytelen
702 if bt + len > capacity then
703 capacity = capacity * 2 + 2
704 nit = new NativeString(capacity)
705 oit.copy_to(nit, 0, 0, from)
706 end
707 oit.copy_to(nit, bt - from, from, from + len)
708 end
709
710 # Shifts the content of the buffer by `len` bytes to the left, starting at `from`
711 #
712 # Internal only, does not modify _bytelen or length, this is the caller's responsability
713 private fun lshift_bytes(from: Int, len: Int) do
714 var it = _items
715 it.copy_to(it, _bytelen - from, from, from - len)
716 end
717
718 redef fun []=(index, item)
719 do
720 assert index >= 0 and index <= length
721 if written then reset
722 is_dirty = true
723 if index == length then
724 add item
725 return
726 end
727 var it = _items
728 var ip = it.char_to_byte_index(index)
729 var c = it.char_at(ip)
730 var clen = c.u8char_len
731 var itemlen = item.u8char_len
732 var size_diff = itemlen - clen
733 if size_diff > 0 then
734 rshift_bytes(ip + clen, size_diff)
735 else if size_diff < 0 then
736 lshift_bytes(ip + clen, -size_diff)
737 end
738 _bytelen += size_diff
739 bytepos += size_diff
740 it.set_char_at(ip, item)
741 end
742
743 redef fun add(c)
744 do
745 if written then reset
746 is_dirty = true
747 var clen = c.u8char_len
748 var bt = _bytelen
749 enlarge(bt + clen)
750 _items.set_char_at(bt, c)
751 _bytelen += clen
752 length += 1
753 end
754
755 redef fun clear do
756 is_dirty = true
757 if written then reset
758 _bytelen = 0
759 length = 0
760 end
761
762 redef fun empty do return new Buffer
763
764 redef fun enlarge(cap)
765 do
766 var c = capacity
767 if cap <= c then return
768 while c <= cap do c = c * 2 + 2
769 # The COW flag can be set at false here, since
770 # it does a copy of the current `Buffer`
771 written = false
772 var bln = _bytelen
773 var a = new NativeString(c+1)
774 if bln > 0 then
775 var it = _items
776 if bln > 0 then it.copy_to(a, bln, 0, 0)
777 end
778 _items = a
779 capacity = c
780 end
781
782 redef fun to_s
783 do
784 written = true
785 var bln = _bytelen
786 if bln == 0 then _items = new NativeString(1)
787 return new FlatString.full(_items, bln, 0, bln - 1, length)
788 end
789
790 redef fun to_cstring
791 do
792 if is_dirty then
793 var bln = _bytelen
794 var new_native = new NativeString(bln + 1)
795 new_native[bln] = 0u8
796 if length > 0 then _items.copy_to(new_native, bln, 0, 0)
797 real_items = new_native
798 is_dirty = false
799 end
800 return real_items
801 end
802
803 # Create a new empty string.
804 init do end
805
806 # Low-level creation a new buffer with given data.
807 #
808 # `_items` will be used as is, without copy, to store the characters of the buffer.
809 # Aliasing issues is the responsibility of the caller.
810 #
811 # If `_items` is shared, `written` should be set to true after the creation
812 # so that a modification will do a copy-on-write.
813 private init with_infos(items: NativeString, capacity, bytelen, length: Int)
814 do
815 self._items = items
816 self.capacity = capacity
817 self._bytelen = bytelen
818 self.length = length
819 end
820
821 # Create a new string copied from `s`.
822 init from(s: Text)
823 do
824 _items = new NativeString(s.bytelen)
825 if s isa FlatText then
826 _items = s._items
827 else
828 for i in substrings do i.as(FlatString)._items.copy_to(_items, i._bytelen, 0, 0)
829 end
830 _bytelen = s.bytelen
831 length = s.length
832 _capacity = _bytelen
833 written = true
834 end
835
836 # Create a new empty string with a given capacity.
837 init with_capacity(cap: Int)
838 do
839 assert cap >= 0
840 _items = new NativeString(cap + 1)
841 capacity = cap
842 _bytelen = 0
843 end
844
845 redef fun append(s)
846 do
847 if s.is_empty then return
848 is_dirty = true
849 var sl = s.bytelen
850 var nln = _bytelen + sl
851 enlarge(nln)
852 if s isa FlatText then
853 s._items.copy_to(_items, sl, s.first_byte, _bytelen)
854 else
855 for i in s.substrings do append i
856 return
857 end
858 _bytelen = nln
859 length += s.length
860 end
861
862 # Copies the content of self in `dest`
863 fun copy(start: Int, len: Int, dest: Buffer, new_start: Int)
864 do
865 var self_chars = self.chars
866 var dest_chars = dest.chars
867 for i in [0..len-1] do
868 dest_chars[new_start+i] = self_chars[start+i]
869 end
870 end
871
872 redef fun substring(from, count)
873 do
874 assert count >= 0
875 if from < 0 then from = 0
876 if (from + count) > length then count = length - from
877 if count <= 0 then return new Buffer
878 var its = _items
879 var bytefrom = its.char_to_byte_index(from)
880 var byteto = its.char_to_byte_index(count + from - 1)
881 byteto += its.char_at(byteto).u8char_len - 1
882 var byte_length = byteto - bytefrom + 1
883 var r_items = new NativeString(byte_length)
884 its.copy_to(r_items, byte_length, bytefrom, 0)
885 return new FlatBuffer.with_infos(r_items, byte_length, byte_length, count)
886 end
887
888 redef fun reverse
889 do
890 written = false
891 var ns = new FlatBuffer.with_capacity(capacity)
892 for i in chars.reverse_iterator do ns.add i
893 _items = ns._items
894 end
895
896 redef fun times(repeats)
897 do
898 var bln = _bytelen
899 var x = new FlatString.full(_items, bln, 0, bln - 1, length)
900 for i in [1 .. repeats[ do
901 append(x)
902 end
903 end
904
905 redef fun upper
906 do
907 if written then reset
908 for i in [0 .. length[ do self[i] = self[i].to_upper
909 end
910
911 redef fun lower
912 do
913 if written then reset
914 for i in [0 .. length[ do self[i] = self[i].to_lower
915 end
916 end
917
918 private class FlatBufferByteReverseIterator
919 super IndexedIterator[Byte]
920
921 var target: FlatBuffer
922
923 var target_items: NativeString is noautoinit
924
925 var curr_pos: Int
926
927 init do target_items = target._items
928
929 redef fun index do return curr_pos
930
931 redef fun is_ok do return curr_pos >= 0
932
933 redef fun item do return target_items[curr_pos]
934
935 redef fun next do curr_pos -= 1
936
937 end
938
939 private class FlatBufferByteView
940 super BufferByteView
941
942 redef type SELFTYPE: FlatBuffer
943
944 redef fun [](index) do return target._items[index]
945
946 redef fun iterator_from(pos) do return new FlatBufferByteIterator(target, pos)
947
948 redef fun reverse_iterator_from(pos) do return new FlatBufferByteReverseIterator(target, pos)
949
950 end
951
952 private class FlatBufferByteIterator
953 super IndexedIterator[Byte]
954
955 var target: FlatBuffer
956
957 var target_items: NativeString is noautoinit
958
959 var curr_pos: Int
960
961 init do target_items = target._items
962
963 redef fun index do return curr_pos
964
965 redef fun is_ok do return curr_pos < target._bytelen
966
967 redef fun item do return target_items[curr_pos]
968
969 redef fun next do curr_pos += 1
970
971 end
972
973 private class FlatBufferCharReverseIterator
974 super IndexedIterator[Char]
975
976 var target: FlatBuffer
977
978 var curr_pos: Int
979
980 redef fun index do return curr_pos
981
982 redef fun is_ok do return curr_pos >= 0
983
984 redef fun item do return target[curr_pos]
985
986 redef fun next do curr_pos -= 1
987
988 end
989
990 private class FlatBufferCharView
991 super BufferCharView
992
993 redef type SELFTYPE: FlatBuffer
994
995 redef fun [](index) do return target[index]
996
997 redef fun []=(index, item)
998 do
999 assert index >= 0 and index <= length
1000 if index == length then
1001 add(item)
1002 return
1003 end
1004 target[index] = item
1005 end
1006
1007 redef fun push(c)
1008 do
1009 target.add(c)
1010 end
1011
1012 redef fun add(c)
1013 do
1014 target.add(c)
1015 end
1016
1017 fun enlarge(cap: Int)
1018 do
1019 target.enlarge(cap)
1020 end
1021
1022 redef fun append(s)
1023 do
1024 var s_length = s.length
1025 if target.capacity < s.length then enlarge(s_length + target.length)
1026 for i in s do target.add i
1027 end
1028
1029 redef fun iterator_from(pos) do return new FlatBufferCharIterator(target, pos)
1030
1031 redef fun reverse_iterator_from(pos) do return new FlatBufferCharReverseIterator(target, pos)
1032
1033 end
1034
1035 private class FlatBufferCharIterator
1036 super IndexedIterator[Char]
1037
1038 var target: FlatBuffer
1039
1040 var max: Int is noautoinit
1041
1042 var curr_pos: Int
1043
1044 init do max = target.length - 1
1045
1046 redef fun index do return curr_pos
1047
1048 redef fun is_ok do return curr_pos <= max
1049
1050 redef fun item do return target[curr_pos]
1051
1052 redef fun next do curr_pos += 1
1053
1054 end
1055
1056 redef class NativeString
1057 redef fun to_s
1058 do
1059 return to_s_with_length(cstring_length)
1060 end
1061
1062 # Returns `self` as a String of `length`.
1063 redef fun to_s_with_length(length): FlatString
1064 do
1065 assert length >= 0
1066 return clean_utf8(length)
1067 end
1068
1069 redef fun to_s_full(bytelen, unilen) do
1070 return new FlatString.full(self, bytelen, 0, bytelen - 1, unilen)
1071 end
1072
1073 # Returns `self` as a new String.
1074 redef fun to_s_with_copy: FlatString
1075 do
1076 var length = cstring_length
1077 var r = clean_utf8(length)
1078 if r.items != self then return r
1079 var new_self = new NativeString(length + 1)
1080 copy_to(new_self, length, 0, 0)
1081 var str = new FlatString.with_infos(new_self, length, 0, length - 1)
1082 new_self[length] = 0u8
1083 str.to_cstring = new_self
1084 return str
1085 end
1086
1087 # Cleans a NativeString if necessary
1088 fun clean_utf8(len: Int): FlatString do
1089 var replacements: nullable Array[Int] = null
1090 var end_length = len
1091 var pos = 0
1092 var chr_ln = 0
1093 while pos < len do
1094 var b = self[pos]
1095 var nxst = length_of_char_at(pos)
1096 var ok_st: Bool
1097 if nxst == 1 then
1098 ok_st = b & 0x80u8 == 0u8
1099 else if nxst == 2 then
1100 ok_st = b & 0xE0u8 == 0xC0u8
1101 else if nxst == 3 then
1102 ok_st = b & 0xF0u8 == 0xE0u8
1103 else
1104 ok_st = b & 0xF8u8 == 0xF0u8
1105 end
1106 if not ok_st then
1107 if replacements == null then replacements = new Array[Int]
1108 replacements.add pos
1109 end_length += 2
1110 pos += 1
1111 chr_ln += 1
1112 continue
1113 end
1114 var ok_c: Bool
1115 var c = char_at(pos)
1116 var cp = c.code_point
1117 if nxst == 1 then
1118 ok_c = cp >= 0 and cp <= 0x7F
1119 else if nxst == 2 then
1120 ok_c = cp >= 0x80 and cp <= 0x7FF
1121 else if nxst == 3 then
1122 ok_c = cp >= 0x800 and cp <= 0xFFFF
1123 ok_c = ok_c and not (cp >= 0xD800 and cp <= 0xDFFF) and cp != 0xFFFE and cp != 0xFFFF
1124 else
1125 ok_c = cp >= 0x10000 and cp <= 0x10FFFF
1126 end
1127 if not ok_c then
1128 if replacements == null then replacements = new Array[Int]
1129 replacements.add pos
1130 end_length += 2
1131 pos += 1
1132 chr_ln += 1
1133 continue
1134 end
1135 pos += c.u8char_len
1136 chr_ln += 1
1137 end
1138 var ret = self
1139 if end_length != len then
1140 ret = new NativeString(end_length)
1141 var old_repl = 0
1142 var off = 0
1143 var repls = replacements.as(not null)
1144 var r = repls.items.as(not null)
1145 var imax = repls.length
1146 for i in [0 .. imax[ do
1147 var repl_pos = r[i]
1148 var chkln = repl_pos - old_repl
1149 copy_to(ret, chkln, old_repl, off)
1150 off += chkln
1151 ret[off] = 0xEFu8
1152 ret[off + 1] = 0xBFu8
1153 ret[off + 2] = 0xBDu8
1154 old_repl = repl_pos + 1
1155 off += 3
1156 end
1157 copy_to(ret, len - old_repl, old_repl, off)
1158 end
1159 return new FlatString.full(ret, end_length, 0, end_length - 1, chr_ln)
1160 end
1161
1162 # Sets the next bytes at position `pos` to the value of `c`, encoded in UTF-8
1163 #
1164 # Very unsafe, make sure to have room for this char prior to calling this function.
1165 private fun set_char_at(pos: Int, c: Char) do
1166 var ln = c.u8char_len
1167 native_set_char(pos, c, ln)
1168 end
1169
1170 private fun native_set_char(pos: Int, c: Char, ln: Int) `{
1171 char* dst = self + pos;
1172 switch(ln){
1173 case 1:
1174 dst[0] = c;
1175 break;
1176 case 2:
1177 dst[0] = 0xC0 | ((c & 0x7C0) >> 6);
1178 dst[1] = 0x80 | (c & 0x3F);
1179 break;
1180 case 3:
1181 dst[0] = 0xE0 | ((c & 0xF000) >> 12);
1182 dst[1] = 0x80 | ((c & 0xFC0) >> 6);
1183 dst[2] = 0x80 | (c & 0x3F);
1184 break;
1185 case 4:
1186 dst[0] = 0xF0 | ((c & 0x1C0000) >> 18);
1187 dst[1] = 0x80 | ((c & 0x3F000) >> 12);
1188 dst[2] = 0x80 | ((c & 0xFC0) >> 6);
1189 dst[3] = 0x80 | (c & 0x3F);
1190 break;
1191 }
1192 `}
1193 end
1194
1195 redef class Int
1196 redef fun to_base(base, signed)
1197 do
1198 var l = digit_count(base)
1199 var s = new FlatBuffer.from(" " * l)
1200 fill_buffer(s, base, signed)
1201 return s.to_s
1202 end
1203
1204 # return displayable int in base 10 and signed
1205 #
1206 # assert 1.to_s == "1"
1207 # assert (-123).to_s == "-123"
1208 redef fun to_s do
1209 # Fast case for common numbers
1210 if self == 0 then return "0"
1211 if self == 1 then return "1"
1212
1213 var nslen = int_to_s_len
1214 var ns = new NativeString(nslen + 1)
1215 ns[nslen] = 0u8
1216 native_int_to_s(ns, nslen + 1)
1217 return new FlatString.full(ns, nslen, 0, nslen - 1, nslen)
1218 end
1219 end
1220
1221 redef class Array[E]
1222
1223 # Fast implementation
1224 redef fun plain_to_s
1225 do
1226 var l = length
1227 if l == 0 then return ""
1228 var its = _items.as(not null)
1229 var first = its[0]
1230 if l == 1 then if first == null then return "" else return first.to_s
1231 var na = new NativeArray[String](l)
1232 var i = 0
1233 var sl = 0
1234 var mypos = 0
1235 while i < l do
1236 var itsi = its[i]
1237 if itsi == null then
1238 i += 1
1239 continue
1240 end
1241 var tmp = itsi.to_s
1242 sl += tmp.bytelen
1243 na[mypos] = tmp
1244 i += 1
1245 mypos += 1
1246 end
1247 var ns = new NativeString(sl + 1)
1248 ns[sl] = 0u8
1249 i = 0
1250 var off = 0
1251 while i < mypos do
1252 var tmp = na[i]
1253 if tmp isa FlatString then
1254 var tpl = tmp._bytelen
1255 tmp._items.copy_to(ns, tpl, tmp._first_byte, off)
1256 off += tpl
1257 else
1258 for j in tmp.substrings do
1259 var s = j.as(FlatString)
1260 var slen = s._bytelen
1261 s._items.copy_to(ns, slen, s._first_byte, off)
1262 off += slen
1263 end
1264 end
1265 i += 1
1266 end
1267 return new FlatString.with_infos(ns, sl, 0, sl - 1)
1268 end
1269 end
1270
1271 redef class NativeArray[E]
1272 redef fun native_to_s do
1273 assert self isa NativeArray[String]
1274 var l = length
1275 var na = self
1276 var i = 0
1277 var sl = 0
1278 var mypos = 0
1279 while i < l do
1280 sl += na[i].bytelen
1281 i += 1
1282 mypos += 1
1283 end
1284 var ns = new NativeString(sl + 1)
1285 ns[sl] = 0u8
1286 i = 0
1287 var off = 0
1288 while i < mypos do
1289 var tmp = na[i]
1290 if tmp isa FlatString then
1291 var tpl = tmp._bytelen
1292 tmp._items.copy_to(ns, tpl, tmp._first_byte, off)
1293 off += tpl
1294 else
1295 for j in tmp.substrings do
1296 var s = j.as(FlatString)
1297 var slen = s._bytelen
1298 s._items.copy_to(ns, slen, s._first_byte, off)
1299 off += slen
1300 end
1301 end
1302 i += 1
1303 end
1304 return new FlatString.with_infos(ns, sl, 0, sl - 1)
1305 end
1306 end
1307
1308 redef class Map[K,V]
1309 redef fun join(sep, couple_sep)
1310 do
1311 if is_empty then return ""
1312
1313 var s = new Buffer # Result
1314
1315 # Concat first item
1316 var i = iterator
1317 var k = i.key
1318 var e = i.item
1319 s.append("{k or else "<null>"}{couple_sep}{e or else "<null>"}")
1320
1321 # Concat other _items
1322 i.next
1323 while i.is_ok do
1324 s.append(sep)
1325 k = i.key
1326 e = i.item
1327 s.append("{k or else "<null>"}{couple_sep}{e or else "<null>"}")
1328 i.next
1329 end
1330 return s.to_s
1331 end
1332 end