mirror of
https://github.com/hyprwm/wlroots-hyprland.git
synced 2024-11-26 06:35:58 +01:00
Use ImageMagick to write PNG screenshot
This commit is contained in:
parent
a87f016017
commit
63af97800f
1 changed files with 30 additions and 9 deletions
|
@ -27,6 +27,7 @@
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <sys/mman.h>
|
#include <sys/mman.h>
|
||||||
|
#include <sys/wait.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
|
@ -184,7 +185,7 @@ static void argb_to_rgba(uint32_t *data, size_t height, size_t stride) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
write_png(int width, int height)
|
write_image(const char *filename, int width, int height)
|
||||||
{
|
{
|
||||||
int output_stride, buffer_stride, i;
|
int output_stride, buffer_stride, i;
|
||||||
void *data, *d, *s;
|
void *data, *d, *s;
|
||||||
|
@ -213,12 +214,33 @@ write_png(int width, int height)
|
||||||
|
|
||||||
argb_to_rgba(data, height, buffer_stride);
|
argb_to_rgba(data, height, buffer_stride);
|
||||||
|
|
||||||
// TODO: call convert
|
char size[10 + 1 + 10 + 2 + 1]; // int32_t are max 10 digits
|
||||||
FILE *f = fopen("wayland-screenshot", "w");
|
sprintf(size, "%dx%d+0", width, height);
|
||||||
fwrite(data, buffer_stride * height, 1, f);
|
|
||||||
fclose(f);
|
|
||||||
|
|
||||||
|
int fd[2];
|
||||||
|
pipe(fd);
|
||||||
|
|
||||||
|
pid_t child = fork();
|
||||||
|
if (child < 0) {
|
||||||
|
fprintf(stderr, "fork() failed\n");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
} else if (child != 0) {
|
||||||
|
close(fd[0]);
|
||||||
|
write(fd[1], data, buffer_stride * height);
|
||||||
|
close(fd[1]);
|
||||||
free(data);
|
free(data);
|
||||||
|
waitpid(child, NULL, 0);
|
||||||
|
} else {
|
||||||
|
close(fd[1]);
|
||||||
|
if (dup2(fd[0], 0) != 0) {
|
||||||
|
fprintf(stderr, "cannot dup the pipe\n");
|
||||||
|
}
|
||||||
|
close(fd[0]);
|
||||||
|
execlp("convert", "convert", "-depth", "8", "-size", size, "rgba:-",
|
||||||
|
"-alpha", "opaque", filename, NULL);
|
||||||
|
fprintf(stderr, "cannot execute convert\n");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static int
|
static int
|
||||||
|
@ -296,7 +318,6 @@ int main(int argc, char *argv[])
|
||||||
wl_display_roundtrip(display);
|
wl_display_roundtrip(display);
|
||||||
}
|
}
|
||||||
|
|
||||||
write_png(width, height);
|
write_image("wayland-screenshot.png", width, height);
|
||||||
|
return EXIT_SUCCESS;
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue