1 files changed, 136 insertions, 102 deletions

diff --git a/tools/grab.c b/tools/grab.c
index e8e91be..bbe9269 100644
--- a/tools/grab.c
+++ b/tools/grab.c
@@ -21,140 +21,174 @@
 #include <stdlib.h>
 #include "uca-plugin-manager.h"
 #include "uca-camera.h"
+#include "ring-buffer.h"
 
-static UcaCamera *camera = NULL;
 
-static void sigint_handler(int signal)
-{
-    printf("Closing down libuca\n");
-    uca_camera_stop_recording(camera, NULL);
-    g_object_unref(camera);
-    exit(signal);
-}
+typedef struct {
+    gint n_frames;
+    gdouble duration;
+} Options;
 
-static void
-print_usage (void)
+
+static gchar *
+get_camera_list (void)
 {
     GList *types;
+    GString *str;
     UcaPluginManager *manager;
 
     manager = uca_plugin_manager_new ();
-    g_print ("Usage: benchmark [ ");
     types = uca_plugin_manager_get_available_cameras (manager);
+    str = g_string_new ("[ ");
+
+    if (types != NULL) {
+        for (GList *it = g_list_first (types); it != NULL; it = g_list_next (it)) {
+            gchar *name = (gchar *) it->data;
 
-    if (types == NULL) {
-        g_print ("] -- no camera plugin found\n");
-        return;
+            if (g_list_next (it) == NULL)
+                g_string_append_printf (str, "%s ]", name);
+            else
+                g_string_append_printf (str, "%s, ", name);
+        }
     }
+    else {
+        g_string_append (str, "]");
+    }
+
+    g_object_unref (manager);
+    return g_string_free (str, FALSE);
+}
 
-    for (GList *it = g_list_first (types); it != NULL; it = g_list_next (it)) {
-        gchar *name = (gchar *) it->data;
-        if (g_list_next (it) == NULL)
-            g_print ("%s ]\n", name);
-        else
-            g_print ("%s, ", name);
+static void
+store_frames (RingBuffer *buffer)
+{
+    guint n_frames;
+    gsize size;
+
+    size = ring_buffer_get_block_size (buffer);
+    n_frames = ring_buffer_get_num_blocks (buffer);
+
+    for (gint i = 0; i < n_frames; i++) {
+        FILE *fp;
+        gchar *filename;
+        gpointer data;
+
+        filename = g_strdup_printf ("frame-%08i.raw", i);
+        fp = fopen(filename, "wb");
+        data = ring_buffer_get_pointer (buffer, i);
+
+        fwrite (data, size, 1, fp);
+        fclose (fp);
+        g_free (filename);
     }
 }
 
-int main(int argc, char *argv[])
+static GError *
+record_frames (UcaCamera *camera, Options *opts)
 {
+    guint roi_width;
+    guint roi_height;
+    guint bits;
+    guint pixel_size;
+    gsize size;
+    gint n_frames;
+    guint n_allocated;
+    GTimer *timer;
+    RingBuffer *buffer;
+    GError *error = NULL;
+
+    g_object_get (G_OBJECT (camera),
+                  "roi-width", &roi_width,
+                  "roi-height", &roi_height,
+                  "sensor-bitdepth", &bits,
+                  NULL);
+
+    pixel_size = bits == 8 ? 1 : 2;
+    size = roi_width * roi_height * pixel_size;
+    n_allocated = opts->n_frames > 0 ? opts->n_frames : 256;
+    buffer = ring_buffer_new (size, n_allocated);
+    timer = g_timer_new();
+
+    g_print("Start recording: %ix%i at %i bits/pixel\n",
+            roi_width, roi_height, bits);
+
+    uca_camera_start_recording(camera, &error);
+
+    if (error != NULL)
+        return error;
+
+    n_frames = 0;
+    g_timer_start(timer);
+
+    while (n_frames < opts->n_frames || g_timer_elapsed (timer, NULL) < opts->duration) {
+        uca_camera_grab (camera, ring_buffer_get_current_pointer (buffer), &error);
+        ring_buffer_proceed (buffer);
+
+        if (error != NULL)
+            return error;
+
+        n_frames++;
+    }
+
+    g_print ("Stop recording\n");
+    uca_camera_stop_recording (camera, &error);
+    store_frames (buffer);
+    ring_buffer_free (buffer);
+    g_timer_destroy (timer);
+
+    return error;
+}
+
+int
+main (int argc, char *argv[])
+{
+    GOptionContext *context;
     UcaPluginManager *manager;
+    UcaCamera *camera;
+    gchar *cam_list;
     GError *error = NULL;
-    (void) signal(SIGINT, sigint_handler);
 
-    guint sensor_width, sensor_height;
-    guint roi_width, roi_height, roi_x, roi_y, roi_width_multiplier, roi_height_multiplier;
-    guint bits;
-    gchar *name;
+    static Options opts = {
+        .n_frames = -1,
+        .duration = -1.0
+    };
+
+    static GOptionEntry entries[] = {
+        { "num-frames", 'n', 0, G_OPTION_ARG_INT, &opts.n_frames, "Number of frames to acquire", "N" },
+        { "time", 't', 0, G_OPTION_ARG_DOUBLE, &opts.duration, "Duration in seconds", NULL },
+        { NULL }
+    };
 
     g_type_init();
 
+    cam_list = get_camera_list ();
+    context = g_option_context_new (cam_list);
+    g_option_context_add_main_entries (context, entries, NULL);
+    g_free (cam_list);
+
+    if (!g_option_context_parse (context, &argc, &argv, &error)) {
+        g_print ("Failed parsing arguments: %s\n", error->message);
+        exit (1);
+    }
+
     if (argc < 2) {
-        print_usage();
-        return 1;
+        g_print ("%s\n", g_option_context_get_help (context, TRUE, NULL));
+        exit (0);
     }
 
     manager = uca_plugin_manager_new ();
     camera = uca_plugin_manager_get_camera (manager, argv[1], &error, NULL);
 
     if (camera == NULL) {
-        g_print("Error during initialization: %s\n", error->message);
-        return 1;
-    }
-
-    g_object_get(G_OBJECT(camera),
-            "sensor-width", &sensor_width,
-            "sensor-height", &sensor_height,
-            "name", &name,
-            NULL);
-
-    g_object_set(G_OBJECT(camera),
-            "exposure-time", 0.001,
-            "roi-x0", 0,
-            "roi-y0", 0,
-            "roi-width", 1000,
-            "roi-height", sensor_height,
-            NULL);
-
-    g_object_get(G_OBJECT(camera),
-            "roi-width", &roi_width,
-            "roi-height", &roi_height,
-            "roi-width-multiplier", &roi_width_multiplier,
-            "roi-height-multiplier", &roi_height_multiplier,
-            "roi-x0", &roi_x,
-            "roi-y0", &roi_y,
-            "sensor-bitdepth", &bits,
-            NULL);
-
-    g_print("Camera: %s\n", name);
-    g_free(name);
-
-    g_print("Sensor: %ix%i px\n", sensor_width, sensor_height);
-    g_print("ROI: %ix%i @ (%i, %i), steps: %i, %i\n",
-            roi_width, roi_height, roi_x, roi_y, roi_width_multiplier, roi_height_multiplier);
-
-    const int pixel_size = bits == 8 ? 1 : 2;
-    gpointer buffer = g_malloc0(roi_width * roi_height * pixel_size);
-    gchar filename[FILENAME_MAX];
-    GTimer *timer = g_timer_new();
-
-    for (int i = 0; i < 1; i++) {
-        gint counter = 0;
-        g_print("Start recording\n");
-        uca_camera_start_recording(camera, &error);
-        g_assert_no_error(error);
-
-        while (counter < 5) {
-            g_print(" grab frame ... ");
-            g_timer_start(timer);
-            uca_camera_grab(camera, buffer, &error);
-
-            if (error != NULL) {
-                g_print("\nError: %s\n", error->message);
-                goto cleanup;
-            }
-
-            g_timer_stop(timer);
-            g_print("done (took %3.5fs)\n", g_timer_elapsed(timer, NULL));
-
-            snprintf(filename, FILENAME_MAX, "frame-%08i.raw", counter++);
-            FILE *fp = fopen(filename, "wb");
-            fwrite(buffer, roi_width * roi_height, pixel_size, fp);
-            fclose(fp);
-            g_usleep(2 * G_USEC_PER_SEC);
-        }
-
-        g_print("Stop recording\n");
-        uca_camera_stop_recording(camera, &error);
-        g_assert_no_error(error);
+        g_print ("Error during initialization: %s\n", error->message);
+        exit (1);
     }
 
-    g_timer_destroy(timer);
+    error = record_frames (camera, &opts);
 
-cleanup:
-    uca_camera_stop_recording(camera, NULL);
-    g_object_unref(camera);
-    g_free(buffer);
+    if (error != NULL)
+        g_print ("Error: %s\n", error->message);
 
+    g_object_unref (camera);
     return error != NULL ? 1 : 0;
 }