use a scheme procedures+eval to manage and transform AST

Removes records for a more-managable scheme-syntax approach. Modules+overriding allows for the IR itself to be represented and evaluated as scheme code during each transformation.

4 files changed, 66 insertions, 0 deletions

diff --git a/src/modules/generator/allocate.scm b/src/modules/generator/allocate.scm
new file mode 100644
index 0000000..193cadb
--- /dev/null
+++ b/src/modules/generator/allocate.scm
@@ -0,0 +1,16 @@
+(define-module (modules generator allocate)
+  #:use-module (modules utils assign-stack)
+  #:export (expansion->allocate))
+
+
+(define (expansion->allocate ast)
+  (define (allocate n)
+    (cond
+     ((null? n) n)
+     ((eq? (car n) 'tmp)
+      (list 'stack (make-location (cadr n))))
+     ((list? (car n))
+      (cons (allocate (car n)) (allocate (cdr n))))
+     (#t
+      (cons (car n) (allocate (cdr n))))))
+  (allocate ast))
diff --git a/src/modules/generator/assembly.scm b/src/modules/generator/assembly.scm
new file mode 100644
index 0000000..50e158e
--- /dev/null
+++ b/src/modules/generator/assembly.scm
@@ -0,0 +1,18 @@
+(define-module (modules generator assembly)
+  #:use-module (modules ast ir)
+  #:use-module (modules utils assign-stack)
+  #:use-module (modules utils merge-instructions)
+  #:export (instrs
+            mov))
+
+
+(define (instrs lst)
+  (cons 'list (cons (list 'alloc (abs (get-frame-size)))
+                    (merge-instr lst))))
+
+(define (mov dst src)
+  (if (and (eq? 'stack (car dst))
+           (eq? 'stack (car src)))
+      (append (list (list 'mov (reg "r10") src))
+              (list (list 'mov dst (reg "r10"))))
+      (list 'mov dst src)))
diff --git a/src/modules/generator/driver.scm b/src/modules/generator/driver.scm
new file mode 100644
index 0000000..dc95268
--- /dev/null
+++ b/src/modules/generator/driver.scm
@@ -0,0 +1,9 @@
+(define-module (modules generator driver)
+  #:use-module (modules generator allocate)
+  #:export (tacky->assembly))
+
+
+(define (tacky->assembly n)
+  (eval (expansion->allocate
+         (eval n (resolve-module '(modules generator expansion))))
+        (resolve-module '(modules generator assembly))))
diff --git a/src/modules/generator/expansion.scm b/src/modules/generator/expansion.scm
new file mode 100644
index 0000000..3beb6db
--- /dev/null
+++ b/src/modules/generator/expansion.scm
@@ -0,0 +1,23 @@
+(define-module (modules generator expansion)
+  #:use-module (modules ast ir)
+  #:use-module (modules utils merge-instructions)
+  #:export (instrs
+            not
+            neg
+            ret))
+
+
+(define (instrs lst)
+  #f
+  (list 'instrs (cons 'list (merge-instr lst))))
+
+(define (not dst src) (unary 'not dst src))
+(define (neg dst src) (unary 'neg dst src))
+
+(define (ret src)
+  (list (list 'mov (list 'reg "eax") src)
+        (list 'ret)))
+
+(define (unary op dst src)
+  (list (list 'mov dst src)
+        (list op dst)))