miqt/cmd/genbindings/emitgo.go

436 lines
12 KiB
Go
Raw Normal View History

2024-08-06 01:03:23 +00:00
package main
import (
"C"
"go/format"
"log"
"sort"
2024-08-06 02:29:12 +00:00
"strings"
2024-08-18 07:01:36 +00:00
"unsafe"
2024-08-06 01:03:23 +00:00
)
func goReservedWord(s string) bool {
switch s {
case "default", "const", "func", "var", "type", "len", "new", "copy", "import":
return true
default:
return false
}
}
func (p CppParameter) RenderTypeGo() string {
if p.Pointer && p.ParameterType == "char" {
return "string"
}
if p.ParameterType == "char" {
return "byte"
}
if p.ParameterType == "QString" {
return "string"
}
if p.ParameterType == "uintptr_t" {
return "uintptr"
}
2024-08-11 04:37:18 +00:00
if t, ok := p.QListOf(); ok {
return "[]" + t.RenderTypeGo()
}
ret := ""
if p.ByRef || p.Pointer {
ret += "*"
}
switch p.ParameterType {
case "char", "qint8", "unsigned char", "uchar", "quint8":
ret += "byte" // Strictly speaking, Go byte is unsigned and char may be signed
case "short", "qint16":
ret += "int16"
case "ushort", "quint16":
ret += "uint16"
case "long":
// Windows ILP32 - 32-bits
// Linux LP64 - 64-bits
if C.sizeof_long == 4 {
ret += "int32"
} else {
ret += "int64"
}
case "ulong", "unsigned long":
if C.sizeof_long == 4 {
ret += "uint32"
} else {
ret += "uint64"
}
case "QRgb":
if C.sizeof_int == 4 {
ret += "uint32"
} else {
ret += "uint64"
}
case "qint32":
ret += "int32"
case "quint32":
ret += "uint32"
case "qlonglong", "qint64":
ret += "int64"
case "qulonglong", "quint64":
ret += "uint64"
case "float":
ret += "float32"
case "double", "qreal":
ret += "float64"
case "qsizetype":
if C.sizeof_size_t == 4 {
ret += "uint32"
} else {
ret += "uint64"
}
2024-08-18 07:01:36 +00:00
case "qintptr":
var ptr *int
if unsafe.Sizeof(ptr) == 8 {
ret += "uint64"
} else {
ret += "uint32"
}
default:
// Do not transform this type
ret += p.ParameterType
}
return ret // ignore const
}
func (p CppParameter) parameterTypeCgo() string {
if p.ParameterType == "QString" {
return "C.char"
}
tmp := strings.Replace(p.RenderTypeCpp(), `*`, "", -1)
if strings.HasPrefix(tmp, "unsigned ") {
tmp = "u" + tmp[9:] // Cgo uses uchar, uint instead of full name
}
return "C." + strings.Replace(tmp, " ", "_", -1)
}
func emitParametersGo(params []CppParameter) string {
2024-08-06 02:29:12 +00:00
tmp := make([]string, 0, len(params))
skipNext := false
for i, p := range params {
if IsArgcArgv(params, i) {
skipNext = true
tmp = append(tmp, "args []string")
} else if skipNext {
// Skip this parameter, already handled
} else {
// Ordinary parameter
tmp = append(tmp, p.ParameterName+" "+p.RenderTypeGo())
}
2024-08-06 02:29:12 +00:00
}
return strings.Join(tmp, ", ")
}
func emitParametersGo2CABIForwarding(m CppMethod) (preamble string, fowarding string) {
tmp := make([]string, 0, len(m.Parameters)+2)
if !m.IsStatic {
tmp = append(tmp, "this.h")
}
skipNext := false
for i, p := range m.Parameters {
if IsArgcArgv(m.Parameters, i) {
skipNext = true
// QApplication constructor. Convert 'args' into Qt's wanted types
// Qt has a warning in the docs saying these pointers must be valid
// for the entire lifetype of QApplication, so, malloc + never free
preamble += "// Convert []string to long-lived int& argc, char** argv, never call free()\n"
preamble += "argc := (*C.int)(C.malloc(8))\n"
preamble += "*argc = len(args)\n"
preamble += "argv := (*[0xffff]*C.char)(C.malloc(c.ulong(8 * len(args))))\n"
preamble += "for i := range args {\n"
preamble += "argv[i] = C.CString(" + p.ParameterName + "[i])\n"
preamble += "}\n"
tmp = append(tmp, "argc, argv")
} else if skipNext {
// Skip this parameter, already handled
skipNext = false
} else if p.ParameterType == "QString" {
// Go: convert string -> char* and len
// CABI: convert char* and len -> real QString
preamble += p.ParameterName + "_Cstring := C.CString(" + p.ParameterName + ")\n"
preamble += "defer C.free(" + p.ParameterName + "_Cstring)\n"
tmp = append(tmp, p.ParameterName+"_Cstring, len("+p.ParameterName+")")
2024-08-11 04:37:18 +00:00
} else if listType, ok := p.QListOf(); ok {
// QList<T>
// Go: convert T[] -> t* and len
// CABI: create a real QList<>
2024-08-15 07:50:30 +00:00
if listType.ParameterType == "QString" {
// Combo
2024-08-11 04:37:18 +00:00
2024-08-15 07:50:30 +00:00
preamble += "// For the C ABI, malloc two C arrays; raw char* pointers and their lengths\n"
preamble += p.ParameterName + "_CArray := (*[0xffff]*" + listType.parameterTypeCgo() + ")(C.malloc(c.ulong(8 * len(" + p.ParameterName + "))))\n"
preamble += p.ParameterName + "_Lengths := (*[0xffff]*C.size_t)(C.malloc(c.ulong(8 * len(" + p.ParameterName + "))))\n"
2024-08-15 07:50:30 +00:00
preamble += "defer C.free(" + p.ParameterName + "_CArray)\n"
preamble += "defer C.free(" + p.ParameterName + "_Lengths)\n"
preamble += "for i := range " + p.ParameterName + "{\n"
preamble += "single_cstring := C.CString(" + p.ParameterName + "[i])\n"
preamble += "defer C.free(single_cstring)\n"
preamble += p.ParameterName + "_CArray[i] = single_cstring\n"
preamble += p.ParameterName + "__Lengths[i] = len(" + p.ParameterName + "[i])\n"
preamble += "}\n"
2024-08-11 04:37:18 +00:00
2024-08-15 07:50:30 +00:00
tmp = append(tmp, p.ParameterName+"_CArray, "+p.ParameterName+"_Lengths, len("+p.ParameterName+")")
} else {
preamble += "// For the C ABI, malloc a C array of raw pointers\n"
preamble += p.ParameterName + "_CArray := (*[0xffff]*" + listType.parameterTypeCgo() + ")(C.malloc(c.ulong(8 * len(" + p.ParameterName + "))))\n"
2024-08-15 07:50:30 +00:00
preamble += "defer C.free(" + p.ParameterName + "_CArray)\n"
preamble += "for i := range " + p.ParameterName + "{\n"
preamble += p.ParameterName + "_CArray[i] = " + p.ParameterName + "[i].cPointer()\n"
preamble += "}\n"
tmp = append(tmp, p.ParameterName+"_CArray, len("+p.ParameterName+")")
}
2024-08-11 04:37:18 +00:00
} else if p.Pointer && p.ParameterType == "char" {
// Single char* argument
preamble += p.ParameterName + "_Cstring := C.CString(" + p.ParameterName + ")\n"
preamble += "defer C.free(" + p.ParameterName + "_Cstring)\n"
tmp = append(tmp, p.ParameterName+"_Cstring")
2024-08-08 07:06:14 +00:00
} else if (p.Pointer || p.ByRef) && p.QtClassType() {
// The C++ type is a pointer to Qt class
// We want our functions to accept the Go wrapper type, and forward as cPointer()
tmp = append(tmp, p.ParameterName+".cPointer()")
} else {
// Default
tmp = append(tmp, p.ParameterName)
}
}
if m.ReturnType.ParameterType == "QString" {
tmp = append(tmp, "&_out, &_out_Strlen")
2024-08-11 04:37:18 +00:00
} else if _, ok := m.ReturnType.QListOf(); ok {
tmp = append(tmp, "&_out, &_out_len")
}
return preamble, strings.Join(tmp, ", ")
}
func emitGo(src *CppParsedHeader, headerName string) (string, error) {
2024-08-06 02:29:12 +00:00
ret := strings.Builder{}
ret.WriteString(`package qt
2024-08-06 02:29:12 +00:00
/*
2024-08-06 02:29:12 +00:00
#cgo CFLAGS: -fPIC
#cgo pkg-config: Qt5Widgets
#include "gen_` + headerName + `"
#include <stdlib.h>
2024-08-06 02:29:12 +00:00
*/
import "C"
2024-08-06 02:29:12 +00:00
%%_IMPORTLIBS_%%
2024-08-06 02:29:12 +00:00
`)
imports := map[string]struct{}{}
for _, c := range src.Classes {
2024-08-06 02:29:12 +00:00
ret.WriteString(`
type ` + c.ClassName + ` struct {
h *C.` + c.ClassName + `
2024-08-10 00:54:26 +00:00
`)
// Embed all inherited types to directly allow calling inherited methods
for _, base := range c.Inherits {
ret.WriteString(base + "\n")
}
ret.WriteString(`
}
func (this *` + c.ClassName + `) cPointer() *C.` + c.ClassName + ` {
if this == nil {
return nil
2024-08-06 02:29:12 +00:00
}
return this.h
2024-08-06 02:29:12 +00:00
}
`)
2024-08-06 02:29:12 +00:00
2024-08-10 00:54:26 +00:00
localInit := "h: ret"
for _, base := range c.Inherits {
localInit += ", " + base + ": " + base + "{h: ret}"
}
ret.WriteString(`
func new` + c.ClassName + `(h *C.` + c.ClassName + `) {
return &` + c.ClassName + `{` + localInit + `}
}
`)
for i, ctor := range c.Ctors {
preamble, forwarding := emitParametersGo2CABIForwarding(ctor)
ret.WriteString(`
// New` + c.ClassName + maybeSuffix(i) + ` constructs a new ` + c.ClassName + ` object.
func New` + c.ClassName + maybeSuffix(i) + `(` + emitParametersGo(ctor.Parameters) + `) {
` + preamble + ` ret := C.` + c.ClassName + `_new` + maybeSuffix(i) + `(` + forwarding + `)
return new` + c.ClassName + `(h)
}
`)
2024-08-06 02:29:12 +00:00
}
for _, m := range c.Methods {
preamble, forwarding := emitParametersGo2CABIForwarding(m)
2024-08-06 02:29:12 +00:00
shouldReturn := "return "
afterword := ""
returnTypeDecl := m.ReturnType.RenderTypeGo() // FIXME handle byRef/const here too
if m.ReturnType.ParameterType == "void" && !m.ReturnType.Pointer {
shouldReturn = ""
returnTypeDecl = ""
2024-08-06 02:29:12 +00:00
} else if m.ReturnType.ParameterType == "void" && m.ReturnType.Pointer {
returnTypeDecl = "interface{}"
} else if m.ReturnType.ParameterType == "char" && m.ReturnType.Pointer {
// Qt functions normally return QString - anything returning char*
// is something like QByteArray.Data() where it returns an unsafe
// internal pointer
imports["unsafe"] = struct{}{}
returnTypeDecl = "unsafe.Pointer"
} else if m.ReturnType.ParameterType == "QString" {
shouldReturn = ""
returnTypeDecl = "string"
preamble += "var _out *C.char = nil\n"
preamble += "var _out_Strlen C.size_t = 0\n"
afterword += "ret := C.GoStringN(_out, _out_Strlen)\n"
afterword += "C.free(_out)\n"
afterword += "return ret"
2024-08-11 04:37:18 +00:00
} else if t, ok := m.ReturnType.QListOf(); ok {
shouldReturn = ""
returnTypeDecl = "[]" + t.RenderTypeGo()
preamble += "var _out **" + t.parameterTypeCgo() + " = nil\n"
2024-08-11 04:37:18 +00:00
preamble += "var _out_len C.size_t = 0\n"
afterword += "ret := make([]" + t.RenderTypeGo() + ", _out_Strlen)\n"
afterword += "for i := 0; i < _out_len; i++ {\n"
afterword += "ret[i] = new" + t.ParameterType + "(_out[i])\n"
afterword += "}\n"
afterword += "C.free(_out)\n"
} else if m.ReturnType.QtClassType() {
// Construct our Go type based on this inner CABI type
shouldReturn = "ret := "
if m.ReturnType.Pointer {
afterword = "return " + m.ReturnType.ParameterType + "{h: ret}"
} else {
// This is return by value, but CABI has new'd it into a
// heap type for us
// To preserve Qt's approximate semantics, add a runtime
// finalizer to automatically Delete once the type goes out
// of Go scope
imports["runtime"] = struct{}{}
afterword = "// Qt uses pass-by-value semantics for this type. Mimic with finalizer\n"
afterword += "ret1 := &" + m.ReturnType.ParameterType + "{h: ret}\n"
afterword += "runtime.SetFinalizer(ret1, func(ret2 *" + m.ReturnType.ParameterType + ") {\n"
afterword += "ret2.Delete()\n"
afterword += "runtime.KeepAlive(ret2.h)\n"
afterword += "})\n"
afterword += "return ret1"
}
}
receiverAndMethod := `(this *` + c.ClassName + `) ` + m.SafeMethodName()
if m.IsStatic {
receiverAndMethod = c.ClassName + `_` + m.SafeMethodName()
}
ret.WriteString(`
func ` + receiverAndMethod + `(` + emitParametersGo(m.Parameters) + `) ` + returnTypeDecl + ` {
` + preamble +
shouldReturn + ` C.` + c.ClassName + `_` + m.SafeMethodName() + `(` + forwarding + `)
` + afterword + `}
`)
// Add Connect() wrappers for signal functions
if m.IsSignal && !m.HasHiddenParams {
imports["unsafe"] = struct{}{}
imports["runtime/cgo"] = struct{}{}
ret.WriteString(`func (this *` + c.ClassName + `) On` + m.SafeMethodName() + `(slot func()) {
var slotWrapper miqtCallbackFunc = func(argc C.int, args *C.void) {
slot()
}
C.` + c.ClassName + `_connect_` + m.SafeMethodName() + `(this.h, unsafe.Pointer(uintptr(cgo.NewHandle(slotWrapper))))
}
`)
}
}
2024-08-06 02:29:12 +00:00
if AllowDelete(c) {
ret.WriteString(`
func (this *` + c.ClassName + `) Delete() {
C.` + c.ClassName + `_Delete(this.h)
}
`)
2024-08-09 22:32:57 +00:00
}
}
2024-08-06 02:29:12 +00:00
goSrc := ret.String()
// Fixup imports
if len(imports) > 0 {
allImports := make([]string, 0, len(imports))
for k, _ := range imports {
allImports = append(allImports, `"`+k+`"`)
}
sort.Strings(allImports)
goSrc = strings.Replace(goSrc, `%%_IMPORTLIBS_%%`, "import (\n\t"+strings.Join(allImports, "\n\t")+"\n)", 1)
} else {
goSrc = strings.Replace(goSrc, `%%_IMPORTLIBS_%%`, "", 1)
}
// Run gofmt over the result
formattedSrc, err := format.Source([]byte(goSrc))
if err != nil {
log.Printf("gofmt failure: %v", err)
formattedSrc = []byte(goSrc)
2024-08-06 02:29:12 +00:00
}
return string(formattedSrc), nil
2024-08-06 01:03:23 +00:00
}