gotui/_examples/treemap/main.go at master · metaspartan/gotui · GitHub

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
package main

import (
	"log"

	ui "github.com/metaspartan/gotui/v5"
	"github.com/metaspartan/gotui/v5/widgets"
)

func main() {
	if err := ui.Init(); err != nil {
		log.Fatalf("failed to initialize gotui: %v", err)
	}
	defer ui.Close()

	tm := widgets.NewTreeMap()
	tm.Title = "Disk Usage"
	tm.SetRect(0, 0, 80, 24)
	tm.TextColor = ui.ColorWhite

	// Sample Hierarchical Data
	root := &widgets.TreeMapNode{
		Children: []*widgets.TreeMapNode{
			{
				Label: "/var",
				Value: 40,
				Style: ui.NewStyle(ui.ColorWhite, ui.ColorRed),
			},
			{
				Label: "/home",
				Value: 30,
				Style: ui.NewStyle(ui.ColorWhite, ui.ColorBlue),
			},
			{
				Label: "/usr",
				Value: 20,
				Style: ui.NewStyle(ui.ColorWhite, ui.ColorGreen),
			},
			{
				Label: "/etc",
				Value: 10,
				Style: ui.NewStyle(ui.ColorWhite, ui.ColorYellow),
			},
		},
	}
	tm.Root = root

	ui.Render(tm)

	uiEvents := ui.PollEvents()
	for {
		e := <-uiEvents
		switch e.ID {
		case "q", "<C-c>":
			return
		}
	}
}