1
0
mirror of https://github.com/nikolaydubina/calendarheatmap.git synced 2025-02-12 16:06:06 +02:00
calendarheatmap/main.go
2020-12-08 07:23:11 +08:00

51 lines
1.3 KiB
Go

package main
import (
"encoding/json"
"flag"
"image/color"
"io/ioutil"
"log"
"os"
"github.com/nikolaydubina/calendarheatmap/charts"
"github.com/nikolaydubina/calendarheatmap/colorscales"
)
func main() {
os.Setenv("CALENDAR_HEATMAP_ASSETS_PATH", "charts/assets")
colorScale := *flag.String("colorscale", "PuBu9", "refer to colorscales for examples")
labels := *flag.Bool("labels", true, "labels for weekday and months")
monthSep := *flag.Bool("monthsep", true, "render month separator")
outputFormat := *flag.String("output", "png", "output format (png, jpeg, gif, svg)")
locale := *flag.String("locale", "en_US", "locale of labels (en_US, ko_KR)")
flag.Parse()
data, err := ioutil.ReadAll(os.Stdin)
if err != nil {
log.Fatal(err)
}
var counts map[string]int
if err := json.Unmarshal(data, &counts); err != nil {
log.Fatal(err)
}
conf := charts.HeatmapConfig{
Counts: counts,
ColorScale: colorscales.LoadColorScale(colorScale),
DrawMonthSeparator: monthSep,
DrawLabels: labels,
Margin: 30,
BoxSize: 150,
TextWidthLeft: 350,
TextHeightTop: 200,
TextColor: color.RGBA{100, 100, 100, 255},
BorderColor: color.RGBA{200, 200, 200, 255},
Locale: locale,
Format: outputFormat,
}
charts.WriteHeatmap(conf, os.Stdout)
}