From bbb6356a4a4f853dc48a54f1668c6712f0ef659f Mon Sep 17 00:00:00 2001
From: Eduardo Julian
Date: Tue, 9 Jun 2020 23:57:50 -0400
Subject: Basic pattern-matching optimizations.

---
 lux-jvm/source/luxc/lang/translation/jvm/case.lux | 54 +++++++++++------------
 1 file changed, 26 insertions(+), 28 deletions(-)

(limited to 'lux-jvm/source/luxc/lang/translation/jvm/case.lux')

diff --git a/lux-jvm/source/luxc/lang/translation/jvm/case.lux b/lux-jvm/source/luxc/lang/translation/jvm/case.lux
index 421f413a0..e21cf9aec 100644
--- a/lux-jvm/source/luxc/lang/translation/jvm/case.lux
+++ b/lux-jvm/source/luxc/lang/translation/jvm/case.lux
@@ -57,22 +57,35 @@
        _.AALOAD
        (_.CHECKCAST runtime.$Stack)))
 
+(def: (leftsI value)
+  (-> Nat Inst)
+  (.case value
+    0 _.ICONST_0
+    1 _.ICONST_1
+    2 _.ICONST_2
+    3 _.ICONST_3
+    4 _.ICONST_4
+    5 _.ICONST_5
+    _ (_.int (.int value))))
+
 (def: (left-projection lefts)
   (-> Nat Inst)
-  (.let [accessI (.case lefts
-                   0
-                   _.AALOAD
-                   
-                   lefts
-                   (_.INVOKESTATIC //.$Runtime "tuple_left" (type.method [(list //.$Tuple runtime.$Index) //.$Value (list)])))]
+  (.let [[indexI accessI] (.case lefts
+                            0
+                            [_.ICONST_0
+                             _.AALOAD]
+                            
+                            lefts
+                            [(leftsI lefts)
+                             (_.INVOKESTATIC //.$Runtime "tuple_left" (type.method [(list //.$Tuple runtime.$Index) //.$Value (list)]))])]
     (|>> (_.CHECKCAST //.$Tuple)
-         (_.int (.int lefts))
+         indexI
          accessI)))
 
 (def: (right-projection lefts)
   (-> Nat Inst)
   (|>> (_.CHECKCAST //.$Tuple)
-       (_.int (.int lefts))
+       (leftsI lefts)
        (_.INVOKESTATIC //.$Runtime "tuple_right" (type.method [(list //.$Tuple runtime.$Index) //.$Value (list)]))))
 
 (def: (path' stack-depth @else @end phase archive path)
@@ -151,33 +164,18 @@
                          pushI))
 
     ## Extra optimization
-    (^ (synthesis.path/seq
-        (synthesis.member/left 0)
-        (synthesis.!bind-top register thenP)))
-    (do phase.monad
-      [then! (path' stack-depth @else @end phase archive thenP)]
-      (wrap (|>> peekI
-                 (_.CHECKCAST //.$Tuple)
-                 (_.int +0)
-                 _.AALOAD
-                 (_.ASTORE register)
-                 then!)))
-
-    ## Extra optimization
-    (^template [<pm> <getter>]
+    (^template [<path> <projection>]
       (^ (synthesis.path/seq
-          (<pm> lefts)
+          (<path> lefts)
           (synthesis.!bind-top register thenP)))
       (do phase.monad
         [then! (path' stack-depth @else @end phase archive thenP)]
         (wrap (|>> peekI
-                   (_.CHECKCAST //.$Tuple)
-                   (_.int (.int lefts))
-                   (_.INVOKESTATIC //.$Runtime <getter> (type.method [(list //.$Tuple runtime.$Index) //.$Value (list)]))
+                   (<projection> lefts)
                    (_.ASTORE register)
                    then!))))
-    ([synthesis.member/left  "tuple_left"]
-     [synthesis.member/right "tuple_right"])
+    ([synthesis.member/left  ..left-projection]
+     [synthesis.member/right ..right-projection])
 
     (#synthesis.Alt leftP rightP)
     (do phase.monad
-- 
cgit v1.2.3