Build: Include <config.h> first in all source files
[sigrok-cli.git] / decode.c
index 8246eeee34e70eabf311baabaf001a28626d89cd..67d8f1ef33c196310bfa24ee1a12f7663e2bbeeb 100644 (file)
--- a/decode.c
+++ b/decode.c
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-#include "sigrok-cli.h"
-#include "config.h"
+#include <config.h>
+#include <stdlib.h>
+#include <string.h>
 #include <glib.h>
+#include "sigrok-cli.h"
 
 #ifdef HAVE_SRD
 static GHashTable *pd_ann_visible = NULL;
 static GHashTable *pd_meta_visible = NULL;
 static GHashTable *pd_binary_visible = NULL;
+static GHashTable *pd_channel_maps = NULL;
 
 extern struct srd_session *srd_sess;
-extern gint opt_loglevel;
-
 
 static int opts_to_gvar(struct srd_decoder *dec, GHashTable *hash,
                GHashTable **options)
@@ -74,45 +75,42 @@ static int opts_to_gvar(struct srd_decoder *dec, GHashTable *hash,
        return ret;
 }
 
-static int probes_to_gvar(struct srd_decoder *dec, GHashTable *hash,
-               GHashTable **probes)
+static int move_hash_element(GHashTable *src, GHashTable *dest, void *key)
 {
-       struct srd_probe *p;
-       GSList *all_probes, *l;
-       GVariant *gvar;
-       gint32 val_int;
-       int ret;
-       char *val_str, *conv;
+       void *orig_key, *value;
 
-       ret = TRUE;
-       *probes = g_hash_table_new_full(g_str_hash, g_str_equal, g_free,
-                       (GDestroyNotify)g_variant_unref);
+       if (!g_hash_table_lookup_extended(src, key, &orig_key, &value))
+               /* Not specified. */
+               return FALSE;
+       g_hash_table_steal(src, orig_key);
+       g_hash_table_insert(dest, orig_key, value);
 
-       all_probes = g_slist_copy(dec->probes);
-       all_probes = g_slist_concat(all_probes, g_slist_copy(dec->opt_probes));
-       for (l = all_probes; l; l = l->next) {
-               p = l->data;
-               if (!(val_str = g_hash_table_lookup(hash, p->id)))
-                       /* Not specified. */
-                       continue;
-               val_int = strtoll(val_str, &conv, 10);
-               if (!conv || conv == val_str) {
-                       g_critical("Protocol decoder '%s' probes '%s' "
-                                       "is not a number.", dec->name, p->id);
-                       ret = FALSE;
-                       break;
-               }
-               gvar = g_variant_new_int32(val_int);
-               g_variant_ref_sink(gvar);
-               g_hash_table_insert(*probes, g_strdup(p->id), gvar);
-               g_hash_table_remove(hash, p->id);
+       return TRUE;
+}
+
+static GHashTable *extract_channel_map(struct srd_decoder *dec, GHashTable *hash)
+{
+       GHashTable *channel_map;
+       struct srd_channel *pdch;
+       GSList *l;
+
+       channel_map = g_hash_table_new_full(g_str_hash, g_str_equal,
+                                         g_free, g_free);
+
+       for (l = dec->channels; l; l = l->next) {
+               pdch = l->data;
+               move_hash_element(hash, channel_map, pdch->id);
+       }
+       for (l = dec->opt_channels; l; l = l->next) {
+               pdch = l->data;
+               move_hash_element(hash, channel_map, pdch->id);
        }
-       g_slist_free(all_probes);
 
-       return ret;
+       return channel_map;
 }
 
-/* Register the given PDs for this session.
+/*
+ * Register the given PDs for this session.
  * Accepts a string of the form: "spi:sck=3:sdata=4,spi:sck=3:sdata=5"
  * That will instantiate two SPI decoders on the clock but different data
  * lines.
@@ -120,17 +118,17 @@ static int probes_to_gvar(struct srd_decoder *dec, GHashTable *hash,
 int register_pds(const char *opt_pds, char *opt_pd_annotations)
 {
        struct srd_decoder *dec;
-       GHashTable *pd_opthash, *options, *probes;
+       GHashTable *pd_opthash, *options, *channels;
        GList *leftover, *l;
        struct srd_decoder_inst *di;
        int ret;
        char **pdtokens, **pdtok, *pd_name;
 
-       pd_ann_visible = g_hash_table_new_full(g_str_hash, g_int_equal,
-                       g_free, NULL);
+       pd_ann_visible = g_hash_table_new_full(g_str_hash, g_str_equal,
+                                              g_free, NULL);
        ret = 0;
        pd_name = NULL;
-       pd_opthash = options = probes = NULL;
+       pd_opthash = options = channels = pd_channel_maps = NULL;
        pdtokens = g_strsplit(opt_pds, ",", 0);
        for (pdtok = pdtokens; *pdtok; pdtok++) {
                if (!(pd_opthash = parse_generic_arg(*pdtok, TRUE))) {
@@ -145,21 +143,23 @@ int register_pds(const char *opt_pds, char *opt_pd_annotations)
                        ret = 1;
                        break;
                }
-               dec = srd_decoder_get_by_id(pd_name);
-
-               /* Convert decoder option and probe values to GVariant. */
-               if (!opts_to_gvar(dec, pd_opthash, &options)) {
+               if (!(dec = srd_decoder_get_by_id(pd_name))) {
+                       g_critical("Failed to get decoder %s by id.", pd_name);
                        ret = 1;
                        break;
                }
-               if (!probes_to_gvar(dec, pd_opthash, &probes)) {
+
+               /* Convert decoder option and channel values to GVariant. */
+               if (!opts_to_gvar(dec, pd_opthash, &options)) {
                        ret = 1;
                        break;
                }
+               channels = extract_channel_map(dec, pd_opthash);
+
                if (g_hash_table_size(pd_opthash) > 0) {
                        leftover = g_hash_table_get_keys(pd_opthash);
                        for (l = leftover; l; l = l->next)
-                               g_critical("Unknown option or probe '%s'", (char *)l->data);
+                               g_critical("Unknown option or channel '%s'", (char *)l->data);
                        g_list_free(leftover);
                        break;
                }
@@ -170,19 +170,25 @@ int register_pds(const char *opt_pds, char *opt_pd_annotations)
                        break;
                }
 
-               /* If no annotation list was specified, add them all in now.
+               if (pdtok == pdtokens) {
+                       /*
+                        * Save the channel setup for later, but only on the
+                        * first decoder (stacked decoders don't get channels).
+                        */
+                       pd_channel_maps = g_hash_table_new_full(g_str_hash,
+                                       g_str_equal, g_free, (GDestroyNotify)g_hash_table_destroy);
+                       g_hash_table_insert(pd_channel_maps, g_strdup(di->inst_id), channels);
+                       channels = NULL;
+               }
+
+               /*
+                * If no annotation list was specified, add them all in now.
                 * This will be pared down later to leave only the last PD
                 * in the stack.
                 */
                if (!opt_pd_annotations)
-                       g_hash_table_insert(pd_ann_visible,
-                                           g_strdup(di->inst_id), GINT_TO_POINTER(-1));
-
-               /* Remap the probes if needed. */
-               if (srd_inst_probe_set_all(di, probes) != SRD_OK) {
-                       ret = 1;
-                       break;
-               }
+                       g_hash_table_insert(pd_ann_visible, g_strdup(di->inst_id),
+                                       g_slist_append(NULL, GINT_TO_POINTER(-1)));
        }
 
        g_strfreev(pdtokens);
@@ -190,14 +196,71 @@ int register_pds(const char *opt_pds, char *opt_pd_annotations)
                g_hash_table_destroy(pd_opthash);
        if (options)
                g_hash_table_destroy(options);
-       if (probes)
-               g_hash_table_destroy(probes);
-       if (pd_name)
-               g_free(pd_name);
+       if (channels)
+               g_hash_table_destroy(channels);
+       g_free(pd_name);
 
        return ret;
 }
 
+static void map_pd_inst_channels(void *key, void *value, void *user_data)
+{
+       GHashTable *channel_map;
+       GHashTable *channel_indices;
+       GSList *channel_list;
+       struct srd_decoder_inst *di;
+       GVariant *var;
+       void *channel_id;
+       void *channel_target;
+       struct sr_channel *ch;
+       GHashTableIter iter;
+
+       channel_map = value;
+       channel_list = user_data;
+
+       di = srd_inst_find_by_id(srd_sess, key);
+       if (!di) {
+               g_critical("Protocol decoder instance \"%s\" not found.",
+                          (char *)key);
+               return;
+       }
+       channel_indices = g_hash_table_new_full(g_str_hash, g_str_equal, g_free,
+                                             (GDestroyNotify)g_variant_unref);
+
+       g_hash_table_iter_init(&iter, channel_map);
+       while (g_hash_table_iter_next(&iter, &channel_id, &channel_target)) {
+               ch = find_channel(channel_list, channel_target);
+               if (!ch) {
+                       g_printerr("cli: No channel with name \"%s\" found.\n",
+                                  (char *)channel_target);
+                       continue;
+               }
+               if (!ch->enabled)
+                       g_printerr("cli: Target channel \"%s\" not enabled.\n",
+                                  (char *)channel_target);
+
+               var = g_variant_new_int32(ch->index);
+               g_variant_ref_sink(var);
+               g_hash_table_insert(channel_indices, g_strdup(channel_id), var);
+       }
+
+       srd_inst_channel_set_all(di, channel_indices);
+}
+
+void map_pd_channels(struct sr_dev_inst *sdi)
+{
+       GSList *channels;
+
+       channels = sr_dev_inst_channels_get(sdi);
+
+       if (pd_channel_maps) {
+               g_hash_table_foreach(pd_channel_maps, &map_pd_inst_channels,
+                                    channels);
+               g_hash_table_destroy(pd_channel_maps);
+               pd_channel_maps = NULL;
+       }
+}
+
 int setup_pd_stack(char *opt_pds, char *opt_pd_stack, char *opt_pd_annotations)
 {
        struct srd_decoder_inst *di_from, *di_to;
@@ -241,13 +304,13 @@ int setup_pd_stack(char *opt_pds, char *opt_pd_stack, char *opt_pd_annotations)
                        if ((ret = srd_inst_stack(srd_sess, di_from, di_to)) != SRD_OK)
                                return 1;
 
-                       /* Don't show annotation from this PD. Only the last PD in
+                       /*
+                        * Don't show annotation from this PD. Only the last PD in
                         * the stack will be left on the annotation list (unless
                         * the annotation list was specifically provided).
                         */
                        if (!opt_pd_annotations)
-                               g_hash_table_remove(pd_ann_visible,
-                                                   di_from->inst_id);
+                               g_hash_table_remove(pd_ann_visible, di_from->inst_id);
 
                        di_from = di_to;
                }
@@ -259,10 +322,10 @@ int setup_pd_stack(char *opt_pds, char *opt_pd_stack, char *opt_pd_annotations)
 
 int setup_pd_annotations(char *opt_pd_annotations)
 {
-       GSList *l;
+       GSList *l, *l_ann;
        struct srd_decoder *dec;
        int ann_class;
-       char **pds, **pdtok, **keyval, **ann_descr;
+       char **pds, **pdtok, **keyval, **annlist, **ann, **ann_descr;
 
        /* Set up custom list of PDs and annotations to show. */
        pds = g_strsplit(opt_pd_annotations, ",", 0);
@@ -276,28 +339,34 @@ int setup_pd_annotations(char *opt_pd_annotations)
                        g_critical("Protocol decoder '%s' has no annotations.", keyval[0]);
                        return 1;
                }
-               ann_class = 0;
-               if (g_strv_length(keyval) == 2) {
-                       for (l = dec->annotations; l; l = l->next, ann_class++) {
-                               ann_descr = l->data;
-                               if (!canon_cmp(ann_descr[0], keyval[1]))
-                                       /* Found it. */
-                                       break;
-                       }
-                       if (!l) {
-                               g_critical("Annotation '%s' not found "
-                                               "for protocol decoder '%s'.", keyval[1], keyval[0]);
-                               return 1;
+               if (g_strv_length(keyval) == 2 && keyval[1][0] != '\0') {
+                       annlist = g_strsplit(keyval[1], ":", 0);
+                       for (ann = annlist; *ann && **ann; ann++) {
+                               ann_class = 0;
+                               for (l = dec->annotations; l; l = l->next, ann_class++) {
+                                       ann_descr = l->data;
+                                       if (!canon_cmp(ann_descr[0], *ann))
+                                               /* Found it. */
+                                               break;
+                               }
+                               if (!l) {
+                                       g_critical("Annotation '%s' not found "
+                                                       "for protocol decoder '%s'.", *ann, keyval[0]);
+                                       return 1;
+                               }
+                               l_ann = g_hash_table_lookup(pd_ann_visible, keyval[0]);
+                               l_ann = g_slist_append(l_ann, GINT_TO_POINTER(ann_class));
+                               g_hash_table_replace(pd_ann_visible, g_strdup(keyval[0]), l_ann);
+                               g_debug("cli: Showing protocol decoder %s annotation "
+                                               "class %d (%s).", keyval[0], ann_class, ann_descr[0]);
                        }
-                       g_debug("cli: Showing protocol decoder %s annotation "
-                                       "class %d (%s).", keyval[0], ann_class, ann_descr[0]);
                } else {
                        /* No class specified: show all of them. */
-                       ann_class = -1;
+                               g_hash_table_insert(pd_ann_visible, g_strdup(keyval[0]),
+                                               g_slist_append(NULL, GINT_TO_POINTER(-1)));
                        g_debug("cli: Showing all annotation classes for protocol "
                                        "decoder %s.", keyval[0]);
                }
-               g_hash_table_insert(pd_ann_visible, g_strdup(keyval[0]), GINT_TO_POINTER(ann_class));
                g_strfreev(keyval);
        }
        g_strfreev(pds);
@@ -331,7 +400,7 @@ int setup_pd_binary(char *opt_pd_binary)
        GSList *l;
        struct srd_decoder *dec;
        int bin_class;
-       char **pds, **pdtok, **keyval, *bin_name;
+       char **pds, **pdtok, **keyval, **bin_name;
 
        pd_binary_visible = g_hash_table_new_full(g_str_hash, g_int_equal,
                        g_free, NULL);
@@ -350,7 +419,7 @@ int setup_pd_binary(char *opt_pd_binary)
                if (g_strv_length(keyval) == 2) {
                        for (l = dec->binary; l; l = l->next, bin_class++) {
                                bin_name = l->data;
-                               if (!canon_cmp(bin_name, keyval[1]))
+                               if (!strcmp(bin_name[0], keyval[1]))
                                        /* Found it. */
                                        break;
                        }
@@ -360,7 +429,7 @@ int setup_pd_binary(char *opt_pd_binary)
                                return 1;
                        }
                        g_debug("cli: Showing protocol decoder %s binary class "
-                                       "%d (%s).", keyval[0], bin_class, bin_name);
+                                       "%d (%s).", keyval[0], bin_class, bin_name[0]);
                } else {
                        /* No class specified: output all of them. */
                        bin_class = -1;
@@ -379,26 +448,32 @@ void show_pd_annotations(struct srd_proto_data *pdata, void *cb_data)
 {
        struct srd_decoder *dec;
        struct srd_proto_data_annotation *pda;
-       gpointer ann_format;
-       int format, i;
+       GSList *ann_list, *l;
+       int i;
        char **ann_descr;
+       gboolean show;
 
-       /* 'cb_data' is not used in this specific callback. */
        (void)cb_data;
 
        if (!pd_ann_visible)
                return;
 
        if (!g_hash_table_lookup_extended(pd_ann_visible, pdata->pdo->di->inst_id,
-                       NULL, &ann_format))
+                       NULL, (void **)&ann_list))
                /* Not in the list of PDs whose annotations we're showing. */
                return;
 
-       format = GPOINTER_TO_INT(ann_format);
        dec = pdata->pdo->di->decoder;
        pda = pdata->data;
-       if (format != -1 && pda->ann_format != format)
-               /* We don't want this particular format from the PD. */
+       show = FALSE;
+       for (l = ann_list; l; l = l->next) {
+               if (GPOINTER_TO_INT(l->data) == -1
+                               || GPOINTER_TO_INT(l->data) == pda->ann_class) {
+                       show = TRUE;
+                       break;
+               }
+       }
+       if (!show)
                return;
 
        if (opt_loglevel <= SR_LOG_WARN) {
@@ -412,7 +487,7 @@ void show_pd_annotations(struct srd_proto_data *pdata, void *cb_data)
                } else {
                        /* Protocol decoder id, annotation class,
                         * all annotation strings. */
-                       ann_descr = g_slist_nth_data(dec->annotations, pda->ann_format);
+                       ann_descr = g_slist_nth_data(dec->annotations, pda->ann_class);
                        printf(" %s: %s:", pdata->pdo->proto_id, ann_descr[0]);
                        for (i = 0; pda->ann_text[i]; i++)
                                printf(" \"%s\"", pda->ann_text[i]);
@@ -424,8 +499,6 @@ void show_pd_annotations(struct srd_proto_data *pdata, void *cb_data)
 
 void show_pd_meta(struct srd_proto_data *pdata, void *cb_data)
 {
-
-       /* 'cb_data' is not used in this specific callback. */
        (void)cb_data;
 
        if (!g_hash_table_lookup_extended(pd_meta_visible,
@@ -447,7 +520,6 @@ void show_pd_binary(struct srd_proto_data *pdata, void *cb_data)
        gpointer classp;
        int class;
 
-       /* 'cb_data' is not used in this specific callback. */
        (void)cb_data;
 
        if (!g_hash_table_lookup_extended(pd_binary_visible,
@@ -466,4 +538,3 @@ void show_pd_binary(struct srd_proto_data *pdata, void *cb_data)
        fflush(stdout);
 }
 #endif
-