Chez inc 3l: transient collection RT shims

transient/persistent!/conj!/assoc!/dissoc!/disj!/pop! as copy-on-write over the
persistent collections (host/chez/transients.ss) — each op rebuilds the persistent
coll (no in-place perf) but the semantics match, so into/frequencies/group-by work.
Adds persistent disj over pset-disj. get/count/contains? are redefined to see
through a transient (frequencies and group-by both do (get tm k) on a transient
map); vector? on a transient vector is false, which group-by relies on.

Prelude parity 1326 -> 1382/2497, 0 new divergences. emit-test exercises the direct
transient ops via run-prelude and the overlay users (frequencies/group-by/into)
end-to-end through the bin/jolt-chez -e binary.

host/chez/rt.ss

@@ -149,3 +149,8 @@
 ;; double/gensym — host-coupled seed natives def-var!'d into clojure.core. Loaded
 ;; LAST because `str` reuses jolt-pr-str (defined just above).
 (load "host/chez/converters.ss")
+
+;; transients (jolt-kl2l): copy-on-write transient collections + persistent disj;
+;; extends get/count/contains? to see through a transient. After collections.ss
+;; (the persistent ops it delegates to).
+(load "host/chez/transients.ss")

host/chez/transients.ss

@@ -0,0 +1,73 @@
+;; transients (jolt-kl2l) — copy-on-write over the persistent collections.
+;;
+;; A first cut: each transient op rebuilds the persistent collection (no in-place
+;; mutation perf), but the SEMANTICS match — the overlay's transient users (into,
+;; frequencies, group-by) get correct results. get/count/contains? are extended to
+;; see THROUGH a transient to its held collection (frequencies/group-by do
+;; (get tm k) on a transient map). vector? on a transient vector is false (it's a
+;; jolt-transient record, not a pvec), which group-by relies on. Loaded after
+;; collections.ss (the persistent ops it delegates to) and converters.ss.
+
+(define-record-type jolt-transient (fields (mutable coll) (mutable active))
+  (nongenerative jolt-transient-v1))
+
+(define (jolt-transient-new coll) (make-jolt-transient coll #t))
+
+(define (jolt-trans-check t who)
+  (unless (jolt-transient? t) (error #f (string-append who ": not a transient") t))
+  (unless (jolt-transient-active t)
+    (error #f (string-append who ": transient used after persistent!"))))
+
+(define (jolt-persistent! t)
+  (jolt-trans-check t "persistent!")
+  (jolt-transient-active-set! t #f)
+  (jolt-transient-coll t))
+
+(define (jolt-conj! t . xs)
+  (jolt-trans-check t "conj!")
+  (jolt-transient-coll-set! t (apply jolt-conj (jolt-transient-coll t) xs))
+  t)
+(define (jolt-assoc! t k v)
+  (jolt-trans-check t "assoc!")
+  (jolt-transient-coll-set! t (jolt-assoc (jolt-transient-coll t) k v))
+  t)
+(define (jolt-dissoc! t . ks)
+  (jolt-trans-check t "dissoc!")
+  (jolt-transient-coll-set! t (apply jolt-dissoc (jolt-transient-coll t) ks))
+  t)
+(define (jolt-disj! t . xs)
+  (jolt-trans-check t "disj!")
+  (jolt-transient-coll-set! t (apply jolt-disj (jolt-transient-coll t) xs))
+  t)
+(define (jolt-pop! t)
+  (jolt-trans-check t "pop!")
+  (jolt-transient-coll-set! t (jolt-pop (jolt-transient-coll t)))
+  t)
+
+;; persistent disj over sets (pset-disj already exists in collections.ss).
+(define (jolt-disj s . xs)
+  (let loop ((s s) (xs xs)) (if (null? xs) s (loop (pset-disj s (car xs)) (cdr xs)))))
+
+;; get / count / contains? see through a transient. Redefine the native procedures
+;; (captured first) so the existing emit lowerings (jolt-get/jolt-count/
+;; jolt-contains?) unwrap a transient before delegating; non-transients are
+;; untouched.
+(define (jolt-deref-transient x) (if (jolt-transient? x) (jolt-transient-coll x) x))
+(define %prev-jolt-get jolt-get)
+(set! jolt-get
+  (case-lambda
+    ((coll k) (%prev-jolt-get (jolt-deref-transient coll) k))
+    ((coll k d) (%prev-jolt-get (jolt-deref-transient coll) k d))))
+(define %prev-jolt-count jolt-count)
+(set! jolt-count (lambda (coll) (%prev-jolt-count (jolt-deref-transient coll))))
+(define %prev-jolt-contains? jolt-contains?)
+(set! jolt-contains? (lambda (coll k) (%prev-jolt-contains? (jolt-deref-transient coll) k)))
+
+(def-var! "clojure.core" "transient" jolt-transient-new)
+(def-var! "clojure.core" "persistent!" jolt-persistent!)
+(def-var! "clojure.core" "conj!" jolt-conj!)
+(def-var! "clojure.core" "assoc!" jolt-assoc!)
+(def-var! "clojure.core" "dissoc!" jolt-dissoc!)
+(def-var! "clojure.core" "disj!" jolt-disj!)
+(def-var! "clojure.core" "pop!" jolt-pop!)
+(def-var! "clojure.core" "disj" jolt-disj)