aboutsummaryrefslogtreecommitdiffstats
path: root/src/scrotty.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/scrotty.c')
-rw-r--r--src/scrotty.c65
1 files changed, 48 insertions, 17 deletions
diff --git a/src/scrotty.c b/src/scrotty.c
index 4297923..591db78 100644
--- a/src/scrotty.c
+++ b/src/scrotty.c
@@ -1,6 +1,6 @@
/**
* scrotty — Screenshot program for Linux's TTY
- * Copyright © 2014 Mattias Andrée (maandree@member.fsf.org)
+ * Copyright © 2014, 2015 Mattias Andrée (maandree@member.fsf.org)
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
@@ -81,12 +81,14 @@ static const char* execname;
/**
* Arguments for `convert`,
* the output file should be printed into `convert_args[2]`.
+ * `NULL` is `convert` shall not be used.
*/
-static char** convert_args;
+static char** convert_args = NULL;
/**
- * Create an PNM-file that is sent to `convert` for convertion to a compressed format
+ * Create an PNM-file that is sent to `convert` for convertion
+ * to a compressed format, or directly to a file
*
* @param fbname The framebuffer device
* @param width The width of the image
@@ -170,7 +172,10 @@ static int save(const char* fbpath, const char* imgpath, long width, long height
{
int pipe_rw[2];
pid_t pid, reaped;
- int saved_errno, status;
+ int saved_errno, status, fd;
+
+ if (convert_args == NULL)
+ goto no_convert;
/* Create a pipe that for sending data into the `convert` program. */
if (pipe(pipe_rw) < 0)
@@ -218,6 +223,21 @@ static int save(const char* fbpath, const char* imgpath, long width, long height
/* Return successfully if and only if `convert` did. */
return status == 0 ? 0 : -1;
+
+
+ /* `convert` shall not be used: */
+
+ no_convert:
+ /* Open output file. */
+ if (fd = open(imgpath, O_WRONLY | O_CREAT | O_TRUNC,
+ S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH), fd == -1)
+ return -1;
+
+ /* Save image. */
+ if (save_pnm(fbpath, width, height, fd) < 0)
+ return saved_errno = errno, close(fd), errno = saved_errno, -1;
+
+ return 0;
}
@@ -443,10 +463,13 @@ static int exec_image(char* flatten_args)
* Take a screenshot of a framebuffer
*
* @param fbno The number of the framebuffer
+ * @param raw Save in PNM rather than in PNG?
* @param filepattern The pattern for the filename, `NULL` for default
+ * @param execpattern The pattern for the command to run to
+ * process the image, `NULL` for default
* @return Zero on success, -1 on error, 1 if the framebuffer does not exist
*/
-static int save_fb(int fbno, const char* filepattern, const char* execpattern)
+static int save_fb(int fbno, int raw, const char* filepattern, const char* execpattern)
{
char fbpath[PATH_MAX];
char imgpath[PATH_MAX];
@@ -468,11 +491,11 @@ static int save_fb(int fbno, const char* filepattern, const char* execpattern)
/* Get output pathname. */
if (filepattern == NULL)
{
- sprintf(imgpath, "fb%i.png", fbno);
+ sprintf(imgpath, "fb%i.%s", fbno, (raw ? "pnm" : "png"));
if (access(imgpath, F_OK) == 0)
for (i = 2;; i++)
{
- sprintf(imgpath, "fb%i.png.%i", fbno, i);
+ sprintf(imgpath, "fb%i.%s.%i", fbno, (raw ? "pnm" : "png"), i);
if (access(imgpath, F_OK))
break;
}
@@ -528,6 +551,7 @@ static int print_help(void)
p("\t--help Print usage information.\n");
p("\t--version Print program name and version.\n");
p("\t--copyright Print copyright information.\n");
+ p("\t--raw Save in PNM rather than in PNG.\n");
p("\t--exec CMD Command to run for each saved image.\n");
p("\n");
p("SPECIAL STRINGS\n");
@@ -578,7 +602,7 @@ static int print_version(void)
static int print_copyright(void)
{
p("scrotty -- Screenshot program for Linux's TTY\n");
- p("Copyright (C) 2014 Mattias Andrée (maandree@member.fsf.org)\n");
+ p("Copyright (C) 2014, 2015 Mattias Andrée (maandree@member.fsf.org)\n");
p("\n");
p("This program is free software: you can redistribute it and/or modify\n");
p("it under the terms of the GNU General Public License as published by\n");
@@ -608,7 +632,8 @@ static int print_copyright(void)
*/
int main(int argc, char* argv[])
{
- int fbno, r, i, dash = argc, exec = -1, help = 0, version = 0, copyright = 0, filepattern = -1;
+ int fbno, r, i, dash = argc, exec = -1, help = 0;
+ int raw = 0, version = 0, copyright = 0, filepattern = -1;
static char convert_args_0[] = "convert";
static char convert_args_1[] = DEVDIR "/stdin";
static char convert_args_2[PATH_MAX];
@@ -621,6 +646,7 @@ int main(int argc, char* argv[])
if (!strcmp(argv[i], "--help")) help = 1;
else if (!strcmp(argv[i], "--version")) version = 1;
else if (!strcmp(argv[i], "--copyright")) copyright = 1;
+ else if (!strcmp(argv[i], "--raw")) raw = 1;
else if (!strcmp(argv[i], "--exec")) exec = ++i;
else if (!strcmp(argv[i], "--"))
{
@@ -643,18 +669,23 @@ int main(int argc, char* argv[])
if (copyright) return -(print_copyright());
/* Create arguments for `convert`. */
- convert_args = alloca((size_t)(4 + (argc - dash)) * sizeof(char*));
- convert_args[0] = convert_args_0;
- convert_args[1] = convert_args_1;
- convert_args[2] = convert_args_2;
- for (i = dash; i < argc; i++)
- convert_args[i - dash + 2] = argv[i];
- convert_args[3 + (argc - dash)] = NULL;
+ if ((!raw) || (dash < argc))
+ {
+ convert_args = alloca((size_t)(4 + (argc - dash)) * sizeof(char*));
+ convert_args[0] = convert_args_0;
+ convert_args[1] = convert_args_1;
+ convert_args[2] = convert_args_2;
+ for (i = dash; i < argc; i++)
+ convert_args[i - dash + 2] = argv[i];
+ convert_args[3 + (argc - dash)] = NULL;
+ }
/* Take a screenshot of each framebuffer. */
for (fbno = 0;; fbno++)
{
- r = save_fb(fbno, filepattern < 0 ? NULL : argv[filepattern], exec < 0 ? NULL : argv[exec]);
+ r = save_fb(fbno, raw,
+ (filepattern < 0 ? NULL : argv[filepattern]),
+ (exec < 0 ? NULL : argv[exec]));
if (r < 0) return perror(execname), 1;
if (r > 0) break;
}