summaryrefslogtreecommitdiffstats
path: root/bin/tools
diff options
context:
space:
mode:
authorMatthias Vogelgesang <matthias.vogelgesang@kit.edu>2017-10-11 10:37:13 +0200
committerMatthias Vogelgesang <matthias.vogelgesang@kit.edu>2017-10-11 10:37:13 +0200
commitc437fcf96e480babe1566f5b0e686e9c915627c6 (patch)
treedfae71f57bfefb7d1e07e717af48f9b487e17ba8 /bin/tools
parentb6bdb5cff6d8e3125fd3c12813b73cd8e34c76e4 (diff)
downloaduca-c437fcf96e480babe1566f5b0e686e9c915627c6.tar.gz
uca-c437fcf96e480babe1566f5b0e686e9c915627c6.tar.bz2
uca-c437fcf96e480babe1566f5b0e686e9c915627c6.tar.xz
uca-c437fcf96e480babe1566f5b0e686e9c915627c6.zip
grab: allow filename to be a format specifier
Diffstat (limited to 'bin/tools')
-rw-r--r--bin/tools/grab.c60
1 files changed, 52 insertions, 8 deletions
diff --git a/bin/tools/grab.c b/bin/tools/grab.c
index 1115d26..f9958ae 100644
--- a/bin/tools/grab.c
+++ b/bin/tools/grab.c
@@ -46,6 +46,27 @@ get_bytes_per_pixel (guint bits_per_pixel)
return bits_per_pixel > 8 ? 2 : 1;
}
+static guint
+count_format_specifiers (const gchar *template)
+{
+ GRegex *regex;
+ GMatchInfo *info;
+ gint count = 0;
+
+ regex = g_regex_new ("%[0-9]+[ui]", 0, 0, NULL);
+ g_regex_match (regex, template, 0, &info);
+
+ while (g_match_info_matches (info)) {
+ count++;
+ g_match_info_next (info, NULL);
+ }
+
+ g_match_info_free (info);
+ g_regex_unref (regex);
+
+ return count > 0 ? (guint) count : 0;
+}
+
#ifdef HAVE_LIBTIFF
static void
write_tiff (UcaRingBuffer *buffer,
@@ -60,6 +81,9 @@ write_tiff (UcaRingBuffer *buffer,
guint bits_per_sample;
gsize bytes_per_pixel;
+ if (count_format_specifiers (opts->filename) > 0)
+ g_warning ("Can only write multi-page TIFF, format specifier is ignored.\n");
+
tif = TIFFOpen (opts->filename, "w");
n_frames = uca_ring_buffer_get_num_blocks (buffer);
rows_per_strip = TIFFDefaultStripSize (tif, (guint32) - 1);
@@ -100,23 +124,43 @@ write_raw (UcaRingBuffer *buffer,
{
guint n_frames;
gsize size;
+ guint num_format_specifiers;
+ gboolean multiple_files;
+ FILE *fp;
size = uca_ring_buffer_get_block_size (buffer);
n_frames = uca_ring_buffer_get_num_blocks (buffer);
+ num_format_specifiers = count_format_specifiers (opts->filename);
+
+ if (num_format_specifiers > 1) {
+ g_printerr ("Can only use zero or one format specifiers. Aborting write.\n");
+ return;
+ }
+
+ multiple_files = num_format_specifiers == 1;
+
+ if (!multiple_files)
+ fp = fopen (opts->filename, "wb");
for (gint i = 0; i < n_frames; i++) {
- FILE *fp;
- gchar *filename;
gpointer data;
- filename = g_strdup_printf ("%s-%08i.raw", opts->filename, i);
- fp = fopen(filename, "wb");
- data = uca_ring_buffer_get_read_pointer (buffer);
+ if (multiple_files) {
+ gchar *filename;
+ filename = g_strdup_printf (opts->filename, i);
+ fp = fopen (filename, "wb");
+ g_free (filename);
+ }
+ data = uca_ring_buffer_get_read_pointer (buffer);
fwrite (data, size, 1, fp);
- fclose (fp);
- g_free (filename);
+
+ if (multiple_files)
+ fclose (fp);
}
+
+ if (!multiple_files)
+ fclose (fp);
}
static GError *
@@ -218,7 +262,7 @@ main (int argc, char *argv[])
static GOptionEntry entries[] = {
{ "num-frames", 'n', 0, G_OPTION_ARG_INT, &opts.n_frames, "Number of frames to acquire", "N" },
- { "output", 'o', 0, G_OPTION_ARG_STRING, &opts.filename, "Output file name", "FILE" },
+ { "output", 'o', 0, G_OPTION_ARG_STRING, &opts.filename, "Output file name template", "FILE" },
#ifdef HAVE_LIBTIFF
{ "write-tiff", 't', 0, G_OPTION_ARG_NONE, &opts.write_tiff, "Write as TIFF", NULL },
#endif