/* See LICENSE file for copyright and license details. */
#include "cg-base.h"
#include <libclut.h>
#include <errno.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 << 61;
/**
* The default class for the program
*/
char default_class[] = PKGNAME "::cg-brilliance::standard";
/**
* Class suffixes
*/
const char *const *class_suffixes = (const char *const[]){NULL};
/**
* -d: keep process alive and remove filter on death
*/
static int dflag = 0;
/**
* -x: remove filter rather than adding it
*/
static int xflag = 0;
/**
* The brilliance of the red channel
*/
static double rvalue = 0;
/**
* The brilliance of the green channel
*/
static double gvalue = 0;
/**
* The brilliance of the blue channel
*/
static double bvalue = 0;
/**
* Print usage information and exit
*/
void
usage(void)
{
fprintf(stderr,
"usage: %s [-M method] [-S site] [-c crtc]... [-R rule] "
"(-x | [-p priority] [-d] (all | red green blue))\n",
argv0);
exit(1);
}
/**
* 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;
default:
usage();
}
} else {
usage();
}
return 0;
(void) arg;
}
/**
* Parse a non-negative double encoded as a string
*
* @param out Output parameter for the value
* @param str The string
* @return 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 || *out < 0 || isinf(*out) || isnan(*out) || *end)
return -1;
if (!*str || !strchr("0123456789.", *str))
return -1;
return 0;
}
/**
* 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 prio The argument associated with the "-p" option
* @return Zero on success, -1 on error
*/
int
handle_args(int argc, char *argv[], char *prio)
{
char *red = NULL;
char *green = NULL;
char *blue = NULL;
int q = xflag + dflag;
if (q > 1 || (xflag && (prio || argc)))
usage();
if (argc == 1) {
red = green = blue = argv[0];
} else if (argc == 3) {
red = argv[0];
green = argv[1];
blue = argv[2];
} else if (argc || !xflag) {
usage();
}
if (argc) {
if (parse_double(&rvalue, red) < 0)
usage();
if (parse_double(&gvalue, blue) < 0)
usage();
if (parse_double(&bvalue, green) < 0)
usage();
}
return 0;
}
/**
* Fill a filter
*
* @param filter The filter to fill
*/
static void
fill_filter(libcoopgamma_filter_t *restrict filter)
{
size_t i;
switch (filter->depth) {
#define X(CONST, MAX, TYPE, MEMBER)\
case CONST:\
for (i = 0; i < filter->ramps.MEMBER.red_size; i++) {\
double val = (double)(filter->ramps.MEMBER.red[i]);\
val *= rvalue;\
filter->ramps.MEMBER.red[i] = val < 0 ? 0 : val > (double)(MAX) ? MAX : (TYPE)val;\
}\
for (i = 0; i < filter->ramps.MEMBER.green_size; i++) {\
double val = (double)(filter->ramps.MEMBER.green[i]);\
val *= gvalue;\
filter->ramps.MEMBER.green[i] = val < 0 ? 0 : val > (double)(MAX) ? MAX : (TYPE)val;\
}\
for (i = 0; i < filter->ramps.MEMBER.blue_size; i++) {\
double val = (double)(filter->ramps.MEMBER.blue[i]);\
val *= bvalue;\
filter->ramps.MEMBER.blue[i] = val < 0 ? 0 : val > (double)(MAX) ? MAX : (TYPE)val;\
}\
break
X(LIBCOOPGAMMA_UINT8, UINT8_MAX, uint8_t, u8);
X(LIBCOOPGAMMA_UINT16, UINT16_MAX, uint16_t, u16);
X(LIBCOOPGAMMA_UINT32, UINT32_MAX, uint32_t, u32);
X(LIBCOOPGAMMA_UINT64, UINT64_MAX, uint64_t, u64);
#undef X
case LIBCOOPGAMMA_FLOAT:
libclut_rgb_brightness(&filter->ramps.f, (float)1, float, rvalue, gvalue, bvalue);
libclut_clip(&filter->ramps.f, (float)1, float, 1, 1, 1);
break;
case LIBCOOPGAMMA_DOUBLE:
libclut_rgb_brightness(&filter->ramps.d, (double)1, double, rvalue, gvalue, bvalue);
libclut_clip(&filter->ramps.d, (double)1, double, 1, 1, 1);
break;
default:
abort();
}
}
/**
* 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;
if (xflag)
for (i = 0; i < filters_n; i++)
crtc_updates[i].filter.lifespan = LIBCOOPGAMMA_REMOVE;
else if (dflag)
for (i = 0; i < filters_n; i++)
crtc_updates[i].filter.lifespan = LIBCOOPGAMMA_UNTIL_DEATH;
else
for (i = 0; i < filters_n; i++)
crtc_updates[i].filter.lifespan = LIBCOOPGAMMA_UNTIL_REMOVAL;
if (!xflag && (r = make_slaves()) < 0)
return r;
for (i = 0, r = 1; i < filters_n; i++) {
if (!crtc_updates[i].master || !crtc_info[crtc_updates[i].crtc].supported)
continue;
if (!xflag)
fill_filter(&crtc_updates[i].filter);
r = update_filter(i, 0);
if (r == -2 || (r == -1 && errno != EAGAIN))
return r;
if (crtc_updates[i].slaves) {
for (j = 0; crtc_updates[i].slaves[j]; j++) {
r = update_filter(crtc_updates[i].slaves[j], 0);
if (r == -2 || (r == -1 && errno != EAGAIN))
return r;
}
}
}
while (r != 1)
if ((r = synchronise(-1)) < 0)
return r;
if (!dflag)
return 0;
if (libcoopgamma_set_nonblocking(&cg, 0) < 0)
return -1;
for (;;) {
if (libcoopgamma_synchronise(&cg, NULL, 0, &j) < 0) {
switch (errno) {
case 0:
break;
case ENOTRECOVERABLE:
goto enotrecoverable;
default:
return -1;
}
}
}
enotrecoverable:
pause();
return -1;
}