1 files changed, 56 insertions, 0 deletions
diff --git a/cmd/jp/scatter.go b/cmd/jp/scatter.go
new file mode 100644
index 0000000..9df8b4a
--- /dev/null
+++ b/cmd/jp/scatter.go
@@ -0,0 +1,56 @@
+package main
+
+import (
+	"reflect"
+
+	"github.com/sgreben/jp/pkg/draw"
+
+	"github.com/sgreben/jp/pkg/plot"
+)
+
+func scatterPlotData(xvv, yvv [][]reflect.Value) (x, y []float64) {
+	for _, xv := range xvv {
+		for i := range xv {
+			if xv[i].IsValid() && xv[i].CanInterface() {
+				xvi, ok := xv[i].Interface().(float64)
+				if ok {
+					x = append(x, xvi)
+				}
+			}
+		}
+	}
+	for _, yv := range yvv {
+		for i := range yv {
+			if yv[i].IsValid() && yv[i].CanInterface() {
+				yvi, ok := yv[i].Interface().(float64)
+				if ok {
+					y = append(y, yvi)
+				}
+			}
+		}
+	}
+	return
+}
+
+func scatterPlot(xvv, yvv [][]reflect.Value, c draw.Canvas) string {
+	x, y := scatterPlotData(xvv, yvv)
+	chart := plot.NewScatterChart(c)
+	data := new(plot.DataTable)
+	data.AddColumn("x")
+	data.AddColumn("y")
+	n := len(x)
+	if len(y) > n {
+		n = len(y)
+	}
+	// If no valid xs are given, use the indices as x values.
+	if len(x) == 0 {
+		x = make([]float64, len(y))
+		for i := 0; i < len(y); i++ {
+			x[i] = float64(i)
+		}
+	}
+	for i := 0; i < n; i++ {
+		data.AddRow(x[i%len(x)], y[i%len(y)])
+	}
+	return chart.Draw(data)
+}