Source file `cross_product.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
open Base

module type S = Cross_product_intf.S

module Make (Outer : Interface.S) (Inner : Interface.S) = struct
  module type S = Interface.S with type 'a t = 'a Inner.t Outer.t

  module Pre = struct
    type 'a t = 'a Inner.t Outer.t [@@deriving sexp_of]

    let map t ~f = Outer.map t ~f:(Inner.map ~f)
    let iter t ~f = Outer.iter t ~f:(Inner.iter ~f)
    let to_list t = List.concat_map ~f:Inner.to_list (Outer.to_list t)

    let port_names_and_widths =
      Outer.map Outer.port_names_and_widths ~f:(fun (n, _) ->
        Inner.map Inner.port_names_and_widths ~f:(fun (m, b) -> n ^ "_" ^ m, b))
    ;;

    let map2 a b ~f = Outer.map2 a b ~f:(Inner.map2 ~f)
    let iter2 a b ~f = Outer.iter2 a b ~f:(Inner.iter2 ~f)
  end

  include Pre
  include Interface.Make (Pre)
end