dotimes: support result form

Sigrid Solveig Haflínudóttir committed 4 months ago

Commit 9c44c721 parent 15a83af2 files · +10 −6Browse files

Showing the changes to src (1 of 2 files). Show all

Unified · Split

      (λ ,formals ,@body)))
 (defmacro (dotimes var . body)
-  (let ((v (car var))
+  (let* {[v   (car var)]
-        (cnt (cadr var)))
+         [cnt (aref var 1)]
-    `(for 0 ,cnt
+         [nv  (length var)]
-          (λ (,v) ,@body))))
+         [res (if (= nv 3) (aref var 2) #.(void))]}
+    (unless (<= nv 3)
+      (compile-error "invalid var expression: " var))
+    `(begin (for 0 ,cnt (λ (,v) ,@body))
+            ,res)))
 (defmacro (with-bindings binds . body)
   (let ((vars (map car binds))

Keyboard shortcuts