termui/widgets/table.go

137 lines
4.4 KiB
Go
Raw Permalink Normal View History

2019-01-23 21:12:10 -07:00
// Copyright 2017 Zack Guo <zack.y.guo@gmail.com>. All rights reserved.
2022-08-28 10:46:41 -06:00
// Use of this source code is governed by a GPLv3 license that can
2019-01-23 21:12:10 -07:00
// be found in the LICENSE file.
package widgets
import (
"image"
. "git.tebitea.media/sashakoshka/termui/v3"
2019-01-23 21:12:10 -07:00
)
2019-01-24 07:21:07 -07:00
/*Table is like:
2019-01-23 21:12:10 -07:00
Awesome Table
Col0 | Col1 | Col2 | Col3 | Col4 | Col5 | Col6 |
Some Item #1 | AAA | 123 | CCCCC | EEEEE | GGGGG | IIIII |
Some Item #2 | BBB | 456 | DDDDD | FFFFF | HHHHH | JJJJJ |
*/
type Table struct {
Block
2019-02-23 18:15:42 -07:00
Rows [][]string
ColumnWidths []int
TextStyle Style
RowSeparator bool
TextAlignment Alignment
RowStyles map[int]Style
FillRow bool
2019-02-28 19:36:03 -07:00
// ColumnResizer is called on each Draw. Can be used for custom column sizing.
ColumnResizer func()
2019-01-23 21:12:10 -07:00
}
func NewTable() *Table {
return &Table{
2019-02-28 19:36:03 -07:00
Block: *NewBlock(),
TextStyle: Theme.Table.Text,
RowSeparator: true,
RowStyles: make(map[int]Style),
ColumnResizer: func() {},
2019-01-23 21:12:10 -07:00
}
}
func (self *Table) Draw(buf *Buffer) {
self.Block.Draw(buf)
2019-02-28 19:36:03 -07:00
self.ColumnResizer()
2019-01-23 21:12:10 -07:00
columnWidths := self.ColumnWidths
if len(columnWidths) == 0 {
columnCount := len(self.Rows[0])
2019-02-23 18:15:42 -07:00
columnWidth := self.Inner.Dx() / columnCount
2019-01-23 21:12:10 -07:00
for i := 0; i < columnCount; i++ {
2019-02-23 18:15:42 -07:00
columnWidths = append(columnWidths, columnWidth)
2019-01-23 21:12:10 -07:00
}
}
yCoordinate := self.Inner.Min.Y
// draw rows
for i := 0; i < len(self.Rows) && yCoordinate < self.Inner.Max.Y; i++ {
row := self.Rows[i]
colXCoordinate := self.Inner.Min.X
2019-02-01 11:12:22 -07:00
rowStyle := self.TextStyle
// get the row style if one exists
if style, ok := self.RowStyles[i]; ok {
rowStyle = style
}
if self.FillRow {
blankCell := NewCell(' ', rowStyle)
buf.Fill(blankCell, image.Rect(self.Inner.Min.X, yCoordinate, self.Inner.Max.X, yCoordinate+1))
}
2019-01-23 21:12:10 -07:00
// draw row cells
for j := 0; j < len(row); j++ {
2019-02-23 17:54:20 -07:00
col := ParseStyles(row[j], rowStyle)
2019-01-23 21:12:10 -07:00
// draw row cell
2019-02-23 18:15:42 -07:00
if len(col) > columnWidths[j] || self.TextAlignment == AlignLeft {
for _, cx := range BuildCellWithXArray(col) {
k, cell := cx.X, cx.Cell
2019-01-23 21:12:10 -07:00
if k == columnWidths[j] || colXCoordinate+k == self.Inner.Max.X {
2019-01-26 03:47:47 -07:00
cell.Rune = ELLIPSES
2019-01-23 21:12:10 -07:00
buf.SetCell(cell, image.Pt(colXCoordinate+k-1, yCoordinate))
break
} else {
buf.SetCell(cell, image.Pt(colXCoordinate+k, yCoordinate))
}
}
2019-02-23 18:15:42 -07:00
} else if self.TextAlignment == AlignCenter {
2019-01-23 21:12:10 -07:00
xCoordinateOffset := (columnWidths[j] - len(col)) / 2
stringXCoordinate := xCoordinateOffset + colXCoordinate
for _, cx := range BuildCellWithXArray(col) {
k, cell := cx.X, cx.Cell
2019-01-23 21:12:10 -07:00
buf.SetCell(cell, image.Pt(stringXCoordinate+k, yCoordinate))
}
2019-02-23 18:15:42 -07:00
} else if self.TextAlignment == AlignRight {
2019-01-23 21:12:10 -07:00
stringXCoordinate := MinInt(colXCoordinate+columnWidths[j], self.Inner.Max.X) - len(col)
for _, cx := range BuildCellWithXArray(col) {
k, cell := cx.X, cx.Cell
2019-01-23 21:12:10 -07:00
buf.SetCell(cell, image.Pt(stringXCoordinate+k, yCoordinate))
}
}
colXCoordinate += columnWidths[j] + 1
}
// draw vertical separators
2019-02-01 11:12:22 -07:00
separatorStyle := self.Block.BorderStyle
2019-01-23 21:12:10 -07:00
separatorXCoordinate := self.Inner.Min.X
2019-02-01 11:12:22 -07:00
verticalCell := NewCell(VERTICAL_LINE, separatorStyle)
for i, width := range columnWidths {
if self.FillRow && i < len(columnWidths)-1 {
verticalCell.Style.Bg = rowStyle.Bg
} else {
verticalCell.Style.Bg = self.Block.BorderStyle.Bg
}
2019-01-23 21:12:10 -07:00
separatorXCoordinate += width
buf.SetCell(verticalCell, image.Pt(separatorXCoordinate, yCoordinate))
separatorXCoordinate++
}
yCoordinate++
// draw horizontal separator
2019-02-01 11:12:22 -07:00
horizontalCell := NewCell(HORIZONTAL_LINE, separatorStyle)
2019-01-23 21:12:10 -07:00
if self.RowSeparator && yCoordinate < self.Inner.Max.Y && i != len(self.Rows)-1 {
buf.Fill(horizontalCell, image.Rect(self.Inner.Min.X, yCoordinate, self.Inner.Max.X, yCoordinate+1))
yCoordinate++
}
}
}