Print control registers in HTL output

1 files changed, 13 insertions, 0 deletions

diff --git a/src/hls/PrintHTL.ml b/src/hls/PrintHTL.ml
index 2bee487..d051aad 100644
--- a/src/hls/PrintHTL.ml
+++ b/src/hls/PrintHTL.ml
@@ -76,6 +76,17 @@ let print_ram pp opt_ram =
     fprintf pp "}\n\n"
   | None -> ()
 
+let print_control pp f =
+  fprintf pp "control {\n";
+  fprintf pp "     st: %s\n" (register f.mod_st);
+  fprintf pp "    stk: %s\n" (register f.mod_stk);
+  fprintf pp " finish: %s\n" (register f.mod_finish);
+  fprintf pp " return: %s\n" (register f.mod_return);
+  fprintf pp "  start: %s\n" (register f.mod_start);
+  fprintf pp "  reset: %s\n" (register f.mod_reset);
+  fprintf pp "    clk: %s\n" (register f.mod_clk);
+  fprintf pp "}\n\n"
+
 let print_module pp id f =
   fprintf pp "%s(%s) {\n" (extern_atom id) (registers f.mod_params);
 
@@ -83,6 +94,8 @@ let print_module pp id f =
   let datapath = ptree_to_list f.mod_datapath in
   let controllogic = ptree_to_list f.mod_controllogic in
 
+  print_control pp f;
+
   print_ram pp f.mod_ram;
 
   fprintf pp "externctrl {\n";