X-Git-Url: https://sigrok.org/gitweb/?p=sigrok-cli.git;a=blobdiff_plain;f=session.c;h=f51bb642aad35149972029991210d906ed3e26f8;hp=d3c7e01d3389f0676b47d277836aa109460b6353;hb=cbbea0871ba979786ea00f0cf9ed4abe3cc0b857;hpb=84c250a2ee6152ec0d857f7be142f873fa44e640 diff --git a/session.c b/session.c index d3c7e01..f51bb64 100644 --- a/session.c +++ b/session.c @@ -17,15 +17,14 @@ * along with this program. If not, see . */ -#include "sigrok-cli.h" #include #include #include #include +#include "sigrok-cli.h" static uint64_t limit_samples = 0; static uint64_t limit_frames = 0; -static char *srzip_and_filename = NULL; #ifdef HAVE_SRD extern struct srd_session *srd_sess; @@ -53,7 +52,7 @@ static int set_limit_time(const struct sr_dev_inst *sdi) } } else if (config_key_has_cap(driver, sdi, NULL, SR_CONF_SAMPLERATE, SR_CONF_GET | SR_CONF_SET)) { - /* Convert to samples based on the samplerate. */ + /* Convert to samples based on the samplerate. */ sr_config_get(driver, sdi, NULL, SR_CONF_SAMPLERATE, &gvar); samplerate = g_variant_get_uint64(gvar); g_variant_unref(gvar); @@ -75,23 +74,19 @@ static int set_limit_time(const struct sr_dev_inst *sdi) return SR_OK; } -const struct sr_output *setup_output_format(const struct sr_dev_inst *sdi) +const struct sr_output *setup_output_format(const struct sr_dev_inst *sdi, FILE **outfile) { const struct sr_output_module *omod; const struct sr_option **options; const struct sr_output *o; GHashTable *fmtargs, *fmtopts; - int size; char *fmtspec; if (!opt_output_format) { if (opt_output_file) { - size = strlen(opt_output_file) + 32; - srzip_and_filename = g_malloc(size); - snprintf(srzip_and_filename, size, "srzip:filename=%s", opt_output_file); - opt_output_format = srzip_and_filename; + opt_output_format = DEFAULT_OUTPUT_FORMAT_FILE; } else { - opt_output_format = DEFAULT_OUTPUT_FORMAT; + opt_output_format = DEFAULT_OUTPUT_FORMAT_NOFILE; } } @@ -107,7 +102,17 @@ const struct sr_output *setup_output_format(const struct sr_dev_inst *sdi) sr_output_options_free(options); } else fmtopts = NULL; - o = sr_output_new(omod, fmtopts, sdi); + o = sr_output_new(omod, fmtopts, sdi, opt_output_file); + + if (opt_output_file) { + if (!sr_output_test_flag(omod, SR_OUTPUT_INTERNAL_IO_HANDLING)) + *outfile = g_fopen(opt_output_file, "wb"); + else + *outfile = NULL; + } else { + *outfile = stdout; + } + if (fmtopts) g_hash_table_destroy(fmtopts); g_hash_table_destroy(fmtargs); @@ -171,23 +176,18 @@ void datafeed_in(const struct sr_dev_inst *sdi, driver = sr_dev_inst_driver_get(sdi); /* If the first packet to come in isn't a header, don't even try. */ - if (packet->type != SR_DF_HEADER && o == NULL) + if (packet->type != SR_DF_HEADER && !o) return; session = cb_data; switch (packet->type) { case SR_DF_HEADER: g_debug("cli: Received SR_DF_HEADER."); - if (!(o = setup_output_format(sdi))) + if (!(o = setup_output_format(sdi, &outfile))) g_critical("Failed to initialize output module."); /* Set up backup analog output module. */ - oa = sr_output_new(sr_output_find("analog"), NULL, sdi); - - if (opt_output_file) - outfile = g_fopen(opt_output_file, "wb"); - else - outfile = stdout; + oa = sr_output_new(sr_output_find("analog"), NULL, sdi, NULL); rcvd_samples_logic = rcvd_samples_analog = 0; @@ -271,7 +271,7 @@ void datafeed_in(const struct sr_dev_inst *sdi, if (opt_pds) { #ifdef HAVE_SRD if (srd_session_send(srd_sess, rcvd_samples_logic, end_sample, - logic->data, input_len) != SRD_OK) + logic->data, input_len, logic->unitsize) != SRD_OK) sr_session_stop(session); #endif } @@ -303,16 +303,18 @@ void datafeed_in(const struct sr_dev_inst *sdi, break; } - if (o && outfile && !opt_pds) { + if (o && !opt_pds) { if (sr_output_send(o, packet, &out) == SR_OK) { if (!out || (out->len == 0 && !opt_output_format && packet->type == SR_DF_ANALOG)) { - /* The user didn't specify an output module, - * but needs to see this analog data. */ + /* + * The user didn't specify an output module, + * but needs to see this analog data. + */ sr_output_send(oa, packet, &out); } - if (out && out->len > 0) { + if (outfile && out && out->len > 0) { fwrite(out->str, 1, out->len, outfile); fflush(outfile); } @@ -321,16 +323,15 @@ void datafeed_in(const struct sr_dev_inst *sdi, } } - /* SR_DF_END needs to be handled after the output module's receive() - * is called, so it can properly clean up that module. */ + /* + * SR_DF_END needs to be handled after the output module's receive() + * is called, so it can properly clean up that module. + */ if (packet->type == SR_DF_END) { g_debug("cli: Received SR_DF_END."); - if (o) { + if (o) sr_output_free(o); - if (srzip_and_filename) - g_free(srzip_and_filename); - } o = NULL; sr_output_free(oa); @@ -368,7 +369,7 @@ int opt_to_gvar(char *key, char *value, struct sr_config *src) } src->key = srci->key; - if ((value == NULL || strlen(value) == 0) && + if ((!value || strlen(value) == 0) && (srci->datatype != SR_T_BOOL)) { g_critical("Option '%s' needs a value.", (char *)key); return -1; @@ -478,7 +479,8 @@ int set_dev_options(struct sr_dev_inst *sdi, GHashTable *args) cg = select_channel_group(sdi); if ((ret = maybe_config_set(sr_dev_inst_driver_get(sdi), sdi, cg, src.key, src.data)) != SR_OK) { - g_critical("Failed to set device option '%s'.", (char *)key); + g_critical("Failed to set device option '%s': %s.", + (char *)key, sr_strerror(ret)); return ret; } } @@ -548,7 +550,7 @@ void run_session(void) g_slist_free(devices); g_slist_free(real_devices); - sr_session_new(&session); + sr_session_new(sr_ctx, &session); sr_session_datafeed_callback_add(session, datafeed_in, NULL); if (sr_dev_open(sdi) != SR_OK) { @@ -610,8 +612,10 @@ void run_session(void) } if (maybe_config_list(driver, sdi, NULL, SR_CONF_LIMIT_SAMPLES, &gvar) == SR_OK) { - /* The device has no compression, or compression is turned - * off, and publishes its sample memory size. */ + /* + * The device has no compression, or compression is turned + * off, and publishes its sample memory size. + */ g_variant_get(gvar, "(tt)", &min_samples, &max_samples); g_variant_unref(gvar); if (limit_samples < min_samples) { @@ -666,4 +670,3 @@ void run_session(void) sr_session_destroy(session); } -