swayfx/swaybg/main.c

345 lines
10 KiB
C
Raw Normal View History

2018-03-29 03:21:50 +11:00
#include <assert.h>
#include <ctype.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
2015-11-29 00:49:02 +11:00
#include <string.h>
#include <time.h>
#include <wayland-client.h>
#include "background-image.h"
2018-03-28 06:25:25 +11:00
#include "cairo.h"
#include "log.h"
#include "pool-buffer.h"
2018-03-28 06:25:25 +11:00
#include "util.h"
#include "wlr-layer-shell-unstable-v1-client-protocol.h"
#include "xdg-output-unstable-v1-client-protocol.h"
struct swaybg_state;
2018-03-28 06:25:25 +11:00
struct swaybg_args {
const char *output;
2018-03-28 06:25:25 +11:00
const char *path;
enum background_mode mode;
const char *fallback;
2018-03-28 06:25:25 +11:00
};
2018-03-28 06:36:56 +11:00
struct swaybg_context {
uint32_t color;
cairo_surface_t *image;
};
struct swaybg_output {
struct wl_output *wl_output;
struct zxdg_output_v1 *xdg_output;
struct swaybg_state *state;
struct wl_list link;
int32_t scale;
};
2018-03-28 06:25:25 +11:00
struct swaybg_state {
const struct swaybg_args *args;
2018-03-28 06:36:56 +11:00
struct swaybg_context context;
2018-03-28 06:25:25 +11:00
struct wl_display *display;
struct wl_compositor *compositor;
struct wl_shm *shm;
struct wl_list outputs;
struct zwlr_layer_shell_v1 *layer_shell;
struct zxdg_output_manager_v1 *xdg_output_manager;
2018-03-28 06:25:25 +11:00
struct swaybg_output *output;
2018-03-28 06:25:25 +11:00
struct wl_surface *surface;
2018-04-04 05:23:51 +10:00
struct wl_region *input_region;
2018-03-28 06:25:25 +11:00
struct zwlr_layer_surface_v1 *layer_surface;
bool run_display;
uint32_t width, height;
struct pool_buffer buffers[2];
struct pool_buffer *current_buffer;
};
bool is_valid_color(const char *color) {
int len = strlen(color);
if (len != 7 || color[0] != '#') {
sway_log(SWAY_ERROR, "%s is not a valid color for swaybg. "
"Color should be specified as #rrggbb (no alpha).", color);
return false;
}
int i;
for (i = 1; i < len; ++i) {
if (!isxdigit(color[i])) {
return false;
}
}
return true;
}
2018-03-28 06:25:25 +11:00
static void render_frame(struct swaybg_state *state) {
int buffer_width = state->width * state->output->scale,
buffer_height = state->height * state->output->scale;
2018-04-03 12:48:13 +10:00
state->current_buffer = get_next_buffer(state->shm,
state->buffers, buffer_width, buffer_height);
if (!state->current_buffer) {
return;
}
2018-03-28 06:25:25 +11:00
cairo_t *cairo = state->current_buffer->cairo;
cairo_save(cairo);
cairo_set_operator(cairo, CAIRO_OPERATOR_CLEAR);
cairo_paint(cairo);
cairo_restore(cairo);
if (state->args->mode == BACKGROUND_MODE_SOLID_COLOR) {
2018-03-28 06:36:56 +11:00
cairo_set_source_u32(cairo, state->context.color);
2018-03-28 06:25:25 +11:00
cairo_paint(cairo);
} else {
if (state->args->fallback && state->context.color) {
cairo_set_source_u32(cairo, state->context.color);
cairo_paint(cairo);
}
render_background_image(cairo, state->context.image,
2018-04-03 12:48:13 +10:00
state->args->mode, buffer_width, buffer_height);
2018-03-28 06:25:25 +11:00
}
wl_surface_set_buffer_scale(state->surface, state->output->scale);
2018-03-28 06:25:25 +11:00
wl_surface_attach(state->surface, state->current_buffer->buffer, 0, 0);
2019-01-17 21:14:36 +11:00
wl_surface_damage_buffer(state->surface, 0, 0, INT32_MAX, INT32_MAX);
2018-03-28 06:25:25 +11:00
wl_surface_commit(state->surface);
}
2018-03-28 06:36:56 +11:00
static bool prepare_context(struct swaybg_state *state) {
if (state->args->mode == BACKGROUND_MODE_SOLID_COLOR) {
2018-03-28 06:36:56 +11:00
state->context.color = parse_color(state->args->path);
return is_valid_color(state->args->path);
}
if (state->args->fallback && is_valid_color(state->args->fallback)) {
state->context.color = parse_color(state->args->fallback);
}
if (!(state->context.image = load_background_image(state->args->path))) {
2018-03-28 06:36:56 +11:00
return false;
}
return true;
}
2018-03-28 06:25:25 +11:00
static void layer_surface_configure(void *data,
struct zwlr_layer_surface_v1 *surface,
uint32_t serial, uint32_t width, uint32_t height) {
struct swaybg_state *state = data;
state->width = width;
state->height = height;
zwlr_layer_surface_v1_ack_configure(surface, serial);
render_frame(state);
2018-03-28 06:25:25 +11:00
}
static void layer_surface_closed(void *data,
struct zwlr_layer_surface_v1 *surface) {
struct swaybg_state *state = data;
zwlr_layer_surface_v1_destroy(state->layer_surface);
wl_surface_destroy(state->surface);
2018-04-04 05:23:51 +10:00
wl_region_destroy(state->input_region);
2018-03-28 06:25:25 +11:00
state->run_display = false;
}
static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
2018-03-28 06:25:25 +11:00
.configure = layer_surface_configure,
.closed = layer_surface_closed,
};
2018-04-04 06:47:06 +10:00
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 output_done(void *data, struct wl_output *output) {
// Who cares
}
static void output_scale(void *data, struct wl_output *wl_output,
int32_t scale) {
struct swaybg_output *output = data;
struct swaybg_state *state = output->state;
output->scale = scale;
if (state->output == output && state->run_display) {
2018-04-04 06:47:06 +10:00
render_frame(state);
}
}
static const struct wl_output_listener output_listener = {
2018-04-04 06:47:06 +10:00
.geometry = output_geometry,
.mode = output_mode,
.done = output_done,
.scale = output_scale,
};
static void xdg_output_handle_logical_position(void *data,
struct zxdg_output_v1 *xdg_output, int32_t x, int32_t y) {
// Who cares
}
static void xdg_output_handle_logical_size(void *data,
struct zxdg_output_v1 *xdg_output, int32_t width, int32_t height) {
// Who cares
}
static void xdg_output_handle_name(void *data,
struct zxdg_output_v1 *xdg_output, const char *name) {
struct swaybg_output *output = data;
struct swaybg_state *state = output->state;
if (strcmp(name, state->args->output) == 0) {
assert(state->output == NULL);
state->output = output;
}
}
static void xdg_output_handle_description(void *data,
struct zxdg_output_v1 *xdg_output, const char *description) {
// Who cares
}
static void xdg_output_handle_done(void *data,
struct zxdg_output_v1 *xdg_output) {
// Who cares
}
static const struct zxdg_output_v1_listener xdg_output_listener = {
.logical_position = xdg_output_handle_logical_position,
.logical_size = xdg_output_handle_logical_size,
.name = xdg_output_handle_name,
.description = xdg_output_handle_description,
.done = xdg_output_handle_done,
};
2018-03-28 06:25:25 +11:00
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 =
2019-01-17 21:14:36 +11:00
wl_registry_bind(registry, name, &wl_compositor_interface, 4);
2018-03-28 06:25:25 +11:00
} else if (strcmp(interface, wl_shm_interface.name) == 0) {
state->shm = wl_registry_bind(registry, name, &wl_shm_interface, 1);
2018-03-28 06:25:25 +11:00
} else if (strcmp(interface, wl_output_interface.name) == 0) {
struct swaybg_output *output = calloc(1, sizeof(struct swaybg_output));
output->state = state;
output->wl_output =
wl_registry_bind(registry, name, &wl_output_interface, 3);
wl_output_add_listener(output->wl_output, &output_listener, output);
wl_list_insert(&state->outputs, &output->link);
2018-03-28 06:25:25 +11:00
} 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, zxdg_output_manager_v1_interface.name) == 0) {
state->xdg_output_manager = wl_registry_bind(registry, name,
&zxdg_output_manager_v1_interface, 2);
2018-03-28 06:25:25 +11:00
}
}
static void handle_global_remove(void *data, struct wl_registry *registry,
uint32_t name) {
// who cares
}
static const struct wl_registry_listener registry_listener = {
.global = handle_global,
.global_remove = handle_global_remove,
};
2015-11-26 01:53:02 +11:00
int main(int argc, const char **argv) {
sway_log_init(SWAY_DEBUG, NULL);
2018-03-28 06:25:25 +11:00
struct swaybg_args args = {0};
struct swaybg_state state = { .args = &args };
wl_list_init(&state.outputs);
if (argc < 4 || argc > 5) {
sway_log(SWAY_ERROR, "Do not run this program manually. "
"See `man 5 sway-output` and look for background options.");
2018-03-28 06:25:25 +11:00
return 1;
}
args.output = argv[1];
2018-03-28 06:25:25 +11:00
args.path = argv[2];
2018-04-03 12:48:13 +10:00
args.mode = parse_background_mode(argv[3]);
if (args.mode == BACKGROUND_MODE_INVALID) {
2018-03-28 06:25:25 +11:00
return 1;
}
args.fallback = argc == 5 ? argv[4] : NULL;
if (!prepare_context(&state)) {
return 1;
}
state.display = wl_display_connect(NULL);
if (!state.display) {
sway_log(SWAY_ERROR, "Unable to connect to the compositor. "
"If your compositor is running, check or set the "
"WAYLAND_DISPLAY environment variable.");
return 1;
}
2018-03-28 06:25:25 +11:00
struct wl_registry *registry = wl_display_get_registry(state.display);
wl_registry_add_listener(registry, &registry_listener, &state);
wl_display_roundtrip(state.display);
if (state.compositor == NULL || state.shm == NULL ||
state.layer_shell == NULL || state.xdg_output_manager == NULL) {
sway_log(SWAY_ERROR, "Missing a required Wayland interface");
return 1;
}
2018-03-28 06:25:25 +11:00
struct swaybg_output *output;
wl_list_for_each(output, &state.outputs, link) {
output->xdg_output = zxdg_output_manager_v1_get_xdg_output(
state.xdg_output_manager, output->wl_output);
zxdg_output_v1_add_listener(output->xdg_output,
&xdg_output_listener, output);
}
// Second roundtrip to get xdg_output properties
2018-04-04 06:47:06 +10:00
wl_display_roundtrip(state.display);
if (state.output == NULL) {
sway_log(SWAY_ERROR, "Cannot find output '%s'", args.output);
return 1;
}
2018-04-04 06:47:06 +10:00
state.surface = wl_compositor_create_surface(state.compositor);
assert(state.surface);
2018-03-28 06:25:25 +11:00
// Empty input region
state.input_region = wl_compositor_create_region(state.compositor);
assert(state.input_region);
2018-04-04 05:23:51 +10:00
wl_surface_set_input_region(state.surface, state.input_region);
2018-03-28 06:25:25 +11:00
state.layer_surface = zwlr_layer_shell_v1_get_layer_surface(
state.layer_shell, state.surface, state.output->wl_output,
2018-03-28 06:25:25 +11:00
ZWLR_LAYER_SHELL_V1_LAYER_BACKGROUND, "wallpaper");
2018-03-29 03:21:50 +11:00
assert(state.layer_surface);
2018-03-28 06:25:25 +11:00
zwlr_layer_surface_v1_set_size(state.layer_surface, 0, 0);
zwlr_layer_surface_v1_set_anchor(state.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);
2018-03-31 09:29:43 +11:00
zwlr_layer_surface_v1_set_exclusive_zone(state.layer_surface, -1);
2018-03-28 06:25:25 +11:00
zwlr_layer_surface_v1_add_listener(state.layer_surface,
&layer_surface_listener, &state);
wl_surface_commit(state.surface);
wl_display_roundtrip(state.display);
2018-04-03 12:48:13 +10:00
state.run_display = true;
2018-03-28 06:25:25 +11:00
while (wl_display_dispatch(state.display) != -1 && state.run_display) {
// This space intentionally left blank
}
return 0;
}