645 lines
14 KiB
Go
645 lines
14 KiB
Go
package wrap
|
|
|
|
import (
|
|
"fmt"
|
|
//"reflect"
|
|
"strings"
|
|
|
|
"gitlab.wow.st/gmp/clast/ast"
|
|
"gitlab.wow.st/gmp/clast/types"
|
|
)
|
|
|
|
var (
|
|
Debug = false
|
|
)
|
|
|
|
type cStruct struct {
|
|
goName, cName string
|
|
}
|
|
|
|
type Wrapper struct {
|
|
Interfaces map[string]Interface
|
|
Tps map[string]*types.Node
|
|
Types map[string]string
|
|
gtMap map[string]string
|
|
ctMap map[string]*types.Node // map from go type as a string.
|
|
goStructTypes map[string]cStruct // map from gotype to struct descr.
|
|
goInterfaceTypes []string
|
|
|
|
cCode strings.Builder // put cGo code here
|
|
goTypes strings.Builder // put Go type declarations here
|
|
goCode strings.Builder // put Go code here
|
|
Processed map[string]bool
|
|
Processed2 map[string]bool
|
|
}
|
|
|
|
var gobuiltinTypes map[string]bool = map[string]bool{
|
|
"byte": true,
|
|
"int": true,
|
|
"float64": true,
|
|
}
|
|
|
|
// translate C builtin types to CGo
|
|
var builtinTypes map[string]string = map[string]string{
|
|
"char": "C.char",
|
|
"signed char": "C.schar",
|
|
"unsigned char": "C.uchar",
|
|
"short": "C.short",
|
|
"unsigned short": "C.ushort",
|
|
"int": "C.int",
|
|
"unsigned int": "C.uint",
|
|
"long": "C.long",
|
|
"unsigned long": "C.ulong",
|
|
"long long": "C.longlong",
|
|
"unsigned long long": "C.ulonglong",
|
|
"float": "C.float",
|
|
"double": "C.double",
|
|
"complex float": "C.complexfloat",
|
|
"complex double": "C.complexdouble",
|
|
}
|
|
|
|
//AddType registers a type that needs a Go wrapper struct
|
|
func (w *Wrapper) AddType(t, class string) {
|
|
//fmt.Printf("Type: %s\n",t)
|
|
if _,ok := builtinTypes[t]; ok {
|
|
return
|
|
}
|
|
nt := w.goType(t,class)
|
|
if Debug {
|
|
fmt.Printf("AddType(): (%s) -> %s\n",t,nt)
|
|
}
|
|
if nt == "" {
|
|
return
|
|
}
|
|
if _,ok := w.Interfaces[nt]; !ok { // not an interface
|
|
return
|
|
}
|
|
}
|
|
|
|
func (w *Wrapper) goType(t,class string) string {
|
|
if len(t) > 1 && t[:2] == "id" { t = "NSObject *" }
|
|
if len(t) > 11 {
|
|
if t[:12] == "instancename" { t = class }
|
|
if t[:12] == "instancetype" { t = class + " *" }
|
|
}
|
|
n,err := types.Parse(t)
|
|
if err != nil {
|
|
//fmt.Printf("Cannot parse type %s\n",t)
|
|
return ""
|
|
}
|
|
if n.HasFunc() {
|
|
//fmt.Printf("Function types not supported (%s)\n",t)
|
|
return ""
|
|
}
|
|
ct := n.CtypeSimplified()
|
|
ret := ""
|
|
if nt, ok := w.gtMap[ct]; ok { // we already know this type
|
|
return nt
|
|
}
|
|
if x := n.PointsTo(); x != nil {
|
|
pt := x.CtypeSimplified()
|
|
// if _,ok := w.Interfaces[pt]; ok {
|
|
// // pointer to Objective-C interface, stop here
|
|
// //fmt.Printf("goType(): %s -> %s\n",t,pt)
|
|
// w.gtMap[ct] = pt
|
|
// w.goInterfaceTypes = append(w.goInterfaceTypes,pt)
|
|
// return pt
|
|
// }
|
|
// pt = x.BaseType().CtypeSimplified()
|
|
// if _,ok := w.Interfaces[pt]; ok {
|
|
// // ultimately points to an interface, so need a wrapper
|
|
// w.AddType(pt,class)
|
|
// ret = w.goType(pt,class) + "Ptr"
|
|
// //fmt.Printf("goType(): %s -> %s\n",t,ret)
|
|
// w.gtMap[ct] = ret
|
|
// return ret
|
|
// }
|
|
// pointer to non-interface type
|
|
ret = "*" + w.goType(pt,class)
|
|
w.gtMap[ct] = ret
|
|
w.ctMap[ret] = n
|
|
return ret
|
|
}
|
|
if x := n.ArrayOf(); x != nil {
|
|
pt := x.CtypeSimplified()
|
|
w.AddType(pt,class)
|
|
ret = "[]" + w.goType(pt,class)
|
|
//fmt.Printf("goType(): %s -> %s\n",t,ret)
|
|
w.gtMap[ct] = ret
|
|
w.ctMap[ret] = n
|
|
return ret
|
|
}
|
|
if bt,ok := builtinTypes[ct]; ok {
|
|
ct = bt
|
|
}
|
|
if _,ok := w.Interfaces[ct]; ok {
|
|
// pointer to Objective-C interface, stop here
|
|
//fmt.Printf("goType(): %s -> %s\n",t,pt)
|
|
w.gtMap[ct] = ct
|
|
w.goInterfaceTypes = append(w.goInterfaceTypes,ct)
|
|
w.ctMap[ct] = n
|
|
w.Types[ct] = t
|
|
return ct
|
|
}
|
|
if n.IsStruct() {
|
|
gt := strings.Title(ct)
|
|
gt = strings.ReplaceAll(gt, " ", "")
|
|
w.gtMap[ct] = gt
|
|
w.goStructTypes[gt] = cStruct{ goName: gt, cName: ct }
|
|
//fmt.Printf("goType(): %s -> %s\n",t,gt)
|
|
w.ctMap[gt] = n
|
|
return gt
|
|
}
|
|
//fmt.Printf("goType(): %s -> %s\n",t,ct)
|
|
w.gtMap[ct] = ct
|
|
w.ctMap[ct] = n
|
|
return ct
|
|
}
|
|
|
|
func NewWrapper(debug bool) *Wrapper {
|
|
Debug = debug
|
|
if Debug { fmt.Println("// Debug mode") }
|
|
return &Wrapper{
|
|
Interfaces: map[string]Interface{},
|
|
Types: map[string]string{},
|
|
gtMap: map[string]string{},
|
|
ctMap: map[string]*types.Node{},
|
|
goStructTypes: map[string]cStruct{},
|
|
Processed: map[string]bool{},
|
|
Processed2: map[string]bool{},
|
|
}
|
|
}
|
|
|
|
type Property struct {
|
|
Name, Type, Attr string
|
|
Tp *types.Type
|
|
}
|
|
|
|
type Parameter struct {
|
|
Pname, Vname, Type string
|
|
Tp *types.Type
|
|
}
|
|
|
|
type Method struct {
|
|
Name, Type, Class string
|
|
Tp *types.Type
|
|
ClassMethod bool
|
|
Parameters []Parameter
|
|
}
|
|
|
|
func (m Method) isVoid() bool {
|
|
return m.Type == "void"
|
|
}
|
|
|
|
func (w Wrapper) isObject(tp string) bool { // takes a goType
|
|
if _,ok := w.Interfaces[tp]; ok {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (w Wrapper) cparamlist(m Method) string {
|
|
ret := make([]string,0)
|
|
if !m.ClassMethod {
|
|
ret = append(ret,"void* obj")
|
|
}
|
|
for _,p := range m.Parameters {
|
|
tp := p.Type
|
|
gtp := w.goType(tp,m.Class)
|
|
//w.goType(tp,m.Class)
|
|
if w.isObject(gtp) {
|
|
tp = "void*"
|
|
}
|
|
ret = append(ret,fmt.Sprintf("%s %s",tp,p.Vname))
|
|
}
|
|
return strings.Join(ret,", ")
|
|
}
|
|
|
|
func (w Wrapper) objcparamlist(m Method) string {
|
|
if len(m.Parameters) == 0 {
|
|
return m.Name
|
|
}
|
|
first := true
|
|
ret := []string{}
|
|
for _,p := range m.Parameters {
|
|
if first {
|
|
ret = append(ret,m.Name + ":" + p.Vname)
|
|
first = false
|
|
} else {
|
|
ret = append(ret, p.Pname + ":" + p.Vname)
|
|
}
|
|
}
|
|
return strings.Join(ret," ")
|
|
}
|
|
|
|
//goreserved is a map telling whether a word is a go reserved word that is not
|
|
//also a C/Objective-C reserved word.
|
|
var goreserved map[string]bool = map[string]bool{
|
|
"range": true,
|
|
}
|
|
|
|
func (m *Method) gpntp() ([]string,[]*types.Type,string) {
|
|
ns := []string{}
|
|
tps := []*types.Type{}
|
|
if !m.ClassMethod {
|
|
ns = append(ns,"o")
|
|
tps = append(tps,types.NewTypeFromString(m.Class + "*",""))
|
|
}
|
|
for _,p := range m.Parameters {
|
|
gname := p.Vname
|
|
if goreserved[gname] {
|
|
gname = gname + "_"
|
|
}
|
|
ns = append(ns,gname)
|
|
tps = append(tps,p.Tp)
|
|
}
|
|
ret := []string{}
|
|
i := 0
|
|
if !m.ClassMethod { i = 1 }
|
|
for ; i < len(ns); i++ {
|
|
ret = append(ret,ns[i] + " " + tps[i].GoType())
|
|
}
|
|
return ns, tps, strings.Join(ret,", ")
|
|
}
|
|
|
|
func (w Wrapper) goparamlist(m Method) (string,[]string,bool) {
|
|
ret := []string{}
|
|
tps := []string{}
|
|
for _,p := range m.Parameters {
|
|
gname := p.Vname
|
|
w.AddType(p.Type,m.Class)
|
|
tp := w.goType(p.Type,m.Class)
|
|
if tp == "" {
|
|
return "UNSUPPORTED TYPE", []string{},false
|
|
}
|
|
if w.isObject(tp) {
|
|
tp = "*" + tp
|
|
}
|
|
if goreserved[gname] {
|
|
gname = gname + "_"
|
|
}
|
|
tps = append(tps,tp)
|
|
ret = append(ret,fmt.Sprintf("%s %s",gname,tp))
|
|
}
|
|
return strings.Join(ret,", "),tps,true
|
|
}
|
|
|
|
func (w Wrapper) goparamnames(m Method) string {
|
|
ret := make([]string,0)
|
|
if !m.ClassMethod {
|
|
ret = append(ret,"o.ptr")
|
|
}
|
|
for _,p := range m.Parameters {
|
|
gname := p.Vname
|
|
if goreserved[gname] {
|
|
gname = gname + "_"
|
|
}
|
|
gt := w.goType(p.Type,m.Class)
|
|
if gt == "" {
|
|
return "UNSUPPORTED TYPE " + p.Type
|
|
}
|
|
if w.isObject(gt) {
|
|
gname = gname + ".ptr"
|
|
} else {
|
|
n := w.ctMap[gt]
|
|
star := ""
|
|
for pt := n.PointsTo(); pt != nil; pt = pt.PointsTo() {
|
|
star = star + "*"
|
|
}
|
|
ct := n.BaseType().CtypeSimplified()
|
|
if n.IsStruct() {
|
|
ct = strings.ReplaceAll(ct,"struct ","")
|
|
ct = "struct_" + ct
|
|
}
|
|
if gt[0] == '*' { // wrap pointers in unsafe.Pointer()
|
|
gname = "unsafe.Pointer(" + gname + ")"
|
|
}
|
|
gname = "(" + star + "C." + ct + ")(" + gname + ")"
|
|
}
|
|
ret = append(ret,gname)
|
|
}
|
|
return strings.Join(ret, ", ")
|
|
}
|
|
|
|
type Interface struct {
|
|
Name, Super string
|
|
Properties map[string]Property
|
|
Methods map[string]Method
|
|
}
|
|
|
|
func (i Interface) IsRoot() bool {
|
|
if i.Super == "" || i.Super == i.Name {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (w *Wrapper) AddInterface(n *ast.ObjCInterfaceDecl) {
|
|
//fmt.Printf("ast.ObjCInterfaceDecl: %s\n",n.Name)
|
|
w.add(n.Name, n.Children())
|
|
}
|
|
|
|
func (w *Wrapper) AddCategory(n *ast.ObjCCategoryDecl) {
|
|
ns := n.Children()
|
|
if len(ns) > 0 {
|
|
switch x := ns[0].(type) {
|
|
case *ast.ObjCInterface:
|
|
w.add(x.Name, ns[1:])
|
|
return
|
|
}
|
|
}
|
|
fmt.Printf("Not adding methods for %s: interface name not found in first child node of category defclaration\n",n.Name)
|
|
}
|
|
|
|
func (w *Wrapper) add(name string, ns []ast.Node) {
|
|
var i Interface
|
|
var ok bool
|
|
if i,ok = w.Interfaces[name]; !ok {
|
|
i = Interface{
|
|
Name: name,
|
|
Properties: map[string]Property{},
|
|
Methods: map[string]Method{},
|
|
}
|
|
w.AddType(name,name) // need this?
|
|
}
|
|
tp := types.NewTypeFromString(name,name)
|
|
tp.Wrap()
|
|
var avail bool
|
|
for _,c := range ns {
|
|
switch x := c.(type) {
|
|
case *ast.ObjCPropertyDecl:
|
|
//fmt.Printf("ObjCPropertyDecl: %s\n",x.Name)
|
|
p := Property{
|
|
Name: x.Name,
|
|
Type: x.Type,
|
|
Tp: types.NewTypeFromString(x.Type,name),
|
|
}
|
|
//_,avail = w.GetParms(x,name) // TODO
|
|
//if avail {
|
|
// w.AddType(x.Type,name)
|
|
i.Properties[p.Name] = p
|
|
//}
|
|
case *ast.ObjCMethodDecl:
|
|
//fmt.Printf("ObjCMethodDecl: %s\n",x.Name)
|
|
m := Method{
|
|
Name: x.Name,
|
|
Type: x.Type,
|
|
Tp: types.NewTypeFromString(x.Type,name),
|
|
Class: name,
|
|
ClassMethod: x.ClassMethod,
|
|
}
|
|
m.Parameters, avail = w.GetParms(x,name)
|
|
if avail {
|
|
// w.AddType(x.Type,name)
|
|
i.Methods[m.Name] = m
|
|
}
|
|
case *ast.ObjCProtocol:
|
|
//fmt.Printf("ast.ObjCProtocol: %s\n",x.Name)
|
|
case *ast.ObjCInterface:
|
|
if x.Super {
|
|
//fmt.Printf("ast.ObjCInterface: %s inherits from %s\n",name,x.Name)
|
|
i.Super = x.Name
|
|
types.SetSuper(name,x.Name)
|
|
}
|
|
case *ast.Unknown:
|
|
//fmt.Printf("(*ast.Unkonwn %s: %s)\n",x.Name,x.Content)
|
|
default:
|
|
//fmt.Println(reflect.TypeOf(x))
|
|
}
|
|
}
|
|
//fmt.Println("Add interface ",i.Name)
|
|
w.Interfaces[i.Name] = i
|
|
}
|
|
|
|
type AvailAttr struct {
|
|
OS, Version string
|
|
Deprecated bool
|
|
}
|
|
|
|
//GetParms returns the parameters of a method declaration and a bool
|
|
//indicating whether the given method is available on MacOS and not
|
|
//deprecated.
|
|
func (w *Wrapper) GetParms(n *ast.ObjCMethodDecl,class string) ([]Parameter,bool) {
|
|
ret := make([]Parameter,0)
|
|
avail := make([]AvailAttr,0)
|
|
j := 0
|
|
for _,c := range n.Children() {
|
|
switch x := c.(type) {
|
|
case *ast.ParmVarDecl:
|
|
p := Parameter{
|
|
Pname: n.Parameters[j],
|
|
Vname: x.Name,
|
|
Type: x.Type,
|
|
Tp: types.NewTypeFromString(x.Type,class),
|
|
}
|
|
ret = append(ret,p)
|
|
j++
|
|
case *ast.AvailabilityAttr:
|
|
avail = append(avail,
|
|
AvailAttr{
|
|
OS: x.OS,
|
|
Version: x.Version,
|
|
Deprecated: x.Unknown1 != "0",
|
|
})
|
|
//fmt.Println("AvailAttr ",avail,x)
|
|
case *ast.UnavailableAttr:
|
|
avail = append(avail,
|
|
AvailAttr{ OS: "macos", Deprecated: true })
|
|
case *ast.Unknown:
|
|
if Debug { fmt.Printf("GetParms(): ast.Unknown: %s\n",x.Name) }
|
|
}
|
|
}
|
|
// check that the method is available for this OS and not deprecated
|
|
a := func() bool {
|
|
if len(avail) == 0 {
|
|
return true
|
|
}
|
|
for _,x := range avail {
|
|
if x.OS == "macos" && x.Deprecated == false {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}()
|
|
if !a {
|
|
return nil, false
|
|
}
|
|
// check that we found the right number of parameters
|
|
if len(ret) != len(n.Parameters) {
|
|
fmt.Printf("Error in method declaration %s: Wrong number of ParmVarDecl children: %d parameters but %d ParmVarDecl children\n",n.Name,len(n.Parameters),len(ret))
|
|
}
|
|
return ret, true
|
|
}
|
|
|
|
// new version of ProcessType
|
|
func (w *Wrapper) pt2(tps ...*types.Type) {
|
|
switch len(tps) {
|
|
case 0:
|
|
return
|
|
case 1:
|
|
break
|
|
default:
|
|
for _,tp := range tps {
|
|
w.pt2(tp)
|
|
}
|
|
return
|
|
}
|
|
tp := tps[0].BaseType()
|
|
if w.Processed2[tp.GoType()] { return }
|
|
w.Processed2[tp.GoType()] = true
|
|
if tp.Node.IsFunction() {
|
|
return
|
|
}
|
|
w.goTypes.WriteString(tps[0].GoTypeDecl())
|
|
}
|
|
|
|
func (w *Wrapper) ProcessType(gotype string) {
|
|
if gotype == "" {
|
|
return
|
|
}
|
|
if gotype[0] == '*' {
|
|
gotype = gotype[1:]
|
|
}
|
|
if w.Processed[gotype] {
|
|
return
|
|
}
|
|
w.Processed[gotype] = true
|
|
if _,ok := gobuiltinTypes[gotype]; ok {
|
|
return
|
|
}
|
|
ctype := w.Types[gotype]
|
|
if _,ok := builtinTypes[ctype]; ok {
|
|
return
|
|
}
|
|
if Debug {
|
|
fmt.Printf("Processing %s (%s)\n",gotype,ctype)
|
|
}
|
|
if i,ok := w.Interfaces[gotype]; ok {
|
|
if Debug {
|
|
fmt.Printf("Have interface for %s. super = %s\n",gotype,i.Super)
|
|
}
|
|
if i.Name != i.Super {
|
|
w.ProcessType(i.Super)
|
|
}
|
|
var fields string
|
|
// if there is an Interface known for this type, decide if it is the
|
|
// root object and if so give it a pointer element:
|
|
if i.IsRoot() {
|
|
fields = "ptr unsafe.Pointer"
|
|
} else {
|
|
fields = i.Super // embed superclass
|
|
}
|
|
w.goTypes.WriteString(fmt.Sprintf(`
|
|
// %s
|
|
type %s struct { %s }
|
|
`,ctype, gotype, fields))
|
|
}
|
|
if s,ok := w.goStructTypes[gotype]; ok {
|
|
ct := strings.ReplaceAll(s.cName," ","_")
|
|
w.goTypes.WriteString(fmt.Sprintf(`
|
|
type %s %s
|
|
`,s.goName,"C." + ct))
|
|
}
|
|
}
|
|
|
|
func (w *Wrapper) ProcessTypes(tps []string) {
|
|
for _,tp := range tps {
|
|
w.ProcessType(tp)
|
|
}
|
|
}
|
|
|
|
func (w *Wrapper) Wrap(toproc []string) {
|
|
|
|
w.cCode.WriteString(`/*
|
|
#cgo CFLAGS: -x objective-c
|
|
#cgo LDFLAGS: -framework Foundation
|
|
|
|
#import <Foundation/Foundation.h>
|
|
`)
|
|
|
|
pInterfaces := map[string]Interface{}
|
|
for _,iface := range toproc {
|
|
pInterfaces[iface] = w.Interfaces[iface]
|
|
}
|
|
for iname,i := range pInterfaces {
|
|
if Debug {
|
|
fmt.Printf("Interface %s: %d properties, %d methods\n",
|
|
iname, len(i.Properties), len(i.Methods))
|
|
}
|
|
|
|
w.goCode.WriteString(fmt.Sprintf(`
|
|
func New%s() *%s {
|
|
ret := &%s{}
|
|
ret.ptr = unsafe.Pointer(C.New%s())
|
|
ret = ret.Init()
|
|
return ret
|
|
}
|
|
`,i.Name,i.Name,i.Name,i.Name))
|
|
|
|
w.cCode.WriteString(fmt.Sprintf(`
|
|
%s*
|
|
New%s() {
|
|
return [%s alloc];
|
|
}
|
|
`, i.Name, i.Name, i.Name))
|
|
|
|
for _,p := range i.Properties {
|
|
if Debug {
|
|
fmt.Printf(" property: %s (%s)\n", p.Name, p.Type)
|
|
}
|
|
w.AddType(p.Type,i.Name)
|
|
}
|
|
for _,m := range i.Methods {
|
|
if Debug {
|
|
fmt.Printf(" method: %s (%s)\n", m.Name, m.Type)
|
|
}
|
|
if m.Tp.Node.IsFunction() {
|
|
continue
|
|
}
|
|
gname := strings.Title(m.Name)
|
|
cname := i.Name + "_" + m.Name
|
|
|
|
w.AddType(m.Type,i.Name)
|
|
//cmtype := w.ctMap[w.goType(m.Type,i.Name)].CtypeSimplified()
|
|
cmtype := m.Tp.CType()
|
|
ns,tps,gplist := m.gpntp()
|
|
w.pt2(tps...)
|
|
w.pt2(m.Tp)
|
|
w.goCode.WriteString(fmt.Sprintf(`
|
|
func (o *%s) %s(%s) %s {
|
|
`,i.Name,gname,gplist,m.Tp.GoType()))
|
|
w.goCode.WriteString(
|
|
types.GoToC(cname,ns,m.Tp,tps) + "}\n\n")
|
|
|
|
cret := ""
|
|
if !m.isVoid() {
|
|
cret = "return "
|
|
}
|
|
var cobj string
|
|
if m.ClassMethod {
|
|
cobj = i.Name
|
|
} else {
|
|
cobj = "(id)obj"
|
|
}
|
|
w.cCode.WriteString(fmt.Sprintf(`
|
|
%s
|
|
%s(%s) {
|
|
%s[%s %s];
|
|
}`, cmtype, cname, w.cparamlist(m), cret, cobj, w.objcparamlist(m)))
|
|
}
|
|
}
|
|
fmt.Println(`package main
|
|
`)
|
|
fmt.Println(w.cCode.String())
|
|
fmt.Println(`
|
|
*/
|
|
import "C"
|
|
|
|
import (
|
|
"unsafe"
|
|
)
|
|
`)
|
|
fmt.Println(w.goTypes.String())
|
|
fmt.Println(w.goCode.String())
|
|
}
|