avo/internal/cmd/avogen/main.go

// Command avogen generates code based on the instruction database.
package main

import (
	"flag"
	"fmt"
	"go/build"
	"log"
	"os"
	"path/filepath"

	"sources.truenas.cloud/code/avo/internal/gen"
	"sources.truenas.cloud/code/avo/internal/inst"
	"sources.truenas.cloud/code/avo/internal/load"
	"sources.truenas.cloud/code/avo/printer"
)

var generators = map[string]gen.Builder{
	"asmtest":     gen.NewAsmTest,
	"godata":      gen.NewGoData,
	"godatatest":  gen.NewGoDataTest,
	"optab":       gen.NewOptab,
	"ctors":       gen.NewCtors,
	"ctorstest":   gen.NewCtorsTest,
	"ctorsstress": gen.NewCtorsStress,
	"ctorsbench":  gen.NewCtorsBench,
	"build":       gen.NewBuild,
	"buildtest":   gen.NewBuildTest,
	"mov":         gen.NewMOV,
}

// Command-line flags.
var (
	bootstrap = flag.Bool("bootstrap", false, "regenerate instruction list from original data")
	datadir   = flag.String(
		"data",
		filepath.Join(build.Default.GOPATH, "src/sources.truenas.cloud/code/avo/internal/data"),
		"path to data directory",
	)
	output = flag.String("output", "", "path to output file (default stdout)")
)

func main() {
	log.SetPrefix("avogen: ")
	log.SetFlags(0)
	flag.Parse()
	if err := run(); err != nil {
		log.Fatal(err)
	}
}

func run() error {
	// Build generator.
	t := flag.Arg(0)
	builder := generators[t]
	if builder == nil {
		return fmt.Errorf("unknown generator type '%s'", t)
	}

	g := builder(printer.NewArgvConfig())

	// Determine output writer.
	w := os.Stdout
	if *output != "" {
		f, err := os.Create(*output)
		if err != nil {
			return err
		}
		defer f.Close()
		w = f
	}

	// Load instructions.
	is := inst.Instructions
	if *bootstrap {
		log.Printf("bootstrap: loading instructions from data directory %s", *datadir)
		l := load.NewLoaderFromDataDir(*datadir)
		r, err := l.Load()
		if err != nil {
			return err
		}
		is = r
	}

	// Generate output.
	b, generr := g.Generate(is)

	// Write.
	if _, err := w.Write(b); err != nil {
		return err
	}

	if generr != nil {
		return generr
	}

	return nil
}