Source file smtml_prelude.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
(* SPDX-License-Identifier: MIT *)
(* Copyright (C) 2023-2026 formalsec *)
(* Written by the Smtml programmers *)

include Prelude

module String = struct
  include String

  let hash x = String_hash.seeded_hash 0 x
end

module Int32 = struct
  include Int32

  (* As they are defined in the stdlib *)
  let hash x = Hashtbl.seeded_hash_param 10 100 0 x
end

module Int64 = struct
  include Int64

  (* As they are defined in the stdlib *)
  let hash x = Hashtbl.seeded_hash_param 10 100 0 x
end

module Option = struct
  include Option

  module Syntax = struct
    let ( let* ) v f = bind v f

    let ( let+ ) v f = map f v
  end
end

module Result = struct
  include Result

  module Syntax = struct
    let ( let* ) v f = Result.bind v f

    let ( let+ ) v f = Result.map f v

    let ( >>= ) = Result.bind

    let ( >>| ) r f = Result.map f r
  end

  let rec list_iter f =
    let open Syntax in
    function
    | [] -> Ok ()
    | hd :: tl ->
      let* () = f hd in
      list_iter f tl

  let list_map f v =
    let open Syntax in
    let rec list_map_cps f v k =
      match v with
      | [] -> k (Ok [])
      | hd :: tl ->
        list_map_cps f tl (fun rest ->
          let* rest in
          let* hd' = f hd in
          k (Ok (hd' :: rest)) )
    in
    list_map_cps f v Fun.id

  let list_filter_map f v =
    let open Syntax in
    let rec list_filter_map_cps f v k =
      match v with
      | [] -> k (Ok [])
      | hd :: tl ->
        list_filter_map_cps f tl (fun rest ->
          let* rest in
          let* v = f hd in
          k (Ok (match v with None -> rest | Some v -> v :: rest)) )
    in
    list_filter_map_cps f v Fun.id
end