256 lines
8.2 KiB
Go
256 lines
8.2 KiB
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 = ©
|
|
}
|
|
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 = ©
|
|
}
|
|
result, plot := ch.Build(spec2)
|
|
if plot != nil {
|
|
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)
|
|
}
|
|
}
|
|
}
|