From bc910f7aef5dac064f3db47ce601b6ef78c14ff5 Mon Sep 17 00:00:00 2001
From: Son Ho
Date: Thu, 25 Nov 2021 14:35:43 +0100
Subject: Implement various substitution functions

---
 src/Identifiers.ml | 27 +++++++++++++++++++--------
 1 file changed, 19 insertions(+), 8 deletions(-)

(limited to 'src/Identifiers.ml')

diff --git a/src/Identifiers.ml b/src/Identifiers.ml
index 6f74e062..23c887c4 100644
--- a/src/Identifiers.ml
+++ b/src/Identifiers.ml
@@ -22,6 +22,8 @@ module type Id = sig
 
   val to_int : id -> int
 
+  val of_int : id -> int
+
   val empty_vector : 'a vector
 
   val vector_to_list : 'a vector -> 'a list
@@ -35,15 +37,23 @@ module type Id = sig
   val nth_opt : 'a vector -> id -> 'a option
 
   val update_nth : 'a vector -> id -> 'a -> 'a vector
+  (** Update the nth element of the vector.
+
+      Raises [Invalid_argument] if the identifier is out of range.
+   *)
 
   val iter : ('a -> unit) -> 'a vector -> unit
 
   val map : ('a -> 'b) -> 'a vector -> 'b vector
 
+  val mapi : (id -> 'a -> 'b) -> 'a vector -> 'b vector
+
   val for_all : ('a -> bool) -> 'a vector -> bool
 
   val exists : ('a -> bool) -> 'a vector -> bool
 
+  module Ord : Map.OrderedType with type t = id
+
   module Set : Set.S with type elt = id
 
   val set_to_string : Set.t -> string
@@ -84,6 +94,8 @@ module IdGen () : Id = struct
 
   let to_int x = x
 
+  let of_int x = x
+
   let empty_vector = []
 
   let vector_to_list v = v
@@ -98,7 +110,7 @@ module IdGen () : Id = struct
 
   let rec update_nth vec id v =
     match (vec, id) with
-    | [], _ -> failwith "Unreachable"
+    | [], _ -> raise (Invalid_argument "Out of range")
     | _ :: vec', 0 -> v :: vec'
     | x :: vec', _ -> x :: update_nth vec' (id - 1) v
 
@@ -106,21 +118,20 @@ module IdGen () : Id = struct
 
   let map = List.map
 
+  let mapi = List.mapi
+
   let for_all = List.for_all
 
   let exists = List.exists
 
-  module Set = Set.Make (struct
+  module Ord = struct
     type t = id
 
     let compare = compare
-  end)
-
-  module Map = Map.Make (struct
-    type t = id
+  end
 
-    let compare = compare
-  end)
+  module Set = Set.Make (Ord)
+  module Map = Map.Make (Ord)
 
   let set_to_string ids =
     let ids = Set.fold (fun id ids -> to_string id :: ids) ids [] in
-- 
cgit v1.2.3