Source file `ppx_z_literals.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
(**************************************************************************)
(*                                                                        *)
(*  SPDX-License-Identifier LGPL-2.1                                      *)
(*  Copyright (C)                                                         *)
(*  CEA (Commissariat à l'énergie atomique et aux énergies alternatives)  *)
(*                                                                        *)
(**************************************************************************)

open Ppxlib

(** We use {!Fc_internal_z} instead of {!Z} to allow using this ppx inside
    [z.ml]. *)
let conv_to_z  loc = [%expr Fc_internal_z.of_int [@alert "-fc_internal_z"]]

let rewriter conv loc s =
  let number =
    Ast_builder.Default.pexp_constant ~loc
      (Parsetree.Pconst_integer (s, None))
  in
  [%expr [%e (conv loc)] [%e number]]

let rule (ch, conv) =
  Ppxlib.Context_free.Rule.(constant Constant_kind.Integer ch (rewriter conv))

let rules =
  List.map rule  [ ('z', conv_to_z) ]

let () =
  Driver.register_transformation ~rules
    "Constant rewriting for Zarith.Z.t from int"