standard/collection: add `to_step` as a decorator of iterators
[nit.git] / lib / standard / collection / abstract_collection.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # Copyright 2004-2008 Jean Privat <jean@pryen.org>
4 #
5 # This file is free software, which comes along with NIT. This software is
6 # distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
7 # without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
8 # PARTICULAR PURPOSE. You can modify it is you want, provided this header
9 # is kept unaltered, and a notification of the changes is added.
10 # You are allowed to redistribute it and sell it, alone or is a part of
11 # another product.
12
13 # Abstract collection classes and services.
14 #
15 # TODO specify the behavior on iterators when collections are modified.
16 module abstract_collection
17
18 import kernel
19
20 # The root of the collection hierarchy.
21 #
22 # Collections modelize finite groups of objects, called elements.
23 #
24 # The specific behavior and representation of collections is determined
25 # by the subclasses of the hierarchy.
26 #
27 # The main service of Collection is to provide a stable `iterator`
28 # method usable to retrieve all the elements of the collection.
29 #
30 # Additional services are provided.
31 # For an implementation point of view, Collection provide a basic
32 # implementation of these services using the `iterator` method.
33 # Subclasses often provide a more efficient implementation.
34 #
35 # Because of the `iterator` method, Collections instances can use
36 # the `for` control structure.
37 #
38 # ~~~nitish
39 # var x: Collection[U]
40 # # ...
41 # for u in x do
42 # # u is a U
43 # # ...
44 # end
45 # ~~~
46 #
47 # that is equivalent with the following:
48 #
49 # ~~~nitish
50 # var x: Collection[U]
51 # # ...
52 # var i = x.iterator
53 # while i.is_ok do
54 # var u = i.item # u is a U
55 # # ...
56 # i.next
57 # end
58 # ~~~
59 interface Collection[E]
60 # Get a new iterator on the collection.
61 fun iterator: Iterator[E] is abstract
62
63 # Is there no item in the collection?
64 #
65 # assert [1,2,3].is_empty == false
66 # assert [1..1[.is_empty == true
67 fun is_empty: Bool do return length == 0
68
69 # Alias for `not is_empty`.
70 #
71 # Some people prefer to have conditions grammatically easier to read.
72 #
73 # assert [1,2,3].not_empty == true
74 # assert [1..1[.not_empty == false
75 fun not_empty: Bool do return not self.is_empty
76
77 # Number of items in the collection.
78 #
79 # assert [10,20,30].length == 3
80 # assert [20..30[.length == 10
81 fun length: Int
82 do
83 var nb = 0
84 for i in self do nb += 1
85 return nb
86 end
87
88 # Is `item` in the collection ?
89 # Comparisons are done with ==
90 #
91 # assert [1,2,3].has(2) == true
92 # assert [1,2,3].has(9) == false
93 # assert [1..5[.has(2) == true
94 # assert [1..5[.has(9) == false
95 fun has(item: E): Bool
96 do
97 for i in self do if i == item then return true
98 return false
99 end
100
101 # Is the collection contain only `item`?
102 # Comparisons are done with ==
103 # Return true if the collection is empty.
104 #
105 # assert [1,1,1].has_only(1) == true
106 # assert [1,2,3].has_only(1) == false
107 # assert [1..1].has_only(1) == true
108 # assert [1..3].has_only(1) == false
109 # assert [3..3[.has_only(1) == true # empty collection
110 #
111 # ENSURE `is_empty implies result == true`
112 fun has_only(item: E): Bool
113 do
114 for i in self do if i != item then return false
115 return true
116 end
117
118 # How many occurrences of `item` are in the collection?
119 # Comparisons are done with ==
120 #
121 # assert [10,20,10].count(10) == 2
122 fun count(item: E): Int
123 do
124 var nb = 0
125 for i in self do if i == item then nb += 1
126 return nb
127 end
128
129 # Return the first item of the collection
130 #
131 # assert [1,2,3].first == 1
132 fun first: E
133 do
134 assert length > 0
135 return iterator.item
136 end
137
138 # Does the collection contain at least each element of `other`?
139 #
140 # assert [1,3,4,2].has_all([1..2]) == true
141 # assert [1,3,4,2].has_all([1..5]) == false
142 #
143 # Repeated elements in the collections are not considered.
144 #
145 # assert [1,1,1].has_all([1]) == true
146 # assert [1..5].has_all([1,1,1]) == true
147 #
148 # Note that the default implementation is general and correct for any lawful Collections.
149 # It is memory-efficient but relies on `has` so may be CPU-inefficient for some kind of collections.
150 fun has_all(other: Collection[E]): Bool
151 do
152 for x in other do if not has(x) then return false
153 return true
154 end
155
156 # Does the collection contain exactly all the elements of `other`?
157 #
158 # The same elements must be present in both `self` and `other`,
159 # but the order of the elements in the collections are not considered.
160 #
161 # assert [1..3].has_exactly([3,1,2]) == true # the same elements
162 # assert [1..3].has_exactly([3,1]) == false # 2 is not in the array
163 # assert [1..2].has_exactly([3,1,2]) == false # 3 is not in the range
164 #
165 # Repeated elements must be present in both collections in the same amount.
166 # So basically it is a multi-set comparison.
167 #
168 # assert [1,2,3,2].has_exactly([1,2,2,3]) == true # the same elements
169 # assert [1,2,3,2].has_exactly([1,2,3]) == false # more 2 in the first array
170 # assert [1,2,3].has_exactly([1,2,2,3]) == false # more 2 in the second array
171 #
172 # Note that the default implementation is general and correct for any lawful Collections.
173 # It is memory-efficient but relies on `count` so may be CPU-inefficient for some kind of collections.
174 fun has_exactly(other: Collection[E]): Bool
175 do
176 if length != other.length then return false
177 for e in self do if self.count(e) != other.count(e) then return false
178 return true
179 end
180 end
181
182 # Iterators generate a series of elements, one at a time.
183 #
184 # They are mainly used with collections and obtained from `Collection::iterator`.
185 interface Iterator[E]
186 # The current item.
187 # Require `is_ok`.
188 fun item: E is abstract
189
190 # Jump to the next item.
191 # Require `is_ok`.
192 fun next is abstract
193
194 # Jump to the next item `step` times.
195 #
196 # ~~~
197 # var i = [11, 22, 33, 44].iterator
198 # assert i.item == 11
199 # i.next_by 2
200 # assert i.item == 33
201 # ~~~
202 #
203 # `next_by` should be used instead of looping on `next` because is takes care
204 # of stopping if the end of iteration is reached prematurely whereas a loop of
205 # `next` will abort because of the precondition on `is_ok`.
206 #
207 # ~~~
208 # i.next_by 100
209 # assert not i.is_ok
210 # ~~~
211 #
212 # If `step` is negative, this method aborts.
213 # But specific subclasses can change this and do something more meaningful instead.
214 #
215 # Require `is_ok`
216 fun next_by(step: Int)
217 do
218 assert step >= 0
219 while is_ok and step > 0 do
220 next
221 step -= 1
222 end
223 end
224
225 # Is there a current item ?
226 fun is_ok: Bool is abstract
227
228 # Iterate over `self`
229 fun iterator: Iterator[E] do return self
230
231 # Post-iteration hook.
232 #
233 # Used to inform `self` that the iteration is over.
234 # Specific iterators can use this to free some resources.
235 #
236 # Is automatically invoked at the end of `for` structures.
237 #
238 # Do nothing by default.
239 fun finish do end
240
241 # A decorator around `self` that advance self a given number of steps instead of one.
242 #
243 # ~~~
244 # var i = [11, 22, 33, 44, 55].iterator
245 # var i2 = i.to_step(2)
246 #
247 # assert i2.item == 11
248 # i2.next
249 # assert i2.item == 33
250 #
251 # assert i.item == 33
252 # ~~~
253 fun to_step(step: Int): Iterator[E] do return new StepIterator[E](self, step)
254 end
255
256 # A basic helper class to specialize specific Iterator decorators
257 abstract class IteratorDecorator[E]
258 super Iterator[E]
259
260 # The underling iterator
261 protected var real: Iterator[E]
262
263 redef fun is_ok do return real.is_ok
264 redef fun item do return real.item
265 redef fun finish do real.finish
266 redef fun next do real.next
267 redef fun next_by(step) do real.next_by(step)
268 end
269
270 # A decorator that advance a given number of steps
271 private class StepIterator[E]
272 super IteratorDecorator[E]
273 var step: Int
274
275 redef fun next do real.next_by(step)
276 redef fun next_by(step) do real.next_by(step * self.step)
277 end
278
279 # A collection that contains only one item.
280 #
281 # Used to pass arguments by reference.
282 #
283 # Also used when one want to give a single element when a full
284 # collection is expected
285 class Container[E]
286 super Collection[E]
287
288 redef fun first do return item
289
290 redef fun is_empty do return false
291
292 redef fun length do return 1
293
294 redef fun has(an_item) do return item == an_item
295
296 redef fun has_only(an_item) do return item == an_item
297
298 redef fun count(an_item)
299 do
300 if item == an_item then
301 return 1
302 else
303 return 0
304 end
305 end
306
307 redef fun iterator do return new ContainerIterator[E](self)
308
309 # The stored item
310 var item: E is writable
311 end
312
313 # This iterator is quite stupid since it is used for only one item.
314 private class ContainerIterator[E]
315 super Iterator[E]
316 redef fun item do return _container.item
317
318 redef fun next do is_ok = false
319
320 redef var is_ok: Bool = true
321
322 var container: Container[E]
323 end
324
325 # Items can be removed from this collection
326 interface RemovableCollection[E]
327 super Collection[E]
328
329 # Remove all items
330 #
331 # var a = [1,2,3]
332 # a.clear
333 # assert a.length == 0
334 #
335 # ENSURE `is_empty`
336 fun clear is abstract
337
338 # Remove an occucence of `item`
339 #
340 # var a = [1,2,3,1,2,3]
341 # a.remove 2
342 # assert a == [1,3,1,2,3]
343 fun remove(item: E) is abstract
344
345 # Remove all occurences of `item`
346 #
347 # var a = [1,2,3,1,2,3]
348 # a.remove_all 2
349 # assert a == [1,3,1,3]
350 fun remove_all(item: E) do while has(item) do remove(item)
351 end
352
353 # Items can be added to these collections.
354 interface SimpleCollection[E]
355 super RemovableCollection[E]
356
357 # Add an item in a collection.
358 #
359 # var a = [1,2]
360 # a.add 3
361 # assert a.has(3) == true
362 # assert a.has(10) == false
363 #
364 # Ensure col.has(item)
365 fun add(item: E) is abstract
366
367 # Add each item of `coll`.
368 # var a = [1,2]
369 # a.add_all([3..5])
370 # assert a.has(4) == true
371 # assert a.has(10) == false
372 fun add_all(coll: Collection[E]) do for i in coll do add(i)
373 end
374
375 # Abstract sets.
376 #
377 # Set is a collection without duplicates (according to `==`)
378 #
379 # var s: Set[String] = new ArraySet[String]
380 # var a = "Hello"
381 # var b = "Hel" + "lo"
382 # # ...
383 # s.add(a)
384 # assert s.has(b) == true
385 interface Set[E]
386 super SimpleCollection[E]
387
388 redef fun has_only(item)
389 do
390 var l = length
391 if l == 1 then
392 return has(item)
393 else if l == 0 then
394 return true
395 else
396 return false
397 end
398 end
399
400 # Only 0 or 1
401 redef fun count(item)
402 do
403 if has(item) then
404 return 1
405 else
406 return 0
407 end
408 end
409
410 # Synonym of remove since there is only one item
411 redef fun remove_all(item) do remove(item)
412
413 # Equality is defined on set and means that each set contains the same elements
414 redef fun ==(other)
415 do
416 if not other isa Set[Object] then return false
417 if other.length != length then return false
418 return has_all(other)
419 end
420
421 # Because of the law between `==` and `hash`, `hash` is redefined to be the sum of the hash of the elements
422 redef fun hash
423 do
424 # 23 is a magic number empirically determined to be not so bad.
425 var res = 23 + length
426 # Note: the order of the elements must not change the hash value.
427 # So, unlike usual hash functions, the accumulator is not combined with itself.
428 for e in self do res += e.hash
429 return res
430 end
431
432 # Returns the union of this set with the `other` set
433 fun union(other: Set[E]): Set[E]
434 do
435 var nhs = new_set
436 nhs.add_all self
437 nhs.add_all other
438 return nhs
439 end
440
441 # Returns the intersection of this set with the `other` set
442 fun intersection(other: Set[E]): Set[E]
443 do
444 var nhs = new_set
445 for v in self do if other.has(v) then nhs.add(v)
446 return nhs
447 end
448
449 # Returns a new instance of `Set`.
450 #
451 # Depends on the subclass, mainly used for copy services
452 # like `union` or `intersection`.
453 protected fun new_set: Set[E] is abstract
454 end
455
456 # MapRead are abstract associative collections: `key` -> `item`.
457 interface MapRead[K, V]
458 # Get the item at `key`
459 #
460 # var x = new HashMap[String, Int]
461 # x["four"] = 4
462 # assert x["four"] == 4
463 # # assert x["five"] #=> abort
464 #
465 # If the key is not in the map, `provide_default_value` is called (that aborts by default)
466 # See `get_or_null` and `get_or_default` for safe variations.
467 fun [](key: K): V is abstract
468
469 # Get the item at `key` or null if `key` is not in the map.
470 #
471 # var x = new HashMap[String, Int]
472 # x["four"] = 4
473 # assert x.get_or_null("four") == 4
474 # assert x.get_or_null("five") == null
475 #
476 # Note: use `has_key` and `[]` if you need the distinction between a key associated with null, and no key.
477 fun get_or_null(key: K): nullable V
478 do
479 if has_key(key) then return self[key]
480 return null
481 end
482
483 # Get the item at `key` or return `default` if not in map
484 #
485 # var x = new HashMap[String, Int]
486 # x["four"] = 4
487 # assert x.get_or_default("four", 40) == 4
488 # assert x.get_or_default("five", 50) == 50
489 #
490 fun get_or_default(key: K, default: V): V
491 do
492 if has_key(key) then return self[key]
493 return default
494 end
495
496 # Is there an item associated with `key`?
497 #
498 # var x = new HashMap[String, Int]
499 # x["four"] = 4
500 # assert x.has_key("four") == true
501 # assert x.has_key("five") == false
502 #
503 # By default it is a synonymous to `keys.has` but could be redefined with a direct implementation.
504 fun has_key(key: K): Bool do return self.keys.has(key)
505
506 # Get a new iterator on the map.
507 fun iterator: MapIterator[K, V] is abstract
508
509 # Return the point of view of self on the values only.
510 # Note that `self` and `values` are views on the same data;
511 # therefore any modification of one is visible on the other.
512 #
513 # var x = new HashMap[String, Int]
514 # x["four"] = 4
515 # assert x.values.has(4) == true
516 # assert x.values.has(5) == false
517 fun values: Collection[V] is abstract
518
519 # Return the point of view of self on the keys only.
520 # Note that `self` and `keys` are views on the same data;
521 # therefore any modification of one is visible on the other.
522 #
523 # var x = new HashMap[String, Int]
524 # x["four"] = 4
525 # assert x.keys.has("four") == true
526 # assert x.keys.has("five") == false
527 fun keys: Collection[K] is abstract
528
529 # Is there no item in the collection?
530 #
531 # var x = new HashMap[String, Int]
532 # assert x.is_empty == true
533 # x["four"] = 4
534 # assert x.is_empty == false
535 fun is_empty: Bool is abstract
536
537 # Number of items in the collection.
538 #
539 # var x = new HashMap[String, Int]
540 # assert x.length == 0
541 # x["four"] = 4
542 # assert x.length == 1
543 # x["five"] = 5
544 # assert x.length == 2
545 fun length: Int is abstract
546
547 # Called by the underling implementation of `[]` to provide a default value when a `key` has no value
548 # By default the behavior is to abort.
549 #
550 # Note: the value is returned *as is*, implementations may want to store the value in the map before returning it
551 # @toimplement
552 protected fun provide_default_value(key: K): V do abort
553
554 # Does `self` and `other` have the same keys associated with the same values?
555 #
556 # ~~~
557 # var a = new HashMap[String, Int]
558 # var b = new ArrayMap[Object, Numeric]
559 # assert a == b
560 # a["one"] = 1
561 # assert a != b
562 # b["one"] = 1
563 # assert a == b
564 # b["one"] = 2
565 # assert a != b
566 # ~~~
567 redef fun ==(other)
568 do
569 if not other isa MapRead[nullable Object, nullable Object] then return false
570 if other.length != self.length then return false
571 for k, v in self do
572 if not other.has_key(k) then return false
573 if other[k] != v then return false
574 end
575 return true
576 end
577
578 # A hashcode based on the hashcode of the keys and the values.
579 #
580 # ~~~
581 # var a = new HashMap[String, Int]
582 # var b = new ArrayMap[Object, Numeric]
583 # a["one"] = 1
584 # b["one"] = 1
585 # assert a.hash == b.hash
586 # ~~~
587 redef fun hash
588 do
589 var res = length
590 for k, v in self do
591 if k != null then res += k.hash * 7
592 if v != null then res += v.hash * 11
593 end
594 return res
595 end
596 end
597
598 # Maps are associative collections: `key` -> `item`.
599 #
600 # The main operator over maps is [].
601 #
602 # var map: Map[String, Int] = new ArrayMap[String, Int]
603 # # ...
604 # map["one"] = 1 # Associate 'one' to '1'
605 # map["two"] = 2 # Associate 'two' to '2'
606 # assert map["one"] == 1
607 # assert map["two"] == 2
608 #
609 # Instances of maps can be used with the for structure
610 #
611 # for key, value in map do
612 # assert (key == "one" and value == 1) or (key == "two" and value == 2)
613 # end
614 #
615 # The keys and values in the map can also be manipulated directly with the `keys` and `values` methods.
616 #
617 # assert map.keys.has("one") == true
618 # assert map.keys.has("tree") == false
619 # assert map.values.has(1) == true
620 # assert map.values.has(3) == false
621 #
622 interface Map[K, V]
623 super MapRead[K, V]
624
625 # Set the `value` at `key`.
626 #
627 # Values can then get retrieved with `[]`.
628 #
629 # var x = new HashMap[String, Int]
630 # x["four"] = 4
631 # assert x["four"] == 4
632 #
633 # If the key was associated with a value, this old value is discarded
634 # and replaced with the new one.
635 #
636 # x["four"] = 40
637 # assert x["four"] == 40
638 # assert x.values.has(4) == false
639 #
640 fun []=(key: K, value: V) is abstract
641
642 # Add each (key,value) of `map` into `self`.
643 # If a same key exists in `map` and `self`, then the value in self is discarded.
644 #
645 # It is the analogous of `SimpleCollection::add_all`
646 #
647 # var x = new HashMap[String, Int]
648 # x["four"] = 4
649 # x["five"] = 5
650 # var y = new HashMap[String, Int]
651 # y["four"] = 40
652 # y["nine"] = 90
653 # x.recover_with y
654 # assert x["four"] == 40
655 # assert x["five"] == 5
656 # assert x["nine"] == 90
657 fun recover_with(map: MapRead[K, V])
658 do
659 var i = map.iterator
660 while i.is_ok do
661 self[i.key] = i.item
662 i.next
663 end
664 end
665
666 # Remove all items
667 #
668 # var x = new HashMap[String, Int]
669 # x["four"] = 4
670 # x.clear
671 # assert x.keys.has("four") == false
672 #
673 # ENSURE `is_empty`
674 fun clear is abstract
675
676 redef fun values: RemovableCollection[V] is abstract
677
678 redef fun keys: RemovableCollection[K] is abstract
679 end
680
681 # Iterators for Map.
682 interface MapIterator[K, V]
683 # The current item.
684 # Require `is_ok`.
685 fun item: V is abstract
686
687 # The key of the current item.
688 # Require `is_ok`.
689 fun key: K is abstract
690
691 # Jump to the next item.
692 # Require `is_ok`.
693 fun next is abstract
694
695 # Is there a current item ?
696 fun is_ok: Bool is abstract
697
698 # Set a new `item` at `key`.
699 #fun item=(item: E) is abstract
700
701 # Post-iteration hook.
702 #
703 # Used to inform `self` that the iteration is over.
704 # Specific iterators can use this to free some resources.
705 #
706 # Is automatically invoked at the end of `for` structures.
707 #
708 # Do nothing by default.
709 fun finish do end
710 end
711
712 # Iterator on a 'keys' point of view of a map
713 class MapKeysIterator[K, V]
714 super Iterator[K]
715 # The original iterator
716 var original_iterator: MapIterator[K, V]
717
718 redef fun is_ok do return self.original_iterator.is_ok
719 redef fun next do self.original_iterator.next
720 redef fun item do return self.original_iterator.key
721 end
722
723 # Iterator on a 'values' point of view of a map
724 class MapValuesIterator[K, V]
725 super Iterator[V]
726 # The original iterator
727 var original_iterator: MapIterator[K, V]
728
729 redef fun is_ok do return self.original_iterator.is_ok
730 redef fun next do self.original_iterator.next
731 redef fun item do return self.original_iterator.item
732 end
733
734 # Sequences are indexed collections.
735 # The first item is 0. The last is `length-1`.
736 #
737 # The order is the main caracteristic of sequence
738 # and all concrete implementation of sequences are basically interchangeable.
739 interface SequenceRead[E]
740 super Collection[E]
741
742 # Get the first item.
743 # Is equivalent with `self[0]`.
744 #
745 # var a = [1,2,3]
746 # assert a.first == 1
747 #
748 # REQUIRE `not is_empty`
749 redef fun first
750 do
751 assert not_empty: not is_empty
752 return self[0]
753 end
754
755 # Return the index-th element of the sequence.
756 # The first element is 0 and the last is `length-1`
757 # If index is invalid, the program aborts
758 #
759 # var a = [10,20,30]
760 # assert a[0] == 10
761 # assert a[1] == 20
762 # assert a[2] == 30
763 #
764 # REQUIRE `index >= 0 and index < length`
765 fun [](index: Int): E is abstract
766
767 # Get the last item.
768 # Is equivalent with `self[length-1]`.
769 #
770 # var a = [1,2,3]
771 # assert a.last == 3
772 #
773 # REQUIRE `not is_empty`
774 fun last: E
775 do
776 assert not_empty: not is_empty
777 return self[length-1]
778 end
779
780 # The index of the first occurrence of `item`.
781 # Return -1 if `item` is not found.
782 # Comparison is done with `==`.
783 #
784 # var a = [10,20,30,10,20,30]
785 # assert a.index_of(20) == 1
786 # assert a.index_of(40) == -1
787 fun index_of(item: E): Int do return index_of_from(item, 0)
788
789 # The index of the last occurrence of `item`.
790 # Return -1 if `item` is not found.
791 # Comparison is done with `==`.
792 #
793 # var a = [10,20,30,10,20,30]
794 # assert a.last_index_of(20) == 4
795 # assert a.last_index_of(40) == -1
796 fun last_index_of(item: E): Int do return last_index_of_from(item, length-1)
797
798 # The index of the first occurrence of `item`, starting from pos.
799 # Return -1 if `item` is not found.
800 # Comparison is done with `==`.
801 #
802 # var a = [10,20,30,10,20,30]
803 # assert a.index_of_from(20, 3) == 4
804 # assert a.index_of_from(20, 4) == 4
805 # assert a.index_of_from(20, 5) == -1
806 fun index_of_from(item: E, pos: Int): Int
807 do
808 var p = 0
809 var i = iterator
810 while i.is_ok do
811 if p>=pos and i.item == item then return i.index
812 i.next
813 p += 1
814 end
815 return -1
816 end
817
818 # The index of the last occurrence of `item` starting from `pos` and decrementing.
819 # Return -1 if `item` is not found.
820 # Comparison is done with `==`.
821 #
822 # var a = [10,20,30,10,20,30]
823 # assert a.last_index_of_from(20, 2) == 1
824 # assert a.last_index_of_from(20, 1) == 1
825 # assert a.last_index_of_from(20, 0) == -1
826 fun last_index_of_from(item: E, pos: Int): Int
827 do
828 var res = -1
829 var p = 0
830 var i = iterator
831 while i.is_ok do
832 if p>pos then break
833 if i.item == item then res = p
834 i.next
835 p += 1
836 end
837 return res
838 end
839
840 # Two sequences are equals if they have the same items in the same order.
841 #
842 # var a = new List[Int]
843 # a.add(1)
844 # a.add(2)
845 # a.add(3)
846 # assert a == [1,2,3]
847 # assert a != [1,3,2]
848 redef fun ==(o)
849 do
850 if not o isa SequenceRead[nullable Object] then return false
851 var l = length
852 if o.length != l then return false
853 var i = 0
854 while i < l do
855 if self[i] != o[i] then return false
856 i += 1
857 end
858 return true
859 end
860
861 # Because of the law between `==` and `hash`, `hash` is redefined to be the sum of the hash of the elements
862 redef fun hash
863 do
864 # The 17 and 2/3 magic numbers were determined empirically.
865 # Note: the standard hash functions djb2, sbdm and fnv1 were also
866 # tested but were comparable (or worse).
867 var res = 17 + length
868 for e in self do
869 res = res * 3 / 2
870 if e != null then res += e.hash
871 end
872 return res
873 end
874
875 redef fun iterator: IndexedIterator[E] is abstract
876
877 # Gets a new Iterator starting at position `pos`
878 #
879 # var iter = [10,20,30,40,50].iterator_from(2)
880 # assert iter.to_a == [30, 40, 50]
881 fun iterator_from(pos: Int): IndexedIterator[E]
882 do
883 var res = iterator
884 while pos > 0 and res.is_ok do
885 res.next
886 pos -= 1
887 end
888 return res
889 end
890
891 # Gets an iterator starting at the end and going backwards
892 #
893 # var reviter = [1,2,3].reverse_iterator
894 # assert reviter.to_a == [3,2,1]
895 fun reverse_iterator: IndexedIterator[E] is abstract
896
897 # Gets an iterator on the chars of self starting from `pos`
898 #
899 # var reviter = [10,20,30,40,50].reverse_iterator_from(2)
900 # assert reviter.to_a == [30,20,10]
901 fun reverse_iterator_from(pos: Int): IndexedIterator[E]
902 do
903 var res = reverse_iterator
904 while pos > 0 and res.is_ok do
905 res.next
906 pos -= 1
907 end
908 return res
909 end
910 end
911
912 # Sequence are indexed collection.
913 # The first item is 0. The last is `length-1`.
914 interface Sequence[E]
915 super SequenceRead[E]
916 super SimpleCollection[E]
917
918 # Set the first item.
919 # Is equivalent with `self[0] = item`.
920 #
921 # var a = [1,2,3]
922 # a.first = 10
923 # assert a == [10,2,3]
924 fun first=(item: E)
925 do self[0] = item end
926
927 # Set the last item.
928 # Is equivalent with `self[length-1] = item`.
929 #
930 # var a = [1,2,3]
931 # a.last = 10
932 # assert a == [1,2,10]
933 #
934 # If the sequence is empty, `last=` is equivalent with `self[0]=` (thus with `first=`)
935 #
936 # var b = new Array[Int]
937 # b.last = 10
938 # assert b == [10]
939 fun last=(item: E)
940 do
941 var l = length
942 if l > 0 then
943 self[l-1] = item
944 else
945 self[0] = item
946 end
947 end
948
949 # A synonym of `push`
950 redef fun add(e) do push(e)
951
952 # Add an item after the last one.
953 #
954 # var a = [1,2,3]
955 # a.push(10)
956 # a.push(20)
957 # assert a == [1,2,3,10,20]
958 fun push(e: E) is abstract
959
960 # Add each item of `coll` after the last.
961 #
962 # var a = [1,2,3]
963 # a.append([7..9])
964 # assert a == [1,2,3,7,8,9]
965 #
966 # Alias of `add_all`
967 fun append(coll: Collection[E]) do add_all(coll)
968
969 # Remove the last item.
970 #
971 # var a = [1,2,3]
972 # assert a.pop == 3
973 # assert a.pop == 2
974 # assert a == [1]
975 #
976 # REQUIRE `not is_empty`
977 fun pop: E is abstract
978
979 # Add an item before the first one.
980 #
981 # var a = [1,2,3]
982 # a.unshift(10)
983 # a.unshift(20)
984 # assert a == [20,10,1,2,3]
985 fun unshift(e: E) is abstract
986
987 # Add all items of `coll` before the first one.
988 #
989 # var a = [1,2,3]
990 # a.prepend([7..9])
991 # assert a == [7,8,9,1,2,3]
992 #
993 # Alias of `insert_at(coll, 0)`
994 fun prepend(coll: Collection[E]) do insert_all(coll, 0)
995
996 # Remove the first item.
997 # The second item thus become the first.
998 #
999 # var a = [1,2,3]
1000 # assert a.shift == 1
1001 # assert a.shift == 2
1002 # assert a == [3]
1003 #
1004 # REQUIRE `not is_empty`
1005 fun shift: E is abstract
1006
1007 # Set the `item` at `index`.
1008 #
1009 # var a = [10,20,30]
1010 # a[1] = 200
1011 # assert a == [10,200,30]
1012 #
1013 # like with `[]`, index should be between `0` and `length-1`
1014 # However, if `index==length`, `[]=` works like `push`.
1015 #
1016 # a[3] = 400
1017 # assert a == [10,200,30,400]
1018 #
1019 # REQUIRE `index >= 0 and index <= length`
1020 fun []=(index: Int, item: E) is abstract
1021
1022 # Insert an element at a given position, following elements are shifted.
1023 #
1024 # var a = [10, 20, 30, 40]
1025 # a.insert(100, 2)
1026 # assert a == [10, 20, 100, 30, 40]
1027 #
1028 # REQUIRE `index >= 0 and index <= length`
1029 # ENSURE `self[index] == item`
1030 fun insert(item: E, index: Int) is abstract
1031
1032 # Insert all elements at a given position, following elements are shifted.
1033 #
1034 # var a = [10, 20, 30, 40]
1035 # a.insert_all([100..102], 2)
1036 # assert a == [10, 20, 100, 101, 102, 30, 40]
1037 #
1038 # REQUIRE `index >= 0 and index <= length`
1039 # ENSURE `self[index] == coll.first`
1040 fun insert_all(coll: Collection[E], index: Int)
1041 do
1042 assert index >= 0 and index < length
1043 if index == length then
1044 add_all(coll)
1045 end
1046 for c in coll do
1047 insert(c, index)
1048 index += 1
1049 end
1050 end
1051
1052 # Remove the item at `index` and shift all following elements
1053 #
1054 # var a = [10,20,30]
1055 # a.remove_at(1)
1056 # assert a == [10,30]
1057 #
1058 # REQUIRE `index >= 0 and index < length`
1059 fun remove_at(index: Int) is abstract
1060 end
1061
1062 # Iterators on indexed collections.
1063 interface IndexedIterator[E]
1064 super Iterator[E]
1065 # The index of the current item.
1066 fun index: Int is abstract
1067 end
1068
1069 # Associative arrays that internally uses couples to represent each (key, value) pairs.
1070 # This is an helper class that some specific implementation of Map may implements.
1071 interface CoupleMap[K, V]
1072 super Map[K, V]
1073
1074 # Return the couple of the corresponding key
1075 # Return null if the key is no associated element
1076 protected fun couple_at(key: K): nullable Couple[K, V] is abstract
1077
1078 # Return a new iteralot on all couples
1079 # Used to provide `iterator` and others
1080 protected fun couple_iterator: Iterator[Couple[K,V]] is abstract
1081
1082 redef fun iterator do return new CoupleMapIterator[K,V](couple_iterator)
1083
1084 redef fun [](key)
1085 do
1086 var c = couple_at(key)
1087 if c == null then
1088 return provide_default_value(key)
1089 else
1090 return c.second
1091 end
1092 end
1093
1094 redef fun has_key(key) do return couple_at(key) != null
1095 end
1096
1097 # Iterator on CoupleMap
1098 #
1099 # Actually it is a wrapper around an iterator of the internal array of the map.
1100 private class CoupleMapIterator[K, V]
1101 super MapIterator[K, V]
1102 redef fun item do return _iter.item.second
1103
1104 #redef fun item=(e) do _iter.item.second = e
1105
1106 redef fun key do return _iter.item.first
1107
1108 redef fun is_ok do return _iter.is_ok
1109
1110 redef fun next
1111 do
1112 _iter.next
1113 end
1114
1115 var iter: Iterator[Couple[K,V]]
1116 end
1117
1118 # Some tools ###################################################################
1119
1120 # Two objects in a simple structure.
1121 class Couple[F, S]
1122
1123 # The first element of the couple.
1124 var first: F is writable
1125
1126 # The second element of the couple.
1127 var second: S is writable
1128 end