]> sigrok.org Git - libsigrok.git/blame - src/hardware/hameg-hmo/api.c
drivers: Add and use STD_CONFIG_LIST().
[libsigrok.git] / src / hardware / hameg-hmo / api.c
CommitLineData
06a3e78a
DJ
1/*
2 * This file is part of the libsigrok project.
3 *
4 * Copyright (C) 2013 poljar (Damir Jelić) <poljarinho@gmail.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
6ec6c43b 20#include <config.h>
13f2b9d7 21#include <stdlib.h>
5a1afc09 22#include "scpi.h"
06a3e78a
DJ
23#include "protocol.h"
24
13f2b9d7
DJ
25#define SERIALCOMM "115200/8n1/flow=1"
26
dd5c48a6 27static struct sr_dev_driver hameg_hmo_driver_info;
e9a62139
DJ
28
29static const char *manufacturers[] = {
30 "HAMEG",
da1726cc 31 "Rohde&Schwarz",
e9a62139
DJ
32};
33
6ec3ef9b
AJ
34static const uint32_t drvopts[] = {
35 SR_CONF_OSCILLOSCOPE,
36};
37
584560f1 38static const uint32_t scanopts[] = {
13f2b9d7
DJ
39 SR_CONF_CONN,
40 SR_CONF_SERIALCOMM,
41};
42
719eff68 43enum {
61e77667
UH
44 CG_INVALID = -1,
45 CG_NONE,
46 CG_ANALOG,
47 CG_DIGITAL,
719eff68
UH
48};
49
e9a62139
DJ
50static int check_manufacturer(const char *manufacturer)
51{
52 unsigned int i;
53
0a1f7b09 54 for (i = 0; i < ARRAY_SIZE(manufacturers); i++)
e9a62139
DJ
55 if (!strcmp(manufacturer, manufacturers[i]))
56 return SR_OK;
57
58 return SR_ERR;
59}
60
ca28abd6 61static struct sr_dev_inst *hmo_probe_serial_device(struct sr_scpi_dev_inst *scpi)
e9a62139
DJ
62{
63 struct sr_dev_inst *sdi;
64 struct dev_context *devc;
65 struct sr_scpi_hw_info *hw_info;
e9a62139
DJ
66
67 sdi = NULL;
68 devc = NULL;
e9a62139
DJ
69 hw_info = NULL;
70
e9a62139
DJ
71 if (sr_scpi_get_hw_id(scpi, &hw_info) != SR_OK) {
72 sr_info("Couldn't get IDN response.");
73 goto fail;
74 }
75
76 if (check_manufacturer(hw_info->manufacturer) != SR_OK)
77 goto fail;
78
aac29cc1 79 sdi = g_malloc0(sizeof(struct sr_dev_inst));
0af636be
UH
80 sdi->vendor = g_strdup(hw_info->manufacturer);
81 sdi->model = g_strdup(hw_info->model);
82 sdi->version = g_strdup(hw_info->firmware_version);
83 sdi->serial_num = g_strdup(hw_info->serial_number);
4f840ce9 84 sdi->driver = &hameg_hmo_driver_info;
b33db61c
SA
85 sdi->inst_type = SR_INST_SCPI;
86 sdi->conn = scpi;
b33db61c 87
e9a62139
DJ
88 sr_scpi_hw_info_free(hw_info);
89 hw_info = NULL;
90
f57d8ffe 91 devc = g_malloc0(sizeof(struct dev_context));
e9a62139 92
e9a62139 93 sdi->priv = devc;
e9a62139
DJ
94
95 if (hmo_init_device(sdi) != SR_OK)
96 goto fail;
97
98 return sdi;
99
100fail:
7b365c47 101 sr_scpi_hw_info_free(hw_info);
4bf93988 102 sr_dev_inst_free(sdi);
b1f83103 103 g_free(devc);
e9a62139
DJ
104
105 return NULL;
106}
107
4f840ce9 108static GSList *scan(struct sr_dev_driver *di, GSList *options)
06a3e78a 109{
41812aca 110 return sr_scpi_scan(di->context, options, hmo_probe_serial_device);
06a3e78a
DJ
111}
112
3553451f 113static void clear_helper(struct dev_context *devc)
13f2b9d7 114{
719eff68 115 hmo_scope_state_free(devc->model_state);
13f2b9d7
DJ
116 g_free(devc->analog_groups);
117 g_free(devc->digital_groups);
13f2b9d7
DJ
118}
119
4f840ce9 120static int dev_clear(const struct sr_dev_driver *di)
06a3e78a 121{
3553451f 122 return std_dev_clear_with_callback(di, (std_dev_clear_callback)clear_helper);
06a3e78a
DJ
123}
124
125static int dev_open(struct sr_dev_inst *sdi)
126{
6402c379 127 if (sr_scpi_open(sdi->conn) != SR_OK)
13f2b9d7 128 return SR_ERR;
06a3e78a 129
719eff68 130 if (hmo_scope_state_get(sdi) != SR_OK)
13f2b9d7 131 return SR_ERR;
06a3e78a 132
06a3e78a
DJ
133 return SR_OK;
134}
135
136static int dev_close(struct sr_dev_inst *sdi)
137{
f1ba6b4b 138 return sr_scpi_close(sdi->conn);
06a3e78a
DJ
139}
140
660e398f 141static int check_channel_group(struct dev_context *devc,
53b4680f 142 const struct sr_channel_group *cg)
13f2b9d7
DJ
143{
144 unsigned int i;
329733d9 145 const struct scope_config *model;
13f2b9d7
DJ
146
147 model = devc->model_config;
148
53b4680f 149 if (!cg)
61e77667 150 return CG_NONE;
13f2b9d7 151
0a1f7b09 152 for (i = 0; i < model->analog_channels; i++)
562b7ae5 153 if (cg == devc->analog_groups[i])
61e77667 154 return CG_ANALOG;
13f2b9d7 155
0a1f7b09 156 for (i = 0; i < model->digital_pods; i++)
562b7ae5 157 if (cg == devc->digital_groups[i])
61e77667 158 return CG_DIGITAL;
13f2b9d7 159
660e398f 160 sr_err("Invalid channel group specified.");
89280b1a 161
61e77667 162 return CG_INVALID;
13f2b9d7
DJ
163}
164
584560f1 165static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
53b4680f 166 const struct sr_channel_group *cg)
06a3e78a 167{
61e77667 168 int ret, cg_type;
13f2b9d7 169 unsigned int i;
13f2b9d7 170 struct dev_context *devc;
329733d9 171 const struct scope_config *model;
14a2f74d 172 struct scope_state *state;
13f2b9d7 173
709468ba 174 if (!sdi)
13f2b9d7
DJ
175 return SR_ERR_ARG;
176
709468ba
UH
177 devc = sdi->priv;
178
61e77667 179 if ((cg_type = check_channel_group(devc, cg)) == CG_INVALID)
13f2b9d7 180 return SR_ERR;
06a3e78a 181
13f2b9d7
DJ
182 ret = SR_ERR_NA;
183 model = devc->model_config;
14a2f74d 184 state = devc->model_state;
06a3e78a 185
06a3e78a 186 switch (key) {
bf622e6d 187 case SR_CONF_NUM_HDIV:
13f2b9d7
DJ
188 *data = g_variant_new_int32(model->num_xdivs);
189 ret = SR_OK;
190 break;
3fd2dca2
DJ
191 case SR_CONF_TIMEBASE:
192 *data = g_variant_new("(tt)", (*model->timebases)[state->timebase][0],
193 (*model->timebases)[state->timebase][1]);
194 ret = SR_OK;
195 break;
13f2b9d7 196 case SR_CONF_NUM_VDIV:
61e77667 197 if (cg_type == CG_NONE) {
660e398f
UH
198 sr_err("No channel group specified.");
199 return SR_ERR_CHANNEL_GROUP;
61e77667 200 } else if (cg_type == CG_ANALOG) {
0a1f7b09 201 for (i = 0; i < model->analog_channels; i++) {
562b7ae5 202 if (cg != devc->analog_groups[i])
082972e8
UH
203 continue;
204 *data = g_variant_new_int32(model->num_ydivs);
205 ret = SR_OK;
206 break;
13f2b9d7
DJ
207 }
208
3fd2dca2
DJ
209 } else {
210 ret = SR_ERR_NA;
211 }
212 break;
213 case SR_CONF_VDIV:
61e77667
UH
214 if (cg_type == CG_NONE) {
215 sr_err("No channel group specified.");
216 return SR_ERR_CHANNEL_GROUP;
217 } else if (cg_type == CG_ANALOG) {
0a1f7b09 218 for (i = 0; i < model->analog_channels; i++) {
562b7ae5 219 if (cg != devc->analog_groups[i])
3fd2dca2
DJ
220 continue;
221 *data = g_variant_new("(tt)",
222 (*model->vdivs)[state->analog_channels[i].vdiv][0],
223 (*model->vdivs)[state->analog_channels[i].vdiv][1]);
224 ret = SR_OK;
225 break;
226 }
227
ccf14618
DJ
228 } else {
229 ret = SR_ERR_NA;
230 }
231 break;
232 case SR_CONF_TRIGGER_SOURCE:
233 *data = g_variant_new_string((*model->trigger_sources)[state->trigger_source]);
234 ret = SR_OK;
235 break;
3fd2dca2
DJ
236 case SR_CONF_TRIGGER_SLOPE:
237 *data = g_variant_new_string((*model->trigger_slopes)[state->trigger_slope]);
238 ret = SR_OK;
239 break;
240 case SR_CONF_HORIZ_TRIGGERPOS:
241 *data = g_variant_new_double(state->horiz_triggerpos);
242 ret = SR_OK;
243 break;
ccf14618 244 case SR_CONF_COUPLING:
61e77667 245 if (cg_type == CG_NONE) {
660e398f
UH
246 sr_err("No channel group specified.");
247 return SR_ERR_CHANNEL_GROUP;
61e77667 248 } else if (cg_type == CG_ANALOG) {
0a1f7b09 249 for (i = 0; i < model->analog_channels; i++) {
562b7ae5 250 if (cg != devc->analog_groups[i])
ccf14618
DJ
251 continue;
252 *data = g_variant_new_string((*model->coupling_options)[state->analog_channels[i].coupling]);
253 ret = SR_OK;
254 break;
255 }
256
13f2b9d7
DJ
257 } else {
258 ret = SR_ERR_NA;
259 }
260 break;
14a2f74d
DJ
261 case SR_CONF_SAMPLERATE:
262 *data = g_variant_new_uint64(state->sample_rate);
263 ret = SR_OK;
264 break;
06a3e78a 265 default:
13f2b9d7 266 ret = SR_ERR_NA;
06a3e78a
DJ
267 }
268
269 return ret;
270}
271
13f2b9d7
DJ
272static GVariant *build_tuples(const uint64_t (*array)[][2], unsigned int n)
273{
274 unsigned int i;
13f2b9d7
DJ
275 GVariant *rational[2];
276 GVariantBuilder gvb;
277
278 g_variant_builder_init(&gvb, G_VARIANT_TYPE_ARRAY);
279
280 for (i = 0; i < n; i++) {
281 rational[0] = g_variant_new_uint64((*array)[i][0]);
282 rational[1] = g_variant_new_uint64((*array)[i][1]);
283
89280b1a 284 /* FIXME: Valgrind reports a memory leak here. */
13f2b9d7
DJ
285 g_variant_builder_add_value(&gvb, g_variant_new_tuple(rational, 2));
286 }
287
288 return g_variant_builder_end(&gvb);
289}
290
584560f1 291static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sdi,
53b4680f 292 const struct sr_channel_group *cg)
06a3e78a 293{
61e77667 294 int ret, cg_type;
89280b1a 295 unsigned int i, j;
965b463d 296 char command[MAX_COMMAND_SIZE], float_str[30];
13f2b9d7 297 struct dev_context *devc;
329733d9 298 const struct scope_config *model;
13f2b9d7 299 struct scope_state *state;
89280b1a 300 const char *tmp;
ca9b9f48 301 uint64_t p, q;
89280b1a 302 double tmp_d;
23e1ea7a 303 gboolean update_sample_rate;
06a3e78a 304
b0baddef 305 if (!sdi)
13f2b9d7
DJ
306 return SR_ERR_ARG;
307
b0baddef
UH
308 devc = sdi->priv;
309
61e77667 310 if ((cg_type = check_channel_group(devc, cg)) == CG_INVALID)
13f2b9d7
DJ
311 return SR_ERR;
312
313 model = devc->model_config;
314 state = devc->model_state;
23e1ea7a 315 update_sample_rate = FALSE;
13f2b9d7
DJ
316
317 ret = SR_ERR_NA;
06a3e78a 318
06a3e78a 319 switch (key) {
13f2b9d7
DJ
320 case SR_CONF_LIMIT_FRAMES:
321 devc->frame_limit = g_variant_get_uint64(data);
322 ret = SR_OK;
323 break;
13f2b9d7 324 case SR_CONF_TRIGGER_SOURCE:
13f2b9d7 325 tmp = g_variant_get_string(data, NULL);
13f2b9d7 326 for (i = 0; (*model->trigger_sources)[i]; i++) {
082972e8
UH
327 if (g_strcmp0(tmp, (*model->trigger_sources)[i]) != 0)
328 continue;
329 state->trigger_source = i;
330 g_snprintf(command, sizeof(command),
331 (*model->scpi_dialect)[SCPI_CMD_SET_TRIGGER_SOURCE],
332 (*model->trigger_sources)[i]);
13f2b9d7 333
082972e8
UH
334 ret = sr_scpi_send(sdi->conn, command);
335 break;
13f2b9d7 336 }
89280b1a 337 break;
13f2b9d7 338 case SR_CONF_VDIV:
61e77667 339 if (cg_type == CG_NONE) {
660e398f
UH
340 sr_err("No channel group specified.");
341 return SR_ERR_CHANNEL_GROUP;
13f2b9d7
DJ
342 }
343
344 g_variant_get(data, "(tt)", &p, &q);
345
346 for (i = 0; i < model->num_vdivs; i++) {
082972e8
UH
347 if (p != (*model->vdivs)[i][0] ||
348 q != (*model->vdivs)[i][1])
349 continue;
0a1f7b09 350 for (j = 1; j <= model->analog_channels; j++) {
562b7ae5 351 if (cg != devc->analog_groups[j - 1])
082972e8 352 continue;
8de2dc3b 353 state->analog_channels[j - 1].vdiv = i;
965b463d 354 g_ascii_formatd(float_str, sizeof(float_str), "%E", (float) p / q);
082972e8
UH
355 g_snprintf(command, sizeof(command),
356 (*model->scpi_dialect)[SCPI_CMD_SET_VERTICAL_DIV],
965b463d 357 j, float_str);
082972e8
UH
358
359 if (sr_scpi_send(sdi->conn, command) != SR_OK ||
360 sr_scpi_get_opc(sdi->conn) != SR_OK)
361 return SR_ERR;
13f2b9d7 362
13f2b9d7
DJ
363 break;
364 }
082972e8
UH
365
366 ret = SR_OK;
367 break;
13f2b9d7 368 }
89280b1a 369 break;
13f2b9d7 370 case SR_CONF_TIMEBASE:
13f2b9d7
DJ
371 g_variant_get(data, "(tt)", &p, &q);
372
373 for (i = 0; i < model->num_timebases; i++) {
082972e8
UH
374 if (p != (*model->timebases)[i][0] ||
375 q != (*model->timebases)[i][1])
376 continue;
8de2dc3b 377 state->timebase = i;
965b463d 378 g_ascii_formatd(float_str, sizeof(float_str), "%E", (float) p / q);
082972e8
UH
379 g_snprintf(command, sizeof(command),
380 (*model->scpi_dialect)[SCPI_CMD_SET_TIMEBASE],
965b463d 381 float_str);
13f2b9d7 382
082972e8 383 ret = sr_scpi_send(sdi->conn, command);
23e1ea7a 384 update_sample_rate = TRUE;
082972e8 385 break;
13f2b9d7 386 }
89280b1a 387 break;
13f2b9d7 388 case SR_CONF_HORIZ_TRIGGERPOS:
89280b1a 389 tmp_d = g_variant_get_double(data);
13f2b9d7 390
89280b1a 391 if (tmp_d < 0.0 || tmp_d > 1.0)
13f2b9d7
DJ
392 return SR_ERR;
393
422a1c0d
DJ
394 state->horiz_triggerpos = tmp_d;
395 tmp_d = -(tmp_d - 0.5) *
396 ((double) (*model->timebases)[state->timebase][0] /
397 (*model->timebases)[state->timebase][1])
398 * model->num_xdivs;
399
400 g_ascii_formatd(float_str, sizeof(float_str), "%E", tmp_d);
13f2b9d7
DJ
401 g_snprintf(command, sizeof(command),
402 (*model->scpi_dialect)[SCPI_CMD_SET_HORIZ_TRIGGERPOS],
422a1c0d 403 float_str);
13f2b9d7
DJ
404
405 ret = sr_scpi_send(sdi->conn, command);
89280b1a 406 break;
13f2b9d7 407 case SR_CONF_TRIGGER_SLOPE:
ca9b9f48 408 tmp = g_variant_get_string(data, NULL);
e3abd15d
SB
409 for (i = 0; (*model->trigger_slopes)[i]; i++) {
410 if (g_strcmp0(tmp, (*model->trigger_slopes)[i]) != 0)
411 continue;
412 state->trigger_slope = i;
413 g_snprintf(command, sizeof(command),
414 (*model->scpi_dialect)[SCPI_CMD_SET_TRIGGER_SLOPE],
415 (*model->trigger_slopes)[i]);
13f2b9d7 416
e3abd15d
SB
417 ret = sr_scpi_send(sdi->conn, command);
418 break;
419 }
89280b1a 420 break;
13f2b9d7 421 case SR_CONF_COUPLING:
61e77667 422 if (cg_type == CG_NONE) {
660e398f
UH
423 sr_err("No channel group specified.");
424 return SR_ERR_CHANNEL_GROUP;
13f2b9d7
DJ
425 }
426
427 tmp = g_variant_get_string(data, NULL);
428
429 for (i = 0; (*model->coupling_options)[i]; i++) {
082972e8
UH
430 if (strcmp(tmp, (*model->coupling_options)[i]) != 0)
431 continue;
0a1f7b09 432 for (j = 1; j <= model->analog_channels; j++) {
562b7ae5 433 if (cg != devc->analog_groups[j - 1])
082972e8
UH
434 continue;
435 state->analog_channels[j-1].coupling = i;
13f2b9d7 436
082972e8
UH
437 g_snprintf(command, sizeof(command),
438 (*model->scpi_dialect)[SCPI_CMD_SET_COUPLING],
439 j, tmp);
440
441 if (sr_scpi_send(sdi->conn, command) != SR_OK ||
442 sr_scpi_get_opc(sdi->conn) != SR_OK)
443 return SR_ERR;
13f2b9d7
DJ
444 break;
445 }
082972e8
UH
446
447 ret = SR_OK;
448 break;
13f2b9d7 449 }
89280b1a 450 break;
06a3e78a
DJ
451 default:
452 ret = SR_ERR_NA;
13f2b9d7 453 break;
06a3e78a
DJ
454 }
455
13f2b9d7
DJ
456 if (ret == SR_OK)
457 ret = sr_scpi_get_opc(sdi->conn);
458
23e1ea7a
DJ
459 if (ret == SR_OK && update_sample_rate)
460 ret = hmo_update_sample_rate(sdi);
461
06a3e78a
DJ
462 return ret;
463}
464
584560f1 465static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi,
53b4680f 466 const struct sr_channel_group *cg)
06a3e78a 467{
6ec3ef9b
AJ
468 int cg_type = CG_NONE;
469 struct dev_context *devc = NULL;
329733d9 470 const struct scope_config *model = NULL;
13f2b9d7 471
4d399734
UH
472 if (sdi) {
473 devc = sdi->priv;
6ec3ef9b
AJ
474 if ((cg_type = check_channel_group(devc, cg)) == CG_INVALID)
475 return SR_ERR;
13f2b9d7 476
6ec3ef9b
AJ
477 model = devc->model_config;
478 }
06a3e78a 479
06a3e78a 480 switch (key) {
226363c4 481 case SR_CONF_SCAN_OPTIONS:
584560f1
BV
482 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
483 scanopts, ARRAY_SIZE(scanopts), sizeof(uint32_t));
226363c4 484 break;
13f2b9d7 485 case SR_CONF_DEVICE_OPTIONS:
61e77667 486 if (cg_type == CG_NONE) {
6ec3ef9b
AJ
487 if (model)
488 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
489 model->devopts, model->num_devopts, sizeof(uint32_t));
490 else
491 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
492 drvopts, ARRAY_SIZE(drvopts), sizeof(uint32_t));
61e77667 493 } else if (cg_type == CG_ANALOG) {
584560f1 494 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
f254bc4b 495 model->analog_devopts, model->num_analog_devopts,
584560f1 496 sizeof(uint32_t));
13f2b9d7 497 } else {
584560f1
BV
498 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
499 NULL, 0, sizeof(uint32_t));
13f2b9d7
DJ
500 }
501 break;
13f2b9d7 502 case SR_CONF_COUPLING:
61e77667 503 if (cg_type == CG_NONE)
660e398f 504 return SR_ERR_CHANNEL_GROUP;
13f2b9d7 505 *data = g_variant_new_strv(*model->coupling_options,
89280b1a 506 g_strv_length((char **)*model->coupling_options));
13f2b9d7 507 break;
13f2b9d7 508 case SR_CONF_TRIGGER_SOURCE:
6ec3ef9b
AJ
509 if (!model)
510 return SR_ERR_ARG;
13f2b9d7 511 *data = g_variant_new_strv(*model->trigger_sources,
89280b1a 512 g_strv_length((char **)*model->trigger_sources));
13f2b9d7 513 break;
bbabdaf1 514 case SR_CONF_TRIGGER_SLOPE:
6ec3ef9b
AJ
515 if (!model)
516 return SR_ERR_ARG;
bbabdaf1
DJ
517 *data = g_variant_new_strv(*model->trigger_slopes,
518 g_strv_length((char **)*model->trigger_slopes));
519 break;
13f2b9d7 520 case SR_CONF_TIMEBASE:
6ec3ef9b
AJ
521 if (!model)
522 return SR_ERR_ARG;
13f2b9d7
DJ
523 *data = build_tuples(model->timebases, model->num_timebases);
524 break;
13f2b9d7 525 case SR_CONF_VDIV:
61e77667 526 if (cg_type == CG_NONE)
660e398f 527 return SR_ERR_CHANNEL_GROUP;
13f2b9d7
DJ
528 *data = build_tuples(model->vdivs, model->num_vdivs);
529 break;
06a3e78a
DJ
530 default:
531 return SR_ERR_NA;
532 }
533
13f2b9d7 534 return SR_OK;
06a3e78a
DJ
535}
536
13f2b9d7 537SR_PRIV int hmo_request_data(const struct sr_dev_inst *sdi)
06a3e78a 538{
13f2b9d7 539 char command[MAX_COMMAND_SIZE];
ba7dd8bb 540 struct sr_channel *ch;
13f2b9d7 541 struct dev_context *devc;
329733d9 542 const struct scope_config *model;
13f2b9d7
DJ
543
544 devc = sdi->priv;
545 model = devc->model_config;
546
ba7dd8bb 547 ch = devc->current_channel->data;
13f2b9d7 548
ba7dd8bb 549 switch (ch->type) {
3f239f08 550 case SR_CHANNEL_ANALOG:
13f2b9d7
DJ
551 g_snprintf(command, sizeof(command),
552 (*model->scpi_dialect)[SCPI_CMD_GET_ANALOG_DATA],
65a6794e
GS
553#ifdef WORDS_BIGENDIAN
554 "MSBF",
555#else
556 "LSBF",
557#endif
ba7dd8bb 558 ch->index + 1);
13f2b9d7 559 break;
3f239f08 560 case SR_CHANNEL_LOGIC:
13f2b9d7
DJ
561 g_snprintf(command, sizeof(command),
562 (*model->scpi_dialect)[SCPI_CMD_GET_DIG_DATA],
ba7dd8bb 563 ch->index < 8 ? 1 : 2);
13f2b9d7
DJ
564 break;
565 default:
ba7dd8bb 566 sr_err("Invalid channel type.");
13f2b9d7
DJ
567 break;
568 }
569
570 return sr_scpi_send(sdi->conn, command);
571}
572
ba7dd8bb 573static int hmo_check_channels(GSList *channels)
13f2b9d7
DJ
574{
575 GSList *l;
ba7dd8bb 576 struct sr_channel *ch;
4889acef
GS
577 gboolean enabled_chan[MAX_ANALOG_CHANNEL_COUNT];
578 gboolean enabled_pod[MAX_DIGITAL_GROUP_COUNT];
579 size_t idx;
580
581 /* Preset "not enabled" for all channels / pods. */
582 for (idx = 0; idx < ARRAY_SIZE(enabled_chan); idx++)
583 enabled_chan[idx] = FALSE;
584 for (idx = 0; idx < ARRAY_SIZE(enabled_pod); idx++)
585 enabled_pod[idx] = FALSE;
586
587 /*
588 * Determine which channels / pods are required for the caller's
589 * specified configuration.
590 */
ba7dd8bb
UH
591 for (l = channels; l; l = l->next) {
592 ch = l->data;
593 switch (ch->type) {
3f239f08 594 case SR_CHANNEL_ANALOG:
4889acef
GS
595 idx = ch->index;
596 if (idx < ARRAY_SIZE(enabled_chan))
597 enabled_chan[idx] = TRUE;
13f2b9d7 598 break;
3f239f08 599 case SR_CHANNEL_LOGIC:
4889acef
GS
600 idx = ch->index / 8;
601 if (idx < ARRAY_SIZE(enabled_pod))
602 enabled_pod[idx] = TRUE;
13f2b9d7 603 break;
13f2b9d7
DJ
604 default:
605 return SR_ERR;
606 }
607 }
608
4889acef
GS
609 /*
610 * Check for resource conflicts. Some channels can be either
611 * analog or digital, but never both at the same time.
612 *
613 * Note that the constraints might depend on the specific model.
614 * These tests might need some adjustment when support for more
615 * models gets added to the driver.
616 */
617 if (enabled_pod[0] && enabled_chan[2])
618 return SR_ERR;
619 if (enabled_pod[1] && enabled_chan[3])
13f2b9d7 620 return SR_ERR;
13f2b9d7
DJ
621 return SR_OK;
622}
623
ba7dd8bb 624static int hmo_setup_channels(const struct sr_dev_inst *sdi)
13f2b9d7
DJ
625{
626 GSList *l;
627 unsigned int i;
23e1ea7a 628 gboolean *pod_enabled, setup_changed;
13f2b9d7 629 char command[MAX_COMMAND_SIZE];
13f2b9d7 630 struct scope_state *state;
329733d9 631 const struct scope_config *model;
ba7dd8bb 632 struct sr_channel *ch;
13f2b9d7 633 struct dev_context *devc;
23f43dff 634 struct sr_scpi_dev_inst *scpi;
13f2b9d7
DJ
635
636 devc = sdi->priv;
23f43dff 637 scpi = sdi->conn;
13f2b9d7
DJ
638 state = devc->model_state;
639 model = devc->model_config;
23e1ea7a 640 setup_changed = FALSE;
13f2b9d7
DJ
641
642 pod_enabled = g_try_malloc0(sizeof(gboolean) * model->digital_pods);
643
ba7dd8bb
UH
644 for (l = sdi->channels; l; l = l->next) {
645 ch = l->data;
646 switch (ch->type) {
3f239f08 647 case SR_CHANNEL_ANALOG:
ba7dd8bb 648 if (ch->enabled == state->analog_channels[ch->index].state)
082972e8
UH
649 break;
650 g_snprintf(command, sizeof(command),
651 (*model->scpi_dialect)[SCPI_CMD_SET_ANALOG_CHAN_STATE],
ba7dd8bb 652 ch->index + 1, ch->enabled);
13f2b9d7 653
23f43dff 654 if (sr_scpi_send(scpi, command) != SR_OK)
082972e8 655 return SR_ERR;
ba7dd8bb 656 state->analog_channels[ch->index].state = ch->enabled;
23e1ea7a 657 setup_changed = TRUE;
89280b1a 658 break;
3f239f08 659 case SR_CHANNEL_LOGIC:
13f2b9d7
DJ
660 /*
661 * A digital POD needs to be enabled for every group of
ba7dd8bb 662 * 8 channels.
13f2b9d7 663 */
ba7dd8bb
UH
664 if (ch->enabled)
665 pod_enabled[ch->index < 8 ? 0 : 1] = TRUE;
13f2b9d7 666
ba7dd8bb 667 if (ch->enabled == state->digital_channels[ch->index])
082972e8
UH
668 break;
669 g_snprintf(command, sizeof(command),
670 (*model->scpi_dialect)[SCPI_CMD_SET_DIG_CHAN_STATE],
ba7dd8bb 671 ch->index, ch->enabled);
13f2b9d7 672
23f43dff 673 if (sr_scpi_send(scpi, command) != SR_OK)
082972e8 674 return SR_ERR;
13f2b9d7 675
ba7dd8bb 676 state->digital_channels[ch->index] = ch->enabled;
23e1ea7a 677 setup_changed = TRUE;
89280b1a 678 break;
13f2b9d7
DJ
679 default:
680 return SR_ERR;
681 }
682 }
683
0a1f7b09 684 for (i = 1; i <= model->digital_pods; i++) {
082972e8
UH
685 if (state->digital_pods[i - 1] == pod_enabled[i - 1])
686 continue;
687 g_snprintf(command, sizeof(command),
688 (*model->scpi_dialect)[SCPI_CMD_SET_DIG_POD_STATE],
689 i, pod_enabled[i - 1]);
23f43dff 690 if (sr_scpi_send(scpi, command) != SR_OK)
082972e8
UH
691 return SR_ERR;
692 state->digital_pods[i - 1] = pod_enabled[i - 1];
23e1ea7a 693 setup_changed = TRUE;
13f2b9d7
DJ
694 }
695
696 g_free(pod_enabled);
697
23e1ea7a
DJ
698 if (setup_changed && hmo_update_sample_rate(sdi) != SR_OK)
699 return SR_ERR;
700
13f2b9d7
DJ
701 return SR_OK;
702}
703
695dc859 704static int dev_acquisition_start(const struct sr_dev_inst *sdi)
13f2b9d7
DJ
705{
706 GSList *l;
1b0f62df 707 gboolean digital_added[MAX_DIGITAL_GROUP_COUNT];
e06875b2 708 size_t group, pod_count;
ba7dd8bb 709 struct sr_channel *ch;
13f2b9d7 710 struct dev_context *devc;
23f43dff 711 struct sr_scpi_dev_inst *scpi;
40a75c8e 712 int ret;
06a3e78a 713
23f43dff 714 scpi = sdi->conn;
13f2b9d7 715 devc = sdi->priv;
13f2b9d7 716
1b0f62df
GS
717 /* Preset empty results. */
718 for (group = 0; group < ARRAY_SIZE(digital_added); group++)
719 digital_added[group] = FALSE;
db81fbb5
SA
720 g_slist_free(devc->enabled_channels);
721 devc->enabled_channels = NULL;
722
1b0f62df 723 /*
e06875b2
GS
724 * Contruct the list of enabled channels. Determine the highest
725 * number of digital pods involved in the acquisition.
1b0f62df 726 */
e06875b2 727 pod_count = 0;
ba7dd8bb
UH
728 for (l = sdi->channels; l; l = l->next) {
729 ch = l->data;
730 if (!ch->enabled)
082972e8 731 continue;
1b0f62df
GS
732 /* Only add a single digital channel per group (pod). */
733 group = ch->index / 8;
734 if (ch->type != SR_CHANNEL_LOGIC || !digital_added[group]) {
ba7dd8bb
UH
735 devc->enabled_channels = g_slist_append(
736 devc->enabled_channels, ch);
e06875b2 737 if (ch->type == SR_CHANNEL_LOGIC) {
1b0f62df 738 digital_added[group] = TRUE;
e06875b2
GS
739 if (pod_count < group + 1)
740 pod_count = group + 1;
741 }
13f2b9d7
DJ
742 }
743 }
ba7dd8bb 744 if (!devc->enabled_channels)
13f2b9d7 745 return SR_ERR;
e06875b2
GS
746 devc->pod_count = pod_count;
747 devc->logic_data = NULL;
13f2b9d7 748
1b0f62df
GS
749 /*
750 * Check constraints. Some channels can be either analog or
751 * digital, but not both at the same time.
752 */
ba7dd8bb
UH
753 if (hmo_check_channels(devc->enabled_channels) != SR_OK) {
754 sr_err("Invalid channel configuration specified!");
40a75c8e
GS
755 ret = SR_ERR_NA;
756 goto free_enabled;
13f2b9d7
DJ
757 }
758
1b0f62df
GS
759 /*
760 * Configure the analog and digital channels and the
761 * corresponding digital pods.
762 */
ba7dd8bb
UH
763 if (hmo_setup_channels(sdi) != SR_OK) {
764 sr_err("Failed to setup channel configuration!");
40a75c8e
GS
765 ret = SR_ERR;
766 goto free_enabled;
13f2b9d7
DJ
767 }
768
1b0f62df
GS
769 /*
770 * Start acquisition on the first enabled channel. The
771 * receive routine will continue driving the acquisition.
772 */
102f1239
BV
773 sr_scpi_source_add(sdi->session, scpi, G_IO_IN, 50,
774 hmo_receive_data, (void *)sdi);
13f2b9d7 775
bee2b016 776 std_session_send_df_header(sdi);
13f2b9d7 777
ba7dd8bb 778 devc->current_channel = devc->enabled_channels;
13f2b9d7
DJ
779
780 return hmo_request_data(sdi);
40a75c8e
GS
781
782free_enabled:
783 g_slist_free(devc->enabled_channels);
784 devc->enabled_channels = NULL;
785 return ret;
06a3e78a
DJ
786}
787
695dc859 788static int dev_acquisition_stop(struct sr_dev_inst *sdi)
06a3e78a 789{
13f2b9d7 790 struct dev_context *devc;
23f43dff 791 struct sr_scpi_dev_inst *scpi;
13f2b9d7 792
bee2b016 793 std_session_send_df_end(sdi);
66e3219d 794
13f2b9d7
DJ
795 devc = sdi->priv;
796
ef1a346b 797 devc->num_frames = 0;
ba7dd8bb
UH
798 g_slist_free(devc->enabled_channels);
799 devc->enabled_channels = NULL;
23f43dff 800 scpi = sdi->conn;
102f1239 801 sr_scpi_source_remove(sdi->session, scpi);
06a3e78a
DJ
802
803 return SR_OK;
804}
805
dd5c48a6 806static struct sr_dev_driver hameg_hmo_driver_info = {
06a3e78a 807 .name = "hameg-hmo",
89280b1a 808 .longname = "Hameg HMO",
06a3e78a 809 .api_version = 1,
c2fdcc25 810 .init = std_init,
700d6b64 811 .cleanup = std_cleanup,
06a3e78a 812 .scan = scan,
c01bf34c 813 .dev_list = std_dev_list,
06a3e78a
DJ
814 .dev_clear = dev_clear,
815 .config_get = config_get,
816 .config_set = config_set,
817 .config_list = config_list,
818 .dev_open = dev_open,
819 .dev_close = dev_close,
820 .dev_acquisition_start = dev_acquisition_start,
821 .dev_acquisition_stop = dev_acquisition_stop,
41812aca 822 .context = NULL,
06a3e78a 823};
dd5c48a6 824SR_REGISTER_DEV_DRIVER(hameg_hmo_driver_info);