Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 1 | #include <stdint.h> |
| 2 | #include <stdio.h> |
| 3 | #include <string.h> |
Edd Barrett | 8146aea | 2014-01-01 23:14:36 +0000 | [diff] [blame] | 4 | #include <stdlib.h> |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 5 | |
Damien | ce89a21 | 2013-10-15 22:25:17 +0100 | [diff] [blame] | 6 | #include "nlr.h" |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 7 | #include "misc.h" |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 8 | #include "mpconfig.h" |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 9 | #include "qstr.h" |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 10 | #include "lexer.h" |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 11 | #include "lexerunix.h" |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 12 | #include "parse.h" |
Damien | 0f08267 | 2013-12-17 18:33:53 +0000 | [diff] [blame] | 13 | #include "obj.h" |
Damien George | 1fb0317 | 2014-01-03 14:22:03 +0000 | [diff] [blame] | 14 | #include "compile.h" |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 15 | #include "runtime0.h" |
| 16 | #include "runtime.h" |
Damien | 92c0656 | 2013-10-22 22:32:27 +0100 | [diff] [blame] | 17 | #include "repl.h" |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 18 | |
Paul Sokolovsky | d674bd5 | 2014-01-04 19:38:19 +0200 | [diff] [blame] | 19 | #if MICROPY_USE_READLINE |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 20 | #include <readline/readline.h> |
Paul Sokolovsky | 903b24f | 2014-01-01 14:54:39 +0200 | [diff] [blame] | 21 | #include <readline/history.h> |
Paul Sokolovsky | fa02767 | 2014-01-01 18:28:01 +0200 | [diff] [blame] | 22 | #endif |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 23 | |
Paul Sokolovsky | e0e79ae | 2014-01-08 02:52:20 +0200 | [diff] [blame] | 24 | extern const mp_obj_fun_native_t mp_builtin_open_obj; |
Paul Sokolovsky | 51ee44a | 2014-01-20 23:49:56 +0200 | [diff] [blame] | 25 | void file_init(); |
Paul Sokolovsky | fc92608 | 2014-01-18 23:47:44 +0200 | [diff] [blame] | 26 | void rawsocket_init(); |
Paul Sokolovsky | a9459bc | 2014-02-02 00:57:06 +0200 | [diff] [blame] | 27 | void time_init(); |
Paul Sokolovsky | 60a9fac | 2014-01-29 00:24:00 +0200 | [diff] [blame] | 28 | void ffi_init(); |
Paul Sokolovsky | e0e79ae | 2014-01-08 02:52:20 +0200 | [diff] [blame] | 29 | |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 30 | static void execute_from_lexer(mp_lexer_t *lex, mp_parse_input_kind_t input_kind, bool is_repl) { |
| 31 | if (lex == NULL) { |
| 32 | return; |
| 33 | } |
| 34 | |
| 35 | if (0) { |
| 36 | // just tokenise |
| 37 | while (!mp_lexer_is_kind(lex, MP_TOKEN_END)) { |
| 38 | mp_token_show(mp_lexer_cur(lex)); |
| 39 | mp_lexer_to_next(lex); |
| 40 | } |
| 41 | mp_lexer_free(lex); |
| 42 | return; |
| 43 | } |
| 44 | |
Damien George | 9528cd6 | 2014-01-15 21:23:31 +0000 | [diff] [blame] | 45 | qstr parse_exc_id; |
| 46 | const char *parse_exc_msg; |
| 47 | mp_parse_node_t pn = mp_parse(lex, input_kind, &parse_exc_id, &parse_exc_msg); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 48 | |
| 49 | if (pn == MP_PARSE_NODE_NULL) { |
| 50 | // parse error |
Damien George | 9528cd6 | 2014-01-15 21:23:31 +0000 | [diff] [blame] | 51 | mp_lexer_show_error_pythonic_prefix(lex); |
| 52 | printf("%s: %s\n", qstr_str(parse_exc_id), parse_exc_msg); |
| 53 | mp_lexer_free(lex); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 54 | return; |
| 55 | } |
| 56 | |
Damien George | 0833500 | 2014-01-18 23:24:36 +0000 | [diff] [blame] | 57 | qstr source_name = mp_lexer_source_name(lex); |
Damien George | 9528cd6 | 2014-01-15 21:23:31 +0000 | [diff] [blame] | 58 | mp_lexer_free(lex); |
| 59 | |
Damien George | cbd2f74 | 2014-01-19 11:48:48 +0000 | [diff] [blame] | 60 | /* |
| 61 | printf("----------------\n"); |
| 62 | mp_parse_node_print(pn, 0); |
| 63 | printf("----------------\n"); |
| 64 | */ |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 65 | |
Damien George | 0833500 | 2014-01-18 23:24:36 +0000 | [diff] [blame] | 66 | mp_obj_t module_fun = mp_compile(pn, source_name, is_repl); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 67 | |
| 68 | if (module_fun == mp_const_none) { |
| 69 | // compile error |
| 70 | return; |
| 71 | } |
| 72 | |
| 73 | // execute it |
| 74 | nlr_buf_t nlr; |
| 75 | if (nlr_push(&nlr) == 0) { |
| 76 | rt_call_function_0(module_fun); |
| 77 | nlr_pop(); |
| 78 | } else { |
| 79 | // uncaught exception |
Damien George | 136b149 | 2014-01-19 12:38:49 +0000 | [diff] [blame] | 80 | mp_obj_print_exception((mp_obj_t)nlr.ret_val); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 81 | } |
| 82 | } |
| 83 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 84 | static char *str_join(const char *s1, int sep_char, const char *s2) { |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 85 | int l1 = strlen(s1); |
| 86 | int l2 = strlen(s2); |
| 87 | char *s = m_new(char, l1 + l2 + 2); |
| 88 | memcpy(s, s1, l1); |
| 89 | if (sep_char != 0) { |
| 90 | s[l1] = sep_char; |
| 91 | l1 += 1; |
| 92 | } |
| 93 | memcpy(s + l1, s2, l2); |
Damien | 92c0656 | 2013-10-22 22:32:27 +0100 | [diff] [blame] | 94 | s[l1 + l2] = 0; |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 95 | return s; |
| 96 | } |
| 97 | |
Paul Sokolovsky | fa02767 | 2014-01-01 18:28:01 +0200 | [diff] [blame] | 98 | static char *prompt(char *p) { |
Paul Sokolovsky | d674bd5 | 2014-01-04 19:38:19 +0200 | [diff] [blame] | 99 | #if MICROPY_USE_READLINE |
Paul Sokolovsky | fa02767 | 2014-01-01 18:28:01 +0200 | [diff] [blame] | 100 | char *line = readline(p); |
| 101 | if (line) { |
| 102 | add_history(line); |
| 103 | } |
| 104 | #else |
| 105 | static char buf[256]; |
| 106 | fputs(p, stdout); |
| 107 | char *s = fgets(buf, sizeof(buf), stdin); |
| 108 | if (!s) { |
| 109 | return NULL; |
| 110 | } |
| 111 | int l = strlen(buf); |
| 112 | if (buf[l - 1] == '\n') { |
| 113 | buf[l - 1] = 0; |
| 114 | } else { |
| 115 | l++; |
| 116 | } |
Paul Sokolovsky | 2b2cb7b | 2014-01-24 16:21:26 +0200 | [diff] [blame] | 117 | char *line = malloc(l); |
Paul Sokolovsky | fa02767 | 2014-01-01 18:28:01 +0200 | [diff] [blame] | 118 | memcpy(line, buf, l); |
| 119 | #endif |
| 120 | return line; |
| 121 | } |
| 122 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 123 | static void do_repl(void) { |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 124 | for (;;) { |
Paul Sokolovsky | fa02767 | 2014-01-01 18:28:01 +0200 | [diff] [blame] | 125 | char *line = prompt(">>> "); |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 126 | if (line == NULL) { |
| 127 | // EOF |
| 128 | return; |
| 129 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 130 | if (mp_repl_is_compound_stmt(line)) { |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 131 | for (;;) { |
Paul Sokolovsky | fa02767 | 2014-01-01 18:28:01 +0200 | [diff] [blame] | 132 | char *line2 = prompt("... "); |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 133 | if (line2 == NULL || strlen(line2) == 0) { |
| 134 | break; |
| 135 | } |
| 136 | char *line3 = str_join(line, '\n', line2); |
Damien | 732407f | 2013-12-29 19:33:23 +0000 | [diff] [blame] | 137 | free(line); |
| 138 | free(line2); |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 139 | line = line3; |
| 140 | } |
| 141 | } |
Damien | fa2162b | 2013-10-20 17:42:00 +0100 | [diff] [blame] | 142 | |
Damien George | b829b5c | 2014-01-25 13:51:19 +0000 | [diff] [blame] | 143 | mp_lexer_t *lex = mp_lexer_new_from_str_len(MP_QSTR__lt_stdin_gt_, line, strlen(line), false); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 144 | execute_from_lexer(lex, MP_PARSE_SINGLE_INPUT, true); |
Paul Sokolovsky | 2b2cb7b | 2014-01-24 16:21:26 +0200 | [diff] [blame] | 145 | free(line); |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 146 | } |
| 147 | } |
| 148 | |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 149 | static void do_file(const char *file) { |
Damien George | 66028ab | 2014-01-03 14:03:48 +0000 | [diff] [blame] | 150 | // hack: set dir for import based on where this file is |
| 151 | { |
| 152 | const char * s = strrchr(file, '/'); |
| 153 | if (s != NULL) { |
| 154 | int len = s - file; |
| 155 | char *dir = m_new(char, len + 1); |
| 156 | memcpy(dir, file, len); |
| 157 | dir[len] = '\0'; |
| 158 | mp_import_set_directory(dir); |
| 159 | } |
| 160 | } |
| 161 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 162 | mp_lexer_t *lex = mp_lexer_new_from_file(file); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 163 | execute_from_lexer(lex, MP_PARSE_FILE_INPUT, false); |
| 164 | } |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 165 | |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 166 | static void do_str(const char *str) { |
Damien George | b829b5c | 2014-01-25 13:51:19 +0000 | [diff] [blame] | 167 | mp_lexer_t *lex = mp_lexer_new_from_str_len(MP_QSTR__lt_stdin_gt_, str, strlen(str), false); |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 168 | execute_from_lexer(lex, MP_PARSE_SINGLE_INPUT, false); |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 169 | } |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 170 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 171 | typedef struct _test_obj_t { |
| 172 | mp_obj_base_t base; |
Damien George | 97209d3 | 2014-01-07 15:58:30 +0000 | [diff] [blame] | 173 | int value; |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 174 | } test_obj_t; |
| 175 | |
Paul Sokolovsky | 76d982e | 2014-01-13 19:19:16 +0200 | [diff] [blame] | 176 | static void test_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 177 | test_obj_t *self = self_in; |
| 178 | print(env, "<test %d>", self->value); |
Damien | a53f694 | 2013-11-02 23:58:38 +0000 | [diff] [blame] | 179 | } |
| 180 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 181 | static mp_obj_t test_get(mp_obj_t self_in) { |
| 182 | test_obj_t *self = self_in; |
| 183 | return mp_obj_new_int(self->value); |
Damien | a53f694 | 2013-11-02 23:58:38 +0000 | [diff] [blame] | 184 | } |
| 185 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 186 | static mp_obj_t test_set(mp_obj_t self_in, mp_obj_t arg) { |
| 187 | test_obj_t *self = self_in; |
| 188 | self->value = mp_obj_get_int(arg); |
| 189 | return mp_const_none; |
Damien | a53f694 | 2013-11-02 23:58:38 +0000 | [diff] [blame] | 190 | } |
| 191 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 192 | static MP_DEFINE_CONST_FUN_OBJ_1(test_get_obj, test_get); |
| 193 | static MP_DEFINE_CONST_FUN_OBJ_2(test_set_obj, test_set); |
| 194 | |
Damien George | 97209d3 | 2014-01-07 15:58:30 +0000 | [diff] [blame] | 195 | static const mp_method_t test_methods[] = { |
| 196 | { "get", &test_get_obj }, |
| 197 | { "set", &test_set_obj }, |
| 198 | { NULL, NULL }, |
| 199 | }; |
| 200 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 201 | static const mp_obj_type_t test_type = { |
| 202 | { &mp_const_type }, |
Damien | a53f694 | 2013-11-02 23:58:38 +0000 | [diff] [blame] | 203 | "Test", |
Paul Sokolovsky | 860ffb0 | 2014-01-05 22:34:09 +0200 | [diff] [blame] | 204 | .print = test_print, |
Damien George | 97209d3 | 2014-01-07 15:58:30 +0000 | [diff] [blame] | 205 | .methods = test_methods, |
Damien | a53f694 | 2013-11-02 23:58:38 +0000 | [diff] [blame] | 206 | }; |
| 207 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 208 | mp_obj_t test_obj_new(int value) { |
| 209 | test_obj_t *o = m_new_obj(test_obj_t); |
| 210 | o->base.type = &test_type; |
| 211 | o->value = value; |
| 212 | return o; |
| 213 | } |
| 214 | |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 215 | int usage(void) { |
| 216 | printf("usage: py [-c <command>] [<filename>]\n"); |
| 217 | return 1; |
| 218 | } |
| 219 | |
Damien George | 4d5b28c | 2014-01-29 18:56:46 +0000 | [diff] [blame] | 220 | mp_obj_t mem_info(void) { |
| 221 | printf("mem: total=%d, current=%d, peak=%d\n", m_get_total_bytes_allocated(), m_get_current_bytes_allocated(), m_get_peak_bytes_allocated()); |
| 222 | return mp_const_none; |
| 223 | } |
| 224 | |
| 225 | mp_obj_t qstr_info(void) { |
| 226 | uint n_pool, n_qstr, n_str_data_bytes, n_total_bytes; |
| 227 | qstr_pool_info(&n_pool, &n_qstr, &n_str_data_bytes, &n_total_bytes); |
| 228 | printf("qstr pool: n_pool=%u, n_qstr=%u, n_str_data_bytes=%u, n_total_bytes=%u\n", n_pool, n_qstr, n_str_data_bytes, n_total_bytes); |
| 229 | return mp_const_none; |
| 230 | } |
| 231 | |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 232 | int main(int argc, char **argv) { |
| 233 | qstr_init(); |
| 234 | rt_init(); |
| 235 | |
Paul Sokolovsky | 625d08a | 2014-02-05 00:59:54 +0200 | [diff] [blame^] | 236 | char *home = getenv("HOME"); |
| 237 | char *path = getenv("MICROPYPATH"); |
| 238 | if (path == NULL) { |
| 239 | path = "~/.micropython/lib:/usr/lib/micropython"; |
| 240 | } |
| 241 | uint path_num = 0; |
| 242 | for (char *p = path; p != NULL; p = strchr(p, ':')) { |
| 243 | path_num++; |
| 244 | if (p != NULL) { |
| 245 | p++; |
| 246 | } |
| 247 | } |
| 248 | sys_path = mp_obj_new_list(path_num, NULL); |
| 249 | mp_obj_t *items; |
| 250 | mp_obj_list_get(sys_path, &path_num, &items); |
| 251 | char *p = path; |
| 252 | for (int i = 0; i < path_num; i++) { |
| 253 | char *p1 = strchr(p, ':'); |
| 254 | if (p1 == NULL) { |
| 255 | p1 = p + strlen(p); |
| 256 | } |
| 257 | if (p[0] == '~' && p[1] == '/' && home != NULL) { |
| 258 | // Expand standalone ~ to $HOME |
| 259 | CHECKBUF(buf, PATH_MAX); |
| 260 | CHECKBUF_APPEND(buf, home, strlen(home)); |
| 261 | CHECKBUF_APPEND(buf, p + 1, p1 - p - 1); |
| 262 | items[i] = MP_OBJ_NEW_QSTR(qstr_from_strn(buf, CHECKBUF_LEN(buf))); |
| 263 | } else { |
| 264 | items[i] = MP_OBJ_NEW_QSTR(qstr_from_strn(p, p1 - p)); |
| 265 | } |
| 266 | p = p1 + 1; |
| 267 | } |
| 268 | |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 269 | mp_obj_t m_sys = mp_obj_new_module(MP_QSTR_sys); |
Paul Sokolovsky | 625d08a | 2014-02-05 00:59:54 +0200 | [diff] [blame^] | 270 | rt_store_attr(m_sys, MP_QSTR_path, sys_path); |
Paul Sokolovsky | fe2690d | 2014-01-20 00:53:46 +0200 | [diff] [blame] | 271 | mp_obj_t py_argv = mp_obj_new_list(0, NULL); |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 272 | rt_store_attr(m_sys, MP_QSTR_argv, py_argv); |
Paul Sokolovsky | fe2690d | 2014-01-20 00:53:46 +0200 | [diff] [blame] | 273 | |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 274 | rt_store_name(qstr_from_str("test"), test_obj_new(42)); |
Damien George | 4d5b28c | 2014-01-29 18:56:46 +0000 | [diff] [blame] | 275 | rt_store_name(qstr_from_str("mem_info"), rt_make_function_n(0, mem_info)); |
| 276 | rt_store_name(qstr_from_str("qstr_info"), rt_make_function_n(0, qstr_info)); |
Damien George | 12eacca | 2014-01-21 21:54:15 +0000 | [diff] [blame] | 277 | |
Paul Sokolovsky | 51ee44a | 2014-01-20 23:49:56 +0200 | [diff] [blame] | 278 | file_init(); |
Paul Sokolovsky | fc92608 | 2014-01-18 23:47:44 +0200 | [diff] [blame] | 279 | rawsocket_init(); |
Paul Sokolovsky | a9459bc | 2014-02-02 00:57:06 +0200 | [diff] [blame] | 280 | #if MICROPY_MOD_TIME |
| 281 | time_init(); |
| 282 | #endif |
Paul Sokolovsky | ed1239f | 2014-02-01 20:06:55 +0200 | [diff] [blame] | 283 | #if MICROPY_MOD_FFI |
Paul Sokolovsky | 60a9fac | 2014-01-29 00:24:00 +0200 | [diff] [blame] | 284 | ffi_init(); |
Paul Sokolovsky | ed1239f | 2014-02-01 20:06:55 +0200 | [diff] [blame] | 285 | #endif |
Damien | a53f694 | 2013-11-02 23:58:38 +0000 | [diff] [blame] | 286 | |
Damien George | 062478e | 2014-01-09 20:57:50 +0000 | [diff] [blame] | 287 | // Here is some example code to create a class and instance of that class. |
| 288 | // First is the Python, then the C code. |
| 289 | // |
| 290 | // class TestClass: |
| 291 | // pass |
| 292 | // test_obj = TestClass() |
| 293 | // test_obj.attr = 42 |
| 294 | mp_obj_t test_class_type, test_class_instance; |
Damien George | 5fa93b6 | 2014-01-22 14:35:10 +0000 | [diff] [blame] | 295 | test_class_type = mp_obj_new_type("TestClass", mp_const_empty_tuple, mp_obj_new_dict(0)); |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 296 | rt_store_name(QSTR_FROM_STR_STATIC("test_obj"), test_class_instance = rt_call_function_0(test_class_type)); |
| 297 | rt_store_attr(test_class_instance, QSTR_FROM_STR_STATIC("attr"), mp_obj_new_int(42)); |
Damien George | 062478e | 2014-01-09 20:57:50 +0000 | [diff] [blame] | 298 | |
Damien George | eb7bfcb | 2014-01-04 15:57:35 +0000 | [diff] [blame] | 299 | /* |
| 300 | printf("bytes:\n"); |
| 301 | printf(" total %d\n", m_get_total_bytes_allocated()); |
| 302 | printf(" cur %d\n", m_get_current_bytes_allocated()); |
| 303 | printf(" peak %d\n", m_get_peak_bytes_allocated()); |
| 304 | */ |
| 305 | |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 306 | if (argc == 1) { |
| 307 | do_repl(); |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 308 | } else { |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 309 | for (int a = 1; a < argc; a++) { |
| 310 | if (argv[a][0] == '-') { |
| 311 | if (strcmp(argv[a], "-c") == 0) { |
| 312 | if (a + 1 >= argc) { |
| 313 | return usage(); |
| 314 | } |
| 315 | do_str(argv[a + 1]); |
| 316 | a += 1; |
| 317 | } else { |
| 318 | return usage(); |
| 319 | } |
| 320 | } else { |
Paul Sokolovsky | fe2690d | 2014-01-20 00:53:46 +0200 | [diff] [blame] | 321 | for (int i = a; i < argc; i++) { |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 322 | rt_list_append(py_argv, MP_OBJ_NEW_QSTR(qstr_from_str(argv[i]))); |
Paul Sokolovsky | fe2690d | 2014-01-20 00:53:46 +0200 | [diff] [blame] | 323 | } |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 324 | do_file(argv[a]); |
Paul Sokolovsky | fe2690d | 2014-01-20 00:53:46 +0200 | [diff] [blame] | 325 | break; |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 326 | } |
| 327 | } |
Damien | 5ac1b2e | 2013-10-18 19:58:12 +0100 | [diff] [blame] | 328 | } |
Damien George | 136f675 | 2014-01-07 14:54:15 +0000 | [diff] [blame] | 329 | |
Damien | 429d719 | 2013-10-04 19:53:11 +0100 | [diff] [blame] | 330 | rt_deinit(); |
| 331 | |
| 332 | //printf("total bytes = %d\n", m_get_total_bytes_allocated()); |
| 333 | return 0; |
| 334 | } |
Damien | 087d218 | 2013-11-09 20:14:30 +0000 | [diff] [blame] | 335 | |
| 336 | // for sqrt |
| 337 | #include <math.h> |
| 338 | machine_float_t machine_sqrt(machine_float_t x) { |
| 339 | return sqrt(x); |
| 340 | } |