added support for commands and scripts

[helm.git] / helm / ocaml / cic_disambiguation / test_parser.ml

diff --git a/helm/ocaml/cic_disambiguation/test_parser.ml b/helm/ocaml/cic_disambiguation/test_parser.ml
index d6afddc670607b1099de6e7384f3bb9f28744458..dde96955506853815b0f4bebe7f0a29a8f428233 100644 (file)
--- a/helm/ocaml/cic_disambiguation/test_parser.ml
+++ b/helm/ocaml/cic_disambiguation/test_parser.ml
@@ -30,7 +30,8 @@ let mode = ref default_mode
 (* let pp_tactical = TacticAstPp.pp_tactical *)
 
 let pp_tactical = TacticAst2Box.tacticalPp
-
+let pp_command = CommandAst.pp_command
+let pp_script = CommandAst.pp_script
 
 let _ =
   try
@@ -39,6 +40,8 @@ let _ =
     | "term" -> mode := `Term
     | "tactic" -> mode := `Tactic
     | "tactical" -> mode := `Tactical
+    | "command" -> mode := `Command
+    | "script" -> mode := `Script
     | _ ->
         prerr_endline "What???????";
         exit 1
@@ -47,32 +50,40 @@ let _ =
 let _ =
   let ic = stdin in
   try
-    while true do
-      try
-        let line = input_line ic in
-        let istream = Stream.of_string line in
-        (match !mode with
-        | `Term ->
-            let term = CicTextualParser2.parse_term istream in
-            print_endline (BoxPp.pp_term term)
-        | `Tactic ->
-            let term = CicTextualParser2.parse_tactic istream in
-            print_endline (TacticAstPp.pp_tactic term)
-        | `Tactical ->
-            let term = CicTextualParser2.parse_tactical istream in
-            print_endline (pp_tactical term)
-        | `Alias ->
-            let env = CicTextualParser2.EnvironmentP3.of_string line in
-            print_endline (CicTextualParser2.EnvironmentP3.to_string env));
-        flush stdout
-      with
-      | CicTextualParser2.Parse_error msg -> prerr_endline msg
-(*
-      | Stdpp.Exc_located ((p_start, p_end), exn) ->
-        prerr_endline (Printf.sprintf "Exception at character %d-%d: %s"
-          p_start p_end (Printexc.to_string exn))
-*)
-    done
+    if !mode =  `Script then begin
+      let script = CicTextualParser2.parse_script (Stream.of_channel ic) in
+      print_endline (pp_script script)
+    end else
+      while true do
+        try
+          let line = input_line ic in
+          let istream = Stream.of_string line in
+          (match !mode with
+          | `Term ->
+              let term = CicTextualParser2.parse_term istream in
+              print_endline (BoxPp.pp_term term)
+          | `Tactic ->
+              let tac = CicTextualParser2.parse_tactic istream in
+              print_endline (TacticAstPp.pp_tactic tac)
+          | `Tactical ->
+              let tac = CicTextualParser2.parse_tactical istream in
+              print_endline (pp_tactical tac)
+          | `Command ->
+              let cmd = CicTextualParser2.parse_command istream in
+              print_endline (pp_command cmd)
+          | `Script -> assert false (* catched above *)
+          | `Alias ->
+              let env = CicTextualParser2.EnvironmentP3.of_string line in
+              print_endline (CicTextualParser2.EnvironmentP3.to_string env));
+          flush stdout
+        with
+        | CicTextualParser2.Parse_error msg -> prerr_endline msg
+  (*
+        | Stdpp.Exc_located ((p_start, p_end), exn) ->
+          prerr_endline (Printf.sprintf "Exception at character %d-%d: %s"
+            p_start p_end (Printexc.to_string exn))
+  *)
+      done
   with End_of_file ->
     close_in ic