init: generator framework

gen/main.go

@@ -0,0 +1,78 @@
+package main
+
+import (
+	"archive/zip"
+	"bufio"
+	"bytes"
+	"flag"
+	"fmt"
+	"io"
+	"io/fs"
+	"net/http"
+	"os"
+	"strings"
+)
+
+func main() {
+	spec := flag.String("spec", "v1.28.2", "The l0 loader spec version tag starting with v or a local level-zero path for dev.")
+	flag.Parse()
+
+	var specdir fs.FS
+
+	if strings.HasPrefix(*spec, "v") {
+		ver := (*spec)[1:]
+		u := fmt.Sprintf("https://github.com/oneapi-src/level-zero/releases/download/v%s/level-zero-win-sdk-%s.zip", ver, ver)
+		resp, err := http.Get(u)
+		if err != nil {
+			panic(err)
+		}
+		data, err := io.ReadAll(resp.Body)
+		_ = resp.Body.Close()
+		if err != nil {
+			panic(err)
+		}
+		r, err := zip.NewReader(bytes.NewReader(data), int64(len(data)))
+		if err != nil {
+			panic(err)
+		}
+		specdir = r
+	} else {
+		specdir = os.DirFS(*spec)
+	}
+
+	fmt.Println("[main] parsing core APIs...")
+	f, err := specdir.Open("include/level_zero/ze_api.h")
+	if err != nil {
+		panic(err)
+	}
+	scanheader("core", bufio.NewScanner(f))
+	_ = f.Close()
+	fmt.Println("[main] finish parsing core")
+
+	fmt.Println("[main] parsing runtime APIs...")
+	f, err = specdir.Open("include/level_zero/zer_api.h")
+	if err != nil {
+		panic(err)
+	}
+	scanheader("rntm", bufio.NewScanner(f))
+	_ = f.Close()
+	fmt.Println("[main] finish parsing runtime")
+
+	fmt.Println("[main] parsing sysman APIs...")
+	f, err = specdir.Open("include/level_zero/zes_api.h")
+	if err != nil {
+		panic(err)
+	}
+	scanheader("sysm", bufio.NewScanner(f))
+	_ = f.Close()
+	fmt.Println("[main] finish parsing sysman")
+
+	fmt.Println("[main] parsing tools APIs...")
+	f, err = specdir.Open("include/level_zero/zet_api.h")
+	if err != nil {
+		panic(err)
+	}
+	scanheader("tols", bufio.NewScanner(f))
+	_ = f.Close()
+	fmt.Println("[main] finish parsing tools")
+}

gen/scan.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"path"
+	"strings"
+)
+
+var infhcache = make(map[string]string, 1024)
+
+func infh(name string) string {
+	if s, ok := infhcache[name]; ok {
+		return s
+	}
+	s := "  [" + name + "]"
+	infhcache[name] = s
+	return s
+}
+
+func scanheader(name string, scan *bufio.Scanner) {
+	ln := 0
+	var regionfile *os.File
+	for scan.Scan() {
+		ln++
+		t := scan.Text()
+		switch {
+		case strings.HasPrefix(t, "#pragma region "):
+			region := strings.TrimSpace(t[15:])
+			if region == "" {
+				panic(fmt.Sprintf("%s L%d: unexpected empty region", name, ln))
+			}
+			fmt.Println(infh(name), "scanning region", region)
+			f, err := os.Create(path.Join(name, region+".go"))
+			if err != nil {
+				panic(fmt.Sprintf("%s L%d: cannot create region %s, err: %v", name, ln, region, err))
+			}
+			f.WriteString("// Code generated by gen. DO NOT EDIT.\n\npackage ")
+			f.WriteString(name)
+			f.WriteString("\n")
+			regionfile = f
+		case strings.HasPrefix(t, "#pragma endregion"):
+			fmt.Println(infh(name), "close region", regionfile.Name())
+			_ = regionfile.Close()
+			regionfile = nil
+		}
+	}
+}