bare-arm, minimal, qemu-arm: Make do_str() take parse-input-kind as arg.

dpgeorge · dpgeorge · commit c4b592d3793b · 2015-06-22T22:38:47.000+01:00
The do_str() function is provided essentially as documentation to show
how to compile and execute a string.  This patch makes do_str take an
extra arg to specify how the string should be interpreted: either as a
single line (ie from a REPL) or as multiple lines (ie from a file).
diff --git a/bare-arm/main.c b/bare-arm/main.c
@@ -7,7 +7,7 @@
 #include "py/runtime.h"
 #include "py/repl.h"
 
-void do_str(const char *src) {
+void do_str(const char *src, mp_parse_input_kind_t input_kind) {
     mp_lexer_t *lex = mp_lexer_new_from_str_len(MP_QSTR__lt_stdin_gt_, src, strlen(src), 0);
     if (lex == NULL) {
         return;
@@ -16,7 +16,7 @@ void do_str(const char *src) {
     nlr_buf_t nlr;
     if (nlr_push(&nlr) == 0) {
         qstr source_name = lex->source_name;
-        mp_parse_node_t pn = mp_parse(lex, MP_PARSE_SINGLE_INPUT);
+        mp_parse_node_t pn = mp_parse(lex, input_kind);
         mp_obj_t module_fun = mp_compile(pn, source_name, MP_EMIT_OPT_NONE, true);
         mp_call_function_0(module_fun);
         nlr_pop();
@@ -28,7 +28,8 @@ void do_str(const char *src) {
 
 int main(int argc, char **argv) {
     mp_init();
-    do_str("print('hello world!', list(x+1 for x in range(10)), end='eol\n')");
+    do_str("print('hello world!', list(x+1 for x in range(10)), end='eol\\n')", MP_PARSE_SINGLE_INPUT);
+    do_str("for i in range(10):\n  print(i)", MP_PARSE_FILE_INPUT);
     mp_deinit();
     return 0;
 }
diff --git a/minimal/main.c b/minimal/main.c
@@ -9,7 +9,7 @@
 #include "py/gc.h"
 #include "pyexec.h"
 
-void do_str(const char *src) {
+void do_str(const char *src, mp_parse_input_kind_t input_kind) {
     mp_lexer_t *lex = mp_lexer_new_from_str_len(MP_QSTR__lt_stdin_gt_, src, strlen(src), 0);
     if (lex == NULL) {
         printf("MemoryError: lexer could not allocate memory\n");
@@ -19,7 +19,7 @@ void do_str(const char *src) {
     nlr_buf_t nlr;
     if (nlr_push(&nlr) == 0) {
         qstr source_name = lex->source_name;
-        mp_parse_node_t pn = mp_parse(lex, MP_PARSE_SINGLE_INPUT);
+        mp_parse_node_t pn = mp_parse(lex, input_kind);
         mp_obj_t module_fun = mp_compile(pn, source_name, MP_EMIT_OPT_NONE, true);
         mp_call_function_0(module_fun);
         nlr_pop();
@@ -51,7 +51,8 @@ int main(int argc, char **argv) {
     #else
     pyexec_friendly_repl();
     #endif
-    //do_str("print('hello world!', list(x+1 for x in range(10)), end='eol\\n')");
+    //do_str("print('hello world!', list(x+1 for x in range(10)), end='eol\\n')", MP_PARSE_SINGLE_INPUT);
+    //do_str("for i in range(10):\r\n  print(i)", MP_PARSE_FILE_INPUT);
     mp_deinit();
     return 0;
 }
diff --git a/qemu-arm/main.c b/qemu-arm/main.c
@@ -12,7 +12,7 @@
 #include "py/gc.h"
 #include "py/repl.h"
 
-void do_str(const char *src) {
+void do_str(const char *src, mp_parse_input_kind_t input_kind) {
     mp_lexer_t *lex = mp_lexer_new_from_str_len(MP_QSTR__lt_stdin_gt_, src, strlen(src), 0);
     if (lex == NULL) {
         return;
@@ -21,7 +21,7 @@ void do_str(const char *src) {
     nlr_buf_t nlr;
     if (nlr_push(&nlr) == 0) {
         qstr source_name = lex->source_name;
-        mp_parse_node_t pn = mp_parse(lex, MP_PARSE_SINGLE_INPUT);
+        mp_parse_node_t pn = mp_parse(lex, input_kind);
         mp_obj_t module_fun = mp_compile(pn, source_name, MP_EMIT_OPT_NONE, true);
         mp_call_function_0(module_fun);
         nlr_pop();
@@ -36,7 +36,7 @@ int main(int argc, char **argv) {
     void *heap = malloc(16 * 1024);
     gc_init(heap, (char*)heap + 16 * 1024);
     mp_init();
-    do_str("print('hello world!')");
+    do_str("print('hello world!')", MP_PARSE_SINGLE_INPUT);
     mp_deinit();
     return 0;
 }
