Add -T|--transform-module and show transform modules in -V output.
[sigrok-cli.git] / main.c
diff --git a/main.c b/main.c
index 9ab3f77cc49baa64dae8e72e1a44ab19b841fb0d..1d3b33df800f1765f6247a7acafe33c603dd94c0 100644 (file)
--- a/main.c
+++ b/main.c
@@ -18,7 +18,7 @@
  */
 
 #include "sigrok-cli.h"
-#include "config.h"
+#include <stdlib.h>
 #include <glib.h>
 
 struct sr_context *sr_ctx = NULL;
@@ -26,87 +26,6 @@ struct sr_context *sr_ctx = NULL;
 struct srd_session *srd_sess = NULL;
 #endif
 
-static gboolean opt_version = FALSE;
-gint opt_loglevel = SR_LOG_WARN; /* Show errors+warnings by default. */
-static gboolean opt_scan_devs = FALSE;
-gboolean opt_wait_trigger = FALSE;
-gchar *opt_input_file = NULL;
-gchar *opt_output_file = NULL;
-gchar *opt_drv = NULL;
-gchar *opt_config = NULL;
-static gchar *opt_probes = NULL;
-gchar *opt_probe_group = NULL;
-gchar *opt_triggers = NULL;
-gchar *opt_pds = NULL;
-#ifdef HAVE_SRD
-static gchar *opt_pd_stack = NULL;
-static gchar *opt_pd_annotations = NULL;
-static gchar *opt_pd_meta = NULL;
-static gchar *opt_pd_binary = NULL;
-#endif
-gchar *opt_input_format = NULL;
-gchar *opt_output_format = NULL;
-static gchar *opt_show = NULL;
-gchar *opt_time = NULL;
-gchar *opt_samples = NULL;
-gchar *opt_frames = NULL;
-gchar *opt_continuous = NULL;
-static gchar *opt_set = NULL;
-
-static GOptionEntry optargs[] = {
-       {"version", 'V', 0, G_OPTION_ARG_NONE, &opt_version,
-                       "Show version and support list", NULL},
-       {"loglevel", 'l', 0, G_OPTION_ARG_INT, &opt_loglevel,
-                       "Set loglevel (5 is most verbose)", NULL},
-       {"driver", 'd', 0, G_OPTION_ARG_STRING, &opt_drv,
-                       "The driver to use", NULL},
-       {"config", 'c', 0, G_OPTION_ARG_STRING, &opt_config,
-                       "Specify device configuration options", NULL},
-       {"input-file", 'i', 0, G_OPTION_ARG_FILENAME, &opt_input_file,
-                       "Load input from file", NULL},
-       {"input-format", 'I', 0, G_OPTION_ARG_STRING, &opt_input_format,
-                       "Input format", NULL},
-       {"output-file", 'o', 0, G_OPTION_ARG_FILENAME, &opt_output_file,
-                       "Save output to file", NULL},
-       {"output-format", 'O', 0, G_OPTION_ARG_STRING, &opt_output_format,
-                       "Output format", NULL},
-       {"probes", 'p', 0, G_OPTION_ARG_STRING, &opt_probes,
-                       "Probes to use", NULL},
-       {"probe-group", 'g', 0, G_OPTION_ARG_STRING, &opt_probe_group,
-                       "Probe groups", NULL},
-       {"triggers", 't', 0, G_OPTION_ARG_STRING, &opt_triggers,
-                       "Trigger configuration", NULL},
-       {"wait-trigger", 'w', 0, G_OPTION_ARG_NONE, &opt_wait_trigger,
-                       "Wait for trigger", NULL},
-#ifdef HAVE_SRD
-       {"protocol-decoders", 'P', 0, G_OPTION_ARG_STRING, &opt_pds,
-                       "Protocol decoders to run", NULL},
-       {"protocol-decoder-stack", 'S', 0, G_OPTION_ARG_STRING, &opt_pd_stack,
-                       "Protocol decoder stack", NULL},
-       {"protocol-decoder-annotations", 'A', 0, G_OPTION_ARG_STRING, &opt_pd_annotations,
-                       "Protocol decoder annotation(s) to show", NULL},
-       {"protocol-decoder-meta", 'M', 0, G_OPTION_ARG_STRING, &opt_pd_meta,
-                       "Protocol decoder meta output to show", NULL},
-       {"protocol-decoder-binary", 'B', 0, G_OPTION_ARG_STRING, &opt_pd_binary,
-                       "Protocol decoder binary output to show", NULL},
-#endif
-       {"scan", 0, 0, G_OPTION_ARG_NONE, &opt_scan_devs,
-                       "Scan for devices", NULL},
-       {"show", 0, 0, G_OPTION_ARG_NONE, &opt_show,
-                       "Show device detail", NULL},
-       {"time", 0, 0, G_OPTION_ARG_STRING, &opt_time,
-                       "How long to sample (ms)", NULL},
-       {"samples", 0, 0, G_OPTION_ARG_STRING, &opt_samples,
-                       "Number of samples to acquire", NULL},
-       {"frames", 0, 0, G_OPTION_ARG_STRING, &opt_frames,
-                       "Number of frames to acquire", NULL},
-       {"continuous", 0, 0, G_OPTION_ARG_NONE, &opt_continuous,
-                       "Sample continuously", NULL},
-       {"set", 0, 0, G_OPTION_ARG_NONE, &opt_set, "Set device options only", NULL},
-       {NULL, 0, 0, 0, NULL, NULL, NULL}
-};
-
-
 static void logger(const gchar *log_domain, GLogLevelFlags log_level,
                   const gchar *message, gpointer cb_data)
 {
@@ -128,29 +47,133 @@ static void logger(const gchar *log_domain, GLogLevelFlags log_level,
 
 }
 
-int select_probes(struct sr_dev_inst *sdi)
+int select_channels(struct sr_dev_inst *sdi)
 {
-       struct sr_probe *probe;
-       GSList *selected_probes, *l;
+       struct sr_channel *ch;
+       GSList *selected_channels, *l, *channels;
 
-       if (!opt_probes)
-               return SR_OK;
+       channels = sr_dev_inst_channels_get(sdi);
 
-       if (!(selected_probes = parse_probestring(sdi, opt_probes)))
-               return SR_ERR;
+       if (opt_channels) {
+               if (!(selected_channels = parse_channelstring(sdi, opt_channels)))
+                       return SR_ERR;
 
-       for (l = sdi->probes; l; l = l->next) {
-               probe = l->data;
-               if (g_slist_find(selected_probes, probe))
-                       probe->enabled = TRUE;
-               else
-                       probe->enabled = FALSE;
+               for (l = channels; l; l = l->next) {
+                       ch = l->data;
+                       if (g_slist_find(selected_channels, ch))
+                               ch->enabled = TRUE;
+                       else
+                               ch->enabled = FALSE;
+               }
+               g_slist_free(selected_channels);
        }
-       g_slist_free(selected_probes);
-
+#ifdef HAVE_SRD
+       map_pd_channels(sdi);
+#endif
        return SR_OK;
 }
 
+gboolean config_key_has_cap(struct sr_dev_driver *driver,
+               const struct sr_dev_inst *sdi, struct sr_channel_group *cg,
+               uint32_t key, uint32_t capability)
+{
+       GVariant *gvar_opts;
+       const uint32_t *opts;
+       gsize num_opts, i;
+
+       if (sr_config_list(driver, sdi, cg, SR_CONF_DEVICE_OPTIONS,
+                       &gvar_opts) != SR_OK)
+               return FALSE;
+
+       opts = g_variant_get_fixed_array(gvar_opts, &num_opts, sizeof(uint32_t));
+       for (i = 0; i < num_opts; i++) {
+               if ((opts[i] & SR_CONF_MASK) == key) {
+                       if ((opts[i] & capability) == capability)
+                               return TRUE;
+                       else
+                               return FALSE;
+               }
+       }
+
+       return FALSE;
+}
+
+int maybe_config_get(struct sr_dev_driver *driver,
+               const struct sr_dev_inst *sdi, struct sr_channel_group *cg,
+               uint32_t key, GVariant **gvar)
+{
+       if (config_key_has_cap(driver, sdi, cg, key, SR_CONF_GET))
+               return sr_config_get(driver, sdi, cg, key, gvar);
+
+       return SR_ERR_NA;
+}
+
+int maybe_config_set(struct sr_dev_driver *driver,
+               const struct sr_dev_inst *sdi, struct sr_channel_group *cg,
+               uint32_t key, GVariant *gvar)
+{
+       if (config_key_has_cap(driver, sdi, cg, key, SR_CONF_SET))
+               return sr_config_set(sdi, cg, key, gvar);
+
+       return SR_ERR_NA;
+}
+
+int maybe_config_list(struct sr_dev_driver *driver,
+               const struct sr_dev_inst *sdi, struct sr_channel_group *cg,
+               uint32_t key, GVariant **gvar)
+{
+       if (config_key_has_cap(driver, sdi, cg, key, SR_CONF_LIST))
+               return sr_config_list(driver, sdi, cg, key, gvar);
+
+       return SR_ERR_NA;
+}
+
+static void get_option(void)
+{
+       struct sr_dev_inst *sdi;
+       struct sr_channel_group *cg;
+       const struct sr_config_info *ci;
+       GSList *devices;
+       GVariant *gvar;
+       GHashTable *devargs;
+       int ret;
+       char *s;
+       struct sr_dev_driver *driver;
+
+       if (!(devices = device_scan())) {
+               g_critical("No devices found.");
+               return;
+       }
+       sdi = devices->data;
+       g_slist_free(devices);
+
+       driver = sr_dev_inst_driver_get(sdi);
+
+       if (sr_dev_open(sdi) != SR_OK) {
+               g_critical("Failed to open device.");
+               return;
+       }
+
+       cg = select_channel_group(sdi);
+       if (!(ci = sr_config_info_name_get(opt_get)))
+               g_critical("Unknown option '%s'", opt_get);
+
+       if ((devargs = parse_generic_arg(opt_config, FALSE)))
+               set_dev_options(sdi, devargs);
+       else devargs = NULL;
+
+       if ((ret = maybe_config_get(driver, sdi, cg, ci->key, &gvar)) != SR_OK)
+               g_critical("Failed to get '%s': %s", opt_get, sr_strerror(ret));
+       s = g_variant_print(gvar, FALSE);
+       printf("%s\n", s);
+       g_free(s);
+
+       g_variant_unref(gvar);
+       sr_dev_close(sdi);
+       if (devargs)
+               g_hash_table_destroy(devargs);
+}
+
 static void set_options(void)
 {
        struct sr_dev_inst *sdi;
@@ -170,6 +193,7 @@ static void set_options(void)
                return;
        }
        sdi = devices->data;
+       g_slist_free(devices);
 
        if (sr_dev_open(sdi) != SR_OK) {
                g_critical("Failed to open device.");
@@ -179,28 +203,16 @@ static void set_options(void)
        set_dev_options(sdi, devargs);
 
        sr_dev_close(sdi);
-       g_slist_free(devices);
        g_hash_table_destroy(devargs);
 
 }
 
 int main(int argc, char **argv)
 {
-       GOptionContext *context;
-       GError *error;
-       int ret;
-       char *help;
-
        g_log_set_default_handler(logger, NULL);
 
-       context = g_option_context_new(NULL);
-       g_option_context_add_main_entries(context, optargs, NULL);
-
-       ret = 1;
-       error = NULL;
-       if (!g_option_context_parse(context, &argc, &argv, &error)) {
-               g_critical("%s", error->message);
-               goto done;
+       if (parse_options(argc, argv)) {
+               return 1;
        }
 
        /* Set the loglevel (amount of messages to output) for libsigrok. */
@@ -251,11 +263,14 @@ int main(int argc, char **argv)
        }
 #endif
 
-       if (setup_output_format() != 0)
-               goto done;
-
        if (opt_version)
                show_version();
+       else if (opt_input_format && opt_show)
+               show_input();
+       else if (opt_output_format && opt_show)
+               show_output();
+       else if (opt_transform_module && opt_show)
+               show_transform();
        else if (opt_scan_devs)
                show_dev_list();
 #ifdef HAVE_SRD
@@ -266,28 +281,23 @@ int main(int argc, char **argv)
                show_dev_detail();
        else if (opt_input_file)
                load_input_file();
+       else if (opt_get)
+               get_option();
        else if (opt_set)
                set_options();
        else if (opt_samples || opt_time || opt_frames || opt_continuous)
                run_session();
-       else {
-               help = g_option_context_get_help(context, TRUE, NULL);
-               printf("%s", help);
-               g_free(help);
-       }
+       else
+               show_help();
 
 #ifdef HAVE_SRD
        if (opt_pds)
                srd_exit();
 #endif
 
-       ret = 0;
-
 done:
        if (sr_ctx)
                sr_exit(sr_ctx);
 
-       g_option_context_free(context);
-
-       return ret;
+       return 0;
 }