[libsigrok.git] / src / hardware / arachnid-labs-re-load-pro / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2015-2016 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, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include <string.h>
#include "protocol.h"

#define READ_TIMEOUT_MS 1000

static int send_cmd(const struct sr_dev_inst *sdi, const char *cmd,
		char *replybuf, int replybufsize)
{
	char *bufptr;
	int len, ret;
	struct sr_serial_dev_inst *serial;

	serial = sdi->conn;

	/* Send the command (blocking, with timeout). */
	if ((ret = serial_write_blocking(serial, cmd,
			strlen(cmd), serial_timeout(serial,
			strlen(cmd)))) < (int)strlen(cmd)) {
		sr_err("Unable to send command.");
		return SR_ERR;
	}

	/* Read the reply (blocking, with timeout). */
	memset(replybuf, 0, replybufsize);
	bufptr = replybuf;
	len = replybufsize;
	ret = serial_readline(serial, &bufptr, &len, READ_TIMEOUT_MS);

	/* If we got 0 characters (possibly one \r or \n), retry once. */
	if (len == 0) {
		len = replybufsize;
		ret = serial_readline(serial, &bufptr, &len, READ_TIMEOUT_MS);
	}

	if (g_str_has_prefix((const char *)&bufptr, "err ")) {
		sr_err("Device replied with an error: '%s'.", bufptr);
		return SR_ERR;
	}

	return ret;
}

SR_PRIV int reloadpro_set_current_limit(const struct sr_dev_inst *sdi,
					float current)
{
	int ret, ma;
	char buf[100];
	char *cmd;

	if (current < 0 || current > 6) {
		sr_err("The current limit must be 0-6 A (was %f A).", current);
		return SR_ERR_ARG;
	}

	/* Hardware expects current in mA, integer (0..6000). */
	ma = (int)(current * 1000);

	sr_err("Setting current limit to %f A (%d mA).", current, ma);

	cmd = g_strdup_printf("set %d\n", ma);
	if ((ret = send_cmd(sdi, cmd, (char *)&buf, sizeof(buf))) < 0) {
		sr_err("Error sending current limit command: %d.", ret);
		g_free(cmd);
		return SR_ERR;
	}
	g_free(cmd);

	return SR_OK;
}

SR_PRIV int reloadpro_set_on_off(const struct sr_dev_inst *sdi, gboolean on)
{
	int ret;
	char buf[100];
	const char *cmd;

	cmd = (on) ? "on\n" : "off\n";
	if ((ret = send_cmd(sdi, cmd, (char *)&buf, sizeof(buf))) < 0) {
		sr_err("Error sending on/off command: %d.", ret);
		return SR_ERR;
	}

	return SR_OK;
}

SR_PRIV int reloadpro_get_current_limit(const struct sr_dev_inst *sdi,
					float *current)
{
	int ret;
	char buf[100];

	if ((ret = send_cmd(sdi, "set\n", (char *)&buf, sizeof(buf))) < 0) {
		sr_err("Error sending current limit query: %d.", ret);
		return SR_ERR;
	}

	/* Hardware sends current in mA, integer (0..6000). */
	*current = g_ascii_strtod(buf + 4, NULL) / 1000;

	return SR_OK;
}

SR_PRIV int reloadpro_get_voltage_current(const struct sr_dev_inst *sdi,
		float *voltage, float *current)
{
	int ret;
	char buf[100];
	char **tokens;

	if ((ret = send_cmd(sdi, "read\n", (char *)&buf, sizeof(buf))) < 0) {
		sr_err("Error sending voltage/current query: %d.", ret);
		return SR_ERR;
	}

	/* Reply: "read <current> <voltage>". */
	tokens = g_strsplit((const char *)&buf, " ", 3);
	if (voltage)
		*voltage = g_ascii_strtod(tokens[2], NULL) / 1000;
	if (current)
		*current = g_ascii_strtod(tokens[1], NULL) / 1000;
	g_strfreev(tokens);

	return SR_OK;
}

static void handle_packet(const struct sr_dev_inst *sdi)
{
	float voltage, current;
	struct sr_datafeed_packet packet;
	struct sr_datafeed_analog analog;
	struct sr_analog_encoding encoding;
	struct sr_analog_meaning meaning;
	struct sr_analog_spec spec;
	struct dev_context *devc;
	char **tokens;
	GSList *l;

	devc = sdi->priv;

	if (g_str_has_prefix((const char *)devc->buf, "overtemp")) {
		sr_warn("Overtemperature condition!");
		devc->otp_active = TRUE;
		return;
	}

	if (g_str_has_prefix((const char *)devc->buf, "undervolt")) {
		sr_warn("Undervoltage condition!");
		devc->uvc_active = TRUE;
		return;
	}

	if (!g_str_has_prefix((const char *)devc->buf, "read ")) {
		sr_dbg("Unknown packet: '%s'.", devc->buf);
		return;
	}

	tokens = g_strsplit((const char *)devc->buf, " ", 3);
	voltage = g_ascii_strtod(tokens[2], NULL) / 1000;
	current = g_ascii_strtod(tokens[1], NULL) / 1000;
	g_strfreev(tokens);

	/* Begin frame. */
	packet.type = SR_DF_FRAME_BEGIN;
	packet.payload = NULL;
	sr_session_send(sdi, &packet);

	sr_analog_init(&analog, &encoding, &meaning, &spec, 4);

	packet.type = SR_DF_ANALOG;
	packet.payload = &analog;
	analog.num_samples = 1;

	/* Voltage */
	l = g_slist_copy(sdi->channels);
	l = g_slist_remove_link(l, g_slist_nth(l, 1));
	meaning.channels = l;
	meaning.mq = SR_MQ_VOLTAGE;
	meaning.mqflags = SR_MQFLAG_DC;
	meaning.unit = SR_UNIT_VOLT;
	analog.data = &voltage;
	sr_session_send(sdi, &packet);
	g_slist_free(l);

	/* Current */
	l = g_slist_copy(sdi->channels);
	l = g_slist_remove_link(l, g_slist_nth(l, 0));
	meaning.channels = l;
	meaning.mq = SR_MQ_CURRENT;
	meaning.mqflags = SR_MQFLAG_DC;
	meaning.unit = SR_UNIT_AMPERE;
	analog.data = &current;
	sr_session_send(sdi, &packet);
	g_slist_free(l);

	/* End frame. */
	packet.type = SR_DF_FRAME_END;
	packet.payload = NULL;
	sr_session_send(sdi, &packet);

	sr_sw_limits_update_samples_read(&devc->limits, 1);
}

static void handle_new_data(const struct sr_dev_inst *sdi)
{
	int len;
	struct dev_context *devc;
	struct sr_serial_dev_inst *serial;

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

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

	if (g_str_has_suffix((const char *)devc->buf, "\n")) {
		handle_packet(sdi);
		memset(devc->buf, 0, RELOADPRO_BUFSIZE);
		devc->buflen = 0;
	}
}

SR_PRIV int reloadpro_receive_data(int fd, int revents, void *cb_data)
{
	struct sr_dev_inst *sdi;
	struct dev_context *devc;

	(void)fd;

	sdi = cb_data;
	devc = sdi->priv;

	if (revents != G_IO_IN)
		return TRUE;

	handle_new_data(sdi);

	if (sr_sw_limits_check(&devc->limits))
		sr_dev_acquisition_stop(sdi);

	return TRUE;
}
Commit	Line	Data
6e8d31d4 UH	1	/*
	2	* This file is part of the libsigrok project.
	3	*
803db07a	4	* Copyright (C) 2015-2016 Uwe Hermann <uwe@hermann-uwe.de>
6e8d31d4 UH	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
2ea1fdf1	17	* along with this program; if not, see <http://www.gnu.org/licenses/>.
6e8d31d4 UH	18	*/
	19
	20	#include <config.h>
803db07a	21	#include <string.h>
6e8d31d4 UH	22	#include "protocol.h"
6e8d31d4 UH	23
803db07a UH	24	#define READ_TIMEOUT_MS 1000
	25
	26	static int send_cmd(const struct sr_dev_inst sdi, const char cmd,
	27	char *replybuf, int replybufsize)
	28	{
	29	char *bufptr;
	30	int len, ret;
	31	struct sr_serial_dev_inst *serial;
	32
	33	serial = sdi->conn;
	34
	35	/* Send the command (blocking, with timeout). */
	36	if ((ret = serial_write_blocking(serial, cmd,
	37	strlen(cmd), serial_timeout(serial,
	38	strlen(cmd)))) < (int)strlen(cmd)) {
	39	sr_err("Unable to send command.");
	40	return SR_ERR;
	41	}
	42
	43	/* Read the reply (blocking, with timeout). */
	44	memset(replybuf, 0, replybufsize);
	45	bufptr = replybuf;
	46	len = replybufsize;
	47	ret = serial_readline(serial, &bufptr, &len, READ_TIMEOUT_MS);
	48
	49	/* If we got 0 characters (possibly one \r or \n), retry once. */
	50	if (len == 0) {
	51	len = replybufsize;
	52	ret = serial_readline(serial, &bufptr, &len, READ_TIMEOUT_MS);
	53	}
	54
	55	if (g_str_has_prefix((const char *)&bufptr, "err ")) {
	56	sr_err("Device replied with an error: '%s'.", bufptr);
	57	return SR_ERR;
	58	}
	59
	60	return ret;
	61	}
	62
	63	SR_PRIV int reloadpro_set_current_limit(const struct sr_dev_inst *sdi,
	64	float current)
	65	{
	66	int ret, ma;
	67	char buf[100];
	68	char *cmd;
	69
	70	if (current < 0 \|\| current > 6) {
	71	sr_err("The current limit must be 0-6 A (was %f A).", current);
	72	return SR_ERR_ARG;
	73	}
	74
	75	/* Hardware expects current in mA, integer (0..6000). */
	76	ma = (int)(current * 1000);
	77
	78	sr_err("Setting current limit to %f A (%d mA).", current, ma);
	79
	80	cmd = g_strdup_printf("set %d\n", ma);
	81	if ((ret = send_cmd(sdi, cmd, (char *)&buf, sizeof(buf))) < 0) {
	82	sr_err("Error sending current limit command: %d.", ret);
	83	g_free(cmd);
	84	return SR_ERR;
	85	}
	86	g_free(cmd);
	87
88	return SR_OK;
89	}
90
8501448c UH	91	SR_PRIV int reloadpro_set_on_off(const struct sr_dev_inst *sdi, gboolean on)
	92	{
	93	int ret;
	94	char buf[100];
	95	const char *cmd;
	96
	97	cmd = (on) ? "on\n" : "off\n";
	98	if ((ret = send_cmd(sdi, cmd, (char *)&buf, sizeof(buf))) < 0) {
	99	sr_err("Error sending on/off command: %d.", ret);
	100	return SR_ERR;
	101	}
	102
	103	return SR_OK;
	104	}
	105
803db07a UH	106	SR_PRIV int reloadpro_get_current_limit(const struct sr_dev_inst *sdi,
	107	float *current)
	108	{
	109	int ret;
	110	char buf[100];
	111
	112	if ((ret = send_cmd(sdi, "set\n", (char *)&buf, sizeof(buf))) < 0) {
	113	sr_err("Error sending current limit query: %d.", ret);
	114	return SR_ERR;
	115	}
	116
	117	/* Hardware sends current in mA, integer (0..6000). */
	118	*current = g_ascii_strtod(buf + 4, NULL) / 1000;
	119
	120	return SR_OK;
	121	}
	122
	123	SR_PRIV int reloadpro_get_voltage_current(const struct sr_dev_inst *sdi,
	124	float voltage, float current)
	125	{
	126	int ret;
	127	char buf[100];
	128	char **tokens;
	129
	130	if ((ret = send_cmd(sdi, "read\n", (char *)&buf, sizeof(buf))) < 0) {
	131	sr_err("Error sending voltage/current query: %d.", ret);
	132	return SR_ERR;
	133	}
	134
	135	/* Reply: "read <current> <voltage>". */
	136	tokens = g_strsplit((const char *)&buf, " ", 3);
	137	if (voltage)
	138	*voltage = g_ascii_strtod(tokens[2], NULL) / 1000;
	139	if (current)
	140	*current = g_ascii_strtod(tokens[1], NULL) / 1000;
	141	g_strfreev(tokens);
	142
	143	return SR_OK;
	144	}
	145
	146	static void handle_packet(const struct sr_dev_inst *sdi)
6e8d31d4	147	{
803db07a UH	148	float voltage, current;
803db07a UH	149	struct sr_datafeed_packet packet;
6e68da51 UH	150	struct sr_datafeed_analog analog;
	151	struct sr_analog_encoding encoding;
	152	struct sr_analog_meaning meaning;
	153	struct sr_analog_spec spec;
6e8d31d4	154	struct dev_context *devc;
803db07a UH	155	char **tokens;
	156	GSList *l;
	157
	158	devc = sdi->priv;
	159
	160	if (g_str_has_prefix((const char *)devc->buf, "overtemp")) {
2217be1d	161	sr_warn("Overtemperature condition!");
803db07a UH	162	devc->otp_active = TRUE;
	163	return;
	164	}
	165
2217be1d UH	166	if (g_str_has_prefix((const char *)devc->buf, "undervolt")) {
	167	sr_warn("Undervoltage condition!");
	168	devc->uvc_active = TRUE;
	169	return;
	170	}
	171
803db07a UH	172	if (!g_str_has_prefix((const char *)devc->buf, "read ")) {
	173	sr_dbg("Unknown packet: '%s'.", devc->buf);
	174	return;
	175	}
	176
	177	tokens = g_strsplit((const char *)devc->buf, " ", 3);
	178	voltage = g_ascii_strtod(tokens[2], NULL) / 1000;
	179	current = g_ascii_strtod(tokens[1], NULL) / 1000;
	180	g_strfreev(tokens);
	181
803db07a UH	182	/* Begin frame. */
803db07a UH	183	packet.type = SR_DF_FRAME_BEGIN;
6e68da51	184	packet.payload = NULL;
803db07a UH	185	sr_session_send(sdi, &packet);
803db07a UH	186
6e68da51 UH	187	sr_analog_init(&analog, &encoding, &meaning, &spec, 4);
	188
	189	packet.type = SR_DF_ANALOG;
803db07a UH	190	packet.payload = &analog;
	191	analog.num_samples = 1;
	192
	193	/* Voltage */
	194	l = g_slist_copy(sdi->channels);
	195	l = g_slist_remove_link(l, g_slist_nth(l, 1));
6e68da51 UH	196	meaning.channels = l;
	197	meaning.mq = SR_MQ_VOLTAGE;
	198	meaning.mqflags = SR_MQFLAG_DC;
	199	meaning.unit = SR_UNIT_VOLT;
803db07a UH	200	analog.data = &voltage;
	201	sr_session_send(sdi, &packet);
	202	g_slist_free(l);
	203
	204	/* Current */
	205	l = g_slist_copy(sdi->channels);
	206	l = g_slist_remove_link(l, g_slist_nth(l, 0));
6e68da51 UH	207	meaning.channels = l;
	208	meaning.mq = SR_MQ_CURRENT;
	209	meaning.mqflags = SR_MQFLAG_DC;
	210	meaning.unit = SR_UNIT_AMPERE;
803db07a UH	211	analog.data = &current;
	212	sr_session_send(sdi, &packet);
	213	g_slist_free(l);
	214
	215	/* End frame. */
	216	packet.type = SR_DF_FRAME_END;
6e68da51	217	packet.payload = NULL;
803db07a UH	218	sr_session_send(sdi, &packet);
803db07a UH	219
014c7f93	220	sr_sw_limits_update_samples_read(&devc->limits, 1);
803db07a UH	221	}
	222
	223	static void handle_new_data(const struct sr_dev_inst *sdi)
	224	{
	225	int len;
	226	struct dev_context *devc;
	227	struct sr_serial_dev_inst *serial;
	228
	229	devc = sdi->priv;
	230	serial = sdi->conn;
	231
	232	/* Try to get as much data as the buffer can hold. */
	233	len = RELOADPRO_BUFSIZE - devc->buflen;
	234	len = serial_read_nonblocking(serial, devc->buf + devc->buflen, len);
	235	if (len == 0)
	236	return; /* No new bytes, nothing to do. */
	237	if (len < 0) {
	238	sr_err("Serial port read error: %d.", len);
	239	return;
	240	}
	241	devc->buflen += len;
	242
	243	if (g_str_has_suffix((const char *)devc->buf, "\n")) {
	244	handle_packet(sdi);
	245	memset(devc->buf, 0, RELOADPRO_BUFSIZE);
	246	devc->buflen = 0;
	247	}
	248	}
	249
	250	SR_PRIV int reloadpro_receive_data(int fd, int revents, void *cb_data)
	251	{
	252	struct sr_dev_inst *sdi;
	253	struct dev_context *devc;
6e8d31d4 UH	254
	255	(void)fd;
	256
803db07a UH	257	sdi = cb_data;
	258	devc = sdi->priv;
	259
	260	if (revents != G_IO_IN)
6e8d31d4 UH	261	return TRUE;
6e8d31d4 UH	262
803db07a UH	263	handle_new_data(sdi);
803db07a UH	264
014c7f93	265	if (sr_sw_limits_check(&devc->limits))
d2f7c417	266	sr_dev_acquisition_stop(sdi);
6e8d31d4 UH	267
	268	return TRUE;
	269	}