From 1f4fe6e731d05e55238c1e9805101f43cb9edebf Mon Sep 17 00:00:00 2001 From: "Suren A. Chilingaryan" Date: Thu, 8 Mar 2012 02:59:46 +0100 Subject: getopt support --- fwbench.c | 258 +++++++++++++++++++++++++++++++++++++++++++++++++++++--------- 1 file changed, 222 insertions(+), 36 deletions(-) diff --git a/fwbench.c b/fwbench.c index b8051d0..ce96d91 100644 --- a/fwbench.c +++ b/fwbench.c @@ -9,6 +9,8 @@ #include #include #include +#include +#include #include @@ -19,11 +21,9 @@ #include "config.h" //#define USE_FIFO -#define FW_BUFFER 4096l #define WRITE_INTERVAL 1 #define WRITE_SUMMARY 5 #define DELTA_TOLERANCE 5 -#define NEWFILE 10 #ifdef USE_UFO_GENERATOR # ifndef USE_FIFO @@ -31,7 +31,6 @@ # endif /* !USE_FIFO */ #endif /* USE_UFO_GENERATOR */ -#define run_time 3600 const char *fifo_name = ".fifo"; struct setup_s { @@ -40,7 +39,13 @@ struct setup_s { size_t bpp; size_t fps; + size_t fpf; size_t iters; + double tolerance; + const char *output; + const char *fs; + + int verbose; volatile int run_started; @@ -58,11 +63,64 @@ struct setup_s { size_t writeouts; fastwriter_t *fw; + char *current_output; }; typedef struct setup_s setup_t; + +typedef enum { + OPT_OUTPUT = 'o', + OPT_BUFFER = 'b', + OPT_SIZE = 's', + OPT_TIME = 't', + OPT_RATE = 'r', + OPT_FRAMES = 'n', + OPT_GEOMETRY = 'g', + OPT_FPF = 'f', + OPT_MISSING = 'm', + OPT_QUIET = 'q', + OPT_HELP = 'h' +} options_t; + + +void Usage(int argc, char* const argv[], const char *format, ...) { + if (format) { + va_list ap; + + va_start(ap, format); + printf("Error %i: ", errno); + vprintf(format, ap); + printf("\n"); + va_end(ap); + + printf("\n"); + } + + printf( +"Usage:\n" +" %s [options]\n" +" -o - Output to file/device [/dev/null]\n" +" use %%zu to replace with frame number\n" +" -f - Number of frames per file [all]\n" +" -b size - Buffer Size (MB)\n" +" -r rate - Write rate (MB/s)\n" +" -s size - Total size of data to write (GB)\n" +" -t run_time - Run time (s)\n" +" -n frames - Number of frames to write\n" +" -g x - Geometry [1024]\n" +" -g - Number of megapixels [0.7]\n" +" -m - Tolerable missing frames [100%%]\n" +" -q - Quiete\n" +" -h - Help\n" +"\n\n", +argv[0]); + + exit(0); +} + + #ifdef USE_UFO_GENERATOR static void handle_error(GError *error) { if (error != NULL) { @@ -73,6 +131,21 @@ static void handle_error(GError *error) { } #endif /* USE_UFO_GENERATOR */ +static void set_verbosity(setup_t *setup, int level) { + setup->verbose = level; +} + +static void set_current_output(setup_t *setup, size_t frame) { + sprintf(setup->current_output, setup->output, frame); +} + +static void set_output(setup_t *setup, const char *name) { + setup->output = name; + setup->fs = name; + setup->current_output = malloc(strlen(setup->output) + 32); + g_assert(setup->current_output); + set_current_output(setup, 0); +} static void set_dim(setup_t *setup, size_t width, size_t height) { setup->bpp = sizeof(float); @@ -82,18 +155,36 @@ static void set_dim(setup_t *setup, size_t width, size_t height) { setup->frame_size = setup->width * setup->height * setup->bpp; } +static void set_pixels(setup_t *setup, size_t pixels) { + int width = sqrt(4 * pixels / 3); + set_dim(setup, width, width * 3 / 4); +} + static void set_speed(setup_t *setup, size_t speed) { setup->fps = 1 + speed / setup->width / setup->height / setup->bpp; } +static void set_iters(setup_t *setup, size_t iters) { + setup->iters = iters; +} + +static void set_time(setup_t *setup, size_t run_time) { + setup->iters = run_time * setup->fps; +} + static void set_size(setup_t *setup, size_t size) { - setup->iters = size / setup->width / setup->height / setup->bpp; + setup->iters = size / setup->frame_size + (size % setup->frame_size)?1:0; +} + +static void set_frames_per_file(setup_t *setup, size_t fpf) { + setup->fpf = fpf; } -static void set_time(setup_t *setup, size_t time) { - setup->iters = time * setup->fps; +static void set_fail_tolerance(setup_t *setup, double fail_rate) { + setup->tolerance = fail_rate; } + static int callback(setup_t *s, size_t result, void *buffer); static void *run(setup_t *setup) { @@ -133,6 +224,8 @@ static void *run(setup_t *setup) { ufo_graph_run(graph, &error); handle_error(error); + + g_thread_exit(NULL); #else /* USE_UFO_GENERATOR */ size_t i; struct timeval tv; @@ -148,7 +241,6 @@ static void *run(setup_t *setup) { int fd = open(fifo_name, O_WRONLY); g_assert(fd >= 0); #endif /* USE_FIFO */ - gettimeofday(&tv, NULL); nexts = tv.tv_sec; nextus = tv.tv_usec + interval; @@ -177,7 +269,6 @@ static void *run(setup_t *setup) { #endif /* USE_UFO_GENERATOR */ - g_thread_exit(NULL); return NULL; } @@ -191,11 +282,17 @@ static int callback(setup_t *s, size_t result, void *buffer) { size_t duration, last_duration, expected; long delta; + gettimeofday(&tv, NULL); if (!s->broken_frame) { err = fastwriter_push(fw, result, buffer); if (err) { if (err == EWOULDBLOCK) { + if (!s->tolerance) { + if (s->verbose >= 0) printf("Lost frame...\n"); + exit(1); + } + if (s->num_read) fastwriter_cancel(fw); s->broken_frame = 1; } else { @@ -217,9 +314,21 @@ static int callback(setup_t *s, size_t result, void *buffer) { } else { err = fastwriter_commit(fw); s->frames++; + + if ((s->fpf)&&((s->frames%s->fpf) == 0)) { + fastwriter_close(s->fw); + set_current_output(s, s->frames); + err = fastwriter_open(s->fw, s->current_output, FASTWRITER_FLAGS_OVERWRITE); + if (err) { + printf("FastWriter returned error %i\n", err); + g_assert(!err); + } + + // reopen } - gettimeofday(&tv, NULL); + } + if (!s->tv_started.tv_sec) { memcpy(&s->tv_started, &tv, sizeof(struct timeval)); if (s->tv_started.tv_usec >= (1000000 / s->fps)) @@ -235,9 +344,20 @@ static int callback(setup_t *s, size_t result, void *buffer) { if (stats.buffer_used > s->buf_max) s->buf_max = stats.buffer_used; if ((tv.tv_sec - s->tv_last_written.tv_sec) >= WRITE_INTERVAL) { - last_duration = (tv.tv_sec - s->tv_last_written.tv_sec) * 1000000 + (tv.tv_usec - s->tv_last_written.tv_usec); - - printf("Lost %6.2lf%% (% 8lu of % 8lu), %9.3lf GB at %8.3lf MB/s, buf:%6.2lf%%\n", 100.*(s->lost - s->last_lost) / (s->lost + s->frames - s->last_lost - s->last_frames), s->lost - s->last_lost, s->lost + s->frames - (s->last_lost + s->last_frames), 1. * s->frame_size * (s->frames - s->last_frames) / 1024 / 1024 / 1024, 1000000. * s->frame_size * (s->frames - s->last_frames) / last_duration / 1024 / 1024, 100.*s->buf_max/stats.buffer_size); + if ((s->tolerance > 0)&&(s->tolerance < 100)) { + double lost = (100. * s->lost / (s->lost + s->frames)); + double last_lost = (100. * (s->lost - s->last_lost) / (s->lost + s->frames - s->last_lost - s->last_frames)); + if ((lost > s->tolerance)||(last_lost > s->tolerance)) { + if (s->verbose >= 0) + printf("Lost %.2lf%% (%lu) frames, total: %.2lf%% (%lu)\n", last_lost, s->lost - s->last_lost, lost, s->lost); + exit(1); + } + } + + if (s->verbose >= 0) { + last_duration = (tv.tv_sec - s->tv_last_written.tv_sec) * 1000000 + (tv.tv_usec - s->tv_last_written.tv_usec); + printf("Lost %6.2lf%% (% 8lu of % 8lu), %9.3lf GB at %8.3lf MB/s, buf:%6.2lf%%\n", 100.*(s->lost - s->last_lost) / (s->lost + s->frames - s->last_lost - s->last_frames), s->lost - s->last_lost, s->lost + s->frames - (s->last_lost + s->last_frames), 1. * s->frame_size * (s->frames - s->last_frames) / 1024 / 1024 / 1024, 1000000. * s->frame_size * (s->frames - s->last_frames) / last_duration / 1024 / 1024, 100.*s->buf_max/stats.buffer_size); + } if (((++s->writeouts)%WRITE_SUMMARY)==0) { duration = (tv.tv_sec - s->tv_started.tv_sec) * 1000000 + (tv.tv_usec - s->tv_started.tv_usec); @@ -246,7 +366,9 @@ static int callback(setup_t *s, size_t result, void *buffer) { if ((delta > DELTA_TOLERANCE)||(delta < -DELTA_TOLERANCE)) printf(" *** Unexpected frame rate: %.2lf (%lu), delta: %li (%lu, %lu)\n", 1000000. * (s->frames + s->lost) / duration, s->fps, delta, s->lost + s->frames, expected); - printf("Total %6.2lf%% (% 8lu of % 8lu), %9.3lf GB at %8.3lf MB/s, buf:%6.2lf%%\n", 100. * s->lost / (s->lost + s->frames), s->lost, s->lost + s->frames, 1. * s->frames * s->frame_size / 1024 / 1024 / 1024, 1000000. * s->frames * s->frame_size / duration / 1024 / 1024, 100.*stats.buffer_max / stats.buffer_size); + if (s->verbose >= 0) { + printf("Total %6.2lf%% (% 8lu of % 8lu), %9.3lf GB at %8.3lf MB/s, buf:%6.2lf%%\n", 100. * s->lost / (s->lost + s->frames), s->lost, s->lost + s->frames, 1. * s->frames * s->frame_size / 1024 / 1024 / 1024, 1000000. * s->frames * s->frame_size / duration / 1024 / 1024, 100.*stats.buffer_max / stats.buffer_size); + } } s->buf_max = 0; @@ -259,7 +381,7 @@ static int callback(setup_t *s, size_t result, void *buffer) { } -int main(int argc, char const* argv[]) +int main(int argc, char* const argv[]) { int err; @@ -269,12 +391,23 @@ int main(int argc, char const* argv[]) #endif /* USE_FIFO */ setup_t setup; + + unsigned char c; + const char *sptr; const char *out = "/dev/null"; + double mpix = 0; size_t width = 1024; size_t height = 768; size_t speed = 850; + size_t size = 0; + size_t run_time = 0; + size_t iters = 0x7FFFFFFF; + size_t fw_buffer = 0; + double fail_rate = 100.; + size_t fpf = 0; + int quiet = 0; + - fastwriter_t *fw; fastwriter_stats_t stats; g_thread_init(NULL); @@ -283,12 +416,49 @@ int main(int argc, char const* argv[]) g_type_init(); #endif /* USE_UFO_GENERATOR */ - if (argc > 1) { - out = argv[1]; - } - - if (argc > 2) { - speed = atoi(argv[2]); + + + while ((c = getopt(argc, argv, "hqo:s:t:r:n:g:f:m:b:")) != (unsigned char)-1) { + switch (c) { + case OPT_OUTPUT: + out = optarg; + break; + case OPT_BUFFER: + fw_buffer = atol(optarg); + break; + case OPT_SIZE: + size = atol(optarg); + break; + case OPT_TIME: + run_time = atol(optarg); + break; + case OPT_FRAMES: + iters = atol(optarg); + break; + case OPT_RATE: + speed = atoi(optarg); + break; + case OPT_GEOMETRY: + sptr = strchr(optarg, 'x'); + if (sptr) { + width = atol(optarg); + height = atol(sptr + 1); + } else { + mpix = atof(optarg); + } + break; + case OPT_FPF: + fpf = atol(optarg); + break; + case OPT_MISSING: + fail_rate = atof(optarg); + break; + case OPT_QUIET: + quiet = 1; + break; + case OPT_HELP: + Usage(argc, argv, NULL); + } } #ifdef USE_FIFO @@ -296,25 +466,41 @@ int main(int argc, char const* argv[]) g_assert(!mkfifo(fifo_name, S_IWUSR | S_IRUSR)); #endif /* USE_FIFO */ - fw = fastwriter_init(out, FASTWRITER_FLAGS_OVERWRITE); - g_assert(fw); + + memset(&setup, 0, sizeof(setup_t)); + + if (quiet) set_verbosity(&setup, -1); + set_output(&setup, out); - err = fastwriter_open(fw, out, FASTWRITER_FLAGS_OVERWRITE); - if (err) printf("FastWriter returned error %i\n", err); - g_assert(!err); + if (mpix) set_pixels(&setup, mpix * 1000000); + else set_dim(&setup, width, height); + + set_speed(&setup, speed * 1024 * 1024); - fastwriter_set_buffer_size(fw, FW_BUFFER * 1024 * 1024); + if (size) set_size(&setup, size * 1024 * 1024 * 1024); + else if (run_time) set_time(&setup, run_time); + else set_iters(&setup, iters); - fastwriter_get_stats(fw, &stats); + set_fail_tolerance(&setup, fail_rate); + set_frames_per_file(&setup, fpf); - memset(&setup, 0, sizeof(setup_t)); - setup.fw = fw; - set_dim(&setup, width, height); - set_speed(&setup, speed * 1024 * 1024); - set_size(&setup, run_time * speed * 1024 * 1024); + setup.fw = fastwriter_init(setup.fs, FASTWRITER_FLAGS_OVERWRITE); + g_assert(setup.fw); + + if (fw_buffer) + fastwriter_set_buffer_size(setup.fw, fw_buffer * 1024 * 1024); + else + fastwriter_set_buffer_size(setup.fw, FASTWRITER_BUFFER_MAX); + + err = fastwriter_open(setup.fw, setup.current_output, FASTWRITER_FLAGS_OVERWRITE); + if (err) printf("FastWriter returned error %i\n", err); + g_assert(!err); + - printf("*** Writing to %s, rate: %lu, data: %lu MB/s, buffer: %lu MB\n", out, setup.fps, speed, stats.buffer_size); + fastwriter_get_stats(setup.fw, &stats); + if (!quiet) + printf("*** Writing to %s, rate: %lu, data: %lu MB/s, buffer: %lu MB\n", out, setup.fps, speed, stats.buffer_size/1024/1024); void *buffer = malloc(setup.frame_size); g_assert(buffer); @@ -345,8 +531,8 @@ int main(int argc, char const* argv[]) #endif /* USE_FIFO */ free(buffer); - fastwriter_close(fw); - fastwriter_destroy(fw); + fastwriter_close(setup.fw); + fastwriter_destroy(setup.fw); #ifdef USE_FIFO unlink(fifo_name); -- cgit v1.2.1