65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
python "github.com/DataDog/go-python3"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
output = flag.String("o", "variadic.c", "output file")
|
||
|
|
caseNumber = flag.Int("n", python.MaxVariadicLength, "number of case in the switch statement")
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
out, err := os.Create(*output)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error opening file %s: %s", *output, err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
defer out.Close()
|
||
|
|
|
||
|
|
_, err = out.WriteString("#include \"Python.h\"\n\n")
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error writing to file %s: %s", *output, err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
out.WriteString(renderTemplate(*caseNumber, "PyObject_CallFunctionObjArgs", "callable"))
|
||
|
|
out.WriteString(renderTemplate(*caseNumber, "PyObject_CallMethodObjArgs", "obj", "name"))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func renderTemplate(n int, functionName string, pyArgs ...string) string {
|
||
|
|
template :=
|
||
|
|
`PyObject* _go_%s(%sint argc, PyObject **argv) {
|
||
|
|
|
||
|
|
PyObject *result = NULL;
|
||
|
|
switch (argc) {
|
||
|
|
%s }
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
`
|
||
|
|
args := ""
|
||
|
|
for _, arg := range pyArgs {
|
||
|
|
args += fmt.Sprintf("PyObject *%s, ", arg)
|
||
|
|
}
|
||
|
|
switchTemplate := ""
|
||
|
|
for i := 0; i < n+1; i++ {
|
||
|
|
switchTemplate += fmt.Sprintf(" case %d:\n", i)
|
||
|
|
args := ""
|
||
|
|
for _, arg := range pyArgs {
|
||
|
|
args += fmt.Sprintf("%s, ", arg)
|
||
|
|
}
|
||
|
|
for j := 0; j < i; j++ {
|
||
|
|
args += fmt.Sprintf("argv[%d], ", j)
|
||
|
|
}
|
||
|
|
args += "NULL"
|
||
|
|
switchTemplate += fmt.Sprintf(" return %s(%s);\n", functionName, args)
|
||
|
|
}
|
||
|
|
return fmt.Sprintf(template, functionName, args, switchTemplate)
|
||
|
|
}
|