[libsigrok.git] / hardware / uni-t-ut32x / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2013 Bert Vermeulen <bert@biot.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 3 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 "protocol.h"

#include <string.h>
#include <math.h>

extern struct sr_dev_driver uni_t_ut32x_driver_info;
static struct sr_dev_driver *di = &uni_t_ut32x_driver_info;

static float parse_temperature(unsigned char *buf)
{
	float temp;
	int i;
	gboolean negative;

	negative = FALSE;
	temp = 0.0;
	for (i = 0; i < 4; i++) {
		if (buf[i] == 0x3a)
			continue;
		if (buf[i] == 0x3b) {
			if (negative) {
				sr_dbg("Double negative sign!");
				return NAN;
			} else {
				negative = TRUE;
				continue;
			}
		}
		if (buf[i] < 0x30 || buf[i] > 0x39) {
			sr_dbg("Invalid digit '%.2x'!", buf[i]);
			return NAN;
		}
		temp *= 10;
		temp += (buf[i] - 0x30);
	}
	temp /= 10;
	if (negative)
		temp = -temp;

	return temp;
}

static void process_packet(struct sr_dev_inst *sdi)
{
	struct dev_context *devc;
	struct sr_datafeed_packet packet;
	struct sr_datafeed_analog analog;
	GString *spew;
	float temp;
	int i;
	gboolean is_valid;

	devc = sdi->priv;
	sr_dbg("Received full 19-byte packet.");
	if (sr_log_loglevel_get() >= SR_LOG_SPEW) {
		spew = g_string_sized_new(60);
		for (i = 0; i < devc->packet_len; i++)
			g_string_append_printf(spew, "%.2x ", devc->packet[i]);
		sr_spew("%s", spew->str);
		g_string_free(spew, TRUE);
	}

	is_valid = TRUE;
	if (devc->packet[1] == 0x3b && devc->packet[2] == 0x3b
			&& devc->packet[3] == 0x3b && devc->packet[4] == 0x3b)
		/* No measurement: missing channel, empty storage location, ... */
		is_valid = FALSE;

	temp = parse_temperature(devc->packet + 1);
	if (isnan(temp))
		is_valid = FALSE;

	if (is_valid) {
		memset(&analog, 0, sizeof(struct sr_datafeed_analog));
		analog.mq = SR_MQ_TEMPERATURE;
		analog.mqflags = 0;
		switch (devc->packet[5] - 0x30) {
		case 1:
			analog.unit = SR_UNIT_CELSIUS;
			break;
		case 2:
			analog.unit = SR_UNIT_FAHRENHEIT;
			break;
		case 3:
			analog.unit = SR_UNIT_KELVIN;
			break;
		default:
			/* We can still pass on the measurement, whatever it is. */
			sr_dbg("Unknown unit 0x%.2x.", devc->packet[5]);
		}
		switch (devc->packet[13] - 0x30) {
		case 0:
			/* Channel T1. */
			analog.channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, 0));
			break;
		case 1:
			/* Channel T2. */
			analog.channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, 1));
			break;
		case 2:
		case 3:
			/* Channel T1-T2. */
			analog.channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, 2));
			analog.mqflags |= SR_MQFLAG_RELATIVE;
			break;
		default:
			sr_err("Unknown channel 0x%.2x.", devc->packet[13]);
			is_valid = FALSE;
		}
		if (is_valid) {
			analog.num_samples = 1;
			analog.data = &temp;
			packet.type = SR_DF_ANALOG;
			packet.payload = &analog;
			sr_session_send(devc->cb_data, &packet);
			g_slist_free(analog.channels);
		}
	}

	/* We count packets even if the temperature was invalid. This way
	 * a sample limit on "Memory" data source still works: unused
	 * memory slots come through as "----" measurements. */
	devc->num_samples++;
	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
		sdi->driver->dev_acquisition_stop((struct sr_dev_inst *)sdi,
				devc->cb_data);
	}

}

SR_PRIV void uni_t_ut32x_receive_transfer(struct libusb_transfer *transfer)
{
	struct dev_context *devc;
	struct sr_dev_inst *sdi;
	int hid_payload_len, ret;

	sdi = transfer->user_data;
	devc = sdi->priv;
	if (transfer->actual_length == 8) {
		/* CH9325 encodes length in low nibble of first byte, with
		 * bytes 1-7 being the (padded) payload. */
		hid_payload_len = transfer->buffer[0] & 0x0f;
		memcpy(devc->packet + devc->packet_len, transfer->buffer + 1,
				hid_payload_len);
		devc->packet_len += hid_payload_len;
		if (devc->packet_len >= 2
				&& devc->packet[devc->packet_len - 2] == 0x0d
				&& devc->packet[devc->packet_len - 1] == 0x0a) {
			/* Got end of packet, but do we have a complete packet? */
			if (devc->packet_len == 19)
				process_packet(sdi);
			/* Either way, done with it. */
			devc->packet_len = 0;
		} else if (devc->packet_len > 19) {
			/* Guard against garbage from the device overrunning
			 * our packet buffer. */
			sr_dbg("Buffer overrun!");
			devc->packet_len = 0;
		}
	}

	/* Get the next transfer (unless we're shutting down). */
	if (sdi->status != SR_ST_STOPPING) {
		if ((ret = libusb_submit_transfer(devc->xfer)) != 0) {
			sr_dbg("Failed to resubmit transfer: %s", libusb_error_name(ret));
			sdi->status = SR_ST_STOPPING;
			libusb_free_transfer(devc->xfer);
		}
	} else
		libusb_free_transfer(devc->xfer);

}

SR_PRIV int uni_t_ut32x_handle_events(int fd, int revents, void *cb_data)
{
	struct drv_context *drvc;
	struct dev_context *devc;
	struct sr_dev_inst *sdi;
	struct sr_datafeed_packet packet;
	struct sr_usb_dev_inst *usb;
	struct timeval tv;
	int len, ret;
	unsigned char cmd[2];

	(void)fd;
	(void)revents;
	drvc = di->priv;

	if (!(sdi = cb_data))
		return TRUE;

	if (!(devc = sdi->priv))
		return TRUE;

	memset(&tv, 0, sizeof(struct timeval));
	libusb_handle_events_timeout_completed(drvc->sr_ctx->libusb_ctx, &tv,
			NULL);

	if (sdi->status == SR_ST_STOPPING) {
		usb_source_remove(drvc->sr_ctx);
		packet.type = SR_DF_END;
		sr_session_send(cb_data, &packet);

		/* Tell the device to stop sending USB packets. */
		usb = sdi->conn;
		cmd[0] = 0x01;
		cmd[1] = CMD_STOP;
		ret = libusb_bulk_transfer(usb->devhdl, EP_OUT, cmd, 2, &len, 5);
		if (ret != 0 || len != 2) {
			/* Warning only, doesn't matter. */
			sr_dbg("Failed to send stop command: %s", libusb_error_name(ret));
		}

		sdi->status = SR_ST_ACTIVE;
		return TRUE;
	}

	return TRUE;
}
Commit	Line	Data
3877dde4 BV	1	/*
	2	* This file is part of the libsigrok project.
	3	*
	4	* Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
	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 3 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, see <http://www.gnu.org/licenses/>.
	18	*/
	19
	20	#include "protocol.h"
	21
d6ff054a BV	22	#include <string.h>
	23	#include <math.h>
	24
	25	extern struct sr_dev_driver uni_t_ut32x_driver_info;
	26	static struct sr_dev_driver *di = &uni_t_ut32x_driver_info;
	27
	28	static float parse_temperature(unsigned char *buf)
3877dde4	29	{
d6ff054a BV	30	float temp;
	31	int i;
	32	gboolean negative;
	33
	34	negative = FALSE;
	35	temp = 0.0;
	36	for (i = 0; i < 4; i++) {
	37	if (buf[i] == 0x3a)
	38	continue;
	39	if (buf[i] == 0x3b) {
	40	if (negative) {
	41	sr_dbg("Double negative sign!");
	42	return NAN;
	43	} else {
	44	negative = TRUE;
	45	continue;
	46	}
	47	}
	48	if (buf[i] < 0x30 \|\| buf[i] > 0x39) {
	49	sr_dbg("Invalid digit '%.2x'!", buf[i]);
	50	return NAN;
	51	}
	52	temp *= 10;
	53	temp += (buf[i] - 0x30);
	54	}
	55	temp /= 10;
	56	if (negative)
	57	temp = -temp;
3877dde4	58
d6ff054a BV	59	return temp;
	60	}
	61
	62	static void process_packet(struct sr_dev_inst *sdi)
	63	{
3877dde4	64	struct dev_context *devc;
d6ff054a BV	65	struct sr_datafeed_packet packet;
	66	struct sr_datafeed_analog analog;
	67	GString *spew;
	68	float temp;
	69	int i;
	70	gboolean is_valid;
3877dde4	71
d6ff054a BV	72	devc = sdi->priv;
	73	sr_dbg("Received full 19-byte packet.");
	74	if (sr_log_loglevel_get() >= SR_LOG_SPEW) {
	75	spew = g_string_sized_new(60);
	76	for (i = 0; i < devc->packet_len; i++)
	77	g_string_append_printf(spew, "%.2x ", devc->packet[i]);
	78	sr_spew("%s", spew->str);
	79	g_string_free(spew, TRUE);
	80	}
3877dde4	81
d6ff054a BV	82	is_valid = TRUE;
	83	if (devc->packet[1] == 0x3b && devc->packet[2] == 0x3b
	84	&& devc->packet[3] == 0x3b && devc->packet[4] == 0x3b)
ba7dd8bb	85	/* No measurement: missing channel, empty storage location, ... */
d6ff054a	86	is_valid = FALSE;
3877dde4	87
d6ff054a BV	88	temp = parse_temperature(devc->packet + 1);
	89	if (isnan(temp))
	90	is_valid = FALSE;
	91
	92	if (is_valid) {
	93	memset(&analog, 0, sizeof(struct sr_datafeed_analog));
	94	analog.mq = SR_MQ_TEMPERATURE;
	95	analog.mqflags = 0;
	96	switch (devc->packet[5] - 0x30) {
	97	case 1:
	98	analog.unit = SR_UNIT_CELSIUS;
	99	break;
	100	case 2:
	101	analog.unit = SR_UNIT_FAHRENHEIT;
	102	break;
	103	case 3:
	104	analog.unit = SR_UNIT_KELVIN;
	105	break;
	106	default:
	107	/* We can still pass on the measurement, whatever it is. */
	108	sr_dbg("Unknown unit 0x%.2x.", devc->packet[5]);
	109	}
	110	switch (devc->packet[13] - 0x30) {
	111	case 0:
ba7dd8bb UH	112	/* Channel T1. */
ba7dd8bb UH	113	analog.channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, 0));
d6ff054a BV	114	break;
d6ff054a BV	115	case 1:
ba7dd8bb UH	116	/* Channel T2. */
ba7dd8bb UH	117	analog.channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, 1));
d6ff054a BV	118	break;
	119	case 2:
	120	case 3:
ba7dd8bb UH	121	/* Channel T1-T2. */
ba7dd8bb UH	122	analog.channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, 2));
d6ff054a BV	123	analog.mqflags \|= SR_MQFLAG_RELATIVE;
	124	break;
	125	default:
ba7dd8bb	126	sr_err("Unknown channel 0x%.2x.", devc->packet[13]);
d6ff054a BV	127	is_valid = FALSE;
	128	}
	129	if (is_valid) {
	130	analog.num_samples = 1;
	131	analog.data = &temp;
	132	packet.type = SR_DF_ANALOG;
	133	packet.payload = &analog;
	134	sr_session_send(devc->cb_data, &packet);
ba7dd8bb	135	g_slist_free(analog.channels);
d6ff054a BV	136	}
	137	}
	138
	139	/* We count packets even if the temperature was invalid. This way
	140	* a sample limit on "Memory" data source still works: unused
	141	* memory slots come through as "----" measurements. */
	142	devc->num_samples++;
	143	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
	144	sdi->driver->dev_acquisition_stop((struct sr_dev_inst *)sdi,
	145	devc->cb_data);
3877dde4 BV	146	}
3877dde4 BV	147
3877dde4	148	}
6513f97f BV	149
	150	SR_PRIV void uni_t_ut32x_receive_transfer(struct libusb_transfer *transfer)
	151	{
d6ff054a BV	152	struct dev_context *devc;
	153	struct sr_dev_inst *sdi;
	154	int hid_payload_len, ret;
	155
	156	sdi = transfer->user_data;
	157	devc = sdi->priv;
	158	if (transfer->actual_length == 8) {
	159	/* CH9325 encodes length in low nibble of first byte, with
	160	* bytes 1-7 being the (padded) payload. */
	161	hid_payload_len = transfer->buffer[0] & 0x0f;
	162	memcpy(devc->packet + devc->packet_len, transfer->buffer + 1,
	163	hid_payload_len);
	164	devc->packet_len += hid_payload_len;
	165	if (devc->packet_len >= 2
	166	&& devc->packet[devc->packet_len - 2] == 0x0d
	167	&& devc->packet[devc->packet_len - 1] == 0x0a) {
	168	/* Got end of packet, but do we have a complete packet? */
	169	if (devc->packet_len == 19)
	170	process_packet(sdi);
	171	/* Either way, done with it. */
	172	devc->packet_len = 0;
	173	} else if (devc->packet_len > 19) {
	174	/* Guard against garbage from the device overrunning
	175	* our packet buffer. */
	176	sr_dbg("Buffer overrun!");
	177	devc->packet_len = 0;
	178	}
	179	}
	180
	181	/* Get the next transfer (unless we're shutting down). */
	182	if (sdi->status != SR_ST_STOPPING) {
	183	if ((ret = libusb_submit_transfer(devc->xfer)) != 0) {
	184	sr_dbg("Failed to resubmit transfer: %s", libusb_error_name(ret));
	185	sdi->status = SR_ST_STOPPING;
	186	libusb_free_transfer(devc->xfer);
	187	}
	188	} else
	189	libusb_free_transfer(devc->xfer);
6513f97f BV	190
	191	}
	192
d6ff054a BV	193	SR_PRIV int uni_t_ut32x_handle_events(int fd, int revents, void *cb_data)
	194	{
	195	struct drv_context *drvc;
	196	struct dev_context *devc;
	197	struct sr_dev_inst *sdi;
	198	struct sr_datafeed_packet packet;
	199	struct sr_usb_dev_inst *usb;
	200	struct timeval tv;
6c60facc	201	int len, ret;
d6ff054a BV	202	unsigned char cmd[2];
	203
	204	(void)fd;
	205	(void)revents;
	206	drvc = di->priv;
	207
	208	if (!(sdi = cb_data))
	209	return TRUE;
	210
	211	if (!(devc = sdi->priv))
	212	return TRUE;
	213
	214	memset(&tv, 0, sizeof(struct timeval));
	215	libusb_handle_events_timeout_completed(drvc->sr_ctx->libusb_ctx, &tv,
	216	NULL);
	217
	218	if (sdi->status == SR_ST_STOPPING) {
6c60facc	219	usb_source_remove(drvc->sr_ctx);
d6ff054a BV	220	packet.type = SR_DF_END;
	221	sr_session_send(cb_data, &packet);
	222
	223	/* Tell the device to stop sending USB packets. */
	224	usb = sdi->conn;
	225	cmd[0] = 0x01;
	226	cmd[1] = CMD_STOP;
	227	ret = libusb_bulk_transfer(usb->devhdl, EP_OUT, cmd, 2, &len, 5);
	228	if (ret != 0 \|\| len != 2) {
	229	/* Warning only, doesn't matter. */
	230	sr_dbg("Failed to send stop command: %s", libusb_error_name(ret));
	231	}
	232
	233	sdi->status = SR_ST_ACTIVE;
	234	return TRUE;
	235	}
	236
	237	return TRUE;
	238	}
	239