[libsigrok.git] / hardware / uni-t-dmm / api.c

/*
 * This file is part of the sigrok project.
 *
 * Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
 *
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 */

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

static const int hwcaps[] = {
	SR_HWCAP_MULTIMETER,
	SR_HWCAP_LIMIT_SAMPLES,
	SR_HWCAP_LIMIT_MSEC,
	SR_HWCAP_CONTINUOUS,
	0,
};

static const char *probe_names[] = {
	"Probe",
	NULL,
};

SR_PRIV struct sr_dev_driver uni_t_ut61d_driver_info;
SR_PRIV struct sr_dev_driver voltcraft_vc820_driver_info;

static struct sr_dev_driver *di_ut61d = &uni_t_ut61d_driver_info;
static struct sr_dev_driver *di_vc820 = &voltcraft_vc820_driver_info;

/* After hw_init() this will point to a device-specific entry (see above). */
static struct sr_dev_driver *di = NULL;

static int clear_instances(void)
{
	/* TODO: Use common code later. */

	return SR_OK;
}

static int hw_init(struct sr_context *sr_ctx, int dmm)
{
	int ret;
	struct drv_context *drvc;

	if (!(drvc = g_try_malloc0(sizeof(struct drv_context)))) {
		sr_err("Driver context malloc failed.");
		return SR_ERR_MALLOC;
	}

	if ((ret = libusb_init(NULL)) < 0) {
		sr_err("Failed to initialize libusb: %s.",
		       libusb_error_name(ret));
		return SR_ERR;
	}

	if (dmm == UNI_T_UT61D)
		di = di_ut61d;
	else if (dmm == VOLTCRAFT_VC820)
		di = di_vc820;
	sr_dbg("Selected '%s' subdriver.", di->name);

	drvc->sr_ctx = sr_ctx;
	di->priv = drvc;

	return SR_OK;
}

static int hw_init_ut61d(struct sr_context *sr_ctx)
{
	return hw_init(sr_ctx, UNI_T_UT61D);
}

static int hw_init_vc820(struct sr_context *sr_ctx)
{
	return hw_init(sr_ctx, VOLTCRAFT_VC820);
}

static GSList *hw_scan(GSList *options)
{
	GSList *devices, *l;
	struct sr_dev_inst *sdi;
	struct dev_context *devc;
	struct drv_context *drvc;
	struct sr_probe *probe;
	struct libusb_device *ldev;
	int i, devcnt;

	(void)options;

	drvc = di->priv;

	devices = NULL;

	if (!(l = sr_usb_connect(NULL, "1a86.e008")))
		return NULL;

	for (i = 0; i < (int)g_slist_length(l); i++) {
		ldev = (struct libusb_device *)l->data;

		if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) {
			sr_err("Device context malloc failed.");
			return NULL;
		}

		devcnt = g_slist_length(drvc->instances);
		if (!(sdi = sr_dev_inst_new(devcnt, SR_ST_INACTIVE,
				di->longname, NULL, NULL))) {
			sr_err("sr_dev_inst_new returned NULL.");
			return NULL;
		}
		sdi->priv = devc;
		sdi->driver = di;
		if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE, "P1")))
			return NULL;
		sdi->probes = g_slist_append(sdi->probes, probe);
		devc->usb = sr_usb_dev_inst_new(
				libusb_get_bus_number(ldev),
				libusb_get_device_address(ldev), NULL);
		drvc->instances = g_slist_append(drvc->instances, l->data);
		devices = g_slist_append(devices, sdi);
	}

	return devices;
}

static GSList *hw_dev_list(void)
{
	struct drv_context *drvc;

	drvc = di->priv;

	return drvc->instances;
}

static int hw_dev_open(struct sr_dev_inst *sdi)
{
	struct dev_context *devc;

	devc = sdi->priv;

	return sr_usb_open(NULL, devc->usb);
}

static int hw_dev_close(struct sr_dev_inst *sdi)
{
	(void)sdi;

	/* TODO */

	return SR_OK;
}

static int hw_cleanup(void)
{
	clear_instances();

	// libusb_exit(NULL);

	return SR_OK;
}

static int hw_info_get(int info_id, const void **data,
		       const struct sr_dev_inst *sdi)
{
	(void)sdi;

	sr_spew("Backend requested info_id %d.", info_id);

	switch (info_id) {
	case SR_DI_HWCAPS:
		*data = hwcaps;
		sr_spew("%s: Returning hwcaps.", __func__);
		break;
	case SR_DI_NUM_PROBES:
		*data = GINT_TO_POINTER(1);
		sr_spew("%s: Returning number of probes.", __func__);
		break;
	case SR_DI_PROBE_NAMES:
		*data = probe_names;
		sr_spew("%s: Returning probe names.", __func__);
		break;
	case SR_DI_SAMPLERATES:
		/* TODO: Get rid of this. */
		*data = NULL;
		sr_spew("%s: Returning samplerates.", __func__);
		return SR_ERR_ARG;
		break;
	case SR_DI_CUR_SAMPLERATE:
		/* TODO: Get rid of this. */
		*data = NULL;
		sr_spew("%s: Returning current samplerate.", __func__);
		return SR_ERR_ARG;
		break;
	default:
		sr_err("%s: Unknown info_id %d.", __func__, info_id);
		return SR_ERR_ARG;
		break;
	}

	return SR_OK;
}

static int hw_dev_config_set(const struct sr_dev_inst *sdi, int hwcap,
			     const void *value)
{
	struct dev_context *devc;

	devc = sdi->priv;

	switch (hwcap) {
	case SR_HWCAP_LIMIT_MSEC:
		/* TODO: Not yet implemented. */
		if (*(const uint64_t *)value == 0) {
			sr_err("Time limit cannot be 0.");
			return SR_ERR;
		}
		devc->limit_msec = *(const uint64_t *)value;
		sr_dbg("Setting time limit to %" PRIu64 "ms.",
		       devc->limit_msec);
		break;
	case SR_HWCAP_LIMIT_SAMPLES:
		if (*(const uint64_t *)value == 0) {
			sr_err("Sample limit cannot be 0.");
			return SR_ERR;
		}
		devc->limit_samples = *(const uint64_t *)value;
		sr_dbg("Setting sample limit to %" PRIu64 ".",
		       devc->limit_samples);
		break;
	default:
		sr_err("Unknown capability: %d.", hwcap);
		return SR_ERR;
		break;
	}

	return SR_OK;
}

static int hw_dev_acquisition_start(const struct sr_dev_inst *sdi,
				    void *cb_data)
{
	struct sr_datafeed_packet packet;
	struct sr_datafeed_header header;
	struct sr_datafeed_meta_analog meta;
	struct dev_context *devc;

	devc = sdi->priv;

	sr_dbg("Starting acquisition.");

	devc->cb_data = cb_data;

	/* Send header packet to the session bus. */
	sr_dbg("Sending SR_DF_HEADER.");
	packet.type = SR_DF_HEADER;
	packet.payload = (uint8_t *)&header;
	header.feed_version = 1;
	gettimeofday(&header.starttime, NULL);
	sr_session_send(devc->cb_data, &packet);

	/* Send metadata about the SR_DF_ANALOG packets to come. */
	sr_dbg("Sending SR_DF_META_ANALOG.");
	packet.type = SR_DF_META_ANALOG;
	packet.payload = &meta;
	meta.num_probes = 1;
	sr_session_send(devc->cb_data, &packet);

	if (!strcmp(di->name, "uni-t-ut61d")) {
		sr_source_add(0, 0, 10 /* poll_timeout */,
			      uni_t_ut61d_receive_data, (void *)sdi);
	} else if (!strcmp(di->name, "voltcraft-vc820")) {
		sr_source_add(0, 0, 10 /* poll_timeout */,
			      voltcraft_vc820_receive_data, (void *)sdi);
	}

	return SR_OK;
}

static int hw_dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
{
	struct sr_datafeed_packet packet;

	(void)sdi;

	sr_dbg("Stopping acquisition.");

	/* Send end packet to the session bus. */
	sr_dbg("Sending SR_DF_END.");
	packet.type = SR_DF_END;
	sr_session_send(cb_data, &packet);

	/* TODO? */
	sr_source_remove(0);

	return SR_OK;
}

SR_PRIV struct sr_dev_driver uni_t_ut61d_driver_info = {
	.name = "uni-t-ut61d",
	.longname = "UNI-T UT61D",
	.api_version = 1,
	.init = hw_init_ut61d,
	.cleanup = hw_cleanup,
	.scan = hw_scan,
	.dev_list = hw_dev_list,
	.dev_clear = clear_instances,
	.dev_open = hw_dev_open,
	.dev_close = hw_dev_close,
	.info_get = hw_info_get,
	.dev_config_set = hw_dev_config_set,
	.dev_acquisition_start = hw_dev_acquisition_start,
	.dev_acquisition_stop = hw_dev_acquisition_stop,
	.priv = NULL,
};

SR_PRIV struct sr_dev_driver voltcraft_vc820_driver_info = {
	.name = "voltcraft-vc820",
	.longname = "Voltcraft VC-820",
	.api_version = 1,
	.init = hw_init_vc820,
	.cleanup = hw_cleanup,
	.scan = hw_scan,
	.dev_list = hw_dev_list,
	.dev_clear = clear_instances,
	.dev_open = hw_dev_open,
	.dev_close = hw_dev_close,
	.info_get = hw_info_get,
	.dev_config_set = hw_dev_config_set,
	.dev_acquisition_start = hw_dev_acquisition_start,
	.dev_acquisition_stop = hw_dev_acquisition_stop,
	.priv = NULL,
};
Commit	Line	Data
79081ec8 UH	1	/*
	2	* This file is part of the sigrok project.
	3	*
	4	* Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License as published by
	8	* the Free Software Foundation; either version 2 of the License, or
	9	* (at your option) any later version.
	10	*
	11	* This program is distributed in the hope that it will be useful,
	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	* GNU General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU General Public License
	17	* along with this program; if not, write to the Free Software
	18	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
	19	*/
	20
	21	#include <stdlib.h>
	22	#include <string.h>
	23	#include "libsigrok.h"
	24	#include "libsigrok-internal.h"
	25	#include "protocol.h"
	26
	27	static const int hwcaps[] = {
	28	SR_HWCAP_MULTIMETER,
	29	SR_HWCAP_LIMIT_SAMPLES,
	30	SR_HWCAP_LIMIT_MSEC,
	31	SR_HWCAP_CONTINUOUS,
	32	0,
	33	};
	34
	35	static const char *probe_names[] = {
	36	"Probe",
	37	NULL,
	38	};
	39
fdbcb86d	40	SR_PRIV struct sr_dev_driver uni_t_ut61d_driver_info;
fdbcb86d	41	SR_PRIV struct sr_dev_driver voltcraft_vc820_driver_info;
6ac5f892 UH	42
6ac5f892 UH	43	static struct sr_dev_driver *di_ut61d = &uni_t_ut61d_driver_info;
fdbcb86d	44	static struct sr_dev_driver *di_vc820 = &voltcraft_vc820_driver_info;
79081ec8	45
6ac5f892 UH	46	/* After hw_init() this will point to a device-specific entry (see above). */
	47	static struct sr_dev_driver *di = NULL;
	48
79081ec8 UH	49	static int clear_instances(void)
	50	{
	51	/* TODO: Use common code later. */
	52
	53	return SR_OK;
	54	}
	55
34f06b90	56	static int hw_init(struct sr_context *sr_ctx, int dmm)
79081ec8 UH	57	{
	58	int ret;
	59	struct drv_context *drvc;
	60
	61	if (!(drvc = g_try_malloc0(sizeof(struct drv_context)))) {
	62	sr_err("Driver context malloc failed.");
	63	return SR_ERR_MALLOC;
	64	}
	65
	66	if ((ret = libusb_init(NULL)) < 0) {
	67	sr_err("Failed to initialize libusb: %s.",
	68	libusb_error_name(ret));
	69	return SR_ERR;
	70	}
	71
fdbcb86d	72	if (dmm == UNI_T_UT61D)
6ac5f892	73	di = di_ut61d;
fdbcb86d	74	else if (dmm == VOLTCRAFT_VC820)
6ac5f892	75	di = di_vc820;
8c1adf37	76	sr_dbg("Selected '%s' subdriver.", di->name);
6ac5f892	77
1ebe4b4e	78	drvc->sr_ctx = sr_ctx;
6ac5f892	79	di->priv = drvc;
79081ec8 UH	80
	81	return SR_OK;
	82	}
	83
34f06b90	84	static int hw_init_ut61d(struct sr_context *sr_ctx)
fdbcb86d	85	{
34f06b90	86	return hw_init(sr_ctx, UNI_T_UT61D);
fdbcb86d UH	87	}
fdbcb86d UH	88
34f06b90	89	static int hw_init_vc820(struct sr_context *sr_ctx)
fdbcb86d	90	{
34f06b90	91	return hw_init(sr_ctx, VOLTCRAFT_VC820);
fdbcb86d UH	92	}
fdbcb86d UH	93
6ac5f892	94	static GSList hw_scan(GSList options)
79081ec8	95	{
41d9427f	96	GSList devices, l;
79081ec8	97	struct sr_dev_inst *sdi;
41d9427f	98	struct dev_context *devc;
79081ec8	99	struct drv_context *drvc;
41d9427f UH	100	struct sr_probe *probe;
	101	struct libusb_device *ldev;
	102	int i, devcnt;
79081ec8 UH	103
	104	(void)options;
	105
6ac5f892	106	drvc = di->priv;
79081ec8	107
41d9427f UH	108	devices = NULL;
	109
	110	if (!(l = sr_usb_connect(NULL, "1a86.e008")))
79081ec8 UH	111	return NULL;
79081ec8 UH	112
41d9427f UH	113	for (i = 0; i < (int)g_slist_length(l); i++) {
	114	ldev = (struct libusb_device *)l->data;
	115
	116	if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) {
	117	sr_err("Device context malloc failed.");
	118	return NULL;
	119	}
	120
	121	devcnt = g_slist_length(drvc->instances);
	122	if (!(sdi = sr_dev_inst_new(devcnt, SR_ST_INACTIVE,
	123	di->longname, NULL, NULL))) {
	124	sr_err("sr_dev_inst_new returned NULL.");
	125	return NULL;
	126	}
	127	sdi->priv = devc;
6ac5f892	128	sdi->driver = di;
41d9427f UH	129	if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE, "P1")))
	130	return NULL;
	131	sdi->probes = g_slist_append(sdi->probes, probe);
	132	devc->usb = sr_usb_dev_inst_new(
	133	libusb_get_bus_number(ldev),
	134	libusb_get_device_address(ldev), NULL);
79081ec8	135	drvc->instances = g_slist_append(drvc->instances, l->data);
41d9427f	136	devices = g_slist_append(devices, sdi);
79081ec8 UH	137	}
	138
	139	return devices;
	140	}
	141
6ac5f892	142	static GSList *hw_dev_list(void)
79081ec8 UH	143	{
	144	struct drv_context *drvc;
	145
6ac5f892	146	drvc = di->priv;
79081ec8 UH	147
	148	return drvc->instances;
	149	}
	150
	151	static int hw_dev_open(struct sr_dev_inst *sdi)
	152	{
41d9427f UH	153	struct dev_context *devc;
	154
	155	devc = sdi->priv;
	156
	157	return sr_usb_open(NULL, devc->usb);
79081ec8 UH	158	}
	159
	160	static int hw_dev_close(struct sr_dev_inst *sdi)
	161	{
	162	(void)sdi;
	163
	164	/* TODO */
	165
	166	return SR_OK;
	167	}
	168
	169	static int hw_cleanup(void)
	170	{
	171	clear_instances();
	172
	173	// libusb_exit(NULL);
	174
	175	return SR_OK;
	176	}
	177
	178	static int hw_info_get(int info_id, const void **data,
	179	const struct sr_dev_inst *sdi)
	180	{
	181	(void)sdi;
	182
	183	sr_spew("Backend requested info_id %d.", info_id);
	184
	185	switch (info_id) {
	186	case SR_DI_HWCAPS:
	187	*data = hwcaps;
	188	sr_spew("%s: Returning hwcaps.", __func__);
	189	break;
	190	case SR_DI_NUM_PROBES:
	191	*data = GINT_TO_POINTER(1);
	192	sr_spew("%s: Returning number of probes.", __func__);
	193	break;
	194	case SR_DI_PROBE_NAMES:
	195	*data = probe_names;
	196	sr_spew("%s: Returning probe names.", __func__);
	197	break;
	198	case SR_DI_SAMPLERATES:
	199	/* TODO: Get rid of this. */
	200	*data = NULL;
	201	sr_spew("%s: Returning samplerates.", __func__);
	202	return SR_ERR_ARG;
	203	break;
	204	case SR_DI_CUR_SAMPLERATE:
	205	/* TODO: Get rid of this. */
	206	*data = NULL;
	207	sr_spew("%s: Returning current samplerate.", __func__);
	208	return SR_ERR_ARG;
	209	break;
	210	default:
	211	sr_err("%s: Unknown info_id %d.", __func__, info_id);
	212	return SR_ERR_ARG;
	213	break;
	214	}
	215
	216	return SR_OK;
	217	}
	218
	219	static int hw_dev_config_set(const struct sr_dev_inst *sdi, int hwcap,
	220	const void *value)
	221	{
222	struct dev_context *devc;
223
224	devc = sdi->priv;
225
226	switch (hwcap) {
227	case SR_HWCAP_LIMIT_MSEC:
228	/* TODO: Not yet implemented. */
229	if ((const uint64_t )value == 0) {
230	sr_err("Time limit cannot be 0.");
231	return SR_ERR;
232	}
233	devc->limit_msec = (const uint64_t )value;
234	sr_dbg("Setting time limit to %" PRIu64 "ms.",
235	devc->limit_msec);
236	break;
237	case SR_HWCAP_LIMIT_SAMPLES:
238	if ((const uint64_t )value == 0) {
239	sr_err("Sample limit cannot be 0.");
240	return SR_ERR;
241	}
242	devc->limit_samples = (const uint64_t )value;
243	sr_dbg("Setting sample limit to %" PRIu64 ".",
244	devc->limit_samples);
245	break;
246	default:
247	sr_err("Unknown capability: %d.", hwcap);
248	return SR_ERR;
249	break;
250	}
251
252	return SR_OK;
253	}
254
255	static int hw_dev_acquisition_start(const struct sr_dev_inst *sdi,
6ac5f892	256	void *cb_data)
79081ec8 UH	257	{
	258	struct sr_datafeed_packet packet;
	259	struct sr_datafeed_header header;
	260	struct sr_datafeed_meta_analog meta;
	261	struct dev_context *devc;
	262
	263	devc = sdi->priv;
	264
	265	sr_dbg("Starting acquisition.");
	266
	267	devc->cb_data = cb_data;
	268
	269	/* Send header packet to the session bus. */
	270	sr_dbg("Sending SR_DF_HEADER.");
	271	packet.type = SR_DF_HEADER;
	272	packet.payload = (uint8_t *)&header;
	273	header.feed_version = 1;
	274	gettimeofday(&header.starttime, NULL);
	275	sr_session_send(devc->cb_data, &packet);
	276
	277	/* Send metadata about the SR_DF_ANALOG packets to come. */
	278	sr_dbg("Sending SR_DF_META_ANALOG.");
	279	packet.type = SR_DF_META_ANALOG;
	280	packet.payload = &meta;
	281	meta.num_probes = 1;
	282	sr_session_send(devc->cb_data, &packet);
	283
6ac5f892	284	if (!strcmp(di->name, "uni-t-ut61d")) {
fdbcb86d UH	285	sr_source_add(0, 0, 10 /* poll_timeout */,
fdbcb86d UH	286	uni_t_ut61d_receive_data, (void *)sdi);
6ac5f892	287	} else if (!strcmp(di->name, "voltcraft-vc820")) {
fdbcb86d UH	288	sr_source_add(0, 0, 10 /* poll_timeout */,
	289	voltcraft_vc820_receive_data, (void *)sdi);
	290	}
79081ec8 UH	291
	292	return SR_OK;
	293	}
	294
69b07d14	295	static int hw_dev_acquisition_stop(struct sr_dev_inst sdi, void cb_data)
79081ec8 UH	296	{
	297	struct sr_datafeed_packet packet;
	298
	299	(void)sdi;
	300
	301	sr_dbg("Stopping acquisition.");
	302
	303	/* Send end packet to the session bus. */
	304	sr_dbg("Sending SR_DF_END.");
	305	packet.type = SR_DF_END;
	306	sr_session_send(cb_data, &packet);
	307
	308	/* TODO? */
	309	sr_source_remove(0);
	310
	311	return SR_OK;
	312	}
	313
fdbcb86d UH	314	SR_PRIV struct sr_dev_driver uni_t_ut61d_driver_info = {
	315	.name = "uni-t-ut61d",
	316	.longname = "UNI-T UT61D",
	317	.api_version = 1,
	318	.init = hw_init_ut61d,
	319	.cleanup = hw_cleanup,
6ac5f892 UH	320	.scan = hw_scan,
6ac5f892 UH	321	.dev_list = hw_dev_list,
fdbcb86d UH	322	.dev_clear = clear_instances,
	323	.dev_open = hw_dev_open,
	324	.dev_close = hw_dev_close,
	325	.info_get = hw_info_get,
	326	.dev_config_set = hw_dev_config_set,
6ac5f892	327	.dev_acquisition_start = hw_dev_acquisition_start,
fdbcb86d UH	328	.dev_acquisition_stop = hw_dev_acquisition_stop,
	329	.priv = NULL,
	330	};
	331
	332	SR_PRIV struct sr_dev_driver voltcraft_vc820_driver_info = {
	333	.name = "voltcraft-vc820",
	334	.longname = "Voltcraft VC-820",
79081ec8	335	.api_version = 1,
fdbcb86d	336	.init = hw_init_vc820,
79081ec8	337	.cleanup = hw_cleanup,
6ac5f892 UH	338	.scan = hw_scan,
6ac5f892 UH	339	.dev_list = hw_dev_list,
79081ec8 UH	340	.dev_clear = clear_instances,
	341	.dev_open = hw_dev_open,
	342	.dev_close = hw_dev_close,
	343	.info_get = hw_info_get,
	344	.dev_config_set = hw_dev_config_set,
6ac5f892	345	.dev_acquisition_start = hw_dev_acquisition_start,
79081ec8 UH	346	.dev_acquisition_stop = hw_dev_acquisition_stop,
	347	.priv = NULL,
	348	};