gopy/bind/vars.go

128 lines
2.7 KiB
Go

// Copyright 2015 The go-python Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package bind
import (
"go/token"
"go/types"
)
type Var struct {
pkg *Package
id string
doc string
name string
tvar *types.Var
}
func (v *Var) Name() string {
return v.name
}
func newVar(p *Package, typ types.Type, name, doc string) *Var {
return &Var{
pkg: p,
id: p.Name() + "_" + name,
doc: doc,
name: name,
tvar: types.NewVar(token.NoPos, p.pkg, name, typ),
}
}
func newVarsFrom(p *Package, tuple *types.Tuple) []*Var {
vars := make([]*Var, 0, tuple.Len())
for i := 0; i < tuple.Len(); i++ {
vars = append(vars, newVarFrom(p, tuple.At(i)))
}
return vars
}
func newVarFrom(p *Package, v *types.Var) *Var {
return newVar(p, v.Type(), v.Name(), p.getDoc("", v))
}
func getTypeString(t types.Type) string {
return types.TypeString(t, func(*types.Package) string { return " " })
}
func (v *Var) Package() *Package { return v.pkg }
func (v *Var) ID() string { return v.id }
func (v *Var) Doc() string { return v.doc }
func (v *Var) GoName() string { return v.name }
func (v *Var) GoType() types.Type { return v.tvar.Type() }
func (v *Var) CType() string { return v.sym.cpyname }
func (v *Var) CGoType() string { return v.sym.cgoname }
func (v *Var) PyCode() string {
return v.sym.pyfmt
}
func (v *Var) isGoString() bool {
switch typ := v.GoType().(type) {
case *types.Basic:
return typ.Kind() == types.String
}
return false
}
func (v *Var) genDecl(g *printer) {
g.Printf("%[1]s c_%[2]s;\n", v.CGoType(), v.Name())
}
func (v *Var) genRecvDecl(g *printer) {
g.Printf("%[1]s c_%[2]s;\n", v.CGoType(), v.Name())
}
func (v *Var) genRecvImpl(g *printer) {
n := v.sym.cpyname
g.Printf("c_%[1]s = ((%[2]s*)self)->cgopy;\n", v.Name(), n)
}
func (v *Var) genRetDecl(g *printer) {
g.Printf("%[1]s c_gopy_ret;\n", v.sym.cgoname)
}
func (v *Var) getArgParse() (string, []string) {
addrs := make([]string, 0, 1)
cnv := v.sym.hasConverter()
if cnv {
addrs = append(addrs, v.sym.py2c)
}
addr := "&c_" + v.Name()
addrs = append(addrs, addr)
return v.sym.pyfmt, addrs
}
func (v *Var) getArgBuildValue() (string, []string) {
args := make([]string, 0, 1)
cnv := v.sym.hasConverter()
if cnv {
args = append(args, ""+v.sym.c2py)
}
arg := "c_" + v.Name()
if cnv {
arg = "&" + arg
}
args = append(args, arg)
return v.sym.pyfmt, args
}
func (v *Var) genFuncPreamble(g *printer) {
}
func (v *Var) getFuncArg() string {
return "c_" + v.Name()
}
func (v *Var) needWrap() bool {
typ := v.GoType()
return needWrapType(typ)
}
func (v *Var) gofmt() string {
return v.sym.gofmt()
}