5945 - branches

1 files changed, 285 insertions, 0 deletions

diff --git a/apps/mu.subx b/apps/mu.subx
index 0022405c..5b883b99 100644
--- a/apps/mu.subx
+++ b/apps/mu.subx
@@ -1224,6 +1224,81 @@ test-convert-function-with-local-var-in-named-block:
     5d/pop-to-ebp
     c3/return
 
+test-convert-function-with-branches-in-block:
+    # empty function decl => function prologue and epilogue
+    #   fn foo x: int {
+    #     {
+    #       break-if->=
+    #       loop-if-addr<
+    #       increment x
+    #       loop
+    #     }
+    #   }
+    # =>
+    #   foo:
+    #     # . prologue
+    #     55/push-ebp
+    #     89/<- %ebp 4/r32/esp
+    #     {
+    #       {
+    #         0f 8d/jump-if->= break/disp32
+    #         0f 82/jump-if-addr< loop/disp32
+    #         ff 0/subop/increment *(ebp+8)
+    #         e9/jump loop/disp32
+    #       }
+    #     }
+    #     # . epilogue
+    #     89/<- %esp 5/r32/ebp
+    #     5d/pop-to-ebp
+    #     c3/return
+    # . prologue
+    55/push-ebp
+    89/<- %ebp 4/r32/esp
+    # setup
+    (clear-stream _test-input-stream)
+    (clear-stream $_test-input-buffered-file->buffer)
+    (clear-stream _test-output-stream)
+    (clear-stream $_test-output-buffered-file->buffer)
+    #
+    (write _test-input-stream "fn foo x: int {\n")
+    (write _test-input-stream "  {\n")
+    (write _test-input-stream "    break-if->=\n")
+    (write _test-input-stream "    loop-if-addr<\n")
+    (write _test-input-stream "    increment x\n")
+    (write _test-input-stream "    loop\n")
+    (write _test-input-stream "  }\n")
+    (write _test-input-stream "}\n")
+    # convert
+    (convert-mu _test-input-buffered-file _test-output-buffered-file)
+    (flush _test-output-buffered-file)
+#?     # dump _test-output-stream {{{
+#?     (write 2 "^")
+#?     (write-stream 2 _test-output-stream)
+#?     (write 2 "$\n")
+#?     (rewind-stream _test-output-stream)
+#?     # }}}
+    # check output
+    (check-next-stream-line-equal _test-output-stream "foo:"                  "F - test-convert-function-with-branches-in-block/0")
+    (check-next-stream-line-equal _test-output-stream "# . prologue"          "F - test-convert-function-with-branches-in-block/1")
+    (check-next-stream-line-equal _test-output-stream "55/push-ebp"           "F - test-convert-function-with-branches-in-block/2")
+    (check-next-stream-line-equal _test-output-stream "89/<- %ebp 4/r32/esp"  "F - test-convert-function-with-branches-in-block/3")
+    (check-next-stream-line-equal _test-output-stream "{"                     "F - test-convert-function-with-branches-in-block/4")
+    (check-next-stream-line-equal _test-output-stream "{"                     "F - test-convert-function-with-branches-in-block/5")
+    (check-next-stream-line-equal _test-output-stream "0f 8d/jump-if->= break/disp32"  "F - test-convert-function-with-branches-in-block/6")
+    (check-next-stream-line-equal _test-output-stream "0f 82/jump-if-addr< loop/disp32"  "F - test-convert-function-with-branches-in-block/7")
+    (check-next-stream-line-equal _test-output-stream "ff 0/subop/increment *(ebp+0x00000008)"  "F - test-convert-function-with-branches-in-block/8")
+    (check-next-stream-line-equal _test-output-stream "e9/jump loop/disp32"   "F - test-convert-function-with-branches-in-block/9")
+    (check-next-stream-line-equal _test-output-stream "}"                     "F - test-convert-function-with-branches-in-block/10")
+    (check-next-stream-line-equal _test-output-stream "}"                     "F - test-convert-function-with-branches-in-block/12")
+    (check-next-stream-line-equal _test-output-stream "# . epilogue"          "F - test-convert-function-with-branches-in-block/13")
+    (check-next-stream-line-equal _test-output-stream "89/<- %esp 5/r32/ebp"  "F - test-convert-function-with-branches-in-block/14")
+    (check-next-stream-line-equal _test-output-stream "5d/pop-to-ebp"         "F - test-convert-function-with-branches-in-block/15")
+    (check-next-stream-line-equal _test-output-stream "c3/return"             "F - test-convert-function-with-branches-in-block/16")
+    # . epilogue
+    89/<- %esp 5/r32/ebp
+    5d/pop-to-ebp
+    c3/return
+
 #######################################################
 # Parsing
 #######################################################
@@ -4974,6 +5049,216 @@ _Primitive-copy-lit-to-mem:
     0/imm32/no-r32
     2/imm32/imm32-is-first-inout
     1/imm32/output-is-write-only
+    _Primitive-break-if-addr</imm32/next
+_Primitive-break-if-addr<:
+    "break-if-addr<"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 82/jump-if-addr< break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-addr>=/imm32/next
+_Primitive-break-if-addr>=:
+    "break-if-addr>="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 83/jump-if-addr>= break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-=/imm32/next
+_Primitive-break-if-=:
+    "break-if-="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 84/jump-if-= break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-!=/imm32/next
+_Primitive-break-if-!=:
+    "break-if-!="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 85/jump-if-!= break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-addr<=/imm32/next
+_Primitive-break-if-addr<=:
+    "break-if-addr<="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 86/jump-if-addr<= break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-addr>/imm32/next
+_Primitive-break-if-addr>:
+    "break-if-addr>"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 87/jump-if-addr> break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-</imm32/next
+_Primitive-break-if-<:
+    "break-if-<"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8c/jump-if-< break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if->=/imm32/next
+_Primitive-break-if->=:
+    "break-if->="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8d/jump-if->= break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if-<=/imm32/next
+_Primitive-break-if-<=:
+    "break-if-<="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8e/jump-if-<= break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-break-if->/imm32/next
+_Primitive-break-if->:
+    "break-if->"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8f/jump-if-> break/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-addr</imm32/next
+_Primitive-loop-if-addr<:
+    "loop-if-addr<"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 82/jump-if-addr< loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-addr>=/imm32/next
+_Primitive-loop-if-addr>=:
+    "loop-if-addr>="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 83/jump-if-addr>= loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-=/imm32/next
+_Primitive-loop-if-=:
+    "loop-if-="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 84/jump-if-= loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-!=/imm32/next
+_Primitive-loop-if-!=:
+    "loop-if-!="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 85/jump-if-!= loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-addr<=/imm32/next
+_Primitive-loop-if-addr<=:
+    "loop-if-addr<="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 86/jump-if-addr<= loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-addr>/imm32/next
+_Primitive-loop-if-addr>:
+    "loop-if-addr>"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 87/jump-if-addr> loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-</imm32/next
+_Primitive-loop-if-<:
+    "loop-if-<"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8c/jump-if-< loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if->=/imm32/next
+_Primitive-loop-if->=:
+    "loop-if->="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8d/jump-if->= loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if-<=/imm32/next
+_Primitive-loop-if-<=:
+    "loop-if-<="/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8e/jump-if-<= loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop-if->/imm32/next
+_Primitive-loop-if->:
+    "loop-if->"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "0f 8f/jump-if-> loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
+    _Primitive-loop/imm32/next  # we probably don't need an unconditional break
+_Primitive-loop:
+    "loop"/imm32/name
+    0/imm32/inouts
+    0/imm32/outputs
+    "e9/jump loop/disp32"/imm32/subx-name
+    0/imm32/no-rm32
+    0/imm32/no-r32
+    0/imm32/no-imm32
+    0/imm32/no-output
     0/imm32/next
 
 Single-int-var-on-stack: