aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorMattias Andrée <maandree@kth.se>2016-08-06 02:14:33 +0200
committerMattias Andrée <maandree@kth.se>2016-08-06 02:14:33 +0200
commitb853d3d10ac3de5de75e384307d67177bbb1091f (patch)
treebac7940b0aba51b07d3bcc98285c7c4b70259b73 /src
parentm (diff)
downloadcg-tools-b853d3d10ac3de5de75e384307d67177bbb1091f.tar.gz
cg-tools-b853d3d10ac3de5de75e384307d67177bbb1091f.tar.bz2
cg-tools-b853d3d10ac3de5de75e384307d67177bbb1091f.tar.xz
Add cg-limits
Signed-off-by: Mattias Andrée <maandree@kth.se>
Diffstat (limited to 'src')
-rw-r--r--src/cg-limits.c739
1 files changed, 739 insertions, 0 deletions
diff --git a/src/cg-limits.c b/src/cg-limits.c
new file mode 100644
index 0000000..4b28268
--- /dev/null
+++ b/src/cg-limits.c
@@ -0,0 +1,739 @@
+/**
+ * cg-tools -- Cooperative gamma-enabled tools
+ * Copyright (C) 2016 Mattias Andrée (maandree@kth.se)
+ *
+ * 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 <http://www.gnu.org/licenses/>.
+ */
+#define _DEFAULT_SOURCE
+#define _BSD_SOURCE
+
+#include "cg-base.h"
+
+#include <libclut.h>
+
+#include <sys/stat.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <pwd.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+
+
+/**
+ * The default filter priority for the program
+ */
+const int64_t default_priority = -(((int64_t)1) << 62);
+
+/**
+ * The default class for the program
+ */
+char* const default_class = PKGNAME "::cg-limits::standard";
+
+
+
+/**
+ * -d: keep process alive and remove filter on death
+ */
+static int dflag = 0;
+
+/**
+ * -x: remove filter rather than adding it
+ */
+static int xflag = 0;
+
+/**
+ * -B: brightness listing file
+ */
+static char* Bflag = NULL;
+
+/**
+ * -C: constrat listing file
+ */
+static char* Cflag = NULL;
+
+/**
+ * The brightness of the red channel
+ */
+static double rbrightness = 0;
+
+/**
+ * The brightness of the green channel
+ */
+static double gbrightness = 0;
+
+/**
+ * The brightness of the blue channel
+ */
+static double bbrightness = 0;
+
+/**
+ * The contrast of the red channel
+ */
+static double rcontrast = 1;
+
+/**
+ * The contrast of the green channel
+ */
+static double gcontrast = 1;
+
+/**
+ * The contrast of the blue channel
+ */
+static double bcontrast = 1;
+
+/**
+ * `NULL`-terminated list of output names
+ * listed in the brightness configuration file
+ */
+static char** brightness_names = NULL;
+
+/**
+ * The brightness of the red channel on monitor
+ * with same index in `brightness_names`
+ */
+static double* rbrightnesses = NULL;
+
+/**
+ * The brightness of the green channel on monitor
+ * with same index in `brightness_names`
+ */
+static double* gbrightnesses = NULL;
+
+/**
+ * The brightness of the blue channel on monitor
+ * with same index in `brightness_names`
+ */
+static double* bbrightnesses = NULL;
+
+/**
+ * `NULL`-terminated list of output names
+ * listed in the contrast configuration file
+ */
+static char** contrast_names = NULL;
+
+/**
+ * The contrast of the red channel on monitor
+ * with same index in `contrast_names`
+ */
+static double* rcontrasts = NULL;
+
+/**
+ * The contrast of the green channel on monitor
+ * with same index in `contrast_names`
+ */
+static double* gcontrasts = NULL;
+
+/**
+ * The contrast of the blue channel on monitor
+ * with same index in `contrast_names`
+ */
+static double* bcontrasts = NULL;
+
+
+
+/**
+ * Print usage information and exit
+ */
+void usage(void)
+{
+ fprintf(stderr,
+ "Usage: %s [-M method] [-S site] [-c crtc]... [-R rule] (-d | [-p priority] [-x] "
+ "([-B brigtness-file] [-C contrast-file] | brightness-all:contrast-all | "
+ "brightness-red:contrast-red brightness-green:contrast-green brightness-blue:contrast-blue))\n",
+ argv0);
+ exit(1);
+}
+
+
+/**
+ * Perform cleanup so valgrind output is clean
+ *
+ * @param ret The value to return
+ * @return `ret` is returned as is
+ */
+static int cleanup(int ret)
+{
+ int saved_errno = errno;
+ if (brightness_names != NULL)
+ {
+ char** p = brightness_names;
+ while (*p)
+ free(*p++);
+ }
+ free(brightness_names);
+ free(rbrightnesses);
+ free(gbrightnesses);
+ free(bbrightnesses);
+ if (contrast_names != NULL)
+ {
+ char** p = contrast_names;
+ while (*p)
+ free(*p++);
+ }
+ free(contrast_names);
+ free(rcontrasts);
+ free(gcontrasts);
+ free(bcontrasts);
+ errno = saved_errno;
+ return ret;
+}
+
+
+/**
+ * Handle a command line option
+ *
+ * @param opt The option, it is a NUL-terminate two-character
+ * string starting with either '-' or '+', if the
+ * argument is not recognised, call `usage`. This
+ * string will not be "-M", "-S", "-c", "-p", or "-R".
+ * @param arg The argument associated with `opt`,
+ * `NULL` there is no next argument, if this
+ * parameter is `NULL` but needed, call `usage`
+ * @return 0 if `arg` was not used,
+ * 1 if `arg` was used,
+ * -1 on error
+ */
+int handle_opt(char* opt, char* arg)
+{
+ if (opt[0] == '-')
+ switch (opt[1])
+ {
+ case 'd':
+ if (dflag || xflag)
+ usage();
+ dflag = 1;
+ break;
+ case 'x':
+ if (xflag || dflag)
+ usage();
+ xflag = 1;
+ break;
+ case 'B':
+ if (Bflag || !(Bflag = arg))
+ usage();
+ return 1;
+ case 'C':
+ if (Cflag || !(Cflag = arg))
+ usage();
+ return 1;
+ default:
+ usage();
+ }
+ return 0;
+}
+
+
+/**
+ * Get the pathname of a configuration file
+ *
+ * @param confname The filename (excluding directory) of the configuration file
+ * @return The full pathname of the configuration file, `NULL` on error
+ */
+static char* get_conf_file(const char* restrict confname)
+{
+ struct passwd* pw;
+ char* path;
+
+ pw = getpwuid(getuid());
+ if ((pw == NULL) || (pw->pw_dir == NULL))
+ return NULL;
+
+ path = malloc(strlen(pw->pw_dir) + strlen(confname) + sizeof("/.config/"));
+ if (path == NULL)
+ return NULL;
+
+ sprintf(path, "%s/.config/%s", pw->pw_dir, confname);
+
+ if (access(path, F_OK) < 0)
+ sprintf(path, "/etc/%s", confname);
+
+ return path;
+}
+
+
+/**
+ * Parse a double encoded as a string
+ *
+ * @param out Output parameter for the value
+ * @param str The string
+ * @retunr Zero on success, -1 if the string is invalid
+ */
+static int parse_double(double* restrict out, const char* restrict str)
+{
+ char* end;
+ errno = 0;
+ *out = strtod(str, &end);
+ if (errno || isinf(*out) || isnan(*out) || *end)
+ return -1;
+ if (!strchr("-0123456789.", *str))
+ return -1;
+ return 0;
+}
+
+
+/**
+ * Parse two doubles encoded as a "%lf:%lf" string
+ *
+ * @param left Output parameter for the first value
+ * @param right Output parameter for the second value
+ * @param str The string
+ * @retunr Zero on success, -1 if the string is invalid
+ */
+static int parse_twidouble(double* restrict left, double* restrict right, const char* restrict str)
+{
+ char* p = strchr(str, ':');
+ int r;
+ if (p == NULL)
+ return -1;
+ *p = '\0';
+ r = -((parse_double(left, str) < 0) || (parse_double(right, p + 1) < 0));
+ *p = ':';
+ return r;
+}
+
+
+/**
+ * Parse configuration file
+ *
+ * @param pathname The pathname of the file
+ * @param names Reference to the list of names
+ * @param rs Reference to the list of red values
+ * @param gs Reference to the list of green values
+ * @param bs Reference to the list of blue values
+ * @return Zero on success, -1 on error
+ */
+static int parse_conf_file(const char* restrict pathname, char*** restrict names,
+ double** restrict rs, double** restrict gs, double** restrict bs)
+{
+ int fd, saved_errno;
+ char* line = NULL;
+ size_t size = 0, lineno = 0, ptr = 0, alloc = 0;
+ ssize_t n;
+ FILE* f = NULL;
+ char* p;
+ char* q;
+ char* r;
+ char* g;
+ char* b;
+
+ fd = open(pathname, O_RDONLY);
+ if (fd == -1)
+ return -1;
+
+ f = fdopen(fd, "rb");
+ if (f == NULL)
+ goto fail;
+
+ while (n = getline(&line, &size, f), n >= 0)
+ {
+ lineno += 1;
+
+ if ((n > 0) && (line[n - 1] == '\n'))
+ line[n - 1] = '\0';
+ p = line;
+ while ((*p == ' ') || (*p == '\t')) p++;
+ if ((!*p) || (*p == '#'))
+ continue;
+
+ r = strpbrk(line, " \t");
+ if (r == NULL)
+ goto bad;
+ while (r[1] == ' ' || r[1] == '\t') r++;
+ g = strpbrk(r + 1, " \t");
+ if (g == NULL)
+ goto bad;
+ while (g[1] == ' ' || g[1] == '\t') g++;
+ b = strpbrk(g + 1, " \t");
+ if (b == NULL)
+ goto bad;
+ while (b[1] == ' ' || b[1] == '\t') b++;
+
+ for (;;)
+ {
+ q = strpbrk(b + 1, " \t");
+ if (q == NULL)
+ break;
+ while (q[1] == ' ' || q[1] == '\t') q++;
+ if (!*q)
+ break;
+ r = g, g = b, b = q;
+ }
+
+ *r++ = '\0';
+ *g++ = '\0';
+ *b++ = '\0';
+
+ q = strpbrk(r, " \t");
+ if (q != NULL)
+ *q = '\0';
+ q = strpbrk(g, " \t");
+ if (q != NULL)
+ *q = '\0';
+ q = strpbrk(b, " \t");
+ if (q != NULL)
+ *q = '\0';
+
+ q = strchr(p, '\0');
+ while ((q != p) && ((q[-1] == ' ') || (q[-1] == '\t')))
+ q--;
+ *q = '\0';
+
+ if (ptr == alloc)
+ {
+ void* new;
+ size_t new_size = alloc ? (alloc << 1) : 4;
+
+ new = realloc(*rs, new_size * sizeof(**rs));
+ if (new == NULL)
+ goto fail;
+ *rs = new;
+
+ new = realloc(*gs, new_size * sizeof(**gs));
+ if (new == NULL)
+ goto fail;
+ *gs = new;
+
+ new = realloc(*bs, new_size * sizeof(**bs));
+ if (new == NULL)
+ goto fail;
+ *bs = new;
+
+ new = realloc(*names, (new_size + 1) * sizeof(**names));
+ if (new == NULL)
+ goto fail;
+ *names = new;
+ memset(*names + alloc, 0, (new_size + 1 - alloc) * sizeof(**names));
+
+ alloc = new_size;
+ }
+
+ if ((parse_double((*rs) + ptr, r) < 0) ||
+ (parse_double((*gs) + ptr, g) < 0) ||
+ (parse_double((*bs) + ptr, b) < 0))
+ goto bad;
+ (*names)[ptr] = malloc(strlen(p) + 1);
+ if ((*names)[ptr] == NULL)
+ goto fail;
+ strcpy((*names)[ptr], p);
+ ptr++;
+
+ continue;
+ bad:
+ fprintf(stderr, "%s: ignoring malformatted line in %s: %zu\n", argv0, pathname, lineno);
+ }
+
+ if (fclose(f) < 0)
+ {
+ f = NULL;
+ goto fail;
+ }
+ close(fd);
+ free(line);
+ return 0;
+ fail:
+ saved_errno = errno;
+ free(line);
+ if (f != NULL)
+ fclose(f);
+ if (fd >= 0)
+ close(fd);
+ errno = saved_errno;
+ return -1;
+}
+
+
+/**
+ * Parse brightness configuration file
+ *
+ * @param pathname The pathname of the file
+ * @return Zero on success, -1 on error
+ */
+static int parse_brightness_file(const char* restrict pathname)
+{
+ return parse_conf_file(pathname, &brightness_names, &rbrightnesses, &gbrightnesses, &bbrightnesses);
+}
+
+
+/**
+ * Parse contrast configuration file
+ *
+ * @param pathname The pathname of the file
+ * @return Zero on success, -1 on error
+ */
+static int parse_contrast_file(const char* restrict pathname)
+{
+ return parse_conf_file(pathname, &contrast_names, &rcontrasts, &gcontrasts, &bcontrasts);
+}
+
+
+/**
+ * This function is called after the last
+ * call to `handle_opt`
+ *
+ * @param argc The number of unparsed arguments
+ * @param argv `NULL` terminated list of unparsed arguments
+ * @param method The argument associated with the "-M" option
+ * @param site The argument associated with the "-S" option
+ * @param crtcs The arguments associated with the "-c" options, `NULL`-terminated
+ * @param prio The argument associated with the "-p" option
+ * @param rule The argument associated with the "-R" option
+ * @return Zero on success, -1 on error
+ */
+int handle_args(int argc, char* argv[], char* method, char* site,
+ char** crtcs, char* prio, char* rule)
+{
+ int free_Bflag = 0, free_Cflag = 0, saved_errno;
+ int q = xflag + dflag;
+ q += (method != NULL) && !strcmp(method, "?");
+ q += (prio != NULL) && !strcmp(prio, "?");
+ q += (rule != NULL) && (!strcmp(rule, "?") || !strcmp(rule, "??"));
+ for (; *crtcs; crtcs++)
+ q += !strcmp(*crtcs, "?");
+ if ((q > 1) || (xflag && ((Bflag != NULL) || (Cflag != NULL) || (argc > 0) || (prio != NULL))))
+ usage();
+ if ((Bflag || Cflag) && argc)
+ usage();
+
+ if (argc == 1)
+ {
+ if (parse_twidouble(&rbrightness, &rcontrast, argv[0]) < 0)
+ usage();
+ bbrightness = gbrightness = rbrightness;
+ bcontrast = gcontrast = rcontrast;
+ }
+ else if (argc == 3)
+ {
+ if (parse_twidouble(&rbrightness, &rcontrast, argv[0]) < 0)
+ usage();
+ if (parse_twidouble(&gbrightness, &gcontrast, argv[1]) < 0)
+ usage();
+ if (parse_twidouble(&bbrightness, &bcontrast, argv[2]) < 0)
+ usage();
+ }
+ else if (argc)
+ usage();
+
+ if (!argc && !Bflag && !xflag)
+ {
+ Bflag = get_conf_file("brightness");
+ if (Bflag == NULL)
+ return -1;
+ free_Bflag = 1;
+ }
+ if (Bflag)
+ if (parse_brightness_file(Bflag) < 0)
+ goto fail;
+ if (free_Bflag)
+ free(Bflag), Bflag = NULL;
+
+ if (!argc && !Cflag && !xflag)
+ {
+ Cflag = get_conf_file("contrast");
+ if (Cflag == NULL)
+ return -1;
+ free_Cflag = 1;
+ }
+ if (Cflag)
+ if (parse_contrast_file(Cflag) < 0)
+ goto fail;
+ if (free_Cflag)
+ free(Cflag), Cflag = NULL;
+
+ return 0;
+ fail:
+ saved_errno = errno;
+ if (free_Bflag)
+ free(Bflag), Bflag = NULL;
+ if (free_Cflag)
+ free(Cflag), Cflag = NULL;
+ errno = saved_errno;
+ return cleanup(-1);
+}
+
+
+/**
+ * Fill a filter
+ *
+ * @param filter The filter to fill
+ * @param rb The red brightness
+ * @param rc The red contrast
+ * @param gb The green brightness
+ * @param gc The green contrast
+ * @param bb The blue brightness
+ * @param bc The blue contrast
+ * @return Zero on success, -1 on error
+ */
+static int fill_filter(libcoopgamma_filter_t* restrict filter,
+ double rb, double rc, double gb, double gc, double bb, double bc)
+{
+ union libcoopgamma_ramps dramps;
+ size_t size;
+
+ if (filter->depth == LIBCOOPGAMMA_DOUBLE)
+ {
+ libclut_rgb_limits(&(filter->ramps.d), (double)1, double, rb, rc, gb, gc, bb, bc);
+ libclut_clip(&(filter->ramps.d), (double)1, double, 1, 1, 1);
+ return 0;
+ }
+ if (filter->depth == LIBCOOPGAMMA_FLOAT)
+ {
+ libclut_rgb_limits(&(filter->ramps.f), (float)1, float, rb, rc, gb, gc, bb, bc);
+ libclut_clip(&(filter->ramps.f), (float)1, float, 1, 1, 1);
+ return 0;
+ }
+
+ size = dramps.d.red_size = filter->ramps.d.red_size;
+ size += dramps.d.green_size = filter->ramps.d.green_size;
+ size += dramps.d.blue_size = filter->ramps.d.blue_size;
+ dramps.d.red = calloc(size, sizeof(double));
+ if (dramps.d.red == NULL)
+ return -1;
+ dramps.d.green = dramps.d.red + dramps.d.red_size;
+ dramps.d.blue = dramps.d.green + dramps.d.green_size;
+
+ libclut_start_over(&(dramps.d), (double)1, double, 1, 1, 1);
+ libclut_rgb_limits(&(dramps.d), (double)1, double, rb, rc, gb, gc, bb, bc);
+ libclut_clip(&(dramps.d), (double)1, double, 1, 1, 1);
+
+ switch (filter->depth)
+ {
+ case LIBCOOPGAMMA_UINT8:
+ libclut_translate(&(filter->ramps.u8), UINT8_MAX, uint8_t, &(dramps.d), (double)1, double);
+ break;
+ case LIBCOOPGAMMA_UINT16:
+ libclut_translate(&(filter->ramps.u16), UINT16_MAX, uint16_t, &(dramps.d), (double)1, double);
+ break;
+ case LIBCOOPGAMMA_UINT32:
+ libclut_translate(&(filter->ramps.u32), UINT32_MAX, uint32_t, &(dramps.d), (double)1, double);
+ break;
+ case LIBCOOPGAMMA_UINT64:
+ libclut_translate(&(filter->ramps.u64), UINT64_MAX, uint64_t, &(dramps.d), (double)1, double);
+ break;
+ default:
+ abort();
+ }
+
+ free(dramps.d.red);
+ return 0;
+}
+
+
+/**
+ * The main function for the program-specific code
+ *
+ * @return 0: Success
+ * -1: Error, `errno` set
+ * -2: Error, `cg.error` set
+ * -3: Error, message already printed
+ */
+int start(void)
+{
+ int r;
+ size_t i, j, k;
+
+ if (xflag)
+ for (i = 0; i < crtcs_n; i++)
+ crtc_updates[i].filter.lifespan = LIBCOOPGAMMA_REMOVE;
+ else if (dflag)
+ for (i = 0; i < crtcs_n; i++)
+ crtc_updates[i].filter.lifespan = LIBCOOPGAMMA_UNTIL_DEATH;
+ else
+ for (i = 0; i < crtcs_n; i++)
+ crtc_updates[i].filter.lifespan = LIBCOOPGAMMA_UNTIL_REMOVAL;
+
+ if (!xflag && ((brightness_names != NULL) || (contrast_names != NULL)))
+ if ((r = make_slaves()) < 0)
+ return cleanup(r);
+
+ if ((brightness_names == NULL) && (contrast_names == NULL))
+ for (i = 0, r = 1; i < crtcs_n; i++)
+ {
+ if (!(crtc_updates[i].master) || !(crtc_info[i].supported))
+ continue;
+ if (!xflag)
+ if ((r = fill_filter(&(crtc_updates[i].filter), rbrightness, rcontrast,
+ gbrightness, gcontrast, bbrightness, bcontrast)) < 0)
+ return cleanup(r);
+ r = update_filter(i, 0);
+ if ((r == -2) || ((r == -1) && (errno != EAGAIN)))
+ return cleanup(r);
+ if (crtc_updates[i].slaves != NULL)
+ for (j = 0; crtc_updates[i].slaves[j] != 0; j++)
+ {
+ r = update_filter(crtc_updates[i].slaves[j], 0);
+ if ((r == -2) || ((r == -1) && (errno != EAGAIN)))
+ return cleanup(r);
+ }
+ }
+ else
+ {
+ char* empty = NULL;
+ char** bnames = brightness_names ? brightness_names : &empty;
+ char** cnames = contrast_names ? contrast_names : &empty;
+ for (i = 0, r = 1; i < crtcs_n; i++)
+ {
+ if (!(crtc_info[i].supported))
+ continue;
+ for (j = 0; bnames[j] != NULL; j++)
+ if (!strcasecmp(crtc_updates[i].filter.crtc, bnames[j]))
+ break;
+ for (k = 0; cnames[k] != NULL; k++)
+ if (!strcasecmp(crtc_updates[k].filter.crtc, cnames[k]))
+ break;
+ if ((bnames[j] != NULL) || (cnames[k] != NULL))
+ {
+ double rb = 0, gb = 0, bb = 0, rc = 1, bc = 1, gc = 1;
+ if (bnames[j] != NULL)
+ rb = rbrightnesses[j], gb = gbrightnesses[j], bb = bbrightnesses[j];
+ if (cnames[j] != NULL)
+ rc = rcontrasts[j], gc = gcontrasts[j], bc = bcontrasts[j];
+ if ((r = fill_filter(&(crtc_updates[i].filter), rb, rc, gb, gc, bb, bc)) < 0)
+ return cleanup(r);
+ r = update_filter(i, 0);
+ if ((r == -2) || ((r == -1) && (errno != EAGAIN)))
+ return cleanup(r);
+ }
+ }
+ }
+
+ while (r != 1)
+ if ((r = synchronise(-1)) < 0)
+ return cleanup(r);
+
+ if (!dflag)
+ return cleanup(0);
+
+ if (libcoopgamma_set_nonblocking(&cg, 0) < 0)
+ return cleanup(-1);
+ for (;;)
+ if (libcoopgamma_synchronise(&cg, NULL, 0, &j) < 0)
+ switch (errno)
+ {
+ case 0:
+ break;
+ case ENOTRECOVERABLE:
+ goto enotrecoverable;
+ default:
+ return cleanup(-1);
+ }
+
+ enotrecoverable:
+ for (;;)
+ if (pause() < 0)
+ return cleanup(-1);
+}
+