99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
// Command avogen generates code based on the instruction database.
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"go/build"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/mmcloughlin/avo/internal/gen"
|
|
"github.com/mmcloughlin/avo/internal/inst"
|
|
"github.com/mmcloughlin/avo/internal/load"
|
|
"github.com/mmcloughlin/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/github.com/mmcloughlin/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
|
|
}
|