X-Git-Url: https://thelambdalab.xyz/gitweb/index.cgi?p=scheme.forth.jl.git;a=blobdiff_plain;f=src%2Fscheme.4th;fp=src%2Fscheme.4th;h=a360b475064434854449066e13fd6ac9b31e4922;hp=2edd16da5fc43309c34387e9c4eb10afa0770ed3;hb=d2810af4877d1f166459c6778a00c4bc22292d1f;hpb=c6aa8bc94536cbac6a7df5ad6872131665aeefeb

diff --git a/src/scheme.4th b/src/scheme.4th
index 2edd16d..a360b47 100644
--- a/src/scheme.4th
+++ b/src/scheme.4th
@@ -564,8 +564,7 @@ global-env obj!
 : continuation->rstack-list
   drop pair-type cdr ;
 
-: restore-param-stack ( continuation -- obj_stack continuation )
-
+: restore-param-stack ( continuation -- obj_stack )
   continuation->pstack-list
   2dup >R >R
 
@@ -587,12 +586,40 @@ global-env obj!
   -2 +loop
 
   2drop
+;
+
+: restore-return-stack ( continuation -- )
+    R> \ store top of return stack on PS
+    continuation->rstack-list
+    2dup >R >R
 
+    ( Allocate stack space first using rsp!,
+      then copy objects from list. )
+
+    car drop
+    rsp0 + rsp!
+
+    R> R> 2dup cdr
+    2swap
+    car drop 0 swap do
+        2dup car drop
+        rsp0 i + 1 + !
+        cdr
+    1- +loop
+
+    2drop
+    trace
+    >R \ restore original top of return stack
 ;
 
-: restore-continuation
+: restore-continuation ( continuation -- )
   \ TODO: replace current parameter and return stacks with
   \ contents of continuation object.
+
+    2dup >R >R
+    restore-param-stack
+    R> R>
+    restore-return-stack
 ;
 
 \ }}}