1 files changed, 16 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))