| // Copyright 2021 The Wuffs Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or |
| // https://www.apache.org/licenses/LICENSE-2.0> or the MIT license |
| // <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your |
| // option. This file may not be copied, modified, or distributed |
| // except according to those terms. |
| // |
| // SPDX-License-Identifier: Apache-2.0 OR MIT |
| |
| //go:build ignore |
| // +build ignore |
| |
| package main |
| |
| // gen.go generates data.go. |
| // |
| // Invoke it via "go generate". |
| |
| import ( |
| "bytes" |
| "fmt" |
| "go/format" |
| "os" |
| "strings" |
| |
| ani "github.com/google/wuffs/lib/armneonintrinsics" |
| ) |
| |
| func main() { |
| if err := main1(); err != nil { |
| os.Stderr.WriteString(err.Error() + "\n") |
| os.Exit(1) |
| } |
| } |
| |
| func main1() error { |
| out.WriteString(header) |
| for i, aniType := range [...]ani.Type{ |
| ani.TypeUint8x8, |
| ani.TypeUint16x4, |
| ani.TypeUint32x2, |
| ani.TypeUint64x1, |
| ani.TypeUint8x16, |
| ani.TypeUint16x8, |
| ani.TypeUint32x4, |
| ani.TypeUint64x2, |
| } { |
| if i != 0 { |
| out.WriteByte('\n') |
| } |
| if err := do(aniType); err != nil { |
| return err |
| } |
| } |
| out.WriteString(footer) |
| |
| formatted, err := format.Source(out.Bytes()) |
| if err != nil { |
| return err |
| } |
| return os.WriteFile("data.go", formatted, 0644) |
| } |
| |
| var ( |
| out bytes.Buffer |
| ) |
| |
| func do(aniType ani.Type) error { |
| wtnRecv := wuffsTypeNames[aniType] |
| outer: |
| for f := ani.FirstFunction(); f.IsValid(); f = f.NextFunction() { |
| if f.FirstArgument().Type() != aniType { |
| continue |
| } |
| fName := f.Name() |
| if strings.HasPrefix(fName, "vreinterpret_") || |
| strings.HasPrefix(fName, "vreinterpretq_") { |
| continue |
| } |
| |
| for arg := f.FirstArgument(); arg.IsValid(); arg = arg.NextArgument() { |
| if wuffsTypeNames[arg.Type()] == "" { |
| fmt.Fprintf(&out, "// %s.%s(etc)\n", wtnRecv, fName) |
| continue outer |
| } |
| } |
| |
| laney := strings.Contains(fName, "_lane") |
| fmt.Fprintf(&out, "\"%s.%s(", wtnRecv, fName) |
| |
| prevArgType := ani.Type(0) |
| for i, arg := 0, f.FirstArgument(); arg.IsValid(); i, arg = i+1, arg.NextArgument() { |
| wtnArg := wuffsTypeNames[arg.Type()] |
| refinement := "" |
| if laney && (wtnArg[0] != 'a') { |
| refinement = refinements[prevArgType] |
| } |
| |
| prevArgType = arg.Type() |
| if i == 0 { |
| continue |
| } else if i > 1 { |
| out.WriteString(", ") |
| } |
| fmt.Fprintf(&out, "%s: %v%s", |
| stripUnderscores(arg.Name()), |
| wuffsTypeNames[arg.Type()], |
| refinement) |
| } |
| out.WriteString(") ") |
| out.WriteString(wuffsTypeNames[f.ReturnType()]) |
| out.WriteString("\",\n") |
| } |
| return nil |
| } |
| |
| func stripUnderscores(s string) string { |
| for (s != "") && (s[0] == '_') { |
| s = s[1:] |
| } |
| return s |
| } |
| |
| var refinements = [256]string{ |
| ani.TypeUint8x8: "[..= 7]", |
| ani.TypeUint16x4: "[..= 3]", |
| ani.TypeUint32x2: "[..= 1]", |
| ani.TypeUint64x1: "[..= 0]", |
| |
| ani.TypeUint8x16: "[..= 15]", |
| ani.TypeUint16x8: "[..= 7]", |
| ani.TypeUint32x4: "[..= 3]", |
| ani.TypeUint64x2: "[..= 1]", |
| } |
| |
| var wuffsTypeNames = [256]string{ |
| ani.TypeInt8: "u8", |
| ani.TypeInt16: "u16", |
| ani.TypeInt32: "u32", |
| ani.TypeInt64: "u64", |
| ani.TypeUint8: "u8", |
| ani.TypeUint16: "u16", |
| ani.TypeUint32: "u32", |
| ani.TypeUint64: "u64", |
| ani.TypeConstInt8: "u8", |
| ani.TypeConstInt16: "u16", |
| ani.TypeConstInt32: "u32", |
| ani.TypeConstInt64: "u64", |
| ani.TypeConstUint8: "u8", |
| ani.TypeConstUint16: "u16", |
| ani.TypeConstUint32: "u32", |
| ani.TypeConstUint64: "u64", |
| |
| ani.TypeUint8x8: "arm_neon_u8x8", |
| ani.TypeUint16x4: "arm_neon_u16x4", |
| ani.TypeUint32x2: "arm_neon_u32x2", |
| ani.TypeUint64x1: "arm_neon_u64x1", |
| |
| ani.TypeUint8x16: "arm_neon_u8x16", |
| ani.TypeUint16x8: "arm_neon_u16x8", |
| ani.TypeUint32x4: "arm_neon_u32x4", |
| ani.TypeUint64x2: "arm_neon_u64x2", |
| } |
| |
| const header = `// Code generated by running "go generate". DO NOT EDIT. |
| |
| // Copyright 2021 The Wuffs Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or |
| // https://www.apache.org/licenses/LICENSE-2.0> or the MIT license |
| // <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your |
| // option. This file may not be copied, modified, or distributed |
| // except according to those terms. |
| // |
| // SPDX-License-Identifier: Apache-2.0 OR MIT |
| |
| package builtin |
| |
| var funcsARMNeon = [...]string { |
| ` |
| |
| const footer = "}\n" |