// saving.go
package main

import (
	"encoding/csv"
	"log"
	"os"
	"sort"
	"strconv"
)

func SaveToCsvFile(data []StatForTime, keymap map[uint8]string, path string, isOnlySum bool) {

	numKeysInt := make([]int, 0)
	for key := range keymap {
		numKeysInt = append(numKeysInt, int(key))
	}
	sort.Ints(numKeysInt)
	numKeys := make([]uint8, 0)
	for _, key := range numKeysInt {
		numKeys = append(numKeys, uint8(key))
	}

	titleLine := make([]string, 0)
	titleLine = append(titleLine, "Time")
	if !isOnlySum {
		for _, key := range numKeys {
			titleLine = append(titleLine, keymap[key])
		}
	}
	titleLine = append(titleLine, "Sum")

	table := make([][]string, 0)
	table = append(table, titleLine)
	for _, rec := range data {
		line := make([]string, 0)
		line = append(line, strconv.Itoa(int(rec.time)))
		var sum int
		for _, key := range numKeys {
			sum += rec.keys[key]
			if !isOnlySum {
				line = append(line, strconv.Itoa(rec.keys[key]))
			}
		}
		line = append(line, strconv.Itoa(sum))
		table = append(table, line)
	}

	csvfile, err := os.Create(path)
	if err != nil {
		log.Fatal(err)
		return
	}
	defer csvfile.Close()
	writer := csv.NewWriter(csvfile)
	writer.WriteAll(table)
}