X-Git-Url: https://code.kerkeslager.com/?a=blobdiff_plain;f=templates%2Fprogram.c;h=f8095bf838a3934733e628d67976060c69109054;hb=5e3bb8aa55a25e2ced307e06f6579c027749ff0e;hp=3410e1bd7b80365621ed37038cb7bfcd92a73992;hpb=3da330f045ed7fcb66ee9d9447de320680263699;p=fur

diff --git a/templates/program.c b/templates/program.c
index 3410e1b..f8095bf 100644
--- a/templates/program.c
+++ b/templates/program.c
@@ -1,5 +1,6 @@
 #include <assert.h>
 #include <inttypes.h>
+#include <setjmp.h>
 #include <stdbool.h>
 #include <stdlib.h>
 #include <string.h>
@@ -34,6 +35,8 @@ struct Environment;
 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 %}
@@ -52,7 +55,11 @@ enum Type
   BOOLEAN,
   CLOSURE,
   INTEGER,
-  STRING
+  LIST,
+  STRING_CONCATENATION,
+  STRING_LITERAL,
+  STRUCTURE,
+  VOID
 };
 
 struct Closure;
@@ -60,7 +67,29 @@ typedef struct Closure Closure;
 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
@@ -68,7 +97,10 @@ 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
@@ -79,6 +111,147 @@ struct Object
 
 const Object builtin$true = { BOOLEAN, (Instance)(bool){ true } };
 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
 {
@@ -106,12 +279,83 @@ void Environment_initialize(Environment* self, Environment* parent)
   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);
   }
 }
@@ -136,7 +380,8 @@ void Environment_mark(Environment* self)
     {
       case BOOLEAN:
       case INTEGER:
-      case STRING:
+      case STRING_LITERAL:
+      case VOID:
         break;
 
       case CLOSURE:
@@ -305,6 +550,13 @@ Environment* EnvironmentPool_allocate(EnvironmentPool* self)
   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;
@@ -316,61 +568,340 @@ Object integerLiteral(int32_t literal)
 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:
@@ -386,14 +917,26 @@ Object builtin$print$implementation(EnvironmentPool* environmentPool, Environmen
         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:
+        printf("nil");
         break;
 
       default:
         assert(false);
     }
+    Object_deinitialize(&output);
   }
 
   // TODO Return something better
@@ -412,11 +955,48 @@ int main(int argc, char** argv)
   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 %}