From 30e9e828777d1cd3f8a4a4d5873ec73a01101f7a Mon Sep 17 00:00:00 2001
From: dyoo <dyoo@kfisler-ra1.(none)>
Date: Thu, 3 Mar 2011 15:26:30 -0500
Subject: [PATCH] I think the output looks more correct.

---
 compile.rkt | 29 +++++++++++++++++------------
 1 file changed, 17 insertions(+), 12 deletions(-)

diff --git a/compile.rkt b/compile.rkt
index b1745a0..89c0d13 100644
--- a/compile.rkt
+++ b/compile.rkt
@@ -300,11 +300,13 @@
                  ;; The last operand needs to be handled specially: it currently lives in 
                  ;; val.  We move the procedure at env[n] over to proc, and move the
                  ;; last operand at 'val into env[n].
-                 (make-instruction-sequence 
-                  `(,(make-AssignImmediateStatement 'proc 
-                                                    (make-EnvLexicalReference n))
-                    ,(make-AssignImmediateStatement (make-EnvOffset n)
-                                                    (make-Reg 'val))))]
+                 (append-instruction-sequences 
+                  (car ops)
+                  (make-instruction-sequence 
+                   `(,(make-AssignImmediateStatement 'proc 
+                                                     (make-EnvLexicalReference n))
+                     ,(make-AssignImmediateStatement (make-EnvOffset n)
+                                                     (make-Reg 'val)))))]
                 [else
                  ;; Otherwise, add instructions to juggle the operator and operands in the stack.
                  (append-instruction-sequences (car ops)
@@ -328,15 +330,18 @@
                                         primitive-branch)))
        
        compiled-branch
-       (compile-proc-appl cenv n target compiled-linkage)
+       (end-with-application-linkage 
+        compiled-linkage
+        cenv
+        (compile-proc-appl cenv n target compiled-linkage))
 
        primitive-branch
-       (end-with-application-linkage linkage
-                                     cenv
-                                     (make-instruction-sequence 
-                                      `(,(make-AssignPrimOpStatement 
-                                          target
-                                          (make-ApplyPrimitiveProcedure n)))))
+       (end-with-linkage linkage
+                         cenv
+                         (make-instruction-sequence 
+                          `(,(make-AssignPrimOpStatement 
+                              target
+                              (make-ApplyPrimitiveProcedure n)))))
        after-call))))