WeatherFXLite.cpp

/*
    weatherfxlite
    A lite replacement for the venerable Bushnell WeatherFX station. RIP.
    Copyright (C) 2024 iAchieved.it LLC

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

#include <math.h>
#include "WeatherFXLite.h"
#include "config.h"
#include <QThread>
#include "ZmqListener.h"
#include "icons/gps_icon.h"


#define CURRENT_CONDITION_TICKS 60
#define CURRENT_FORECAST_TICKS 60

WeatherFXLite::WeatherFXLite() {

  // Listen for optional GPS coordinates to be published on ZeroMQ
  QString zmqAddress("tcp://localhost:11205");
  ZmqListener* listener = new ZmqListener(zmqAddress);
  QThread* thread = new QThread();
  listener->moveToThread(thread);
  connect(thread,
          SIGNAL(started()),
          listener,
          SLOT(startListening()));
  connect(listener,
          SIGNAL(gpsMessageReceived(QString)),
          this,
          SLOT(gpsCoordinatesReceived(QString)));

  // Set up the UI
  window = new QWidget();
  ui.setupUi(window);

  // Pin the condition label to its 60pt height so font auto-shrinking doesn't reflow the layout
  {
    QFont f = ui.currentCondition->font();
    f.setPointSize(60);
    ui.currentCondition->setFixedHeight(QFontMetrics(f).height());
  }

  // Pin the temperature label to the widest 2-digit width so 3-digit values trigger auto-shrinking
  {
    QFont f = ui.currentTemperature->font();
    f.setPointSize(114);
    ui.currentTemperature->setFixedWidth(QFontMetrics(f).horizontalAdvance("99°"));
  }

  // Pin the wind speed label to the widest 2-digit width so 3-digit values trigger auto-shrinking
  {
    QFont f = ui.windSpeed->font();
    f.setPointSize(36);
    ui.windSpeed->setFixedWidth(QFontMetrics(f).horizontalAdvance("99"));
  }

  window->setStyleSheet("background-color:black;");

#ifdef METRIC
  ui.windUnits->setText("kph");
#endif

#ifdef Q_OS_LINUX
  window->setWindowFlags(Qt::FramelessWindowHint);
  window->setWindowState(Qt::WindowFullScreen);
  QRect screenRect = QApplication::desktop()->screenGeometry(1);
  window->move(QPoint(screenRect.x(), screenRect.y()));
#endif

  ui.versionLabel->setText("v1.0.2");

  window->show();

  // Use the Apple WeatherKit API
  weatherAPI = new WeatherKitAPI();  
  
  timer = new QTimer(this);

  connect(weatherAPI, SIGNAL(currentConditionsUpdate()), this, SLOT(updateWeatherDisplay()));
  connect(weatherAPI, SIGNAL(currentForecastUpdate()), this, SLOT(updateForecastDisplay()));
  connect(timer, SIGNAL(timeout()), this, SLOT(timerTick()));

  // Set up GPS scene
  gpsScene = new QGraphicsScene();
  QPixmap icon;
  if (icon.loadFromData(gps_png, gps_png_len, "png")) {
    gpsItem = new QGraphicsPixmapItem(icon);
    gpsScene->addItem(gpsItem);
  }
  ui.gpsAvailable->hide();
  ui.gpsAvailable->setScene(gpsScene);

  timer->start(1000); // 1 second

  // Start the thread to listen for GPS coordinates
  thread->start();

}

void WeatherFXLite::gpsCoordinatesReceived(const QString& message) {

  qDebug() << "Received GPS coordinates: " << message;

  lastGPSUpdate = QDateTime().currentDateTime().toSecsSinceEpoch();
  qDebug() << "Last GPS update: " << lastGPSUpdate;

  weatherAPI->setCurrentLocation(message);

  ui.gpsAvailable->show();

}

void WeatherFXLite::updateWeatherDisplay(void) {

  CurrentConditions current = weatherAPI->getCurrentConditions();

  ui.currentCondition->setText(current.condition);
  {
    QFont f = ui.currentCondition->font();
    f.setPointSize(60);
    QFontMetrics fm(f);
    while (fm.horizontalAdvance(current.condition) > ui.currentCondition->width() && f.pointSize() > 20) {
      f.setPointSize(f.pointSize() - 1);
      fm = QFontMetrics(f);
    }
    ui.currentCondition->setFont(f);
  }
  {
    QString tempText = QString::number(current.temperature) + QString("°");
    ui.currentTemperature->setText(tempText);
    QFont f = ui.currentTemperature->font();
    f.setPointSize(114);
    QFontMetrics fm(f);
    while (fm.horizontalAdvance(tempText) > ui.currentTemperature->width() && f.pointSize() > 60) {
      f.setPointSize(f.pointSize() - 1);
      fm = QFontMetrics(f);
    }
    ui.currentTemperature->setFont(f);
  }
  {
    QString speedText = QString::number(current.windSpeed);
    ui.windSpeed->setText(speedText);
    QFont f = ui.windSpeed->font();
    f.setPointSize(36);
    QFontMetrics fm(f);
    while (fm.horizontalAdvance(speedText) > ui.windSpeed->width() && f.pointSize() > 18) {
      f.setPointSize(f.pointSize() - 1);
      fm = QFontMetrics(f);
    }
    ui.windSpeed->setFont(f);
  }
  ui.windArrow->setDirection(current.windDirection);

  std::string background = "background-color:" + backgroundForTemperature(current.temperature) + ";";
  window->setStyleSheet(background.c_str());

  if (scene == nullptr) {
    scene = new QGraphicsScene();
  }

  if (item) {
    delete item;
  }

  QPixmap icon;
  if (icon.loadFromData(current.icon.data, current.icon.len, "png")) {
    item = new QGraphicsPixmapItem(icon);
    scene->addItem(item);
    ui.graphicsView->setScene(scene);
  } else {
    qDebug() << "Unable to load icon data";
  }

  qint64 now = QDateTime().currentDateTime().toSecsSinceEpoch();
  if (now - lastGPSUpdate > 60) {
    qDebug() << "GPS data stale";
    ui.gpsAvailable->hide();
  }

}

void WeatherFXLite::updateForecastDisplay(void) {
  CurrentConditions current = weatherAPI->getCurrentConditions();
  ui.hiTemp->setText(QString::number(current.high) + QString("°"));
  ui.loTemp->setText(QString::number(current.low) +  QString("°"));
  ui.precipChance->setText(QString::number(current.precipitationChance) + "%");
  if (current.precipWeatherIcon.data) {
    QPixmap px;
    px.loadFromData(current.precipWeatherIcon.data, current.precipWeatherIcon.len, "png");
    ui.precipIcon->setPixmap(px.scaled(48, 48, Qt::KeepAspectRatio, Qt::SmoothTransformation));
  }
}

void WeatherFXLite::timerTick(void) {

  currentConditionTicks++; currentForecastTicks++;

  QDateTime now = QDateTime().currentDateTime();

#ifdef TWENTYFOUR_HOUR_FORMAT
  QString timeNow = now.toString("HH:mm");
#else
  QString timeNow = now.toString("h:mm A");
#endif

  QString dateNow = now.toString("MMM dd");
  ui.currentTime->setText(timeNow);
  ui.currentDate->setText(dateNow);

  if (currentForecastTicks % CURRENT_FORECAST_TICKS == 0 || boot) {
    weatherAPI->updateCurrentForecast();
    currentForecastTicks = 0;
  }

  if (currentConditionTicks % CURRENT_CONDITION_TICKS == 0 || boot) {
    weatherAPI->updateCurrentConditions();
    currentConditionTicks = 0;
    boot = false;
  }

}

const std::string backgrounds[] = {
  "#ABA5C2", // 0s
  "#ABA5C2", // 10s
  "#ABA5C2", // 20s
  "#0089C6", // 30s
  "#262A62", // 40s
  "#35713D", // 50s
  "#74AB46", // 60s
  "#F3D13C", // 70s
  "#D37733", // 80s
  "#BB352B"  // 90+
};

/// @brief Look up background color for temperature
/// @param temp 
/// @return 
std::string WeatherFXLite::backgroundForTemperature(short temp) {

#ifdef METRIC
  // Convert to Fahrenheit before indexing
  temp = floor((temp * 9.0) / 5.0 + 32);
#endif

  qDebug() << "Temperature: " << temp;

  int tIndex = temp/10;
  if (tIndex <= 0) tIndex = 0;
  if (tIndex >= 9) tIndex = 9;
  return backgrounds[tIndex];
}