serotonin

This is a small program that takes a small subset of a Python like language and converts it to a normal form.

Example 1

So far, as of 10/4/23, you can parse programs like this:

pyparser.rkt> (define p3 (open-input-string "x = 4 y = 5 z = 4 if True then z + x else z+y"))
pyparser.rkt> (the-parser (lambda () (the-lexer/tokens p3)))
(py-module
 (list
  (py-assign (py-id 'x) (py-num 4))
  (py-assign (py-id 'y) (py-num 5))
  (py-assign (py-id 'z) (py-num 4))
  (py-if-exp (py-bool 'True) (py-plus (py-id 'z) (py-id 'x)) (py-plus (py-id 'z) (py-id 'y)))))

Example 2

toANormal.rkt> (define p1 (open-input-string "x = 10 + -50 y = 4 + -60 z = 4 + 5 k = 5 + -3"))
toANormal.rkt> (define ast (the-parser (lambda () (the-lexer/tokens p1))))
toANormal.rkt> ast
(py-module
 (list
  (py-assign (py-id 'x) (py-plus (py-num 10) (py-neg 50)))
  (py-assign (py-id 'y) (py-plus (py-num 4) (py-neg 60)))
  (py-assign (py-id 'z) (py-plus (py-num 4) (py-num 5)))
  (py-assign (py-id 'k) (py-plus (py-num 5) (py-neg 3)))))
toANormal.rkt> (program-to-a-normal-form ast)
(list
 (atomic-assignment (atomic (py-id "temp_0")) (py-neg 50))
 (atomic-assignment (atomic (py-id 'x)) (atomic-plus (atomic (py-id "temp_0")) (atomic (py-num 10))))
 (atomic-assignment (atomic (py-id "temp_1")) (py-neg 60))
 (atomic-assignment (atomic (py-id 'y)) (atomic-plus (atomic (py-id "temp_1")) (atomic (py-num 4))))
 (atomic-assignment (atomic (py-id 'z)) (atomic-plus (atomic (py-num 4)) (atomic (py-num 5))))
 (atomic-assignment (atomic (py-id "temp_2")) (py-neg 3))
 (atomic-assignment (atomic (py-id 'k)) (atomic-plus (atomic (py-id "temp_2")) (atomic (py-num 5)))))
toANormal.rkt>

Example 3

to-anf.rkt> (define p (open-input-string "if if if if True then True else False then True else False then True else False then 1 else 3"))
to-anf.rkt> (define ast (the-parser (lambda () (the-lexer/tokens p))))
to-anf.rkt> ast
(py-module
 (list
  (py-if-exp
   (py-if-exp
    (py-if-exp
     (py-if-exp (py-bool 'True) (py-bool 'True) (py-bool 'False))
     (py-bool 'True)
     (py-bool 'False))
    (py-bool 'True)
    (py-bool 'False))
   (py-num 1)
   (py-num 3))))
to-anf.rkt> (define anf (syntax-ast-to-anf ast))
to-anf.rkt> anf 
(list
 (atomic-assignment
  (atomic (py-id "temp_0"))
  (list
   (atomic-assignment (atomic (py-id "temp_2")) (atomic (py-bool 'True)))
   (atomic-assignment
    (atomic (py-id "temp-3"))
    (anf-if-exp
     (atomic (py-id "temp_2"))
     (atomic (py-bool 'True))
     (atomic (py-bool 'False))))
   (anf-if-exp
    (atomic (py-id "temp-3"))
    (atomic (py-bool 'True))
    (atomic (py-bool 'False)))))
 (atomic-assignment
  (atomic (py-id "temp-1"))
  (anf-if-exp
   (atomic (py-id "temp_0"))
   (atomic (py-bool 'True))
   (atomic (py-bool 'False))))
 (anf-if-exp (atomic (py-id "temp-1")) (atomic (py-num 1)) (atomic (py-num 3))))
to-anf.rkt> (restructure-anf-ast anf)
(list
 (list
  (atomic-assignment (atomic (py-id "temp_2")) (atomic (py-bool 'True)))
  (atomic-assignment
   (atomic (py-id "temp-3"))
   (anf-if-exp
    (atomic (py-id "temp_2"))
    (atomic (py-bool 'True))
    (atomic (py-bool 'False))))
  (atomic-assignment
   (atomic (py-id "temp_0"))
   (anf-if-exp
    (atomic (py-id "temp-3"))
    (atomic (py-bool 'True))
    (atomic (py-bool 'False)))))
 (atomic-assignment
  (atomic (py-id "temp-1"))
  (anf-if-exp
   (atomic (py-id "temp_0"))
   (atomic (py-bool 'True))
   (atomic (py-bool 'False))))
 (anf-if-exp (atomic (py-id "temp-1")) (atomic (py-num 1)) (atomic (py-num 3))))