[libsigrok.git] / src / hardware / center-3xx / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2013 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 <config.h>
#include "protocol.h"

#define NUM_CHANNELS 4

struct center_info {
	float temp[NUM_CHANNELS];
	gboolean rec, std, max, min, maxmin, t1t2, rel, hold, lowbat, celsius;
	gboolean memfull, autooff;
	gboolean mode_std, mode_rel, mode_max, mode_min, mode_maxmin;
};

static int center_send(struct sr_serial_dev_inst *serial, const char *cmd)
{
	int ret;

	if ((ret = serial_write_blocking(serial, cmd, strlen(cmd),
			serial_timeout(serial, strlen(cmd)))) < 0) {
		sr_err("Error sending '%s' command: %d.", cmd, ret);
		return SR_ERR;
	}

	return SR_OK;
}

SR_PRIV gboolean center_3xx_packet_valid(const uint8_t *buf)
{
	return (buf[0] == 0x02 && buf[44] == 0x03);
}

static void log_packet(const uint8_t *buf, int idx)
{
	int i;
	GString *s;

	s = g_string_sized_new(100);
	g_string_printf(s, "Packet: ");
	for (i = 0; i < center_devs[idx].packet_size; i++)
	        g_string_append_printf(s, "%02x ", buf[i]);
	sr_spew("%s", s->str);
	g_string_free(s, TRUE);
}

static int packet_parse(const uint8_t *buf, int idx, struct center_info *info)
{
	int i;
	uint16_t temp_u16;

	log_packet(buf, idx);

	/* Byte 0: Always 0x02. */

	/* Byte 1: Various status bits. */
	info->rec         = (buf[1] & (1 << 0)) != 0;
	info->mode_std    = (((buf[1] >> 1) & 0x3) == 0);
	info->mode_max    = (((buf[1] >> 1) & 0x3) == 1);
	info->mode_min    = (((buf[1] >> 1) & 0x3) == 2);
	info->mode_maxmin = (((buf[1] >> 1) & 0x3) == 3);
	/* TODO: Rel. Not available on all models. */
	info->t1t2        = (buf[1] & (1 << 3)) != 0;
	info->rel         = (buf[1] & (1 << 4)) != 0;
	info->hold        = (buf[1] & (1 << 5)) != 0;
	info->lowbat      = (buf[1] & (1 << 6)) != 0;
	info->celsius     = (buf[1] & (1 << 7)) != 0;

	/* Byte 2: Further status bits. */
	info->memfull     = (buf[2] & (1 << 0)) != 0;
	info->autooff     = (buf[2] & (1 << 7)) != 0;

	/* Byte 7+8/9+10/11+12/13+14: channel T1/T2/T3/T4 temperature. */
	for (i = 0; i < NUM_CHANNELS; i++) {
		temp_u16 = buf[8 + (i * 2)];
		temp_u16 |= ((uint16_t)buf[7 + (i * 2)] << 8);
		info->temp[i] = (float)temp_u16;
	}

	/* Byte 43: Specifies whether we need to divide the value(s) by 10. */
	for (i = 0; i < NUM_CHANNELS; i++) {
		/* Bit = 0: Divide by 10. Bit = 1: Don't divide by 10. */
		if ((buf[43] & (1 << i)) == 0)
			info->temp[i] /= 10;
	}

	/* Bytes 39-42: Overflow/overlimit bits, depending on mode. */
	for (i = 0; i < NUM_CHANNELS; i++) {
		if (info->mode_std && ((buf[39] & (1 << i)) != 0))
			info->temp[i] = INFINITY;
		/* TODO: Rel. Not available on all models. */
		// if (info->mode_rel && ((buf[40] & (1 << i)) != 0))
		// 	info->temp[i] = INFINITY;
		if (info->mode_max && ((buf[41] & (1 << i)) != 0))
			info->temp[i] = INFINITY;
		if (info->mode_min && ((buf[42] & (1 << i)) != 0))
			info->temp[i] = INFINITY;
		/* TODO: Minmax? */
	}

	/* Byte 44: Always 0x03. */

	return SR_OK;
}

static int handle_packet(const uint8_t *buf, struct sr_dev_inst *sdi, int idx)
{
	struct sr_datafeed_packet packet;
	struct sr_datafeed_analog_old analog;
	struct dev_context *devc;
	struct center_info info;
	GSList *l;
	int i, ret;

	devc = sdi->priv;

	memset(&analog, 0, sizeof(struct sr_datafeed_analog_old));
	memset(&info, 0, sizeof(struct center_info));

	ret = packet_parse(buf, idx, &info);
	if (ret < 0) {
		sr_err("Failed to parse packet.");
		return SR_ERR;
	}

	/* Common values for all 4 channels. */
	packet.type = SR_DF_ANALOG_OLD;
	packet.payload = &analog;
	analog.mq = SR_MQ_TEMPERATURE;
	analog.unit = (info.celsius) ? SR_UNIT_CELSIUS : SR_UNIT_FAHRENHEIT;
	analog.num_samples = 1;

	/* Send the values for T1 - T4. */
	for (i = 0; i < NUM_CHANNELS; i++) {
		l = NULL;
		l = g_slist_append(l, g_slist_nth_data(sdi->channels, i));
		analog.channels = l;
		analog.data = &(info.temp[i]);
		sr_session_send(sdi, &packet);
		g_slist_free(l);
	}

	devc->num_samples++;

	return SR_OK;
}

/* Return TRUE if a full packet was parsed, FALSE otherwise. */
static gboolean handle_new_data(struct sr_dev_inst *sdi, int idx)
{
	struct dev_context *devc;
	struct sr_serial_dev_inst *serial;
	int len, i, offset = 0, ret = FALSE;

	devc = sdi->priv;
	serial = sdi->conn;

	/* Try to get as much data as the buffer can hold. */
	len = SERIAL_BUFSIZE - devc->buflen;
	len = serial_read_nonblocking(serial, devc->buf + devc->buflen, len);
	if (len < 1) {
		sr_err("Serial port read error: %d.", len);
		return FALSE;
	}

	devc->buflen += len;

	/* Now look for packets in that data. */
	while ((devc->buflen - offset) >= center_devs[idx].packet_size) {
		if (center_devs[idx].packet_valid(devc->buf + offset)) {
			handle_packet(devc->buf + offset, sdi, idx);
			offset += center_devs[idx].packet_size;
			ret = TRUE;
		} else {
			offset++;
		}
	}

	/* If we have any data left, move it to the beginning of our buffer. */
	for (i = 0; i < devc->buflen - offset; i++)
		devc->buf[i] = devc->buf[offset + i];
	devc->buflen -= offset;

	return ret;
}

static int receive_data(int fd, int revents, int idx, void *cb_data)
{
	struct sr_dev_inst *sdi;
	struct dev_context *devc;
	int64_t t;
	static gboolean request_new_packet = TRUE;
	struct sr_serial_dev_inst *serial;

	(void)fd;

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

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

	serial = sdi->conn;

	if (revents == G_IO_IN) {
		/* New data arrived. */
		request_new_packet = handle_new_data(sdi, idx);
	} else {
		/*
		 * Timeout. Send "A" to request a packet, but then don't send
		 * further "A" commands until we received a full packet first.
		 */
		if (request_new_packet) {
			center_send(serial, "A");
			request_new_packet = FALSE;
		}
	}

	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
		sr_info("Requested number of samples reached.");
		sdi->driver->dev_acquisition_stop(sdi);
		return TRUE;
	}

	if (devc->limit_msec) {
		t = (g_get_monotonic_time() - devc->starttime) / 1000;
		if (t > (int64_t)devc->limit_msec) {
			sr_info("Requested time limit reached.");
			sdi->driver->dev_acquisition_stop(sdi);
			return TRUE;
		}
	}

	return TRUE;
}

#define RECEIVE_DATA(ID_UPPER) \
SR_PRIV int receive_data_##ID_UPPER(int fd, int revents, void *cb_data) { \
	return receive_data(fd, revents, ID_UPPER, cb_data); }

/* Driver-specific receive_data() wrappers */
RECEIVE_DATA(CENTER_309)
RECEIVE_DATA(VOLTCRAFT_K204)
Commit	Line	Data
4433145f UH	1	/*
	2	* This file is part of the libsigrok project.
	3	*
	4	* Copyright (C) 2013 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
6ec6c43b	21	#include <config.h>
4433145f UH	22	#include "protocol.h"
4433145f UH	23
07ffa5b3 UH	24	#define NUM_CHANNELS 4
07ffa5b3 UH	25
4a8bbed7	26	struct center_info {
07ffa5b3	27	float temp[NUM_CHANNELS];
4a8bbed7 UH	28	gboolean rec, std, max, min, maxmin, t1t2, rel, hold, lowbat, celsius;
	29	gboolean memfull, autooff;
	30	gboolean mode_std, mode_rel, mode_max, mode_min, mode_maxmin;
	31	};
	32
	33	static int center_send(struct sr_serial_dev_inst serial, const char cmd)
4433145f	34	{
4a8bbed7 UH	35	int ret;
4a8bbed7 UH	36
84d328ac UH	37	if ((ret = serial_write_blocking(serial, cmd, strlen(cmd),
84d328ac UH	38	serial_timeout(serial, strlen(cmd)))) < 0) {
4a8bbed7 UH	39	sr_err("Error sending '%s' command: %d.", cmd, ret);
	40	return SR_ERR;
	41	}
	42
	43	return SR_OK;
	44	}
	45
	46	SR_PRIV gboolean center_3xx_packet_valid(const uint8_t *buf)
	47	{
	48	return (buf[0] == 0x02 && buf[44] == 0x03);
	49	}
	50
	51	static void log_packet(const uint8_t *buf, int idx)
	52	{
	53	int i;
	54	GString *s;
	55
	56	s = g_string_sized_new(100);
	57	g_string_printf(s, "Packet: ");
	58	for (i = 0; i < center_devs[idx].packet_size; i++)
	59	g_string_append_printf(s, "%02x ", buf[i]);
	60	sr_spew("%s", s->str);
	61	g_string_free(s, TRUE);
	62	}
	63
	64	static int packet_parse(const uint8_t buf, int idx, struct center_info info)
	65	{
	66	int i;
	67	uint16_t temp_u16;
	68
	69	log_packet(buf, idx);
	70
	71	/* Byte 0: Always 0x02. */
	72
	73	/* Byte 1: Various status bits. */
	74	info->rec = (buf[1] & (1 << 0)) != 0;
	75	info->mode_std = (((buf[1] >> 1) & 0x3) == 0);
	76	info->mode_max = (((buf[1] >> 1) & 0x3) == 1);
	77	info->mode_min = (((buf[1] >> 1) & 0x3) == 2);
	78	info->mode_maxmin = (((buf[1] >> 1) & 0x3) == 3);
	79	/* TODO: Rel. Not available on all models. */
	80	info->t1t2 = (buf[1] & (1 << 3)) != 0;
	81	info->rel = (buf[1] & (1 << 4)) != 0;
	82	info->hold = (buf[1] & (1 << 5)) != 0;
	83	info->lowbat = (buf[1] & (1 << 6)) != 0;
	84	info->celsius = (buf[1] & (1 << 7)) != 0;
	85
	86	/* Byte 2: Further status bits. */
	87	info->memfull = (buf[2] & (1 << 0)) != 0;
	88	info->autooff = (buf[2] & (1 << 7)) != 0;
	89
	90	/* Byte 7+8/9+10/11+12/13+14: channel T1/T2/T3/T4 temperature. */
07ffa5b3	91	for (i = 0; i < NUM_CHANNELS; i++) {
4a8bbed7 UH	92	temp_u16 = buf[8 + (i * 2)];
	93	temp_u16 \|= ((uint16_t)buf[7 + (i * 2)] << 8);
	94	info->temp[i] = (float)temp_u16;
	95	}
	96
	97	/* Byte 43: Specifies whether we need to divide the value(s) by 10. */
07ffa5b3	98	for (i = 0; i < NUM_CHANNELS; i++) {
4a8bbed7 UH	99	/* Bit = 0: Divide by 10. Bit = 1: Don't divide by 10. */
	100	if ((buf[43] & (1 << i)) == 0)
	101	info->temp[i] /= 10;
	102	}
	103
	104	/* Bytes 39-42: Overflow/overlimit bits, depending on mode. */
07ffa5b3	105	for (i = 0; i < NUM_CHANNELS; i++) {
4a8bbed7 UH	106	if (info->mode_std && ((buf[39] & (1 << i)) != 0))
	107	info->temp[i] = INFINITY;
	108	/* TODO: Rel. Not available on all models. */
	109	// if (info->mode_rel && ((buf[40] & (1 << i)) != 0))
	110	// info->temp[i] = INFINITY;
	111	if (info->mode_max && ((buf[41] & (1 << i)) != 0))
	112	info->temp[i] = INFINITY;
	113	if (info->mode_min && ((buf[42] & (1 << i)) != 0))
	114	info->temp[i] = INFINITY;
	115	/* TODO: Minmax? */
	116	}
	117
	118	/* Byte 44: Always 0x03. */
	119
	120	return SR_OK;
	121	}
	122
	123	static int handle_packet(const uint8_t buf, struct sr_dev_inst sdi, int idx)
	124	{
	125	struct sr_datafeed_packet packet;
5faebab2	126	struct sr_datafeed_analog_old analog;
4a8bbed7 UH	127	struct dev_context *devc;
	128	struct center_info info;
	129	GSList *l;
	130	int i, ret;
	131
	132	devc = sdi->priv;
4433145f	133
5faebab2	134	memset(&analog, 0, sizeof(struct sr_datafeed_analog_old));
4a8bbed7 UH	135	memset(&info, 0, sizeof(struct center_info));
	136
	137	ret = packet_parse(buf, idx, &info);
	138	if (ret < 0) {
	139	sr_err("Failed to parse packet.");
	140	return SR_ERR;
	141	}
	142
ba7dd8bb	143	/* Common values for all 4 channels. */
5faebab2	144	packet.type = SR_DF_ANALOG_OLD;
4a8bbed7 UH	145	packet.payload = &analog;
	146	analog.mq = SR_MQ_TEMPERATURE;
	147	analog.unit = (info.celsius) ? SR_UNIT_CELSIUS : SR_UNIT_FAHRENHEIT;
	148	analog.num_samples = 1;
	149
	150	/* Send the values for T1 - T4. */
07ffa5b3	151	for (i = 0; i < NUM_CHANNELS; i++) {
4a8bbed7	152	l = NULL;
ba7dd8bb UH	153	l = g_slist_append(l, g_slist_nth_data(sdi->channels, i));
ba7dd8bb UH	154	analog.channels = l;
4a8bbed7	155	analog.data = &(info.temp[i]);
695dc859	156	sr_session_send(sdi, &packet);
4a8bbed7 UH	157	g_slist_free(l);
	158	}
	159
	160	devc->num_samples++;
	161
	162	return SR_OK;
	163	}
	164
	165	/* Return TRUE if a full packet was parsed, FALSE otherwise. */
	166	static gboolean handle_new_data(struct sr_dev_inst *sdi, int idx)
	167	{
	168	struct dev_context *devc;
	169	struct sr_serial_dev_inst *serial;
	170	int len, i, offset = 0, ret = FALSE;
	171
	172	devc = sdi->priv;
	173	serial = sdi->conn;
	174
	175	/* Try to get as much data as the buffer can hold. */
	176	len = SERIAL_BUFSIZE - devc->buflen;
3582ce8a	177	len = serial_read_nonblocking(serial, devc->buf + devc->buflen, len);
4a8bbed7 UH	178	if (len < 1) {
	179	sr_err("Serial port read error: %d.", len);
	180	return FALSE;
	181	}
	182
	183	devc->buflen += len;
	184
	185	/* Now look for packets in that data. */
	186	while ((devc->buflen - offset) >= center_devs[idx].packet_size) {
	187	if (center_devs[idx].packet_valid(devc->buf + offset)) {
	188	handle_packet(devc->buf + offset, sdi, idx);
	189	offset += center_devs[idx].packet_size;
	190	ret = TRUE;
	191	} else {
	192	offset++;
	193	}
	194	}
	195
	196	/* If we have any data left, move it to the beginning of our buffer. */
	197	for (i = 0; i < devc->buflen - offset; i++)
	198	devc->buf[i] = devc->buf[offset + i];
	199	devc->buflen -= offset;
	200
	201	return ret;
	202	}
	203
	204	static int receive_data(int fd, int revents, int idx, void *cb_data)
	205	{
	206	struct sr_dev_inst *sdi;
4433145f	207	struct dev_context *devc;
4a8bbed7 UH	208	int64_t t;
	209	static gboolean request_new_packet = TRUE;
	210	struct sr_serial_dev_inst *serial;
	211
	212	(void)fd;
4433145f UH	213
	214	if (!(sdi = cb_data))
	215	return TRUE;
	216
	217	if (!(devc = sdi->priv))
	218	return TRUE;
	219
4a8bbed7 UH	220	serial = sdi->conn;
4a8bbed7 UH	221
4433145f	222	if (revents == G_IO_IN) {
4a8bbed7 UH	223	/* New data arrived. */
	224	request_new_packet = handle_new_data(sdi, idx);
	225	} else {
	226	/*
	227	* Timeout. Send "A" to request a packet, but then don't send
	228	* further "A" commands until we received a full packet first.
	229	*/
	230	if (request_new_packet) {
	231	center_send(serial, "A");
	232	request_new_packet = FALSE;
	233	}
	234	}
	235
	236	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
	237	sr_info("Requested number of samples reached.");
695dc859	238	sdi->driver->dev_acquisition_stop(sdi);
4a8bbed7 UH	239	return TRUE;
	240	}
	241
	242	if (devc->limit_msec) {
	243	t = (g_get_monotonic_time() - devc->starttime) / 1000;
	244	if (t > (int64_t)devc->limit_msec) {
	245	sr_info("Requested time limit reached.");
695dc859	246	sdi->driver->dev_acquisition_stop(sdi);
4a8bbed7 UH	247	return TRUE;
4a8bbed7 UH	248	}
4433145f UH	249	}
	250
	251	return TRUE;
	252	}
4a8bbed7 UH	253
	254	#define RECEIVE_DATA(ID_UPPER) \
	255	SR_PRIV int receive_data_##ID_UPPER(int fd, int revents, void *cb_data) { \
	256	return receive_data(fd, revents, ID_UPPER, cb_data); }
	257
	258	/* Driver-specific receive_data() wrappers */
	259	RECEIVE_DATA(CENTER_309)
	260	RECEIVE_DATA(VOLTCRAFT_K204)