Return stack restore working.
[scheme.forth.jl.git] / src / scheme.4th
index 135bdab..0dec215 100644 (file)
@@ -564,6 +564,10 @@ global-env obj!
 : continuation->rstack-list
   drop pair-type cdr ;
 
+: stack-list-len ( stack-list -- n )
+    car drop
+;
+
 : restore-param-stack ( continuation -- obj_stack )
   continuation->pstack-list
   2dup >R >R
@@ -588,15 +592,35 @@ global-env obj!
   2drop
 ;
 
+: list->pad ( list n -- )
+
+    pad + 1- \ final dest addr
+    pad      \ initial dest addr
+    swap
+    do
+        2dup cdr 2swap car
+        drop i !
+    -1 +loop
+
+    2drop
+;
+
 : restore-return-stack ( continuation -- )
 
-    R> -rot \ store top of return stack on PS
     continuation->rstack-list
-    2dup print 2dup
 
-    \ TODO: Implement body of return stack restoration
+    2dup stack-list-len -rot ( n stack-list )
+    2dup cdr 2swap stack-list-len  ( n list n )
+
+    list->pad  ( n )
 
-    >R \ restore original top of return stack
+    dup RSP0 + RSP! \ expand return stack to accommodate entries
+
+    ( n )
+    0   \ initial offset
+    do
+        pad i + @ RSP0 i 1+ + !
+    loop
 ;
 
 : restore-continuation ( continuation -- )
@@ -604,12 +628,11 @@ global-env obj!
   \ contents of continuation object.
 
     2dup >R >R
+
     restore-param-stack
 
-    ." ====== PARAM STACK RESTORED ======" cr
-    trace
-    
     R> R>
+
     restore-return-stack
 ;