giu/PlotSpec.go at master · AllenDang/giu · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package giu

import (
	"image/color"

	"github.com/AllenDang/cimgui-go/implot"
)

//go:generate go run github.com/dmarkham/enumer@latest -linecomment -type=PlotProperty .

// PlotProperty represents a property of a single plot on the canvas.
// See PlotSpec.
type PlotProperty byte

// Plot properties. These are used to set various properties of plots on the canvas. See PlotSpec.
const (
	PlotPropertyLineColor        PlotProperty = PlotProperty(implot.PropLineColor)
	PlotPropertyLineColors       PlotProperty = PlotProperty(implot.PropLineColors)
	PlotPropertyLineWeight       PlotProperty = PlotProperty(implot.PropLineWeight)
	PlotPropertyFillColor        PlotProperty = PlotProperty(implot.PropFillColor)
	PlotPropertyFillColors       PlotProperty = PlotProperty(implot.PropFillColors)
	PlotPropertyFillAlpha        PlotProperty = PlotProperty(implot.PropFillAlpha)
	PlotPropertyMarker           PlotProperty = PlotProperty(implot.PropMarker)
	PlotPropertyMarkerSize       PlotProperty = PlotProperty(implot.PropMarkerSize)
	PlotPropertyMarkerSizes      PlotProperty = PlotProperty(implot.PropMarkerSizes)
	PlotPropertyMarkerLineColor  PlotProperty = PlotProperty(implot.PropMarkerLineColor)
	PlotPropertyMarkerLineColors PlotProperty = PlotProperty(implot.PropMarkerLineColors)
	PlotPropertyMarkerFillColor  PlotProperty = PlotProperty(implot.PropMarkerFillColor)
	PlotPropertyMarkerFillColors PlotProperty = PlotProperty(implot.PropMarkerFillColors)
	PlotPropertySize             PlotProperty = PlotProperty(implot.PropSize)
	PlotPropertyOffset           PlotProperty = PlotProperty(implot.PropOffset)
	PlotPropertyStride           PlotProperty = PlotProperty(implot.PropStride)
	PlotPropertyFlags            PlotProperty = PlotProperty(implot.PropFlags)
)

// PlotSpec is a wrapper for implot.Spec. It allows to set various properties for plots.
type PlotSpec struct {
	spec *implot.Spec
}

// NewPlotSpec creates a new PlotSpec instance with default values.
func NewPlotSpec() *PlotSpec {
	return &PlotSpec{spec: implot.NewSpec()}
}

// SetProperty allows to set chosen property to value. Type of value could be:
// - uint16
// - uint32
// - *uint32
// - uint64
// - uint8
// - vec4
// - float64
// - int16
// - int
// - int64
// TODO: This could be made a generic method as soon as go adds them (probably go 1.27).
func (ps *PlotSpec) SetProperty(property PlotProperty, value any) *PlotSpec {
	switch value := value.(type) {
	case uint16:
		ps.spec.SetPropU16(implot.Prop(property), value)
	case uint32:
		ps.spec.SetPropU32(implot.Prop(property), value)
	case *uint32:
		ps.spec.SetPropU32Ptr(implot.Prop(property), value)
	case uint64:
		ps.spec.SetPropU64(implot.Prop(property), value)
	case uint8:
		ps.spec.SetPropU8(implot.Prop(property), value)
	case color.Color:
		ps.spec.SetPropVec4(implot.Prop(property), ToVec4Color(value))
	case float64:
		ps.spec.SetPropdouble(implot.Prop(property), value)
	case int16:
		ps.spec.SetPropS16(implot.Prop(property), value)
	case int:
		ps.spec.SetPropS32(implot.Prop(property), value)
	case int64:
		ps.spec.SetPropS64(implot.Prop(property), value)
	default:
		panic("unsupported type")
	}

	return ps
}

// GetSpec returns underlying instance of implot.Spec (to use with implot functions).
func (ps *PlotSpec) GetSpec() *implot.Spec {
	return ps.spec
}