stdlib/strings: Access iterator through its position constructor.
[nit.git] / lib / standard / string.nit
index 7774dd9..524d4d4 100644 (file)
@@ -5,7 +5,7 @@
 #
 # This file is free software, which comes along with NIT.  This software is
 # distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
-# without  even  the implied warranty of  MERCHANTABILITY or  FITNESS FOR A 
+# without  even  the implied warranty of  MERCHANTABILITY or  FITNESS FOR A
 # PARTICULAR PURPOSE.  You can modify it is you want,  provided this header
 # is kept unaltered, and a notification of the changes is added.
 # You  are  allowed  to  redistribute it and sell it, alone or is a part of
@@ -14,6 +14,7 @@
 # Basic manipulations of strings of characters
 module string
 
+import math
 intrude import collection # FIXME should be collection::array
 
 `{
@@ -56,7 +57,7 @@ abstract class AbstractString
                if from < count then
                        var r = new Buffer.with_capacity(count - from)
                        while from < count do
-                               r.push(_items[from])
+                               r.chars.push(_items[from])
                                from += 1
                        end
                        return r.to_s
@@ -144,7 +145,7 @@ abstract class AbstractString
                var i = 0
                var neg = false
 
-               for c in self
+               for c in self.chars
                do
                        var v = c.to_i
                        if v > base then
@@ -175,7 +176,7 @@ abstract class AbstractString
        fun is_numeric: Bool
        do
                var has_point_or_comma = false
-               for i in self
+               for i in self.chars
                do
                        if not i.is_numeric
                        then
@@ -196,7 +197,7 @@ abstract class AbstractString
        fun to_upper: String
        do
                var s = new Buffer.with_capacity(length)
-               for i in self do s.add(i.to_upper)
+               for i in self.chars do s.add(i.to_upper)
                return s.to_s
        end
 
@@ -206,7 +207,7 @@ abstract class AbstractString
        fun to_lower : String
        do
                var s = new Buffer.with_capacity(length)
-               for i in self do s.add(i.to_lower)
+               for i in self.chars do s.add(i.to_lower)
                return s.to_s
        end
 
@@ -217,18 +218,18 @@ abstract class AbstractString
        #     assert "\na\nb\tc\t".trim          == "a\nb\tc"
        fun trim: String
        do
-               if self._length == 0 then return self.to_s
+               if self.length == 0 then return self.to_s
                # find position of the first non white space char (ascii < 32) from the start of the string
                var start_pos = 0
-               while self[start_pos].ascii <= 32 do
+               while self.chars[start_pos].ascii <= 32 do
                        start_pos += 1
-                       if start_pos == _length then return ""
+                       if start_pos == length then return ""
                end
                # find position of the first non white space char from the end of the string
                var end_pos = length - 1
-               while self[end_pos].ascii <= 32 do
+               while self.chars[end_pos].ascii <= 32 do
                        end_pos -= 1
-                       if end_pos == start_pos then return self[start_pos].to_s
+                       if end_pos == start_pos then return self.chars[start_pos].to_s
                end
                return self.substring(start_pos, end_pos - start_pos + 1)
        end
@@ -247,7 +248,7 @@ abstract class AbstractString
        do
                var res = new Buffer
                var underscore = false
-               for c in self do
+               for c in self.chars do
                        if (c >= 'a' and c <= 'z') or (c >='A' and c <= 'Z') then
                                res.add(c)
                                underscore = false
@@ -280,7 +281,7 @@ abstract class AbstractString
        fun escape_to_c: String
        do
                var b = new Buffer
-               for c in self do
+               for c in self.chars do
                        if c == '\n' then
                                b.append("\\n")
                        else if c == '\0' then
@@ -377,6 +378,10 @@ abstract class StringCharView
 
        redef fun length do return target.length
 
+       redef fun iterator: IndexedIterator[Char] do return self.iterator_from(0)
+
+       fun iterator_from(pos: Int): IndexedIterator[Char] is abstract
+
        redef fun has(c: Char): Bool
        do
                for i in self do
@@ -437,11 +442,13 @@ class String
 
                var realFrom = _index_from + from
 
-               if (realFrom + count) > _index_to then return new String.from_substring(realFrom, _index_to, _items)
+               if (realFrom + count) > _index_to then return new String.with_infos(_items, _index_to - realFrom + 1, realFrom, _index_to)
 
                if count == 0 then return ""
 
-               return new String.from_substring(realFrom, realFrom + count - 1, _items)
+               var to = realFrom + count - 1
+
+               return new String.with_infos(_items, to - realFrom + 1, realFrom, to)
        end
 
        redef fun substring_from(from: Int): String
@@ -549,20 +556,6 @@ class String
        #              String Specific Methods           #
        ##################################################
 
-       # Creates a String object as a substring of another String
-       #
-       # From : index to start at
-       #
-       # To : Index to stop at (from + count -1)
-       #
-       private init from_substring(from: Int, to: Int, internalString: NativeString)
-       do
-               _items = internalString
-               _index_from = from
-               _index_to = to
-               _length = to - from + 1
-       end
-
        private init with_infos(items: NativeString, len: Int, from: Int, to: Int)
        do
                self._items = items
@@ -758,7 +751,7 @@ private class FlatStringCharView
                return target._items[index + target._index_from]
        end
 
-       redef fun iterator: IndexedIterator[Char] do return new FlatStringIterator.with_pos(target, 0)
+       redef fun iterator_from(start) do return new FlatStringIterator.with_pos(target, start)
 
 end
 
@@ -831,8 +824,8 @@ class Buffer
                var l1 = length
                var l2 = s.length
                while i < l1 and i < l2 do
-                       var c1 = self[i].ascii
-                       var c2 = s[i].ascii
+                       var c1 = self.chars[i].ascii
+                       var c2 = s.chars[i].ascii
                        if c1 < c2 then
                                return true
                        else if c2 < c1 then
@@ -895,11 +888,6 @@ private class FlatBufferCharView
 
        redef type SELFTYPE: Buffer
 
-       init(tgt: Buffer)
-       do
-               self.target = tgt
-       end
-
        redef fun [](index) do return target._items[index]
 
        redef fun []=(index, item)
@@ -934,7 +922,7 @@ private class FlatBufferCharView
                if target.capacity < s.length then enlarge(s_length + target.length)
        end
 
-       redef fun iterator: IndexedIterator[Char] do return new FlatBufferIterator.with_pos(target, 0)
+       redef fun iterator_from(pos) do return new FlatBufferIterator.with_pos(target, pos)
 
 end
 
@@ -1004,13 +992,13 @@ redef class Bool
        #     assert true.to_s         == "true"
        #     assert false.to_s        == "false"
        redef fun to_s
-       do 
-               if self then 
-                       return once "true" 
-               else 
-                       return once "false" 
+       do
+               if self then
+                       return once "true"
+               else
+                       return once "false"
                end
-       end   
+       end
 end
 
 redef class Int
@@ -1022,17 +1010,17 @@ redef class Int
                # Sign
                if self < 0 then
                        n = - self
-                       s[0] = '-'
+                       s.chars[0] = '-'
                else if self == 0 then
-                       s[0] = '0'
+                       s.chars[0] = '0'
                        return
                else
                        n = self
                end
                # Fill digits
                var pos = digit_count(base) - 1
-               while pos >= 0 and n > 0 do 
-                       s[pos] = (n % base).to_c
+               while pos >= 0 and n > 0 do
+                       s.chars[pos] = (n % base).to_c
                        n = n / base # /
                        pos -= 1
                end
@@ -1067,10 +1055,11 @@ redef class Float
        # Pretty print self, print needoed decimals up to a max of 3.
        redef fun to_s do
                var str = to_precision( 3 )
+               if is_inf != 0 or is_nan then return str
                var len = str.length
                for i in [0..len-1] do
                        var j = len-1-i
-                       var c = str[j]
+                       var c = str.chars[j]
                        if c == '0' then
                                continue
                        else if c == '.' then
@@ -1085,6 +1074,15 @@ redef class Float
        # `self` representation with `nb` digits after the '.'.
        fun to_precision(nb: Int): String
        do
+               if is_nan then return "nan"
+
+               var isinf = self.is_inf
+               if isinf == 1 then
+                       return "inf"
+               else if isinf == -1 then
+                       return  "-inf"
+               end
+
                if nb == 0 then return self.to_i.to_s
                var f = self
                for i in [0..nb[ do f = f * 10.0
@@ -1106,7 +1104,7 @@ redef class Float
                end
        end
 
-       fun to_precision_native(nb: Int): String import NativeString::to_s `{
+       fun to_precision_native(nb: Int): String import NativeString.to_s `{
                int size;
                char *str;
 
@@ -1123,7 +1121,7 @@ redef class Char
        redef fun to_s
        do
                var s = new Buffer.with_capacity(1)
-               s[0] = self
+               s.chars[0] = self
                return s.to_s
        end
 
@@ -1168,14 +1166,14 @@ redef class Collection[E]
        fun join(sep: String): String
        do
                if is_empty then return ""
-               
+
                var s = new Buffer # Result
 
                # Concat first item
                var i = iterator
                var e = i.item
                if e != null then s.append(e.to_s)
-               
+
                # Concat other items
                i.next
                while i.is_ok do
@@ -1216,7 +1214,7 @@ redef class Map[K,V]
        fun join(sep: String, couple_sep: String): String
        do
                if is_empty then return ""
-               
+
                var s = new Buffer # Result
 
                # Concat first item