feat: add multi line text draw

table.go

@@ -0,0 +1,145 @@
+// MIT License
+
+// Copyright (c) 2022 Tree Xie
+
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+
+// The above copyright notice and this permission notice shall be included in all
+// copies or substantial portions of the Software.
+
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+package charts
+
+import (
+	"errors"
+
+	"github.com/wcharczuk/go-chart/v2"
+)
+
+type TableOption struct {
+	// draw
+	Draw *Draw
+	// The width of table
+	Width int
+	// The header of table
+	Header []string
+	// The style of table
+	Style        chart.Style
+	ColumnWidths []float64
+	// 是否仅测量高度
+	measurement bool
+}
+
+var ErrTableColumnWidthInvalid = errors.New("table column width is invalid")
+
+func tableDivideWidth(width, size int, columnWidths []float64) ([]int, error) {
+	widths := make([]int, size)
+
+	autoFillCount := size
+	restWidth := width
+	if len(columnWidths) != 0 {
+		for index, v := range columnWidths {
+			if v <= 0 {
+				continue
+			}
+			autoFillCount--
+			// 小于1的表示占比
+			if v < 1 {
+				widths[index] = int(v * float64(width))
+			} else {
+				widths[index] = int(v)
+			}
+			restWidth -= widths[index]
+		}
+	}
+	if restWidth < 0 {
+		return nil, ErrTableColumnWidthInvalid
+	}
+	// 填充其它未指定的宽度
+	if autoFillCount > 0 {
+		autoWidth := restWidth / autoFillCount
+		for index, v := range widths {
+			if v == 0 {
+				widths[index] = autoWidth
+			}
+		}
+	}
+	widthSum := 0
+	for _, v := range widths {
+		widthSum += v
+	}
+	if widthSum > width {
+		return nil, ErrTableColumnWidthInvalid
+	}
+	return widths, nil
+}
+
+func TableMeasure(opt TableOption) (chart.Box, error) {
+	d, err := NewDraw(DrawOption{
+		Width:  opt.Width,
+		Height: 600,
+	})
+	if err != nil {
+		return chart.BoxZero, err
+	}
+	opt.Draw = d
+	opt.measurement = true
+	return tableRender(opt)
+}
+
+func tableRender(opt TableOption) (chart.Box, error) {
+	if opt.Draw == nil {
+		return chart.BoxZero, errors.New("draw can not be nil")
+	}
+	if len(opt.Header) == 0 {
+		return chart.BoxZero, errors.New("header can not be nil")
+	}
+	width := opt.Width
+	if width == 0 {
+		width = opt.Draw.Box.Width()
+	}
+
+	columnWidths, err := tableDivideWidth(width, len(opt.Header), opt.ColumnWidths)
+	if err != nil {
+		return chart.BoxZero, err
+	}
+
+	d := opt.Draw
+	style := opt.Style
+	y := 0
+	x := 0
+
+	headerMaxHeight := 0
+	for index, text := range opt.Header {
+		var box chart.Box
+		w := columnWidths[index]
+		y0 := y + int(opt.Style.FontSize)
+		if opt.measurement {
+			box = d.measureTextFit(text, x, y0, w, style)
+		} else {
+			box = d.textFit(text, x, y0, w, style)
+		}
+		if box.Height() > headerMaxHeight {
+			headerMaxHeight = box.Height()
+		}
+		x += w
+	}
+	y += headerMaxHeight
+
+	return chart.Box{
+		Right:  width,
+		Bottom: y,
+	}, nil
+}