linuxSmartie/src/main.c

#include <signal.h>
#include <pthread.h>
#include <stdlib.h>
#include "log.h"
#include <unistd.h>
#include <stdatomic.h>
#include <sys/signalfd.h>
#include "lcd.h"
#include "cpu.h"
#include "sys.h"
#include "utils.h"

#include <string.h>

#pragma GCC diagnostic ignored "-Wunused-result"

static constexpr uint16_t refresh_interval_ms = 500;

struct control_t {
	pthread_mutex_t sensors_mtx;
	pthread_cond_t sensors_ready;
	atomic_bool g_running; // if using c11, you need atomic_store() and atomic_load();
} control;

struct sensors_t {
	struct time_date_t td;
	double loadavg[3];
	struct memory_t ram;
	struct disk_t home;
	struct network_t net;
	struct syssensors_t sys;
	struct cpu_sensors_t cpu;
	// gpu::gpuSensors gpu;
	double hass_state;
};

static void *sensors_thread(void *arg) {
	applog(LOG_START, "sensors thread started, refresh interval is %dms", refresh_interval_ms);
	struct sensors_t* sensors = arg;

	struct timespec next;
	clock_gettime(CLOCK_REALTIME, &next);

	// set up paths for the sensors
	strcpy(sensors->home.mountpoint, "/home/drath");
	strcpy(sensors->net.device, "enp7s0");
	cpu_detect(&sensors->cpu, "0000:00:18.3");
	sys_detect(&sensors->sys, "nct6775.656");

	while (control.g_running) { // NOLINT
		pthread_mutex_lock(&control.sensors_mtx);

		sys_get_load_avg(sensors->loadavg);
		sys_get_time(&sensors->td);
		sys_get_ram(&sensors->ram);
		sys_get_disk(&sensors->home);
		sys_get_network(&sensors->net);
		sys_update_sensors(&sensors->sys);

		cpu_update(&sensors->cpu);

		pthread_mutex_unlock(&control.sensors_mtx);
		pthread_cond_signal(&control.sensors_ready);

		timespec_add_ms(&next, refresh_interval_ms);
		clock_nanosleep(CLOCK_REALTIME, TIMER_ABSTIME, &next, nullptr);
	}

	applog(LOG_WRN, "sensors thread stopped");
	return nullptr;
}

static void *display_thread(void *arg) {
	applog(LOG_START, "display thread started");
	struct sensors_t* sensors = arg;
	sensors->hass_state = 1;

	struct lcd_t lcd = {
		.port = "/dev/serial/by-id/usb-Raspberry_Pi_Pico_E6616407E37AA12F-if00",
		.baud = 0010002, // from termios-baud.h, B115200
	};

	lcd_init(&lcd);

	// basic lcd setup
	lcd_set_backlight(&lcd, true);
	lcd_set_contrast(&lcd, 206);
	lcd_set_custom_char(&lcd, 0, &(lcd_glyph_t){16, 6, 9, 8, 8, 9, 6, 0});     // celcius
	lcd_set_custom_char(&lcd, 1, &(lcd_glyph_t){0, 0, 4, 10, 10, 17, 31, 0});  // delta
	lcd_set_custom_char(&lcd, 2, &(lcd_glyph_t){0, 0, 8, 21, 21, 2, 0, 0});    // home (tilde)
	lcd_set_custom_char(&lcd, 3, &(lcd_glyph_t){4, 14, 21, 4, 4, 4, 0});       // up arrow
	lcd_set_custom_char(&lcd, 4, &(lcd_glyph_t){4, 4, 4, 21, 14, 4, 0});       // down arrow

	lcd_write_buf(&lcd, 1, 9, "Waiting for sensors..."); // need to center this still
	lcd_flush_buf(&lcd);
	lcd_clear_buf(&lcd);

	while (control.g_running) { // NOLINT its updated globally bud
		struct timespec ts;
		clock_gettime(CLOCK_REALTIME, &ts);
		timespec_add_ms(&ts, (uint16_t)(refresh_interval_ms * 1.25));

		pthread_mutex_lock(&control.sensors_mtx); // lock sensor struct
		pthread_cond_timedwait(&control.sensors_ready, &control.sensors_mtx, &ts); // wait until data ready

		const uint16_t avg_fan_speed = (sensors->sys.fan_speed[0] + sensors->sys.fan_speed[1]) / 2;

		lcd_writef(&lcd, 0, 0, "%s   %s", sensors->td.date, sensors->td.time);
		lcd_writef(&lcd, 26, 0, "%.2f %.2f %.2f", sensors->loadavg[0], sensors->loadavg[1], sensors->loadavg[2]);

		lcd_writef(&lcd, 0, 1, "%d/%d%c", sensors->cpu.tctl, sensors->cpu.tccd1, 0);
		lcd_writef(&lcd, 7, 1, "%s", P_BYTES(sensors->ram.used, 0));
		lcd_writef(&lcd, 20, 1, "%d%c", sensors->sys.t_sensor, 0);

		lcd_writef(&lcd, 0, 2, "%.0fW", sensors->cpu.ptotal);
		lcd_writef(&lcd, 5, 2, "%dR", avg_fan_speed);
		lcd_writef(&lcd, 12, 2, "%c%s %c%s", 3, P_BYTES(sensors->net.upload, 1), 4, P_BYTES(sensors->net.download, 1));

		lcd_writef(&lcd, 0, 3, "%dMHz", sensors->cpu.avg_freq);
		lcd_writef(&lcd, 14, 3, "%c%s", 2, P_BYTES(sensors->home.usage.free, 1));

		pthread_mutex_unlock(&control.sensors_mtx); // unlock sensor struct

		lcd_flush_buf(&lcd);
		lcd_clear_buf(&lcd);
	}

	lcd_write_buf(&lcd, 2, 16, "Goodbye"); // center
	lcd_flush_buf(&lcd);
	lcd_close(&lcd);
	applog(LOG_WRN, "display thread stopped");
	return nullptr;
}

int main(void) {
	control.g_running = true;
	applog(LOG_START, "linuxSmartie started");

	/* signal setup */
	sigset_t sigset;
	sigemptyset(&sigset);
	sigaddset(&sigset, SIGINT);
	sigaddset(&sigset, SIGTERM);
	sigaddset(&sigset, SIGQUIT);
	pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
	struct signalfd_siginfo info;
	const int sfd = signalfd(-1, &sigset, SFD_CLOEXEC);

	/* control vars setup */
	pthread_mutex_init(&control.sensors_mtx, nullptr);
	pthread_cond_init(&control.sensors_ready, nullptr);

	/* allocate mem for the main struct */
	struct sensors_t *sensors = malloc(sizeof(struct sensors_t));

	if (sensors == nullptr) {
		applog(LOG_ERR, "cant allocate mem for the sensor struct");
		free(sensors);
		return -1;
	}
	applog(LOG_INF, "sensor struct allocated, size is %lu bytes", sizeof(struct sensors_t));

	/* thread setup */
	pthread_t display_pthread;
	pthread_t sensors_pthread;

	/* actually start the threads */
	pthread_create(&display_pthread, nullptr, display_thread, sensors);
	pthread_create(&sensors_pthread, nullptr, sensors_thread, sensors);

	read(sfd, &info, sizeof(info)); // halts on this until SIGTERM

	/* cleanup on exit signal */
	applog(LOG_WRN, "got stop signal, shutting down...");
	close(sfd);
	control.g_running = false;
	pthread_join(display_pthread, nullptr);
	pthread_join(sensors_pthread, nullptr);
	free(sensors);
	return 0;
}