[sledge] Add Trm.Map

Reviewed By: ngorogiannis

Differential Revision: D24532355

fbshipit-source-id: 5939e41e5

sledge/src/trm.ml

@@ -386,6 +386,15 @@ end = struct
   let vars e = Iter.from_labelled_iter (iter_vars e)
 end
 
+module T = struct
+  type t = Trm.t [@@deriving compare, sexp]
+end
+
+module Map = struct
+  include Map.Make (T)
+  include Provide_of_sexp (T)
+end
+
 type arith = Arith.t
 
 include Trm

sledge/src/trm.mli

@@ -41,6 +41,12 @@ end
 module Arith :
   Arithmetic.S with type var := Var.t with type trm := t with type t = arith
 
+module Map : sig
+  include Map.S with type key := t
+
+  val t_of_sexp : (Sexp.t -> 'a) -> Sexp.t -> 'a t
+end
+
 val ppx : Var.strength -> t pp
 val pp : t pp
 val pp_diff : (t * t) pp