Implement Set stdlib (#335)

* Implement Set stdlib

* Rename an argument of the function Set.fold

* Add docs for Set stdlib

* Correct the usage of articles in the docs

* Fix bug

* Fix the link to Set stdlib section

Co-authored-by: Radosław Rowicki <35342116+radrow@users.noreply.github.com>
This commit is contained in:
Gaith Hallak 2021-08-07 11:08:45 +03:00 committed by GitHub
parent 4896ad3b36
commit 3029bf31cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 175 additions and 0 deletions

View File

@ -6,6 +6,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## [Unreleased]
### Added
- `Set` stdlib
- `Option.force_msg`
### Changed
### Removed

View File

@ -39,6 +39,7 @@ include "List.aes"
- [Triple](#Triple)
- [BLS12_381](#BLS12_381)
- [Frac](#Frac)
- [Set](#set-stdlib)
# Builtin namespaces
@ -2274,3 +2275,125 @@ accept arbitrary `frac`s from the surface you should report it as a
If you expect getting calls with malformed `frac`s in your contract, you should use
this function to verify the input.
## <a name='set-stdlib'>Set</a>
### Types
```
record set('a) = { to_map : map('a, unit) }
```
### Functions
#### new
```
Set.new() : set('a)
```
Returns an empty set
#### member
```
member(e : 'a, s : set('a)) : bool
```
Checks if the element `e` is present in the set `s`
#### insert
```
insert(e : 'a, s : set('a)) : set('a)
```
Inserts the element `e` in the set `s`
#### delete
```
Set.delete(e : 'a, s : set('a)) : set('a)
```
Removes the element `e` from the set `s`
#### size
```
size(s : set('a)) : int
```
Returns the number of elements in the set `s`
#### to_list
```
Set.to_list(s : set('a)) : list('a)
```
Returns a list containing the elements of the set `s`
#### from_list
```
Set.from_list(l : list('a)) : set('a)
```
Turns the list `l` into a set
#### filter
```
Set.filter(p : 'a => bool, s : set('a)) : set('a)
```
Filters out elements of `s` that fulfill predicate `p`
#### fold
```
Set.fold(f : ('a, 'b) => 'b, acc : 'b, s : set('a)) : 'b
```
Folds the function `f` over every element in the set `s` and returns the final value of the accumulator `acc`.
#### subtract
```
Set.subtract(s1 : set('a), s2 : set('a)) : set('a)
```
Returns the elements of `s1` that are not members of `s2`
#### intersection
```
Set.intersection(s1 : set('a), s2 : set('a)) : set('a)
```
Returns the intersection of the two sets `s1` and `s2`
#### intersection_list
```
Set.intersection_list(sets : list(set('a))) : set('a)
```
Returns the intersection of all the sets in the given list
#### union
```
Set.union(s1 : set('a), s2 : set('a)) : set('a)
```
Returns the union of the two sets `s1` and `s2`
#### union_list
```
Set.union_list(sets : list(set('a))) : set('a)
```
Returns the union of all the sets in the given list

51
priv/stdlib/Set.aes Normal file
View File

@ -0,0 +1,51 @@
include "List.aes"
include "Option.aes"
include "Pair.aes"
namespace Set =
record set('a) = { to_map : map('a, unit) }
function new() : set('a) =
{ to_map = {} }
function member(e : 'a, s : set('a)) : bool =
Map.member(e, s.to_map)
function insert(e : 'a, s : set('a)) : set('a) =
{ to_map = s.to_map{[e] = ()} }
function delete(e : 'a, s : set('a)) : set('a) =
{ to_map = Map.delete(e, s.to_map) }
function size(s : set('a)) : int =
Map.size(s.to_map)
function to_list(s : set('a)) : list('a) =
List.map(Pair.fst, Map.to_list(s.to_map))
function from_list(l : list('a)) : set('a) =
{ to_map = Map.from_list(List.map((x) => (x, ()), l)) }
function filter(p : 'a => bool, s : set('a)) : set('a) =
from_list(List.filter(p, to_list(s)))
function fold(f : ('a, 'b) => 'b, acc : 'b, s : set('a)) : 'b =
List.foldr(f, acc, to_list(s))
function subtract(s1 : set('a), s2 : set('a)) : set('a) =
filter((x) => !member(x, s2), s1)
function intersection(s1 : set('a), s2 : set('a)) : set('a) =
filter((x) => member(x, s2), s1)
function intersection_list(sets : list(set('a))) : set('a) =
List.foldr(
intersection,
Option.default(new(), List.first(sets)),
Option.default([], List.tail(sets)))
function union(s1 : set('a), s2 : set('a)) : set('a) =
from_list(to_list(s1) ++ to_list(s2))
function union_list(sets : list(set('a))) : set('a) =
List.foldr(union, new(), sets)