MujicaUI/chart/indicator_pane.go

package chart

import (
	"fmt"
	"log/slog"
	"math"
	"strings"

	"github.com/egoist/mygo/ui"

	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/core"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/data"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/icons"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/input"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/internal"
	"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/theme"
)

// IndicatorPaneSpec is one lower pane: MACD, RSI, KDJ or OBV.
type IndicatorPaneSpec struct {
	Kind IndicatorKind
	// Params nil uses the kind's defaults.
	Params []float64
}

// IndicatorPanes holds stacked indicators and their height shares.
type IndicatorPanes struct {
	Panes []IndicatorPaneSpec
	// Sizes reset to equal shares when pane count changes.
	Sizes []float32
}

// IndicatorPaneOptions configure an IndicatorPane.
type IndicatorPaneOptions struct {
	// Height defaults to 140 per pane.
	Height float32
	// Label prefixes each pane's accessible name, as the symbol.
	Label    string
	Disabled bool
	Status   data.DataStatus
	Error    string
	Empty    string
}

// IndicatorPaneResult reports the stack's events.
type IndicatorPaneResult struct {
	Element ui.Element
	// Removed reports the closed pane; -1 means none.
	Removed int
	// Resized reports finished height changes, for saving.
	Resized      bool
	Point        int
	Pane         int
	PointChanged bool
}

// IndicatorPane stacks resizable technical indicators using the shared window.
func IndicatorPane(c *ui.Context, candles []Candle, view *MarketView, panes *IndicatorPanes, opts IndicatorPaneOptions) IndicatorPaneResult {
	if view == nil || panes == nil {
		panic("mujica: IndicatorPane needs a view and panes")
	}
	box := ui.Column(c).FillWidth().Disabled(opts.Disabled)
	if box.IsDisabled() {
		copy := *panes
		copy.Panes = append([]IndicatorPaneSpec(nil), panes.Panes...)
		copy.Sizes = append([]float32(nil), panes.Sizes...)
		panes = &copy
	}
	if opts.Height < 0 {
		panic("mujica: IndicatorPane Height must be non-negative")
	}
	res := IndicatorPaneResult{Removed: -1, Point: -1, Pane: -1}
	n := len(panes.Panes)
	for _, p := range panes.Panes {
		if _ = p.Kind.String(); indicatorInfo[p.Kind].overlay {
			panic(fmt.Sprintf("mujica: %v is an overlay, not a pane indicator", p.Kind))
		}
		indicatorParams(p.Kind, p.Params)
	}
	if n == 0 {
		res.Element = box
		return res
	}
	if len(panes.Sizes) != n {
		panes.Sizes = nil
	}
	u := core.Density(c).Unit()
	lineH := core.FontSize(c, theme.CaptionSize) * theme.LineHeight
	chrome := 2*max(theme.MinHitSize, lineH) + max(u*5, lineH) + 4*u
	minimum := chrome + 40
	h := opts.Height
	if h == 0 {
		h = 140 * float32(n)
	}
	build := func(i int) func() {
		return func() {
			col := ui.Column(c).Fill().MinHeight(0)
			b := col.Bounds()
			internal.Remeasure(c, col, b.W, b.H)
			plotH := max(40, b.H-chrome)
			col.Children(func() {
				closed, point := indicatorPaneOne(c, candles, view, panes.Panes[i], opts, plotH)
				if point.Point >= 0 {
					res.Point = point.Point
					res.Pane = i
				}
				res.PointChanged = res.PointChanged || point.Changed()
				if closed {
					res.Removed = i
				}
			})
		}
	}
	box.Height(h).MinHeight(float32(n)*minimum + float32(n-1)*theme.BorderWidth)
	box.Children(func() {
		if n == 1 {
			build(0)()
			return
		}
		list := make([]input.ResizablePanel, n)
		for i, p := range panes.Panes {
			list[i] = input.ResizablePanel{Label: indicatorName(p.Kind, indicatorParams(p.Kind, p.Params)), Min: minimum, Build: build(i)}
		}
		r := input.ResizablePanelGroup(c, &panes.Sizes, list, input.ResizablePanelGroupOptions{Vertical: true})
		r.Element.Fill()
		res.Resized = r.Committed
	})
	if res.Removed >= 0 {
		slog.Debug("mujica: indicator_pane removed", "kind", panes.Panes[res.Removed].Kind)
		panes.Panes = append(panes.Panes[:res.Removed:res.Removed], panes.Panes[res.Removed+1:]...)
		panes.Sizes = nil
		c.Invalidate()
	}
	res.Element = box
	return res
}

// indicatorPaneOne draws one indicator and reports its events.
func indicatorPaneOne(c *ui.Context, candles []Candle, view *MarketView, spec IndicatorPaneSpec, opts IndicatorPaneOptions, height float32) (bool, ChartResult) {
	k := core.Tokens(c)
	params := indicatorParams(spec.Kind, spec.Params)
	name := indicatorName(spec.Kind, params)
	closed := false
	ui.Row(c).AlignItems(ui.Center).Gap(core.Density(c).Unit()).Children(func() {
		ui.Text(c, name).FontSize(core.FontSize(c, theme.CaptionSize)).FontWeight(600).TextColor(k.Text).Grow(1)
		closed = core.IconAction(c, icons.Must("x"), fmt.Sprintf(core.Msg(c, "indicator_pane.remove"), name)).Clicked()
	})
	label := strings.TrimSpace(opts.Label + " " + name)
	ch := NewChart(c, ChartFrame{Label: label, Height: height, Status: opts.Status, Error: opts.Error, Empty: opts.Empty})
	series := ChartMemo(ch, "series", fmt.Sprintf("%s|%v|%v", ChartID(candles), spec.Kind, params), func() []ChartSeries {
		marketCheckCandles(candles)
		return indicatorSeries(spec.Kind, params, candles)
	})
	m := &mktCore{c: c, k: k, ch: ch, candles: candles, view: view}
	m.window()
	pal := internal.Palette(k)
	lines := make([]mktLine, len(series))
	keys := make([]ChartLegendItem, len(series))
	for i, s := range series {
		lines[i] = mktLine{name: s.Name, values: s.Values, color: pal[(i+1)%len(pal)], shape: i}
		keys[i] = ChartLegendItem{Name: s.Name, Color: lines[i].color, Shape: i}
	}
	yr := ChartMemo(ch, "y", [5]any{ChartID(candles), m.from, m.to, spec.Kind, fmt.Sprint(params)}, func() [2]float64 {
		lo, hi := math.Inf(1), math.Inf(-1)
		for _, l := range lines {
			for i := m.from; i < m.to; i++ {
				if v := l.values[i]; !math.IsNaN(v) {
					lo, hi = min(lo, v), max(hi, v)
				}
			}
		}
		if spec.Kind == IndicatorRSI {
			lo, hi = 0, 100
		}
		return [2]float64{lo, hi}
	})
	lo, hi := yr[0], yr[1]
	spec2 := ChartSpec{Marks: len(candles), Legend: keys}
	if m.to <= m.from || math.IsInf(lo, 0) {
		spec2.Marks = 0
	}
	m.lo, m.hi = lo, hi
	spec2.Pick = func(r ui.Rect, x, y float32) int {
		g := m.paneGeom(r)
		if !g.plot.Contains(x, y) {
			return -1
		}
		return g.index(m, x)
	}
	spec2.Readout = func(i int) string {
		parts := []string{candles[i].Time.Format("2006-01-02 15:04")}
		for _, l := range lines {
			parts = append(parts, l.name+" "+ChartValue(c, l.values[i], func(v float64) string { return fmt.Sprintf("%.2f", v) }))
		}
		return strings.Join(parts, " · ")
	}
	spec2.Draw = func(p *ui.Painter, r ui.Rect, active int) {
		g := m.paneGeom(r)
		size := core.FontSize(c, theme.CaptionSize)
		internal.AxisY(p, g.plot, g.ys, g.ticks, func(v float64) string { return volumeShort(v, g.step) }, size, k)
		p.Clip(g.plot, 0, func() {
			for i, l := range lines {
				if spec.Kind == IndicatorMACD && i == 2 {
					indicatorPaneBars(p, g, m, l.values, c)
					continue
				}
				mktPaintLine(p, g, m, l.values, l.color, i)
			}
		})
		if active >= m.from && active < m.to {
			x := internal.Snap(p, g.x(m, active))
			p.Line(x, g.plot.Y, x, g.plot.Y+g.plot.H, 1, k.TextMuted)
		}
	}
	m.st = *ui.Local(ch.Res.Element, "mujica.mkt", func() *mktState { return &mktState{grab: -1, focus: -1, alertFocus: -1} })
	if ch.Res.Element.IsDisabled() {
		copy := *view
		m.view = &copy
	}
	result, plot := ch.Build(spec2)
	if plot.Valid() {
		m.input(plot)
	}
	return closed && !ch.Res.Element.IsDisabled(), result
}

// paneGeom is geom without the time axis, for lower panes.
func (m *mktCore) paneGeom(r ui.Rect) mktGeom {
	plot := ChartInset(r, 60, 4)
	lo, hi, step := internal.Nice(m.lo, m.hi, 2)
	return mktGeom{plot: plot, band: internal.Band{N: m.to - m.from, R0: plot.X, R1: plot.X + plot.W, Pad: 0.3},
		ys: internal.Linear{D0: lo, D1: hi, R0: float64(plot.Y + plot.H), R1: float64(plot.Y)}, ticks: internal.Ticks(lo, hi, step), step: step}
}

// indicatorPaneBars draws MACD's histogram: hollow bars above zero, filled below.
func indicatorPaneBars(p *ui.Painter, g mktGeom, m *mktCore, vals []float64, c *ui.Context) {
	k := core.Tokens(c)
	zero := g.ys.Map(0)
	w := max(1, g.band.Width())
	for i := m.from; i < m.to; i++ {
		v := vals[i]
		if math.IsNaN(v) {
			continue
		}
		y := g.ys.Map(v)
		x := internal.Snap(p, g.x(m, i))
		bar := ui.Rect{X: x - w/2, Y: min(y, zero), W: w, H: max(1, float32(math.Abs(float64(y-zero))))}
		if v >= 0 && w >= 3 {
			p.Fill(bar, k.Surface, 0)
			p.Stroke(bar, core.TrendColor(c, 1), 0, 1)
		} else {
			p.Fill(bar, core.TrendColor(c, core.TrendSign(v)), 0)
		}
	}
}