[libsigrok.git] / src / hardware / lecroy-xstream / protocol.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2017 Sven Schnelle <svens@stackframe.org>
 *
 * 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 <config.h>
#include <math.h>
#include <stdlib.h>
#include "scpi.h"
#include "protocol.h"

struct lecroy_wavedesc_2_x {
	uint16_t comm_type;
	uint16_t comm_order; /* 1 - little endian */
	uint32_t wave_descriptor_length;
	uint32_t user_text_len;
	uint32_t res_desc1;
	uint32_t trigtime_array_length;
	uint32_t ris_time1_array_length;
	uint32_t res_array1;
	uint32_t wave_array1_length;
	uint32_t wave_array2_length;
	uint32_t wave_array3_length;
	uint32_t wave_array4_length;
	char instrument_name[16];
	uint32_t instrument_number;
	char trace_label[16];
	uint32_t reserved;
	uint32_t wave_array_count;
	uint32_t points_per_screen;
	uint32_t first_valid_point;
	uint32_t last_valid_point;
	uint32_t first_point;
	uint32_t sparsing_factor;
	uint32_t segment_index;
	uint32_t subarray_count;
	uint32_t sweeps_per_acq;
	uint16_t points_per_pair;
	uint16_t pair_offset;
	float vertical_gain;
	float vertical_offset;
	float max_value;
	float min_value;
	uint16_t nominal_bits;
	uint16_t nom_subarray_count;
	float horiz_interval;
	double horiz_offset;
	double pixel_offset;
	char vertunit[48];
	char horunit[48];
	uint32_t reserved1;
	double trigger_time;
} __attribute__((packed));

struct lecroy_wavedesc {
	char descriptor_name[16];
	char template_name[16];
	union {
		struct lecroy_wavedesc_2_x version_2_x;
	};
} __attribute__((packed));

static const char *coupling_options[] = {
	"A1M", ///< AC with 1 MOhm termination
	"D50", ///< DC with 50 Ohm termination
	"D1M", ///< DC with 1 MOhm termination
	"GND",
	"OVL",
};

static const char *scope_trigger_slopes[] = {
	"POS", "NEG",
};

static const char *trigger_sources[] = {
	"C1", "C2", "C3", "C4", "LINE", "EXT",
};

static const uint64_t timebases[][2] = {
	/* picoseconds */
	{ 20, 1000000000000 },
	{ 50, 1000000000000 },
	{ 100, 1000000000000 },
	{ 200, 1000000000000 },
	{ 500, 1000000000000 },
	/* nanoseconds */
	{ 1, 1000000000 },
	{ 2, 1000000000 },
	{ 5, 1000000000 },
	{ 10, 1000000000 },
	{ 20, 1000000000 },
	{ 50, 1000000000 },
	{ 100, 1000000000 },
	{ 200, 1000000000 },
	{ 500, 1000000000 },
	/* microseconds */
	{ 1, 1000000 },
	{ 2, 1000000 },
	{ 5, 1000000 },
	{ 10, 1000000 },
	{ 20, 1000000 },
	{ 50, 1000000 },
	{ 100, 1000000 },
	{ 200, 1000000 },
	{ 500, 1000000 },
	/* milliseconds */
	{ 1, 1000 },
	{ 2, 1000 },
	{ 5, 1000 },
	{ 10, 1000 },
	{ 20, 1000 },
	{ 50, 1000 },
	{ 100, 1000 },
	{ 200, 1000 },
	{ 500, 1000 },
	/* seconds */
	{ 1, 1 },
	{ 2, 1 },
	{ 5, 1 },
	{ 10, 1 },
	{ 20, 1 },
	{ 50, 1 },
	{ 100, 1 },
	{ 200, 1 },
	{ 500, 1 },
	{ 1000, 1 },
};

static const uint64_t vdivs[][2] = {
	/* millivolts */
	{ 1, 1000 },
	{ 2, 1000 },
	{ 5, 1000 },
	{ 10, 1000 },
	{ 20, 1000 },
	{ 50, 1000 },
	{ 100, 1000 },
	{ 200, 1000 },
	{ 500, 1000 },
	/* volts */
	{ 1, 1 },
	{ 2, 1 },
	{ 5, 1 },
	{ 10, 1 },
	{ 20, 1 },
	{ 50, 1 },
};

static const char *scope_analog_channel_names[] = {
	"CH1", "CH2", "CH3", "CH4",
};

static const struct scope_config scope_models[] = {
	{
		 /* Default config */
		.name = {NULL},

		.analog_channels = 4,
		.analog_names = &scope_analog_channel_names,

		.coupling_options = &coupling_options,
		.num_coupling_options = ARRAY_SIZE(coupling_options),

		.trigger_sources = &trigger_sources,
		.num_trigger_sources = ARRAY_SIZE(trigger_sources),

		.trigger_slopes = &scope_trigger_slopes,
		.num_trigger_slopes = ARRAY_SIZE(scope_trigger_slopes),

		.timebases = &timebases,
		.num_timebases = ARRAY_SIZE(timebases),

		.vdivs = &vdivs,
		.num_vdivs = ARRAY_SIZE(vdivs),

		.num_xdivs = 10,
		.num_ydivs = 8,
	},
};

static void scope_state_dump(const struct scope_config *config,
		struct scope_state *state)
{
	unsigned int i;
	char *tmp;

	for (i = 0; i < config->analog_channels; i++) {
		tmp = sr_voltage_string((*config->vdivs)[state->analog_channels[i].vdiv][0],
				(*config->vdivs)[state->analog_channels[i].vdiv][1]);
		sr_info("State of analog channel %d -> %s : %s (coupling) %s (vdiv) %2.2e (offset)",
				i + 1, state->analog_channels[i].state ? "On" : "Off",
				(*config->coupling_options)[state->analog_channels[i].coupling],
				tmp, state->analog_channels[i].vertical_offset);
	}

	tmp = sr_period_string((*config->timebases)[state->timebase][0],
			(*config->timebases)[state->timebase][1]);
	sr_info("Current timebase: %s", tmp);
	g_free(tmp);

	tmp = sr_samplerate_string(state->sample_rate);
	sr_info("Current samplerate: %s", tmp);
	g_free(tmp);

	sr_info("Current trigger: %s (source), %s (slope) %.2f (offset)",
			(*config->trigger_sources)[state->trigger_source],
			(*config->trigger_slopes)[state->trigger_slope],
			state->horiz_triggerpos);
}

static int scope_state_get_array_option(const char *resp,
		const char *(*array)[], unsigned int n, int *result)
{
	unsigned int i;

	for (i = 0; i < n; i++) {
		if (!g_strcmp0(resp, (*array)[i])) {
			*result = i;
			return SR_OK;
		}
	}

	return SR_ERR;
}

/**
 * This function takes a value of the form "2.000E-03" and returns the index
 * of an array where a matching pair was found.
 *
 * @param value The string to be parsed.
 * @param array The array of s/f pairs.
 * @param array_len The number of pairs in the array.
 * @param result The index at which a matching pair was found.
 *
 * @return SR_ERR on any parsing error, SR_OK otherwise.
 */
static int array_float_get(gchar *value, const uint64_t array[][2],
		int array_len, unsigned int *result)
{
	struct sr_rational rval;
	struct sr_rational aval;

	if (sr_parse_rational(value, &rval) != SR_OK)
		return SR_ERR;

	for (int i = 0; i < array_len; i++) {
		sr_rational_set(&aval, array[i][0], array[i][1]);
		if (sr_rational_eq(&rval, &aval)) {
			*result = i;
			return SR_OK;
		}
	}

	return SR_ERR;
}

static int analog_channel_state_get(struct sr_scpi_dev_inst *scpi,
		const struct scope_config *config, struct scope_state *state)
{
	unsigned int i, j;
	char command[MAX_COMMAND_SIZE];
	char *tmp_str;

	for (i = 0; i < config->analog_channels; i++) {
		g_snprintf(command, sizeof(command), "C%d:TRACE?", i + 1);

		if (sr_scpi_get_bool(scpi, command,
				&state->analog_channels[i].state) != SR_OK)
			return SR_ERR;

		g_snprintf(command, sizeof(command), "C%d:VDIV?", i + 1);

		if (sr_scpi_get_string(scpi, command, &tmp_str) != SR_OK)
			return SR_ERR;

		if (array_float_get(tmp_str, ARRAY_AND_SIZE(vdivs), &j) != SR_OK) {
			g_free(tmp_str);
			sr_err("Could not determine array index for vertical div scale.");
			return SR_ERR;
		}

		g_free(tmp_str);
		state->analog_channels[i].vdiv = j;

		g_snprintf(command, sizeof(command), "C%d:OFFSET?", i + 1);

		if (sr_scpi_get_float(scpi, command, &state->analog_channels[i].vertical_offset) != SR_OK)
			return SR_ERR;

		g_snprintf(command, sizeof(command), "C%d:COUPLING?", i + 1);

		if (sr_scpi_get_string(scpi, command, &tmp_str) != SR_OK)
			return SR_ERR;


		if (scope_state_get_array_option(tmp_str, config->coupling_options,
				 config->num_coupling_options,
				 &state->analog_channels[i].coupling) != SR_OK)
			return SR_ERR;

		g_free(tmp_str);
	}

	return SR_OK;
}

SR_PRIV int lecroy_xstream_update_sample_rate(const struct sr_dev_inst *sdi)
{
	struct dev_context *devc;
	struct scope_state *state;
	const struct scope_config *config;
	float memsize, timediv;

	devc = sdi->priv;
	state = devc->model_state;
	config = devc->model_config;

	if (sr_scpi_get_float(sdi->conn, "MEMORY_SIZE?", &memsize) != SR_OK)
		return SR_ERR;

	if (sr_scpi_get_float(sdi->conn, "TIME_DIV?", &timediv) != SR_OK)
		return SR_ERR;

	state->sample_rate = 1 / ((timediv * config->num_xdivs) / memsize);

	return SR_OK;
}

SR_PRIV int lecroy_xstream_state_get(struct sr_dev_inst *sdi)
{
	struct dev_context *devc;
	struct scope_state *state;
	const struct scope_config *config;
	unsigned int i;
	char *tmp_str, *tmp_str2, *tmpp, *p, *key;
	char command[MAX_COMMAND_SIZE];
	char *trig_source = NULL;

	devc = sdi->priv;
	config = devc->model_config;
	state = devc->model_state;

	sr_info("Fetching scope state");

	if (analog_channel_state_get(sdi->conn, config, state) != SR_OK)
		return SR_ERR;

	if (sr_scpi_get_string(sdi->conn, "TIME_DIV?", &tmp_str) != SR_OK)
		return SR_ERR;

	if (array_float_get(tmp_str, ARRAY_AND_SIZE(timebases), &i) != SR_OK) {
		g_free(tmp_str);
		sr_err("Could not determine array index for timbase scale.");
		return SR_ERR;
	}
	g_free(tmp_str);
	state->timebase = i;

	if (sr_scpi_get_string(sdi->conn, "TRIG_SELECT?", &tmp_str) != SR_OK)
		return SR_ERR;

	key = tmpp = NULL;
	tmp_str2 = tmp_str;
	i = 0;
	while ((p = strtok_r(tmp_str2, ",", &tmpp))) {
		tmp_str2 = NULL;
		if (i == 0) {
			/* trigger type */
		} else if (i & 1) {
			key = p;
			/* key */
		} else if (!(i & 1)) {
			if (!strcmp(key, "SR"))
				trig_source = p;
		}
		i++;
	}

	if (!trig_source || scope_state_get_array_option(trig_source,
			config->trigger_sources, config->num_trigger_sources,
			&state->trigger_source) != SR_OK)
		return SR_ERR;

	g_snprintf(command, sizeof(command), "%s:TRIG_SLOPE?", trig_source);
	if (sr_scpi_get_string(sdi->conn, command, &tmp_str) != SR_OK)
		return SR_ERR;

	if (scope_state_get_array_option(tmp_str, config->trigger_slopes,
			config->num_trigger_slopes, &state->trigger_slope) != SR_OK)
		return SR_ERR;

	if (sr_scpi_get_float(sdi->conn, "TRIG_DELAY?", &state->horiz_triggerpos) != SR_OK)
		return SR_ERR;

	if (lecroy_xstream_update_sample_rate(sdi) != SR_OK)
		return SR_ERR;

	sr_info("Fetching finished.");

	scope_state_dump(config, state);

	return SR_OK;
}

static struct scope_state *scope_state_new(const struct scope_config *config)
{
	struct scope_state *state;

	state = g_malloc0(sizeof(struct scope_state));
	state->analog_channels = g_malloc0_n(config->analog_channels,
			sizeof(struct analog_channel_state));
	return state;
}

SR_PRIV void lecroy_xstream_state_free(struct scope_state *state)
{
	g_free(state->analog_channels);
	g_free(state);
}

SR_PRIV int lecroy_xstream_init_device(struct sr_dev_inst *sdi)
{
	char command[MAX_COMMAND_SIZE];
	int model_index;
	unsigned int i, j;
	struct sr_channel *ch;
	struct dev_context *devc;
	gboolean channel_enabled;

	devc = sdi->priv;
	model_index = -1;

	/* Find the exact model. */
	for (i = 0; i < ARRAY_SIZE(scope_models); i++) {
		for (j = 0; scope_models[i].name[j]; j++) {
			if (!strcmp(sdi->model, scope_models[i].name[j])) {
				model_index = i;
				break;
			}
		}
		if (model_index != -1)
			break;
	}

	if (model_index == -1) {
		sr_dbg("Unknown LeCroy device, using default config.");
		for (i = 0; i < ARRAY_SIZE(scope_models); i++)
			if (scope_models[i].name[0] == NULL)
				model_index = i;
	}

	/* Set the desired response and format modes. */
	sr_scpi_send(sdi->conn, "COMM_HEADER OFF");
	sr_scpi_send(sdi->conn, "COMM_FORMAT DEF9,WORD,BIN");

	devc->analog_groups = g_malloc0(sizeof(struct sr_channel_group*) *
				scope_models[model_index].analog_channels);

	/* Add analog channels. */
	for (i = 0; i < scope_models[model_index].analog_channels; i++) {
		g_snprintf(command, sizeof(command), "C%d:TRACE?", i + 1);

		if (sr_scpi_get_bool(sdi->conn, command, &channel_enabled) != SR_OK)
			return SR_ERR;

		g_snprintf(command, sizeof(command), "C%d:VDIV?", i + 1);

		ch = sr_channel_new(sdi, i, SR_CHANNEL_ANALOG, channel_enabled,
			(*scope_models[model_index].analog_names)[i]);

		devc->analog_groups[i] = g_malloc0(sizeof(struct sr_channel_group));

		devc->analog_groups[i]->name = g_strdup(
			(char *)(*scope_models[model_index].analog_names)[i]);
		devc->analog_groups[i]->channels = g_slist_append(NULL, ch);

		sdi->channel_groups = g_slist_append(sdi->channel_groups,
			devc->analog_groups[i]);
	}

	devc->model_config = &scope_models[model_index];
	devc->frame_limit = 0;
	devc->model_state = scope_state_new(devc->model_config);

	return SR_OK;
}

static int lecroy_waveform_2_x_to_analog(GByteArray *data,
		struct lecroy_wavedesc *desc, struct sr_datafeed_analog *analog)
{
	struct sr_analog_encoding *encoding = analog->encoding;
	struct sr_analog_meaning *meaning = analog->meaning;
	struct sr_analog_spec *spec = analog->spec;
	float *data_float;
	int16_t *waveform_data;
	unsigned int i, num_samples;

	data_float = g_malloc(desc->version_2_x.wave_array_count * sizeof(float));
	num_samples = desc->version_2_x.wave_array_count;

	waveform_data = (int16_t*)(data->data +
		+ desc->version_2_x.wave_descriptor_length
		+ desc->version_2_x.user_text_len);

	for (i = 0; i < num_samples; i++)
		data_float[i] = (float)waveform_data[i]
			* desc->version_2_x.vertical_gain
			+ desc->version_2_x.vertical_offset;

	analog->data = data_float;
	analog->num_samples = num_samples;

	encoding->unitsize = sizeof(float);
	encoding->is_signed = TRUE;
	encoding->is_float = TRUE;
	encoding->is_bigendian = FALSE;
	encoding->scale.p = 1;
	encoding->scale.q = 1;
	encoding->offset.p = 0;
	encoding->offset.q = 1;

	encoding->digits = 6;
	encoding->is_digits_decimal = FALSE;

	if (strcmp(desc->version_2_x.vertunit, "A")) {
		meaning->mq = SR_MQ_CURRENT;
		meaning->unit = SR_UNIT_AMPERE;
	} else {
		/* Default to voltage. */
		meaning->mq = SR_MQ_VOLTAGE;
		meaning->unit = SR_UNIT_VOLT;
	}

	meaning->mqflags = 0;
	spec->spec_digits = 3;

	return SR_OK;
}

static int lecroy_waveform_to_analog(GByteArray *data,
		struct sr_datafeed_analog *analog)
{
	struct lecroy_wavedesc *desc;

	if (data->len < sizeof(struct lecroy_wavedesc))
		return SR_ERR;

	desc = (struct lecroy_wavedesc*)data->data;

	if (!strncmp(desc->template_name, "LECROY_2_2", 16) ||
	    !strncmp(desc->template_name, "LECROY_2_3", 16)) {
		return lecroy_waveform_2_x_to_analog(data, desc, analog);
	}

	sr_err("Waveformat template '%.16s' not supported.", desc->template_name);
	return SR_ERR;
}

SR_PRIV int lecroy_xstream_receive_data(int fd, int revents, void *cb_data)
{
	struct sr_channel *ch;
	struct sr_dev_inst *sdi;
	struct dev_context *devc;
	struct sr_datafeed_packet packet;
	GByteArray *data;
	struct sr_datafeed_analog analog;
	struct sr_analog_encoding encoding;
	struct sr_analog_meaning meaning;
	struct sr_analog_spec spec;
	char buf[8];

	(void)fd;
	(void)revents;

	data = NULL;

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

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

	ch = devc->current_channel->data;

	/*
	 * Send "frame begin" packet upon reception of data for the
	 * first enabled channel.
	 */
	if (devc->current_channel == devc->enabled_channels) {
		packet.type = SR_DF_FRAME_BEGIN;
		sr_session_send(sdi, &packet);
	}

	if (ch->type != SR_CHANNEL_ANALOG)
		return SR_ERR;

	/* Pass on the received data of the channel(s). */
	if (sr_scpi_read_data(sdi->conn, buf, 4) != 4) {
		sr_err("Reading header failed.");
		return TRUE;
	}

	if (sr_scpi_get_block(sdi->conn, NULL, &data) != SR_OK) {
		if (data)
			g_byte_array_free(data, TRUE);
		return TRUE;
	}

	analog.encoding = &encoding;
	analog.meaning = &meaning;
	analog.spec = &spec;

	if (lecroy_waveform_to_analog(data, &analog) != SR_OK)
		return SR_ERR;

	meaning.channels = g_slist_append(NULL, ch);
	packet.payload = &analog;
	packet.type = SR_DF_ANALOG;
	sr_session_send(sdi, &packet);

	g_byte_array_free(data, TRUE);
	data = NULL;

	g_slist_free(meaning.channels);
	g_free(analog.data);

	/*
	 * Advance to the next enabled channel. When data for all enabled
	 * channels was received, then flush potentially queued logic data,
	 * and send the "frame end" packet.
	 */
	if (devc->current_channel->next) {
		devc->current_channel = devc->current_channel->next;
		lecroy_xstream_request_data(sdi);
		return TRUE;
	}

	packet.type = SR_DF_FRAME_END;
	sr_session_send(sdi, &packet);

	/*
	 * End of frame was reached. Stop acquisition after the specified
	 * number of frames, or continue reception by starting over at
	 * the first enabled channel.
	 */
	if (++devc->num_frames == devc->frame_limit) {
		sr_dev_acquisition_stop(sdi);
	} else {
		devc->current_channel = devc->enabled_channels;
		lecroy_xstream_request_data(sdi);
	}

	return TRUE;
}
Commit	Line	Data
e3b83c5e SS	1	/*
	2	* This file is part of the libsigrok project.
	3	*
	4	* Copyright (C) 2017 Sven Schnelle <svens@stackframe.org>
	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 <config.h>
3f2c7c94 SS	21	#include <math.h>
	22	#include <stdlib.h>
	23	#include "scpi.h"
e3b83c5e SS	24	#include "protocol.h"
e3b83c5e SS	25
3f2c7c94 SS	26	struct lecroy_wavedesc_2_x {
	27	uint16_t comm_type;
	28	uint16_t comm_order; /* 1 - little endian */
	29	uint32_t wave_descriptor_length;
	30	uint32_t user_text_len;
	31	uint32_t res_desc1;
	32	uint32_t trigtime_array_length;
	33	uint32_t ris_time1_array_length;
	34	uint32_t res_array1;
	35	uint32_t wave_array1_length;
	36	uint32_t wave_array2_length;
	37	uint32_t wave_array3_length;
	38	uint32_t wave_array4_length;
	39	char instrument_name[16];
	40	uint32_t instrument_number;
	41	char trace_label[16];
	42	uint32_t reserved;
	43	uint32_t wave_array_count;
	44	uint32_t points_per_screen;
	45	uint32_t first_valid_point;
	46	uint32_t last_valid_point;
	47	uint32_t first_point;
	48	uint32_t sparsing_factor;
	49	uint32_t segment_index;
	50	uint32_t subarray_count;
	51	uint32_t sweeps_per_acq;
	52	uint16_t points_per_pair;
	53	uint16_t pair_offset;
	54	float vertical_gain;
	55	float vertical_offset;
	56	float max_value;
	57	float min_value;
	58	uint16_t nominal_bits;
	59	uint16_t nom_subarray_count;
	60	float horiz_interval;
	61	double horiz_offset;
	62	double pixel_offset;
	63	char vertunit[48];
	64	char horunit[48];
	65	uint32_t reserved1;
	66	double trigger_time;
	67	} __attribute__((packed));
	68
	69	struct lecroy_wavedesc {
	70	char descriptor_name[16];
	71	char template_name[16];
	72	union {
	73	struct lecroy_wavedesc_2_x version_2_x;
	74	};
	75	} __attribute__((packed));
	76
6d13a46c	77	static const char *coupling_options[] = {
952c7376 SA	78	"A1M", ///< AC with 1 MOhm termination
	79	"D50", ///< DC with 50 Ohm termination
	80	"D1M", ///< DC with 1 MOhm termination
3f2c7c94 SS	81	"GND",
3f2c7c94 SS	82	"OVL",
3f2c7c94 SS	83	};
	84
	85	static const char *scope_trigger_slopes[] = {
692716f5	86	"POS", "NEG",
3f2c7c94 SS	87	};
3f2c7c94 SS	88
6d13a46c	89	static const char *trigger_sources[] = {
692716f5	90	"C1", "C2", "C3", "C4", "LINE", "EXT",
3f2c7c94 SS	91	};
3f2c7c94 SS	92
76f0fa5d	93	static const uint64_t timebases[][2] = {
3f2c7c94 SS	94	/* picoseconds */
	95	{ 20, 1000000000000 },
	96	{ 50, 1000000000000 },
	97	{ 100, 1000000000000 },
	98	{ 200, 1000000000000 },
	99	{ 500, 1000000000000 },
	100	/* nanoseconds */
	101	{ 1, 1000000000 },
	102	{ 2, 1000000000 },
	103	{ 5, 1000000000 },
	104	{ 10, 1000000000 },
	105	{ 20, 1000000000 },
	106	{ 50, 1000000000 },
	107	{ 100, 1000000000 },
	108	{ 200, 1000000000 },
	109	{ 500, 1000000000 },
	110	/* microseconds */
	111	{ 1, 1000000 },
	112	{ 2, 1000000 },
	113	{ 5, 1000000 },
	114	{ 10, 1000000 },
	115	{ 20, 1000000 },
	116	{ 50, 1000000 },
	117	{ 100, 1000000 },
	118	{ 200, 1000000 },
	119	{ 500, 1000000 },
	120	/* milliseconds */
	121	{ 1, 1000 },
	122	{ 2, 1000 },
	123	{ 5, 1000 },
	124	{ 10, 1000 },
	125	{ 20, 1000 },
	126	{ 50, 1000 },
	127	{ 100, 1000 },
	128	{ 200, 1000 },
	129	{ 500, 1000 },
	130	/* seconds */
	131	{ 1, 1 },
	132	{ 2, 1 },
	133	{ 5, 1 },
	134	{ 10, 1 },
	135	{ 20, 1 },
	136	{ 50, 1 },
	137	{ 100, 1 },
	138	{ 200, 1 },
	139	{ 500, 1 },
	140	{ 1000, 1 },
	141	};
	142
76f0fa5d	143	static const uint64_t vdivs[][2] = {
3f2c7c94 SS	144	/* millivolts */
	145	{ 1, 1000 },
	146	{ 2, 1000 },
	147	{ 5, 1000 },
	148	{ 10, 1000 },
	149	{ 20, 1000 },
	150	{ 50, 1000 },
	151	{ 100, 1000 },
	152	{ 200, 1000 },
	153	{ 500, 1000 },
	154	/* volts */
	155	{ 1, 1 },
	156	{ 2, 1 },
	157	{ 5, 1 },
	158	{ 10, 1 },
	159	{ 20, 1 },
	160	{ 50, 1 },
	161	};
	162
	163	static const char *scope_analog_channel_names[] = {
f8195cb2	164	"CH1", "CH2", "CH3", "CH4",
3f2c7c94 SS	165	};
	166
	167	static const struct scope_config scope_models[] = {
	168	{
e7d2cd1e SA	169	/* Default config */
e7d2cd1e SA	170	.name = {NULL},
3f2c7c94 SS	171
	172	.analog_channels = 4,
	173	.analog_names = &scope_analog_channel_names,
	174
6d13a46c	175	.coupling_options = &coupling_options,
692716f5 UH	176	.num_coupling_options = ARRAY_SIZE(coupling_options),
692716f5 UH	177
6d13a46c	178	.trigger_sources = &trigger_sources,
692716f5 UH	179	.num_trigger_sources = ARRAY_SIZE(trigger_sources),
692716f5 UH	180
3f2c7c94	181	.trigger_slopes = &scope_trigger_slopes,
692716f5	182	.num_trigger_slopes = ARRAY_SIZE(scope_trigger_slopes),
3f2c7c94	183
76f0fa5d	184	.timebases = &timebases,
6d13a46c	185	.num_timebases = ARRAY_SIZE(timebases),
3f2c7c94	186
76f0fa5d	187	.vdivs = &vdivs,
6d13a46c	188	.num_vdivs = ARRAY_SIZE(vdivs),
3f2c7c94 SS	189
	190	.num_xdivs = 10,
	191	.num_ydivs = 8,
	192	},
	193	};
	194
	195	static void scope_state_dump(const struct scope_config *config,
952c7376	196	struct scope_state *state)
3f2c7c94 SS	197	{
	198	unsigned int i;
	199	char *tmp;
	200
	201	for (i = 0; i < config->analog_channels; i++) {
76f0fa5d	202	tmp = sr_voltage_string((*config->vdivs)[state->analog_channels[i].vdiv][0],
952c7376	203	(*config->vdivs)[state->analog_channels[i].vdiv][1]);
3f2c7c94	204	sr_info("State of analog channel %d -> %s : %s (coupling) %s (vdiv) %2.2e (offset)",
952c7376 SA	205	i + 1, state->analog_channels[i].state ? "On" : "Off",
	206	(*config->coupling_options)[state->analog_channels[i].coupling],
	207	tmp, state->analog_channels[i].vertical_offset);
3f2c7c94 SS	208	}
3f2c7c94 SS	209
76f0fa5d	210	tmp = sr_period_string((*config->timebases)[state->timebase][0],
952c7376	211	(*config->timebases)[state->timebase][1]);
3f2c7c94 SS	212	sr_info("Current timebase: %s", tmp);
	213	g_free(tmp);
	214
	215	tmp = sr_samplerate_string(state->sample_rate);
	216	sr_info("Current samplerate: %s", tmp);
	217	g_free(tmp);
	218
	219	sr_info("Current trigger: %s (source), %s (slope) %.2f (offset)",
952c7376 SA	220	(*config->trigger_sources)[state->trigger_source],
	221	(*config->trigger_slopes)[state->trigger_slope],
	222	state->horiz_triggerpos);
3f2c7c94 SS	223	}
	224
	225	static int scope_state_get_array_option(const char *resp,
952c7376	226	const char (array)[], unsigned int n, int *result)
3f2c7c94 SS	227	{
	228	unsigned int i;
	229
692716f5	230	for (i = 0; i < n; i++) {
3f2c7c94 SS	231	if (!g_strcmp0(resp, (*array)[i])) {
	232	*result = i;
	233	return SR_OK;
	234	}
	235	}
	236
	237	return SR_ERR;
	238	}
	239
	240	/**
	241	* This function takes a value of the form "2.000E-03" and returns the index
	242	* of an array where a matching pair was found.
	243	*
	244	* @param value The string to be parsed.
	245	* @param array The array of s/f pairs.
	246	* @param array_len The number of pairs in the array.
	247	* @param result The index at which a matching pair was found.
	248	*
	249	* @return SR_ERR on any parsing error, SR_OK otherwise.
	250	*/
76f0fa5d	251	static int array_float_get(gchar *value, const uint64_t array[][2],
3f2c7c94 SS	252	int array_len, unsigned int *result)
	253	{
	254	struct sr_rational rval;
76f0fa5d	255	struct sr_rational aval;
3f2c7c94 SS	256
	257	if (sr_parse_rational(value, &rval) != SR_OK)
	258	return SR_ERR;
	259
	260	for (int i = 0; i < array_len; i++) {
76f0fa5d UH	261	sr_rational_set(&aval, array[i][0], array[i][1]);
76f0fa5d UH	262	if (sr_rational_eq(&rval, &aval)) {
3f2c7c94 SS	263	*result = i;
	264	return SR_OK;
	265	}
	266	}
	267
	268	return SR_ERR;
	269	}
	270
	271	static int analog_channel_state_get(struct sr_scpi_dev_inst *scpi,
952c7376	272	const struct scope_config config, struct scope_state state)
3f2c7c94 SS	273	{
	274	unsigned int i, j;
	275	char command[MAX_COMMAND_SIZE];
	276	char *tmp_str;
	277
	278	for (i = 0; i < config->analog_channels; i++) {
ea257cdc	279	g_snprintf(command, sizeof(command), "C%d:TRACE?", i + 1);
3f2c7c94 SS	280
3f2c7c94 SS	281	if (sr_scpi_get_bool(scpi, command,
ea257cdc	282	&state->analog_channels[i].state) != SR_OK)
3f2c7c94 SS	283	return SR_ERR;
3f2c7c94 SS	284
ea257cdc	285	g_snprintf(command, sizeof(command), "C%d:VDIV?", i + 1);
3f2c7c94 SS	286
	287	if (sr_scpi_get_string(scpi, command, &tmp_str) != SR_OK)
	288	return SR_ERR;
	289
952c7376	290	if (array_float_get(tmp_str, ARRAY_AND_SIZE(vdivs), &j) != SR_OK) {
3f2c7c94 SS	291	g_free(tmp_str);
	292	sr_err("Could not determine array index for vertical div scale.");
	293	return SR_ERR;
	294	}
	295
	296	g_free(tmp_str);
	297	state->analog_channels[i].vdiv = j;
	298
ea257cdc	299	g_snprintf(command, sizeof(command), "C%d:OFFSET?", i + 1);
3f2c7c94 SS	300
	301	if (sr_scpi_get_float(scpi, command, &state->analog_channels[i].vertical_offset) != SR_OK)
	302	return SR_ERR;
	303
ea257cdc	304	g_snprintf(command, sizeof(command), "C%d:COUPLING?", i + 1);
3f2c7c94 SS	305
	306	if (sr_scpi_get_string(scpi, command, &tmp_str) != SR_OK)
	307	return SR_ERR;
	308
	309
	310	if (scope_state_get_array_option(tmp_str, config->coupling_options,
692716f5	311	config->num_coupling_options,
ea257cdc	312	&state->analog_channels[i].coupling) != SR_OK)
3f2c7c94 SS	313	return SR_ERR;
	314
	315	g_free(tmp_str);
	316	}
	317
	318	return SR_OK;
	319	}
	320
	321	SR_PRIV int lecroy_xstream_update_sample_rate(const struct sr_dev_inst *sdi)
e3b83c5e	322	{
e3b83c5e	323	struct dev_context *devc;
3f2c7c94 SS	324	struct scope_state *state;
	325	const struct scope_config *config;
	326	float memsize, timediv;
	327
	328	devc = sdi->priv;
	329	state = devc->model_state;
	330	config = devc->model_config;
	331
	332	if (sr_scpi_get_float(sdi->conn, "MEMORY_SIZE?", &memsize) != SR_OK)
ea257cdc	333	return SR_ERR;
3f2c7c94 SS	334
	335	if (sr_scpi_get_float(sdi->conn, "TIME_DIV?", &timediv) != SR_OK)
	336	return SR_ERR;
	337
ea257cdc	338	state->sample_rate = 1 / ((timediv * config->num_xdivs) / memsize);
3f2c7c94 SS	339
	340	return SR_OK;
	341	}
	342
	343	SR_PRIV int lecroy_xstream_state_get(struct sr_dev_inst *sdi)
	344	{
	345	struct dev_context *devc;
ea257cdc UH	346	struct scope_state *state;
ea257cdc UH	347	const struct scope_config *config;
3f2c7c94 SS	348	unsigned int i;
	349	char tmp_str, tmp_str2, tmpp, p, *key;
	350	char command[MAX_COMMAND_SIZE];
	351	char *trig_source = NULL;
	352
	353	devc = sdi->priv;
	354	config = devc->model_config;
	355	state = devc->model_state;
	356
	357	sr_info("Fetching scope state");
	358
	359	if (analog_channel_state_get(sdi->conn, config, state) != SR_OK)
	360	return SR_ERR;
	361
	362	if (sr_scpi_get_string(sdi->conn, "TIME_DIV?", &tmp_str) != SR_OK)
	363	return SR_ERR;
	364
53012da6	365	if (array_float_get(tmp_str, ARRAY_AND_SIZE(timebases), &i) != SR_OK) {
3f2c7c94 SS	366	g_free(tmp_str);
	367	sr_err("Could not determine array index for timbase scale.");
	368	return SR_ERR;
	369	}
	370	g_free(tmp_str);
	371	state->timebase = i;
	372
	373	if (sr_scpi_get_string(sdi->conn, "TRIG_SELECT?", &tmp_str) != SR_OK)
	374	return SR_ERR;
	375
7002e64a	376	key = tmpp = NULL;
3f2c7c94 SS	377	tmp_str2 = tmp_str;
3f2c7c94 SS	378	i = 0;
ea257cdc	379	while ((p = strtok_r(tmp_str2, ",", &tmpp))) {
3f2c7c94 SS	380	tmp_str2 = NULL;
	381	if (i == 0) {
	382	/* trigger type */
	383	} else if (i & 1) {
	384	key = p;
	385	/* key */
	386	} else if (!(i & 1)) {
	387	if (!strcmp(key, "SR"))
	388	trig_source = p;
	389	}
	390	i++;
	391	}
	392
952c7376 SA	393	if (!trig_source \|\| scope_state_get_array_option(trig_source,
	394	config->trigger_sources, config->num_trigger_sources,
	395	&state->trigger_source) != SR_OK)
3f2c7c94 SS	396	return SR_ERR;
3f2c7c94 SS	397
3f2c7c94 SS	398	g_snprintf(command, sizeof(command), "%s:TRIG_SLOPE?", trig_source);
	399	if (sr_scpi_get_string(sdi->conn, command, &tmp_str) != SR_OK)
	400	return SR_ERR;
	401
952c7376 SA	402	if (scope_state_get_array_option(tmp_str, config->trigger_slopes,
952c7376 SA	403	config->num_trigger_slopes, &state->trigger_slope) != SR_OK)
3f2c7c94 SS	404	return SR_ERR;
3f2c7c94 SS	405
952c7376	406	if (sr_scpi_get_float(sdi->conn, "TRIG_DELAY?", &state->horiz_triggerpos) != SR_OK)
3f2c7c94 SS	407	return SR_ERR;
	408
	409	if (lecroy_xstream_update_sample_rate(sdi) != SR_OK)
	410	return SR_ERR;
	411
	412	sr_info("Fetching finished.");
	413
	414	scope_state_dump(config, state);
	415
	416	return SR_OK;
	417	}
	418
	419	static struct scope_state scope_state_new(const struct scope_config config)
	420	{
	421	struct scope_state *state;
	422
	423	state = g_malloc0(sizeof(struct scope_state));
	424	state->analog_channels = g_malloc0_n(config->analog_channels,
	425	sizeof(struct analog_channel_state));
	426	return state;
	427	}
	428
	429	SR_PRIV void lecroy_xstream_state_free(struct scope_state *state)
	430	{
	431	g_free(state->analog_channels);
	432	g_free(state);
	433	}
	434
	435	SR_PRIV int lecroy_xstream_init_device(struct sr_dev_inst *sdi)
	436	{
	437	char command[MAX_COMMAND_SIZE];
	438	int model_index;
	439	unsigned int i, j;
	440	struct sr_channel *ch;
	441	struct dev_context *devc;
	442	gboolean channel_enabled;
	443
	444	devc = sdi->priv;
	445	model_index = -1;
	446
	447	/* Find the exact model. */
	448	for (i = 0; i < ARRAY_SIZE(scope_models); i++) {
	449	for (j = 0; scope_models[i].name[j]; j++) {
	450	if (!strcmp(sdi->model, scope_models[i].name[j])) {
	451	model_index = i;
	452	break;
	453	}
	454	}
	455	if (model_index != -1)
	456	break;
	457	}
	458
	459	if (model_index == -1) {
e7d2cd1e SA	460	sr_dbg("Unknown LeCroy device, using default config.");
	461	for (i = 0; i < ARRAY_SIZE(scope_models); i++)
	462	if (scope_models[i].name[0] == NULL)
	463	model_index = i;
3f2c7c94 SS	464	}
3f2c7c94 SS	465
6158728c SP	466	/* Set the desired response and format modes. */
6158728c SP	467	sr_scpi_send(sdi->conn, "COMM_HEADER OFF");
952c7376	468	sr_scpi_send(sdi->conn, "COMM_FORMAT DEF9,WORD,BIN");
6158728c	469
3f2c7c94	470	devc->analog_groups = g_malloc0(sizeof(struct sr_channel_group)
ea257cdc	471	scope_models[model_index].analog_channels);
3f2c7c94 SS	472
	473	/* Add analog channels. */
	474	for (i = 0; i < scope_models[model_index].analog_channels; i++) {
ea257cdc	475	g_snprintf(command, sizeof(command), "C%d:TRACE?", i + 1);
3f2c7c94 SS	476
	477	if (sr_scpi_get_bool(sdi->conn, command, &channel_enabled) != SR_OK)
	478	return SR_ERR;
	479
ea257cdc	480	g_snprintf(command, sizeof(command), "C%d:VDIV?", i + 1);
3f2c7c94 SS	481
3f2c7c94 SS	482	ch = sr_channel_new(sdi, i, SR_CHANNEL_ANALOG, channel_enabled,
952c7376	483	(*scope_models[model_index].analog_names)[i]);
3f2c7c94 SS	484
	485	devc->analog_groups[i] = g_malloc0(sizeof(struct sr_channel_group));
	486
	487	devc->analog_groups[i]->name = g_strdup(
	488	(char )(scope_models[model_index].analog_names)[i]);
	489	devc->analog_groups[i]->channels = g_slist_append(NULL, ch);
	490
	491	sdi->channel_groups = g_slist_append(sdi->channel_groups,
952c7376	492	devc->analog_groups[i]);
3f2c7c94 SS	493	}
	494
	495	devc->model_config = &scope_models[model_index];
	496	devc->frame_limit = 0;
952c7376	497	devc->model_state = scope_state_new(devc->model_config);
3f2c7c94 SS	498
	499	return SR_OK;
	500	}
	501
	502	static int lecroy_waveform_2_x_to_analog(GByteArray *data,
952c7376	503	struct lecroy_wavedesc desc, struct sr_datafeed_analog analog)
3f2c7c94 SS	504	{
	505	struct sr_analog_encoding *encoding = analog->encoding;
	506	struct sr_analog_meaning *meaning = analog->meaning;
	507	struct sr_analog_spec *spec = analog->spec;
	508	float *data_float;
	509	int16_t *waveform_data;
	510	unsigned int i, num_samples;
	511
ea257cdc	512	data_float = g_malloc(desc->version_2_x.wave_array_count * sizeof(float));
3f2c7c94 SS	513	num_samples = desc->version_2_x.wave_array_count;
3f2c7c94 SS	514
952c7376 SA	515	waveform_data = (int16_t*)(data->data +
	516	+ desc->version_2_x.wave_descriptor_length
	517	+ desc->version_2_x.user_text_len);
3f2c7c94	518
ea257cdc	519	for (i = 0; i < num_samples; i++)
3f2c7c94 SS	520	data_float[i] = (float)waveform_data[i]
	521	* desc->version_2_x.vertical_gain
	522	+ desc->version_2_x.vertical_offset;
	523
3f2c7c94 SS	524	analog->data = data_float;
	525	analog->num_samples = num_samples;
	526
	527	encoding->unitsize = sizeof(float);
	528	encoding->is_signed = TRUE;
	529	encoding->is_float = TRUE;
	530	encoding->is_bigendian = FALSE;
	531	encoding->scale.p = 1;
	532	encoding->scale.q = 1;
	533	encoding->offset.p = 0;
	534	encoding->offset.q = 1;
	535
	536	encoding->digits = 6;
	537	encoding->is_digits_decimal = FALSE;
	538
	539	if (strcmp(desc->version_2_x.vertunit, "A")) {
	540	meaning->mq = SR_MQ_CURRENT;
	541	meaning->unit = SR_UNIT_AMPERE;
	542	} else {
ea257cdc	543	/* Default to voltage. */
3f2c7c94 SS	544	meaning->mq = SR_MQ_VOLTAGE;
	545	meaning->unit = SR_UNIT_VOLT;
	546	}
	547
	548	meaning->mqflags = 0;
	549	spec->spec_digits = 3;
ea257cdc	550
3f2c7c94 SS	551	return SR_OK;
	552	}
	553
	554	static int lecroy_waveform_to_analog(GByteArray *data,
952c7376	555	struct sr_datafeed_analog *analog)
3f2c7c94 SS	556	{
	557	struct lecroy_wavedesc *desc;
	558
	559	if (data->len < sizeof(struct lecroy_wavedesc))
	560	return SR_ERR;
	561
952c7376	562	desc = (struct lecroy_wavedesc*)data->data;
3f2c7c94 SS	563
	564	if (!strncmp(desc->template_name, "LECROY_2_2", 16) \|\|
	565	!strncmp(desc->template_name, "LECROY_2_3", 16)) {
	566	return lecroy_waveform_2_x_to_analog(data, desc, analog);
	567	}
	568
952c7376	569	sr_err("Waveformat template '%.16s' not supported.", desc->template_name);
3f2c7c94 SS	570	return SR_ERR;
	571	}
	572
	573	SR_PRIV int lecroy_xstream_receive_data(int fd, int revents, void *cb_data)
	574	{
	575	struct sr_channel *ch;
	576	struct sr_dev_inst *sdi;
	577	struct dev_context *devc;
	578	struct sr_datafeed_packet packet;
	579	GByteArray *data;
	580	struct sr_datafeed_analog analog;
	581	struct sr_analog_encoding encoding;
	582	struct sr_analog_meaning meaning;
	583	struct sr_analog_spec spec;
	584	char buf[8];
ea257cdc	585
e3b83c5e	586	(void)fd;
3f2c7c94 SS	587	(void)revents;
	588
	589	data = NULL;
e3b83c5e SS	590
	591	if (!(sdi = cb_data))
	592	return TRUE;
	593
	594	if (!(devc = sdi->priv))
	595	return TRUE;
	596
3f2c7c94 SS	597	ch = devc->current_channel->data;
	598
	599	/*
	600	* Send "frame begin" packet upon reception of data for the
	601	* first enabled channel.
	602	*/
	603	if (devc->current_channel == devc->enabled_channels) {
	604	packet.type = SR_DF_FRAME_BEGIN;
	605	sr_session_send(sdi, &packet);
	606	}
	607
	608	if (ch->type != SR_CHANNEL_ANALOG)
	609	return SR_ERR;
ea257cdc UH	610
ea257cdc UH	611	/* Pass on the received data of the channel(s). */
3f2c7c94	612	if (sr_scpi_read_data(sdi->conn, buf, 4) != 4) {
ea257cdc	613	sr_err("Reading header failed.");
3f2c7c94 SS	614	return TRUE;
	615	}
	616
	617	if (sr_scpi_get_block(sdi->conn, NULL, &data) != SR_OK) {
	618	if (data)
	619	g_byte_array_free(data, TRUE);
	620	return TRUE;
	621	}
	622
	623	analog.encoding = &encoding;
	624	analog.meaning = &meaning;
	625	analog.spec = &spec;
	626
	627	if (lecroy_waveform_to_analog(data, &analog) != SR_OK)
	628	return SR_ERR;
	629
	630	meaning.channels = g_slist_append(NULL, ch);
	631	packet.payload = &analog;
	632	packet.type = SR_DF_ANALOG;
	633	sr_session_send(sdi, &packet);
	634
	635	g_byte_array_free(data, TRUE);
	636	data = NULL;
	637
	638	g_slist_free(meaning.channels);
	639	g_free(analog.data);
	640
3f2c7c94 SS	641	/*
	642	* Advance to the next enabled channel. When data for all enabled
	643	* channels was received, then flush potentially queued logic data,
	644	* and send the "frame end" packet.
	645	*/
	646	if (devc->current_channel->next) {
	647	devc->current_channel = devc->current_channel->next;
	648	lecroy_xstream_request_data(sdi);
	649	return TRUE;
	650	}
	651
	652	packet.type = SR_DF_FRAME_END;
	653	sr_session_send(sdi, &packet);
	654
	655	/*
	656	* End of frame was reached. Stop acquisition after the specified
	657	* number of frames, or continue reception by starting over at
	658	* the first enabled channel.
	659	*/
	660	if (++devc->num_frames == devc->frame_limit) {
d2f7c417	661	sr_dev_acquisition_stop(sdi);
3f2c7c94 SS	662	} else {
	663	devc->current_channel = devc->enabled_channels;
	664	lecroy_xstream_request_data(sdi);
e3b83c5e SS	665	}
	666
	667	return TRUE;
	668	}