Этот коммит содержится в:
Ayke van Laethem 2018-04-13 20:03:35 +02:00
родитель 30931ab3a6
коммит 9d8d0b9e03

60
tgo.go
Просмотреть файл

@ -229,7 +229,7 @@ func (c *Compiler) parseInstr(pkgName string, instr ssa.Instruction) error {
if err != nil { if err != nil {
return err return err
} }
c.builder.CreateRet(*val) c.builder.CreateRet(val)
return nil return nil
} else { } else {
return errors.New("todo: return value") return errors.New("todo: return value")
@ -242,7 +242,7 @@ func (c *Compiler) parseInstr(pkgName string, instr ssa.Instruction) error {
} }
} }
func (c *Compiler) parseBuiltin(pkgName string, instr *ssa.CallCommon, call *ssa.Builtin) (*llvm.Value, error) { func (c *Compiler) parseBuiltin(pkgName string, instr *ssa.CallCommon, call *ssa.Builtin) (llvm.Value, error) {
fmt.Printf(" builtin: %#v\n", call) fmt.Printf(" builtin: %#v\n", call)
name := call.Name() name := call.Name()
@ -255,27 +255,27 @@ func (c *Compiler) parseBuiltin(pkgName string, instr *ssa.CallCommon, call *ssa
fmt.Printf(" arg: %s\n", arg); fmt.Printf(" arg: %s\n", arg);
expr, err := c.parseExpr(pkgName, arg) expr, err := c.parseExpr(pkgName, arg)
if err != nil { if err != nil {
return nil, err return llvm.Value{}, err
} }
switch expr.Type() { switch expr.Type() {
case c.stringPtrType: case c.stringPtrType:
c.builder.CreateCall(c.printstringFunc, []llvm.Value{*expr}, "") c.builder.CreateCall(c.printstringFunc, []llvm.Value{expr}, "")
case c.intType: case c.intType:
c.builder.CreateCall(c.printintFunc, []llvm.Value{*expr}, "") c.builder.CreateCall(c.printintFunc, []llvm.Value{expr}, "")
default: default:
return nil, errors.New("unknown arg type") return llvm.Value{}, errors.New("unknown arg type")
} }
} }
if name == "println" { if name == "println" {
c.builder.CreateCall(c.printnlFunc, nil, "") c.builder.CreateCall(c.printnlFunc, nil, "")
} }
return nil, nil // print() or println() returns void return llvm.Value{}, nil // print() or println() returns void
default: default:
return nil, errors.New("todo: builtin: " + name) return llvm.Value{}, errors.New("todo: builtin: " + name)
} }
} }
func (c *Compiler) parseFunctionCall(pkgName string, call *ssa.CallCommon, fn *ssa.Function) (*llvm.Value, error) { func (c *Compiler) parseFunctionCall(pkgName string, call *ssa.CallCommon, fn *ssa.Function) (llvm.Value, error) {
fmt.Printf(" function: %s\n", fn) fmt.Printf(" function: %s\n", fn)
name := fn.Name() name := fn.Name()
@ -285,23 +285,22 @@ func (c *Compiler) parseFunctionCall(pkgName string, call *ssa.CallCommon, fn *s
} }
target := c.mod.NamedFunction(name) target := c.mod.NamedFunction(name)
if target.IsNil() { if target.IsNil() {
return nil, errors.New("undefined function: " + name) return llvm.Value{}, errors.New("undefined function: " + name)
} }
var params []llvm.Value var params []llvm.Value
for _, param := range call.Args { for _, param := range call.Args {
val, err := c.parseExpr(pkgName, param) val, err := c.parseExpr(pkgName, param)
if err != nil { if err != nil {
return nil, err return llvm.Value{}, err
} }
params = append(params, *val) params = append(params, val)
} }
val := c.builder.CreateCall(target, params, "") return c.builder.CreateCall(target, params, ""), nil
return &val, nil
} }
func (c *Compiler) parseCall(pkgName string, instr *ssa.Call) (*llvm.Value, error) { func (c *Compiler) parseCall(pkgName string, instr *ssa.Call) (llvm.Value, error) {
fmt.Printf(" call: %s\n", instr) fmt.Printf(" call: %s\n", instr)
switch call := instr.Common().Value.(type) { switch call := instr.Common().Value.(type) {
@ -310,32 +309,30 @@ func (c *Compiler) parseCall(pkgName string, instr *ssa.Call) (*llvm.Value, erro
case *ssa.Function: case *ssa.Function:
return c.parseFunctionCall(pkgName, instr.Common(), call) return c.parseFunctionCall(pkgName, instr.Common(), call)
default: default:
return nil, errors.New("todo: unknown call type: " + fmt.Sprintf("%#v", call)) return llvm.Value{}, errors.New("todo: unknown call type: " + fmt.Sprintf("%#v", call))
} }
} }
func (c *Compiler) parseBinOp(pkgName string, binop *ssa.BinOp) (*llvm.Value, error) { func (c *Compiler) parseBinOp(pkgName string, binop *ssa.BinOp) (llvm.Value, error) {
x, err := c.parseExpr(pkgName, binop.X) x, err := c.parseExpr(pkgName, binop.X)
if err != nil { if err != nil {
return nil, err return llvm.Value{}, err
} }
y, err := c.parseExpr(pkgName, binop.Y) y, err := c.parseExpr(pkgName, binop.Y)
if err != nil { if err != nil {
return nil, err return llvm.Value{}, err
} }
switch binop.Op { switch binop.Op {
case token.ADD: case token.ADD:
val := c.builder.CreateBinOp(llvm.Add, *x, *y, "") return c.builder.CreateBinOp(llvm.Add, x, y, ""), nil
return &val, nil
case token.MUL: case token.MUL:
val := c.builder.CreateBinOp(llvm.Mul, *x, *y, "") return c.builder.CreateBinOp(llvm.Mul, x, y, ""), nil
return &val, nil
default: default:
return nil, errors.New("todo: unknown binop") return llvm.Value{}, errors.New("todo: unknown binop")
} }
} }
func (c *Compiler) parseExpr(pkgName string, expr ssa.Value) (*llvm.Value, error) { func (c *Compiler) parseExpr(pkgName string, expr ssa.Value) (llvm.Value, error) {
fmt.Printf(" expr: %v\n", expr) fmt.Printf(" expr: %v\n", expr)
switch expr := expr.(type) { switch expr := expr.(type) {
case *ssa.Const: case *ssa.Const:
@ -348,14 +345,12 @@ func (c *Compiler) parseExpr(pkgName string, expr ssa.Value) (*llvm.Value, error
ptr := llvm.AddGlobal(c.mod, strObj.Type(), ".str") ptr := llvm.AddGlobal(c.mod, strObj.Type(), ".str")
ptr.SetInitializer(strObj) ptr.SetInitializer(strObj)
ptr.SetLinkage(llvm.InternalLinkage) ptr.SetLinkage(llvm.InternalLinkage)
ptrCast := llvm.ConstPointerCast(ptr, c.stringPtrType) return llvm.ConstPointerCast(ptr, c.stringPtrType), nil
return &ptrCast, nil
case constant.Int: case constant.Int:
n, _ := constant.Int64Val(expr.Value) // TODO: do something with the 'exact' return value? n, _ := constant.Int64Val(expr.Value) // TODO: do something with the 'exact' return value?
val := llvm.ConstInt(c.intType, uint64(n), true) return llvm.ConstInt(c.intType, uint64(n), true), nil
return &val, nil
default: default:
return nil, errors.New("todo: unknown constant") return llvm.Value{}, errors.New("todo: unknown constant")
} }
case *ssa.BinOp: case *ssa.BinOp:
return c.parseBinOp(pkgName, expr) return c.parseBinOp(pkgName, expr)
@ -364,10 +359,9 @@ func (c *Compiler) parseExpr(pkgName string, expr ssa.Value) (*llvm.Value, error
case *ssa.Parameter: case *ssa.Parameter:
fn := c.funcs[expr.Parent()] fn := c.funcs[expr.Parent()]
llvmFn := c.mod.NamedFunction(fn.name) llvmFn := c.mod.NamedFunction(fn.name)
param := llvmFn.Param(fn.params[expr]) return llvmFn.Param(fn.params[expr]), nil
return &param, nil
default: default:
return nil, errors.New("todo: unknown expression: " + fmt.Sprintf("%#v", expr)) return llvm.Value{}, errors.New("todo: unknown expression: " + fmt.Sprintf("%#v", expr))
} }
} }