git.gag.com Git - fw/altos/blob - src/lisp/ao_lisp_poly.c

   1 /*
   2  * Copyright © 2016 Keith Packard <keithp@keithp.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  */
  14
  15 #include "ao_lisp.h"
  16
  17 #if 0
  18 #define DBG(...) printf (__VA_ARGS__)
  19 #else
  20 #define DBG(...)
  21 #endif
  22
  23 struct ao_lisp_funcs {
  24         void (*print)(ao_poly);
  25         void (*patom)(ao_poly);
  26 };
  27
  28 static const struct ao_lisp_funcs ao_lisp_funcs[AO_LISP_NUM_TYPE] = {
  29         [AO_LISP_CONS] = {
  30                 .print = ao_lisp_cons_print,
  31                 .patom = ao_lisp_cons_patom,
  32         },
  33         [AO_LISP_STRING] = {
  34                 .print = ao_lisp_string_print,
  35                 .patom = ao_lisp_string_patom,
  36         },
  37         [AO_LISP_INT] = {
  38                 .print = ao_lisp_int_print,
  39                 .patom = ao_lisp_int_print,
  40         },
  41         [AO_LISP_ATOM] = {
  42                 .print = ao_lisp_atom_print,
  43                 .patom = ao_lisp_atom_print,
  44         },
  45         [AO_LISP_BUILTIN] = {
  46                 .print = ao_lisp_builtin_print,
  47                 .patom = ao_lisp_builtin_print,
  48         },
  49         [AO_LISP_FRAME] = {
  50                 .print = ao_lisp_frame_print,
  51                 .patom = ao_lisp_frame_print,
  52         },
  53         [AO_LISP_LAMBDA] = {
  54                 .print = ao_lisp_lambda_print,
  55                 .patom = ao_lisp_lambda_print,
  56         },
  57         [AO_LISP_STACK] = {
  58                 .print = ao_lisp_stack_print,
  59                 .patom = ao_lisp_stack_print,
  60         },
  61 };
  62
  63 static const struct ao_lisp_funcs *
  64 funcs(ao_poly p)
  65 {
  66         uint8_t type = ao_lisp_poly_type(p);
  67
  68         if (type < AO_LISP_NUM_TYPE)
  69                 return &ao_lisp_funcs[type];
  70         return NULL;
  71 }
  72
  73 void
  74 ao_lisp_poly_print(ao_poly p)
  75 {
  76         const struct ao_lisp_funcs *f = funcs(p);
  77
  78         if (f && f->print)
  79                 f->print(p);
  80 }
  81
  82 void
  83 ao_lisp_poly_patom(ao_poly p)
  84 {
  85         const struct ao_lisp_funcs *f = funcs(p);
  86
  87         if (f && f->patom)
  88                 f->patom(p);
  89 }
  90
  91 void *
  92 ao_lisp_ref(ao_poly poly) {
  93         if (poly == AO_LISP_NIL)
  94                 return NULL;
  95         if (poly & AO_LISP_CONST)
  96                 return (void *) (ao_lisp_const + (poly & AO_LISP_REF_MASK) - 4);
  97         return (void *) (ao_lisp_pool + (poly & AO_LISP_REF_MASK) - 4);
  98 }
  99
 100 ao_poly
 101 ao_lisp_poly(const void *addr, ao_poly type) {
 102         const uint8_t   *a = addr;
 103         if (a == NULL)
 104                 return AO_LISP_NIL;
 105         if (AO_LISP_IS_CONST(a))
 106                 return AO_LISP_CONST | (a - ao_lisp_const + 4) | type;
 107         return (a - ao_lisp_pool + 4) | type;
 108 }