2022-01-23 18:57:19 -05:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
|
|
|
"encoding/csv"
|
2022-02-03 12:41:53 -05:00
|
|
|
"flag"
|
2022-01-23 18:57:19 -05:00
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"io/ioutil"
|
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
"strconv"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
type ActivityLine struct {
|
|
|
|
Date time.Time
|
|
|
|
CaloriesEstimate uint64
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
2022-02-03 12:41:53 -05:00
|
|
|
var csvPathArg string
|
|
|
|
var endDateArg string
|
|
|
|
var startDateArg string
|
|
|
|
var outputPath string
|
|
|
|
flag.StringVar(&csvPathArg, "csv", "heat.csv", "Specify csv path, default is heat.csv.")
|
|
|
|
flag.StringVar(&outputPath, "o", "heatmap.svg", "Specify output path, default is heatmeap.svg.")
|
|
|
|
flag.StringVar(&startDateArg, "s", "", "Specify start date (2006-01-02). Default is 1 year previous to end date.")
|
|
|
|
flag.StringVar(&endDateArg, "e", "", "Specify end date (2006-01-02). Default is today.")
|
|
|
|
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
endDate := time.Date(time.Now().Year(), time.Now().Month(), time.Now().Day(), 0, 0, 0, 0, time.Local)
|
|
|
|
if endDateArg != "" {
|
|
|
|
var err error
|
|
|
|
endDate, err = time.Parse("2006-01-02", endDateArg)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println("failed to parse end time")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
start := time.Date(endDate.Year()-1, endDate.Month(), endDate.Day(), 0, 0, 0, 0, time.Local)
|
|
|
|
if startDateArg != "" {
|
|
|
|
var err error
|
|
|
|
start, err = time.Parse("2006-01-02", startDateArg)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println("failed to parse start time")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
csvFile, _ := os.Open(csvPathArg)
|
2022-01-23 18:57:19 -05:00
|
|
|
reader := csv.NewReader(bufio.NewReader(csvFile))
|
|
|
|
var length int = 17
|
|
|
|
var days = make(map[time.Time]uint64)
|
|
|
|
for {
|
|
|
|
line, error := reader.Read()
|
|
|
|
if error == io.EOF {
|
|
|
|
break
|
|
|
|
} else if error != nil {
|
|
|
|
log.Fatal(error)
|
|
|
|
}
|
|
|
|
thisTime, error := time.Parse("2006-01-02", line[0])
|
|
|
|
thisTime = time.Date(thisTime.Year(), thisTime.Month(), thisTime.Day(), 0, 0, 0, 0, time.Local)
|
|
|
|
if error != nil {
|
|
|
|
fmt.Println(error)
|
|
|
|
break
|
|
|
|
}
|
|
|
|
cals, error := strconv.ParseUint(line[1], 10, 64)
|
|
|
|
if error != nil {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
days[thisTime] = days[thisTime] + cals
|
|
|
|
}
|
|
|
|
svg := "<svg width=\"960\" height=\"152\"><g transform=\"translate(29.5,24)\"><g fill=\"none\" stroke=\"#ccc\">"
|
|
|
|
datePointer := start
|
|
|
|
var week int = 0
|
|
|
|
|
|
|
|
var monthChange = false
|
|
|
|
var monthLabelSpots = make(map[string]int)
|
|
|
|
for {
|
|
|
|
|
|
|
|
var calsOnDay = days[datePointer]
|
|
|
|
|
|
|
|
if calsOnDay > 0 {
|
|
|
|
svg = fmt.Sprintf("%s\n<rect width=\"%v\" height=\"%v\" x=\"%v\" y=\"%v\" fill=\"#00FF00\"></rect>", svg, length, length, week*length, length*int(datePointer.Weekday()))
|
|
|
|
} else {
|
|
|
|
svg = fmt.Sprintf("%s\n<rect width=\"%v\" height=\"%v\" x=\"%v\" y=\"%v\" ></rect>", svg, length, length, week*length, length*int(datePointer.Weekday()))
|
|
|
|
}
|
|
|
|
datePointer = datePointer.Add(time.Hour * 24)
|
|
|
|
|
2022-02-03 12:41:53 -05:00
|
|
|
if datePointer.After(endDate) {
|
2022-01-23 18:57:19 -05:00
|
|
|
break
|
|
|
|
}
|
|
|
|
if datePointer.Day() == 1 {
|
|
|
|
monthChange = true
|
|
|
|
}
|
|
|
|
if datePointer.Weekday() == time.Sunday {
|
|
|
|
week++
|
|
|
|
if monthChange {
|
|
|
|
monthLabelSpots[datePointer.Month().String()] = week * length
|
|
|
|
monthChange = false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
svg = fmt.Sprintf("%s\n</g></g>", svg)
|
|
|
|
|
|
|
|
//week labels
|
|
|
|
//tODO: parameterize
|
|
|
|
svg = fmt.Sprintf("%s<g transform=\"translate(6, 16)\"><text x=\"0\" y=\"30\">Mon</text><text x=\"0\" y=\"64\">Wed</text><text x=\"0\" y=\"97\">Fri</text></g>", svg)
|
|
|
|
|
|
|
|
//month labels
|
|
|
|
svg = fmt.Sprintf("%s<g transform=\"translate(30,14)\">", svg)
|
|
|
|
for monthName := range monthLabelSpots {
|
|
|
|
svg = fmt.Sprintf("%s<text x=\"%v\" y=\"0\">%v</text>", svg, monthLabelSpots[monthName], monthName[:3])
|
|
|
|
}
|
|
|
|
svg = fmt.Sprintf("%s</g>", svg)
|
|
|
|
|
|
|
|
svg = fmt.Sprintf("%s</svg>", svg)
|
2022-02-03 12:41:53 -05:00
|
|
|
ioutil.WriteFile(outputPath, []byte(svg), 0644)
|
2022-01-23 18:57:19 -05:00
|
|
|
}
|