open Internal.Utilities.Collections.Tagged

open System.Collections.Generic

type internal Zmap<'Key,'T> = Internal.Utilities.Collections.Tagged.Map<'Key,'T>

module internal Zmap =

let empty (ord: IComparer<'T>) = Map<_,_,_>.Empty(ord)

let add k v (m: Zmap<_,_>) = m.Add(k,v)

let find k (m: Zmap<_,_>) = m.[k]

let tryFind k (m: Zmap<_,_>) = m.TryFind(k)

let remove k (m: Zmap<_,_>) = m.Remove(k)

let mem k (m: Zmap<_,_>) = m.ContainsKey(k)

let iter f (m: Zmap<_,_>) = m.Iterate(f)

let first f (m: Zmap<_,_>) = m.First(fun k v -> if f k v then Some (k,v) else None)

let exists f (m: Zmap<_,_>) = m.Exists(f)

let forall f (m: Zmap<_,_>) = m.ForAll(f)

let map f (m: Zmap<_,_>) = m.MapRange(f)

let mapi f (m: Zmap<_,_>) = m.Map(f)

let fold f (m: Zmap<_,_>) x = m.Fold f x

let toList (m: Zmap<_,_>) = m.ToList()

let foldSection lo hi f (m: Zmap<_,_>) x = m.FoldSection lo hi f x

let isEmpty (m: Zmap<_,_>) = m.IsEmpty

let foldMap f z (m: Zmap<_,_>) =

let m,z = m.FoldAndMap (fun k v z -> let z,v' = f z k v in v',z) z in

z,m

let choose f (m: Zmap<_,_>) = m.First(f)

let chooseL f (m: Zmap<_,_>) =

m.Fold (fun k v s -> match f k v with None -> s | Some x -> x :: s) []

let ofList ord xs = Internal.Utilities.Collections.Tagged.Map<_,_>.FromList(ord,xs)

let keys (m: Zmap<_,_>) = m.Fold (fun k _ s -> k :: s) []

let values (m: Zmap<_,_>) = m.Fold (fun _ v s -> v :: s) []

let memberOf m k = mem k m