[libsigrok.git] / src / hardware / demo / api.c

/*
 * This file is part of the libsigrok project.
 *
 * 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
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <libsigrok/libsigrok.h>
#include "libsigrok-internal.h"
#include "protocol.h"

#define DEFAULT_NUM_LOGIC_CHANNELS	8
#define DEFAULT_ENABLED_LOGIC_CHANNELS	8
#define DEFAULT_LOGIC_PATTERN		PATTERN_SIGROK

#define DEFAULT_NUM_ANALOG_CHANNELS	4
#define DEFAULT_ENABLED_ANALOG_CHANNELS	4
#define DEFAULT_ANALOG_AMPLITUDE	10

/* Note: No spaces allowed because of sigrok-cli. */
static const char *logic_pattern_str[] = {
	"sigrok",
	"random",
	"incremental",
	"walking-one",
	"walking-zero",
	"all-low",
	"all-high",
	"squid",
};

static const uint32_t drvopts[] = {
	SR_CONF_DEMO_DEV,
	SR_CONF_LOGIC_ANALYZER,
	SR_CONF_OSCILLOSCOPE,
};

static const uint32_t scanopts[] = {
	SR_CONF_NUM_LOGIC_CHANNELS,
	SR_CONF_NUM_ANALOG_CHANNELS,
};

static const uint32_t devopts[] = {
	SR_CONF_CONTINUOUS,
	SR_CONF_LIMIT_SAMPLES | SR_CONF_GET | SR_CONF_SET,
	SR_CONF_LIMIT_MSEC | SR_CONF_GET | SR_CONF_SET,
	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_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,
};

static const uint64_t samplerates[] = {
	SR_HZ(1),
	SR_GHZ(1),
	SR_HZ(1),
};

static GSList *scan(struct sr_dev_driver *di, GSList *options)
{
	struct dev_context *devc;
	struct sr_dev_inst *sdi;
	struct sr_channel *ch;
	struct sr_channel_group *cg, *acg;
	struct sr_config *src;
	struct analog_gen *ag;
	GSList *l;
	int num_logic_channels, num_analog_channels, pattern, i;
	char channel_name[16];
	gboolean enabled;

	num_logic_channels = DEFAULT_NUM_LOGIC_CHANNELS;
	num_analog_channels = DEFAULT_NUM_ANALOG_CHANNELS;
	for (l = options; l; l = l->next) {
		src = l->data;
		switch (src->key) {
		case SR_CONF_NUM_LOGIC_CHANNELS:
			num_logic_channels = g_variant_get_int32(src->data);
			break;
		case SR_CONF_NUM_ANALOG_CHANNELS:
			num_analog_channels = g_variant_get_int32(src->data);
			break;
		}
	}

	sdi = g_malloc0(sizeof(struct sr_dev_inst));
	sdi->status = SR_ST_INACTIVE;
	sdi->model = g_strdup("Demo device");

	devc = g_malloc0(sizeof(struct dev_context));
	devc->cur_samplerate = SR_KHZ(200);
	devc->num_logic_channels = num_logic_channels;
	devc->logic_unitsize = (devc->num_logic_channels + 7) / 8;
	devc->logic_pattern = DEFAULT_LOGIC_PATTERN;
	devc->num_analog_channels = num_analog_channels;

	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);
			enabled = (i < DEFAULT_ENABLED_LOGIC_CHANNELS) ? TRUE : FALSE;
			ch = sr_channel_new(sdi, i, SR_CHANNEL_LOGIC, enabled, channel_name);
			cg->channels = g_slist_append(cg->channels, ch);
		}
		sdi->channel_groups = g_slist_append(NULL, cg);
	}

	/* Analog channels, channel groups and pattern generators. */
	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);
			enabled = (i < DEFAULT_ENABLED_ANALOG_CHANNELS) ? TRUE : FALSE;
			ch = sr_channel_new(sdi, i + num_logic_channels, SR_CHANNEL_ANALOG,
					enabled, 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;
			sr_analog_init(&ag->packet, &ag->encoding, &ag->meaning, &ag->spec, 2);
			ag->packet.meaning->channels = cg->channels;
			ag->packet.meaning->mq = 0;
			ag->packet.meaning->mqflags = 0;
			ag->packet.meaning->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;

	return std_scan_complete(di, g_slist_append(NULL, sdi));
}

static int dev_open(struct sr_dev_inst *sdi)
{
	sdi->status = SR_ST_ACTIVE;

	return SR_OK;
}

static int dev_close(struct sr_dev_inst *sdi)
{
	sdi->status = SR_ST_INACTIVE;

	return SR_OK;
}

static void clear_helper(void *priv)
{
	struct dev_context *devc;
	GHashTableIter iter;
	void *value;

	devc = priv;

	/* Analog generators. */
	g_hash_table_iter_init(&iter, devc->ch_ag);
	while (g_hash_table_iter_next(&iter, NULL, &value))
		g_free(value);
	g_hash_table_unref(devc->ch_ag);
	g_free(devc);
}

static int dev_clear(const struct sr_dev_driver *di)
{
	return std_dev_clear(di, clear_helper);
}

static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
		const struct sr_channel_group *cg)
{
	struct dev_context *devc;
	struct sr_channel *ch;
	struct analog_gen *ag;
	int pattern;

	if (!sdi)
		return SR_ERR_ARG;

	devc = sdi->priv;
	switch (key) {
	case SR_CONF_SAMPLERATE:
		*data = g_variant_new_uint64(devc->cur_samplerate);
		break;
	case SR_CONF_LIMIT_SAMPLES:
		*data = g_variant_new_uint64(devc->limit_samples);
		break;
	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;
		/* Any channel in the group will do. */
		ch = cg->channels->data;
		if (ch->type == SR_CHANNEL_LOGIC) {
			pattern = devc->logic_pattern;
			*data = g_variant_new_string(logic_pattern_str[pattern]);
		} else if (ch->type == SR_CHANNEL_ANALOG) {
			ag = g_hash_table_lookup(devc->ch_ag, ch);
			pattern = ag->pattern;
			*data = g_variant_new_string(analog_pattern_str[pattern]);
		} else
			return SR_ERR_BUG;
		break;
	case SR_CONF_AMPLITUDE:
		if (!cg)
			return SR_ERR_CHANNEL_GROUP;
		/* Any channel in the group will do. */
		ch = cg->channels->data;
		if (ch->type != SR_CHANNEL_ANALOG)
			return SR_ERR_ARG;
		ag = g_hash_table_lookup(devc->ch_ag, ch);
		*data = g_variant_new_double(ag->amplitude);
		break;
	default:
		return SR_ERR_NA;
	}

	return SR_OK;
}

static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sdi,
		const struct sr_channel_group *cg)
{
	struct dev_context *devc;
	struct analog_gen *ag;
	struct sr_channel *ch;
	GSList *l;
	int logic_pattern, analog_pattern, ret;
	unsigned int i;
	const char *stropt;

	devc = sdi->priv;

	if (sdi->status != SR_ST_ACTIVE)
		return SR_ERR_DEV_CLOSED;

	ret = SR_OK;
	switch (key) {
	case SR_CONF_SAMPLERATE:
		devc->cur_samplerate = g_variant_get_uint64(data);
		break;
	case SR_CONF_LIMIT_SAMPLES:
		devc->limit_msec = 0;
		devc->limit_samples = g_variant_get_uint64(data);
		break;
	case SR_CONF_LIMIT_MSEC:
		devc->limit_msec = g_variant_get_uint64(data);
		devc->limit_samples = 0;
		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)
			return SR_ERR_CHANNEL_GROUP;
		stropt = g_variant_get_string(data, NULL);
		logic_pattern = analog_pattern = -1;
		for (i = 0; i < ARRAY_SIZE(logic_pattern_str); i++) {
			if (!strcmp(stropt, logic_pattern_str[i])) {
				logic_pattern = i;
				break;
			}
		}
		for (i = 0; i < ARRAY_SIZE(analog_pattern_str); i++) {
			if (!strcmp(stropt, analog_pattern_str[i])) {
				analog_pattern = i;
				break;
			}
		}
		if (logic_pattern == -1 && analog_pattern == -1)
			return SR_ERR_ARG;
		for (l = cg->channels; l; l = l->next) {
			ch = l->data;
			if (ch->type == SR_CHANNEL_LOGIC) {
				if (logic_pattern == -1)
					return SR_ERR_ARG;
				sr_dbg("Setting logic pattern to %s",
						logic_pattern_str[logic_pattern]);
				devc->logic_pattern = logic_pattern;
				/* Might as well do this now, these are static. */
				if (logic_pattern == PATTERN_ALL_LOW)
					memset(devc->logic_data, 0x00, LOGIC_BUFSIZE);
				else if (logic_pattern == PATTERN_ALL_HIGH)
					memset(devc->logic_data, 0xff, LOGIC_BUFSIZE);
			} else if (ch->type == SR_CHANNEL_ANALOG) {
				if (analog_pattern == -1)
					return SR_ERR_ARG;
				sr_dbg("Setting analog pattern for channel %s to %s",
						ch->name, analog_pattern_str[analog_pattern]);
				ag = g_hash_table_lookup(devc->ch_ag, ch);
				ag->pattern = analog_pattern;
			} else
				return SR_ERR_BUG;
		}
		break;
	case SR_CONF_AMPLITUDE:
		if (!cg)
			return SR_ERR_CHANNEL_GROUP;
		for (l = cg->channels; l; l = l->next) {
			ch = l->data;
			if (ch->type != SR_CHANNEL_ANALOG)
				return SR_ERR_ARG;
			ag = g_hash_table_lookup(devc->ch_ag, ch);
			ag->amplitude = g_variant_get_double(data);
		}
		break;
	default:
		ret = SR_ERR_NA;
	}

	return ret;
}

static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
		const struct sr_channel_group *cg)
{
	struct sr_channel *ch;
	GVariant *gvar;
	GVariantBuilder gvb;

	if (key == SR_CONF_SCAN_OPTIONS) {
		*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
				scanopts, ARRAY_SIZE(scanopts), sizeof(uint32_t));
		return SR_OK;
	}

	if (key == SR_CONF_DEVICE_OPTIONS && !sdi) {
		*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
				drvopts, ARRAY_SIZE(drvopts), sizeof(uint32_t));
		return SR_OK;
	}

	if (!sdi)
		return SR_ERR_ARG;

	if (!cg) {
		switch (key) {
		case SR_CONF_DEVICE_OPTIONS:
			*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
					devopts, ARRAY_SIZE(devopts), sizeof(uint32_t));
			break;
		case SR_CONF_SAMPLERATE:
			g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}"));
			gvar = g_variant_new_fixed_array(G_VARIANT_TYPE("t"), samplerates,
					ARRAY_SIZE(samplerates), sizeof(uint64_t));
			g_variant_builder_add(&gvb, "{sv}", "samplerate-steps", gvar);
			*data = g_variant_builder_end(&gvb);
			break;
		default:
			return SR_ERR_NA;
		}
	} else {
		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_UINT32,
						devopts_cg_logic, ARRAY_SIZE(devopts_cg_logic),
						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));
			else if (ch->type == SR_CHANNEL_ANALOG)
				*data = g_variant_new_strv(analog_pattern_str,
						ARRAY_SIZE(analog_pattern_str));
			else
				return SR_ERR_BUG;
			break;
		default:
			return SR_ERR_NA;
		}
	}

	return SR_OK;
}

static int dev_acquisition_start(const struct sr_dev_inst *sdi)
{
	struct dev_context *devc;
	GHashTableIter iter;
	void *value;

	if (sdi->status != SR_ST_ACTIVE)
		return SR_ERR_DEV_CLOSED;

	devc = sdi->priv;
	devc->sent_samples = 0;

	g_hash_table_iter_init(&iter, devc->ch_ag);
	while (g_hash_table_iter_next(&iter, NULL, &value))
		demo_generate_analog_pattern(value, devc->cur_samplerate);

	sr_session_source_add(sdi->session, -1, 0, 100,
			demo_prepare_data, (struct sr_dev_inst *)sdi);

	std_session_send_df_header(sdi);

	/* We use this timestamp to decide how many more samples to send. */
	devc->start_us = g_get_monotonic_time();
	devc->spent_us = 0;
	devc->step = 0;

	return SR_OK;
}

static int dev_acquisition_stop(struct sr_dev_inst *sdi)
{
	sr_dbg("Stopping acquisition.");
	sr_session_source_remove(sdi->session, -1);
	std_session_send_df_end(sdi);

	return SR_OK;
}

static struct sr_dev_driver demo_driver_info = {
	.name = "demo",
	.longname = "Demo driver and pattern generator",
	.api_version = 1,
	.init = std_init,
	.cleanup = std_cleanup,
	.scan = scan,
	.dev_list = std_dev_list,
	.dev_clear = dev_clear,
	.config_get = config_get,
	.config_set = config_set,
	.config_list = config_list,
	.dev_open = dev_open,
	.dev_close = dev_close,
	.dev_acquisition_start = dev_acquisition_start,
	.dev_acquisition_stop = dev_acquisition_stop,
	.context = NULL,
};
SR_REGISTER_DEV_DRIVER(demo_driver_info);
Commit	Line	Data
6239c175	1	/*
50985c20	2	* This file is part of the libsigrok project.
6239c175 UH	3	*
6239c175 UH	4	* Copyright (C) 2010 Uwe Hermann <uwe@hermann-uwe.de>
fc96e6f8	5	* Copyright (C) 2011 Olivier Fauchon <olivier@aixmarseille.com>
c216d623	6	* Copyright (C) 2012 Alexandru Gagniuc <mr.nuke.me@gmail.com>
7a8a1aba	7	* Copyright (C) 2015 Bartosz Golaszewski <bgolaszewski@baylibre.com>
6239c175 UH	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
2ea1fdf1	20	* along with this program; if not, see <http://www.gnu.org/licenses/>.
6239c175 UH	21	*/
6239c175 UH	22
6ec6c43b	23	#include <config.h>
6239c175 UH	24	#include <stdlib.h>
6239c175 UH	25	#include <string.h>
4374219b	26	#include <math.h>
c1aae900	27	#include <libsigrok/libsigrok.h>
45c59c8b	28	#include "libsigrok-internal.h"
ba508e22	29	#include "protocol.h"
92bcedf6	30
7db90279	31	#define DEFAULT_NUM_LOGIC_CHANNELS 8
b1e6eec6 GS	32	#define DEFAULT_ENABLED_LOGIC_CHANNELS 8
b1e6eec6 GS	33	#define DEFAULT_LOGIC_PATTERN PATTERN_SIGROK
c03ed397	34
b1e6eec6 GS	35	#define DEFAULT_NUM_ANALOG_CHANNELS 4
b1e6eec6 GS	36	#define DEFAULT_ENABLED_ANALOG_CHANNELS 4
d9251a2c	37	#define DEFAULT_ANALOG_AMPLITUDE 10
85b5af06	38
77463bd3	39	/* Note: No spaces allowed because of sigrok-cli. */
8b2d41ed	40	static const char *logic_pattern_str[] = {
61c39f54 BV	41	"sigrok",
	42	"random",
	43	"incremental",
77463bd3 SA	44	"walking-one",
77463bd3 SA	45	"walking-zero",
61c39f54 BV	46	"all-low",
61c39f54 BV	47	"all-high",
81d53a29	48	"squid",
61c39f54 BV	49	};
61c39f54 BV	50
1e4a7cac BV	51	static const uint32_t drvopts[] = {
	52	SR_CONF_DEMO_DEV,
	53	SR_CONF_LOGIC_ANALYZER,
	54	SR_CONF_OSCILLOSCOPE,
	55	};
	56
584560f1	57	static const uint32_t scanopts[] = {
3f239f08 UH	58	SR_CONF_NUM_LOGIC_CHANNELS,
3f239f08 UH	59	SR_CONF_NUM_ANALOG_CHANNELS,
e15f48c2	60	};
85b5af06	61
390795c0	62	static const uint32_t devopts[] = {
e91bb0a6	63	SR_CONF_CONTINUOUS,
5827f61b BV	64	SR_CONF_LIMIT_SAMPLES \| SR_CONF_GET \| SR_CONF_SET,
5827f61b BV	65	SR_CONF_LIMIT_MSEC \| SR_CONF_GET \| SR_CONF_SET,
390795c0	66	SR_CONF_SAMPLERATE \| SR_CONF_GET \| SR_CONF_SET \| SR_CONF_LIST,
7a8a1aba BG	67	SR_CONF_AVERAGING \| SR_CONF_GET \| SR_CONF_SET,
7a8a1aba BG	68	SR_CONF_AVG_SAMPLES \| SR_CONF_GET \| SR_CONF_SET,
390795c0 BV	69	};
	70
	71	static const uint32_t devopts_cg_logic[] = {
f12d9979	72	SR_CONF_PATTERN_MODE \| SR_CONF_GET \| SR_CONF_SET \| SR_CONF_LIST,
49224c28 BV	73	};
49224c28 BV	74
e2b99f04 SA	75	static const uint32_t devopts_cg_analog_group[] = {
	76	SR_CONF_AMPLITUDE \| SR_CONF_GET \| SR_CONF_SET,
	77	};
	78
	79	static const uint32_t devopts_cg_analog_channel[] = {
f12d9979 BV	80	SR_CONF_PATTERN_MODE \| SR_CONF_GET \| SR_CONF_SET \| SR_CONF_LIST,
f12d9979 BV	81	SR_CONF_AMPLITUDE \| SR_CONF_GET \| SR_CONF_SET,
7a1da331 BV	82	};
7a1da331 BV	83
d00088ca BV	84	static const uint64_t samplerates[] = {
	85	SR_HZ(1),
	86	SR_GHZ(1),
	87	SR_HZ(1),
4bfbf9fc BV	88	};
4bfbf9fc BV	89
4f840ce9	90	static GSList scan(struct sr_dev_driver di, GSList *options)
6239c175	91	{
33ef7573	92	struct dev_context *devc;
c07f60e7	93	struct sr_dev_inst *sdi;
ba7dd8bb	94	struct sr_channel *ch;
49224c28	95	struct sr_channel_group cg, acg;
c07f60e7	96	struct sr_config *src;
8b2d41ed	97	struct analog_gen *ag;
43376f33	98	GSList *l;
ba7dd8bb UH	99	int num_logic_channels, num_analog_channels, pattern, i;
ba7dd8bb UH	100	char channel_name[16];
b1e6eec6	101	gboolean enabled;
067d0716	102
3f239f08 UH	103	num_logic_channels = DEFAULT_NUM_LOGIC_CHANNELS;
3f239f08 UH	104	num_analog_channels = DEFAULT_NUM_ANALOG_CHANNELS;
c07f60e7 BV	105	for (l = options; l; l = l->next) {
	106	src = l->data;
	107	switch (src->key) {
3f239f08	108	case SR_CONF_NUM_LOGIC_CHANNELS:
ba7dd8bb	109	num_logic_channels = g_variant_get_int32(src->data);
c07f60e7	110	break;
3f239f08	111	case SR_CONF_NUM_ANALOG_CHANNELS:
ba7dd8bb	112	num_analog_channels = g_variant_get_int32(src->data);
c07f60e7 BV	113	break;
	114	}
	115	}
85b5af06	116
aac29cc1	117	sdi = g_malloc0(sizeof(struct sr_dev_inst));
45884333	118	sdi->status = SR_ST_INACTIVE;
4b664cd6	119	sdi->model = g_strdup("Demo device");
e15f48c2	120
a49a320d	121	devc = g_malloc0(sizeof(struct dev_context));
8b2d41ed	122	devc->cur_samplerate = SR_KHZ(200);
ba7dd8bb UH	123	devc->num_logic_channels = num_logic_channels;
ba7dd8bb UH	124	devc->logic_unitsize = (devc->num_logic_channels + 7) / 8;
b1e6eec6	125	devc->logic_pattern = DEFAULT_LOGIC_PATTERN;
ba7dd8bb	126	devc->num_analog_channels = num_analog_channels;
8b2d41ed	127
f18e0db3 LPC	128	if (num_logic_channels > 0) {
	129	/* Logic channels, all in one channel group. */
	130	cg = g_malloc0(sizeof(struct sr_channel_group));
	131	cg->name = g_strdup("Logic");
	132	for (i = 0; i < num_logic_channels; i++) {
	133	sprintf(channel_name, "D%d", i);
b1e6eec6 GS	134	enabled = (i < DEFAULT_ENABLED_LOGIC_CHANNELS) ? TRUE : FALSE;
b1e6eec6 GS	135	ch = sr_channel_new(sdi, i, SR_CHANNEL_LOGIC, enabled, channel_name);
f18e0db3 LPC	136	cg->channels = g_slist_append(cg->channels, ch);
	137	}
	138	sdi->channel_groups = g_slist_append(NULL, cg);
87ca93c5 BV	139	}
87ca93c5 BV	140
ba7dd8bb	141	/* Analog channels, channel groups and pattern generators. */
f18e0db3 LPC	142	if (num_analog_channels > 0) {
	143	pattern = 0;
	144	/* An "Analog" channel group with all analog channels in it. */
	145	acg = g_malloc0(sizeof(struct sr_channel_group));
	146	acg->name = g_strdup("Analog");
	147	sdi->channel_groups = g_slist_append(sdi->channel_groups, acg);
	148
	149	devc->ch_ag = g_hash_table_new(g_direct_hash, g_direct_equal);
	150	for (i = 0; i < num_analog_channels; i++) {
	151	snprintf(channel_name, 16, "A%d", i);
b1e6eec6	152	enabled = (i < DEFAULT_ENABLED_ANALOG_CHANNELS) ? TRUE : FALSE;
f18e0db3	153	ch = sr_channel_new(sdi, i + num_logic_channels, SR_CHANNEL_ANALOG,
b1e6eec6	154	enabled, channel_name);
f18e0db3 LPC	155	acg->channels = g_slist_append(acg->channels, ch);
	156
	157	/* Every analog channel gets its own channel group as well. */
	158	cg = g_malloc0(sizeof(struct sr_channel_group));
	159	cg->name = g_strdup(channel_name);
	160	cg->channels = g_slist_append(NULL, ch);
	161	sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
	162
	163	/* Every channel gets a generator struct. */
	164	ag = g_malloc(sizeof(struct analog_gen));
	165	ag->amplitude = DEFAULT_ANALOG_AMPLITUDE;
3be044aa	166	sr_analog_init(&ag->packet, &ag->encoding, &ag->meaning, &ag->spec, 2);
4b770103 UH	167	ag->packet.meaning->channels = cg->channels;
	168	ag->packet.meaning->mq = 0;
	169	ag->packet.meaning->mqflags = 0;
	170	ag->packet.meaning->unit = SR_UNIT_VOLT;
f18e0db3 LPC	171	ag->packet.data = ag->pattern_data;
	172	ag->pattern = pattern;
	173	ag->avg_val = 0.0f;
	174	ag->num_avgs = 0;
	175	g_hash_table_insert(devc->ch_ag, ch, ag);
	176
	177	if (++pattern == ARRAY_SIZE(analog_pattern_str))
	178	pattern = 0;
	179	}
33ef7573	180	}
33ef7573 JH	181
	182	sdi->priv = devc;
	183
43376f33	184	return std_scan_complete(di, g_slist_append(NULL, sdi));
6239c175 UH	185	}
6239c175 UH	186
6078d2c9	187	static int dev_open(struct sr_dev_inst *sdi)
6239c175	188	{
e73ffd42	189	sdi->status = SR_ST_ACTIVE;
697785d1	190
e46b8fb1	191	return SR_OK;
6239c175 UH	192	}
6239c175 UH	193
6078d2c9	194	static int dev_close(struct sr_dev_inst *sdi)
6239c175	195	{
decfe89d	196	sdi->status = SR_ST_INACTIVE;
697785d1 UH	197
697785d1 UH	198	return SR_OK;
6239c175 UH	199	}
6239c175 UH	200
ed0b7fed BV	201	static void clear_helper(void *priv)
	202	{
	203	struct dev_context *devc;
49224c28 BV	204	GHashTableIter iter;
49224c28 BV	205	void *value;
ed0b7fed BV	206
ed0b7fed BV	207	devc = priv;
49224c28 BV	208
	209	/* Analog generators. */
	210	g_hash_table_iter_init(&iter, devc->ch_ag);
	211	while (g_hash_table_iter_next(&iter, NULL, &value))
	212	g_free(value);
	213	g_hash_table_unref(devc->ch_ag);
ed0b7fed BV	214	g_free(devc);
	215	}
	216
6ab68731	217	static int dev_clear(const struct sr_dev_driver *di)
6239c175	218	{
ed0b7fed	219	return std_dev_clear(di, clear_helper);
6239c175 UH	220	}
6239c175 UH	221
584560f1	222	static int config_get(uint32_t key, GVariant *data, const struct sr_dev_inst sdi,
53b4680f	223	const struct sr_channel_group *cg)
6239c175	224	{
c07f60e7	225	struct dev_context *devc;
ba7dd8bb	226	struct sr_channel *ch;
2388ae86 BV	227	struct analog_gen *ag;
2388ae86 BV	228	int pattern;
6f57fd96	229
2388ae86 BV	230	if (!sdi)
2388ae86 BV	231	return SR_ERR_ARG;
8f996b89	232
c07f60e7	233	devc = sdi->priv;
584560f1	234	switch (key) {
123e1313	235	case SR_CONF_SAMPLERATE:
a7684294	236	*data = g_variant_new_uint64(devc->cur_samplerate);
6239c175	237	break;
2474d87e	238	case SR_CONF_LIMIT_SAMPLES:
a7684294	239	*data = g_variant_new_uint64(devc->limit_samples);
2474d87e BV	240	break;
2474d87e BV	241	case SR_CONF_LIMIT_MSEC:
a7684294	242	*data = g_variant_new_uint64(devc->limit_msec);
2474d87e	243	break;
7a8a1aba BG	244	case SR_CONF_AVERAGING:
	245	*data = g_variant_new_boolean(devc->avg);
	246	break;
	247	case SR_CONF_AVG_SAMPLES:
	248	*data = g_variant_new_uint64(devc->avg_samples);
	249	break;
2474d87e	250	case SR_CONF_PATTERN_MODE:
53b4680f	251	if (!cg)
660e398f	252	return SR_ERR_CHANNEL_GROUP;
49224c28	253	/* Any channel in the group will do. */
ba7dd8bb	254	ch = cg->channels->data;
3f239f08	255	if (ch->type == SR_CHANNEL_LOGIC) {
2388ae86 BV	256	pattern = devc->logic_pattern;
2388ae86 BV	257	*data = g_variant_new_string(logic_pattern_str[pattern]);
3f239f08	258	} else if (ch->type == SR_CHANNEL_ANALOG) {
49224c28	259	ag = g_hash_table_lookup(devc->ch_ag, ch);
2388ae86 BV	260	pattern = ag->pattern;
	261	*data = g_variant_new_string(analog_pattern_str[pattern]);
	262	} else
	263	return SR_ERR_BUG;
c07f60e7	264	break;
dddabe37 BV	265	case SR_CONF_AMPLITUDE:
	266	if (!cg)
	267	return SR_ERR_CHANNEL_GROUP;
49224c28	268	/* Any channel in the group will do. */
dddabe37 BV	269	ch = cg->channels->data;
	270	if (ch->type != SR_CHANNEL_ANALOG)
	271	return SR_ERR_ARG;
49224c28	272	ag = g_hash_table_lookup(devc->ch_ag, ch);
dddabe37 BV	273	*data = g_variant_new_double(ag->amplitude);
dddabe37 BV	274	break;
7dfcf010	275	default:
bd6fbf62	276	return SR_ERR_NA;
6239c175 UH	277	}
6239c175 UH	278
dfb0fa1a	279	return SR_OK;
6239c175 UH	280	}
6239c175 UH	281
584560f1	282	static int config_set(uint32_t key, GVariant data, const struct sr_dev_inst sdi,
53b4680f	283	const struct sr_channel_group *cg)
6239c175	284	{
8b2d41ed	285	struct dev_context *devc;
4374219b	286	struct analog_gen *ag;
ba7dd8bb	287	struct sr_channel *ch;
49224c28 BV	288	GSList *l;
49224c28 BV	289	int logic_pattern, analog_pattern, ret;
61c39f54	290	unsigned int i;
1b79df2f	291	const char *stropt;
6239c175	292
8b2d41ed	293	devc = sdi->priv;
6239c175	294
e73ffd42 BV	295	if (sdi->status != SR_ST_ACTIVE)
	296	return SR_ERR_DEV_CLOSED;
	297
2388ae86	298	ret = SR_OK;
584560f1	299	switch (key) {
2388ae86	300	case SR_CONF_SAMPLERATE:
a7684294	301	devc->cur_samplerate = g_variant_get_uint64(data);
2388ae86 BV	302	break;
2388ae86 BV	303	case SR_CONF_LIMIT_SAMPLES:
a7684294 JH	304	devc->limit_msec = 0;
a7684294 JH	305	devc->limit_samples = g_variant_get_uint64(data);
2388ae86 BV	306	break;
2388ae86 BV	307	case SR_CONF_LIMIT_MSEC:
a7684294 JH	308	devc->limit_msec = g_variant_get_uint64(data);
a7684294 JH	309	devc->limit_samples = 0;
2388ae86	310	break;
7a8a1aba BG	311	case SR_CONF_AVERAGING:
	312	devc->avg = g_variant_get_boolean(data);
	313	sr_dbg("%s averaging", devc->avg ? "Enabling" : "Disabling");
	314	break;
	315	case SR_CONF_AVG_SAMPLES:
	316	devc->avg_samples = g_variant_get_uint64(data);
	317	sr_dbg("Setting averaging rate to %" PRIu64, devc->avg_samples);
	318	break;
2388ae86	319	case SR_CONF_PATTERN_MODE:
53b4680f	320	if (!cg)
660e398f	321	return SR_ERR_CHANNEL_GROUP;
d00088ca	322	stropt = g_variant_get_string(data, NULL);
49224c28 BV	323	logic_pattern = analog_pattern = -1;
	324	for (i = 0; i < ARRAY_SIZE(logic_pattern_str); i++) {
	325	if (!strcmp(stropt, logic_pattern_str[i])) {
	326	logic_pattern = i;
	327	break;
8b2d41ed	328	}
49224c28 BV	329	}
	330	for (i = 0; i < ARRAY_SIZE(analog_pattern_str); i++) {
	331	if (!strcmp(stropt, analog_pattern_str[i])) {
	332	analog_pattern = i;
	333	break;
8b2d41ed	334	}
49224c28 BV	335	}
	336	if (logic_pattern == -1 && analog_pattern == -1)
	337	return SR_ERR_ARG;
	338	for (l = cg->channels; l; l = l->next) {
	339	ch = l->data;
	340	if (ch->type == SR_CHANNEL_LOGIC) {
	341	if (logic_pattern == -1)
	342	return SR_ERR_ARG;
	343	sr_dbg("Setting logic pattern to %s",
	344	logic_pattern_str[logic_pattern]);
	345	devc->logic_pattern = logic_pattern;
	346	/* Might as well do this now, these are static. */
	347	if (logic_pattern == PATTERN_ALL_LOW)
	348	memset(devc->logic_data, 0x00, LOGIC_BUFSIZE);
	349	else if (logic_pattern == PATTERN_ALL_HIGH)
	350	memset(devc->logic_data, 0xff, LOGIC_BUFSIZE);
	351	} else if (ch->type == SR_CHANNEL_ANALOG) {
	352	if (analog_pattern == -1)
	353	return SR_ERR_ARG;
	354	sr_dbg("Setting analog pattern for channel %s to %s",
	355	ch->name, analog_pattern_str[analog_pattern]);
	356	ag = g_hash_table_lookup(devc->ch_ag, ch);
	357	ag->pattern = analog_pattern;
	358	} else
	359	return SR_ERR_BUG;
	360	}
2388ae86	361	break;
dddabe37 BV	362	case SR_CONF_AMPLITUDE:
	363	if (!cg)
	364	return SR_ERR_CHANNEL_GROUP;
49224c28 BV	365	for (l = cg->channels; l; l = l->next) {
	366	ch = l->data;
	367	if (ch->type != SR_CHANNEL_ANALOG)
	368	return SR_ERR_ARG;
	369	ag = g_hash_table_lookup(devc->ch_ag, ch);
	370	ag->amplitude = g_variant_get_double(data);
	371	}
dddabe37	372	break;
2388ae86	373	default:
bd6fbf62	374	ret = SR_ERR_NA;
6239c175 UH	375	}
	376
	377	return ret;
	378	}
	379
584560f1	380	static int config_list(uint32_t key, GVariant *data, const struct sr_dev_inst sdi,
53b4680f	381	const struct sr_channel_group *cg)
a1c743fc	382	{
ba7dd8bb	383	struct sr_channel *ch;
d00088ca BV	384	GVariant *gvar;
d00088ca BV	385	GVariantBuilder gvb;
a1c743fc	386
7a1da331	387	if (key == SR_CONF_SCAN_OPTIONS) {
584560f1 BV	388	*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
584560f1 BV	389	scanopts, ARRAY_SIZE(scanopts), sizeof(uint32_t));
7a1da331 BV	390	return SR_OK;
	391	}
	392
390795c0 BV	393	if (key == SR_CONF_DEVICE_OPTIONS && !sdi) {
390795c0 BV	394	*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
1e4a7cac	395	drvopts, ARRAY_SIZE(drvopts), sizeof(uint32_t));
390795c0 BV	396	return SR_OK;
	397	}
	398
7a1da331 BV	399	if (!sdi)
	400	return SR_ERR_ARG;
	401
53b4680f	402	if (!cg) {
7a1da331 BV	403	switch (key) {
7a1da331 BV	404	case SR_CONF_DEVICE_OPTIONS:
584560f1	405	*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
1e4a7cac	406	devopts, ARRAY_SIZE(devopts), sizeof(uint32_t));
7a1da331 BV	407	break;
	408	case SR_CONF_SAMPLERATE:
	409	g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}"));
	410	gvar = g_variant_new_fixed_array(G_VARIANT_TYPE("t"), samplerates,
	411	ARRAY_SIZE(samplerates), sizeof(uint64_t));
	412	g_variant_builder_add(&gvb, "{sv}", "samplerate-steps", gvar);
	413	*data = g_variant_builder_end(&gvb);
	414	break;
	415	default:
	416	return SR_ERR_NA;
	417	}
	418	} else {
ba7dd8bb	419	ch = cg->channels->data;
7a1da331 BV	420	switch (key) {
7a1da331 BV	421	case SR_CONF_DEVICE_OPTIONS:
49224c28	422	if (ch->type == SR_CHANNEL_LOGIC)
8249889d	423	*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
49224c28	424	devopts_cg_logic, ARRAY_SIZE(devopts_cg_logic),
584560f1	425	sizeof(uint32_t));
e2b99f04 SA	426	else if (ch->type == SR_CHANNEL_ANALOG) {
	427	if (strcmp(cg->name, "Analog") == 0)
	428	*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
	429	devopts_cg_analog_group, ARRAY_SIZE(devopts_cg_analog_group),
	430	sizeof(uint32_t));
	431	else
	432	*data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
	433	devopts_cg_analog_channel, ARRAY_SIZE(devopts_cg_analog_channel),
	434	sizeof(uint32_t));
	435	}
49224c28 BV	436	else
49224c28 BV	437	return SR_ERR_BUG;
7a1da331 BV	438	break;
7a1da331 BV	439	case SR_CONF_PATTERN_MODE:
e2b99f04 SA	440	/* The analog group (with all 4 channels) shall not have a pattern property. */
	441	if (strcmp(cg->name, "Analog") == 0)
	442	return SR_ERR_NA;
	443
3f239f08	444	if (ch->type == SR_CHANNEL_LOGIC)
7a1da331 BV	445	*data = g_variant_new_strv(logic_pattern_str,
7a1da331 BV	446	ARRAY_SIZE(logic_pattern_str));
3f239f08	447	else if (ch->type == SR_CHANNEL_ANALOG)
7a1da331 BV	448	*data = g_variant_new_strv(analog_pattern_str,
7a1da331 BV	449	ARRAY_SIZE(analog_pattern_str));
2388ae86 BV	450	else
2388ae86 BV	451	return SR_ERR_BUG;
7a1da331 BV	452	break;
	453	default:
	454	return SR_ERR_NA;
	455	}
a1c743fc BV	456	}
	457
	458	return SR_OK;
	459	}
	460
695dc859	461	static int dev_acquisition_start(const struct sr_dev_inst *sdi)
6239c175	462	{
61c39f54	463	struct dev_context *devc;
49224c28 BV	464	GHashTableIter iter;
49224c28 BV	465	void *value;
85b5af06	466
e73ffd42 BV	467	if (sdi->status != SR_ST_ACTIVE)
	468	return SR_ERR_DEV_CLOSED;
	469
61c39f54	470	devc = sdi->priv;
a49a320d	471	devc->sent_samples = 0;
85b5af06	472
49224c28 BV	473	g_hash_table_iter_init(&iter, devc->ch_ag);
49224c28 BV	474	while (g_hash_table_iter_next(&iter, NULL, &value))
ba508e22	475	demo_generate_analog_pattern(value, devc->cur_samplerate);
4374219b	476
98c01fe1	477	sr_session_source_add(sdi->session, -1, 0, 100,
ba508e22	478	demo_prepare_data, (struct sr_dev_inst *)sdi);
85b5af06	479
bee2b016	480	std_session_send_df_header(sdi);
f366e86c	481
3b203673	482	/* We use this timestamp to decide how many more samples to send. */
a49a320d DE	483	devc->start_us = g_get_monotonic_time();
a49a320d DE	484	devc->spent_us = 0;
471ac344	485	devc->step = 0;
3b203673	486
e46b8fb1	487	return SR_OK;
6239c175 UH	488	}
6239c175 UH	489
695dc859	490	static int dev_acquisition_stop(struct sr_dev_inst *sdi)
6239c175	491	{
a84f6ad3	492	sr_dbg("Stopping acquisition.");
027bf077	493	sr_session_source_remove(sdi->session, -1);
bee2b016	494	std_session_send_df_end(sdi);
7fd3e859	495
3010f21c	496	return SR_OK;
6239c175 UH	497	}
6239c175 UH	498
dd5c48a6	499	static struct sr_dev_driver demo_driver_info = {
e519ba86 UH	500	.name = "demo",
	501	.longname = "Demo driver and pattern generator",
	502	.api_version = 1,
c2fdcc25	503	.init = std_init,
700d6b64	504	.cleanup = std_cleanup,
6078d2c9	505	.scan = scan,
c01bf34c	506	.dev_list = std_dev_list,
6ab68731	507	.dev_clear = dev_clear,
035a1078 BV	508	.config_get = config_get,
035a1078 BV	509	.config_set = config_set,
a1c743fc	510	.config_list = config_list,
6078d2c9 UH	511	.dev_open = dev_open,
	512	.dev_close = dev_close,
	513	.dev_acquisition_start = dev_acquisition_start,
	514	.dev_acquisition_stop = dev_acquisition_stop,
41812aca	515	.context = NULL,
6239c175	516	};
dd5c48a6	517	SR_REGISTER_DEV_DRIVER(demo_driver_info);