mes/mes.c

/* -*-comment-start: "//";comment-end:""-*-
 * Mes --- Maxwell Equations of Software
 * Copyright © 2016 Jan Nieuwenhuizen <janneke@gnu.org>
 *
 * This file is part of Mes.
 *
 * Mes is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or (at
 * your option) any later version.
 *
 * Mes is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Mes.  If not, see <http://www.gnu.org/licenses/>.
 */

#define _GNU_SOURCE
#include <assert.h>
#include <ctype.h>
#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdbool.h>

#define DEBUG 0
#define QUASIQUOTE 1
#define QUASISYNTAX 0
#define ENV_CACHE 0
#define FIXED_PRIMITIVES 1
#define READER 1

#if READER
int ARENA_SIZE = 1000000;
#else
int ARENA_SIZE = 100000;
#endif
int MAX_ARENA_SIZE = 20000000;
int GC_SAFETY = 100;

typedef int SCM;
enum type_t {CHAR, FUNCTION, KEYWORD, MACRO, NUMBER, PAIR, SPECIAL, STRING, SYMBOL, REF, VALUES, VECTOR, BROKEN_HEART};
typedef SCM (*function0_t) (void);
typedef SCM (*function1_t) (SCM);
typedef SCM (*function2_t) (SCM, SCM);
typedef SCM (*function3_t) (SCM, SCM, SCM);
typedef SCM (*functionn_t) (SCM);
typedef struct function_t {
  union {
    function0_t function0;
    function1_t function1;
    function2_t function2;
    function3_t function3;
    functionn_t functionn;
  };
  int arity;
} function;
struct scm_t;
typedef struct scm_t {
  enum type_t type;
  union {
    char const *name;
    SCM string;
    SCM car;
    SCM ref;
    int length;
  };
  union {
    int value;
    int function;
    SCM cdr;
    SCM macro;
    SCM vector;
    int hits;
  };
} scm;

scm scm_nil = {SPECIAL, "()"};
scm scm_f = {SPECIAL, "#f"};
scm scm_t = {SPECIAL, "#t"};
scm scm_dot = {SPECIAL, "."};
scm scm_arrow = {SPECIAL, "=>"};
scm scm_undefined = {SPECIAL, "*undefined*"};
scm scm_unspecified = {SPECIAL, "*unspecified*"};
scm scm_closure = {SPECIAL, "*closure*"};
scm scm_circular = {SPECIAL, "*circular*"};
#if BOOT
scm scm_label = {
  SPECIAL, "label"};
#endif
scm scm_begin = {SPECIAL, "*begin*"};

scm scm_symbol_lambda = {SYMBOL, "lambda"};
scm scm_symbol_begin = {SYMBOL, "begin"};
scm scm_symbol_if = {SYMBOL, "if"};
scm scm_symbol_define = {SYMBOL, "define"};
scm scm_symbol_define_macro = {SYMBOL, "define-macro"};
scm scm_symbol_set_x = {SYMBOL, "set!"};

scm scm_symbol_quote = {SYMBOL, "quote"};
scm scm_symbol_quasiquote = {SYMBOL, "quasiquote"};
scm scm_symbol_unquote = {SYMBOL, "unquote"};
scm scm_symbol_unquote_splicing = {SYMBOL, "unquote-splicing"};

scm scm_symbol_sc_expand = {SYMBOL, "sc-expand"};
scm scm_symbol_expand_macro = {SYMBOL, "expand-macro"};
scm scm_symbol_sc_expander_alist = {SYMBOL, "*sc-expander-alist*"};
scm scm_symbol_noexpand = {SYMBOL, "noexpand"};
scm scm_symbol_syntax = {SYMBOL, "syntax"};
scm scm_symbol_quasisyntax = {SYMBOL, "quasisyntax"};
scm scm_symbol_unsyntax = {SYMBOL, "unsyntax"};
scm scm_symbol_unsyntax_splicing = {SYMBOL, "unsyntax-splicing"};

scm scm_symbol_call_with_values = {SYMBOL, "call-with-values"};
scm scm_symbol_current_module = {SYMBOL, "current-module"};
scm scm_symbol_primitive_load = {SYMBOL, "primitive-load"};
scm scm_symbol_read_input_file = {SYMBOL, "read-input-file"};

scm scm_symbol_the_unquoters = {SYMBOL, "*the-unquoters*"};

scm scm_symbol_car = {SYMBOL, "car"};
scm scm_symbol_cdr = {SYMBOL, "cdr"};
scm scm_symbol_null_p = {SYMBOL, "null?"};
scm scm_symbol_eq_p = {SYMBOL, "eq?"};
scm scm_symbol_cons = {SYMBOL, "cons"};

scm char_eof = {CHAR, .name="*eof*", .value=-1};
scm char_nul = {CHAR, .name="nul", .value=0};
scm char_backspace = {CHAR, .name="backspace", .value=8};
scm char_tab = {CHAR, .name="tab", .value=9};
scm char_newline = {CHAR, .name="newline", .value=10};
scm char_vt = {CHAR, .name="vt", .value=11};
scm char_page = {CHAR, .name="page", .value=12};
scm char_return = {CHAR, .name="return", .value=13};
scm char_space = {CHAR, .name="space", .value=32};

scm g_free = {NUMBER, .value=0};
scm *g_cells;
scm *g_news = 0;

#include "mes.symbols.h"

SCM tmp;
SCM tmp_num;
SCM tmp_num2;
SCM tmp_num3;
SCM tmp_num4;

function functions[200];
int g_function = 0;

SCM g_symbols = 0;
SCM stack = 0;
SCM r0 = 0; // a/env
SCM r1 = 0; // param 1
SCM r2 = 0; // param 2
SCM r3 = 0; // param 3

#include "define.h"
#include "display.h"
#include "lib.h"
#include "math.h"
#include "mes.h"
#include "posix.h"
#include "quasiquote.h"
#include "reader.h"
#include "string.h"
#include "type.h"

#define CAR(x) g_cells[x].car
#define CDR(x) g_cells[x].cdr
#define HITS(x) g_cells[x].hits
#define LENGTH(x) g_cells[x].length
#define NAME(x) g_cells[x].name
#define STRING(x) g_cells[x].string
#define TYPE(x) g_cells[x].type
#define MACRO(x) g_cells[x].macro
#define REF(x) g_cells[x].ref
#define VALUE(x) g_cells[x].value
#define VECTOR(x) g_cells[x].vector
#define FUNCTION(x) functions[g_cells[x].function]
#define NCAR(x) g_news[x].car
#define NTYPE(x) g_news[x].type

#define CAAR(x) CAR (CAR (x))
#define CDAR(x) CDR (CAR (x))
#define CAAR(x) CAR (CAR (x))
#define CADAR(x) CAR (CDR (CAR (x)))
#define CADDR(x) CAR (CDR (CDR (x)))
#define CDADAR(x) CAR (CDR (CAR (CDR (x))))
#define CADR(x) CAR (CDR (x))

SCM display_ (FILE* f, SCM x);
SCM vm_call (function0_t f, SCM p1, SCM p2, SCM a);

SCM
alloc (int n)
{
  assert (g_free.value + n < ARENA_SIZE);
  SCM x = g_free.value;
  g_free.value += n;
  return x;
}

SCM
make_cell (SCM type, SCM car, SCM cdr)
{
  SCM x = alloc (1);
  assert (TYPE (type) == NUMBER);
  TYPE (x) = VALUE (type);
  if (VALUE (type) == CHAR || VALUE (type) == NUMBER) {
    if (car) CAR (x) = CAR (car);
    if (cdr) CDR (x) = CDR (cdr);
  } else if (VALUE (type) == FUNCTION) {
    if (car) CAR (x) = car;
    if (cdr) CDR (x) = CDR (cdr);
  } else {
    CAR (x) = car;
    CDR (x) = cdr;
  }
  return x;
}

SCM
cons (SCM x, SCM y)
{
  g_cells[tmp_num].value = PAIR;
  return make_cell (tmp_num, x, y);
}

SCM
car (SCM x)
{
  assert (TYPE (x) == PAIR);
  return CAR (x);
}

SCM
cdr (SCM x)
{
  assert (TYPE (x) == PAIR);
  return CDR (x);
}

SCM
eq_p (SCM x, SCM y)
{
  return (x == y
          || ((TYPE (x) == KEYWORD && TYPE (y) == KEYWORD
               && STRING (x) == STRING (y)))
          || (TYPE (x) == CHAR && TYPE (y) == CHAR
              && VALUE (x) == VALUE (y))
          || (TYPE (x) == NUMBER && TYPE (y) == NUMBER
              && VALUE (x) == VALUE (y)))
    ? cell_t : cell_f;
}

SCM
set_car_x (SCM x, SCM e)
{
  assert (TYPE (x) == PAIR);
  CAR (x) = e;
  return cell_unspecified;
}

SCM
set_cdr_x (SCM x, SCM e)
{
  assert (TYPE (x) == PAIR);
  CDR (x) = e;
  return cell_unspecified;
}

SCM
set_env_x (SCM x, SCM e, SCM a)
{
  SCM p = assert_defined (x, assq (x, a));
  return set_cdr_x (p, e);
}

SCM
quote (SCM x)
{
  return cons (cell_symbol_quote, x);
}

SCM
quasiquote (SCM x)
{
  return cons (cell_symbol_quasiquote, x);
}

SCM
quasisyntax (SCM x)
{
  return cons (cell_symbol_quasisyntax, x);
}

SCM
pairlis (SCM x, SCM y, SCM a)
{
  if (x == cell_nil)
    return a;
  if (pair_p (x) == cell_f)
    return cons (cons (x, y), a);
  return cons (cons (car (x), car (y)),
               pairlis (cdr (x), cdr (y), a));
}

SCM
assq (SCM x, SCM a)
{
  while (a != cell_nil && eq_p (x, CAAR (a)) == cell_f)
    {
      if (TYPE (a) == BROKEN_HEART || TYPE (CAR (a)) == BROKEN_HEART)
        fprintf (stderr, "oops, broken heart\n");
      a = CDR (a);
    }
  return a != cell_nil ? car (a) : cell_f;
}

SCM
assq_ref_cache (SCM x, SCM a)
{
  x = assq (x, a);
  if (x == cell_f) return cell_undefined;
  return cdr (x);
}

SCM
assert_defined (SCM x, SCM e)
{
  if (e == cell_undefined)
    {
      fprintf (stderr, "eval: unbound variable:");
      display_ (stderr, x);
      fprintf (stderr, "\n");
      assert (!"unbound variable");
    }
  return e;
}

SCM
vm_evlis_env ()
{
  if (r1 == cell_nil) return cell_nil;
  if (TYPE (r1) != PAIR) return eval_env (r1, r0);
  r2 = eval_env (car (r1), r0);
  r1 = evlis_env (cdr (r1), r0);
  return cons (r2, r1);
}

SCM
vm_call_lambda ()
{
  return vm_call (vm_begin_env, r1, cell_undefined, r0);
}

SCM
call_lambda (SCM e, SCM x, SCM aa, SCM a) ///((internal))
{
  SCM cl = cons (cons (cell_closure, x), x);
  r1 = e;
  r0 = cl;
  r2 = a;
  r3 = aa;
  return vm_call_lambda ();
}

SCM
vm_apply_env ()
{
  if (TYPE (r1) != PAIR)
    {
      if (TYPE (r1) == FUNCTION) return call (r1, r2);
      if (r1 == cell_symbol_call_with_values)
        return call_with_values_env (car (r2), cadr (r2), r0);
      if (r1 == cell_symbol_current_module) return r0;
    }
  else
    switch (car (r1))
      {
      case cell_symbol_lambda:
        {
          SCM args = cadr (r1);
          SCM body = cddr (r1);
          SCM p = pairlis (args, r2, r0);
          return call_lambda (body, p, p, r0);
        }
      case cell_closure:
        {
          SCM args = caddr (r1);
          SCM body = cdddr (r1);
          SCM aa = cdadr (r1);
          aa = cdr (aa);
          SCM p = pairlis (args, r2, aa);
          return call_lambda (body, p, aa, r0);
        }
#if BOOT
      case cell_symbol_label:
        return apply_env (caddr (r1), r2, cons (cons (cadr (r1), caddr (r1)), r0));
#endif
      }
  SCM e = eval_env (r1, r0);
  char const* type = 0;
  if (e == cell_f || e == cell_t) type = "bool";
  if (TYPE (e) == CHAR) type = "char";
  if (TYPE (e) == NUMBER) type = "number";
  if (TYPE (e) == STRING) type = "string";
  if (e == cell_unspecified) type = "*unspecified*";
  if (e == cell_undefined) type =  "*undefined*";
  if (type)
    {
      fprintf (stderr, "cannot apply: %s: ", type);
      display_ (stderr, e);
      fprintf (stderr, " [");
      display_ (stderr, r1);
      fprintf (stderr, "]\n");
      assert (!"cannot apply");
    }
  return apply_env (e, r2, r0);
}

SCM
vm_eval_env ()
{
  switch (TYPE (r1))
    {
    case PAIR:
      {
        switch (car (r1))
          {
#if FIXED_PRIMITIVES
          case cell_symbol_car: return car (eval_env (CADR (r1), r0));
          case cell_symbol_cdr: return cdr (eval_env (CADR (r1), r0));
          case cell_symbol_cons: {SCM m = evlis_env (CDR (r1), r0);
              return cons (CAR (m), CADR (m));}
          case cell_symbol_null_p: return null_p (eval_env (CADR (r1), r0));
#endif // FIXED_PRIMITIVES
          case cell_symbol_quote: return cadr (r1);
#if QUASISYNTAX
          case cell_symbol_syntax: return r1;
#endif
          case cell_symbol_begin: return begin_env (r1, r0);
          case cell_symbol_lambda:
            return make_closure (cadr (r1), cddr (r1), assq (cell_closure, r0));
          case cell_closure: return r1;
          case cell_symbol_if: return if_env (cdr (r1), r0);
#if !BOOT
          case cell_symbol_define: return define_env (r1, r0);
          case cell_symbol_define_macro: return define_env (r1, r0);
#endif
#if 1 //!BOOT
          case cell_symbol_set_x: {
            SCM x = eval_env (caddr (r1), r0); return set_env_x (cadr (r1), x, r0);
          }
#endif
#if QUASIQUOTE
          case cell_symbol_unquote: return eval_env (cadr (r1), r0);
          case cell_symbol_quasiquote: return eval_quasiquote (cadr (r1), add_unquoters (r0));
#endif //QUASIQUOTE
#if QUASISYNTAX
          case cell_symbol_unsyntax: return eval_env (cadr (r1), r0);
          case cell_symbol_quasisyntax: return eval_quasisyntax (cadr (r1), add_unsyntaxers (r0));
#endif //QUASISYNTAX
          default: {
            SCM x = expand_macro_env (r1, r0);
            if (x != r1) return eval_env (x, r0);
            SCM m = evlis_env (CDR (r1), r0);
            return apply_env (car (r1), m, r0);
          }
          }
      }
    case SYMBOL: return assert_defined (r1, assq_ref_cache (r1, r0));
    default: return r1;
    }
}

SCM
vm_expand_macro_env ()
{
  if (TYPE (CAR (r1)) == STRING && string_to_symbol (CAR (r1)) == cell_symbol_noexpand)
    return cadr (r1);

  SCM macro;
  SCM expanders;
  if (TYPE (r1) == PAIR
      && (macro = lookup_macro (car (r1), r0)) != cell_f)
    return apply_env (macro, CDR (r1), r0);
  else if (TYPE (r1) == PAIR
           && TYPE (CAR (r1)) == SYMBOL
           && ((expanders = assq_ref_cache (cell_symbol_sc_expander_alist, r0)) != cell_undefined)
           && ((macro = assq (CAR (r1), expanders)) != cell_f))
    {
      SCM sc_expand = assq_ref_cache (cell_symbol_expand_macro, r0);
      if (sc_expand != cell_undefined && sc_expand != cell_f)
        r1 = apply_env (sc_expand, cons (r1, cell_nil), r0);
    }
  return r1;
}

SCM
vm_begin_env ()
{
  SCM r = cell_unspecified;
  while (r1 != cell_nil) {
    if (TYPE (r1) == PAIR && TYPE (CAR (r1)) == PAIR)
      {
        if (caar (r1) == cell_symbol_begin)
          r1 = append2 (cdar (r1), cdr (r1));
        else if (caar (r1) == cell_symbol_primitive_load)
          {
            SCM f = read_input_file_env (r0);
            r1 = append2 (f, cdr (r1));
          }
      }
    r = eval_env (car (r1), r0);
    r1 = CDR (r1);
  }
  return r;
}

SCM
vm_if_env ()
{
  SCM x = eval_env (car (r1), r0);
  if (x != cell_f)
    return eval_env (cadr (r1), r0);
  if (cddr (r1) != cell_nil)
    return eval_env (caddr (r1), r0);
  return cell_unspecified;
}

SCM
vm_call_with_values_env ()
{
  SCM v = apply_env (r1, cell_nil, r0);
  if (TYPE (v) == VALUES)
    v = CDR (v);
  return apply_env (r2, v, r0);
}

SCM
call (SCM fn, SCM x)
{
  if ((FUNCTION (fn).arity > 0 || FUNCTION (fn).arity == -1)
      && x != cell_nil && TYPE (CAR (x)) == VALUES)
    x = cons (CADAR (x), CDR (x));
  if ((FUNCTION (fn).arity > 1 || FUNCTION (fn).arity == -1)
      && x != cell_nil && TYPE (CDR (x)) == PAIR && TYPE (CADR (x)) == VALUES)
    x = cons (CAR (x), cons (CDADAR (x), CDR (x)));
  switch (FUNCTION (fn).arity)
    {
    case 0: return FUNCTION (fn).function0 ();
    case 1: return FUNCTION (fn).function1 (car (x));
    case 2: return FUNCTION (fn).function2 (car (x), cadr (x));
    case 3: return FUNCTION (fn).function3 (car (x), cadr (x), caddr (x));
    case -1: return FUNCTION (fn).functionn (x);
    }
  return cell_unspecified;
}

SCM
gc_frame (SCM stack)
{
  SCM frame = car (stack);
  r1 = car (frame);
  r2 = cadr (frame);
  r3 = caddr (frame);
  r0 = cadddr (frame);
  return frame;
}

SCM
gc_stack (SCM a)
{
  SCM frame = cons (r1, cons (r2, cons (r3, cons (r0, cell_nil))));
  stack = cons (frame, stack);
  stack = gc (stack);
  gc_frame (stack);
  stack = cdr (stack);
  return stack;
}

SCM
vm_call (function0_t f, SCM p1, SCM p2, SCM a)
{
  SCM frame = cons (r1, cons (r2, cons (r3, cons (r0, cell_nil))));
  stack = cons (frame, stack);
  r1 = p1;
  r2 = p2;
  r0 = a;
  if (g_free.value + GC_SAFETY > ARENA_SIZE)
    gc_stack (stack);

  SCM r = f ();
  frame = gc_frame (stack);
  stack = cdr (stack);
  return r;
}

SCM
evlis_env (SCM m, SCM a)
{
  return vm_call (vm_evlis_env, m, cell_undefined, a);
}

SCM
apply_env (SCM fn, SCM x, SCM a)
{
  return vm_call (vm_apply_env, fn, x, a);
}

SCM
eval_env (SCM e, SCM a)
{
  return vm_call (vm_eval_env, e, cell_undefined, a);
}

SCM
expand_macro_env (SCM e, SCM a)
{
  return vm_call (vm_expand_macro_env, e, cell_undefined, a);
}

SCM
begin_env (SCM e, SCM a)
{
  return vm_call (vm_begin_env, e, cell_undefined, a);
}

SCM
if_env (SCM e, SCM a)
{
  return vm_call (vm_if_env, e, cell_undefined, a);
}

SCM
call_with_values_env (SCM producer, SCM consumer, SCM a)
{
  return vm_call (vm_call_with_values_env, producer, consumer, a);
}

SCM
append2 (SCM x, SCM y)
{
  if (x == cell_nil) return y;
  assert (TYPE (x) == PAIR);
  return cons (car (x), append2 (cdr (x), y));
}

SCM
append (SCM x) ///((arity . n))
 {
  if (x == cell_nil) return cell_nil;
  return append2 (car (x), append (cdr (x)));
 }

SCM
make_char (int x)
{
  g_cells[tmp_num].value = CHAR;
  g_cells[tmp_num2].value = x;
  return make_cell (tmp_num, tmp_num2, tmp_num2);
}

SCM
make_function (SCM name, SCM id, SCM arity)
{
  g_cells[tmp_num3].value = FUNCTION;
  function *f = (function*)malloc (sizeof (function));
  f->arity = VALUE (arity);
  g_cells[tmp_num4].value = (long)f;
  return make_cell (tmp_num3, name, tmp_num4);
}

SCM
make_keyword (SCM s)
{
  SCM x = internal_lookup_symbol (s);
  x = x ? x : internal_make_symbol (s);
  g_cells[tmp_num].value = KEYWORD;
  return make_cell (tmp_num, STRING (x), 0);
}

SCM
make_macro (SCM name, SCM x)
{
  g_cells[tmp_num].value = MACRO;
  return make_cell (tmp_num, STRING (name), x);
}

SCM
make_number (int x)
{
  g_cells[tmp_num].value = NUMBER;
  g_cells[tmp_num2].value = x;
  return make_cell (tmp_num, tmp_num2, tmp_num2);
}

SCM
make_ref (SCM x)
{
  g_cells[tmp_num].value = REF;
  return make_cell (tmp_num, x, x);
}

SCM
make_string (SCM x)
{
  g_cells[tmp_num].value = STRING;
  return make_cell (tmp_num, x, 0);
}

SCM
cstring_to_list (char const* s)
{
  SCM p = cell_nil;
  int i = strlen (s);
  while (i--)
    p = cons (make_char (s[i]), p);
  return p;
}

SCM
null_p (SCM x)
{
  return x == cell_nil ? cell_t : cell_f;
}

SCM
internal_make_symbol (SCM s)
{
  g_cells[tmp_num].value = SYMBOL;
  SCM x = make_cell (tmp_num, s, 0);
  g_symbols = cons (x, g_symbols);
  return x;
}

SCM
make_symbol (SCM s)
{
  SCM x = internal_lookup_symbol (s);
  return x ? x : internal_make_symbol (s);
}

SCM
make_vector (SCM n)
{
  int k = VALUE (n);
  g_cells[tmp_num].value = VECTOR;
  SCM v = alloc (k);
  SCM x = make_cell (tmp_num, k, v);
  for (int i=0; i<k; i++) g_cells[v+i] = g_cells[vector_entry (cell_unspecified)];
  return x;
}

SCM
values (SCM x) ///((arity . n))
{
  SCM v = cons (0, x);
  TYPE (v) = VALUES;
  return v;
}

SCM
vector_length (SCM x)
{
  assert (TYPE (x) == VECTOR);
  return make_number (LENGTH (x));
}

SCM
vector_ref (SCM x, SCM i)
{
  assert (TYPE (x) == VECTOR);
  assert (VALUE (i) < LENGTH (x));
  SCM e = VECTOR (x) + VALUE (i);
  if (TYPE (e) == REF) e = g_cells[e].ref;
  if (TYPE (e) == CHAR) e = make_char (VALUE (e));
  if (TYPE (e) == NUMBER) e = make_number (VALUE (e));
  return e;
}

SCM
vector_entry (SCM x) {
  if (TYPE (x) == PAIR || TYPE (x) == SPECIAL || TYPE (x) == STRING || TYPE (x) == SYMBOL || TYPE (x) == VECTOR) x = make_ref (x);
  return x;
}

SCM
vector_set_x (SCM x, SCM i, SCM e)
{
  assert (TYPE (x) == VECTOR);
  assert (VALUE (i) < LENGTH (x));
  g_cells[VECTOR (x)+g_cells[i].value] = g_cells[vector_entry (e)];
  return cell_unspecified;
}

SCM
list_to_vector (SCM x)
{
  VALUE (tmp_num) = VALUE (length (x));
  SCM v = make_vector (tmp_num);
  SCM p = VECTOR (v);
  while (x != cell_nil)
    {
      g_cells[p++] = g_cells[vector_entry (car (x))];
      x = cdr (x);
    }
  return v;
}

FILE *g_stdin;
int
getchar ()
{
  return getc (g_stdin);
}

int
ungetchar (int c)
{
  return ungetc (c, g_stdin);
}

int
peekchar ()
{
  int c = getchar ();
  ungetchar (c);
  return c;
}

SCM
peek_byte ()
{
  return make_number (peekchar ());
}

SCM
read_byte ()
{
  return make_number (getchar ());
}

SCM
unread_byte (SCM i)
{
  return ungetchar (VALUE (i));
}

SCM
write_char (SCM x) ///((arity . n))
{
  SCM c = car (x);
  SCM p = cdr (x);
  int fd = 1;
  if (TYPE (p) == PAIR && TYPE (car (p)) == NUMBER) fd = VALUE (car (p));
  FILE *f = fd == 1 ? stdout : stderr;
  assert (TYPE (c) == NUMBER || TYPE (c) == CHAR);
  fputc (VALUE (c), f);
  return c;
}

SCM
symbol_to_list (SCM x)
{
  assert (TYPE (x) == SYMBOL);
  return STRING (x);
}

SCM
char_to_integer (SCM x)
{
  assert (TYPE (x) == CHAR);
  return make_number (VALUE (x));
}

SCM
integer_to_char (SCM x)
{
  assert (TYPE (x) == NUMBER);
  return make_char (VALUE (x));
}

void
make_tmps (scm* cells)
{
  tmp = g_free.value++;
  cells[tmp].type = CHAR;
  tmp_num = g_free.value++;
  cells[tmp_num].type = NUMBER;
  tmp_num2 = g_free.value++;
  cells[tmp_num2].type = NUMBER;
  tmp_num3 = g_free.value++;
  cells[tmp_num3].type = NUMBER;
  tmp_num4 = g_free.value++;
  cells[tmp_num4].type = NUMBER;
}

// Jam Collector
SCM g_symbol_max;
bool g_debug = false;

SCM
gc_up_arena ()
{
  ARENA_SIZE *= 2;
  void *p = realloc (g_cells-1, 2*ARENA_SIZE*sizeof(scm));
  if (!p)
    {
      if (g_debug) fprintf (stderr, "cannot up arena: %s: arena=%d\n", strerror (errno), 2*ARENA_SIZE);
      return cell_unspecified;
    }
  g_cells = (scm*)p;
  g_cells++;
  gc_init_news ();
}

SCM
gc ()
{
  if (g_debug) fprintf (stderr, "***gc[%d]...", g_free.value);
  g_free.value = 1;
  if (g_cells < g_news && ARENA_SIZE < MAX_ARENA_SIZE) gc_up_arena ();
  for (int i=g_free.value; i<g_symbol_max; i++)
    gc_copy (i);
  make_tmps (g_news);
  g_symbols = gc_copy (g_symbols);
  SCM new = gc_copy (stack);
  if (g_debug) fprintf (stderr, "new=%d\n", new, stack);
  stack = new;
  return gc_loop (1);
}

SCM
gc_loop (SCM scan)
{
  while (scan < g_free.value)
    {
      if (NTYPE (scan) == KEYWORD
          || NTYPE (scan) == MACRO
          || NTYPE (scan) == PAIR
          || NTYPE (scan) == REF
          || scan == 1 // null
          || NTYPE (scan) == SPECIAL
          || NTYPE (scan) == STRING
          || NTYPE (scan) == SYMBOL)
        {
          SCM car = gc_copy (g_news[scan].car);
          gc_relocate_car (scan, car);
        }
      if ((NTYPE (scan) == MACRO
           || NTYPE (scan) == PAIR
           || NTYPE (scan) == VALUES)
          && g_news[scan].cdr) // allow for 0 terminated list of symbols
        {
          SCM cdr = gc_copy (g_news[scan].cdr);
          gc_relocate_cdr (scan, cdr);
        }
      scan++;
    }
  return gc_flip ();
}

SCM
gc_copy (SCM old)
{
  if (TYPE (old) == BROKEN_HEART) return g_cells[old].car;
  SCM new = g_free.value++;
  g_news[new] = g_cells[old];
  if (NTYPE (new) == VECTOR)
    {
      g_news[new].vector = g_free.value;
      for (int i=0; i<LENGTH (old); i++)
        g_news[g_free.value++] = g_cells[VECTOR (old)+i];
    }
  g_cells[old].type = BROKEN_HEART;
  g_cells[old].car = new;
  return new;
}

SCM
gc_relocate_car (SCM new, SCM car)
{
  g_news[new].car = car;
  return cell_unspecified;
}

SCM
gc_relocate_cdr (SCM new, SCM cdr)
{
  g_news[new].cdr = cdr;
  return cell_unspecified;
}

SCM
gc_flip ()
{
  scm *cells = g_cells;
  g_cells = g_news;
  g_news = cells;
  if (g_debug) fprintf (stderr, " => jam[%d]\n", g_free.value);
  return stack;
}

SCM
gc_show ()
{
  fprintf (stderr, "cells: ");
  scm *t = g_cells;
  display_ (stderr, -1);
  fprintf (stderr, "\n");
  if (g_news)
    {
      fprintf (stderr, "news: ");
      g_cells = g_news;
      display_ (stderr, -1);
      fprintf (stderr, "\n");
    }
  g_cells = t;
  return cell_unspecified;
}

// Environment setup
SCM
acons (SCM key, SCM value, SCM alist)
{
  return cons (cons (key, value), alist);
}

SCM
add_environment (SCM a, char const *name, SCM x)
{
  return acons (make_symbol (cstring_to_list (name)), x, a);
}

SCM
gc_init_cells ()
{
  g_cells = (scm *)malloc (2*ARENA_SIZE*sizeof(scm));
  g_cells[0].type = VECTOR;
  g_cells[0].length = 1000;
  g_cells[0].vector = 0;
  g_cells++;
  g_cells[0].type = CHAR;
  g_cells[0].value = 'c';
}

SCM
gc_init_news ()
{
  g_news = g_cells-1 + ARENA_SIZE;
  g_news[0].type = VECTOR;
  g_news[0].length = 1000;
  g_news[0].vector = 0;
  g_news++;
  g_news[0].type = CHAR;
  g_news[0].value = 'n';
}

SCM
mes_symbols () ///((internal))
{
  gc_init_cells ();
  gc_init_news ();

#include "mes.symbols.i"

  g_symbol_max = g_free.value;
  make_tmps (g_cells);

  g_symbols = 0;
  for (int i=1; i<g_symbol_max; i++)
    g_symbols = cons (i, g_symbols);

  SCM a = cell_nil;

#if BOOT
  a = acons (cell_symbol_label, cell_t, a);
#endif
  a = acons (cell_symbol_begin, cell_begin, a);
  a = add_environment (a, "sc-expand", cell_f);
  a = acons (cell_closure, a, a);

  internal_lookup_symbol (cell_nil);

  return a;
}

SCM
mes_builtins (SCM a)
{
#include "mes.i"

#include "define.i"
#include "display.i"
#include "lib.i"
#include "math.i"
#include "posix.i"
#include "quasiquote.i"
#include "reader.i"
#include "string.i"
#include "type.i"

#include "define.environment.i"
#include "display.environment.i"
#include "lib.environment.i"
#include "math.environment.i"
#include "mes.environment.i"
#include "posix.environment.i"
  //#include "quasiquote.environment.i"
#include "reader.environment.i"
#include "string.environment.i"
#include "type.environment.i"

#if QUASIQUOTE
  SCM cell_unquote = assq_ref_cache (cell_symbol_unquote, a);
  SCM cell_unquote_splicing = assq_ref_cache (cell_symbol_unquote_splicing, a);
  SCM the_unquoters = cons (cons (cell_symbol_unquote, cell_unquote),
                            cons (cons (cell_symbol_unquote_splicing, cell_unquote_splicing),
                                  cell_nil));
  a = acons (cell_symbol_the_unquoters, the_unquoters, a);
#endif

  a = add_environment (a, "*dot*", cell_dot);
  a = add_environment (a, "*foo-bar-baz*", cell_nil); // FIXME: some off-by one?

  return a;
}

SCM
mes_stack (SCM a) ///((internal))
{
  r0 = a;
  r1 = make_char (0);
  r2 = make_char (0);
  r3 = make_char (0);
  stack = cons (cell_nil, cell_nil);
  return r0;
}

SCM
mes_environment () ///((internal))
{
  SCM a = mes_symbols ();
  return mes_stack (a);
}

SCM
make_lambda (SCM args, SCM body)
{
  return cons (cell_symbol_lambda, cons (args, body));
}

SCM
make_closure (SCM args, SCM body, SCM a)
{
  return cons (cell_closure, cons (cons (cell_circular, a), cons (args, body)));
}

SCM
lookup_macro (SCM x, SCM a)
{
  if (TYPE (x) != SYMBOL) return cell_f;
  SCM m = assq_ref_cache (x, a);
  if (macro_p (m) == cell_t) return MACRO (m);
  return cell_f;
}

SCM
read_input_file_env_ (SCM e, SCM a)
{
  if (e == cell_nil) return e;
  return cons (e, read_input_file_env_ (read_env (a), a));
}

SCM
read_input_file_env (SCM a)
{
  r0 = a;
#if READER
  return read_input_file_env_ (read_env (r0), r0);
#endif
  return apply_env (cell_symbol_read_input_file, cell_nil, r0);
}

SCM
load_env (SCM a) ///((internal))
{
  r0 =a;
#if !READER
  g_stdin = fopen ("module/mes/read-0.mes", "r");
  g_stdin = g_stdin ? g_stdin : fopen (PREFIX "module/mes/read-0.mes", "r");
#endif
  if (!g_function) r0 = mes_builtins (r0);
  r3 = read_input_file_env (r0);
  g_stdin = stdin;
  return r3;
}

SCM
bload_env (SCM a) ///((internal))
{
  g_stdin = fopen ("module/mes/read-0.mo", "r");
  g_stdin = g_stdin ? g_stdin : fopen (PREFIX "module/mes/read-0.mo", "r");
  char *p = (char*)g_cells;
  assert (getchar () == 'M');
  assert (getchar () == 'E');
  assert (getchar () == 'S');
  stack = getchar () << 8;
  stack += getchar ();
  int c = getchar ();
  while (c != EOF)
    {
      *p++ = c;
      c = getchar ();
    }
  g_free.value = (p-(char*)g_cells) / sizeof (scm);
  gc_frame (stack);
  g_symbols = r1;
  g_stdin = stdin;

  r0 = mes_builtins (r0);
  return r3;
}

int
dump ()
{
  r1 = g_symbols;
  SCM frame = cons (r1, cons (r2, cons (r3, cons (r0, cell_nil))));
  stack = cons (frame, stack);
  stack = gc (stack);
  gc_frame (stack);
  char *p = (char*)g_cells;
  fputc ('M', stdout);
  fputc ('E', stdout);
  fputc ('S', stdout);
  fputc (stack >> 8, stdout);
  fputc (stack % 256, stdout);
  for (int i=0; i<g_free.value * sizeof(scm); i++)
    fputc (*p++, stdout);
  return 0;
}

#include "type.c"
#include "define.c"
#include "display.c"
#include "lib.c"
#include "math.c"
#include "posix.c"
#include "quasiquote.c"
#include "reader.c"
#include "string.c"

int
main (int argc, char *argv[])
{
  g_debug = getenv ("MES_DEBUG");
  if (getenv ("MES_ARENA")) ARENA_SIZE = atoi (getenv ("MES_ARENA"));
  if (argc > 1 && !strcmp (argv[1], "--help")) return puts ("Usage: mes < FILE\n");
  if (argc > 1 && !strcmp (argv[1], "--version")) return puts ("Mes 0.3\n");
  g_stdin = stdin;
  r0 = mes_environment ();
  SCM program = (argc > 1 && !strcmp (argv[1], "--load"))
    ? bload_env (r0) : load_env (r0);
  if (argc > 1 && !strcmp (argv[1], "--dump")) return dump ();
  display_ (stderr, begin_env (program, r0));
  fputs ("", stderr);
  gc (stack);
  if (g_debug) fprintf (stderr, "\nstats: [%d]\n", g_free.value);
  return 0;
}