uic: recursive emit basic go implementation

cmd/miqt-uic/types.go

@@ -76,7 +76,7 @@ type UiFile struct {
 
 	Class       string        `xml:"class"`
 	Version     string        `xml:"version,attr"` // e.g. 4.0
-	Widget      []UiWidget    `xml:"widget"`
+	Widget      UiWidget      `xml:"widget"`       // There's only one root widget
 	Resources   UiResources   `xml:"resources"`
 	Connections UiConnections `xml:"connections"`
 }

cmd/miqt-uic/ui2go.go

@@ -1,7 +1,9 @@
 package main
 
 import (
+	"fmt"
 	"go/format"
+	"strconv"
 	"strings"
 )
 
@@ -14,6 +16,7 @@ func collectClassNames_Widget(u UiWidget) []string {
 		ret = append(ret, collectClassNames_Widget(w)...)
 	}
 	if u.Layout != nil {
+		ret = append(ret, u.Layout.Name+" *qt."+u.Layout.Class)
 		for _, li := range u.Layout.Items {
 			ret = append(ret, collectClassNames_Widget(li.Widget)...)
 		}
@@ -24,12 +27,153 @@ func collectClassNames_Widget(u UiWidget) []string {
 	return ret
 }
 
-func collectClassNames(u UiFile) []string {
+func generateString(s *UiString, parentName string) string {
-	var ret []string
+	if s.Notr || parentName == "" {
-	for _, w := range u.Widget {
+		return strconv.Quote(s.Value)
-		ret = append(ret, collectClassNames_Widget(w)...)
 	}
-	return ret
+	return parentName + `.Tr(` + strconv.Quote(s.Value) + `)`
+}
+
+func generateWidget(w UiWidget, parentName string) (string, error) {
+	ret := strings.Builder{}
+
+	ret.WriteString(`
+	ui.` + w.Name + ` = qt.New` + w.Class + `(` + parentName + `)
+	ui.` + w.Name + `.SetObjectName(` + strconv.Quote(w.Name) + `)
+	`)
+
+	// Properties
+	for _, prop := range w.Properties {
+		setterFunc := `.Set` + strings.ToUpper(string(prop.Name[0])) + prop.Name[1:]
+
+		if prop.Name == "geometry" {
+			if !(prop.RectVal.X == 0 && prop.RectVal.Y == 0) {
+				// Set all 4x properties
+				ret.WriteString(`ui.` + w.Name + `.SetGeometry(qt.NewQRect(` + fmt.Sprintf("%d, %d, %d, %d", prop.RectVal.X, prop.RectVal.Y, prop.RectVal.Width, prop.RectVal.Height) + "))\n")
+
+			} else if !(prop.RectVal.Width == 0 && prop.RectVal.Height == 0) {
+				// Only width/height were supplied
+				ret.WriteString(`ui.` + w.Name + `.Resize(` + fmt.Sprintf("%d, %d", prop.RectVal.Width, prop.RectVal.Height) + ")\n")
+
+			}
+
+		} else if prop.StringVal != nil {
+			//  "windowTitle", "title", "text"
+			ret.WriteString(`ui.` + w.Name + setterFunc + `(` + generateString(prop.StringVal, parentName) + ")\n")
+
+		} else if prop.EnumVal != nil {
+			// "frameShape"
+			ret.WriteString(`ui.` + w.Name + setterFunc + `(qt.` + strings.Replace(*prop.EnumVal, `::`, `__`, -1) + ")\n")
+
+		} else {
+			ret.WriteString("/* miqt-uic: no handler for property '" + prop.Name + "' */\n")
+		}
+	}
+
+	// Layout
+	if w.Layout != nil {
+		ret.WriteString(`
+		ui.` + w.Layout.Name + ` = qt.New` + w.Layout.Class + `(` + w.Name + `)
+		ui.` + w.Layout.Name + `.SetObjectName(` + strconv.Quote(w.Layout.Name) + `)
+		`)
+
+		for _, child := range w.Layout.Items {
+
+			// Layout items have the parent as the real QWidget parent and are
+			// separately assigned to the layout afterwards
+
+			nest, err := generateWidget(child.Widget, `ui.`+w.Name+`.QWidget`) // MIQT uses .QWidget to cast down
+			if err != nil {
+				return "", fmt.Errorf(w.Name+": %w", err)
+			}
+
+			ret.WriteString(nest)
+
+			// Assign to layout
+
+			switch w.Layout.Class {
+			case `QFormLayout`:
+				// Row and Column are always populated.
+				rowPos := fmt.Sprintf("%d", *child.Row)
+				var colPos string
+				if *child.Column == 0 {
+					colPos = `qt.QFormLayout__LabelRow`
+				} else if *child.Column == 1 {
+					colPos = `qt.QFormLayout__FieldRow`
+				} else {
+					ret.WriteString("/* miqt-uic: QFormLayout does not understand column index */\n")
+					continue
+				}
+
+				// For QFormLayout it's SetWidget
+				ret.WriteString(`
+				ui.` + w.Layout.Name + `.SetWidget(` + rowPos + `, ` + colPos + `, ui.` + child.Widget.Name + `)
+					`)
+
+			case `QGridLayout`:
+				// For QGridLayout it's AddWidget
+				// FIXME in Miqt this function has optionals, needs to be called with the correct arity
+				// TODO support rowSpan, columnSpan
+				ret.WriteString(`
+				ui.` + w.Layout.Name + `.AddWidget(ui.` + child.Widget.Name + `, ` + fmt.Sprintf("%d, %d", *child.Row, *child.Column) + `)
+					`)
+
+			default:
+				ret.WriteString("/* miqt-uic: no handler for layout '" + w.Layout.Class + "' */\n")
+
+			}
+		}
+	}
+
+	// Actions
+
+	for _, a := range w.Actions {
+		ret.WriteString(`
+		ui.` + a.Name + ` = qt.NewQAction(` + parentName + `)
+		ui.` + a.Name + `.SetObjectName(` + strconv.Quote(a.Name) + `)
+		`)
+	}
+
+	// AddActions
+	// TODO
+	// w.AddActions
+
+	// Items
+
+	for _, itm := range w.Items {
+		// FIXME add details
+		_ = itm
+		ret.WriteString("ui." + w.Name + `.AddItem()` + "\n")
+	}
+
+	// Attributes
+	// TODO
+	// w.Attributes
+
+	// Columns
+	// TODO
+	// w.Columns
+
+	// Recurse children
+	setCentralWidget := ""
+	for _, child := range w.Widgets {
+		nest, err := generateWidget(child, `ui.`+w.Name)
+		if err != nil {
+			return "", fmt.Errorf(w.Name+": %w", err)
+		}
+
+		ret.WriteString(nest)
+
+		// QMainWindow CentralWidget handling
+		// The first listed class can be the central widget.
+		// TODO should it be the first child with a layout? But need to handle windows with no layout
+		if w.Class == `QMainWindow` && setCentralWidget == "" {
+			ret.WriteString(`ui.` + w.Name + `.SetCentralWidget(ui.` + child.Name + ") // Set central widget \n")
+			setCentralWidget = w.Name
+		}
+	}
+
+	return ret.String(), nil
 }
 
 func generate(packageName string, goGenerateArgs string, u UiFile) ([]byte, error) {
@@ -46,25 +190,47 @@ import (
 )
 
 type ` + u.Class + `Ui struct {
-	` + strings.Join(collectClassNames(u), "\n") + `
+	` + strings.Join(collectClassNames_Widget(u.Widget), "\n") + `
 }
 
+// New` + u.Class + `Ui creates all Qt widget classes for ` + u.Class + `.
 func New` + u.Class + `Ui() *` + u.Class + `Ui {
 	ui := &` + u.Class + `Ui{}
 	`)
 
-	// ...
+	nest, err := generateWidget(u.Widget, "")
+	if err != nil {
+		return nil, err
+	}
+	ret.WriteString(nest)
 
 	ret.WriteString(`
 	return ui
 }
 
+// RetranslateUi reapplies all text translations.
 func (u *` + u.Class + `Ui) RetranslateUi() {
-	panic("TODO")
+	`)
+
+	// Trap and repeat all lines from `.nest` that include .Tr(.
+	for _, line := range strings.Split(nest, "\n") {
+		if strings.Contains(line, `.Tr(`) {
+			ret.WriteString(line + "\n")
+		}
+	}
+
+	ret.WriteString(`
 }
 
 	`)
 
 	output := ret.String()
-	return format.Source([]byte(output))
+
+	formatted, err := format.Source([]byte(output))
+	if err != nil {
+		// Return unformatted so it can be fixed
+		return []byte(output), nil
+	}
+
+	return formatted, nil
 }