initial commit

uiterm/tree.go

@@ -0,0 +1,140 @@
+package uiterm
+
+import (
+	"strings"
+
+	"github.com/nsf/termbox-go"
+)
+
+type TreeItem interface {
+	TreeItemStyle(active bool) (Attribute, Attribute)
+	String() string
+}
+
+type renderedTreeItem struct {
+	//String string
+	Level int
+	Item  TreeItem
+}
+
+type TreeFunc func(item TreeItem) []TreeItem
+type TreeListener func(ui *Ui, tree *Tree, item TreeItem)
+
+type Tree struct {
+	Fg        Attribute
+	Bg        Attribute
+	Generator TreeFunc
+	Listener  TreeListener
+
+	lines          []renderedTreeItem
+	activeLine     int
+	active         bool
+	x0, y0, x1, y1 int
+}
+
+func bounded(i, lower, upper int) int {
+	if i < lower {
+		return lower
+	}
+	if i > upper {
+		return upper
+	}
+	return i
+}
+
+func (t *Tree) SetBounds(ui *Ui, x0, y0, x1, y1 int) {
+	t.x0 = x0
+	t.y0 = y0
+	t.x1 = x1
+	t.y1 = y1
+}
+
+func (t *Tree) Rebuild() {
+	if t.Generator == nil {
+		t.lines = []renderedTreeItem{}
+		return
+	}
+
+	lines := []renderedTreeItem{}
+	for _, item := range t.Generator(nil) {
+		children := t.rebuild_rec(item, 0)
+		if children != nil {
+			lines = append(lines, children...)
+		}
+	}
+	t.lines = lines
+	t.activeLine = bounded(t.activeLine, 0, len(t.lines)-1)
+}
+
+func (t *Tree) rebuild_rec(parent TreeItem, level int) []renderedTreeItem {
+	if parent == nil {
+		return nil
+	}
+	lines := []renderedTreeItem{
+		renderedTreeItem{
+			Level: level,
+			Item:  parent,
+		},
+	}
+	for _, item := range t.Generator(parent) {
+		children := t.rebuild_rec(item, level+1)
+		if children != nil {
+			lines = append(lines, children...)
+		}
+	}
+	return lines
+}
+
+func (t *Tree) Draw(ui *Ui) {
+	if t.lines == nil {
+		t.Rebuild()
+	}
+
+	line := 0
+	for y := t.y0; y < t.y1; y++ {
+		var reader *strings.Reader
+		var item TreeItem
+		level := 0
+		if line < len(t.lines) {
+			item = t.lines[line].Item
+			level = t.lines[line].Level
+			reader = strings.NewReader(item.String())
+		}
+		for x := t.x0; x < t.x1; x++ {
+			var chr rune = ' '
+			fg := t.Fg
+			bg := t.Bg
+			dx := x - t.x0
+			dy := y - t.y0
+			if reader != nil && level*2 <= dx {
+				if ch, _, err := reader.ReadRune(); err == nil {
+					chr = ch
+					fg, bg = item.TreeItemStyle(t.active && t.activeLine == dy)
+				}
+			}
+			termbox.SetCell(x, y, chr, termbox.Attribute(fg), termbox.Attribute(bg))
+		}
+		line++
+	}
+}
+
+func (t *Tree) SetActive(ui *Ui, active bool) {
+	t.active = active
+}
+
+func (t *Tree) KeyEvent(ui *Ui, mod Modifier, key Key) {
+	switch key {
+	case KeyArrowUp:
+		t.activeLine = bounded(t.activeLine-1, 0, len(t.lines)-1)
+	case KeyArrowDown:
+		t.activeLine = bounded(t.activeLine+1, 0, len(t.lines)-1)
+	case KeyEnter:
+		if t.Listener != nil && t.activeLine >= 0 && t.activeLine < len(t.lines) {
+			t.Listener(ui, t, t.lines[t.activeLine].Item)
+		}
+	}
+	ui.Refresh()
+}
+
+func (t *Tree) CharacterEvent(ui *Ui, ch rune) {
+}