From b2ebdea44a3b5eafd13246cfa815ec98b836e5a9 Mon Sep 17 00:00:00 2001 From: Kenny Levinsen Date: Sun, 13 Sep 2020 03:18:14 +0200 Subject: [PATCH] Implement time ramp --- main.c | 280 ++++++++++++++++++++++++++++++++++++++++++++++++++------- 1 file changed, 247 insertions(+), 33 deletions(-) diff --git a/main.c b/main.c index 5f6ac12..1328390 100644 --- a/main.c +++ b/main.c @@ -1,5 +1,7 @@ +#define _XOPEN_SOURCE 700 #define _POSIX_C_SOURCE 200809L #include +#include #include #include #include @@ -8,10 +10,37 @@ #include #include #include +#include +#include #include "wlr-gamma-control-unstable-v1-client-protocol.h" +enum state { + HIGH_TEMP, + LOW_TEMP, + ANIMATING_TO_HIGH, + ANIMATING_TO_LOW, +}; + +struct context { + double gamma; + time_t start_time; + time_t stop_time; + int high_temp; + int low_temp; + int duration; + + int cur_temp; + bool new_output; + enum state state; + time_t animation_start; + + struct wl_list outputs; +}; + struct output { + struct context *context; struct wl_output *wl_output; + int id; struct zwlr_gamma_control_v1 *gamma_control; uint32_t ramp_size; int table_fd; @@ -19,7 +48,6 @@ struct output { struct wl_list link; }; -static struct wl_list outputs; static struct zwlr_gamma_control_manager_v1 *gamma_control_manager = NULL; static int create_anonymous_file(off_t size) { @@ -68,6 +96,7 @@ static void gamma_control_handle_gamma_size(void *data, struct output *output = data; output->ramp_size = ramp_size; output->table_fd = create_gamma_table(ramp_size, &output->table); + output->context->new_output = true; if (output->table_fd < 0) { exit(EXIT_FAILURE); } @@ -76,7 +105,6 @@ static void gamma_control_handle_gamma_size(void *data, static void gamma_control_handle_failed(void *data, struct zwlr_gamma_control_v1 *gamma_control) { fprintf(stderr, "failed to set gamma table\n"); - exit(EXIT_FAILURE); } static const struct zwlr_gamma_control_v1_listener gamma_control_listener = { @@ -84,13 +112,29 @@ static const struct zwlr_gamma_control_v1_listener gamma_control_listener = { .failed = gamma_control_handle_failed, }; +static bool setup_output(struct output *output) { + if (gamma_control_manager == NULL || output->gamma_control != NULL) { + return false; + } + output->gamma_control = zwlr_gamma_control_manager_v1_get_gamma_control( + gamma_control_manager, output->wl_output); + zwlr_gamma_control_v1_add_listener(output->gamma_control, + &gamma_control_listener, output); + return true; +} + static void registry_handle_global(void *data, struct wl_registry *registry, uint32_t name, const char *interface, uint32_t version) { + struct context *ctx = (struct context *)data; if (strcmp(interface, wl_output_interface.name) == 0) { struct output *output = calloc(1, sizeof(struct output)); + output->id = name; output->wl_output = wl_registry_bind(registry, name, &wl_output_interface, 1); - wl_list_insert(&outputs, &output->link); + output->table_fd = -1; + output->context = ctx; + wl_list_insert(&ctx->outputs, &output->link); + setup_output(output); } else if (strcmp(interface, zwlr_gamma_control_manager_v1_interface.name) == 0) { gamma_control_manager = wl_registry_bind(registry, name, @@ -100,7 +144,20 @@ static void registry_handle_global(void *data, struct wl_registry *registry, static void registry_handle_global_remove(void *data, struct wl_registry *registry, uint32_t name) { - // Who cares? + struct context *ctx = (struct context *)data; + struct output *output, *tmp; + wl_list_for_each_safe(output, tmp, &ctx->outputs, link) { + if (output->id == name) { + if (output->gamma_control != NULL) { + zwlr_gamma_control_v1_destroy(output->gamma_control); + } + if (output->table_fd != -1) { + close(output->table_fd); + } + wl_list_remove(&output->link); + break; + } + } } static const struct wl_registry_listener registry_listener = { @@ -178,6 +235,13 @@ static void xyz_to_srgb(double x, double y, double z, double *r, double *g, doub *b = srgb_gamma(clamp(0.0556434 * x - 0.2040259 * y + 1.0572252 * z), 2.2); } +static void srgb_normalize(double *r, double *g, double *b) { + double maxw = fmaxl(*r, fmaxl(*g, *b)); + *r /= maxw; + *g /= maxw; + *b /= maxw; +} + static void calc_whitepoint(int temp, double *rw, double *gw, double *bw) { if (temp == 6500) { *rw = *gw = *bw = 1.0; @@ -193,31 +257,193 @@ static void calc_whitepoint(int temp, double *rw, double *gw, double *bw) { double z = 1.0 - x - y; xyz_to_srgb(x, y, z, rw, gw, bw); + srgb_normalize(rw, gw, bw); +} - double maxw = fmaxl(*rw, fmaxl(*gw, *bw)); - *rw /= maxw; - *gw /= maxw; - *bw /= maxw; +static void set_temperature(struct context *ctx) { + double rw, gw, bw; + calc_whitepoint(ctx->cur_temp, &rw, &gw, &bw); + + struct output *output; + wl_list_for_each(output, &ctx->outputs, link) { + if (output->gamma_control == NULL || output->table_fd == -1) { + continue; + } + fill_gamma_table(output->table, output->ramp_size, + rw, gw, bw, ctx->gamma); + lseek(output->table_fd, 0, SEEK_SET); + zwlr_gamma_control_v1_set_gamma(output->gamma_control, + output->table_fd); + } +} + +static void update_temperature(struct context *ctx) { + time_t now = time(NULL); + time_t t = now % 86400; + int temp, temp_pos; + double time_pos; + + switch (ctx->state) { + case HIGH_TEMP: + if (t > ctx->stop_time || t < ctx->start_time) { + ctx->state = ANIMATING_TO_LOW; + ctx->animation_start = now; + } + temp = ctx->high_temp; + break; + case LOW_TEMP: + if (t > ctx->start_time || t < ctx->stop_time) { + ctx->state = ANIMATING_TO_HIGH; + ctx->animation_start = now; + } + temp = ctx->low_temp; + break; + case ANIMATING_TO_HIGH: + if (now > ctx->animation_start + ctx->duration) { + ctx->state = HIGH_TEMP; + } + time_pos = clamp(((double)now - (double)ctx->animation_start) / (double)ctx->duration); + temp_pos = (double)(ctx->high_temp - ctx->low_temp) * time_pos; + temp = ctx->low_temp + temp_pos; + break; + case ANIMATING_TO_LOW: + if (now > ctx->animation_start + ctx->duration) { + ctx->state = LOW_TEMP; + } + time_pos = clamp(((double)now - (double)ctx->animation_start) / (double)ctx->duration); + temp_pos = (double)(ctx->high_temp - ctx->low_temp) * time_pos; + temp = ctx->high_temp - temp_pos; + break; + } + + if (temp != ctx->cur_temp || ctx->new_output) { + fprintf(stderr, "state: %d, temp: %d, cur: %d\n", ctx->state, temp, ctx->cur_temp); + ctx->cur_temp = temp; + set_temperature(ctx); + } +} + +static int increments(struct context *ctx) { + int diff = ctx->high_temp - ctx->low_temp; + diff /= 50; + int time = (ctx->duration * 1000) / diff; + return time; +} + +static int time_to_next_event(struct context *ctx) { + switch (ctx->state) { + case ANIMATING_TO_HIGH: + case ANIMATING_TO_LOW: + return increments(ctx); + default: + return 300000; + } +} + +static int display_poll(struct wl_display *display, short int events, int timeout) { + struct pollfd pfd[1]; + pfd[0].fd = wl_display_get_fd(display); + pfd[0].events = events; + + int ret; + do { + ret = poll(pfd, 1, timeout); + } while (ret == -1 && errno == EINTR); + + return ret; +} + +int display_dispatch_with_timeout(struct wl_display *display, int timeout); +int display_dispatch_with_timeout(struct wl_display *display, int timeout) { + if (wl_display_prepare_read(display) == -1) { + return wl_display_dispatch_pending(display); + } + + int ret; + while (true) { + ret = wl_display_flush(display); + if (ret != -1 || errno != EAGAIN) { + break; + } + + if (display_poll(display, POLLOUT, -1) == -1) { + wl_display_cancel_read(display); + return -1; + } + } + + if (ret < 0 && errno != EPIPE) { + wl_display_cancel_read(display); + return -1; + } + + if (display_poll(display, POLLIN, timeout) == -1) { + wl_display_cancel_read(display); + return -1; + } + + if (wl_display_read_events(display) == -1) { + return -1; + } + + return wl_display_dispatch_pending(display); } static const char usage[] = "usage: %s [options]\n" " -h show this help message\n" -" -t set temperature (default: 6500)\n" +" -T set high temperature (default: 6500)\n" +" -t set low temperature (default: 3500)\n" +" -S set ramp up time (default: 6:00)\n" +" -s set ramp down time (default: 18:00)\n" +" -d set ramping duration in minutes (default: 30)\n" " -g set gamma (default: 1)\n"; int main(int argc, char *argv[]) { - wl_list_init(&outputs); - double gamma = 1; - long temperature = 6500; + tzset(); + + // Initialize defaults + struct context ctx = { + .gamma = 1.0, + .start_time = 6 * 60 * 60, + .stop_time = 18 * 60 * 60, + .high_temp = 6500, + .low_temp = 3500, + .duration = 30 * 60, + .state = HIGH_TEMP, + }; + wl_list_init(&ctx.outputs); + int opt; - while ((opt = getopt(argc, argv, "ht:g:")) != -1) { + time_t now = time(NULL); + struct tm tm = { 0 }; + struct tm current = { 0 }; + localtime_r(&now, ¤t); + while ((opt = getopt(argc, argv, "hT:t:S:s:g:d:")) != -1) { switch (opt) { + case 'T': + ctx.high_temp = strtol(optarg, NULL, 10); + break; case 't': - temperature = strtol(optarg, NULL, 10); + ctx.low_temp = strtol(optarg, NULL, 10); + break; + case 'S': + memcpy(&tm, ¤t, sizeof tm); + if (strptime(optarg, "%H:%M", &tm) != NULL) { + ctx.start_time = mktime(&tm) % 86400; + } + break; + case 's': + memcpy(&tm, ¤t, sizeof tm); + if (strptime(optarg, "%H:%M", &tm) != NULL) { + ctx.stop_time = mktime(&tm) % 86400; + } + break; + case 'd': + ctx.duration = strtod(optarg, NULL) * 60; break; case 'g': - gamma = strtod(optarg, NULL); + ctx.gamma = strtod(optarg, NULL); break; case 'h': default: @@ -233,7 +459,7 @@ int main(int argc, char *argv[]) { } struct wl_registry *registry = wl_display_get_registry(display); - wl_registry_add_listener(registry, ®istry_listener, NULL); + wl_registry_add_listener(registry, ®istry_listener, &ctx); wl_display_roundtrip(display); if (gamma_control_manager == NULL) { @@ -243,27 +469,15 @@ int main(int argc, char *argv[]) { } struct output *output; - wl_list_for_each(output, &outputs, link) { - output->gamma_control = zwlr_gamma_control_manager_v1_get_gamma_control( - gamma_control_manager, output->wl_output); - zwlr_gamma_control_v1_add_listener(output->gamma_control, - &gamma_control_listener, output); + wl_list_for_each(output, &ctx.outputs, link) { + setup_output(output); } wl_display_roundtrip(display); - /* Approximate white point */ - double rw, gw, bw; - calc_whitepoint(temperature, &rw, &gw, &bw); - wl_list_for_each(output, &outputs, link) { - fill_gamma_table(output->table, output->ramp_size, - rw, gw, bw, gamma); - zwlr_gamma_control_v1_set_gamma(output->gamma_control, - output->table_fd); - } - - while (wl_display_dispatch(display) != -1) { - // This space is intentionnally left blank + update_temperature(&ctx); + while (display_dispatch_with_timeout(display, time_to_next_event(&ctx)) != -1) { + update_temperature(&ctx); } return EXIT_SUCCESS;