#include <assert.h>
#include <inttypes.h>
+#include <setjmp.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
typedef struct Environment Environment;
struct EnvironmentPool;
typedef struct EnvironmentPool EnvironmentPool;
+struct Stack;
+typedef struct Stack Stack;
const char* const STRING_LITERAL_LIST[] = {
{% for string_literal in string_literal_list %}
BOOLEAN,
CLOSURE,
INTEGER,
- STRING,
+ LIST,
+ STRING_CONCATENATION,
+ STRING_LITERAL,
+ STRUCTURE,
VOID
};
struct Closure
{
Environment* closed;
- Object (*call)(EnvironmentPool*, Environment*, size_t, Object*);
+ Object (*call)(EnvironmentPool*, Environment*, size_t, Stack*, const unsigned long, jmp_buf);
+};
+
+struct List;
+typedef struct List List;
+struct List
+{
+ size_t allocated;
+ size_t length;
+ Object* items;
+};
+
+struct StringConcatenation;
+typedef struct StringConcatenation StringConcatenation;
+
+struct Structure;
+typedef struct Structure Structure;
+struct Structure
+{
+ size_t reference_count;
+ size_t length;
+ const char** symbol_list;
+ Object* value_list;
};
union Instance
bool boolean;
Closure closure;
int32_t integer;
- const char* string;
+ List list;
+ StringConcatenation* string_concatenation;
+ const char* string_literal;
+ Structure* structure;
};
struct Object
const Object builtin$false = { BOOLEAN, (Instance)(bool){ false } };
const Object builtin$nil = { VOID, { 0 } };
+struct StringConcatenation
+{
+ size_t referenceCount;
+ Object left;
+ Object right;
+};
+
+Object List_construct(size_t allocate)
+{
+ Object* items = malloc(sizeof(Object) * allocate);
+ Object result = { LIST, (Instance)(List){ allocate, 0, items } };
+ return result;
+}
+
+void List_append(Object* list, Object item)
+{
+ assert(list->type == LIST);
+
+ if(list->instance.list.allocated == list->instance.list.length)
+ {
+ list->instance.list.allocated *= 2;
+ list->instance.list.items = realloc(
+ list->instance.list.items,
+ sizeof(Object) * list->instance.list.allocated
+ );
+ }
+
+ list->instance.list.items[list->instance.list.length] = item;
+ list->instance.list.length++;
+}
+
+Object List_get(Object* list, Object index)
+{
+ assert(list->type == LIST);
+ assert(index.type == INTEGER);
+
+ return list->instance.list.items[index.instance.integer];
+}
+
+struct Stack
+{
+ uint16_t length;
+ Object items[256];
+};
+
+void Stack_initialize(Stack* self)
+{
+ self->length = 0;
+}
+
+Stack* Stack_construct()
+{
+ Stack* result = malloc(sizeof(Stack));
+ Stack_initialize(result);
+ return result;
+}
+
+void Stack_destruct(Stack* self)
+{
+ free(self);
+}
+
+bool Stack_any(Stack* self)
+{
+ return self->length > 0;
+}
+
+void Stack_push(Stack* self, Object item)
+{
+ assert(self->length < 256);
+ self->items[self->length] = item;
+ self->length++;
+}
+
+Object Stack_pop(Stack* self)
+{
+ assert(self->length > 0);
+ self->length--;
+ return self->items[self->length];
+}
+
+Object Object_rereference(Object self)
+{
+ switch(self.type)
+ {
+ case BOOLEAN:
+ case CLOSURE:
+ case INTEGER:
+ case STRING_LITERAL:
+ case VOID:
+ return self;
+
+ case STRING_CONCATENATION:
+ self.instance.string_concatenation->referenceCount++;
+ return self;
+
+ case STRUCTURE:
+ self.instance.structure->reference_count++;
+ return self;
+
+ default:
+ assert(false);
+ }
+}
+
+Object Structure_construct(size_t length, const char** symbol_list, Object* value_list)
+{
+ Structure* structure = malloc(sizeof(Structure));
+ structure->reference_count = 1;
+ structure->length = length;
+ structure->symbol_list = malloc(sizeof(const char*) * length);
+ structure->value_list = malloc(sizeof(Object) * length);
+
+ // TODO Don't allow assignment of mutable structures, as this screws up reference counting
+ for(size_t i = 0; i < length; i++)
+ {
+ structure->symbol_list[i] = symbol_list[i];
+ structure->value_list[i] = Object_rereference(value_list[i]);
+ }
+
+ Object result = { STRUCTURE, (Instance)structure };
+
+ return result;
+}
+
+Object Structure_get(Object* self, const char* symbol)
+{
+ assert(self->type == STRUCTURE);
+
+ for(size_t i = 0; i < self->instance.structure->length; i++)
+ {
+ if(self->instance.structure->symbol_list[i] == symbol)
+ {
+ return self->instance.structure->value_list[i];
+ }
+ }
+
+ assert(false);
+}
+
struct EnvironmentNode
{
const char* key;
self->live = true;
}
+void Object_deinitialize(Object* self)
+{
+ switch(self->type)
+ {
+ case BOOLEAN:
+ break;
+ case CLOSURE:
+ break;
+ case INTEGER:
+ break;
+ case STRING_LITERAL:
+ break;
+ case VOID:
+ break;
+
+ case LIST:
+ for(size_t i = 0; i < self->instance.list.length; i++) {
+ Object_deinitialize(&(self->instance.list.items[i]));
+ }
+
+ free(self->instance.list.items);
+ break;
+
+ case STRING_CONCATENATION:
+ self->instance.string_concatenation->referenceCount--;
+
+ if(self->instance.string_concatenation->referenceCount == 0)
+ {
+ Object_deinitialize(&(self->instance.string_concatenation->left));
+ Object_deinitialize(&(self->instance.string_concatenation->right));
+ free(self->instance.string_concatenation);
+ }
+ break;
+
+ case STRUCTURE:
+ self->instance.structure->reference_count--;
+
+ if(self->instance.structure->reference_count == 0)
+ {
+ for(size_t i = 0; i < self->instance.structure->length; i++)
+ {
+ Object_deinitialize(&(self->instance.structure->value_list[i]));
+ }
+ free(self->instance.structure->symbol_list);
+ free(self->instance.structure->value_list);
+ free(self->instance.structure);
+ }
+ break;
+
+ default:
+ assert(false);
+ }
+}
+
+typedef uint32_t StackSnapshot;
+
+StackSnapshot Stack_takeSnapshot(Stack* self)
+{
+ return (StackSnapshot) self->length;
+}
+
+void Stack_rewind(Stack* self, StackSnapshot snapshot)
+{
+ while(self->length > snapshot)
+ {
+ Object item = Stack_pop(self);
+ Object_deinitialize(&item);
+ }
+}
+
void Environment_deinitialize(Environment* self)
{
EnvironmentNode* next;
for(EnvironmentNode* node = self->root; node != NULL; node = next)
{
next = node->next;
+ Object_deinitialize(&(node->value));
free(node);
}
}
{
case BOOLEAN:
case INTEGER:
- case STRING:
+ case STRING_LITERAL:
case VOID:
break;
return EnvironmentPool_allocate(previous->overflow);
}
+Environment* Environment_construct(EnvironmentPool* environmentPool, Environment* parent)
+{
+ Environment* environment = EnvironmentPool_allocate(environmentPool);
+ Environment_initialize(environment, parent);
+ return environment;
+}
+
Object integerLiteral(int32_t literal)
{
Object result;
Object stringLiteral(const char* literal)
{
Object result;
- result.type = STRING;
- result.instance.string = literal;
+ result.type = STRING_LITERAL;
+ result.instance.string_literal = literal;
return result;
}
-// TODO Make this conditionally added
-Object operator$negate(Object input)
+{% if 'pow' in builtins %}
+Object builtin$pow$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
{
- assert(input.type == INTEGER);
+ // Must unload items in reverse order
+ Object exponent = Stack_pop(stack);
+ Object base = Stack_pop(stack);
+
+ assert(base.type == INTEGER);
+ assert(exponent.type == INTEGER);
Object result;
result.type = INTEGER;
- result.instance.integer = -input.instance.integer;
+ result.instance.integer = pow(base.instance.integer, exponent.instance.integer);
return result;
}
-{% for id in infix_declarations %}
-Object operator${{ id.name }}(Object left, Object right)
+Object builtin$pow = { CLOSURE, (Instance)(Closure){ NULL, builtin$pow$implementation } };
+{% endif %}
+
+Object builtin$negate$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
{
- assert(left.type == {{ id.in_type.upper() }});
- assert(right.type == {{ id.in_type.upper() }});
+ assert(argc == 1);
+
+ Object argument = Stack_pop(stack);
+
+ assert(argument.type == INTEGER);
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (-argument.instance.integer)
+ };
- Object result;
- result.type = {{ id.out_type.upper() }};
- result.instance.{{ id.out_type.lower() }} = left.instance.{{ id.in_type.lower() }} {{ id.operator }} right.instance.{{ id.in_type.lower() }};
return result;
}
+Object builtin$negate = { CLOSURE, (Instance)(Closure){ NULL, builtin$negate$implementation } };
+
+{% for op in ['lt', 'gt', 'lte', 'gte', 'eq', 'neq'] %}
+Object builtin${{ op }}$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == INTEGER);
+ assert(right.type == INTEGER);
+
+ {% if op == 'lt' %}
+ if(left.instance.integer < right.instance.integer)
+ {% elif op == 'gt' %}
+ if(left.instance.integer > right.instance.integer)
+ {% elif op == 'lte' %}
+ if(left.instance.integer <= right.instance.integer)
+ {% elif op == 'gte' %}
+ if(left.instance.integer >= right.instance.integer)
+ {% elif op == 'eq' %}
+ if(left.instance.integer == right.instance.integer)
+ {% elif op == 'neq' %}
+ if(left.instance.integer != right.instance.integer)
+ {% endif %}
+ {
+ return builtin$true;
+ }
+ else
+ {
+ return builtin$false;
+ }
+}
+Object builtin${{ op }} = { CLOSURE, (Instance)(Closure){ NULL, builtin${{ op }}$implementation } };
{% endfor %}
-{% if 'pow' in builtins %}
-Object builtin$pow$implementation(EnvironmentPool* environmentPool, Environment* parent, size_t argc, Object* args)
+Object builtin$concat$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
{
assert(argc == 2);
- Object base = args[0];
- Object exponent = args[1];
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
- assert(base.type == INTEGER);
- assert(exponent.type == INTEGER);
+ switch(left.type) {
+ case STRING_CONCATENATION:
+ case STRING_LITERAL:
+ break;
- Object result;
- result.type = INTEGER;
- result.instance.integer = pow(base.instance.integer, exponent.instance.integer);
+ default:
+ assert(false);
+ }
+
+ switch(right.type) {
+ case STRING_CONCATENATION:
+ case STRING_LITERAL:
+ break;
+
+ default:
+ assert(false);
+ }
+
+ StringConcatenation* concatenation = malloc(sizeof(StringConcatenation));
+ concatenation->referenceCount = 1;
+ concatenation->left = Object_rereference(left);
+ concatenation->right = Object_rereference(right);
+
+ Object result = { STRING_CONCATENATION, (Instance)concatenation };
return result;
}
+Object builtin$concat = { CLOSURE, (Instance)(Closure){ NULL, builtin$concat$implementation } };
+
+Object builtin$add$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
-Object builtin$pow = { CLOSURE, (Instance)(Closure){ NULL, builtin$pow$implementation } };
-{% endif %}
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == INTEGER);
+ assert(right.type == INTEGER);
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (left.instance.integer + right.instance.integer)
+ };
+
+ return result;
+}
+Object builtin$add = { CLOSURE, (Instance)(Closure){ NULL, builtin$add$implementation } };
+
+Object builtin$subtract$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == INTEGER);
+ assert(right.type == INTEGER);
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (left.instance.integer - right.instance.integer)
+ };
+
+ return result;
+}
+Object builtin$subtract = { CLOSURE, (Instance)(Closure){ NULL, builtin$subtract$implementation } };
+
+Object builtin$multiply$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == INTEGER);
+ assert(right.type == INTEGER);
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (left.instance.integer * right.instance.integer)
+ };
+
+ return result;
+}
+Object builtin$multiply = { CLOSURE, (Instance)(Closure){ NULL, builtin$multiply$implementation } };
+
+Object builtin$integer_divide$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == INTEGER);
+ assert(right.type == INTEGER);
+
+ if(right.instance.integer == 0)
+ {
+ fprintf(stderr, "DivisionByZeroError on line %zu\n", line);
+ longjmp(parentJump, 1);
+ }
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (left.instance.integer / right.instance.integer)
+ };
+
+ return result;
+}
+Object builtin$integer_divide = { CLOSURE, (Instance)(Closure){ NULL, builtin$integer_divide$implementation } };
+
+Object builtin$modular_divide$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == INTEGER);
+ assert(right.type == INTEGER);
+
+ if(right.instance.integer == 0)
+ {
+ fprintf(stderr, "DivisionByZeroError on line %zu\n", line);
+ longjmp(parentJump, 1);
+ }
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (left.instance.integer % right.instance.integer)
+ };
+
+ return result;
+}
+Object builtin$modular_divide = { CLOSURE, (Instance)(Closure){ NULL, builtin$modular_divide$implementation } };
+
+Object builtin$field$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ assert(left.type == STRUCTURE);
+ assert(right.type == STRING_LITERAL);
+
+ Object result = (Object){
+ INTEGER,
+ (Instance)(int32_t) (left.instance.integer % right.instance.integer)
+ };
+
+ return result;
+}
+Object builtin$field = { CLOSURE, (Instance)(Closure){ NULL, builtin$field$implementation } };
+
+Object builtin$get$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
+{
+ assert(argc == 2);
+
+ Object right = Stack_pop(stack);
+ Object left = Stack_pop(stack);
+
+ return List_get(&left, right);
+}
+Object builtin$get = { CLOSURE, (Instance)(Closure){ NULL, builtin$get$implementation } };
{% if 'print' in builtins %}
-Object builtin$print$implementation(EnvironmentPool* environmentPool, Environment* parent, size_t argc, Object* args)
+Object builtin$print$implementation(
+ EnvironmentPool* environmentPool,
+ Environment* parent,
+ size_t argc,
+ Stack* stack,
+ const unsigned long line,
+ jmp_buf parentJump)
{
+ Stack reverse_stack;
+ Stack_initialize(&reverse_stack);
+
for(size_t i = 0; i < argc; i++)
{
- Object output = args[i];
+ Stack_push(&reverse_stack, Stack_pop(stack));
+ }
+
+ while(reverse_stack.length > 0)
+ {
+ Object output = Stack_pop(&reverse_stack);
switch(output.type)
{
case BOOLEAN:
printf("%" PRId32, output.instance.integer);
break;
- case STRING:
+ case STRING_CONCATENATION:
+ Stack_push(stack, output.instance.string_concatenation->left);
+ builtin$print$implementation(NULL, NULL, 1, stack, line, parentJump);
+ Stack_push(stack, output.instance.string_concatenation->right);
+ builtin$print$implementation(NULL, NULL, 1, stack, line, parentJump);
+ break;
+
+ case STRING_LITERAL:
// Using fwrite instead of printf to handle size_t length
- printf("%s", output.instance.string);
+ printf("%s", output.instance.string_literal);
break;
case VOID:
default:
assert(false);
}
+ Object_deinitialize(&output);
}
// TODO Return something better
Environment* environment = EnvironmentPool_allocate(environmentPool);
Environment_initialize(environment, NULL);
+ Stack stackMemory;
+ Stack* stack = &stackMemory;
+ Stack_initialize(stack);
+
+ jmp_buf jump;
+ if(setjmp(jump) != 0)
+ {
+ fprintf(stderr, "\tin __main__\n");
+
+ while(Stack_any(stack))
+ {
+ Object item = Stack_pop(stack);
+ Object_deinitialize(&item);
+ }
+ Environment_setLive(environment, false);
+ EnvironmentPool_destruct(environmentPool);
+
+ // TODO We would like to return something nonzero here, but that messes up Valgrind so we couldn't catch memory leaks
+ return 0;
+ }
+
// TODO Use the symbol from SYMBOL_LIST
{% for builtin in builtins %}
Environment_set(environment, "{{ builtin }}", builtin${{ builtin }});
{% endfor %}
+ Environment_set(environment, "true", builtin$true);
+ Environment_set(environment, "false", builtin$false);
+ Environment_set(environment, "__add__", builtin$add);
+ Environment_set(environment, "__subtract__", builtin$subtract);
+ Environment_set(environment, "__multiply__", builtin$multiply);
+ Environment_set(environment, "__integer_divide__", builtin$integer_divide);
+ Environment_set(environment, "__modular_divide__", builtin$modular_divide);
+ Environment_set(environment, "__negate__", builtin$negate);
+ Environment_set(environment, "__concat__", builtin$concat);
+ Environment_set(environment, "__field__", builtin$field);
+ Environment_set(environment, "__get__", builtin$get);
+
+ {% for op in ['lt', 'gt', 'lte', 'gte', 'eq', 'neq'] %}
+ Environment_set(environment, "__{{ op }}__", builtin${{ op }});
+ {% endfor %}
+
{% for statement in statements %}
{{ statement }}
{% endfor %}