mirror of
https://github.com/swaywm/swaybg
synced 2024-09-28 03:20:26 +02:00
469d9d4d77
Calling get_buffer_size on a newly created swaybg_output, before a config is assigned to a swaybg_output, is unnecessary and yields a null pointer dereference.
715 lines
21 KiB
C
715 lines
21 KiB
C
#include <assert.h>
|
|
#include <ctype.h>
|
|
#include <getopt.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
#include <wayland-client.h>
|
|
#include "background-image.h"
|
|
#include "cairo_util.h"
|
|
#include "log.h"
|
|
#include "pool-buffer.h"
|
|
#include "wlr-layer-shell-unstable-v1-client-protocol.h"
|
|
#include "viewporter-client-protocol.h"
|
|
#include "single-pixel-buffer-v1-client-protocol.h"
|
|
#include "fractional-scale-v1-client-protocol.h"
|
|
|
|
/*
|
|
* If `color` is a hexadecimal string of the form 'rrggbb' or '#rrggbb',
|
|
* `*result` will be set to the uint32_t version of the color. Otherwise,
|
|
* return false and leave `*result` unmodified.
|
|
*/
|
|
static bool parse_color(const char *color, uint32_t *result) {
|
|
if (color[0] == '#') {
|
|
++color;
|
|
}
|
|
|
|
int len = strlen(color);
|
|
if (len != 6) {
|
|
return false;
|
|
}
|
|
for (int i = 0; i < len; ++i) {
|
|
if (!isxdigit(color[i])) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
uint32_t val = (uint32_t)strtoul(color, NULL, 16);
|
|
*result = (val << 8) | 0xFF;
|
|
return true;
|
|
}
|
|
|
|
struct swaybg_state {
|
|
struct wl_display *display;
|
|
struct wl_compositor *compositor;
|
|
struct wl_shm *shm;
|
|
struct zwlr_layer_shell_v1 *layer_shell;
|
|
struct wp_viewporter *viewporter;
|
|
struct wp_single_pixel_buffer_manager_v1 *single_pixel_buffer_manager;
|
|
struct wp_fractional_scale_manager_v1 *fract_scale_manager;
|
|
struct wl_list configs; // struct swaybg_output_config::link
|
|
struct wl_list outputs; // struct swaybg_output::link
|
|
struct wl_list images; // struct swaybg_image::link
|
|
bool run_display;
|
|
};
|
|
|
|
struct swaybg_image {
|
|
struct wl_list link;
|
|
const char *path;
|
|
bool load_required;
|
|
};
|
|
|
|
struct swaybg_output_config {
|
|
char *output;
|
|
const char *image_path;
|
|
struct swaybg_image *image;
|
|
enum background_mode mode;
|
|
uint32_t color;
|
|
struct wl_list link;
|
|
};
|
|
|
|
struct swaybg_output {
|
|
uint32_t wl_name;
|
|
struct wl_output *wl_output;
|
|
char *name;
|
|
char *identifier;
|
|
|
|
struct swaybg_state *state;
|
|
struct swaybg_output_config *config;
|
|
|
|
struct wl_surface *surface;
|
|
struct zwlr_layer_surface_v1 *layer_surface;
|
|
struct wp_viewport *viewport;
|
|
struct wp_fractional_scale_v1 *fract_scale;
|
|
|
|
uint32_t width, height;
|
|
int32_t scale;
|
|
uint32_t pref_fract_scale;
|
|
|
|
uint32_t configure_serial;
|
|
bool dirty, needs_ack;
|
|
// dimensions of the wl_buffer attached to the wl_surface
|
|
uint32_t buffer_width, buffer_height;
|
|
|
|
struct wl_list link;
|
|
};
|
|
|
|
// Create a wl_buffer with the specified dimensions and content
|
|
static struct wl_buffer *draw_buffer(const struct swaybg_output *output,
|
|
cairo_surface_t *surface, uint32_t buffer_width, uint32_t buffer_height) {
|
|
if (buffer_width == 1 && buffer_height == 1 &&
|
|
output->config->mode == BACKGROUND_MODE_SOLID_COLOR &&
|
|
output->state->single_pixel_buffer_manager) {
|
|
// create and return single pixel buffer
|
|
uint8_t r8 = (output->config->color >> 24) & 0xFF;
|
|
uint8_t g8 = (output->config->color >> 16) & 0xFF;
|
|
uint8_t b8 = (output->config->color >> 8) & 0xFF;
|
|
uint8_t a8 = (output->config->color >> 0) & 0xFF;
|
|
uint32_t f = 0xFFFFFFFF / 0xFF; // division result is an integer
|
|
uint32_t r32 = r8 * f;
|
|
uint32_t g32 = g8 * f;
|
|
uint32_t b32 = b8 * f;
|
|
uint32_t a32 = a8 * f;
|
|
return wp_single_pixel_buffer_manager_v1_create_u32_rgba_buffer(
|
|
output->state->single_pixel_buffer_manager, r32, g32, b32, a32);
|
|
}
|
|
|
|
|
|
struct pool_buffer buffer;
|
|
if (!create_buffer(&buffer, output->state->shm,
|
|
buffer_width, buffer_height, WL_SHM_FORMAT_ARGB8888)) {
|
|
return NULL;
|
|
}
|
|
|
|
cairo_t *cairo = buffer.cairo;
|
|
cairo_save(cairo);
|
|
cairo_set_operator(cairo, CAIRO_OPERATOR_CLEAR);
|
|
cairo_paint(cairo);
|
|
cairo_restore(cairo);
|
|
if (output->config->mode == BACKGROUND_MODE_SOLID_COLOR) {
|
|
cairo_set_source_u32(cairo, output->config->color);
|
|
cairo_paint(cairo);
|
|
} else {
|
|
if (output->config->color) {
|
|
cairo_set_source_u32(cairo, output->config->color);
|
|
cairo_paint(cairo);
|
|
}
|
|
|
|
if (surface) {
|
|
render_background_image(cairo, surface,
|
|
output->config->mode, buffer_width, buffer_height);
|
|
}
|
|
}
|
|
|
|
// return wl_buffer for caller to use and destroy
|
|
struct wl_buffer *wl_buf = buffer.buffer;
|
|
buffer.buffer = NULL;
|
|
destroy_buffer(&buffer);
|
|
return wl_buf;
|
|
}
|
|
|
|
#define FRACT_DENOM 120
|
|
|
|
// Return the size of the buffer that should be attached to this output
|
|
static void get_buffer_size(const struct swaybg_output *output,
|
|
uint32_t *buffer_width, uint32_t *buffer_height) {
|
|
if (output->config->mode == BACKGROUND_MODE_SOLID_COLOR &&
|
|
output->state->viewporter) {
|
|
*buffer_width = 1;
|
|
*buffer_height = 1;
|
|
} else if (output->pref_fract_scale && output->state->viewporter) {
|
|
// rounding mode is 'round half up'
|
|
*buffer_width = (output->width * output->pref_fract_scale +
|
|
FRACT_DENOM / 2) / FRACT_DENOM;
|
|
*buffer_height = (output->height * output->pref_fract_scale +
|
|
FRACT_DENOM / 2) / FRACT_DENOM;
|
|
} else {
|
|
*buffer_width = output->width * output->scale;
|
|
*buffer_height = output->height * output->scale;
|
|
}
|
|
}
|
|
|
|
static void render_frame(struct swaybg_output *output, cairo_surface_t *surface) {
|
|
uint32_t buffer_width, buffer_height;
|
|
get_buffer_size(output, &buffer_width, &buffer_height);
|
|
|
|
// Attach a new buffer if the desired size has changed
|
|
struct wl_buffer *buf = NULL;
|
|
if (buffer_width != output->buffer_width ||
|
|
buffer_height != output->buffer_height) {
|
|
buf = draw_buffer(output, surface,
|
|
buffer_width, buffer_height);
|
|
if (!buf) {
|
|
return;
|
|
}
|
|
|
|
wl_surface_attach(output->surface, buf, 0, 0);
|
|
wl_surface_damage_buffer(output->surface, 0, 0,
|
|
buffer_width, buffer_height);
|
|
|
|
output->buffer_width = buffer_width;
|
|
output->buffer_height = buffer_height;
|
|
}
|
|
|
|
if (output->viewport) {
|
|
wp_viewport_set_destination(output->viewport, output->width, output->height);
|
|
} else {
|
|
wl_surface_set_buffer_scale(output->surface, output->scale);
|
|
}
|
|
wl_surface_commit(output->surface);
|
|
if (buf) {
|
|
wl_buffer_destroy(buf);
|
|
}
|
|
}
|
|
|
|
static void destroy_swaybg_image(struct swaybg_image *image) {
|
|
if (!image) {
|
|
return;
|
|
}
|
|
wl_list_remove(&image->link);
|
|
free(image);
|
|
}
|
|
|
|
static void destroy_swaybg_output_config(struct swaybg_output_config *config) {
|
|
if (!config) {
|
|
return;
|
|
}
|
|
wl_list_remove(&config->link);
|
|
free(config->output);
|
|
free(config);
|
|
}
|
|
|
|
static void destroy_swaybg_output(struct swaybg_output *output) {
|
|
if (!output) {
|
|
return;
|
|
}
|
|
wl_list_remove(&output->link);
|
|
if (output->layer_surface != NULL) {
|
|
zwlr_layer_surface_v1_destroy(output->layer_surface);
|
|
}
|
|
if (output->surface != NULL) {
|
|
wl_surface_destroy(output->surface);
|
|
}
|
|
if (output->viewport != NULL) {
|
|
wp_viewport_destroy(output->viewport);
|
|
}
|
|
if (output->fract_scale != NULL) {
|
|
wp_fractional_scale_v1_destroy(output->fract_scale);
|
|
}
|
|
wl_output_destroy(output->wl_output);
|
|
free(output->name);
|
|
free(output->identifier);
|
|
free(output);
|
|
}
|
|
|
|
static void layer_surface_configure(void *data,
|
|
struct zwlr_layer_surface_v1 *surface,
|
|
uint32_t serial, uint32_t width, uint32_t height) {
|
|
struct swaybg_output *output = data;
|
|
output->width = width;
|
|
output->height = height;
|
|
output->dirty = true;
|
|
output->configure_serial = serial;
|
|
output->needs_ack = true;
|
|
}
|
|
|
|
static void layer_surface_closed(void *data,
|
|
struct zwlr_layer_surface_v1 *surface) {
|
|
struct swaybg_output *output = data;
|
|
swaybg_log(LOG_DEBUG, "Destroying output %s (%s)",
|
|
output->name, output->identifier);
|
|
destroy_swaybg_output(output);
|
|
}
|
|
|
|
static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
|
|
.configure = layer_surface_configure,
|
|
.closed = layer_surface_closed,
|
|
};
|
|
|
|
static void fract_preferred_scale(void *data, struct wp_fractional_scale_v1 *f,
|
|
uint32_t scale) {
|
|
struct swaybg_output *output = data;
|
|
output->pref_fract_scale = scale;
|
|
}
|
|
|
|
static const struct wp_fractional_scale_v1_listener fract_scale_listener = {
|
|
.preferred_scale = fract_preferred_scale
|
|
};
|
|
|
|
static void output_geometry(void *data, struct wl_output *output, int32_t x,
|
|
int32_t y, int32_t width_mm, int32_t height_mm, int32_t subpixel,
|
|
const char *make, const char *model, int32_t transform) {
|
|
// Who cares
|
|
}
|
|
|
|
static void output_mode(void *data, struct wl_output *output, uint32_t flags,
|
|
int32_t width, int32_t height, int32_t refresh) {
|
|
// Who cares
|
|
}
|
|
|
|
static void create_layer_surface(struct swaybg_output *output) {
|
|
output->surface = wl_compositor_create_surface(output->state->compositor);
|
|
assert(output->surface);
|
|
|
|
// Empty input region
|
|
struct wl_region *input_region =
|
|
wl_compositor_create_region(output->state->compositor);
|
|
assert(input_region);
|
|
wl_surface_set_input_region(output->surface, input_region);
|
|
wl_region_destroy(input_region);
|
|
|
|
if (output->state->fract_scale_manager) {
|
|
output->fract_scale = wp_fractional_scale_manager_v1_get_fractional_scale(
|
|
output->state->fract_scale_manager, output->surface);
|
|
assert(output->fract_scale);
|
|
wp_fractional_scale_v1_add_listener(output->fract_scale,
|
|
&fract_scale_listener, output);
|
|
}
|
|
|
|
if (output->state->viewporter &&
|
|
(output->config->mode == BACKGROUND_MODE_SOLID_COLOR ||
|
|
output->state->fract_scale_manager)) {
|
|
output->viewport = wp_viewporter_get_viewport(
|
|
output->state->viewporter, output->surface);
|
|
}
|
|
|
|
output->layer_surface = zwlr_layer_shell_v1_get_layer_surface(
|
|
output->state->layer_shell, output->surface, output->wl_output,
|
|
ZWLR_LAYER_SHELL_V1_LAYER_BACKGROUND, "wallpaper");
|
|
assert(output->layer_surface);
|
|
|
|
zwlr_layer_surface_v1_set_size(output->layer_surface, 0, 0);
|
|
zwlr_layer_surface_v1_set_anchor(output->layer_surface,
|
|
ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP |
|
|
ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT |
|
|
ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM |
|
|
ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT);
|
|
zwlr_layer_surface_v1_set_exclusive_zone(output->layer_surface, -1);
|
|
zwlr_layer_surface_v1_add_listener(output->layer_surface,
|
|
&layer_surface_listener, output);
|
|
wl_surface_commit(output->surface);
|
|
}
|
|
|
|
static void output_done(void *data, struct wl_output *wl_output) {
|
|
struct swaybg_output *output = data;
|
|
if (!output->config) {
|
|
swaybg_log(LOG_DEBUG, "Could not find config for output %s (%s)",
|
|
output->name, output->identifier);
|
|
destroy_swaybg_output(output);
|
|
} else if (!output->layer_surface) {
|
|
swaybg_log(LOG_DEBUG, "Found config %s for output %s (%s)",
|
|
output->config->output, output->name, output->identifier);
|
|
create_layer_surface(output);
|
|
}
|
|
}
|
|
|
|
static void output_scale(void *data, struct wl_output *wl_output,
|
|
int32_t scale) {
|
|
struct swaybg_output *output = data;
|
|
output->scale = scale;
|
|
if (output->state->run_display && output->width > 0 && output->height > 0) {
|
|
output->dirty = true;
|
|
}
|
|
}
|
|
|
|
static void find_config(struct swaybg_output *output, const char *name) {
|
|
struct swaybg_output_config *config = NULL;
|
|
wl_list_for_each(config, &output->state->configs, link) {
|
|
if (strcmp(config->output, name) == 0) {
|
|
output->config = config;
|
|
return;
|
|
} else if (!output->config && strcmp(config->output, "*") == 0) {
|
|
output->config = config;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void output_name(void *data, struct wl_output *wl_output,
|
|
const char *name) {
|
|
struct swaybg_output *output = data;
|
|
output->name = strdup(name);
|
|
|
|
// If description was sent first, the config may already be populated. If
|
|
// there is an identifier config set, keep it.
|
|
if (!output->config || strcmp(output->config->output, "*") == 0) {
|
|
find_config(output, name);
|
|
}
|
|
}
|
|
|
|
static void output_description(void *data, struct wl_output *wl_output,
|
|
const char *description) {
|
|
struct swaybg_output *output = data;
|
|
|
|
// wlroots currently sets the description to `make model serial (name)`
|
|
// If this changes in the future, this will need to be modified.
|
|
char *paren = strrchr(description, '(');
|
|
if (paren) {
|
|
size_t length = paren - description;
|
|
output->identifier = malloc(length);
|
|
if (!output->identifier) {
|
|
swaybg_log(LOG_ERROR, "Failed to allocate output identifier");
|
|
return;
|
|
}
|
|
strncpy(output->identifier, description, length);
|
|
output->identifier[length - 1] = '\0';
|
|
|
|
find_config(output, output->identifier);
|
|
}
|
|
}
|
|
|
|
static const struct wl_output_listener output_listener = {
|
|
.geometry = output_geometry,
|
|
.mode = output_mode,
|
|
.done = output_done,
|
|
.scale = output_scale,
|
|
.name = output_name,
|
|
.description = output_description,
|
|
};
|
|
|
|
static void handle_global(void *data, struct wl_registry *registry,
|
|
uint32_t name, const char *interface, uint32_t version) {
|
|
struct swaybg_state *state = data;
|
|
if (strcmp(interface, wl_compositor_interface.name) == 0) {
|
|
state->compositor =
|
|
wl_registry_bind(registry, name, &wl_compositor_interface, 4);
|
|
} else if (strcmp(interface, wl_shm_interface.name) == 0) {
|
|
state->shm = wl_registry_bind(registry, name, &wl_shm_interface, 1);
|
|
} else if (strcmp(interface, wl_output_interface.name) == 0) {
|
|
struct swaybg_output *output = calloc(1, sizeof(struct swaybg_output));
|
|
output->state = state;
|
|
output->scale = 1;
|
|
output->wl_name = name;
|
|
output->wl_output =
|
|
wl_registry_bind(registry, name, &wl_output_interface, 4);
|
|
wl_output_add_listener(output->wl_output, &output_listener, output);
|
|
wl_list_insert(&state->outputs, &output->link);
|
|
} else if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
|
|
state->layer_shell =
|
|
wl_registry_bind(registry, name, &zwlr_layer_shell_v1_interface, 1);
|
|
} else if (strcmp(interface, wp_viewporter_interface.name) == 0) {
|
|
state->viewporter = wl_registry_bind(registry, name,
|
|
&wp_viewporter_interface, 1);
|
|
} else if (strcmp(interface,
|
|
wp_single_pixel_buffer_manager_v1_interface.name) == 0) {
|
|
state->single_pixel_buffer_manager = wl_registry_bind(registry, name,
|
|
&wp_single_pixel_buffer_manager_v1_interface, 1);
|
|
} else if (strcmp(interface, wp_fractional_scale_manager_v1_interface.name) == 0) {
|
|
state->fract_scale_manager = wl_registry_bind(registry, name,
|
|
&wp_fractional_scale_manager_v1_interface, 1);
|
|
}
|
|
}
|
|
|
|
static void handle_global_remove(void *data, struct wl_registry *registry,
|
|
uint32_t name) {
|
|
struct swaybg_state *state = data;
|
|
struct swaybg_output *output, *tmp;
|
|
wl_list_for_each_safe(output, tmp, &state->outputs, link) {
|
|
if (output->wl_name == name) {
|
|
swaybg_log(LOG_DEBUG, "Destroying output %s (%s)",
|
|
output->name, output->identifier);
|
|
destroy_swaybg_output(output);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static const struct wl_registry_listener registry_listener = {
|
|
.global = handle_global,
|
|
.global_remove = handle_global_remove,
|
|
};
|
|
|
|
static bool store_swaybg_output_config(struct swaybg_state *state,
|
|
struct swaybg_output_config *config) {
|
|
struct swaybg_output_config *oc = NULL;
|
|
wl_list_for_each(oc, &state->configs, link) {
|
|
if (strcmp(config->output, oc->output) == 0) {
|
|
// Merge on top
|
|
if (config->image_path) {
|
|
oc->image_path = config->image_path;
|
|
}
|
|
if (config->color) {
|
|
oc->color = config->color;
|
|
}
|
|
if (config->mode != BACKGROUND_MODE_INVALID) {
|
|
oc->mode = config->mode;
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
// New config, just add it
|
|
wl_list_insert(&state->configs, &config->link);
|
|
return true;
|
|
}
|
|
|
|
static void parse_command_line(int argc, char **argv,
|
|
struct swaybg_state *state) {
|
|
static struct option long_options[] = {
|
|
{"color", required_argument, NULL, 'c'},
|
|
{"help", no_argument, NULL, 'h'},
|
|
{"image", required_argument, NULL, 'i'},
|
|
{"mode", required_argument, NULL, 'm'},
|
|
{"output", required_argument, NULL, 'o'},
|
|
{"version", no_argument, NULL, 'v'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
const char *usage =
|
|
"Usage: swaybg <options...>\n"
|
|
"\n"
|
|
" -c, --color RRGGBB Set the background color.\n"
|
|
" -h, --help Show help message and quit.\n"
|
|
" -i, --image <path> Set the image to display.\n"
|
|
" -m, --mode <mode> Set the mode to use for the image.\n"
|
|
" -o, --output <name> Set the output to operate on or * for all.\n"
|
|
" -v, --version Show the version number and quit.\n"
|
|
"\n"
|
|
"Background Modes:\n"
|
|
" stretch, fit, fill, center, tile, or solid_color\n";
|
|
|
|
struct swaybg_output_config *config = calloc(1, sizeof(struct swaybg_output_config));
|
|
config->output = strdup("*");
|
|
config->mode = BACKGROUND_MODE_INVALID;
|
|
wl_list_init(&config->link); // init for safe removal
|
|
|
|
int c;
|
|
while (1) {
|
|
int option_index = 0;
|
|
c = getopt_long(argc, argv, "c:hi:m:o:v", long_options, &option_index);
|
|
if (c == -1) {
|
|
break;
|
|
}
|
|
switch (c) {
|
|
case 'c': // color
|
|
if (!parse_color(optarg, &config->color)) {
|
|
swaybg_log(LOG_ERROR, "%s is not a valid color for swaybg. "
|
|
"Color should be specified as rrggbb or #rrggbb (no alpha).", optarg);
|
|
continue;
|
|
}
|
|
break;
|
|
case 'i': // image
|
|
config->image_path = optarg;
|
|
break;
|
|
case 'm': // mode
|
|
config->mode = parse_background_mode(optarg);
|
|
if (config->mode == BACKGROUND_MODE_INVALID) {
|
|
swaybg_log(LOG_ERROR, "Invalid mode: %s", optarg);
|
|
}
|
|
break;
|
|
case 'o': // output
|
|
if (config && !store_swaybg_output_config(state, config)) {
|
|
// Empty config or merged on top of an existing one
|
|
destroy_swaybg_output_config(config);
|
|
}
|
|
config = calloc(1, sizeof(struct swaybg_output_config));
|
|
config->output = strdup(optarg);
|
|
config->mode = BACKGROUND_MODE_INVALID;
|
|
wl_list_init(&config->link); // init for safe removal
|
|
break;
|
|
case 'v': // version
|
|
fprintf(stdout, "swaybg version " SWAYBG_VERSION "\n");
|
|
exit(EXIT_SUCCESS);
|
|
break;
|
|
default:
|
|
fprintf(c == 'h' ? stdout : stderr, "%s", usage);
|
|
exit(c == 'h' ? EXIT_SUCCESS : EXIT_FAILURE);
|
|
}
|
|
}
|
|
if (config && !store_swaybg_output_config(state, config)) {
|
|
// Empty config or merged on top of an existing one
|
|
destroy_swaybg_output_config(config);
|
|
}
|
|
|
|
// Check for invalid options
|
|
if (optind < argc) {
|
|
config = NULL;
|
|
struct swaybg_output_config *tmp = NULL;
|
|
wl_list_for_each_safe(config, tmp, &state->configs, link) {
|
|
destroy_swaybg_output_config(config);
|
|
}
|
|
// continue into empty list
|
|
}
|
|
if (wl_list_empty(&state->configs)) {
|
|
fprintf(stderr, "%s", usage);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
// Set default mode and remove empties
|
|
config = NULL;
|
|
struct swaybg_output_config *tmp = NULL;
|
|
wl_list_for_each_safe(config, tmp, &state->configs, link) {
|
|
if (!config->image_path && !config->color) {
|
|
destroy_swaybg_output_config(config);
|
|
} else if (config->mode == BACKGROUND_MODE_INVALID) {
|
|
config->mode = config->image_path
|
|
? BACKGROUND_MODE_STRETCH
|
|
: BACKGROUND_MODE_SOLID_COLOR;
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
swaybg_log_init(LOG_DEBUG);
|
|
|
|
struct swaybg_state state = {0};
|
|
wl_list_init(&state.configs);
|
|
wl_list_init(&state.outputs);
|
|
wl_list_init(&state.images);
|
|
|
|
parse_command_line(argc, argv, &state);
|
|
|
|
// Identify distinct image paths which will need to be loaded
|
|
struct swaybg_image *image;
|
|
struct swaybg_output_config *config;
|
|
wl_list_for_each(config, &state.configs, link) {
|
|
if (!config->image_path) {
|
|
continue;
|
|
}
|
|
wl_list_for_each(image, &state.images, link) {
|
|
if (strcmp(image->path, config->image_path) == 0) {
|
|
config->image = image;
|
|
break;
|
|
}
|
|
}
|
|
if (config->image) {
|
|
continue;
|
|
}
|
|
image = calloc(1, sizeof(struct swaybg_image));
|
|
image->path = config->image_path;
|
|
wl_list_insert(&state.images, &image->link);
|
|
config->image = image;
|
|
}
|
|
|
|
state.display = wl_display_connect(NULL);
|
|
if (!state.display) {
|
|
swaybg_log(LOG_ERROR, "Unable to connect to the compositor. "
|
|
"If your compositor is running, check or set the "
|
|
"WAYLAND_DISPLAY environment variable.");
|
|
return 1;
|
|
}
|
|
|
|
struct wl_registry *registry = wl_display_get_registry(state.display);
|
|
wl_registry_add_listener(registry, ®istry_listener, &state);
|
|
if (wl_display_roundtrip(state.display) < 0) {
|
|
swaybg_log(LOG_ERROR, "wl_display_roundtrip failed");
|
|
return 1;
|
|
}
|
|
if (state.compositor == NULL || state.shm == NULL ||
|
|
state.layer_shell == NULL) {
|
|
swaybg_log(LOG_ERROR, "Missing a required Wayland interface");
|
|
return 1;
|
|
}
|
|
|
|
state.run_display = true;
|
|
while (wl_display_dispatch(state.display) != -1 && state.run_display) {
|
|
// Send acks, and determine which images need to be loaded
|
|
struct swaybg_output *output;
|
|
wl_list_for_each(output, &state.outputs, link) {
|
|
if (output->needs_ack) {
|
|
output->needs_ack = false;
|
|
zwlr_layer_surface_v1_ack_configure(
|
|
output->layer_surface,
|
|
output->configure_serial);
|
|
}
|
|
|
|
if (output->dirty) {
|
|
uint32_t buffer_width, buffer_height;
|
|
get_buffer_size(output, &buffer_width, &buffer_height);
|
|
bool buffer_change = output->buffer_width != buffer_width ||
|
|
output->buffer_height != buffer_height;
|
|
if (output->config->image && buffer_change) {
|
|
output->config->image->load_required = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Load images, render associated frames, and unload
|
|
wl_list_for_each(image, &state.images, link) {
|
|
if (!image->load_required) {
|
|
continue;
|
|
}
|
|
|
|
cairo_surface_t *surface = load_background_image(image->path);
|
|
if (!surface) {
|
|
swaybg_log(LOG_ERROR, "Failed to load image: %s", image->path);
|
|
continue;
|
|
}
|
|
|
|
wl_list_for_each(output, &state.outputs, link) {
|
|
if (output->dirty && output->config->image == image) {
|
|
output->dirty = false;
|
|
render_frame(output, surface);
|
|
}
|
|
}
|
|
|
|
image->load_required = false;
|
|
cairo_surface_destroy(surface);
|
|
}
|
|
|
|
// Redraw outputs without associated image
|
|
wl_list_for_each(output, &state.outputs, link) {
|
|
if (output->dirty) {
|
|
output->dirty = false;
|
|
render_frame(output, NULL);
|
|
}
|
|
}
|
|
}
|
|
|
|
struct swaybg_output *output, *tmp_output;
|
|
wl_list_for_each_safe(output, tmp_output, &state.outputs, link) {
|
|
destroy_swaybg_output(output);
|
|
}
|
|
|
|
struct swaybg_output_config *tmp_config = NULL;
|
|
wl_list_for_each_safe(config, tmp_config, &state.configs, link) {
|
|
destroy_swaybg_output_config(config);
|
|
}
|
|
|
|
struct swaybg_image *tmp_image;
|
|
wl_list_for_each_safe(image, tmp_image, &state.images, link) {
|
|
destroy_swaybg_image(image);
|
|
}
|
|
|
|
return 0;
|
|
}
|