]> sigrok.org Git - libsigrok.git/blobdiff - src/hardware/demo/demo.c
Put driver pointers into special section
[libsigrok.git] / src / hardware / demo / demo.c
index 21715ea0826e52fcbdd19dc05976ae51b8793f02..9fa744b1f4e965b448b5509ba277d068adedbc75 100644 (file)
@@ -4,6 +4,7 @@
  * Copyright (C) 2010 Uwe Hermann <uwe@hermann-uwe.de>
  * Copyright (C) 2011 Olivier Fauchon <olivier@aixmarseille.com>
  * Copyright (C) 2012 Alexandru Gagniuc <mr.nuke.me@gmail.com>
+ * Copyright (C) 2015 Bartosz Golaszewski <bgolaszewski@baylibre.com>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  */
 
+#include <config.h>
 #include <stdlib.h>
-#include <unistd.h>
 #include <string.h>
 #include <math.h>
-#ifdef _WIN32
-#include <io.h>
-#include <fcntl.h>
-#define pipe(fds) _pipe(fds, 4096, _O_BINARY)
-#endif
-#include "libsigrok.h"
+#include <libsigrok/libsigrok.h>
 #include "libsigrok-internal.h"
 
 #define LOG_PREFIX "demo"
@@ -42,7 +38,7 @@
 /* Size of the analog pattern space per channel. */
 #define ANALOG_BUFSIZE       4096
 
-#define DEFAULT_ANALOG_AMPLITUDE 25
+#define DEFAULT_ANALOG_AMPLITUDE 10
 #define ANALOG_SAMPLES_PER_PERIOD 20
 
 /* Logic patterns we can generate. */
@@ -102,20 +98,19 @@ struct analog_gen {
        float amplitude;
        float pattern_data[ANALOG_BUFSIZE];
        unsigned int num_samples;
-       struct sr_datafeed_analog packet;
+       struct sr_datafeed_analog_old packet;
+       float avg_val; /* Average value */
+       unsigned num_avgs; /* Number of samples averaged */
 };
 
 /* Private, per-device-instance driver context. */
 struct dev_context {
-       int pipe_fds[2];
-       GIOChannel *channel;
        uint64_t cur_samplerate;
-       gboolean continuous;
        uint64_t limit_samples;
        uint64_t limit_msec;
-       uint64_t logic_counter;
-       uint64_t analog_counter;
-       int64_t starttime;
+       uint64_t sent_samples;
+       int64_t start_us;
+       int64_t spent_us;
        uint64_t step;
        /* Logic */
        int32_t num_logic_channels;
@@ -126,6 +121,14 @@ struct dev_context {
        /* Analog */
        int32_t num_analog_channels;
        GHashTable *ch_ag;
+       gboolean avg; /* True if averaging is enabled */
+       uint64_t avg_samples;
+};
+
+static const uint32_t drvopts[] = {
+       SR_CONF_DEMO_DEV,
+       SR_CONF_LOGIC_ANALYZER,
+       SR_CONF_OSCILLOSCOPE,
 };
 
 static const uint32_t scanopts[] = {
@@ -134,23 +137,23 @@ static const uint32_t scanopts[] = {
 };
 
 static const uint32_t devopts[] = {
-       SR_CONF_DEMO_DEV,
-       SR_CONF_LOGIC_ANALYZER,
-       SR_CONF_OSCILLOSCOPE,
        SR_CONF_CONTINUOUS,
        SR_CONF_LIMIT_SAMPLES | SR_CONF_GET | SR_CONF_SET,
        SR_CONF_LIMIT_MSEC | SR_CONF_GET | SR_CONF_SET,
-};
-
-static const uint32_t devopts_global[] = {
        SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
+       SR_CONF_AVERAGING | SR_CONF_GET | SR_CONF_SET,
+       SR_CONF_AVG_SAMPLES | SR_CONF_GET | SR_CONF_SET,
 };
 
 static const uint32_t devopts_cg_logic[] = {
        SR_CONF_PATTERN_MODE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
 };
 
-static const uint32_t devopts_cg_analog[] = {
+static const uint32_t devopts_cg_analog_group[] = {
+       SR_CONF_AMPLITUDE | SR_CONF_GET | SR_CONF_SET,
+};
+
+static const uint32_t devopts_cg_analog_channel[] = {
        SR_CONF_PATTERN_MODE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
        SR_CONF_AMPLITUDE | SR_CONF_GET | SR_CONF_SET,
 };
@@ -161,7 +164,7 @@ static const uint64_t samplerates[] = {
        SR_HZ(1),
 };
 
-static uint8_t pattern_sigrok[] = {
+static const uint8_t pattern_sigrok[] = {
        0x4c, 0x92, 0x92, 0x92, 0x64, 0x00, 0x00, 0x00,
        0x82, 0xfe, 0xfe, 0x82, 0x00, 0x00, 0x00, 0x00,
        0x7c, 0x82, 0x82, 0x92, 0x74, 0x00, 0x00, 0x00,
@@ -172,16 +175,7 @@ static uint8_t pattern_sigrok[] = {
        0xbe, 0xbe, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
 };
 
-SR_PRIV struct sr_dev_driver demo_driver_info;
-static struct sr_dev_driver *di = &demo_driver_info;
-
-static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data);
-
-
-static int init(struct sr_context *sr_ctx)
-{
-       return std_init(sr_ctx, di, LOG_PREFIX);
-}
+static int dev_acquisition_stop(struct sr_dev_inst *sdi);
 
 static void generate_analog_pattern(struct analog_gen *ag, uint64_t sample_rate)
 {
@@ -207,26 +201,24 @@ static void generate_analog_pattern(struct analog_gen *ag, uint64_t sample_rate)
                }
                ag->num_samples = last_end;
                break;
-
        case PATTERN_SINE:
                frequency = (double) sample_rate / ANALOG_SAMPLES_PER_PERIOD;
 
                /* Make sure the number of samples we put out is an integer
                 * multiple of our period size */
                /* FIXME we actually need only one period. A ringbuffer would be
-                * usefull here.*/
+                * useful here. */
                while (num_samples % ANALOG_SAMPLES_PER_PERIOD != 0)
                        num_samples--;
 
                for (i = 0; i < num_samples; i++) {
                        t = (double) i / (double) sample_rate;
                        ag->pattern_data[i] = ag->amplitude *
-                                               sin(2 * M_PI * frequency * t);
+                                               sin(2 * G_PI * frequency * t);
                }
 
                ag->num_samples = num_samples;
                break;
-
        case PATTERN_TRIANGLE:
                frequency = (double) sample_rate / ANALOG_SAMPLES_PER_PERIOD;
 
@@ -235,13 +227,12 @@ static void generate_analog_pattern(struct analog_gen *ag, uint64_t sample_rate)
 
                for (i = 0; i < num_samples; i++) {
                        t = (double) i / (double) sample_rate;
-                       ag->pattern_data[i] = (2 * ag->amplitude / M_PI) *
-                                               asin(sin(2 * M_PI * frequency * t));
+                       ag->pattern_data[i] = (2 * ag->amplitude / G_PI) *
+                                               asin(sin(2 * G_PI * frequency * t));
                }
 
                ag->num_samples = num_samples;
                break;
-
        case PATTERN_SAWTOOTH:
                frequency = (double) sample_rate / ANALOG_SAMPLES_PER_PERIOD;
 
@@ -259,7 +250,7 @@ static void generate_analog_pattern(struct analog_gen *ag, uint64_t sample_rate)
        }
 }
 
-static GSList *scan(GSList *options)
+static GSList *scan(struct sr_dev_driver *di, GSList *options)
 {
        struct drv_context *drvc;
        struct dev_context *devc;
@@ -272,7 +263,7 @@ static GSList *scan(GSList *options)
        int num_logic_channels, num_analog_channels, pattern, i;
        char channel_name[16];
 
-       drvc = di->priv;
+       drvc = di->context;
 
        num_logic_channels = DEFAULT_NUM_LOGIC_CHANNELS;
        num_analog_channels = DEFAULT_NUM_ANALOG_CHANNELS;
@@ -290,68 +281,67 @@ static GSList *scan(GSList *options)
 
        devices = NULL;
 
-       sdi = sr_dev_inst_new();
-       sdi->status = SR_ST_ACTIVE;
+       sdi = g_malloc0(sizeof(struct sr_dev_inst));
+       sdi->status = SR_ST_INACTIVE;
        sdi->model = g_strdup("Demo device");
        sdi->driver = di;
 
-       devc = g_malloc(sizeof(struct dev_context));
+       devc = g_malloc0(sizeof(struct dev_context));
        devc->cur_samplerate = SR_KHZ(200);
-       devc->limit_samples = 0;
-       devc->limit_msec = 0;
-       devc->step = 0;
-       devc->continuous = FALSE;
        devc->num_logic_channels = num_logic_channels;
        devc->logic_unitsize = (devc->num_logic_channels + 7) / 8;
        devc->logic_pattern = PATTERN_SIGROK;
        devc->num_analog_channels = num_analog_channels;
 
-       /* Logic channels, all in one channel group. */
-       cg = g_malloc0(sizeof(struct sr_channel_group));
-       cg->name = g_strdup("Logic");
-       for (i = 0; i < num_logic_channels; i++) {
-               sprintf(channel_name, "D%d", i);
-               if (!(ch = sr_channel_new(i, SR_CHANNEL_LOGIC, TRUE, channel_name)))
-                       return NULL;
-               sdi->channels = g_slist_append(sdi->channels, ch);
-               cg->channels = g_slist_append(cg->channels, ch);
+       if (num_logic_channels > 0) {
+               /* Logic channels, all in one channel group. */
+               cg = g_malloc0(sizeof(struct sr_channel_group));
+               cg->name = g_strdup("Logic");
+               for (i = 0; i < num_logic_channels; i++) {
+                       sprintf(channel_name, "D%d", i);
+                       ch = sr_channel_new(sdi, i, SR_CHANNEL_LOGIC, TRUE, channel_name);
+                       cg->channels = g_slist_append(cg->channels, ch);
+               }
+               sdi->channel_groups = g_slist_append(NULL, cg);
        }
-       sdi->channel_groups = g_slist_append(NULL, cg);
 
        /* Analog channels, channel groups and pattern generators. */
-       pattern = 0;
-       /* An "Analog" channel group with all analog channels in it. */
-       acg = g_malloc0(sizeof(struct sr_channel_group));
-       acg->name = g_strdup("Analog");
-       sdi->channel_groups = g_slist_append(sdi->channel_groups, acg);
-
-       devc->ch_ag = g_hash_table_new(g_direct_hash, g_direct_equal);
-       for (i = 0; i < num_analog_channels; i++) {
-               snprintf(channel_name, 16, "A%d", i);
-               ch = sr_channel_new(i + num_logic_channels, SR_CHANNEL_ANALOG,
-                               TRUE, channel_name);
-               sdi->channels = g_slist_append(sdi->channels, ch);
-               acg->channels = g_slist_append(acg->channels, ch);
-
-               /* Every analog channel gets its own channel group as well. */
-               cg = g_malloc0(sizeof(struct sr_channel_group));
-               cg->name = g_strdup(channel_name);
-               cg->channels = g_slist_append(NULL, ch);
-               sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
-
-               /* Every channel gets a generator struct. */
-               ag = g_malloc(sizeof(struct analog_gen));
-               ag->amplitude = DEFAULT_ANALOG_AMPLITUDE;
-               ag->packet.channels = cg->channels;
-               ag->packet.mq = 0;
-               ag->packet.mqflags = 0;
-               ag->packet.unit = SR_UNIT_VOLT;
-               ag->packet.data = ag->pattern_data;
-               ag->pattern = pattern;
-               g_hash_table_insert(devc->ch_ag, ch, ag);
-
-               if (++pattern == ARRAY_SIZE(analog_pattern_str))
-                       pattern = 0;
+       if (num_analog_channels > 0) {
+               pattern = 0;
+               /* An "Analog" channel group with all analog channels in it. */
+               acg = g_malloc0(sizeof(struct sr_channel_group));
+               acg->name = g_strdup("Analog");
+               sdi->channel_groups = g_slist_append(sdi->channel_groups, acg);
+
+               devc->ch_ag = g_hash_table_new(g_direct_hash, g_direct_equal);
+               for (i = 0; i < num_analog_channels; i++) {
+                       snprintf(channel_name, 16, "A%d", i);
+                       ch = sr_channel_new(sdi, i + num_logic_channels, SR_CHANNEL_ANALOG,
+                                       TRUE, channel_name);
+                       acg->channels = g_slist_append(acg->channels, ch);
+
+                       /* Every analog channel gets its own channel group as well. */
+                       cg = g_malloc0(sizeof(struct sr_channel_group));
+                       cg->name = g_strdup(channel_name);
+                       cg->channels = g_slist_append(NULL, ch);
+                       sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
+
+                       /* Every channel gets a generator struct. */
+                       ag = g_malloc(sizeof(struct analog_gen));
+                       ag->amplitude = DEFAULT_ANALOG_AMPLITUDE;
+                       ag->packet.channels = cg->channels;
+                       ag->packet.mq = 0;
+                       ag->packet.mqflags = 0;
+                       ag->packet.unit = SR_UNIT_VOLT;
+                       ag->packet.data = ag->pattern_data;
+                       ag->pattern = pattern;
+                       ag->avg_val = 0.0f;
+                       ag->num_avgs = 0;
+                       g_hash_table_insert(devc->ch_ag, ch, ag);
+
+                       if (++pattern == ARRAY_SIZE(analog_pattern_str))
+                               pattern = 0;
+               }
        }
 
        sdi->priv = devc;
@@ -361,11 +351,6 @@ static GSList *scan(GSList *options)
        return devices;
 }
 
-static GSList *dev_list(void)
-{
-       return ((struct drv_context *)(di->priv))->instances;
-}
-
 static int dev_open(struct sr_dev_inst *sdi)
 {
        sdi->status = SR_ST_ACTIVE;
@@ -396,7 +381,7 @@ static void clear_helper(void *priv)
        g_free(devc);
 }
 
-static int cleanup(void)
+static int dev_clear(const struct sr_dev_driver *di)
 {
        return std_dev_clear(di, clear_helper);
 }
@@ -423,6 +408,12 @@ static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *s
        case SR_CONF_LIMIT_MSEC:
                *data = g_variant_new_uint64(devc->limit_msec);
                break;
+       case SR_CONF_AVERAGING:
+               *data = g_variant_new_boolean(devc->avg);
+               break;
+       case SR_CONF_AVG_SAMPLES:
+               *data = g_variant_new_uint64(devc->avg_samples);
+               break;
        case SR_CONF_PATTERN_MODE:
                if (!cg)
                        return SR_ERR_CHANNEL_GROUP;
@@ -475,17 +466,22 @@ static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sd
        switch (key) {
        case SR_CONF_SAMPLERATE:
                devc->cur_samplerate = g_variant_get_uint64(data);
-               sr_dbg("Setting samplerate to %" PRIu64, devc->cur_samplerate);
                break;
        case SR_CONF_LIMIT_SAMPLES:
                devc->limit_msec = 0;
                devc->limit_samples = g_variant_get_uint64(data);
-               sr_dbg("Setting sample limit to %" PRIu64, devc->limit_samples);
                break;
        case SR_CONF_LIMIT_MSEC:
                devc->limit_msec = g_variant_get_uint64(data);
                devc->limit_samples = 0;
-               sr_dbg("Setting time limit to %" PRIu64"ms", devc->limit_msec);
+               break;
+       case SR_CONF_AVERAGING:
+               devc->avg = g_variant_get_boolean(data);
+               sr_dbg("%s averaging", devc->avg ? "Enabling" : "Disabling");
+               break;
+       case SR_CONF_AVG_SAMPLES:
+               devc->avg_samples = g_variant_get_uint64(data);
+               sr_dbg("Setting averaging rate to %" PRIu64, devc->avg_samples);
                break;
        case SR_CONF_PATTERN_MODE:
                if (!cg)
@@ -555,8 +551,6 @@ static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *
        GVariant *gvar;
        GVariantBuilder gvb;
 
-       (void)sdi;
-
        if (key == SR_CONF_SCAN_OPTIONS) {
                *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
                                scanopts, ARRAY_SIZE(scanopts), sizeof(uint32_t));
@@ -565,7 +559,7 @@ static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *
 
        if (key == SR_CONF_DEVICE_OPTIONS && !sdi) {
                *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
-                               devopts, ARRAY_SIZE(devopts), sizeof(uint32_t));
+                               drvopts, ARRAY_SIZE(drvopts), sizeof(uint32_t));
                return SR_OK;
        }
 
@@ -576,7 +570,7 @@ static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *
                switch (key) {
                case SR_CONF_DEVICE_OPTIONS:
                        *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
-                                       devopts_global, ARRAY_SIZE(devopts_global), sizeof(uint32_t));
+                                       devopts, ARRAY_SIZE(devopts), sizeof(uint32_t));
                        break;
                case SR_CONF_SAMPLERATE:
                        g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}"));
@@ -589,22 +583,31 @@ static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *
                        return SR_ERR_NA;
                }
        } else {
-               /* Any channel in the group will do. */
                ch = cg->channels->data;
                switch (key) {
                case SR_CONF_DEVICE_OPTIONS:
                        if (ch->type == SR_CHANNEL_LOGIC)
-                               *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
+                               *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
                                                devopts_cg_logic, ARRAY_SIZE(devopts_cg_logic),
                                                sizeof(uint32_t));
-                       else if (ch->type == SR_CHANNEL_ANALOG)
-                               *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
-                                               devopts_cg_analog, ARRAY_SIZE(devopts_cg_analog),
-                                               sizeof(uint32_t));
+                       else if (ch->type == SR_CHANNEL_ANALOG) {
+                               if (strcmp(cg->name, "Analog") == 0)
+                                       *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
+                                                       devopts_cg_analog_group, ARRAY_SIZE(devopts_cg_analog_group),
+                                                       sizeof(uint32_t));
+                               else
+                                       *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
+                                                       devopts_cg_analog_channel, ARRAY_SIZE(devopts_cg_analog_channel),
+                                                       sizeof(uint32_t));
+                       }
                        else
                                return SR_ERR_BUG;
                        break;
                case SR_CONF_PATTERN_MODE:
+                       /* The analog group (with all 4 channels) shall not have a pattern property. */
+                       if (strcmp(cg->name, "Analog") == 0)
+                               return SR_ERR_NA;
+
                        if (ch->type == SR_CHANNEL_LOGIC)
                                *data = g_variant_new_strv(logic_pattern_str,
                                                ARRAY_SIZE(logic_pattern_str));
@@ -663,6 +666,66 @@ static void logic_generator(struct sr_dev_inst *sdi, uint64_t size)
        }
 }
 
+static void send_analog_packet(struct analog_gen *ag,
+                              struct sr_dev_inst *sdi,
+                              uint64_t *analog_sent,
+                              uint64_t analog_pos,
+                              uint64_t analog_todo)
+{
+       struct sr_datafeed_packet packet;
+       struct dev_context *devc;
+       uint64_t sending_now, to_avg;
+       int ag_pattern_pos;
+       unsigned int i;
+
+       devc = sdi->priv;
+       packet.type = SR_DF_ANALOG_OLD;
+       packet.payload = &ag->packet;
+
+       if (!devc->avg) {
+               ag_pattern_pos = analog_pos % ag->num_samples;
+               sending_now = MIN(analog_todo, ag->num_samples-ag_pattern_pos);
+               ag->packet.data = ag->pattern_data + ag_pattern_pos;
+               ag->packet.num_samples = sending_now;
+               sr_session_send(sdi, &packet);
+
+               /* Whichever channel group gets there first. */
+               *analog_sent = MAX(*analog_sent, sending_now);
+       } else {
+               ag_pattern_pos = analog_pos % ag->num_samples;
+               to_avg = MIN(analog_todo, ag->num_samples-ag_pattern_pos);
+
+               for (i = 0; i < to_avg; i++) {
+                       ag->avg_val = (ag->avg_val +
+                                       *(ag->pattern_data +
+                                         ag_pattern_pos + i)) / 2;
+                       ag->num_avgs++;
+                       /* Time to send averaged data? */
+                       if (devc->avg_samples > 0 &&
+                           ag->num_avgs >= devc->avg_samples)
+                               goto do_send;
+               }
+
+               if (devc->avg_samples == 0) {
+                       /* We're averaging all the samples, so wait with
+                        * sending until the very end.
+                        */
+                       *analog_sent = ag->num_avgs;
+                       return;
+               }
+
+do_send:
+               ag->packet.data = &ag->avg_val;
+               ag->packet.num_samples = 1;
+
+               sr_session_send(sdi, &packet);
+               *analog_sent = ag->num_avgs;
+
+               ag->num_avgs = 0;
+               ag->avg_val = 0.0f;
+       }
+}
+
 /* Callback handling data */
 static int prepare_data(int fd, int revents, void *cb_data)
 {
@@ -671,38 +734,56 @@ static int prepare_data(int fd, int revents, void *cb_data)
        struct sr_datafeed_packet packet;
        struct sr_datafeed_logic logic;
        struct analog_gen *ag;
-       int ag_pattern_pos;
        GHashTableIter iter;
        void *value;
-       uint64_t logic_todo, analog_todo, expected_samplenum, analog_sent, sending_now;
-       int64_t time, elapsed;
+       uint64_t samples_todo, logic_done, analog_done, analog_sent, sending_now;
+       int64_t elapsed_us, limit_us, todo_us;
 
        (void)fd;
        (void)revents;
 
        sdi = cb_data;
        devc = sdi->priv;
-       logic_todo = analog_todo = 0;
 
-       /* How many samples should we have sent by now? */
-       time = g_get_monotonic_time();
-       elapsed = time - devc->starttime;
-       expected_samplenum = elapsed * devc->cur_samplerate / 1000000;
+       /* Just in case. */
+       if (devc->cur_samplerate <= 0
+                       || (devc->num_logic_channels <= 0
+                       && devc->num_analog_channels <= 0)) {
+               dev_acquisition_stop(sdi);
+               return G_SOURCE_CONTINUE;
+       }
 
-       /* But never more than the limit, if there is one. */
-       if (!devc->continuous)
-               expected_samplenum = MIN(expected_samplenum, devc->limit_samples);
+       /* What time span should we send samples for? */
+       elapsed_us = g_get_monotonic_time() - devc->start_us;
+       limit_us = 1000 * devc->limit_msec;
+       if (limit_us > 0 && limit_us < elapsed_us)
+               todo_us = MAX(0, limit_us - devc->spent_us);
+       else
+               todo_us = MAX(0, elapsed_us - devc->spent_us);
+
+       /* How many samples are outstanding since the last round? */
+       samples_todo = (todo_us * devc->cur_samplerate + G_USEC_PER_SEC - 1)
+                       / G_USEC_PER_SEC;
+       if (devc->limit_samples > 0) {
+               if (devc->limit_samples < devc->sent_samples)
+                       samples_todo = 0;
+               else if (devc->limit_samples - devc->sent_samples < samples_todo)
+                       samples_todo = devc->limit_samples - devc->sent_samples;
+       }
+       /* Calculate the actual time covered by this run back from the sample
+        * count, rounded towards zero. This avoids getting stuck on a too-low
+        * time delta with no samples being sent due to round-off.
+        */
+       todo_us = samples_todo * G_USEC_PER_SEC / devc->cur_samplerate;
 
-       /* Of those, how many do we still have to send? */
-       if (devc->num_logic_channels)
-               logic_todo = expected_samplenum - devc->logic_counter;
-       if (devc->num_analog_channels)
-               analog_todo = expected_samplenum - devc->analog_counter;
+       logic_done  = devc->num_logic_channels  > 0 ? 0 : samples_todo;
+       analog_done = devc->num_analog_channels > 0 ? 0 : samples_todo;
 
-       while (logic_todo || analog_todo) {
+       while (logic_done < samples_todo || analog_done < samples_todo) {
                /* Logic */
-               if (logic_todo > 0) {
-                       sending_now = MIN(logic_todo, LOGIC_BUFSIZE / devc->logic_unitsize);
+               if (logic_done < samples_todo) {
+                       sending_now = MIN(samples_todo - logic_done,
+                                       LOGIC_BUFSIZE / devc->logic_unitsize);
                        logic_generator(sdi, sending_now * devc->logic_unitsize);
                        packet.type = SR_DF_LOGIC;
                        packet.payload = &logic;
@@ -710,127 +791,100 @@ static int prepare_data(int fd, int revents, void *cb_data)
                        logic.unitsize = devc->logic_unitsize;
                        logic.data = devc->logic_data;
                        sr_session_send(sdi, &packet);
-                       logic_todo -= sending_now;
-                       devc->logic_counter += sending_now;
+                       logic_done += sending_now;
                }
 
                /* Analog, one channel at a time */
-               if (analog_todo > 0) {
+               if (analog_done < samples_todo) {
                        analog_sent = 0;
 
+                       g_hash_table_iter_init(&iter, devc->ch_ag);
+                       while (g_hash_table_iter_next(&iter, NULL, &value)) {
+                               send_analog_packet(value, sdi, &analog_sent,
+                                               devc->sent_samples + analog_done,
+                                               samples_todo - analog_done);
+                       }
+                       analog_done += analog_sent;
+               }
+       }
+       /* At this point, both logic_done and analog_done should be
+        * exactly equal to samples_todo, or else.
+        */
+       if (logic_done != samples_todo || analog_done != samples_todo) {
+               sr_err("BUG: Sample count mismatch.");
+               return G_SOURCE_REMOVE;
+       }
+       devc->sent_samples += samples_todo;
+       devc->spent_us += todo_us;
+
+       if ((devc->limit_samples > 0 && devc->sent_samples >= devc->limit_samples)
+                       || (limit_us > 0 && devc->spent_us >= limit_us)) {
+
+               /* If we're averaging everything - now is the time to send data */
+               if (devc->avg_samples == 0) {
                        g_hash_table_iter_init(&iter, devc->ch_ag);
                        while (g_hash_table_iter_next(&iter, NULL, &value)) {
                                ag = value;
-                               packet.type = SR_DF_ANALOG;
+                               packet.type = SR_DF_ANALOG_OLD;
                                packet.payload = &ag->packet;
-                               ag_pattern_pos = devc->analog_counter % ag->num_samples;
-                               sending_now = MIN(analog_todo, ag->num_samples-ag_pattern_pos);
-                               ag->packet.data = ag->pattern_data + ag_pattern_pos;
-                               ag->packet.num_samples = sending_now;
+                               ag->packet.data = &ag->avg_val;
+                               ag->packet.num_samples = 1;
                                sr_session_send(sdi, &packet);
-
-                               /* Whichever channel group gets there first. */
-                               analog_sent = MAX(analog_sent, sending_now);
                        }
-                       analog_todo -= analog_sent;
-                       devc->analog_counter += analog_sent;
                }
-       }
-
-       if (!devc->continuous
-                       && (!devc->num_logic_channels || devc->logic_counter >= devc->limit_samples)
-                       && (!devc->num_analog_channels || devc->analog_counter >= devc->limit_samples)) {
                sr_dbg("Requested number of samples reached.");
-               dev_acquisition_stop(sdi, cb_data);
-               return TRUE;
+               dev_acquisition_stop(sdi);
        }
 
-       return TRUE;
+       return G_SOURCE_CONTINUE;
 }
 
-static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
+static int dev_acquisition_start(const struct sr_dev_inst *sdi)
 {
        struct dev_context *devc;
        GHashTableIter iter;
        void *value;
 
-       (void)cb_data;
-
        if (sdi->status != SR_ST_ACTIVE)
                return SR_ERR_DEV_CLOSED;
 
        devc = sdi->priv;
-       devc->continuous = !devc->limit_samples;
-       devc->logic_counter = devc->analog_counter = 0;
-
-       /*
-        * Setting two channels connected by a pipe is a remnant from when the
-        * demo driver generated data in a thread, and collected and sent the
-        * data in the main program loop.
-        * They are kept here because it provides a convenient way of setting
-        * up a timeout-based polling mechanism.
-        */
-       if (pipe(devc->pipe_fds)) {
-               sr_err("%s: pipe() failed", __func__);
-               return SR_ERR;
-       }
+       devc->sent_samples = 0;
 
        g_hash_table_iter_init(&iter, devc->ch_ag);
        while (g_hash_table_iter_next(&iter, NULL, &value))
                generate_analog_pattern(value, devc->cur_samplerate);
 
-       devc->channel = g_io_channel_unix_new(devc->pipe_fds[0]);
-       g_io_channel_set_flags(devc->channel, G_IO_FLAG_NONBLOCK, NULL);
+       sr_session_source_add(sdi->session, -1, 0, 100,
+                       prepare_data, (struct sr_dev_inst *)sdi);
 
-       /* Set channel encoding to binary (default is UTF-8). */
-       g_io_channel_set_encoding(devc->channel, NULL, NULL);
-
-       /* Make channels unbuffered. */
-       g_io_channel_set_buffered(devc->channel, FALSE);
-
-       sr_session_source_add_channel(sdi->session, devc->channel,
-                       G_IO_IN | G_IO_ERR, 40, prepare_data, (void *)sdi);
-
-       /* Send header packet to the session bus. */
        std_session_send_df_header(sdi, LOG_PREFIX);
 
        /* We use this timestamp to decide how many more samples to send. */
-       devc->starttime = g_get_monotonic_time();
+       devc->start_us = g_get_monotonic_time();
+       devc->spent_us = 0;
 
        return SR_OK;
 }
 
-static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
+static int dev_acquisition_stop(struct sr_dev_inst *sdi)
 {
-       struct dev_context *devc;
-       struct sr_datafeed_packet packet;
-
-       (void)cb_data;
-
-       devc = sdi->priv;
        sr_dbg("Stopping acquisition.");
-
-       sr_session_source_remove_channel(sdi->session, devc->channel);
-       g_io_channel_shutdown(devc->channel, FALSE, NULL);
-       g_io_channel_unref(devc->channel);
-       devc->channel = NULL;
-
-       /* Send last packet. */
-       packet.type = SR_DF_END;
-       sr_session_send(sdi, &packet);
+       sr_session_source_remove(sdi->session, -1);
+       std_session_send_df_end(sdi, LOG_PREFIX);
 
        return SR_OK;
 }
 
-SR_PRIV struct sr_dev_driver demo_driver_info = {
+static struct sr_dev_driver demo_driver_info = {
        .name = "demo",
        .longname = "Demo driver and pattern generator",
        .api_version = 1,
-       .init = init,
-       .cleanup = cleanup,
+       .init = std_init,
+       .cleanup = std_cleanup,
        .scan = scan,
-       .dev_list = dev_list,
-       .dev_clear = NULL,
+       .dev_list = std_dev_list,
+       .dev_clear = dev_clear,
        .config_get = config_get,
        .config_set = config_set,
        .config_list = config_list,
@@ -838,5 +892,6 @@ SR_PRIV struct sr_dev_driver demo_driver_info = {
        .dev_close = dev_close,
        .dev_acquisition_start = dev_acquisition_start,
        .dev_acquisition_stop = dev_acquisition_stop,
-       .priv = NULL,
+       .context = NULL,
 };
+SR_REGISTER_DEV_DRIVER(demo_driver_info);