我正在尝试转储 OCaml 项目中所有标识符的类型信息,基本上它与遍历类型化抽象语法树相同(https://github.com/ocaml/ocaml/blob/trunk/typing/typedtree.mli) . 由于我是 OCaml 编译器代码库的新手,我不确定编译器是否提供了 api,因此我们可以轻松编写插件来完成这项工作,或者我们必须破解编译器代码?此外,这如何与 OCamlbuild 交互?感谢您的任何提示或建议。
2 回答
假设您已经structure
以某种方式获得了类型化的 AST。
经典的方法就是自己写一个大的递归函数来遍历AST。
但是现在TypedtreeIter
OCaml 编译器源代码中有可用的模块,并且它暴露在compiler-libs
. 对于简单的遍历,这非常方便。
TypedtreeIter
提供了一个仿函数来在类型化 AST 上构建您自己的迭代器。这是一个非常简单的示例,用于打印所有模式标识符及其类型:
(* ocamlfind ocamlc -package compiler-libs.common -c example.ml *)
open Typedtree
open TypedtreeIter
module MyIteratorArgument = struct
include DefaultIteratorArgument
let enter_pattern p = match p.pat_desc with
| Tpat_var (id, _) ->
Format.printf "@[<2>%s@ : %a@]@."
(Ident.name id)
Printtyp.type_scheme p.pat_type
| _ -> ()
end
module Iterator = TypedtreeIter.MakeIterator(MyIteratorArgument)
模块类型TypedtreeIter.IteratorArgument
是指定你的迭代器对每个 AST 构造做什么。你有两点来执行你的工作:当遍历进入一个构造时,以及当它从构造中退出时。例如pattern
,您有enter_pattern
和exit_pattern
。您无需担心递归遍历本身:这是 functor 的工作MakeIterator
。给一个IteratorArgument
模块,它会递归地连接所有模块,enter_*
并exit_*
返回一个带有一堆迭代器的模块。
通常你只对 AST 的某些部分感兴趣而想跳过其他部分。DefaultIteratorArgument
是一个模块,它enter_*
什么exit_*
也不做。你的IteratorArgument
模块应该包括DefaultIteratorArgument
继承这个默认行为,然后只实现一些特殊的部分。
如果您不仅想遍历类型化的 AST,还想修改其中的某些部分,请使用TypedtreeMap
而不是TypedtreeIter
. TypedtreeMap
在https://bitbucket.org/camlspotter/compiler-libs-hack/src/340072a7c14cbce624b98a57bf8c4c6509c40a31/overload/mod.ml?at=default有一个小例子。
(我不使用 ocamlbuild,所以我无能为力。)
OCaml 提供自己的编译器作为名为compiler-libs
. 它包含所有内容,允许您从具体语法转向可执行,所有中间步骤都在您的控制之下,当然包括类型树。
坏消息是它没有记录。我建议你使用utop
或merlin
探索这个库。
使用 ocamlbuild 不需要任何特殊的东西compiler-libs
,它是一个常规库。