[libsigrok.git] / hardware / mic-985xx / 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 "protocol.h"

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

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

	return SR_OK;
}

SR_PRIV int mic_cmd_get_device_info(struct sr_serial_dev_inst *serial)
{
	return mic_send(serial, "I\r");
}

static int mic_cmd_set_realtime_mode(struct sr_serial_dev_inst *serial)
{
	return mic_send(serial, "S 1 M 2 32 3\r");
}

SR_PRIV gboolean packet_valid_temp(const uint8_t *buf)
{
	if (buf[0] != 'v' || buf[1] != ' ' || buf[5] != '\r')
		return FALSE;

	if (!isdigit(buf[2]) || !isdigit(buf[3]) || !isdigit(buf[4]))
		return FALSE;

	return TRUE;
}

SR_PRIV gboolean packet_valid_temp_hum(const uint8_t *buf)
{
	if (buf[0] != 'v' || buf[1] != ' ' || buf[5] != ' ' || buf[9] != '\r')
		return FALSE;

	if (!isdigit(buf[2]) || !isdigit(buf[3]) || !isdigit(buf[4]))
		return FALSE;

	if (!isdigit(buf[6]) || !isdigit(buf[7]) || !isdigit(buf[8]))
		return FALSE;

	return TRUE;
}

static int packet_parse(const char *buf, int idx, float *temp, float *humidity)
{
	char tmp[4];

	/* Packet format MIC98581: "v ttt\r". */
	/* Packet format MIC98583: "v ttt hhh\r". */

	/* TODO: Sanity check on buf. For now we assume well-formed ASCII. */

	tmp[3] = '\0';

	strncpy((char *)&tmp, &buf[2], 3);
	*temp = g_ascii_strtoull((const char *)&tmp, NULL, 10) / 10;

	if (mic_devs[idx].has_humidity) {
		strncpy((char *)&tmp, &buf[6], 3);
		*humidity = g_ascii_strtoull((const char *)&tmp, NULL, 10) / 10;
	}

	return SR_OK;
}

static int handle_packet(const uint8_t *buf, struct sr_dev_inst *sdi, int idx)
{
	float temperature, humidity;
	struct sr_datafeed_packet packet;
	struct sr_datafeed_analog analog;
	struct dev_context *devc;
	GSList *l;
	int ret;

	(void)idx;

	devc = sdi->priv;

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

	/* Clear 'analog', otherwise it'll contain random garbage. */
	memset(&analog, 0, sizeof(struct sr_datafeed_analog));

	/* Common values for both probes. */
	packet.type = SR_DF_ANALOG;
	packet.payload = &analog;
	analog.num_samples = 1;

	/* Temperature. */
	l = g_slist_copy(sdi->probes);
	l = g_slist_remove_link(l, g_slist_nth(l, 1));
	analog.probes = l;
	analog.mq = SR_MQ_TEMPERATURE;
	analog.unit = SR_UNIT_CELSIUS; /* TODO: Use C/F correctly. */
	analog.data = &temperature;
	sr_session_send(devc->cb_data, &packet);
	g_slist_free(l);

	/* Humidity. */
	if (mic_devs[idx].has_humidity) {
		l = g_slist_copy(sdi->probes);
		l = g_slist_remove_link(l, g_slist_nth(l, 0));
		analog.probes = l;
		analog.mq = SR_MQ_RELATIVE_HUMIDITY;
		analog.unit = SR_UNIT_PERCENTAGE;
		analog.data = &humidity;
		sr_session_send(devc->cb_data, &packet);
		g_slist_free(l);
	}

	devc->num_samples++;

	return SR_OK;
}

static void handle_new_data(struct sr_dev_inst *sdi, int idx)
{
	struct dev_context *devc;
	int len, i, offset = 0;

	devc = sdi->priv;

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

	devc->buflen += len;

	/* Now look for packets in that data. */
	while ((devc->buflen - offset) >= mic_devs[idx].packet_size) {
		if (mic_devs[idx].packet_valid(devc->buf + offset)) {
			handle_packet(devc->buf + offset, sdi, idx);
			offset += mic_devs[idx].packet_size;
		} 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;
}

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 first_time = TRUE;

	(void)fd;

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

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

	if (revents == G_IO_IN) {
		/* New data arrived. */
		handle_new_data(sdi, idx);
	} else {
		/* Timeout. */
		if (first_time) {
			mic_cmd_set_realtime_mode(devc->serial);
			first_time = FALSE;
		}
	}

	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
		sr_info("Requested number of samples reached.");
		sdi->driver->dev_acquisition_stop(sdi, cb_data);
		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, cb_data);
			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(MIC_98581)
RECEIVE_DATA(MIC_98583)
Commit	Line	Data
7ec5b549 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
	21	#include "protocol.h"
	22
0cd8e231	23	static int mic_send(struct sr_serial_dev_inst serial, const char cmd)
7ec5b549	24	{
0cd8e231 UH	25	int ret;
	26
	27	if ((ret = serial_write(serial, cmd, strlen(cmd))) < 0) {
	28	sr_err("Error sending '%s' command: %d.", cmd, ret);
	29	return SR_ERR;
	30	}
	31
	32	return SR_OK;
	33	}
	34
	35	SR_PRIV int mic_cmd_get_device_info(struct sr_serial_dev_inst *serial)
	36	{
	37	return mic_send(serial, "I\r");
	38	}
	39
	40	static int mic_cmd_set_realtime_mode(struct sr_serial_dev_inst *serial)
	41	{
	42	return mic_send(serial, "S 1 M 2 32 3\r");
	43	}
	44
6f3e5335 UH	45	SR_PRIV gboolean packet_valid_temp(const uint8_t *buf)
	46	{
	47	if (buf[0] != 'v' \|\| buf[1] != ' ' \|\| buf[5] != '\r')
	48	return FALSE;
	49
	50	if (!isdigit(buf[2]) \|\| !isdigit(buf[3]) \|\| !isdigit(buf[4]))
	51	return FALSE;
	52
	53	return TRUE;
	54	}
	55
	56	SR_PRIV gboolean packet_valid_temp_hum(const uint8_t *buf)
0cd8e231 UH	57	{
	58	if (buf[0] != 'v' \|\| buf[1] != ' ' \|\| buf[5] != ' ' \|\| buf[9] != '\r')
	59	return FALSE;
	60
	61	if (!isdigit(buf[2]) \|\| !isdigit(buf[3]) \|\| !isdigit(buf[4]))
	62	return FALSE;
	63
	64	if (!isdigit(buf[6]) \|\| !isdigit(buf[7]) \|\| !isdigit(buf[8]))
	65	return FALSE;
	66
	67	return TRUE;
	68	}
	69
6f3e5335	70	static int packet_parse(const char buf, int idx, float temp, float *humidity)
0cd8e231 UH	71	{
	72	char tmp[4];
	73
6f3e5335 UH	74	/* Packet format MIC98581: "v ttt\r". */
6f3e5335 UH	75	/* Packet format MIC98583: "v ttt hhh\r". */
0cd8e231 UH	76
	77	/* TODO: Sanity check on buf. For now we assume well-formed ASCII. */
	78
	79	tmp[3] = '\0';
	80
	81	strncpy((char *)&tmp, &buf[2], 3);
	82	temp = g_ascii_strtoull((const char )&tmp, NULL, 10) / 10;
	83
6f3e5335 UH	84	if (mic_devs[idx].has_humidity) {
	85	strncpy((char *)&tmp, &buf[6], 3);
	86	humidity = g_ascii_strtoull((const char )&tmp, NULL, 10) / 10;
	87	}
0cd8e231 UH	88
	89	return SR_OK;
	90	}
	91
	92	static int handle_packet(const uint8_t buf, struct sr_dev_inst sdi, int idx)
	93	{
	94	float temperature, humidity;
	95	struct sr_datafeed_packet packet;
	96	struct sr_datafeed_analog analog;
	97	struct dev_context *devc;
	98	GSList *l;
	99	int ret;
7ec5b549	100
0cd8e231 UH	101	(void)idx;
	102
	103	devc = sdi->priv;
	104
6f3e5335	105	ret = packet_parse((const char *)buf, idx, &temperature, &humidity);
0cd8e231 UH	106	if (ret < 0) {
	107	sr_err("Failed to parse packet.");
	108	return SR_ERR;
	109	}
	110
ff17e6ba UH	111	/* Clear 'analog', otherwise it'll contain random garbage. */
	112	memset(&analog, 0, sizeof(struct sr_datafeed_analog));
	113
0cd8e231 UH	114	/* Common values for both probes. */
	115	packet.type = SR_DF_ANALOG;
	116	packet.payload = &analog;
	117	analog.num_samples = 1;
	118
	119	/* Temperature. */
	120	l = g_slist_copy(sdi->probes);
	121	l = g_slist_remove_link(l, g_slist_nth(l, 1));
	122	analog.probes = l;
	123	analog.mq = SR_MQ_TEMPERATURE;
	124	analog.unit = SR_UNIT_CELSIUS; /* TODO: Use C/F correctly. */
	125	analog.data = &temperature;
	126	sr_session_send(devc->cb_data, &packet);
	127	g_slist_free(l);
	128
	129	/* Humidity. */
6f3e5335 UH	130	if (mic_devs[idx].has_humidity) {
	131	l = g_slist_copy(sdi->probes);
	132	l = g_slist_remove_link(l, g_slist_nth(l, 0));
	133	analog.probes = l;
	134	analog.mq = SR_MQ_RELATIVE_HUMIDITY;
	135	analog.unit = SR_UNIT_PERCENTAGE;
	136	analog.data = &humidity;
	137	sr_session_send(devc->cb_data, &packet);
	138	g_slist_free(l);
	139	}
0cd8e231 UH	140
	141	devc->num_samples++;
	142
	143	return SR_OK;
	144	}
	145
	146	static void handle_new_data(struct sr_dev_inst *sdi, int idx)
	147	{
7ec5b549	148	struct dev_context *devc;
0cd8e231 UH	149	int len, i, offset = 0;
	150
	151	devc = sdi->priv;
	152
	153	/* Try to get as much data as the buffer can hold. */
	154	len = SERIAL_BUFSIZE - devc->buflen;
	155	len = serial_read(devc->serial, devc->buf + devc->buflen, len);
	156	if (len < 1) {
	157	sr_err("Serial port read error: %d.", len);
	158	return;
	159	}
	160
	161	devc->buflen += len;
	162
	163	/* Now look for packets in that data. */
6f3e5335 UH	164	while ((devc->buflen - offset) >= mic_devs[idx].packet_size) {
6f3e5335 UH	165	if (mic_devs[idx].packet_valid(devc->buf + offset)) {
0cd8e231	166	handle_packet(devc->buf + offset, sdi, idx);
6f3e5335	167	offset += mic_devs[idx].packet_size;
0cd8e231 UH	168	} else {
	169	offset++;
	170	}
	171	}
	172
	173	/* If we have any data left, move it to the beginning of our buffer. */
	174	for (i = 0; i < devc->buflen - offset; i++)
	175	devc->buf[i] = devc->buf[offset + i];
	176	devc->buflen -= offset;
	177	}
	178
	179	static int receive_data(int fd, int revents, int idx, void *cb_data)
	180	{
	181	struct sr_dev_inst *sdi;
	182	struct dev_context *devc;
	183	int64_t t;
	184	static gboolean first_time = TRUE;
	185
	186	(void)fd;
7ec5b549 UH	187
	188	if (!(sdi = cb_data))
	189	return TRUE;
	190
	191	if (!(devc = sdi->priv))
	192	return TRUE;
	193
	194	if (revents == G_IO_IN) {
0cd8e231 UH	195	/* New data arrived. */
	196	handle_new_data(sdi, idx);
	197	} else {
	198	/* Timeout. */
	199	if (first_time) {
	200	mic_cmd_set_realtime_mode(devc->serial);
	201	first_time = FALSE;
	202	}
	203	}
	204
	205	if (devc->limit_samples && devc->num_samples >= devc->limit_samples) {
	206	sr_info("Requested number of samples reached.");
	207	sdi->driver->dev_acquisition_stop(sdi, cb_data);
	208	return TRUE;
	209	}
	210
	211	if (devc->limit_msec) {
	212	t = (g_get_monotonic_time() - devc->starttime) / 1000;
	213	if (t > (int64_t)devc->limit_msec) {
	214	sr_info("Requested time limit reached.");
	215	sdi->driver->dev_acquisition_stop(sdi, cb_data);
	216	return TRUE;
	217	}
7ec5b549 UH	218	}
	219
	220	return TRUE;
	221	}
0cd8e231 UH	222
	223	#define RECEIVE_DATA(ID_UPPER) \
	224	SR_PRIV int receive_data_##ID_UPPER(int fd, int revents, void *cb_data) { \
	225	return receive_data(fd, revents, ID_UPPER, cb_data); }
	226
	227	/* Driver-specific receive_data() wrappers */
6f3e5335	228	RECEIVE_DATA(MIC_98581)
0cd8e231	229	RECEIVE_DATA(MIC_98583)