aboutsummaryrefslogtreecommitdiffhomepage
path: root/lib/dAst.ml
blob: 0fe323d0134637e0ea222d9bb68d62d35f12827c (plain)
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
(************************************************************************)
(*  v      *   The Coq Proof Assistant  /  The Coq Development Team     *)
(* <O___,, *   INRIA - CNRS - LIX - LRI - PPS - Copyright 1999-2017     *)
(*   \VV/  **************************************************************)
(*    //   *      This file is distributed under the terms of the       *)
(*         *       GNU Lesser General Public License Version 2.1        *)
(************************************************************************)

open CAst

type ('a, _) thunk =
| Value : 'a -> ('a, 'b) thunk
| Thunk : 'a Lazy.t -> ('a, [ `thunk ]) thunk

type ('a, 'b) t = ('a, 'b) thunk CAst.t

let map_thunk (type s) f : (_, s) thunk -> (_, s) thunk = function
| Value x -> Value (f x)
| Thunk k -> Thunk (lazy (f (Lazy.force k)))

let get_thunk (type s) : ('a, s) thunk -> 'a = function
| Value x -> x
| Thunk k -> Lazy.force k

let get x = get_thunk x.v

let make ?loc v = CAst.make ?loc (Value v)

let delay ?loc v = CAst.make ?loc (Thunk (Lazy.from_fun v))

let map f n = CAst.map (fun x -> map_thunk f x) n

let map_with_loc f n =
  CAst.map_with_loc (fun ?loc x -> map_thunk (fun x -> f ?loc x) x) n

let map_from_loc f (loc, x) =
  make ?loc (f ?loc x)

let with_val f n = f (get n)

let with_loc_val f n = f ?loc:n.CAst.loc (get n)