attach term Type to values
This commit is contained in:
parent
8937201f5c
commit
a5040a6248
46
vm/cmds.go
46
vm/cmds.go
@ -11,7 +11,7 @@ type buildCmd struct {
|
|||||||
pattern lang.Tuple
|
pattern lang.Tuple
|
||||||
inTypeFn func(lang.Term) (llvm.Type, error)
|
inTypeFn func(lang.Term) (llvm.Type, error)
|
||||||
outTypeFn func(lang.Term) (llvm.Type, error)
|
outTypeFn func(lang.Term) (llvm.Type, error)
|
||||||
buildFn func(lang.Term) (llvm.Value, error)
|
buildFn func(lang.Term) (val, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cmd buildCmd) matches(t lang.Term) bool {
|
func (cmd buildCmd) matches(t lang.Term) bool {
|
||||||
@ -32,7 +32,7 @@ func (cmd buildCmd) outType(t lang.Term) (llvm.Type, error) {
|
|||||||
return cmd.outTypeFn(t)
|
return cmd.outTypeFn(t)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cmd buildCmd) build(t lang.Term) (llvm.Value, error) {
|
func (cmd buildCmd) build(t lang.Term) (val, error) {
|
||||||
return cmd.buildFn(t)
|
return cmd.buildFn(t)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -55,14 +55,17 @@ func buildCmds(mod *Module) []buildCmd {
|
|||||||
outTypeFn: func(t lang.Term) (llvm.Type, error) {
|
outTypeFn: func(t lang.Term) (llvm.Type, error) {
|
||||||
return llvm.Int64Type(), nil
|
return llvm.Int64Type(), nil
|
||||||
},
|
},
|
||||||
buildFn: func(t lang.Term) (llvm.Value, error) {
|
buildFn: func(t lang.Term) (val, error) {
|
||||||
con := t.(lang.Const)
|
con := t.(lang.Const)
|
||||||
coni, err := strconv.ParseInt(string(con), 10, 64)
|
coni, err := strconv.ParseInt(string(con), 10, 64)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return llvm.Value{}, err
|
return val{}, err
|
||||||
}
|
}
|
||||||
|
return val{
|
||||||
// TODO why does this have to be cast?
|
// TODO why does this have to be cast?
|
||||||
return llvm.ConstInt(llvm.Int64Type(), uint64(coni), false), nil
|
v: llvm.ConstInt(llvm.Int64Type(), uint64(coni), false),
|
||||||
|
typ: lang.AInt,
|
||||||
|
}, nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|
||||||
@ -82,28 +85,36 @@ func buildCmds(mod *Module) []buildCmd {
|
|||||||
}
|
}
|
||||||
return llvm.StructType(typs, false), nil
|
return llvm.StructType(typs, false), nil
|
||||||
},
|
},
|
||||||
buildFn: func(t lang.Term) (llvm.Value, error) {
|
buildFn: func(t lang.Term) (val, error) {
|
||||||
tup := t.(lang.Tuple)
|
tup := t.(lang.Tuple)
|
||||||
// if the tuple is empty then it is a void
|
// if the tuple is empty then it is a void
|
||||||
if len(tup) == 0 {
|
if len(tup) == 0 {
|
||||||
return llvm.Undef(llvm.VoidType()), nil
|
return val{
|
||||||
|
v: llvm.Undef(llvm.VoidType()),
|
||||||
|
typ: lang.Tuple{lang.ATuple, lang.Tuple{}},
|
||||||
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
vals := make([]llvm.Value, len(tup))
|
vals := make([]val, len(tup))
|
||||||
typs := make([]llvm.Type, len(tup))
|
typs := make([]llvm.Type, len(tup))
|
||||||
|
ttyps := make([]lang.Term, len(tup))
|
||||||
for i := range tup {
|
for i := range tup {
|
||||||
if vals[i], err = mod.build(tup[i]); err != nil {
|
if vals[i], err = mod.build(tup[i]); err != nil {
|
||||||
return llvm.Value{}, err
|
return val{}, err
|
||||||
}
|
}
|
||||||
typs[i] = vals[i].Type()
|
typs[i] = vals[i].v.Type()
|
||||||
|
ttyps[i] = vals[i].typ
|
||||||
}
|
}
|
||||||
|
|
||||||
str := llvm.Undef(llvm.StructType(typs, false))
|
str := llvm.Undef(llvm.StructType(typs, false))
|
||||||
for i := range vals {
|
for i := range vals {
|
||||||
str = mod.b.CreateInsertValue(str, vals[i], i, "")
|
str = mod.b.CreateInsertValue(str, vals[i].v, i, "")
|
||||||
}
|
}
|
||||||
return str, nil
|
return val{
|
||||||
|
v: str,
|
||||||
|
typ: lang.Tuple{lang.ATuple, lang.Tuple(ttyps)},
|
||||||
|
}, nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|
||||||
@ -112,17 +123,20 @@ func buildCmds(mod *Module) []buildCmd {
|
|||||||
outTypeFn: func(t lang.Term) (llvm.Type, error) {
|
outTypeFn: func(t lang.Term) (llvm.Type, error) {
|
||||||
return llvm.Int64Type(), nil
|
return llvm.Int64Type(), nil
|
||||||
},
|
},
|
||||||
buildFn: func(t lang.Term) (llvm.Value, error) {
|
buildFn: func(t lang.Term) (val, error) {
|
||||||
tup := t.(lang.Tuple)
|
tup := t.(lang.Tuple)
|
||||||
v1, err := mod.build(tup[0])
|
v1, err := mod.build(tup[0])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return llvm.Value{}, err
|
return val{}, err
|
||||||
}
|
}
|
||||||
v2, err := mod.build(tup[1])
|
v2, err := mod.build(tup[1])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return llvm.Value{}, err
|
return val{}, err
|
||||||
}
|
}
|
||||||
return mod.b.CreateAdd(v1, v2, ""), nil
|
return val{
|
||||||
|
v: mod.b.CreateAdd(v1.v, v2.v, ""),
|
||||||
|
typ: v1.typ,
|
||||||
|
}, nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
13
vm/vm.go
13
vm/vm.go
@ -10,8 +10,8 @@ import (
|
|||||||
"llvm.org/llvm/bindings/go/llvm"
|
"llvm.org/llvm/bindings/go/llvm"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Val holds onto a value which has been created within the VM
|
type val struct {
|
||||||
type Val struct {
|
typ lang.Term
|
||||||
v llvm.Value
|
v llvm.Value
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -87,14 +87,15 @@ func (mod *Module) outType(t lang.Term) (llvm.Type, error) {
|
|||||||
return cmd.outType(t.(lang.Tuple)[1])
|
return cmd.outType(t.(lang.Tuple)[1])
|
||||||
}
|
}
|
||||||
|
|
||||||
func (mod *Module) build(t lang.Term) (llvm.Value, error) {
|
func (mod *Module) build(t lang.Term) (val, error) {
|
||||||
cmd, err := mod.matchingBuildCmd(t)
|
cmd, err := mod.matchingBuildCmd(t)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return llvm.Value{}, err
|
return val{}, err
|
||||||
}
|
}
|
||||||
return cmd.build(t.(lang.Tuple)[1])
|
return cmd.build(t.(lang.Tuple)[1])
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO make this return a val once we get function types
|
||||||
func (mod *Module) buildFn(tt ...lang.Term) (llvm.Value, error) {
|
func (mod *Module) buildFn(tt ...lang.Term) (llvm.Value, error) {
|
||||||
if len(tt) == 0 {
|
if len(tt) == 0 {
|
||||||
return llvm.Value{}, errors.New("function cannot be empty")
|
return llvm.Value{}, errors.New("function cannot be empty")
|
||||||
@ -120,13 +121,13 @@ func (mod *Module) buildFn(tt ...lang.Term) (llvm.Value, error) {
|
|||||||
block := llvm.AddBasicBlock(fn, "")
|
block := llvm.AddBasicBlock(fn, "")
|
||||||
mod.b.SetInsertPoint(block, block.FirstInstruction())
|
mod.b.SetInsertPoint(block, block.FirstInstruction())
|
||||||
|
|
||||||
var out llvm.Value
|
var out val
|
||||||
for _, t := range tt {
|
for _, t := range tt {
|
||||||
if out, err = mod.build(t); err != nil {
|
if out, err = mod.build(t); err != nil {
|
||||||
return llvm.Value{}, err
|
return llvm.Value{}, err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
mod.b.CreateRet(out)
|
mod.b.CreateRet(out.v)
|
||||||
return fn, nil
|
return fn, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user