1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
|
package gen
import (
"bufio"
"reflect"
"strings"
"github.com/mmcloughlin/addchain/acc/ir"
"github.com/mmcloughlin/addchain/acc/printer"
)
// Function is a function provided to templates.
type Function struct {
Name string
Description string
Func interface{}
}
// Signature returns the function signature.
func (f *Function) Signature() string {
return reflect.ValueOf(f.Func).Type().String()
}
// Functions is the list of functions provided to templates.
var Functions = []*Function{
{
Name: "add",
Description: "If the input operation is an `ir.Add` then return it, otherwise return `nil`",
Func: func(op ir.Op) ir.Op {
if a, ok := op.(ir.Add); ok {
return a
}
return nil
},
},
{
Name: "double",
Description: "If the input operation is an `ir.Double` then return it, otherwise return `nil`",
Func: func(op ir.Op) ir.Op {
if d, ok := op.(ir.Double); ok {
return d
}
return nil
},
},
{
Name: "shift",
Description: "If the input operation is an `ir.Shift` then return it, otherwise return `nil`",
Func: func(op ir.Op) ir.Op {
if s, ok := op.(ir.Shift); ok {
return s
}
return nil
},
},
{
Name: "inc",
Description: "Increment an integer",
Func: func(n int) int { return n + 1 },
},
{
Name: "format",
Description: "Formats an addition chain script (`*ast.Chain`) as a string",
Func: printer.String,
},
{
Name: "split",
Description: "Calls `strings.Split`",
Func: strings.Split,
},
{
Name: "join",
Description: "Calls `strings.Join`",
Func: strings.Join,
},
{
Name: "lines",
Description: "Split input string into lines",
Func: func(s string) []string {
var lines []string
scanner := bufio.NewScanner(strings.NewReader(s))
for scanner.Scan() {
lines = append(lines, scanner.Text())
}
return lines
},
},
}
|