aboutsummaryrefslogtreecommitdiff
path: root/new-luxc/source
diff options
context:
space:
mode:
authorEduardo Julian2017-09-17 00:38:24 -0400
committerEduardo Julian2017-09-17 00:38:24 -0400
commitc95fa2cc7db042fdde7250479727650f43b087a1 (patch)
treecf4cc5a1829fa717b4dad17683251af56c54afa3 /new-luxc/source
parent18fa9ac1ded14e8e6b96609ff1fb6f98af47580f (diff)
- Added pattern-matching compilation.
Diffstat (limited to '')
-rw-r--r--new-luxc/source/luxc/generator/case.jvm.lux214
-rw-r--r--new-luxc/source/luxc/generator/host/jvm/inst.lux49
-rw-r--r--new-luxc/source/luxc/generator/runtime.jvm.lux177
-rw-r--r--new-luxc/source/luxc/generator/structure.jvm.lux2
4 files changed, 411 insertions, 31 deletions
diff --git a/new-luxc/source/luxc/generator/case.jvm.lux b/new-luxc/source/luxc/generator/case.jvm.lux
new file mode 100644
index 000000000..1f351325e
--- /dev/null
+++ b/new-luxc/source/luxc/generator/case.jvm.lux
@@ -0,0 +1,214 @@
+(;module:
+ lux
+ (lux (control [monad #+ do])
+ [macro "lux/" Monad<Lux>])
+ (luxc (lang ["ls" synthesis])
+ (generator (host ["$" jvm]
+ (jvm ["$t" type]
+ ["$i" inst]))
+ [expr]))
+ [../runtime])
+
+(def: $Object $;Type ($t;class "java.lang.Object" (list)))
+
+(def: (pop-altI stack-depth)
+ (-> Nat $;Inst)
+ (case stack-depth
+ +0 id
+ +1 $i;POP
+ +2 $i;POP2
+ _ ## (n.> +2)
+ (|>. $i;POP2
+ (pop-altI (n.- +2 stack-depth)))))
+
+(def: peekI
+ $;Inst
+ (|>. $i;DUP
+ ($i;INVOKESTATIC ../runtime;runtime-name
+ "pm_peek"
+ ($t;method (list ../runtime;$Stack)
+ (#;Some $Object)
+ (list))
+ false)))
+
+(def: popI
+ $;Inst
+ (|>. ($i;INVOKESTATIC ../runtime;runtime-name
+ "pm_pop"
+ ($t;method (list ../runtime;$Stack)
+ (#;Some ../runtime;$Stack)
+ (list))
+ false)))
+
+(def: pushI
+ $;Inst
+ (|>. ($i;INVOKESTATIC ../runtime;runtime-name
+ "pm_push"
+ ($t;method (list ../runtime;$Stack $Object)
+ (#;Some ../runtime;$Stack)
+ (list))
+ false)))
+
+(def: (generate-pattern' stack-depth @else @end path)
+ (-> Nat $;Label $;Label ls;Path (Lux $;Inst))
+ (case path
+ (#ls;ExecP bodyS)
+ (do macro;Monad<Lux>
+ [bodyI (expr;generate bodyS)]
+ (wrap (|>. (pop-altI stack-depth)
+ bodyI
+ ($i;GOTO @end))))
+
+ #ls;UnitP
+ (lux/wrap popI)
+
+ (#ls;BindP register)
+ (lux/wrap (|>. peekI
+ ($i;ASTORE register)
+ popI))
+
+ (#ls;BoolP value)
+ (lux/wrap (let [jumpI (if value $i;IFEQ $i;IFNE)]
+ (|>. peekI
+ ($i;unwrap #$;Boolean)
+ (jumpI @else))))
+
+ (^template [<tag> <prep>]
+ (<tag> value)
+ (lux/wrap (|>. peekI
+ ($i;unwrap #$;Long)
+ ($i;long (|> value <prep>))
+ $i;LCMP
+ ($i;IFNE @else))))
+ ([#ls;NatP (:! Int)]
+ [#ls;IntP (: Int)]
+ [#ls;DegP (:! Int)])
+
+ (#ls;FracP value)
+ (lux/wrap (|>. peekI
+ ($i;unwrap #$;Double)
+ ($i;double value)
+ $i;DCMPL
+ ($i;IFNE @else)))
+
+ (#ls;TextP value)
+ (lux/wrap (|>. peekI
+ ($i;string value)
+ ($i;INVOKEVIRTUAL "java.lang.Object"
+ "equals"
+ ($t;method (list $Object)
+ (#;Some $t;boolean)
+ (list))
+ false)
+ ($i;IFEQ @else)))
+
+ (#ls;TupleP idx subP)
+ (do macro;Monad<Lux>
+ [subI (generate-pattern' stack-depth @else @end subP)
+ #let [[idx tail?] (case idx
+ (#;Left idx)
+ [idx false]
+
+ (#;Right idx)
+ [idx true])]]
+ (wrap (case idx
+ +0
+ (|>. peekI
+ ($i;CHECKCAST ($t;descriptor ../runtime;$Tuple))
+ ($i;int 0)
+ $i;AALOAD
+ pushI
+ subI)
+
+ _
+ (|>. peekI
+ ($i;CHECKCAST ($t;descriptor ../runtime;$Tuple))
+ ($i;int (nat-to-int idx))
+ ($i;INVOKESTATIC ../runtime;runtime-name
+ (if tail? "pm_right" "pm_left")
+ ($t;method (list ../runtime;$Tuple $t;int)
+ (#;Some $Object)
+ (list))
+ false)
+ pushI
+ subI))))
+
+ (#ls;VariantP idx subP)
+ (do macro;Monad<Lux>
+ [subI (generate-pattern' stack-depth @else @end subP)
+ #let [[idx last?] (case idx
+ (#;Left idx)
+ [idx false]
+
+ (#;Right idx)
+ [idx true])
+ flagI (if last?
+ ($i;string "")
+ $i;NULL)]]
+ (wrap (<| $i;with-label (function [@success])
+ $i;with-label (function [@fail])
+ (|>. peekI
+ ($i;CHECKCAST ($t;descriptor ../runtime;$Variant))
+ ($i;int (nat-to-int idx))
+ flagI
+ ($i;INVOKESTATIC ../runtime;runtime-name "pm_variant"
+ ($t;method (list ../runtime;$Variant ../runtime;$Tag ../runtime;$Flag)
+ (#;Some ../runtime;$Datum)
+ (list))
+ false)
+ $i;DUP
+ ($i;IFNULL @fail)
+ ($i;GOTO @success)
+ ($i;label @fail)
+ $i;POP
+ ($i;GOTO @else)
+ ($i;label @success)
+ pushI
+ subI))))
+
+ (#ls;SeqP leftP rightP)
+ (do macro;Monad<Lux>
+ [leftI (generate-pattern' stack-depth @else @end leftP)
+ rightI (generate-pattern' stack-depth @else @end rightP)]
+ (wrap (|>. leftI
+ rightI)))
+
+ (#ls;AltP leftP rightP)
+ (do macro;Monad<Lux>
+ [@alt-else $i;make-label
+ leftI (generate-pattern' (n.inc stack-depth) @alt-else @end leftP)
+ rightI (generate-pattern' stack-depth @else @end rightP)]
+ (wrap (|>. $i;DUP
+ leftI
+ ($i;label @alt-else)
+ $i;POP
+ rightI)))
+ ))
+
+(def: (generate-pattern path @end)
+ (-> ls;Path $;Label (Lux $;Inst))
+ (do macro;Monad<Lux>
+ [@else $i;make-label
+ pathI (generate-pattern' +1 @else @end path)]
+ (wrap (|>. pathI
+ ($i;label @else)
+ $i;POP
+ ($i;INVOKESTATIC ../runtime;runtime-name
+ "pm_fail"
+ ($t;method (list) #;None (list))
+ false)
+ $i;NULL
+ ($i;GOTO @end)))))
+
+(def: #export (generate valueS path)
+ (-> ls;Synthesis ls;Path (Lux $;Inst))
+ (do macro;Monad<Lux>
+ [@end $i;make-label
+ valueI (expr;generate valueS)
+ pathI (generate-pattern path @end)]
+ (wrap (|>. valueI
+ $i;NULL
+ $i;SWAP
+ pushI
+ pathI
+ ($i;label @end)))))
diff --git a/new-luxc/source/luxc/generator/host/jvm/inst.lux b/new-luxc/source/luxc/generator/host/jvm/inst.lux
index 6085ff72b..0f947925c 100644
--- a/new-luxc/source/luxc/generator/host/jvm/inst.lux
+++ b/new-luxc/source/luxc/generator/host/jvm/inst.lux
@@ -3,6 +3,7 @@
(lux (control monad
["p" parser])
(data text/format
+ ["R" result]
(coll [list "L/" Functor<List>]))
[host #+ do-to]
[macro]
@@ -25,14 +26,16 @@
<stack> (declare DUP DUP2 DUP2_X1 DUP2_X2
POP POP2
SWAP)
- <jump> (declare IF_ICMPEQ IF_ACMPEQ IFNULL
- IFEQ IFLT IFLE IFGT IFGE
+ <jump> (declare IF_ICMPEQ IF_ICMPGT IF_ICMPLT IF_ACMPEQ IFNULL
+ IFEQ IFNE IFLT IFLE IFGT IFGE
GOTO)
<var> (declare ILOAD LLOAD DLOAD ALOAD
- ISTORE LSTORE)
+ ISTORE LSTORE ASTORE)
<arithmethic> (declare IADD ISUB
- LADD LSUB LMUL LDIV LREM LCMP
- DADD DSUB DMUL DDIV DREM DCMPG)
+ LADD LSUB LMUL LDIV LREM
+ LCMP
+ DADD DSUB DMUL DDIV DREM
+ DCMPG DCMPL)
<return> (declare RETURN IRETURN LRETURN DRETURN ARETURN)]
(host;import org.objectweb.asm.Opcodes
<primitive>
@@ -101,9 +104,16 @@
(visitMethodInsn [int String String String boolean] void)
(visitLabel [Label] void)
(visitJumpInsn [int Label] void)
- (visitTryCatchBlock [Label Label Label String] void))
+ (visitTryCatchBlock [Label Label Label String] void)
+ (visitTableSwitchInsn [int int Label (Array Label)] void)
+ )
## [Insts]
+(def: #export make-label
+ (Lux Label)
+ (function [compiler]
+ (#R;Success [compiler (Label.new [])])))
+
(def: #export (with-label action)
(-> (-> Label $;Inst) $;Inst)
(action (Label.new [])))
@@ -149,10 +159,12 @@
[IADD] [ISUB]
## Long arithmethic
- [LADD] [LSUB] [LMUL] [LDIV] [LREM] [LCMP]
+ [LADD] [LSUB] [LMUL] [LDIV] [LREM]
+ [LCMP]
## Double arithmetic
- [DADD] [DSUB] [DMUL] [DDIV] [DREM] [DCMPG]
+ [DADD] [DSUB] [DMUL] [DDIV] [DREM]
+ [DCMPG] [DCMPL]
## Conversions
[I2L] [L2I] [L2D] [D2L] [I2C]
@@ -175,7 +187,7 @@
(MethodVisitor.visitVarInsn [(prefix <name>) (nat-to-int register)]))))]
[ILOAD] [LLOAD] [DLOAD] [ALOAD]
- [ISTORE] [LSTORE]
+ [ISTORE] [LSTORE] [ASTORE]
)
(do-template [<name> <inst>]
@@ -237,11 +249,26 @@
(do-to visitor
(MethodVisitor.visitJumpInsn [(prefix <name>) @where]))))]
- [IF_ICMPEQ] [IF_ACMPEQ] [IFNULL]
- [IFEQ] [IFLT] [IFLE] [IFGT] [IFGE]
+ [IF_ICMPEQ] [IF_ICMPGT] [IF_ICMPLT] [IF_ACMPEQ] [IFNULL]
+ [IFEQ] [IFNE] [IFLT] [IFLE] [IFGT] [IFGE]
[GOTO]
)
+(def: #export (TABLESWITCH min max default labels)
+ (-> Int Int $;Label (List $;Label) $;Inst)
+ (function [visitor]
+ (let [num-labels (list;size labels)
+ labels-array (host;array Label num-labels)
+ _ (loop [idx +0]
+ (if (n.< num-labels idx)
+ (exec (host;array-write idx
+ (assume (list;nth idx labels))
+ labels-array)
+ (recur (n.inc idx)))
+ []))]
+ (do-to visitor
+ (MethodVisitor.visitTableSwitchInsn [min max default labels-array])))))
+
(def: #export (try @from @to @handler exception)
(-> $;Label $;Label $;Label Text $;Inst)
(function [visitor]
diff --git a/new-luxc/source/luxc/generator/runtime.jvm.lux b/new-luxc/source/luxc/generator/runtime.jvm.lux
index e094334c0..68e18deaa 100644
--- a/new-luxc/source/luxc/generator/runtime.jvm.lux
+++ b/new-luxc/source/luxc/generator/runtime.jvm.lux
@@ -47,6 +47,12 @@
(def: $Object $;Type ($t;class "java.lang.Object" (list)))
(def: $Object-Array $;Type ($t;array +1 $Object))
(def: $String $;Type ($t;class "java.lang.String" (list)))
+(def: #export $Stack $;Type ($t;array +1 $Object))
+(def: #export $Tuple $;Type ($t;array +1 $Object))
+(def: #export $Variant $;Type ($t;array +1 $Object))
+(def: #export $Tag $;Type $t;int)
+(def: #export $Flag $;Type $Object)
+(def: #export $Datum $;Type $Object)
(def: logI
$;Inst
@@ -55,41 +61,45 @@
(|>. outI ($i;string "LOG: ") (printI "print")
outI $i;SWAP (printI "println"))))
-(def: sum-method
+(def: variant-method
$;Method
($t;method (list $t;int $Object $Object) (#;Some $Object-Array) (list)))
+(def: variant-makeI
+ $;Inst
+ ($i;INVOKESTATIC runtime-name "variant_make" variant-method false))
+
(def: #export someI
$;Inst
(|>. ($i;int 1)
($i;string "")
$i;DUP2_X1
$i;POP2
- ($i;INVOKESTATIC runtime-name "sum_make" sum-method false)))
+ variant-makeI))
(def: #export noneI
$;Inst
(|>. ($i;int 0)
$i;NULL
($i;string unit)
- ($i;INVOKESTATIC runtime-name "sum_make" sum-method false)))
+ variant-makeI))
-(def: add-adt-methods
+(def: adt-methods
$;Def
- (let [store-tag (|>. $i;DUP ($i;int 0) ($i;ILOAD +0) ($i;wrap #$;Int) $i;AASTORE)
- store-flag (|>. $i;DUP ($i;int 1) ($i;ALOAD +1) $i;AASTORE)
- store-value (|>. $i;DUP ($i;int 2) ($i;ALOAD +2) $i;AASTORE)]
- (|>. ($d;method #$;Public $;staticM "sum_make"
+ (let [store-tagI (|>. $i;DUP ($i;int 0) ($i;ILOAD +0) ($i;wrap #$;Int) $i;AASTORE)
+ store-flagI (|>. $i;DUP ($i;int 1) ($i;ALOAD +1) $i;AASTORE)
+ store-valueI (|>. $i;DUP ($i;int 2) ($i;ALOAD +2) $i;AASTORE)]
+ (|>. ($d;method #$;Public $;staticM "variant_make"
($t;method (list $t;int $Object $Object)
- (#;Some ($t;array +1 $Object))
+ (#;Some $Variant)
(list))
(|>. ($i;array $Object +3)
- store-tag
- store-flag
- store-value
+ store-tagI
+ store-flagI
+ store-valueI
$i;ARETURN)))))
-(def: add-nat-methods
+(def: nat-methods
$;Def
(let [compare-nat-method ($t;method (list $t;long $t;long) (#;Some $t;int) (list))
less-thanI (function [@where] (|>. ($i;INVOKESTATIC runtime-name "compare_nat" compare-nat-method false) ($i;IFLT @where)))
@@ -183,7 +193,7 @@
(def: frac-shiftI $;Inst ($i;double (math;pow 32.0 2.0)))
-(def: add-frac-methods
+(def: frac-methods
$;Def
(|>. ($d;method #$;Public $;staticM "decode_frac" ($t;method (list $String) (#;Some $Object-Array) (list))
(<| $i;with-label (function [@from])
@@ -229,7 +239,7 @@
(def: deg-method $;Method ($t;method (list $t;long $t;long) (#;Some $t;long) (list)))
(def: clz-method $;Method ($t;method (list $t;long) (#;Some $t;int) (list)))
-(def: add-deg-methods
+(def: deg-methods
$;Def
(let [## "And" mask corresponding to -1 (FFFF...), on the low 32 bits.
low-half (|>. ($i;int -1) $i;I2L $i;LAND)
@@ -301,14 +311,143 @@
$i;DRETURN)))
)))
+(def: pm-methods
+ $;Def
+ (let [tuple-sizeI (|>. ($i;ALOAD +0) $i;ARRAYLENGTH)
+ tuple-elemI (|>. ($i;ALOAD +0) ($i;ILOAD +1) $i;AALOAD)
+ expected-last-sizeI (|>. ($i;ILOAD +1) ($i;int 1) $i;IADD)
+ tuple-tailI (|>. ($i;ALOAD +0) tuple-sizeI ($i;int 1) $i;ISUB $i;AALOAD ($i;CHECKCAST ($t;descriptor $Tuple)))]
+ (|>. ($d;method #$;Public $;staticM "pm_fail" ($t;method (list) #;None (list))
+ (|>. ($i;NEW "java.lang.IllegalStateException")
+ $i;DUP
+ ($i;string "Invalid expression for pattern-matching.")
+ ($i;INVOKESPECIAL "java.lang.IllegalStateException" "<init>" ($t;method (list $String) #;None (list)) false)
+ $i;ATHROW))
+ ($d;method #$;Public $;staticM "pm_push" ($t;method (list $Stack $Object) (#;Some $Stack) (list))
+ (|>. ($i;int 2)
+ ($i;ANEWARRAY "java.lang.Object")
+ $i;DUP
+ ($i;int 0)
+ ($i;ALOAD +0)
+ $i;AASTORE
+ $i;DUP
+ ($i;int 1)
+ ($i;ALOAD +1)
+ $i;AASTORE
+ $i;ARETURN))
+ ($d;method #$;Public $;staticM "pm_pop" ($t;method (list $Stack) (#;Some $Stack) (list))
+ (|>. ($i;ALOAD +0)
+ ($i;int 0)
+ $i;AALOAD
+ ($i;CHECKCAST ($t;descriptor $Stack))
+ $i;ARETURN))
+ ($d;method #$;Public $;staticM "pm_peek" ($t;method (list $Stack) (#;Some $Object) (list))
+ (|>. ($i;ALOAD +0)
+ ($i;int 1)
+ $i;AALOAD
+ $i;ARETURN))
+ ($d;method #$;Public $;staticM "pm_variant" ($t;method (list $Stack $t;int $Object) (#;Some $Object) (list))
+ (<| $i;with-label (function [@begin])
+ $i;with-label (function [@just-return])
+ $i;with-label (function [@then])
+ $i;with-label (function [@further])
+ $i;with-label (function [@shorten])
+ $i;with-label (function [@wrong])
+ (let [variant-partI (: (-> Nat $;Inst)
+ (function [idx]
+ (|>. ($i;int (nat-to-int idx)) $i;AALOAD)))
+ tagI (: $;Inst
+ (|>. (variant-partI +0) ($i;unwrap #$;Int)))
+ flagI (variant-partI +1)
+ datumI (variant-partI +2)
+ shortenI (|>. ($i;ALOAD +0) tagI ## Get tag
+ ($i;ILOAD +1) $i;ISUB ## Shorten tag
+ ($i;ALOAD +0) flagI ## Get flag
+ ($i;ALOAD +0) datumI ## Get value
+ variant-makeI ## Build sum
+ $i;ARETURN)
+ update-tagI (|>. $i;ISUB ($i;ISTORE +1))
+ update-variantI (|>. ($i;ALOAD +0) datumI ($i;CHECKCAST ($t;descriptor $Variant)) ($i;ASTORE +0))
+ wrongI (|>. $i;NULL $i;ARETURN)
+ return-datumI (|>. ($i;ALOAD +0) datumI $i;ARETURN)])
+ (|>. ($i;label @begin)
+ ($i;ILOAD +1) ## tag
+ ($i;ALOAD +0) tagI ## tag, sumT
+ $i;DUP2 ($i;IF_ICMPEQ @then)
+ $i;DUP2 ($i;IF_ICMPGT @further)
+ $i;DUP2 ($i;IF_ICMPLT @shorten)
+ ## $i;POP2
+ wrongI
+ ($i;label @then) ## tag, sumT
+ ($i;ALOAD +2) ## tag, sumT, wants-last?
+ ($i;ALOAD +0) flagI ## tag, sumT, wants-last?, is-last?
+ ($i;IF_ACMPEQ @just-return) ## tag, sumT
+ ($i;label @further) ## tag, sumT
+ ($i;ALOAD +0) flagI ## tag, sumT, last?
+ ($i;IFNULL @wrong) ## tag, sumT
+ update-tagI
+ update-variantI
+ ($i;GOTO @begin)
+ ($i;label @just-return) ## tag, sumT
+ ## $i;POP2
+ return-datumI
+ ($i;label @shorten) ## tag, sumT
+ ($i;ALOAD +2) ($i;IFNULL @wrong)
+ ## $i;POP2
+ shortenI
+ ($i;label @wrong) ## tag, sumT
+ ## $i;POP2
+ wrongI)))
+ ($d;method #$;Public $;staticM "pm_left" ($t;method (list $Tuple $t;int) (#;Some $Object) (list))
+ (<| $i;with-label (function [@begin])
+ $i;with-label (function [@not-recursive])
+ (let [updated-idxI (|>. $i;SWAP $i;ISUB)])
+ (|>. ($i;label @begin)
+ tuple-sizeI
+ expected-last-sizeI
+ $i;DUP2 ($i;IF_ICMPGT @not-recursive)
+ ## Recursive
+ updated-idxI ($i;ISTORE +1)
+ tuple-tailI ($i;ASTORE +0)
+ ($i;GOTO @begin)
+ ($i;label @not-recursive)
+ ## $i;POP2
+ tuple-elemI
+ $i;ARETURN)))
+ ($d;method #$;Public $;staticM "pm_right" ($t;method (list $Tuple $t;int) (#;Some $Object) (list))
+ (<| $i;with-label (function [@begin])
+ $i;with-label (function [@tail])
+ $i;with-label (function [@slice])
+ (let [updated-idxI (|>. ($i;ILOAD +1) ($i;int 1) $i;IADD tuple-sizeI $i;ISUB)
+ sliceI (|>. ($i;ALOAD +0) ($i;ILOAD +1) tuple-sizeI
+ ($i;INVOKESTATIC "java.util.Arrays" "copyOfRange" ($t;method (list $Object-Array $t;int $t;int) (#;Some $Object-Array) (list)) false))])
+ (|>. ($i;label @begin)
+ tuple-sizeI
+ expected-last-sizeI
+ $i;DUP2 ($i;IF_ICMPEQ @tail)
+ ($i;IF_ICMPGT @slice)
+ ## Must recurse
+ tuple-tailI ($i;ASTORE +0)
+ updated-idxI ($i;ISTORE +1)
+ ($i;GOTO @begin)
+ ($i;label @slice)
+ sliceI
+ $i;ARETURN
+ ($i;label @tail)
+ ## $i;POP2
+ tuple-elemI
+ $i;ARETURN)))
+ )))
+
(def: #export generate
(Lux &common;Bytecode)
(do Monad<Lux>
[_ (wrap [])
#let [bytecode ($d;class #$;V1.6 #$;Public $;finalC runtime-name (list) ["java.lang.Object" (list)] (list)
- (|>. add-adt-methods
- add-nat-methods
- add-frac-methods
- add-deg-methods))]
+ (|>. adt-methods
+ nat-methods
+ frac-methods
+ deg-methods
+ pm-methods))]
_ (&common;store-class runtime-name bytecode)]
(wrap bytecode)))
diff --git a/new-luxc/source/luxc/generator/structure.jvm.lux b/new-luxc/source/luxc/generator/structure.jvm.lux
index 64848546f..9adff1a55 100644
--- a/new-luxc/source/luxc/generator/structure.jvm.lux
+++ b/new-luxc/source/luxc/generator/structure.jvm.lux
@@ -52,7 +52,7 @@
(flagI tail?)
memberI
($i;INVOKESTATIC ../runtime;runtime-name
- "sum_make"
+ "variant_make"
($t;method (list $t;int $Object $Object)
(#;Some ($t;array +1 $Object))
(list))