view src/multimap_fn.sml @ 2235:0aae15c2a05a

Refactored a lot and fixed an and/or swap, but still not good on current test.
author Ziv Scully <ziv@mit.edu>
date Mon, 29 Jun 2015 01:33:47 -0700
parents 365727ff68f4
children
line wrap: on
line source
functor MultimapFn (structure KeyMap : ORD_MAP structure ValSet : ORD_SET) = struct
    type key = KeyMap.Key.ord_key
    type item = ValSet.item
    type itemSet = ValSet.set
    type multimap = ValSet.set KeyMap.map
    val empty : multimap = KeyMap.empty
    fun insertSet (kToVs : multimap, k : key, vs : itemSet) : multimap =
        KeyMap.unionWith ValSet.union (kToVs, KeyMap.singleton (k, vs))
    fun insert (kToVs : multimap, k : key, v : item) : multimap =
        insertSet (kToVs, k, ValSet.singleton v)
    fun findSet (kToVs : multimap, k : key) =
        case KeyMap.find (kToVs, k) of
            SOME vs => vs
          | NONE => ValSet.empty
    val findList : multimap * key -> item list = ValSet.listItems o findSet
end