tree widget

1 files changed, 67 insertions, 6 deletions

diff --git a/gui/widget/widget.go b/gui/widget/widget.go
index 2e7c671..3fd9637 100644
--- a/gui/widget/widget.go
+++ b/gui/widget/widget.go
@@ -11,6 +11,8 @@ import (
 	"image/draw"
 
 	"volute/gui"
+	"volute/gui/layout"
+	"volute/gui/text"
 	"volute/gui/win"
 )
 
@@ -23,12 +25,71 @@ var (
 	interpolator = xdraw.ApproxBiLinear
 )
 
-func Label(text string, r image.Rectangle, env gui.Env, wg *sync.WaitGroup) {
+type Node[T any] struct {
+	Label    string
+	Value    T
+	Children []Node[T]
+
+	expanded bool
+}
+
+func Tree[T any](trees []Node[T], r image.Rectangle, focus FocusSlave, mux *gui.Mux, wg *sync.WaitGroup) {
+	defer wg.Done()
+
+	var nodes []string
+	for _, root := range trees {
+		nodes = append(nodes, flatten(root, 0)...)
+	}
+
+	bounds := layout.Grid{
+		Rows:        populate(make([]int, len(nodes)), 1),
+		Background:  color.Gray{255},
+		Gap:         1,
+		Split:       layout.EvenSplit,
+		SplitRows:   layout.TextRowSplit,
+		Margin:      0,
+		Border:      0,
+		BorderColor: color.Gray{16},
+		Flip:        false,
+	}.Lay(r)
+	for i := range nodes {
+		wg.Add(1)
+		go Label(nodes[i], bounds[i], mux.MakeEnv(), wg)
+	}
+
+	/*
+		globalFocus := focus;
+		localFocus := NewFocusMaster([]int{1, 1, 1});
+		defer localFocus.Close()
+	*/
+	// TODO
+}
+
+func flatten[T any](root Node[T], depth int) []string {
+	indent := string(populate(make([]byte, 2*depth), ' '))
+	nodes := []string{indent + root.Label}
+	root.expanded = true // TODO: remove me
+	if root.expanded {
+		for _, c := range root.Children {
+			nodes = append(nodes, flatten(c, depth+1)...)
+		}
+	}
+	return nodes
+}
+
+func populate[T any](arr []T, v T) []T {
+	for i := range arr {
+		arr[i] = v
+	}
+	return arr
+}
+
+func Label(str string, r image.Rectangle, env gui.Env, wg *sync.WaitGroup) {
 	defer wg.Done()
 	defer close(env.Draw())
 
 	redraw := func(drw draw.Image) image.Rectangle {
-		drawText([]byte(text), drw, r, BLACK, WHITE)
+		text.Draw([]byte(str), drw, r, BLACK, WHITE)
 		return r
 	}
 
@@ -93,12 +154,12 @@ Loop:
 	}
 }
 
-func inputDraw(text []byte, focused bool, r image.Rectangle) func(draw.Image) image.Rectangle {
+func inputDraw(str []byte, focused bool, r image.Rectangle) func(draw.Image) image.Rectangle {
 	return func(drw draw.Image) image.Rectangle {
 		if focused {
-			drawText(text, drw, r, GREEN, FOCUS_COLOR)
+			text.Draw(str, drw, r, GREEN, FOCUS_COLOR)
 		} else {
-			drawText(text, drw, r, GREEN, WHITE)
+			text.Draw(str, drw, r, GREEN, WHITE)
 		}
 		return r
 	}
@@ -128,7 +189,7 @@ Loop:
 
 func outputDraw(v float64, r image.Rectangle) func(draw.Image) image.Rectangle {
 	return func(drw draw.Image) image.Rectangle {
-		drawText([]byte(fmt.Sprintf("%.3f", v)), drw, r, BLACK, WHITE)
+		text.Draw([]byte(fmt.Sprintf("%.3f", v)), drw, r, BLACK, WHITE)
 		return r
 	}
 }