Fixed bug in sexp arity

This commit is contained in:
Dmitry Boulytchev 2018-11-05 20:17:11 +03:00 committed by danyabeerzun
parent dc38319a13
commit 99fdd176f4
6 changed files with 42 additions and 31 deletions

View file

@ -4,3 +4,4 @@
1 1
2 2
3 3
5

View file

@ -14,8 +14,17 @@ case 3 of
esac; esac;
case `a (1, 2, 3) of case `a (1, 2, 3) of
`b -> write (1) `a -> write (1)
| a@`a (_, _, _) -> case a of | a@`a (_, _, _) -> case a of
`a (x, y, z) -> write (x); write (y); write (z) `a (x, y, z) -> write (x); write (y); write (z)
esac esac
esac;
case `a (1, 2, 3, 4, 5) of
`a -> write (0)
| `a (_) -> write (1)
| `a (_, _) -> write (2)
| `a (_, _, _) -> write (3)
| `a (_, _, _, _) -> write (4)
| `a (_, _, _, _, _) -> write (5)
esac esac

View file

@ -218,9 +218,9 @@ extern void* Bsexp (int n, ...) {
return d->contents; return d->contents;
} }
extern int Btag (void *d, int t) { extern int Btag (void *d, int t, int n) {
data *r = TO_DATA(d); data *r = TO_DATA(d);
return BOX(TAG(r->tag) == SEXP_TAG && TO_SEXP(d)->tag == t); return BOX(TAG(r->tag) == SEXP_TAG && TO_SEXP(d)->tag == t && LEN(r->tag) == n);
} }
extern void Bsta (int n, int v, void *s, ...) { extern void Bsta (int n, int v, void *s, ...) {

View file

@ -130,7 +130,7 @@ module Builtin =
| Value.Sexp (_, a) -> List.nth a i | Value.Sexp (_, a) -> List.nth a i
) )
) )
| ".length" -> (st, i, o, Some (Value.of_int (match List.hd args with Value.Array a -> List.length a | Value.String s -> String.length s))) | ".length" -> (st, i, o, Some (Value.of_int (match List.hd args with Value.Sexp (_, a) | Value.Array a -> List.length a | Value.String s -> String.length s)))
| ".array" -> (st, i, o, Some (Value.of_array args)) | ".array" -> (st, i, o, Some (Value.of_array args))
| ".stringval" -> let [a] = args in (st, i, o, Some (Value.of_string @@ Value.string_val a)) | ".stringval" -> let [a] = args in (st, i, o, Some (Value.of_string @@ Value.string_val a))
| "isArray" -> let [a] = args in (st, i, o, Some (Value.of_int @@ match a with Value.Array _ -> 1 | _ -> 0)) | "isArray" -> let [a] = args in (st, i, o, Some (Value.of_int @@ match a with Value.Array _ -> 1 | _ -> 0))

View file

@ -20,7 +20,7 @@ open Language
(* drops the top element off *) | DROP (* drops the top element off *) | DROP
(* duplicates the top element *) | DUP (* duplicates the top element *) | DUP
(* swaps two top elements *) | SWAP (* swaps two top elements *) | SWAP
(* checks the tag of S-expression *) | TAG of string (* checks the tag and arity of S-expression *) | TAG of string * int
(* enters a scope *) | ENTER of string list (* enters a scope *) | ENTER of string list
(* leaves a scope *) | LEAVE (* leaves a scope *) | LEAVE
with show with show
@ -79,8 +79,8 @@ let rec eval env ((cstack, stack, ((st, i, o) as c)) as conf) = function
| DUP -> eval env (cstack, List.hd stack :: stack, c) prg' | DUP -> eval env (cstack, List.hd stack :: stack, c) prg'
| SWAP -> let x::y::stack' = stack in | SWAP -> let x::y::stack' = stack in
eval env (cstack, y::x::stack', c) prg' eval env (cstack, y::x::stack', c) prg'
| TAG t -> let x::stack' = stack in | TAG (t, n) -> let x::stack' = stack in
eval env (cstack, (Value.of_int @@ match x with Value.Sexp (t', _) when t' = t -> 1 | _ -> 0) :: stack', c) prg' eval env (cstack, (Value.of_int @@ match x with Value.Sexp (t', a) when t' = t && List.length a = n -> 1 | _ -> 0) :: stack', c) prg'
| ENTER xs -> let vs, stack' = split (List.length xs) stack in | ENTER xs -> let vs, stack' = split (List.length xs) stack in
eval env (cstack, stack', (State.push st (List.fold_left (fun s (x, v) -> State.bind x v s) State.undefined (List.combine xs vs)) xs, i, o)) prg' eval env (cstack, stack', (State.push st (List.fold_left (fun s (x, v) -> State.bind x v s) State.undefined (List.combine xs vs)) xs, i, o)) prg'
@ -140,7 +140,7 @@ let compile (defs, p) =
| Stmt.Pattern.Sexp (t, ps) -> | Stmt.Pattern.Sexp (t, ps) ->
let ltag , env = env#get_label in let ltag , env = env#get_label in
let ldrop, env = env#get_label in let ldrop, env = env#get_label in
let tag = [DUP; TAG t; CJMP ("nz", ltag); LABEL ldrop; DROP; JMP lfalse; LABEL ltag] in let tag = [DUP; TAG (t, List.length ps); CJMP ("nz", ltag); LABEL ldrop; DROP; JMP lfalse; LABEL ltag] in
let _, env, code = let _, env, code =
List.fold_left List.fold_left
(fun (i, env, code) p -> (fun (i, env, code) p ->

View file

@ -322,10 +322,11 @@ let compile env code =
let x, y = env#peek2 in let x, y = env#peek2 in
env, [Push x; Push y; Pop x; Pop y] env, [Push x; Push y; Pop x; Pop y]
| TAG t -> | TAG (t, n) ->
let s, env = env#allocate in let s1, env = env#allocate in
let env, code = call env ".tag" 2 false in let s2, env = env#allocate in
env, [Mov (L env#hash t, s)] @ code let env, code = call env ".tag" 3 false in
env, [Mov (L env#hash t, s1); Mov (L n, s2)] @ code
| ENTER xs -> | ENTER xs ->
let env, code = let env, code =