From 18bf177fd843f49c21abbe79385917f853ed28b4 Mon Sep 17 00:00:00 2001
From: gucio321 <gucio321@protonmail.com>
Date: Tue, 25 Jun 2024 10:29:57 +0200
Subject: [PATCH] enable/disable 2nd and 3rd axis if needed

---
 Plot.go | 78 +++++++++++++++++++++++++++++++++++++++------------------
 1 file changed, 54 insertions(+), 24 deletions(-)

diff --git a/Plot.go b/Plot.go
index dc2147b0..4b0966bc 100644
--- a/Plot.go
+++ b/Plot.go
@@ -6,6 +6,20 @@ import (
 	imgui "github.com/AllenDang/cimgui-go"
 )
 
+type (
+	PlotXAxis = imgui.PlotAxisEnum
+	PlotYAxis = imgui.PlotAxisEnum
+)
+
+const (
+	AxisX1 = PlotXAxis(imgui.AxisX1)
+	AxisX2 = PlotXAxis(imgui.AxisX2)
+	AxisX3 = PlotXAxis(imgui.AxisX3)
+	AxisY1 = PlotYAxis(imgui.AxisY1)
+	AxisY2 = PlotYAxis(imgui.AxisY2)
+	AxisY3 = PlotYAxis(imgui.AxisY3)
+)
+
 // PlotWidget is implemented by all the particular plots, which can be used
 // in (*PlotCanvasWidget).Plots.
 type PlotWidget interface {
@@ -75,6 +89,32 @@ func Plot(title string) *PlotCanvasWidget {
 	}
 }
 
+func (p *PlotCanvasWidget) SetXAxisLabel(axis PlotXAxis, label string) *PlotCanvasWidget {
+	switch axis {
+	case AxisX1:
+		p.xLabel = label
+	case AxisX2:
+		p.y2Label = label
+	case AxisX3:
+		p.y3Label = label
+	}
+
+	return p
+}
+
+func (p *PlotCanvasWidget) SetYAxisLabel(axis PlotYAxis, label string) *PlotCanvasWidget {
+	switch axis {
+	case AxisY1:
+		p.yLabel = label
+	case AxisY2:
+		p.y2Label = label
+	case AxisY3:
+		p.y3Label = label
+	}
+
+	return p
+}
+
 // AxisLimits sets X and Y axis limits.
 func (p *PlotCanvasWidget) AxisLimits(xmin, xmax, ymin, ymax float64, cond ExecCondition) *PlotCanvasWidget {
 	p.xMin = xmin
@@ -222,17 +262,21 @@ func (p *PlotCanvasWidget) Build() {
 			imgui.PlotAxisFlags(p.yFlags),
 		)
 
-		imgui.PlotSetupAxisV(
-			imgui.AxisY2,
-			Context.FontAtlas.RegisterString(p.y2Label),
-			imgui.PlotAxisFlags(p.y2Flags),
-		)
+		if p.y2Label != "" {
+			imgui.PlotSetupAxisV(
+				imgui.AxisY2,
+				Context.FontAtlas.RegisterString(p.y2Label),
+				imgui.PlotAxisFlags(p.y2Flags),
+			)
+		}
 
-		imgui.PlotSetupAxisV(
-			imgui.AxisY3,
-			Context.FontAtlas.RegisterString(p.y3Label),
-			imgui.PlotAxisFlags(p.y3Flags),
-		)
+		if p.y3Label != "" {
+			imgui.PlotSetupAxisV(
+				imgui.AxisY3,
+				Context.FontAtlas.RegisterString(p.y3Label),
+				imgui.PlotAxisFlags(p.y3Flags),
+			)
+		}
 
 		for _, plot := range p.plots {
 			plot.Plot()
@@ -242,20 +286,6 @@ func (p *PlotCanvasWidget) Build() {
 	}
 }
 
-type (
-	PlotXAxis = imgui.PlotAxisEnum
-	PlotYAxis = imgui.PlotAxisEnum
-)
-
-const (
-	AxisX1 = PlotXAxis(imgui.AxisX1)
-	AxisX2 = PlotXAxis(imgui.AxisX2)
-	AxisX3 = PlotXAxis(imgui.AxisX3)
-	AxisY1 = PlotYAxis(imgui.AxisY1)
-	AxisY2 = PlotYAxis(imgui.AxisY2)
-	AxisY3 = PlotYAxis(imgui.AxisY3)
-)
-
 func SwitchPlotAxes(x PlotXAxis, y PlotYAxis) PlotWidget {
 	return Custom(func() {
 		imgui.PlotSetAxes(imgui.PlotAxisEnum(x), imgui.PlotAxisEnum(y))