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 struct ao_lisp_funcs {
  18         void (*print)(ao_poly);
  19         void (*patom)(ao_poly);
  20 };
  21
  22 static const struct ao_lisp_funcs ao_lisp_funcs[AO_LISP_NUM_TYPE] = {
  23         [AO_LISP_CONS] = {
  24                 .print = ao_lisp_cons_print,
  25                 .patom = ao_lisp_cons_patom,
  26         },
  27         [AO_LISP_STRING] = {
  28                 .print = ao_lisp_string_print,
  29                 .patom = ao_lisp_string_patom,
  30         },
  31         [AO_LISP_INT] = {
  32                 .print = ao_lisp_int_print,
  33                 .patom = ao_lisp_int_print,
  34         },
  35         [AO_LISP_ATOM] = {
  36                 .print = ao_lisp_atom_print,
  37                 .patom = ao_lisp_atom_print,
  38         },
  39         [AO_LISP_BUILTIN] = {
  40                 .print = ao_lisp_builtin_print,
  41                 .patom = ao_lisp_builtin_print,
  42         },
  43         [AO_LISP_FRAME] = {
  44                 .print = ao_lisp_frame_print,
  45                 .patom = ao_lisp_frame_print,
  46         },
  47         [AO_LISP_LAMBDA] = {
  48                 .print = ao_lisp_lambda_print,
  49                 .patom = ao_lisp_lambda_print,
  50         },
  51         [AO_LISP_STACK] = {
  52                 .print = ao_lisp_stack_print,
  53                 .patom = ao_lisp_stack_print,
  54         },
  55 };
  56
  57 static const struct ao_lisp_funcs *
  58 funcs(ao_poly p)
  59 {
  60         uint8_t type = ao_lisp_poly_type(p);
  61
  62         if (type < AO_LISP_NUM_TYPE)
  63                 return &ao_lisp_funcs[type];
  64         return NULL;
  65 }
  66
  67 void
  68 ao_lisp_poly_print(ao_poly p)
  69 {
  70         const struct ao_lisp_funcs *f = funcs(p);
  71
  72         if (f && f->print)
  73                 f->print(p);
  74 }
  75
  76 void
  77 ao_lisp_poly_patom(ao_poly p)
  78 {
  79         const struct ao_lisp_funcs *f = funcs(p);
  80
  81         if (f && f->patom)
  82                 f->patom(p);
  83 }
  84
  85 void *
  86 ao_lisp_ref(ao_poly poly) {
  87         if (poly == AO_LISP_NIL)
  88                 return NULL;
  89         if (poly & AO_LISP_CONST)
  90                 return (void *) (ao_lisp_const + (poly & AO_LISP_REF_MASK) - 4);
  91         return (void *) (ao_lisp_pool + (poly & AO_LISP_REF_MASK) - 4);
  92 }
  93
  94 ao_poly
  95 ao_lisp_poly(const void *addr, ao_poly type) {
  96         const uint8_t   *a = addr;
  97         if (a == NULL)
  98                 return AO_LISP_NIL;
  99         if (AO_LISP_IS_CONST(a))
 100                 return AO_LISP_CONST | (a - ao_lisp_const + 4) | type;
 101         return (a - ao_lisp_pool + 4) | type;
 102 }