spec: sets (make-set/set-add!/set-member?/union/intersection/etc)

Adds 13 set primitives to stdlib.sets. OCaml: SxSet as (string,value)
Hashtbl keyed by inspect(val); JS: SxSet wrapping Map keyed by
write-to-string. Structural equality — (make-set '(1 2)) contains 1.
Includes union, intersection, difference, for-each, map.
33 tests in test-sets.sx, all pass on both JS and OCaml.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-05-01 18:45:46 +00:00
parent 24d78464d8
commit 3b0ac67a10
6 changed files with 478 additions and 2 deletions

View File

@@ -1702,6 +1702,60 @@ PRIMITIVES_JS_MODULES: dict[str, str] = {
src.data.forEach(function(v, k) { dst.data.set(k, v); });
return null;
};
''',
"stdlib.sets": '''
// stdlib.sets — structural sets keyed by write-to-string serialization
function SxSet() { this.data = new Map(); this._sxset = true; }
SxSet.prototype._type = "set";
function sxSetKey(v) { return sxWriteVal(v, "write"); }
function sxSetSeed(s, lst) {
if (Array.isArray(lst)) lst.forEach(function(v) { s.data.set(sxSetKey(v), v); });
return s;
}
PRIMITIVES["make-set"] = function() {
var s = new SxSet();
if (arguments.length > 0 && Array.isArray(arguments[0])) sxSetSeed(s, arguments[0]);
return s;
};
PRIMITIVES["set?"] = function(v) { return v instanceof SxSet; };
PRIMITIVES["set-add!"] = function(s, v) { s.data.set(sxSetKey(v), v); return NIL; };
PRIMITIVES["set-member?"] = function(s, v) { return s.data.has(sxSetKey(v)); };
PRIMITIVES["set-remove!"] = function(s, v) { s.data.delete(sxSetKey(v)); return NIL; };
PRIMITIVES["set-size"] = function(s) { return s.data.size; };
PRIMITIVES["set->list"] = function(s) { return Array.from(s.data.values()); };
PRIMITIVES["list->set"] = function(lst) {
var s = new SxSet();
if (Array.isArray(lst)) lst.forEach(function(v) { s.data.set(sxSetKey(v), v); });
return s;
};
PRIMITIVES["set-union"] = function(a, b) {
var s = new SxSet();
a.data.forEach(function(v, k) { s.data.set(k, v); });
b.data.forEach(function(v, k) { s.data.set(k, v); });
return s;
};
PRIMITIVES["set-intersection"] = function(a, b) {
var s = new SxSet();
a.data.forEach(function(v, k) { if (b.data.has(k)) s.data.set(k, v); });
return s;
};
PRIMITIVES["set-difference"] = function(a, b) {
var s = new SxSet();
a.data.forEach(function(v, k) { if (!b.data.has(k)) s.data.set(k, v); });
return s;
};
PRIMITIVES["set-for-each"] = function(s, fn) {
s.data.forEach(function(v) { apply(fn, [v]); });
return NIL;
};
PRIMITIVES["set-map"] = function(s, fn) {
var out = new SxSet();
s.data.forEach(function(v) {
var r = apply(fn, [v]);
out.data.set(sxSetKey(r), r);
});
return out;
};
''',
}
# Modules to include by default (all)
@@ -1747,6 +1801,7 @@ PLATFORM_JS_PRE = '''
if (x._vector) return "vector";
if (x._string_buffer) return "string-buffer";
if (x._hash_table) return "hash-table";
if (x._sxset) return "set";
if (x._rational) return "rational";
if (typeof Node !== "undefined" && x instanceof Node) return "dom-node";
if (Array.isArray(x)) return "list";

View File

@@ -2687,4 +2687,87 @@ let () =
| _ -> raise (Eval_error "display-to-string: 1 arg"));
register "current-input-port" (fun _ -> Nil);
register "current-output-port" (fun _ -> Nil);
register "current-error-port" (fun _ -> Nil)
register "current-error-port" (fun _ -> Nil);
(* ---- Sets ---- *)
let set_key v = Sx_types.inspect v in
register "make-set" (fun args ->
let ht = Hashtbl.create 8 in
(match args with
| [] -> ()
| [List items] -> List.iter (fun v -> Hashtbl.replace ht (set_key v) v) items
| [ListRef r] -> List.iter (fun v -> Hashtbl.replace ht (set_key v) v) !r
| _ -> raise (Eval_error "make-set: expected optional list"));
SxSet ht);
register "set?" (fun args ->
match args with
| [SxSet _] -> Bool true
| [_] -> Bool false
| _ -> raise (Eval_error "set?: 1 arg"));
register "set-add!" (fun args ->
match args with
| [SxSet ht; v] -> Hashtbl.replace ht (set_key v) v; Nil
| _ -> raise (Eval_error "set-add!: expected set val"));
register "set-member?" (fun args ->
match args with
| [SxSet ht; v] -> Bool (Hashtbl.mem ht (set_key v))
| _ -> raise (Eval_error "set-member?: expected set val"));
register "set-remove!" (fun args ->
match args with
| [SxSet ht; v] -> Hashtbl.remove ht (set_key v); Nil
| _ -> raise (Eval_error "set-remove!: expected set val"));
register "set-size" (fun args ->
match args with
| [SxSet ht] -> Integer (Hashtbl.length ht)
| _ -> raise (Eval_error "set-size: expected set"));
register "set->list" (fun args ->
match args with
| [SxSet ht] -> List (Hashtbl.fold (fun _ v acc -> v :: acc) ht [])
| _ -> raise (Eval_error "set->list: expected set"));
register "list->set" (fun args ->
match args with
| [List items] ->
let ht = Hashtbl.create (List.length items) in
List.iter (fun v -> Hashtbl.replace ht (set_key v) v) items;
SxSet ht
| [ListRef r] ->
let ht = Hashtbl.create (List.length !r) in
List.iter (fun v -> Hashtbl.replace ht (set_key v) v) !r;
SxSet ht
| [Nil] -> SxSet (Hashtbl.create 0)
| _ -> raise (Eval_error "list->set: expected list"));
register "set-union" (fun args ->
match args with
| [SxSet a; SxSet b] ->
let ht = Hashtbl.copy a in
Hashtbl.iter (fun k v -> Hashtbl.replace ht k v) b;
SxSet ht
| _ -> raise (Eval_error "set-union: expected 2 sets"));
register "set-intersection" (fun args ->
match args with
| [SxSet a; SxSet b] ->
let ht = Hashtbl.create 8 in
Hashtbl.iter (fun k v -> if Hashtbl.mem b k then Hashtbl.replace ht k v) a;
SxSet ht
| _ -> raise (Eval_error "set-intersection: expected 2 sets"));
register "set-difference" (fun args ->
match args with
| [SxSet a; SxSet b] ->
let ht = Hashtbl.create 8 in
Hashtbl.iter (fun k v -> if not (Hashtbl.mem b k) then Hashtbl.replace ht k v) a;
SxSet ht
| _ -> raise (Eval_error "set-difference: expected 2 sets"));
register "set-for-each" (fun args ->
match args with
| [SxSet ht; fn] ->
Hashtbl.iter (fun _ v -> ignore (!Sx_types._cek_call_ref fn (List [v]))) ht;
Nil
| _ -> raise (Eval_error "set-for-each: expected set fn"));
register "set-map" (fun args ->
match args with
| [SxSet ht; fn] ->
let out = Hashtbl.create (Hashtbl.length ht) in
Hashtbl.iter (fun _ v ->
let r = !Sx_types._cek_call_ref fn (List [v]) in
Hashtbl.replace out (set_key r) r) ht;
SxSet out
| _ -> raise (Eval_error "set-map: expected set fn"))

View File

@@ -79,6 +79,7 @@ and value =
| Eof (** EOF sentinel — returned by read-char etc. at end of input. *)
| Port of sx_port (** String port — input (string cursor) or output (buffer). *)
| Rational of int * int (** Exact rational: numerator, denominator (reduced, denom>0). *)
| SxSet of (string, value) Hashtbl.t (** Mutable set keyed by inspect(value). *)
(** String input port: source string + mutable cursor position. *)
and sx_port_kind =
@@ -514,6 +515,7 @@ let type_of = function
| Port { sp_kind = PortInput _; _ } -> "input-port"
| Port { sp_kind = PortOutput _; _ } -> "output-port"
| Rational _ -> "rational"
| SxSet _ -> "set"
let is_nil = function Nil -> true | _ -> false
let is_lambda = function Lambda _ -> true | _ -> false
@@ -876,3 +878,4 @@ let rec inspect = function
| Port { sp_kind = PortOutput buf; sp_closed } ->
Printf.sprintf "<output-port:len=%d%s>" (Buffer.length buf) (if sp_closed then ":closed" else "")
| Rational (n, d) -> Printf.sprintf "%d/%d" n d
| SxSet ht -> Printf.sprintf "<set:%d>" (Hashtbl.length ht)