Add ability to select multiple screens

Signed-off-by: Mattias Andrée <m@maandree.se>

1 files changed, 42 insertions, 21 deletions

diff --git a/src/backend-direct.c b/src/backend-direct.c
index de58548..37c82c6 100644
--- a/src/backend-direct.c
+++ b/src/backend-direct.c
@@ -140,7 +140,7 @@ struct gamma_state {
 	/**
 	 * Whether a parition has been selected
 	 */
-	unsigned partition_selected : 1;
+	unsigned partitions_selected : 1;
 
 	/**
 	 * Whether CRTCs have been selected
@@ -158,11 +158,6 @@ struct gamma_state {
 	int method;
 
 	/**
-	 * Selected parition (if `partition_selected`)
-	 */
-	size_t selected_partition;
-
-	/**
 	 * Number of selected CRTCs
 	 */
 	size_t ncrtcs;
@@ -178,6 +173,11 @@ struct gamma_state {
 	size_t npartitions;
 
 	/**
+	 * Selected paritions (if `partitions_selected`)
+	 */
+	size_t *selected_partitions;
+
+	/**
 	 * Selected CRTC numbers
 	 * 
 	 * Deallocated by when no longer needed
@@ -274,7 +274,7 @@ direct_print_help(FILE *f, int method)
 		fprintf(f, "  card=%s     %s\n", _("N     "), _("Graphics card to apply adjustments to"));
 	} else if (caps.multiple_partitions) {
 		/* TRANSLATORS: "N" represents "ordinal"; right-pad with spaces to preserve display width */
-		fprintf(f, "  screen=%s   %s\n", _("N     "), _("X screen to apply adjustments to"));
+		fprintf(f, "  screen=%s   %s\n", _("N     "), _("List of comma-separated X screens to apply adjustments to"));
 	}
 
 	if (caps.multiple_crtcs) {
@@ -327,21 +327,38 @@ direct_set_site(struct gamma_state *state, const char *key, const char *value)
 int
 direct_set_partitions(struct gamma_state *state, const char *key, const char *value)
 {
-	const char *end;
+	const char *end, *p;
 	uintmax_t num;
+	size_t count;
 
 	/* Check previously unspecified */
-	if (state->partition_selected)
+	if (state->partitions_selected)
 		weprintf(_("`%s' option specified multiple times, using last selection."), key);
-	state->partition_selected = 1;
+	state->partitions_selected = 1;
 
-	/* Parse number */
+	/* Check if all are selected */
+	state->npartitions = 0;
+	free(state->selected_partitions);
+	state->selected_partitions = NULL;
+	if (!*value || !strcasecmp(value, "all"))
+		return 0;
+
+	/* Get number count */
+	for (p = value, count = 1; *p; p++)
+		if (*p == ',')
+			count++;
+	state->selected_partitions = ecalloc(count, sizeof(*state->selected_partitions));
+
+	/* Parse numbers */
 	errno = 0;
-	num = strtoumax(value, (void *)&end, 10);
-	state->selected_partition = (size_t)num;
-	if (num > (uintmax_t)SIZE_MAX || *end || !isdigit(*value) || errno) {
-		weprintf(_("Invalid value of `%s' option: `%s'."), key, value);
-		return -1;
+	for (p = value; *p; p = end) {
+		num = strtoumax(p, (void *)&end, 10);
+		state->selected_partitions[state->npartitions++] = (size_t)num;
+		if (num > (uintmax_t)SIZE_MAX || (*end && *end != ',') || !isdigit(*p) || errno) {
+			weprintf(_("Invalid value of `%s' option: `%s'."), key, value);
+			return -1;
+		}
+		end = &end[*end == ','];
 	}
 
 	return 0;
@@ -478,11 +495,14 @@ direct_start(struct gamma_state *state)
 	}
 
 	/* Allocate partition states */
-	if (state->partition_selected) {
-		state->partitions = ecalloc(1, sizeof(*state->partitions));
-		state->partitions[0].state.partition = state->selected_partition;
-		state->partitions[0].state.data = NULL;
-		state->npartitions = 1;
+	if (state->npartitions) {
+		state->partitions = ecalloc(state->npartitions, sizeof(*state->partitions));
+		for (i = 0; i < state->npartitions; i++) {
+			state->partitions[i].state.partition = state->selected_partitions[i];
+			state->partitions[i].state.data = NULL;
+		}
+		free(state->selected_partitions);
+		state->selected_partitions = NULL;
 	} else if (!state->site.partitions_available) {
 		if (state->partitions_are_graphics_cards)
 			weprintf(_("No graphics card found."));
@@ -908,6 +928,7 @@ direct_free(struct gamma_state *state)
 				libgamma_partition_destroy(&state->partitions[i].state);
 		free(state->partitions);
 	}
+	free(state->selected_partitions);
 	free(state->selected_crtcs);
 	if (state->selected_edids) {
 		for (i = 0; i < state->nedids; i++)