aboutsummaryrefslogtreecommitdiff
path: root/swaygrab/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'swaygrab/main.c')
-rw-r--r--swaygrab/main.c298
1 files changed, 0 insertions, 298 deletions
diff --git a/swaygrab/main.c b/swaygrab/main.c
deleted file mode 100644
index 1b699bb9..00000000
--- a/swaygrab/main.c
+++ /dev/null
@@ -1,298 +0,0 @@
-#define _XOPEN_SOURCE 700
-#define _POSIX_C_SOURCE 199309L
-#include <stdio.h>
-#include <stdbool.h>
-#include <stdlib.h>
-#include <string.h>
-#include <getopt.h>
-#include <unistd.h>
-#include <stdint.h>
-#include <math.h>
-#include <time.h>
-#include <sys/wait.h>
-#include <json-c/json.h>
-#include "log.h"
-#include "ipc-client.h"
-#include "util.h"
-#include "swaygrab/json.h"
-
-void sway_terminate(int exit_code) {
- exit(exit_code);
-}
-
-void grab_and_apply_magick(const char *file, const char *payload,
- int socketfd, int raw) {
- uint32_t len = strlen(payload);
- char *pixels = ipc_single_command(socketfd,
- IPC_SWAY_GET_PIXELS, payload, &len);
- uint32_t *u32pixels = (uint32_t *)(pixels + 1);
- uint32_t width = u32pixels[0];
- uint32_t height = u32pixels[1];
- len -= 9;
- pixels += 9;
-
- if (width == 0 || height == 0) {
- // indicates geometry was clamped by WLC because it was outside of the output's area
- json_object *obj = json_tokener_parse(payload);
- json_object *output;
- json_object_object_get_ex(obj, "output", &output);
- const char *name = json_object_get_string(output);
- json_object_put(obj);
- sway_abort("Unknown output %s.", name);
- }
-
- if (raw) {
- fwrite(pixels, 1, len, stdout);
- fflush(stdout);
- free(pixels - 9);
- return;
- }
-
- char size[10 + 1 + 10 + 2 + 1]; // int32_t are max 10 digits
- sprintf(size, "%dx%d+0", width, height);
-
- pid_t child;
- int fd[2];
- pipe(fd);
-
- if ((child = fork()) < 0) {
- sway_log(L_ERROR, "Swaygrab failed to fork.");
- exit(EXIT_FAILURE);
- } else if (child != 0) {
- close(fd[0]);
- write(fd[1], pixels, len);
- close(fd[1]);
- free(pixels - 9);
- waitpid(child, NULL, 0);
- } else {
- close(fd[1]);
- if (dup2(fd[0], 0) != 0) {
- sway_log(L_ERROR, "Could not fdup the pipe");
- }
- close(fd[0]);
- execlp("convert", "convert", "-depth", "8", "-size", size, "rgba:-", "-flip", file, NULL);
- sway_log(L_ERROR, "Swaygrab could not run convert.");
- exit(EXIT_FAILURE);
- }
-}
-
-void grab_and_apply_movie_magic(const char *file, const char *payload,
- int socketfd, int raw, int framerate) {
- if (raw) {
- sway_log(L_ERROR, "Raw capture data is not yet supported. Proceeding with ffmpeg normally.");
- }
-
- uint32_t len = strlen(payload);
- char *pixels = ipc_single_command(socketfd,
- IPC_SWAY_GET_PIXELS, payload, &len);
- uint32_t *u32pixels = (uint32_t *)(pixels + 1);
- uint32_t width = u32pixels[0];
- uint32_t height = u32pixels[1];
- pixels += 9;
-
- if (width == 0 || height == 0) {
- // indicates geometry was clamped by WLC because it was outside of the output's area
- json_object *obj = json_tokener_parse(payload);
- json_object *output;
- json_object_object_get_ex(obj, "output", &output);
- const char *name = json_object_get_string(output);
- json_object_put(obj);
- sway_abort("Unknown output %s.", name);
- }
-
- char *ffmpeg_opts = getenv("SWAYGRAB_FFMPEG_OPTS");
- if(!ffmpeg_opts) {
- ffmpeg_opts = "";
- }
-
- const char *fmt = "ffmpeg %s -f rawvideo -framerate %d "
- "-video_size %dx%d -pixel_format argb "
- "-i pipe:0 -r %d -vf vflip %s";
- char *cmd = malloc(strlen(fmt) - 8 /*args*/
- + strlen(ffmpeg_opts) + numlen(width) + numlen(height)
- + numlen(framerate) * 2 + strlen(file) + 1);
- sprintf(cmd, fmt, ffmpeg_opts, framerate, width, height, framerate, file);
-
- long ns = (long)(1000000000 * (1.0 / framerate));
- struct timespec start, finish, ts;
- ts.tv_sec = 0;
-
- FILE *f = popen(cmd, "w");
- fwrite(pixels, 1, len, f);
- free(pixels - 9);
- int sleep = 0;
- while (sleep != -1) {
- clock_gettime(CLOCK_MONOTONIC, &start);
- len = strlen(payload);
- pixels = ipc_single_command(socketfd,
- IPC_SWAY_GET_PIXELS, payload, &len);
- pixels += 9;
- len -= 9;
-
- fwrite(pixels, 1, len, f);
-
- free(pixels - 9);
- clock_gettime(CLOCK_MONOTONIC, &finish);
- ts.tv_nsec = ns;
- double fts = (double)finish.tv_sec + 1.0e-9*finish.tv_nsec;
- double sts = (double)start.tv_sec + 1.0e-9*start.tv_nsec;
- long diff = (fts - sts) * 1000000000;
- ts.tv_nsec = ns - diff;
- if (ts.tv_nsec < 0) {
- ts.tv_nsec = 0;
- }
- sleep = nanosleep(&ts, NULL);
- }
- fflush(f);
-
- fclose(f);
- free(cmd);
-}
-
-char *default_filename(const char *extension) {
- int ext_len = strlen(extension);
- int len = 28 + ext_len; // format: "2015-12-17-180040_swaygrab.ext"
- char *filename = malloc(len * sizeof(char));
- time_t t = time(NULL);
-
- struct tm *lt = localtime(&t);
- strftime(filename, len, "%Y-%m-%d-%H%M%S_swaygrab.", lt);
- strncat(filename, extension, ext_len);
-
- return filename;
-}
-
-int main(int argc, char **argv) {
- static int capture = 0, raw = 0;
- char *socket_path = NULL;
- char *output = NULL;
- int framerate = 30;
- bool grab_focused = false;
-
- init_log(L_INFO);
-
- static struct option long_options[] = {
- {"help", no_argument, NULL, 'h'},
- {"capture", no_argument, NULL, 'c'},
- {"output", required_argument, NULL, 'o'},
- {"version", no_argument, NULL, 'v'},
- {"socket", required_argument, NULL, 's'},
- {"raw", no_argument, NULL, 'r'},
- {"rate", required_argument, NULL, 'R'},
- {"focused", no_argument, NULL, 'f'},
- {0, 0, 0, 0}
- };
-
- const char *usage =
- "Usage: swaygrab [options] [file]\n"
- "\n"
- " -h, --help Show help message and quit.\n"
- " -c, --capture Capture video.\n"
- " -o, --output <output> Output source.\n"
- " -v, --version Show the version number and quit.\n"
- " -s, --socket <socket> Use the specified socket.\n"
- " -R, --rate <rate> Specify framerate (default: 30)\n"
- " -r, --raw Write raw rgba data to stdout.\n"
- " -f, --focused Grab the focused container.\n";
-
- int c;
- while (1) {
- int option_index = 0;
- c = getopt_long(argc, argv, "hco:vs:R:rf", long_options, &option_index);
- if (c == -1) {
- break;
- }
- switch (c) {
- case 'f':
- grab_focused = true;
- break;
- case 's': // Socket
- socket_path = strdup(optarg);
- break;
- case 'r':
- raw = 1;
- break;
- case 'o': // output
- output = strdup(optarg);
- break;
- case 'c':
- capture = 1;
- break;
- case 'R': // Frame rate
- framerate = atoi(optarg);
- break;
- case 'v':
- fprintf(stdout, "sway version " SWAY_VERSION "\n");
- exit(EXIT_SUCCESS);
- break;
- default:
- fprintf(stderr, "%s", usage);
- exit(EXIT_FAILURE);
- }
- }
-
- if (!socket_path) {
- socket_path = get_socketpath();
- if (!socket_path) {
- sway_abort("Unable to retrieve socket path");
- }
- }
-
- char *file = NULL;
- if (raw) {
- if (optind >= argc + 1) {
- sway_abort("Invalid usage. See `man swaygrab` %d %d", argc, optind);
- }
- } else if (optind < argc) {
- file = strdup(argv[optind]);
- }
-
- int socketfd = ipc_open_socket(socket_path);
- free(socket_path);
-
- init_json_tree(socketfd);
-
- struct wlc_geometry *geo;
-
- if (grab_focused) {
- output = get_focused_output();
- json_object *con = get_focused_container();
- json_object *name;
- json_object_object_get_ex(con, "name", &name);
- geo = get_container_geometry(con);
- free(con);
- } else {
- if (!output) {
- output = get_focused_output();
- }
- geo = get_container_geometry(get_output_container(output));
- // the geometry of the output in the get_tree response is relative to a global (0, 0).
- // we need it to be relative to itself, so set origin to (0, 0) always.
- geo->origin.x = 0;
- geo->origin.y = 0;
- }
-
- const char *payload = create_payload(output, geo);
-
- free(geo);
-
- if (!file) {
- if (!capture) {
- file = default_filename("png");
- } else {
- file = default_filename("webm");
- }
- }
-
- if (!capture) {
- grab_and_apply_magick(file, payload, socketfd, raw);
- } else {
- grab_and_apply_movie_magic(file, payload, socketfd, raw, framerate);
- }
-
- free_json_tree();
- free(output);
- free(file);
- close(socketfd);
- return 0;
-}
generated by cgit v1.2.3 (git 2.39.1) at 2025-03-31 07:29:17 +0000