diff --git a/generateGui.go b/generateGui.go index d5e0d2f..8ddba01 100644 --- a/generateGui.go +++ b/generateGui.go @@ -96,6 +96,18 @@ func guiMain(w io.Writer, FRUITS string, FRUIT string) { fmt.Fprintln(w, " return sf") fmt.Fprintln(w, "}") + fmt.Fprintln(w, "func (sf *"+FRUIT+"StringFunc) SetTitle(title string) {") + fmt.Fprintln(w, " sf.title = title") + fmt.Fprintln(w, "}") + + fmt.Fprintln(w, "func (sf *"+FRUIT+"IntFunc) SetTitle(title string) {") + fmt.Fprintln(w, " sf.title = title") + fmt.Fprintln(w, "}") + + fmt.Fprintln(w, "func (sf *"+FRUIT+"TimeFunc) SetTitle(title string) {") + fmt.Fprintln(w, " sf.title = title") + fmt.Fprintln(w, "}") + fmt.Fprintln(w, "func (t *"+FRUITS+"Table) AddIntFunc(title string, f func(*"+FRUIT+") int) *"+FRUIT+"IntFunc {") fmt.Fprintln(w, " t.pb.Order = append(t.pb.Order, title)") fmt.Fprintln(w, "") @@ -234,12 +246,13 @@ func guiStringFuncs(w io.Writer, ZOOPB string, FRUITS string, FRUIT string) { func guiAddStringFunc(w io.Writer, FRUITS string, FRUIT string, BRAND string) { fmt.Fprintln(w, "") - fmt.Fprintln(w, "func (t *"+FRUITS+"Table) Add"+BRAND+"() {") + fmt.Fprintln(w, "func (t *"+FRUITS+"Table) Add"+BRAND+"() *"+FRUIT+"StringFunc {") fmt.Fprintln(w, " // t.pb.Order = append(t.pb.Order, "+BRAND+")") fmt.Fprintln(w, "") - fmt.Fprintln(w, " t.AddStringFunc(\""+BRAND+"\", func(m *"+FRUIT+") string {") + fmt.Fprintln(w, " sf := t.AddStringFunc(\""+BRAND+"\", func(m *"+FRUIT+") string {") fmt.Fprintln(w, " return m."+BRAND+"") fmt.Fprintln(w, " })") + fmt.Fprintln(w, " return sf") fmt.Fprintln(w, "}") /* fmt.Fprintln(w, "func (t *"+FRUITS+"Table) AddMemory() {")