Merge pull request #111 from wanzysky/master

table widget

_example/table.go

@@ -0,0 +1,52 @@
+package main
+
+import "github.com/gizak/termui"
+
+func main() {
+	err := termui.Init()
+	if err != nil {
+		panic(err)
+	}
+	defer termui.Close()
+	rows_1 := [][]string{
+		[]string{"header1", "header2", "header3"},
+		[]string{"孙嘉你好吗", "Go-lang is so cool", "Im working on Ruby"},
+		[]string{"2016", "10", "11"},
+	}
+
+	table_1 := termui.NewTable()
+	table_1.Rows = rows_1
+	table_1.FgColor = termui.ColorWhite
+	table_1.BgColor = termui.ColorDefault
+	table_1.Y = 0
+	table_1.X = 0
+	table_1.Width = 62
+	table_1.Height = 7
+
+	termui.Render(table_1)
+
+	rows := [][]string{
+		[]string{"header1", "header2", "header3"},
+		[]string{"Foundations", "Go-lang is so cool", "Im working on Ruby"},
+		[]string{"2016", "11", "11"},
+	}
+
+	table := termui.NewTable()
+	table.Rows = rows
+	table.FgColor = termui.ColorWhite
+	table.BgColor = termui.ColorDefault
+	table.TextAlign = termui.AlignCenter
+	table.Seperator = false
+	table.Analysis()
+	table.SetSize()
+	table.BgColors[2] = termui.ColorRed
+	table.Y = 20
+	table.X = 0
+	table.Border = true
+
+	termui.Render(table)
+	termui.Handle("/sys/kbd/q", func(termui.Event) {
+		termui.StopLoop()
+	})
+	termui.Loop()
+}

table.go

@@ -0,0 +1,170 @@
+package termui
+
+import "strings"
+
+/*
+	table := termui.NewTable()
+	table.Rows = rows
+	table.FgColor = termui.ColorWhite
+	table.BgColor = termui.ColorDefault
+	table.Height = 7
+	table.Width = 62
+	table.Y = 0
+	table.X = 0
+	table.Border = true
+*/
+
+type Table struct {
+	Block
+	Rows      [][]string
+	CellWidth []int
+	FgColor   Attribute
+	BgColor   Attribute
+	FgColors  []Attribute
+	BgColors  []Attribute
+	Seperator bool
+	TextAlign Align
+}
+
+func NewTable() *Table {
+	table := &Table{Block: *NewBlock()}
+	table.FgColor = ColorWhite
+	table.BgColor = ColorDefault
+	table.Seperator = true
+	return table
+}
+
+func (table *Table) Analysis() {
+	length := len(table.Rows)
+	if length < 1 {
+		return
+	}
+
+	if len(table.FgColors) == 0 {
+		table.FgColors = make([]Attribute, len(table.Rows))
+	}
+	if len(table.BgColors) == 0 {
+		table.BgColors = make([]Attribute, len(table.Rows))
+	}
+
+	row_width := len(table.Rows[0])
+	cellWidthes := make([]int, row_width)
+
+	for index, row := range table.Rows {
+		for i, str := range row {
+			if cellWidthes[i] < len(str) {
+				cellWidthes[i] = len(str)
+			}
+		}
+
+		if table.FgColors[index] == 0 {
+			table.FgColors[index] = table.FgColor
+		}
+
+		if table.BgColors[index] == 0 {
+			table.BgColors[index] = table.BgColor
+		}
+	}
+
+	table.CellWidth = cellWidthes
+
+	//width_sum := 2
+	//for i, width := range cellWidthes {
+	//	width_sum += (width + 2)
+	//	for u, row := range table.Rows {
+	//		switch table.TextAlign {
+	//		case "right":
+	//			row[i] = fmt.Sprintf(" %*s ", width, table.Rows[u][i])
+	//		case "center":
+	//			word_width := len(table.Rows[u][i])
+	//			offset := (width - word_width) / 2
+	//			row[i] = fmt.Sprintf(" %*s ", width, fmt.Sprintf("%-*s", offset+word_width, table.Rows[u][i]))
+	//		default: // left
+	//			row[i] = fmt.Sprintf(" %-*s ", width, table.Rows[u][i])
+	//		}
+	//	}
+	//}
+
+	//if table.Width == 0 {
+	//	table.Width = width_sum
+	//}
+}
+
+func (table *Table) SetSize() {
+	length := len(table.Rows)
+	if table.Seperator {
+		table.Height = length*2 + 1
+	} else {
+		table.Height = length + 2
+	}
+	table.Width = 2
+	if length != 0 {
+		for _, cell_width := range table.CellWidth {
+			table.Width += cell_width + 3
+		}
+	}
+}
+
+func (table *Table) CalculatePosition(x int, y int, x_coordinate *int, y_coordibate *int, cell_beginning *int) {
+	if table.Seperator {
+		*y_coordibate = table.innerArea.Min.Y + y*2
+	} else {
+		*y_coordibate = table.innerArea.Min.Y + y
+	}
+	if x == 0 {
+		*cell_beginning = table.innerArea.Min.X
+	} else {
+		*cell_beginning += table.CellWidth[x-1] + 3
+	}
+
+	switch table.TextAlign {
+	case AlignRight:
+		*x_coordinate = *cell_beginning + (table.CellWidth[x] - len(table.Rows[y][x])) + 2
+	case AlignCenter:
+		*x_coordinate = *cell_beginning + (table.CellWidth[x]-len(table.Rows[y][x]))/2 + 2
+	default:
+		*x_coordinate = *cell_beginning + 2
+	}
+}
+
+func (table *Table) Buffer() Buffer {
+	buffer := table.Block.Buffer()
+	table.Analysis()
+
+	pointer_x := table.innerArea.Min.X + 2
+	pointer_y := table.innerArea.Min.Y
+	border_pointer_x := table.innerArea.Min.X
+	for y, row := range table.Rows {
+		for x, cell := range row {
+			table.CalculatePosition(x, y, &pointer_x, &pointer_y, &border_pointer_x)
+			backgraound := DefaultTxBuilder.Build(strings.Repeat(" ", table.CellWidth[x]+3), table.BgColors[y], table.BgColors[y])
+			cells := DefaultTxBuilder.Build(cell, table.FgColors[y], table.BgColors[y])
+
+			for i, back := range backgraound {
+				buffer.Set(border_pointer_x+i, pointer_y, back)
+			}
+
+			coordinate_x := pointer_x
+			for _, printer := range cells {
+				buffer.Set(coordinate_x, pointer_y, printer)
+				coordinate_x += printer.Width()
+			}
+
+			if x != 0 {
+				devidors := DefaultTxBuilder.Build("|", table.FgColors[y], table.BgColors[y])
+				for _, devidor := range devidors {
+					buffer.Set(border_pointer_x, pointer_y, devidor)
+				}
+			}
+		}
+
+		if table.Seperator {
+			border := DefaultTxBuilder.Build(strings.Repeat("─", table.Width-2), table.FgColor, table.BgColor)
+			for i, cell := range border {
+				buffer.Set(i+1, pointer_y+1, cell)
+			}
+		}
+	}
+
+	return buffer
+}