#include "script/expr.h"
#include "script/gc.h"
-
-static char *string_duplicate(const char *str,
- const char *str_end)
-{
- if (str_end != NULL && str > str_end) {
- return NULL;
- }
-
- const size_t n = str_end == NULL ? strlen(str) : (size_t) (str_end - str);
- char *dup_str = malloc(sizeof(char) * (n + 1));
- if (dup_str == NULL) {
- return NULL;
- }
-
- memcpy(dup_str, str, n);
- dup_str[n] = '\0';
-
- return dup_str;
-}
+#include "str.h"
struct Expr atom_as_expr(struct Atom *atom)
{
return expr;
}
-static struct Atom *clone_atom(Gc *gc, struct Atom *atom)
-{
- assert(atom);
-
- switch(atom->type) {
- case ATOM_NUMBER:
- return create_number_atom(gc, atom->num);
-
- case ATOM_STRING:
- return create_string_atom(gc, atom->str, NULL);
-
- case ATOM_SYMBOL:
- return create_symbol_atom(gc, atom->sym, NULL);
- }
-
- return atom;
-}
-
-struct Expr clone_expr(Gc *gc, struct Expr expr)
-{
- switch (expr.type) {
- case EXPR_ATOM:
- return atom_as_expr(clone_atom(gc, expr.atom));
-
- case EXPR_CONS:
- return cons_as_expr(
- create_cons(gc,
- clone_expr(gc, expr.cons->car),
- clone_expr(gc, expr.cons->cdr)));
-
- default: {}
- }
-
- return expr;
-}
-
struct Expr void_expr(void)
{
struct Expr expr = {
return expr;
}
-void print_atom_as_sexpr(struct Atom *atom)
+void print_atom_as_sexpr(FILE *stream, struct Atom *atom)
{
assert(atom);
switch (atom->type) {
case ATOM_SYMBOL:
- printf("%s", atom->sym);
+ fprintf(stream, "%s", atom->sym);
break;
case ATOM_NUMBER:
- printf("%f", atom->num);
+ fprintf(stream, "%ld", atom->num);
break;
case ATOM_STRING:
- printf("\"%s\"", atom->str);
+ fprintf(stream, "\"%s\"", atom->str);
+ break;
+
+ case ATOM_NATIVE:
+ fprintf(stream, "<native>");
break;
}
}
-void print_cons_as_sexpr(struct Cons *head)
+void print_cons_as_sexpr(FILE *stream, struct Cons *head)
{
assert(head);
struct Cons *cons = head;
- printf("(");
- print_expr_as_sexpr(cons->car);
+ fprintf(stream, "(");
+ print_expr_as_sexpr(stream, cons->car);
while (cons->cdr.type == EXPR_CONS) {
cons = cons->cdr.cons;
- printf(" ");
- print_expr_as_sexpr(cons->car);
+ fprintf(stream, " ");
+ print_expr_as_sexpr(stream, cons->car);
}
if (cons->cdr.atom->type != ATOM_SYMBOL ||
strcmp("nil", cons->cdr.atom->sym) != 0) {
- printf(" . ");
- print_expr_as_sexpr(cons->cdr);
+ fprintf(stream, " . ");
+ print_expr_as_sexpr(stream, cons->cdr);
}
- printf(")");
+ fprintf(stream, ")");
}
-void print_expr_as_sexpr(struct Expr expr)
+void print_expr_as_sexpr(FILE *stream, struct Expr expr)
{
- /* TODO(#296): print_expr_as_sexpr doesn't support lists */
switch (expr.type) {
case EXPR_ATOM:
- print_atom_as_sexpr(expr.atom);
+ print_atom_as_sexpr(stream, expr.atom);
break;
case EXPR_CONS:
- print_cons_as_sexpr(expr.cons);
+ print_cons_as_sexpr(stream, expr.cons);
break;
case EXPR_VOID:
free(cons);
}
-struct Atom *create_number_atom(Gc *gc, float num)
+struct Atom *create_number_atom(Gc *gc, long int num)
{
struct Atom *atom = malloc(sizeof(struct Atom));
if (atom == NULL) {
return NULL;
}
+struct Atom *create_native_atom(Gc *gc, NativeFunction fun, void *param)
+{
+ struct Atom *atom = malloc(sizeof(struct Atom));
+
+ if (atom == NULL) {
+ goto error;
+ }
+
+ atom->type = ATOM_NATIVE;
+ atom->native.fun = fun;
+ atom->native.param = param;
+
+ if (gc_add_expr(gc, atom_as_expr(atom)) < 0) {
+ goto error;
+ }
+
+ return atom;
+
+error:
+ if (atom != NULL) {
+ free(atom);
+ }
+
+ return NULL;
+}
+
void destroy_atom(struct Atom *atom)
{
switch (atom->type) {
free(atom->str);
} break;
+ case ATOM_NATIVE:
case ATOM_NUMBER: {
/* Nothing */
} break;
free(atom);
}
+
+static int atom_as_sexpr(struct Atom *atom, char *output, size_t n)
+{
+ assert(atom);
+ assert(output);
+
+ switch (atom->type) {
+ case ATOM_SYMBOL:
+ return snprintf(output, n, "%s", atom->sym);
+
+ case ATOM_NUMBER:
+ return snprintf(output, n, "%ld", atom->num);
+
+ case ATOM_STRING:
+ return snprintf(output, n, "\"%s\"", atom->str);
+
+ case ATOM_NATIVE:
+ return snprintf(output, n, "<native>");
+ }
+
+ return 0;
+}
+
+static int cons_as_sexpr(struct Cons *head, char *output, size_t n)
+{
+ assert(head);
+ assert(output);
+
+ /* TODO(#378): cons_as_sexpr does not handle encoding errors of snprintf */
+
+ struct Cons *cons = head;
+
+ int m = (int) n;
+
+ int c = snprintf(output, n, "(");
+ if (m - c <= c) {
+ return c;
+ }
+
+ c += expr_as_sexpr(cons->car, output + c, (size_t) (m - c));
+ if (m - c <= 0) {
+ return c;
+ }
+
+ while (cons->cdr.type == EXPR_CONS) {
+ cons = cons->cdr.cons;
+
+ c += snprintf(output + c, (size_t) (m - c), " ");
+ if (m - c <= 0) {
+ return c;
+ }
+
+ c += expr_as_sexpr(cons->car, output + c, (size_t) (m - c));
+ if (m - c <= 0) {
+ return c;
+ }
+ }
+
+ if (cons->cdr.atom->type != ATOM_SYMBOL ||
+ strcmp("nil", cons->cdr.atom->sym) != 0) {
+
+ c += snprintf(output + c, (size_t) (m - c), " . ");
+ if (m - c <= 0) {
+ return c;
+ }
+
+ c += expr_as_sexpr(cons->cdr, output + c, (size_t) (m - c));
+ if (m - c <= 0) {
+ return c;
+ }
+ }
+
+ c += snprintf(output + c, (size_t) (m - c), ")");
+ if (m - c <= 0) {
+ return c;
+ }
+
+ return c;
+}
+
+int expr_as_sexpr(struct Expr expr, char *output, size_t n)
+{
+ switch(expr.type) {
+ case EXPR_ATOM:
+ return atom_as_sexpr(expr.atom, output, n);
+
+ case EXPR_CONS:
+ return cons_as_sexpr(expr.cons, output, n);
+
+ case EXPR_VOID:
+ return 0;
+ }
+
+ return 0;
+}
projects / nothing.git / blobdiff