Modify assembly pass to expand new binary operators

2 files changed, 32 insertions, 13 deletions

diff --git a/src/backend/generator/driver.scm b/src/backend/generator/driver.scm
index 6625388..ac060f7 100644
--- a/src/backend/generator/driver.scm
+++ b/src/backend/generator/driver.scm
@@ -3,7 +3,10 @@
   #:export (tacky->assembly))
 
 
+;; (define (tacky->assembly n)
+;;   (eval (expansion->allocate
+;;          (eval n (resolve-module '(backend generator expansion))))
+;;         (resolve-module '(backend generator assembly))))
+
 (define (tacky->assembly n)
-  (eval (expansion->allocate
-         (eval n (resolve-module '(backend generator expansion))))
-        (resolve-module '(backend generator assembly))))
+  (eval n (resolve-module '(backend generator expansion))))
diff --git a/src/backend/generator/expansion.scm b/src/backend/generator/expansion.scm
index 1745e23..dd2962c 100644
--- a/src/backend/generator/expansion.scm
+++ b/src/backend/generator/expansion.scm
@@ -2,22 +2,38 @@
   #:use-module (backend ast ir)
   #:use-module (backend utils merge-instructions)
   #:export (instrs-ir
-            not-ir
-            neg-ir
-            ret-ir))
+            instr-ir))
 
 
+(define eax (reg-ir "eax"))
+(define edx (reg-ir "edx"))
+
 (define (instrs-ir lst)
-  #f
+  (display lst)
   (list 'instrs-ir (cons 'list (merge-instr lst))))
 
-(define (not-ir src dst) (unary 'not-ir src dst))
-(define (neg-ir src dst) (unary 'neg-ir src dst))
+(define (instr-ir op dst . srcs)
+  (case op
+    ((ret-ir) (handle-ret dst))
+    ((neg-ir not-ir) (handle-unary op (car srcs) dst))
+    ((plus-ir minus-ir mult-ir) (handle-binary op (car srcs) (cadr srcs) dst))
+    ((div-ir mod-ir) (handle-binary-q/r op (car srcs) (cadr srcs) dst))))
 
-(define (ret-ir src)
-  (list (list 'mov-ir src (list 'reg-ir "eax"))
-        (list 'ret-ir)))
+(define (handle-ret src)
+  (list (list 'instr-ir 'mov-ir src eax)
+        (list 'instr-ir 'ret-ir)))
 
-(define (unary op src dst)
+(define (handle-unary op src dst)
   (list (list 'mov-ir src dst)
         (list op dst)))
+
+(define (handle-binary op src1 src2 dst)
+  (list (list 'instr-ir 'mov-ir src1 dst)
+        (list 'instr-ir op src2 dst)))
+
+(define (handle-binary-q/r op src1 src2 dst)
+  (map (lambda (x) (cons 'instr-ir x))
+       (list (list 'mov-ir src1 eax)
+             (list 'cdq)
+             (list op src2)
+             (list 'mov-ir (if (eq? op div-ir) eax edx) dst))))