diff options
Diffstat (limited to 'py/emitcommon.c')
| -rw-r--r-- | py/emitcommon.c | 171 |
1 files changed, 171 insertions, 0 deletions
diff --git a/py/emitcommon.c b/py/emitcommon.c new file mode 100644 index 0000000000..1fd8697c3a --- /dev/null +++ b/py/emitcommon.c @@ -0,0 +1,171 @@ +#include <unistd.h> +#include <stdio.h> +#include <stdint.h> +#include <string.h> +#include <assert.h> + +#include "misc.h" +#include "lexer.h" +#include "machine.h" +#include "parse.h" +#include "scope.h" +#include "runtime.h" +#include "emit.h" + +#define EMIT(fun, arg...) (emit_##fun(emit, ##arg)) + +void emit_common_declare_global(pass_kind_t pass, scope_t *scope, qstr qstr) { + if (pass == PASS_1) { + if (scope->kind == SCOPE_MODULE) { + printf("SyntaxError?: can't declare global in outer code\n"); + return; + } + bool added; + id_info_t *id_info = scope_find_or_add_id(scope, qstr, &added); + if (!added) { + printf("SyntaxError?: identifier already declared something\n"); + return; + } + id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT; + + // if the id exists in the global scope, set its kind to EXPLICIT_GLOBAL + id_info = scope_find_global(scope, qstr); + if (id_info != NULL) { + id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT; + } + } +} + +void emit_common_declare_nonlocal(pass_kind_t pass, scope_t *scope, qstr qstr) { + if (pass == PASS_1) { + if (scope->kind == SCOPE_MODULE) { + printf("SyntaxError?: can't declare nonlocal in outer code\n"); + return; + } + bool added; + id_info_t *id_info = scope_find_or_add_id(scope, qstr, &added); + if (!added) { + printf("SyntaxError?: identifier already declared something\n"); + return; + } + id_info_t *id_info2 = scope_find_local_in_parent(scope, qstr); + if (id_info2 == NULL || !(id_info2->kind == ID_INFO_KIND_LOCAL || id_info2->kind == ID_INFO_KIND_CELL || id_info2->kind == ID_INFO_KIND_FREE)) { + printf("SyntaxError: no binding for nonlocal '%s' found\n", qstr_str(qstr)); + return; + } + id_info->kind = ID_INFO_KIND_FREE; + scope_close_over_in_parents(scope, qstr); + } +} + +void emit_common_load_id(pass_kind_t pass, scope_t *scope, qstr qstr___class__, emitter_t *emit, qstr qstr) { + id_info_t *id_info = NULL; + if (pass == PASS_1) { + // name adding/lookup + bool added; + id_info = scope_find_or_add_id(scope, qstr, &added); + if (added) { + if (strcmp(qstr_str(qstr), "AssertionError") == 0) { + id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT; + // TODO how much of a hack is this? + } else if (strcmp(qstr_str(qstr), "super") == 0 && scope->kind == SCOPE_FUNCTION) { + // special case, super is a global, and also counts as use of __class__ + id_info->kind = ID_INFO_KIND_GLOBAL_EXPLICIT; + id_info_t *id_info2 = scope_find_local_in_parent(scope, qstr___class__); + if (id_info2 != NULL) { + id_info2 = scope_find_or_add_id(scope, qstr___class__, &added); + if (added) { + id_info2->kind = ID_INFO_KIND_FREE; + scope_close_over_in_parents(scope, qstr___class__); + } + } + } else { + id_info_t *id_info2 = scope_find_local_in_parent(scope, qstr); + if (id_info2 != NULL && (id_info2->kind == ID_INFO_KIND_LOCAL || id_info2->kind == ID_INFO_KIND_CELL || id_info2->kind == ID_INFO_KIND_FREE)) { + id_info->kind = ID_INFO_KIND_FREE; + scope_close_over_in_parents(scope, qstr); + } else { + id_info->kind = ID_INFO_KIND_GLOBAL_IMPLICIT; + } + } + } + } else { + id_info = scope_find(scope, qstr); + } + + assert(id_info != NULL); // TODO can this ever fail? + + // call the emit backend with the correct code + if (id_info == NULL || id_info->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) { + EMIT(load_name, qstr); + } else if (id_info->kind == ID_INFO_KIND_GLOBAL_EXPLICIT) { + EMIT(load_global, qstr); + } else if (id_info->kind == ID_INFO_KIND_LOCAL) { + EMIT(load_fast, qstr, id_info->local_num); + } else if (id_info->kind == ID_INFO_KIND_CELL || id_info->kind == ID_INFO_KIND_FREE) { + EMIT(load_deref, qstr); + } else { + assert(0); + } +} + +static id_info_t *get_id_for_modification(pass_kind_t pass, scope_t *scope, emitter_t *emit, qstr qstr) { + id_info_t *id_info = NULL; + if (pass == PASS_1) { + // name adding/lookup + bool added; + id_info = scope_find_or_add_id(scope, qstr, &added); + if (added) { + if (scope->kind == SCOPE_MODULE || scope->kind == SCOPE_CLASS) { + id_info->kind = ID_INFO_KIND_GLOBAL_IMPLICIT; + } else { + id_info->kind = ID_INFO_KIND_LOCAL; + } + } else if (scope->kind >= SCOPE_FUNCTION && scope->kind <= SCOPE_GEN_EXPR && id_info->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) { + // rebind as a local variable + id_info->kind = ID_INFO_KIND_LOCAL; + } + } else { + id_info = scope_find(scope, qstr); + } + + assert(id_info != NULL); // TODO can this ever fail? + + return id_info; +} + +void emit_common_store_id(pass_kind_t pass, scope_t *scope, emitter_t *emit, qstr qstr) { + // create/get the id info + id_info_t *id = get_id_for_modification(pass, scope, emit, qstr); + + // call the emit backend with the correct code + if (id == NULL || id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) { + EMIT(store_name, qstr); + } else if (id->kind == ID_INFO_KIND_GLOBAL_EXPLICIT) { + EMIT(store_global, qstr); + } else if (id->kind == ID_INFO_KIND_LOCAL) { + EMIT(store_fast, qstr, id->local_num); + } else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) { + EMIT(store_deref, qstr); + } else { + assert(0); + } +} + +void emit_common_delete_id(pass_kind_t pass, scope_t *scope, emitter_t *emit, qstr qstr) { + // create/get the id info + id_info_t *id = get_id_for_modification(pass, scope, emit, qstr); + + // call the emit backend with the correct code + if (id == NULL || id->kind == ID_INFO_KIND_GLOBAL_IMPLICIT) { + EMIT(delete_name, qstr); + } else if (id->kind == ID_INFO_KIND_GLOBAL_EXPLICIT) { + EMIT(delete_global, qstr); + } else if (id->kind == ID_INFO_KIND_LOCAL) { + EMIT(delete_fast, qstr, id->local_num); + } else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) { + EMIT(delete_deref, qstr); + } else { + assert(0); + } +} |