src/hardware/demo/api.c

   1 /*
   2  * This file is part of the libsigrok project.
   3  *
   4  * Copyright (C) 2010 Uwe Hermann <uwe@hermann-uwe.de>
   5  * Copyright (C) 2011 Olivier Fauchon <olivier@aixmarseille.com>
   6  * Copyright (C) 2012 Alexandru Gagniuc <mr.nuke.me@gmail.com>
   7  * Copyright (C) 2015 Bartosz Golaszewski <bgolaszewski@baylibre.com>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23 #include <config.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <math.h>
  27 #include <libsigrok/libsigrok.h>
  28 #include "libsigrok-internal.h"
  29 #include "protocol.h"
  30
  31 #define DEFAULT_NUM_LOGIC_CHANNELS      8
  32 #define DEFAULT_LOGIC_PATTERN           PATTERN_SIGROK
  33
  34 #define DEFAULT_NUM_ANALOG_CHANNELS     4
  35 #define DEFAULT_ANALOG_AMPLITUDE        10
  36
  37 /* Note: No spaces allowed because of sigrok-cli. */
  38 static const char *logic_pattern_str[] = {
  39         "sigrok",
  40         "random",
  41         "incremental",
  42         "walking-one",
  43         "walking-zero",
  44         "all-low",
  45         "all-high",
  46         "squid",
  47 };
  48
  49 static const uint32_t drvopts[] = {
  50         SR_CONF_DEMO_DEV,
  51         SR_CONF_LOGIC_ANALYZER,
  52         SR_CONF_OSCILLOSCOPE,
  53 };
  54
  55 static const uint32_t scanopts[] = {
  56         SR_CONF_NUM_LOGIC_CHANNELS,
  57         SR_CONF_NUM_ANALOG_CHANNELS,
  58 };
  59
  60 static const uint32_t devopts[] = {
  61         SR_CONF_CONTINUOUS,
  62         SR_CONF_LIMIT_SAMPLES | SR_CONF_GET | SR_CONF_SET,
  63         SR_CONF_LIMIT_MSEC | SR_CONF_GET | SR_CONF_SET,
  64         SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  65         SR_CONF_AVERAGING | SR_CONF_GET | SR_CONF_SET,
  66         SR_CONF_AVG_SAMPLES | SR_CONF_GET | SR_CONF_SET,
  67 };
  68
  69 static const uint32_t devopts_cg_logic[] = {
  70         SR_CONF_PATTERN_MODE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  71 };
  72
  73 static const uint32_t devopts_cg_analog_group[] = {
  74         SR_CONF_AMPLITUDE | SR_CONF_GET | SR_CONF_SET,
  75 };
  76
  77 static const uint32_t devopts_cg_analog_channel[] = {
  78         SR_CONF_PATTERN_MODE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  79         SR_CONF_AMPLITUDE | SR_CONF_GET | SR_CONF_SET,
  80 };
  81
  82 static const uint64_t samplerates[] = {
  83         SR_HZ(1),
  84         SR_GHZ(1),
  85         SR_HZ(1),
  86 };
  87
  88 static GSList *scan(struct sr_dev_driver *di, GSList *options)
  89 {
  90         struct dev_context *devc;
  91         struct sr_dev_inst *sdi;
  92         struct sr_channel *ch;
  93         struct sr_channel_group *cg, *acg;
  94         struct sr_config *src;
  95         struct analog_gen *ag;
  96         GSList *l;
  97         int num_logic_channels, num_analog_channels, pattern, i;
  98         char channel_name[16];
  99
 100         num_logic_channels = DEFAULT_NUM_LOGIC_CHANNELS;
 101         num_analog_channels = DEFAULT_NUM_ANALOG_CHANNELS;
 102         for (l = options; l; l = l->next) {
 103                 src = l->data;
 104                 switch (src->key) {
 105                 case SR_CONF_NUM_LOGIC_CHANNELS:
 106                         num_logic_channels = g_variant_get_int32(src->data);
 107                         break;
 108                 case SR_CONF_NUM_ANALOG_CHANNELS:
 109                         num_analog_channels = g_variant_get_int32(src->data);
 110                         break;
 111                 }
 112         }
 113
 114         sdi = g_malloc0(sizeof(struct sr_dev_inst));
 115         sdi->status = SR_ST_INACTIVE;
 116         sdi->model = g_strdup("Demo device");
 117
 118         devc = g_malloc0(sizeof(struct dev_context));
 119         devc->cur_samplerate = SR_KHZ(200);
 120         devc->num_logic_channels = num_logic_channels;
 121         devc->logic_unitsize = (devc->num_logic_channels + 7) / 8;
 122         devc->logic_pattern = DEFAULT_LOGIC_PATTERN;
 123         devc->num_analog_channels = num_analog_channels;
 124
 125         if (num_logic_channels > 0) {
 126                 /* Logic channels, all in one channel group. */
 127                 cg = g_malloc0(sizeof(struct sr_channel_group));
 128                 cg->name = g_strdup("Logic");
 129                 for (i = 0; i < num_logic_channels; i++) {
 130                         sprintf(channel_name, "D%d", i);
 131                         ch = sr_channel_new(sdi, i, SR_CHANNEL_LOGIC, TRUE, channel_name);
 132                         cg->channels = g_slist_append(cg->channels, ch);
 133                 }
 134                 sdi->channel_groups = g_slist_append(NULL, cg);
 135         }
 136
 137         /* Analog channels, channel groups and pattern generators. */
 138         devc->ch_ag = g_hash_table_new(g_direct_hash, g_direct_equal);
 139         if (num_analog_channels > 0) {
 140                 pattern = 0;
 141                 /* An "Analog" channel group with all analog channels in it. */
 142                 acg = g_malloc0(sizeof(struct sr_channel_group));
 143                 acg->name = g_strdup("Analog");
 144                 sdi->channel_groups = g_slist_append(sdi->channel_groups, acg);
 145
 146                 for (i = 0; i < num_analog_channels; i++) {
 147                         snprintf(channel_name, 16, "A%d", i);
 148                         ch = sr_channel_new(sdi, i + num_logic_channels, SR_CHANNEL_ANALOG,
 149                                         TRUE, channel_name);
 150                         acg->channels = g_slist_append(acg->channels, ch);
 151
 152                         /* Every analog channel gets its own channel group as well. */
 153                         cg = g_malloc0(sizeof(struct sr_channel_group));
 154                         cg->name = g_strdup(channel_name);
 155                         cg->channels = g_slist_append(NULL, ch);
 156                         sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
 157
 158                         /* Every channel gets a generator struct. */
 159                         ag = g_malloc(sizeof(struct analog_gen));
 160                         ag->ch = ch;
 161                         ag->amplitude = DEFAULT_ANALOG_AMPLITUDE;
 162                         sr_analog_init(&ag->packet, &ag->encoding, &ag->meaning, &ag->spec, 2);
 163                         ag->packet.meaning->channels = cg->channels;
 164                         ag->packet.meaning->mq = 0;
 165                         ag->packet.meaning->mqflags = 0;
 166                         ag->packet.meaning->unit = SR_UNIT_VOLT;
 167                         ag->packet.data = ag->pattern_data;
 168                         ag->pattern = pattern;
 169                         ag->avg_val = 0.0f;
 170                         ag->num_avgs = 0;
 171                         g_hash_table_insert(devc->ch_ag, ch, ag);
 172
 173                         if (++pattern == ARRAY_SIZE(analog_pattern_str))
 174                                 pattern = 0;
 175                 }
 176         }
 177
 178         sdi->priv = devc;
 179
 180         return std_scan_complete(di, g_slist_append(NULL, sdi));
 181 }
 182
 183 static void clear_helper(void *priv)
 184 {
 185         struct dev_context *devc;
 186         GHashTableIter iter;
 187         void *value;
 188
 189         devc = priv;
 190
 191         /* Analog generators. */
 192         g_hash_table_iter_init(&iter, devc->ch_ag);
 193         while (g_hash_table_iter_next(&iter, NULL, &value))
 194                 g_free(value);
 195         g_hash_table_unref(devc->ch_ag);
 196 }
 197
 198 static int dev_clear(const struct sr_dev_driver *di)
 199 {
 200         return std_dev_clear_with_callback(di, clear_helper);
 201 }
 202
 203 static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
 204                 const struct sr_channel_group *cg)
 205 {
 206         struct dev_context *devc;
 207         struct sr_channel *ch;
 208         struct analog_gen *ag;
 209         int pattern;
 210
 211         if (!sdi)
 212                 return SR_ERR_ARG;
 213
 214         devc = sdi->priv;
 215         switch (key) {
 216         case SR_CONF_SAMPLERATE:
 217                 *data = g_variant_new_uint64(devc->cur_samplerate);
 218                 break;
 219         case SR_CONF_LIMIT_SAMPLES:
 220                 *data = g_variant_new_uint64(devc->limit_samples);
 221                 break;
 222         case SR_CONF_LIMIT_MSEC:
 223                 *data = g_variant_new_uint64(devc->limit_msec);
 224                 break;
 225         case SR_CONF_AVERAGING:
 226                 *data = g_variant_new_boolean(devc->avg);
 227                 break;
 228         case SR_CONF_AVG_SAMPLES:
 229                 *data = g_variant_new_uint64(devc->avg_samples);
 230                 break;
 231         case SR_CONF_PATTERN_MODE:
 232                 if (!cg)
 233                         return SR_ERR_CHANNEL_GROUP;
 234                 /* Any channel in the group will do. */
 235                 ch = cg->channels->data;
 236                 if (ch->type == SR_CHANNEL_LOGIC) {
 237                         pattern = devc->logic_pattern;
 238                         *data = g_variant_new_string(logic_pattern_str[pattern]);
 239                 } else if (ch->type == SR_CHANNEL_ANALOG) {
 240                         ag = g_hash_table_lookup(devc->ch_ag, ch);
 241                         pattern = ag->pattern;
 242                         *data = g_variant_new_string(analog_pattern_str[pattern]);
 243                 } else
 244                         return SR_ERR_BUG;
 245                 break;
 246         case SR_CONF_AMPLITUDE:
 247                 if (!cg)
 248                         return SR_ERR_CHANNEL_GROUP;
 249                 /* Any channel in the group will do. */
 250                 ch = cg->channels->data;
 251                 if (ch->type != SR_CHANNEL_ANALOG)
 252                         return SR_ERR_ARG;
 253                 ag = g_hash_table_lookup(devc->ch_ag, ch);
 254                 *data = g_variant_new_double(ag->amplitude);
 255                 break;
 256         default:
 257                 return SR_ERR_NA;
 258         }
 259
 260         return SR_OK;
 261 }
 262
 263 static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sdi,
 264                 const struct sr_channel_group *cg)
 265 {
 266         struct dev_context *devc;
 267         struct analog_gen *ag;
 268         struct sr_channel *ch;
 269         GSList *l;
 270         int logic_pattern, analog_pattern, ret;
 271         unsigned int i;
 272         const char *stropt;
 273
 274         devc = sdi->priv;
 275
 276         ret = SR_OK;
 277         switch (key) {
 278         case SR_CONF_SAMPLERATE:
 279                 devc->cur_samplerate = g_variant_get_uint64(data);
 280                 break;
 281         case SR_CONF_LIMIT_SAMPLES:
 282                 devc->limit_msec = 0;
 283                 devc->limit_samples = g_variant_get_uint64(data);
 284                 break;
 285         case SR_CONF_LIMIT_MSEC:
 286                 devc->limit_msec = g_variant_get_uint64(data);
 287                 devc->limit_samples = 0;
 288                 break;
 289         case SR_CONF_AVERAGING:
 290                 devc->avg = g_variant_get_boolean(data);
 291                 sr_dbg("%s averaging", devc->avg ? "Enabling" : "Disabling");
 292                 break;
 293         case SR_CONF_AVG_SAMPLES:
 294                 devc->avg_samples = g_variant_get_uint64(data);
 295                 sr_dbg("Setting averaging rate to %" PRIu64, devc->avg_samples);
 296                 break;
 297         case SR_CONF_PATTERN_MODE:
 298                 if (!cg)
 299                         return SR_ERR_CHANNEL_GROUP;
 300                 stropt = g_variant_get_string(data, NULL);
 301                 logic_pattern = analog_pattern = -1;
 302                 for (i = 0; i < ARRAY_SIZE(logic_pattern_str); i++) {
 303                         if (!strcmp(stropt, logic_pattern_str[i])) {
 304                                 logic_pattern = i;
 305                                 break;
 306                         }
 307                 }
 308                 for (i = 0; i < ARRAY_SIZE(analog_pattern_str); i++) {
 309                         if (!strcmp(stropt, analog_pattern_str[i])) {
 310                                 analog_pattern = i;
 311                                 break;
 312                         }
 313                 }
 314                 if (logic_pattern == -1 && analog_pattern == -1)
 315                         return SR_ERR_ARG;
 316                 for (l = cg->channels; l; l = l->next) {
 317                         ch = l->data;
 318                         if (ch->type == SR_CHANNEL_LOGIC) {
 319                                 if (logic_pattern == -1)
 320                                         return SR_ERR_ARG;
 321                                 sr_dbg("Setting logic pattern to %s",
 322                                                 logic_pattern_str[logic_pattern]);
 323                                 devc->logic_pattern = logic_pattern;
 324                                 /* Might as well do this now, these are static. */
 325                                 if (logic_pattern == PATTERN_ALL_LOW)
 326                                         memset(devc->logic_data, 0x00, LOGIC_BUFSIZE);
 327                                 else if (logic_pattern == PATTERN_ALL_HIGH)
 328                                         memset(devc->logic_data, 0xff, LOGIC_BUFSIZE);
 329                         } else if (ch->type == SR_CHANNEL_ANALOG) {
 330                                 if (analog_pattern == -1)
 331                                         return SR_ERR_ARG;
 332                                 sr_dbg("Setting analog pattern for channel %s to %s",
 333                                                 ch->name, analog_pattern_str[analog_pattern]);
 334                                 ag = g_hash_table_lookup(devc->ch_ag, ch);
 335                                 ag->pattern = analog_pattern;
 336                         } else
 337                                 return SR_ERR_BUG;
 338                 }
 339                 break;
 340         case SR_CONF_AMPLITUDE:
 341                 if (!cg)
 342                         return SR_ERR_CHANNEL_GROUP;
 343                 for (l = cg->channels; l; l = l->next) {
 344                         ch = l->data;
 345                         if (ch->type != SR_CHANNEL_ANALOG)
 346                                 return SR_ERR_ARG;
 347                         ag = g_hash_table_lookup(devc->ch_ag, ch);
 348                         ag->amplitude = g_variant_get_double(data);
 349                 }
 350                 break;
 351         default:
 352                 ret = SR_ERR_NA;
 353         }
 354
 355         return ret;
 356 }
 357
 358 static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
 359                 const struct sr_channel_group *cg)
 360 {
 361         struct sr_channel *ch;
 362         GVariant *gvar;
 363         GVariantBuilder gvb;
 364
 365         if (key == SR_CONF_SCAN_OPTIONS) {
 366                 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
 367                                 scanopts, ARRAY_SIZE(scanopts), sizeof(uint32_t));
 368                 return SR_OK;
 369         }
 370
 371         if (key == SR_CONF_DEVICE_OPTIONS && !sdi) {
 372                 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
 373                                 drvopts, ARRAY_SIZE(drvopts), sizeof(uint32_t));
 374                 return SR_OK;
 375         }
 376
 377         if (!sdi)
 378                 return SR_ERR_ARG;
 379
 380         if (!cg) {
 381                 switch (key) {
 382                 case SR_CONF_DEVICE_OPTIONS:
 383                         *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
 384                                         devopts, ARRAY_SIZE(devopts), sizeof(uint32_t));
 385                         break;
 386                 case SR_CONF_SAMPLERATE:
 387                         g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}"));
 388                         gvar = g_variant_new_fixed_array(G_VARIANT_TYPE("t"), samplerates,
 389                                         ARRAY_SIZE(samplerates), sizeof(uint64_t));
 390                         g_variant_builder_add(&gvb, "{sv}", "samplerate-steps", gvar);
 391                         *data = g_variant_builder_end(&gvb);
 392                         break;
 393                 default:
 394                         return SR_ERR_NA;
 395                 }
 396         } else {
 397                 ch = cg->channels->data;
 398                 switch (key) {
 399                 case SR_CONF_DEVICE_OPTIONS:
 400                         if (ch->type == SR_CHANNEL_LOGIC)
 401                                 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
 402                                                 devopts_cg_logic, ARRAY_SIZE(devopts_cg_logic),
 403                                                 sizeof(uint32_t));
 404                         else if (ch->type == SR_CHANNEL_ANALOG) {
 405                                 if (strcmp(cg->name, "Analog") == 0)
 406                                         *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
 407                                                         devopts_cg_analog_group, ARRAY_SIZE(devopts_cg_analog_group),
 408                                                         sizeof(uint32_t));
 409                                 else
 410                                         *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
 411                                                         devopts_cg_analog_channel, ARRAY_SIZE(devopts_cg_analog_channel),
 412                                                         sizeof(uint32_t));
 413                         }
 414                         else
 415                                 return SR_ERR_BUG;
 416                         break;
 417                 case SR_CONF_PATTERN_MODE:
 418                         /* The analog group (with all 4 channels) shall not have a pattern property. */
 419                         if (strcmp(cg->name, "Analog") == 0)
 420                                 return SR_ERR_NA;
 421
 422                         if (ch->type == SR_CHANNEL_LOGIC)
 423                                 *data = g_variant_new_strv(logic_pattern_str,
 424                                                 ARRAY_SIZE(logic_pattern_str));
 425                         else if (ch->type == SR_CHANNEL_ANALOG)
 426                                 *data = g_variant_new_strv(analog_pattern_str,
 427                                                 ARRAY_SIZE(analog_pattern_str));
 428                         else
 429                                 return SR_ERR_BUG;
 430                         break;
 431                 default:
 432                         return SR_ERR_NA;
 433                 }
 434         }
 435
 436         return SR_OK;
 437 }
 438
 439 static int dev_acquisition_start(const struct sr_dev_inst *sdi)
 440 {
 441         struct dev_context *devc;
 442         GSList *l;
 443         struct sr_channel *ch;
 444         int bitpos;
 445         uint8_t mask;
 446         GHashTableIter iter;
 447         void *value;
 448
 449         devc = sdi->priv;
 450         devc->sent_samples = 0;
 451
 452         /*
 453          * Determine the numbers of logic and analog channels that are
 454          * involved in the acquisition. Determine an offset and a mask to
 455          * remove excess logic data content before datafeed submission.
 456          */
 457         devc->enabled_logic_channels = 0;
 458         devc->enabled_analog_channels = 0;
 459         for (l = sdi->channels; l; l = l->next) {
 460                 ch = l->data;
 461                 if (!ch->enabled)
 462                         continue;
 463                 if (ch->type == SR_CHANNEL_ANALOG) {
 464                         devc->enabled_analog_channels++;
 465                         continue;
 466                 }
 467                 if (ch->type != SR_CHANNEL_LOGIC)
 468                         continue;
 469                 /*
 470                  * TODO: Need we create a channel map here, such that the
 471                  * session datafeed packets will have a dense representation
 472                  * of the enabled channels' data? For example store channels
 473                  * D3 and D5 in bit positions 0 and 1 respectively, when all
 474                  * other channels are disabled? The current implementation
 475                  * generates a sparse layout, might provide data for logic
 476                  * channels that are disabled while it might suppress data
 477                  * from enabled channels at the same time.
 478                  */
 479                 devc->enabled_logic_channels++;
 480         }
 481         devc->first_partial_logic_index = devc->enabled_logic_channels / 8;
 482         bitpos = devc->enabled_logic_channels % 8;
 483         mask = (1 << bitpos) - 1;
 484         devc->first_partial_logic_mask = mask;
 485         sr_dbg("num logic %zu, partial off %zu, mask 0x%02x.",
 486                 devc->enabled_logic_channels,
 487                 devc->first_partial_logic_index,
 488                 devc->first_partial_logic_mask);
 489
 490         /*
 491          * Have the waveform for analog patterns pre-generated. It's
 492          * supposed to be periodic, so the generator just needs to
 493          * access the prepared sample data (DDS style).
 494          */
 495         g_hash_table_iter_init(&iter, devc->ch_ag);
 496         while (g_hash_table_iter_next(&iter, NULL, &value))
 497                 demo_generate_analog_pattern(value, devc->cur_samplerate);
 498
 499         sr_session_source_add(sdi->session, -1, 0, 100,
 500                         demo_prepare_data, (struct sr_dev_inst *)sdi);
 501
 502         std_session_send_df_header(sdi);
 503
 504         /* We use this timestamp to decide how many more samples to send. */
 505         devc->start_us = g_get_monotonic_time();
 506         devc->spent_us = 0;
 507         devc->step = 0;
 508
 509         return SR_OK;
 510 }
 511
 512 static int dev_acquisition_stop(struct sr_dev_inst *sdi)
 513 {
 514         sr_session_source_remove(sdi->session, -1);
 515         std_session_send_df_end(sdi);
 516
 517         return SR_OK;
 518 }
 519
 520 static struct sr_dev_driver demo_driver_info = {
 521         .name = "demo",
 522         .longname = "Demo driver and pattern generator",
 523         .api_version = 1,
 524         .init = std_init,
 525         .cleanup = std_cleanup,
 526         .scan = scan,
 527         .dev_list = std_dev_list,
 528         .dev_clear = dev_clear,
 529         .config_get = config_get,
 530         .config_set = config_set,
 531         .config_list = config_list,
 532         .dev_open = std_dummy_dev_open,
 533         .dev_close = std_dummy_dev_close,
 534         .dev_acquisition_start = dev_acquisition_start,
 535         .dev_acquisition_stop = dev_acquisition_stop,
 536         .context = NULL,
 537 };
 538 SR_REGISTER_DEV_DRIVER(demo_driver_info);