From 94c18ad233b9e16ee9263ba8b717630b4adbf91e Mon Sep 17 00:00:00 2001
From: Bernhard Schommer <bernhardschommer@gmail.com>
Date: Fri, 5 Jan 2018 16:51:32 +0100
Subject: Change AsmToJson to be similar to other printers.

---
 arm/AsmToJSON.ml  | 14 +++++++++++---
 arm/AsmToJSON.mli |  8 ++++++--
 2 files changed, 17 insertions(+), 5 deletions(-)

(limited to 'arm')

diff --git a/arm/AsmToJSON.ml b/arm/AsmToJSON.ml
index 85ef0603..c11162d6 100644
--- a/arm/AsmToJSON.ml
+++ b/arm/AsmToJSON.ml
@@ -302,9 +302,17 @@ let pp_instructions pp ic =
   in
     pp_jarray instruction pp ic
 
-let pp_program pp prog =
-  reset_id ();
-  JsonAST.pp_program pp pp_instructions prog
+let destination : string option ref = ref None
+let sdump_folder : string ref = ref ""
+
+let print_if prog sourcename =
+  match !destination with
+  | None -> ()
+  | Some f ->
+    let f = Filename.concat !sdump_folder f in
+    let oc = open_out f in
+    JsonAST.pp_ast (Format.formatter_of_out_channel oc) pp_instructions prog sourcename;
+    close_out oc
 
 let pp_mnemonics pp =
   JsonAST.pp_mnemonics pp mnemonic_names
diff --git a/arm/AsmToJSON.mli b/arm/AsmToJSON.mli
index 058a4e83..52c055c4 100644
--- a/arm/AsmToJSON.mli
+++ b/arm/AsmToJSON.mli
@@ -10,6 +10,10 @@
 (*                                                                     *)
 (* *********************************************************************)
 
-val pp_program: Format.formatter -> (Asm.coq_function AST.fundef, 'a) AST.program -> unit
-
 val pp_mnemonics: Format.formatter -> unit
+
+val print_if:  (Asm.coq_function AST.fundef, 'a) AST.program  -> string -> unit
+
+val destination: string option ref
+
+val sdump_folder : string ref
-- 
cgit