Files

52 lines
1.4 KiB
Go

package chart
import (
"bytes"
"image/png"
"log/slog"
"math"
"github.com/egoist/mygo/ui"
"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/core"
)
// ChartExportOptions configure a ChartExport.
type ChartExportOptions struct {
// Width and Height are the image size in DIPs; required.
Width, Height int
// Scale is device pixels per DIP; 0 means 2.
Scale float32
}
// ChartExport returns a PNG rendered in the current theme.
func ChartExport(c *ui.Context, chart func(c *ui.Context), opts ChartExportOptions) ([]byte, error) {
if chart == nil || opts.Width <= 0 || opts.Height <= 0 || opts.Scale < 0 || math.IsNaN(float64(opts.Scale)) || math.IsInf(float64(opts.Scale), 0) {
panic("mujica: ChartExport needs a chart, a positive size and Scale ≥ 0")
}
scale := opts.Scale
if scale == 0 {
scale = 2
}
settings, tokens := core.CurrentSettings(c), core.Tokens(c)
ready := false
tester := ui.NewTester(func(c *ui.Context) {
if !ready {
return
}
core.Use(c, settings)
ui.Column(c).Fill().Background(tokens.Surface).Padding(core.Density(c).Unit() * 4).Children(func() { chart(c) })
}, opts.Width, opts.Height)
tester.SetScale(scale)
tester.SetPreferences(c.Preferences())
ready = true
tester.Frame()
img := tester.Image()
var buf bytes.Buffer
if err := png.Encode(&buf, img); err != nil {
return nil, err
}
slog.Debug("mujica: chart_export png", "bytes", buf.Len(), "width", opts.Width, "height", opts.Height)
return buf.Bytes(), nil
}