display-epaper/img.go

328 lines
6.8 KiB
Go

package main
import (
"context"
"embed"
"fmt"
"github.com/Crocmagnon/display-epaper/epd"
"github.com/Crocmagnon/display-epaper/fete"
"github.com/Crocmagnon/display-epaper/home_assistant"
"github.com/Crocmagnon/display-epaper/quotes"
"github.com/Crocmagnon/display-epaper/transports"
"github.com/Crocmagnon/display-epaper/weather"
"github.com/llgcode/draw2d"
"github.com/llgcode/draw2d/draw2dimg"
"image"
"image/color"
"log"
"math"
"strconv"
"strings"
"sync"
"time"
)
//go:embed icons
var icons embed.FS
const (
leftX = 20
rightX = 530
)
func getImg(ctx context.Context, nowFunc func() time.Time, transportsClient *transports.Client, feteClient *fete.Client, weatherClient *weather.Client, hassClient *home_assistant.Client) (*image.RGBA, error) {
var (
bus *transports.Passages
tram *transports.Passages
velovRoc *transports.Station
fetes *fete.Fete
wthr *weather.Prevision
msg string
)
wg := &sync.WaitGroup{}
wg.Add(6)
go func() {
defer wg.Done()
ctx, cancel := context.WithTimeout(ctx, 20*time.Second)
defer cancel()
var err error
bus, err = transportsClient.GetTCLPassages(ctx, 290)
if err != nil {
log.Println("error getting bus:", err)
}
}()
go func() {
defer wg.Done()
ctx, cancel := context.WithTimeout(ctx, 20*time.Second)
defer cancel()
var err error
tram, err = transportsClient.GetTCLPassages(ctx, 34068)
if err != nil {
log.Println("error getting tram:", err)
}
}()
go func() {
defer wg.Done()
ctx, cancel := context.WithTimeout(ctx, 20*time.Second)
defer cancel()
var err error
velovRoc, err = transportsClient.GetVelovStation(ctx, 10044)
if err != nil {
log.Println("error getting velov:", err)
}
}()
go func() {
defer wg.Done()
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
var err error
fetes, err = feteClient.GetFete(ctx, nowFunc())
if err != nil {
log.Println("error getting fetes:", err)
}
}()
go func() {
defer wg.Done()
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
var err error
wthr, err = weatherClient.GetWeather(ctx)
if err != nil {
log.Println("error getting weather:", err)
}
}()
go func() {
defer wg.Done()
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
var err error
msg, err = hassClient.GetState(ctx, "input_text.e_paper_message")
if err != nil {
log.Println("error getting hass message:", err)
}
}()
img := newWhite()
gc := draw2dimg.NewGraphicContext(img)
gc.SetFillRule(draw2d.FillRuleWinding)
gc.SetFillColor(color.RGBA{255, 255, 255, 255})
gc.SetStrokeColor(color.RGBA{0, 0, 0, 255})
wg.Wait()
if msg == "" {
msg = quotes.GetQuote(nowFunc())
}
drawTCL(gc, bus, 55)
drawTCL(gc, tram, 190)
drawVelov(gc, velovRoc, 350)
drawDate(gc, nowFunc())
drawFete(gc, fetes)
drawWeather(gc, wthr)
drawMsg(gc, msg)
return img, nil
}
func drawMsg(gc *draw2dimg.GraphicContext, quote string) {
text(gc, quote, 15, leftX, 450)
}
func drawWeather(gc *draw2dimg.GraphicContext, wthr *weather.Prevision) {
if wthr == nil {
return
}
if len(wthr.Daily) == 0 || len(wthr.Daily[0].Weather) == 0 {
log.Println("missing daily or daily weather")
return
}
daily := wthr.Daily[0]
dailyWeather := daily.Weather[0]
err := drawWeatherIcon(gc, dailyWeather)
if err != nil {
log.Println("Failed to draw weather icon:", err)
}
text(gc, formatTemp(wthr.Current.Temp), 23, leftX, 120)
text(gc, "max "+formatTemp(daily.Temp.Max), 18, 120, 45)
text(gc, fmt.Sprintf("pluie %v%%", int(math.Round(daily.Pop*100))), 18, 120, 80)
text(gc, dailyWeather.Description, 18, 120, 115)
}
func drawWeatherIcon(gc *draw2dimg.GraphicContext, dailyWeather weather.Weather) error {
icon := strings.TrimSuffix(dailyWeather.Icon, "d")
icon = strings.TrimSuffix(icon, "n")
f, err := icons.Open(fmt.Sprintf("icons/%sd.png", icon))
if err != nil {
return fmt.Errorf("opening icon: %w", err)
}
img, _, err := image.Decode(f)
if err != nil {
return fmt.Errorf("decoding icon: %w", err)
}
gc.DrawImage(img)
return nil
}
func formatTemp(temp float64) string {
return fmt.Sprintf("%v°C", int(math.Round(temp)))
}
func drawVelov(gc *draw2dimg.GraphicContext, station *transports.Station, yOffset float64) {
if station == nil {
return
}
text(gc, station.Name, 23, rightX, yOffset)
text(gc, fmt.Sprintf("V : %v - P : %v", station.BikesAvailable, station.DocksAvailable), 23, rightX, yOffset+30)
}
func drawDate(gc *draw2dimg.GraphicContext, now time.Time) {
text(gc, now.Format("15:04"), 110, leftX, 300)
text(gc, getDate(now), 30, leftX, 345)
}
func drawFete(gc *draw2dimg.GraphicContext, fetes *fete.Fete) {
if fetes == nil {
return
}
text(gc, fmt.Sprintf("On fête les %s", fetes.Name), 18, leftX, 380)
}
func drawTCL(gc *draw2dimg.GraphicContext, passages *transports.Passages, yoffset float64) {
if passages == nil {
return
}
for i, passage := range passages.Passages {
x := float64(rightX + i*120)
text(gc, passage.Ligne, 23, x, yoffset)
for j, delay := range passage.Delays {
y := yoffset + float64(j+1)*35
text(gc, delay, 23, x, y)
if j >= 2 { // limit number of delays displayed
break
}
}
}
}
func text(gc *draw2dimg.GraphicContext, s string, size, x, y float64) {
gc.SetFillColor(color.RGBA{0, 0, 0, 255})
gc.SetFontData(draw2d.FontData{Name: fontName})
gc.SetFontSize(size)
gc.FillStringAt(s, x, y)
}
func newWhite() *image.RGBA {
img := image.NewRGBA(image.Rect(0, 0, epd.Width, epd.Height))
for y := 0; y < epd.Height; y++ {
for x := 0; x < epd.Width; x++ {
img.Set(x, y, color.White)
}
}
return img
}
func rect(gc *draw2dimg.GraphicContext, x1, y1, x2, y2 float64) {
gc.BeginPath()
gc.MoveTo(x1, y1)
gc.LineTo(x2, y1)
gc.LineTo(x2, y2)
gc.LineTo(x1, y2)
gc.Close()
gc.FillStroke()
}
func getDate(now time.Time) string {
return fmt.Sprintf("%v %v %v", getDow(now), getDay(now), getMonth(now))
}
func getDow(now time.Time) string {
switch now.Weekday() {
case time.Monday:
return "lun"
case time.Tuesday:
return "mar"
case time.Wednesday:
return "mer"
case time.Thursday:
return "jeu"
case time.Friday:
return "ven"
case time.Saturday:
return "sam"
case time.Sunday:
return "dim"
}
return "?"
}
func getDay(now time.Time) string {
if now.Day() == 1 {
return "1er"
}
return strconv.Itoa(now.Day())
}
func getMonth(t time.Time) string {
switch t.Month() {
case time.January:
return "jan."
case time.February:
return "fev."
case time.March:
return "mars"
case time.April:
return "avr."
case time.May:
return "mai"
case time.June:
return "juin"
case time.July:
return "juil."
case time.August:
return "août"
case time.September:
return "sept."
case time.October:
return "oct."
case time.November:
return "nov."
case time.December:
return "dec."
}
return "?"
}