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