git.gag.com Git - fw/altos/blob - src/scheme/ao_scheme_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_scheme.h"
  16
  17 static void ao_scheme_invalid_write(FILE *out, ao_poly p, bool write) {
  18         fprintf(out, "??? type %d poly 0x%04x ???", ao_scheme_poly_type (p), p);
  19         (void) write;
  20         ao_scheme_abort();
  21 }
  22
  23 static void (*const ao_scheme_write_funcs[AO_SCHEME_NUM_TYPE]) (FILE *out, ao_poly p, bool write) = {
  24         [AO_SCHEME_CONS] = ao_scheme_cons_write,
  25 #ifdef AO_SCHEME_FEATURE_BIGINT
  26         [AO_SCHEME_BIGINT] = ao_scheme_bigint_write,
  27 #endif
  28         [AO_SCHEME_INT] = ao_scheme_int_write,
  29         [AO_SCHEME_ATOM] = ao_scheme_atom_write,
  30         [AO_SCHEME_BUILTIN] = ao_scheme_builtin_write,
  31         [AO_SCHEME_FRAME] = ao_scheme_frame_write,
  32         [AO_SCHEME_FRAME_VALS] = ao_scheme_invalid_write,
  33         [AO_SCHEME_LAMBDA] = ao_scheme_lambda_write,
  34         [AO_SCHEME_STACK] = ao_scheme_stack_write,
  35         [AO_SCHEME_BOOL] = ao_scheme_bool_write,
  36         [AO_SCHEME_STRING] = ao_scheme_string_write,
  37 #ifdef AO_SCHEME_FEATURE_FLOAT
  38         [AO_SCHEME_FLOAT] = ao_scheme_float_write,
  39 #endif
  40 #ifdef AO_SCHEME_FEATURE_VECTOR
  41         [AO_SCHEME_VECTOR] = ao_scheme_vector_write,
  42 #endif
  43 #ifdef AO_SCHEME_FEATURE_PORT
  44         [AO_SCHEME_PORT] = ao_scheme_port_write,
  45 #endif
  46 };
  47
  48 void (*ao_scheme_poly_write_func(ao_poly p))(FILE *out, ao_poly p, bool write)
  49 {
  50         uint8_t type = ao_scheme_poly_type(p);
  51
  52         if (type < AO_SCHEME_NUM_TYPE)
  53                 return ao_scheme_write_funcs[type];
  54         return ao_scheme_invalid_write;
  55 }
  56
  57 void *
  58 ao_scheme_ref(ao_poly poly) {
  59         if (poly == AO_SCHEME_NIL)
  60                 return NULL;
  61         if (poly & AO_SCHEME_CONST)
  62                 return (void *) (ao_scheme_const + (poly & AO_SCHEME_REF_MASK) - 4);
  63         return (void *) (ao_scheme_pool + (poly & AO_SCHEME_REF_MASK) - 4);
  64 }
  65
  66 ao_poly
  67 ao_scheme_poly(const void *addr, ao_poly type) {
  68         const uint8_t   *a = addr;
  69         if (a == NULL)
  70                 return AO_SCHEME_NIL;
  71         if (ao_scheme_is_const_addr(a))
  72                 return AO_SCHEME_CONST | (a - ao_scheme_const + 4) | type;
  73         return (a - ao_scheme_pool + 4) | type;
  74 }