gen.go

//go:build ignore
// +build ignore

package main

import (
	"bytes"
	"go/format"
	"os"
	"strings"
	"text/template"
)

type Dtype struct {
	TypeU    string
	TypeL    string
	TypeCode string
}

var (
	Ftypes = []Dtype{
		{"Complex128", "complex128", "c16"},
		{"Complex64", "complex64", "c8"},
		{"Float64", "float64", "f8"},
		{"Float32", "float32", "f4"},
		{"Uint64", "uint64", "u8"},
		{"Uint32", "uint32", "u4"},
		{"Uint16", "uint16", "u2"},
		{"Uint8", "uint8", "u1"},
		{"Int64", "int64", "i8"},
		{"Int32", "int32", "i4"},
		{"Int16", "int16", "i2"},
		{"Int8", "int8", "i1"},
		{"Bytes", "byte", "i1"},
	}
)

func main() {

	tmplFname := os.Args[1]

	tmpl, err := template.ParseFiles(tmplFname)
	if err != nil {
		panic(err)
	}

	var buf bytes.Buffer
	err = tmpl.Execute(&buf, Ftypes)
	if err != nil {
		panic(err)
	}

	var p []byte
	p, err = format.Source(buf.Bytes())
	if err != nil {
		panic(err)
	}

	var outfile string
	if strings.Contains(tmplFname, "test") {
		outfile = strings.Replace(tmplFname, "_test.template", "_gen_test.go", 1)
	} else {
		outfile = strings.Replace(tmplFname, ".template", "_gen.go", 1)
	}
	if tmplFname == outfile {
		panic("Can't overwrite template file\n")
	}

	out, err := os.Create(outfile)
	if err != nil {
		panic(err)
	}

	out.WriteString("// GENERATED CODE, DO NOT EDIT\n\n")
	_, err = out.Write(p)
	if err != nil {
		panic(err)
	}

	out.Close()
}