Add bytecode generator

src/main.c

@@ -31,6 +31,13 @@
 #include "lexer.h"
 #include "parser.h"
 #include "pretty-printer.h"
+#include "bytecode-generator.h"
+
+#define DUMP_TOKENS   (1u << 0)
+#define DUMP_AST      (1u << 1)
+#define DUMP_BYTECODE (1u << 2)
+
+#define MAX_STRINGS 100
 
 void fake_exit (void);
 
@@ -85,12 +92,27 @@ fake_exit (void)
 #endif
 }
 
+static inline void
+check_for_several_dumps (uint8_t dump)
+{
+  bool was_bit = 0;
+  for (; dump; dump >>= 1)
+    {
+      if (dump & 1u)
+        {
+          if (was_bit)
+            jerry_Exit (ERR_SEVERAL_FILES);
+          else
+            was_bit = true;
+        }
+    }
+}
+
 int
 main (int argc, char **argv)
 {
   statement st;
-  bool dump_tokens = false;
-  bool dump_ast = false;
+  uint8_t dump = 0;
   const char *file_name = NULL;
 #ifdef __HOST
   FILE *file = NULL;
@@ -102,25 +124,26 @@ main (int argc, char **argv)
     for (int i = 1; i < argc; i++)
     {
       if (!__strcmp ("-t", argv[i]))
-        dump_tokens = true;
+        dump |= DUMP_TOKENS;
       else if (!__strcmp ("-a", argv[i]))
-        dump_ast = true;
+        dump |= DUMP_AST;
+      else if (!__strcmp ("-b", argv[i]))
+        dump |= DUMP_BYTECODE;
       else if (file_name == NULL)
         file_name = argv[i];
       else
         jerry_Exit (ERR_SEVERAL_FILES);
     }
 
+  check_for_several_dumps (dump);
+
+  if (!dump)
+    dump |= DUMP_BYTECODE;
+
+#ifdef __HOST
   if (file_name == NULL)
     jerry_Exit (ERR_NO_FILES);
 
-  if (dump_tokens && dump_ast)
-    jerry_Exit (ERR_SEVERAL_FILES);
-
-  if (!dump_tokens)
-    dump_ast = true;
-
-#ifdef __HOST
   file = __fopen (file_name, "r");
 
   if (file == NULL)
@@ -133,7 +156,7 @@ main (int argc, char **argv)
   lexer_set_source (generated_source);
 #endif
 
-  if (dump_ast)
+  if (dump & DUMP_AST)
     {
       parser_init ();
       st = parser_parse_statement ();
@@ -146,7 +169,7 @@ main (int argc, char **argv)
         }
     }
 
-  if (dump_tokens)
+  if (dump & DUMP_TOKENS)
     {
       token tok = lexer_next_token ();
       while (tok.type != TOK_EOF)
@@ -156,6 +179,39 @@ main (int argc, char **argv)
         }
     }
 
+  if (dump & DUMP_BYTECODE)
+    {
+      const char *strings[MAX_STRINGS];
+      uint8_t strings_num;
+      // First run parser to fill list of strings
+      token tok = lexer_next_token ();
+      while (tok.type != TOK_EOF)
+        tok = lexer_next_token ();
+
+      strings_num = lexer_get_strings (strings);
+
+      // Reset lexer
+#ifdef __HOST
+      __rewind (file);
+      lexer_set_file (file);
+#else
+      lexer_set_source (generated_source);
+#endif
+
+      parser_init ();
+      generator_init ();
+      generator_dump_strings (strings, strings_num);
+      st = parser_parse_statement ();
+      JERRY_ASSERT (!is_statement_null (st));
+      __printf (" ST_TYPE = %d", st.type == STMT_EOF);
+      while (st.type != STMT_EOF)
+        {
+          generator_dump_statement (st);
+          st = parser_parse_statement ();
+          JERRY_ASSERT (!is_statement_null (st));
+        }
+    }
+
   //gen_bytecode (generated_source);
   //gen_bytecode ();
   //run_int ();