Added if expression statements
[fur] / templates / program.c
index ecabb17..4ac6289 100644 (file)
@@ -4,37 +4,72 @@
 #include <stdlib.h>
 #include <string.h>
 
+/* Some terminology used in function names:
+ * - initialize: These functions take a pointer and potentially some other arguments, and use those
+ *   to initialize the value pointed to by self. Initialize functions DO NOT allocate the function,
+ *   so they can be used to initialize stack-allocated variables.
+ * - construct: This allocates a value for a pointer, initializes it, and returns it. This is for
+ *   heap-allocated values. It may be as simple as allocating the memory, calling an initialize, and
+ *   returning it.
+ * - deinitialize: These functions dereference or free any objects pointed to by the self pointer's
+ *   value, but they don't actually free the self pointer. This is useful for stack-allocated objects
+ *   which point to heap-allocated objects.
+ * - destruct: This dereferences or frees memory pointed to by the self argument, and all the
+ *   pointers on the self argument.
+ */
+
 {% for standard_library in standard_libraries %}
 #include <{{standard_library}}>
 {% endfor %}
 
-struct String;
-typedef struct String String;
 enum Type;
 typedef enum Type Type;
 union Instance;
 typedef union Instance Instance;
 struct Object;
 typedef struct Object Object;
-struct Runtime;
-typedef struct Runtime Runtime;
+struct EnvironmentNode;
+typedef struct EnvironmentNode EnvironmentNode;
+struct Environment;
+typedef struct Environment Environment;
+struct EnvironmentPool;
+typedef struct EnvironmentPool EnvironmentPool;
 
-struct String
-{
-  size_t length;
-  char* characters;
+const char* const STRING_LITERAL_LIST[] = {
+{% for string_literal in string_literal_list %}
+  "{{ string_literal }}",
+{% endfor %}
+};
+
+const char* const SYMBOL_LIST[] = {
+{% for symbol in symbol_list %}
+  "{{ symbol }}",
+{% endfor %}
 };
 
 enum Type
 {
+  BOOLEAN,
+  CLOSURE,
   INTEGER,
-  STRING
+  STRING,
+  VOID
+};
+
+struct Closure;
+typedef struct Closure Closure;
+struct Closure
+{
+  Environment* closed;
+  Object (*call)(EnvironmentPool*, Environment*, size_t, Object*);
 };
 
 union Instance
 {
+  bool boolean;
+  Closure closure;
   int32_t integer;
-  String* string;
+  const char* string;
 };
 
 struct Object
@@ -43,52 +78,234 @@ struct Object
   Instance instance;
 };
 
-struct Runtime
+const Object builtin$true = { BOOLEAN, (Instance)(bool){ true } };
+const Object builtin$false = { BOOLEAN, (Instance)(bool){ false } };
+const Object builtin$nil = { VOID, { 0 } };
+
+struct EnvironmentNode
+{
+  const char* key;
+  Object value;
+  EnvironmentNode* next;
+};
+
+struct Environment
+{
+  bool mark;
+  bool live;
+
+  Environment* parent;
+  EnvironmentNode* root;
+};
+
+void Environment_initialize(Environment* self, Environment* parent)
+{
+  self->parent = parent;
+  self->root = NULL;
+
+  // We are currently only ever initializing environments at the beginning of running functions, so
+  // for now at least we can assume that we want it to be live immediately.
+  self->live = true;
+}
+
+void Environment_deinitialize(Environment* self)
+{
+  EnvironmentNode* next;
+  for(EnvironmentNode* node = self->root; node != NULL; node = next)
+  {
+    next = node->next;
+    free(node);
+  }
+}
+
+void Environment_setLive(Environment* self, bool live)
+{
+  self->live = live;
+}
+
+void Environment_mark(Environment* self)
+{
+  if(self == NULL) return;
+  if(self->mark) return; // Prevents infinite recursion in the case of cycles
+
+  self->mark = true;
+
+  Environment_mark(self->parent);
+
+  for(EnvironmentNode* node = self->root; node != NULL; node = node->next)
+  {
+    switch(node->value.type)
+    {
+      case BOOLEAN:
+      case INTEGER:
+      case STRING:
+      case VOID:
+        break;
+
+      case CLOSURE:
+        Environment_mark(node->value.instance.closure.closed);
+        break;
+
+      default:
+        assert(false);
+    }
+  }
+}
+
+// This need not be thread safe because environments exist on one thread only
+void Environment_set(Environment* self, const char* const key, Object value)
+{
+  EnvironmentNode* node = malloc(sizeof(EnvironmentNode));
+  node->key = key;
+  node->value = value;
+  node->next = self->root;
+  self->root = node;
+}
+
+Object Environment_get(Environment* self, const char* const symbol)
 {
-  size_t permanentStringsLength;
-  size_t permanentStringsAllocated;
-  String** permanentStrings;
+  for(EnvironmentNode* node = self->root; node != NULL; node = node->next)
+  {
+    // We can compare pointers because pointers are unique in the SYMBOL_LIST
+    if(node->key == symbol)
+    {
+      return node->value;
+    }
+  }
+
+  if(self->parent != NULL)
+  {
+    return Environment_get(self->parent, symbol);
+  }
+
+  // TODO Handle symbol errors
+  assert(false);
+}
+
+# define POOL_SIZE 64
+struct EnvironmentPool
+{
+  int8_t freeIndex;
+  bool allocatedFlags[POOL_SIZE];
+  Environment environments[POOL_SIZE];
+  EnvironmentPool* overflow;
 };
 
-Runtime* Runtime_construct()
+EnvironmentPool* EnvironmentPool_construct();
+void EnvironmentPool_initialize(EnvironmentPool*);
+void EnvironmentPool_deinitialize(EnvironmentPool*);
+void EnvironmentPool_destruct(EnvironmentPool*);
+
+EnvironmentPool* EnvironmentPool_construct()
 {
-  Runtime* result = malloc(sizeof(Runtime));
-  result->permanentStringsLength = 0;
-  result->permanentStringsAllocated = 0;
-  result->permanentStrings = NULL;
+  EnvironmentPool* result = malloc(sizeof(EnvironmentPool));
+  EnvironmentPool_initialize(result);
   return result;
 }
 
-void Runtime_destruct(Runtime* self)
+void EnvironmentPool_initialize(EnvironmentPool* self)
+{
+  self->overflow = NULL;
+  self->freeIndex = 0;
+
+  for(size_t i = 0; i < POOL_SIZE; i++)
+  {
+    self->allocatedFlags[i] = false;
+    self->environments[i].live = false;
+  }
+}
+
+void EnvironmentPool_deinitialize(EnvironmentPool* self)
 {
-  free(self->permanentStrings);
+  // We can assume if this is being called, none of the Environments are live
+  for(int8_t i = 0; i < POOL_SIZE; i++)
+  {
+    if(self->allocatedFlags[i]) Environment_deinitialize(&(self->environments[i]));
+  }
+
+  EnvironmentPool_destruct(self->overflow);
+}
+
+void EnvironmentPool_destruct(EnvironmentPool* self)
+{
+  if(self == NULL) return;
+  EnvironmentPool_deinitialize(self);
   free(self);
 }
 
-void Runtime_addPermanentString(Runtime* self, String* string)
+void EnvironmentPool_GC(EnvironmentPool* self)
 {
-  // TODO Make this function thread-safe
-  if(self->permanentStringsLength == self->permanentStringsAllocated)
+  // Unmark all the environments
+  for(EnvironmentPool* current = self; current != NULL; current = current->overflow)
   {
-    if(self->permanentStringsAllocated == 0)
+    for(int8_t i = 0; i < POOL_SIZE; i++)
     {
-      self->permanentStringsAllocated = 8;
+      current->environments[i].mark = false;
     }
-    else
+  }
+
+  // Mark live enviroments and environments referenced by live environments
+  for(EnvironmentPool* current = self; current != NULL; current = current->overflow)
+  {
+    for(int8_t i = 0; i < POOL_SIZE; i++)
     {
-      self->permanentStringsAllocated = self->permanentStringsAllocated * 2;
+      if(current->environments[i].live)
+      {
+        Environment_mark(&(current->environments[i]));
+      }
     }
+  }
 
-    self->permanentStrings = realloc(
-      self->permanentStrings,
-      sizeof(String*) * self->permanentStringsAllocated
-    );
+  // TODO We never free pools until the very end--we could free a pool if two pools are empty
+  for(EnvironmentPool* current = self; current != NULL; current = current->overflow)
+  {
+    for(int8_t i = POOL_SIZE - 1; i >= 0; i--)
+    {
+      if(!current->environments[i].mark && current->allocatedFlags[i])
+      {
+        Environment_deinitialize(&(current->environments[i]));
+        current->allocatedFlags[i] = false;
+        current->freeIndex = i;
+      }
+    }
+  }
+}
 
-    // TODO Handle realloc returning NULL
+Environment* EnvironmentPool_allocate(EnvironmentPool* self)
+{
+  for(EnvironmentPool* current = self; current != NULL; current = current->overflow)
+  {
+    for(; current->freeIndex < POOL_SIZE; current->freeIndex++)
+    {
+      if(!current->allocatedFlags[current->freeIndex])
+      {
+        current->allocatedFlags[current->freeIndex] = true;
+        return &(current->environments[current->freeIndex]);
+      }
+    }
   }
 
-  self->permanentStrings[self->permanentStringsLength] = string;
-  self->permanentStringsLength++;
+  EnvironmentPool_GC(self);
+
+  EnvironmentPool* previous;
+  for(EnvironmentPool* current = self; current != NULL; current = current->overflow)
+  {
+    for(; current->freeIndex < POOL_SIZE; current->freeIndex++)
+    {
+      if(!current->allocatedFlags[current->freeIndex])
+      {
+        current->allocatedFlags[current->freeIndex] = true;
+        return &(current->environments[current->freeIndex]);
+      }
+      else
+      {
+        previous = current;
+      }
+    }
+  }
+
+  previous->overflow = EnvironmentPool_construct();
+  return EnvironmentPool_allocate(previous->overflow);
 }
 
 Object integerLiteral(int32_t literal)
@@ -99,49 +316,119 @@ Object integerLiteral(int32_t literal)
   return result;
 }
 
-Object stringLiteral(Runtime* runtime, const char* literal)
+Object stringLiteral(const char* literal)
 {
-  String* resultString = malloc(sizeof(String));
-  resultString->length = strlen(literal);
-  resultString->characters = malloc(resultString->length);
-  memcpy(resultString->characters, literal, resultString->length);
-  Runtime_addPermanentString(runtime, resultString);
-
   Object result;
   result.type = STRING;
-  result.instance.string = resultString;
+  result.instance.string = literal;
+  return result;
+}
+
+// TODO Make this conditionally added
+Object operator$negate(Object input)
+{
+  assert(input.type == INTEGER);
+
+  Object result;
+  result.type = INTEGER;
+  result.instance.integer = -input.instance.integer;
+  return result;
+}
+
+{% for id in infix_declarations %}
+Object operator${{ id.name }}(Object left, Object right)
+{
+  assert(left.type == {{ id.in_type.upper() }});
+  assert(right.type == {{ id.in_type.upper() }});
+
+  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;
 }
+{% endfor %}
+
+{% if 'pow' in builtins %}
+Object builtin$pow$implementation(EnvironmentPool* environmentPool, Environment* parent, size_t argc, Object* args)
+{
+  assert(argc == 2);
+
+  Object base = args[0];
+  Object exponent = args[1];
+
+  assert(base.type == INTEGER);
+  assert(exponent.type == INTEGER);
+
+  Object result;
+  result.type = INTEGER;
+  result.instance.integer = pow(base.instance.integer, exponent.instance.integer);
+  return result;
+}
+
+Object builtin$pow = { CLOSURE, (Instance)(Closure){ NULL, builtin$pow$implementation } };
+{% endif %}
 
 {% if 'print' in builtins %}
-void builtin$print(Object output)
+Object builtin$print$implementation(EnvironmentPool* environmentPool, Environment* parent, size_t argc, Object* args)
 {
-  switch(output.type)
+  for(size_t i = 0; i < argc; i++)
   {
-    case INTEGER:
-      printf("%" PRId32, output.instance.integer);
-      break;
+    Object output = args[i];
+    switch(output.type)
+    {
+      case BOOLEAN:
+        fputs(output.instance.boolean ? "true" : "false", stdout);
+        break;
+
+      case CLOSURE:
+        // TODO Print something better
+        printf("<Closure>");
+        break;
+
+      case INTEGER:
+        printf("%" PRId32, output.instance.integer);
+        break;
+
+      case STRING:
+        // Using fwrite instead of printf to handle size_t length
+        printf("%s", output.instance.string);
+        break;
 
-    case STRING:
-      // Using fwrite instead of printf to handle size_t length
-      fwrite(output.instance.string->characters, 1, output.instance.string->length, stdout);
-      break;
+      case VOID:
+        printf("nil");
+        break;
 
-    default:
-      assert(false);
+      default:
+        assert(false);
+    }
   }
+
+  // TODO Return something better
+  return builtin$false;
 }
+
+Object builtin$print = { CLOSURE, (Instance)(Closure){ NULL, builtin$print$implementation } };
 {% endif %}
+{% for function_definition in function_definition_list %}
+{{ function_definition }}
+{% endfor %}
 
 int main(int argc, char** argv)
 {
-  Runtime* runtime = Runtime_construct();
+  EnvironmentPool* environmentPool = EnvironmentPool_construct();
+  Environment* environment = EnvironmentPool_allocate(environmentPool);
+  Environment_initialize(environment, NULL);
+
+  // TODO Use the symbol from SYMBOL_LIST
+  {% for builtin in builtins %}
+  Environment_set(environment, "{{ builtin }}", builtin${{ builtin }});
+  {% endfor %}
 
   {% for statement in statements %}
   {{ statement }}
   {% endfor %}
 
-  Runtime_destruct(runtime);
-
+  Environment_setLive(environment, false);
+  EnvironmentPool_destruct(environmentPool);
   return 0;
 }