go(stats): add functions for manipulating Stats

stats/stats.go

@@ -3,6 +3,14 @@
 
 package stats
 
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+)
+
 type FuncStats struct {
 	BenchName string
 	Results   []float64
@@ -15,3 +23,38 @@ type Stats struct {
 	Iterations     int
 	Generations    int
 }
+
+func GetFuncStats(funcName string, results []float64) FuncStats {
+	f := FuncStats{
+		BenchName: funcName,
+		Results:   results,
+	}
+
+	return f
+}
+
+func GetStats(algo string, dimens int, benchFuncStats []FuncStats, iterations, generations int) Stats {
+	s := Stats{
+		Algo:           algo,
+		Dimens:         dimens,
+		BenchFuncStats: benchFuncStats,
+		Iterations:     iterations,
+		Generations:    generations,
+	}
+
+	return s
+}
+
+func SaveStats(stats Stats) {
+	if j, err := json.MarshalIndent(stats, "", "  "); err != nil {
+		log.Fatal(err)
+	} else {
+		fmt.Fprintln(os.Stderr, j)
+
+		log.Println("saving json stats to: test.json")
+
+		if err = ioutil.WriteFile("test.json", j, 0o600); err != nil {
+			log.Println("error saving stats to file:", err)
+		}
+	}
+}