You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.6 KiB
59 lines
1.6 KiB
(*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*)
|
|
|
|
open Import0
|
|
include Map_intf
|
|
|
|
module Make (Key : sig
|
|
type t [@@deriving compare, sexp_of]
|
|
end) : S with type key = Key.t = struct
|
|
module KeyMap = Core.Map.Make_plain (Key)
|
|
module Key = KeyMap.Key
|
|
|
|
type key = Key.t
|
|
|
|
include KeyMap.Tree
|
|
|
|
let compare = compare_direct
|
|
|
|
let to_map t =
|
|
Core.Map.Using_comparator.of_tree ~comparator:Key.comparator t
|
|
|
|
let of_map m = Base.Map.Using_comparator.to_tree m
|
|
|
|
let merge_skewed x y ~combine =
|
|
of_map (Core.Map.merge_skewed (to_map x) (to_map y) ~combine)
|
|
|
|
let find_and_remove m k =
|
|
let found = ref None in
|
|
let m =
|
|
change m k ~f:(fun v ->
|
|
found := v ;
|
|
None )
|
|
in
|
|
Option.map ~f:(fun v -> (v, m)) !found
|
|
|
|
let pp pp_k pp_v fs m =
|
|
Format.fprintf fs "@[<1>[%a]@]"
|
|
(List.pp ",@ " (fun fs (k, v) ->
|
|
Format.fprintf fs "@[%a @<2>↦ %a@]" pp_k k pp_v v ))
|
|
(to_alist m)
|
|
|
|
let pp_diff ~data_equal pp_key pp_val pp_diff_val fs (x, y) =
|
|
let pp_diff_elt fs = function
|
|
| k, `Left v ->
|
|
Format.fprintf fs "-- [@[%a@ @<2>↦ %a@]]" pp_key k pp_val v
|
|
| k, `Right v ->
|
|
Format.fprintf fs "++ [@[%a@ @<2>↦ %a@]]" pp_key k pp_val v
|
|
| k, `Unequal vv ->
|
|
Format.fprintf fs "[@[%a@ @<2>↦ %a@]]" pp_key k pp_diff_val vv
|
|
in
|
|
let sd = Sequence.to_list (symmetric_diff ~data_equal x y) in
|
|
if not (List.is_empty sd) then
|
|
Format.fprintf fs "[@[<hv>%a@]];@ " (List.pp ";@ " pp_diff_elt) sd
|
|
end
|