]> git.scottworley.com Git - batteryviewer/blame - batteryviewer.c
Extract draw_data()
[batteryviewer] / batteryviewer.c
CommitLineData
ef0963b8
SW
1// For vasprintf
2#define _GNU_SOURCE
3
5a945372 4#include "chart.h"
0f9e308a
SW
5#include <ctype.h>
6#include <errno.h>
eda1f8c1 7#include <glob.h>
79ac753b 8#include <gtk/gtk.h>
0f9e308a 9#include <math.h>
ef0963b8 10#include <stdarg.h>
0f9e308a
SW
11#include <stdio.h>
12#include <stdlib.h>
eda1f8c1 13#include <string.h>
0f9e308a 14
ef0963b8
SW
15char *sasprintf(const char *fmt, ...) {
16 char *result;
17 va_list args;
18 va_start(args, fmt);
19 if (vasprintf(&result, fmt, args) == -1) {
20 fprintf(stderr, "Can't allocate memory!?\n");
21 abort();
22 }
23 va_end(args);
24 return result;
25}
26
fc1a2773 27struct State {
7281755a
SW
28 char *voltage_filename;
29 char *current_filename;
fc1a2773
SW
30 BVChart *voltage;
31 BVChart *current;
32};
33
0f9e308a
SW
34static float fatof(const char *filename) {
35 FILE *f = fopen(filename, "r");
36 if (f == NULL) {
37 return NAN;
38 }
39
40 const int bufsize = 1024;
41 char buf[bufsize];
42 const int read = fread(buf, sizeof(char), bufsize, f);
43 if (read == 0 || read == bufsize || ferror(f)) {
44 fclose(f);
45 return NAN;
46 }
47 fclose(f);
48
49 char *end;
50 errno = 0;
51 float val = strtof(buf, &end);
52 int parsed = end - buf;
53 gboolean parsed_all = parsed == read;
54 gboolean parsed_all_but_space = parsed == read - 1 && isspace(*end);
55 gboolean parse_ok = parsed_all || parsed_all_but_space;
56 if (errno != 0 || parsed == 0 || !parse_ok) {
57 return NAN;
58 }
59 return val;
60}
61
7281755a 62static gboolean collect_data(struct State *state) {
51a39b46 63 float now = g_get_monotonic_time() / 1e6;
cf9a7d38
SW
64 float voltage = fatof(state->voltage_filename);
65 float current = fatof(state->current_filename);
dc8fff78
SW
66 if (!isnan(voltage)) {
67 bv_chart_add_point(state->voltage, now, voltage);
68 gtk_widget_queue_draw(GTK_WIDGET(state->voltage));
69 }
70 if (!isnan(current)) {
71 bv_chart_add_point(state->current, now, current);
72 gtk_widget_queue_draw(GTK_WIDGET(state->current));
73 }
6f50819e
SW
74 return TRUE;
75}
76
fc1a2773 77static void activate(GtkApplication *app, gpointer user_data) {
79ac753b 78
cecfa926 79 GtkWidget *window = gtk_application_window_new(app);
a8d35c4c 80 gtk_window_set_title(GTK_WINDOW(window), "BatteryViewer");
79ac753b
SW
81 gtk_window_set_default_size(GTK_WINDOW(window), 200, 200);
82
1bd5b327
SW
83 GtkWidget *box = gtk_box_new(GTK_ORIENTATION_VERTICAL, 1);
84 gtk_container_add(GTK_CONTAINER(window), box);
85
fc1a2773
SW
86 struct State *state = (struct State *)user_data;
87 state->voltage = BV_CHART(bv_chart_new());
1bd5b327
SW
88 gboolean expand = TRUE;
89 gboolean fill = TRUE;
90 guint padding = 0;
fc1a2773
SW
91 gtk_box_pack_start(GTK_BOX(box), GTK_WIDGET(state->voltage), expand, fill,
92 padding);
1bd5b327 93
fc1a2773 94 state->current = BV_CHART(bv_chart_new());
fc1a2773
SW
95 gtk_box_pack_end(GTK_BOX(box), GTK_WIDGET(state->current), expand, fill,
96 padding);
79ac753b 97
a9a80e4e 98 gtk_widget_show_all(window);
6f50819e
SW
99
100 g_timeout_add_seconds(1, (GSourceFunc)collect_data, user_data);
79ac753b
SW
101}
102
eda1f8c1
SW
103char *find_battery_dir() {
104 glob_t globbuf = {.gl_offs = 0};
105 char pattern[] = "/sys/class/power_supply/*[Bb][Aa][Tt]*";
106 if (glob(pattern, GLOB_ERR, NULL, &globbuf) != 0 || globbuf.gl_pathc == 0) {
107 fprintf(stderr, "Could not find battery dir %s\n", pattern);
108 exit(1);
109 }
110 if (globbuf.gl_pathc != 1) {
111 // TODO: Filter for directories that have {voltage,current}_now files.
112 // TODO: Support multiple batteries
113 fprintf(stderr, "Multiple batteries found. Arbitrarily choosing %s\n",
114 globbuf.gl_pathv[0]);
115 }
116 char *dir = strdup(globbuf.gl_pathv[0]);
117 globfree(&globbuf);
118 return dir;
119}
120
79ac753b 121int main(int argc, char **argv) {
eda1f8c1 122 char *battery_dir = find_battery_dir();
7281755a 123 struct State state = {
ef0963b8
SW
124 .voltage_filename = sasprintf("%s/voltage_now", battery_dir),
125 .current_filename = sasprintf("%s/current_now", battery_dir),
7281755a 126 };
79ac753b 127
a8d35c4c
SW
128 GtkApplication *app = gtk_application_new("com.scottworley.batteryviewer",
129 G_APPLICATION_DEFAULT_FLAGS);
fc1a2773 130 g_signal_connect(app, "activate", G_CALLBACK(activate), &state);
cecfa926 131 int status = g_application_run(G_APPLICATION(app), argc, argv);
79ac753b
SW
132 g_object_unref(app);
133
134 return status;
135}