codegen presets: add cgo preamble

AllenDang · Nov 16, 2024 · a838dac · a838dac
1 parent af475a3
commit a838dac
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 8 deletions.
diff --git a/cmd/codegen/cimgui-go-preset.json b/cmd/codegen/cimgui-go-preset.json
@@ -73,5 +73,8 @@
         ],
         "OriginReplace": {
                 "TextEditor_GetText": "TextEditor_GetText_alloc"
-        }
+        },
+        "ExtraCGOPreamble": [
+                "#include \"../imgui/extra_types.h\""
+        ]
 }
diff --git a/cmd/codegen/gengo_callbacks.go b/cmd/codegen/gengo_callbacks.go
@@ -457,10 +457,10 @@ func (g *callbacksGenerator) saveToDisk() error {
 	fmt.Fprintf(result,
 		`// #include <stdlib.h>
 // #include <memory.h>
-// #include "../imgui/extra_types.h"
 // #include "wrapper.h"
 // #include "typedefs.h"
-`)
+%s
+`, g.ctx.preset.MergeCGoPreamble())
 
 	fmt.Fprintf(result, g.cgoSb.String())
 

diff --git a/cmd/codegen/gengo_funcs.go b/cmd/codegen/gengo_funcs.go
@@ -44,6 +44,7 @@ func GenerateGoFuncs(
 		enumNames:     context.enumNames,
 		refTypedefs:   context.refTypedefs,
 		context:       context,
+		sb:            &strings.Builder{},
 	}
 
 	generator.writeFuncsFileHeader()
@@ -105,7 +106,7 @@ type goFuncsGenerator struct {
 	enumNames     map[CIdentifier]bool
 	refTypedefs   map[CIdentifier]bool
 
-	sb                 strings.Builder
+	sb                 *strings.Builder
 	convertedFuncCount int
 	shouldGenerate     bool
 
@@ -116,14 +117,14 @@ type goFuncsGenerator struct {
 func (g *goFuncsGenerator) writeFuncsFileHeader() {
 	g.sb.WriteString(getGoPackageHeader(g.context))
 
-	g.sb.WriteString(
-		`// #include "../imgui/extra_types.h"
-// #include "structs_accessor.h"
+	fmt.Fprintf(g.sb,
+		`// #include "structs_accessor.h"
 // #include "wrapper.h"
+%s
 import "C"
 import "unsafe"
 
-`)
+`, g.context.preset.MergeCGoPreamble())
 }
 
 func (g *goFuncsGenerator) GenerateFunction(f FuncDef, args []GoIdentifier, argWrappers []ArgumentWrapperData) (noErrors bool) {

diff --git a/cmd/codegen/presets.go b/cmd/codegen/presets.go
@@ -30,4 +30,17 @@ type Preset struct {
 	// Introduced to replace TextEditor_GetText -> TextEditor_GetText_alloc
 	// but could be re-used to force use of another function than json tells us to use.
 	OriginReplace map[CIdentifier]CIdentifier
+	// ExtraCGOPreamble allows to specify additional C code elements in Go files.
+	// For example could be used to include extra files.
+	// For ease of use, its in form of []string. These lines will be merged and prefixed (if appliable) with '//'
+	ExtraCGOPreamble []string
+}
+
+func (p *Preset) MergeCGoPreamble() string {
+	result := ""
+	for _, line := range p.ExtraCGOPreamble {
+		result += "// " + line + "\n"
+	}
+
+	return result
 }