]> sigrok.org Git - libsigrok.git/blame - src/hardware/hameg-hmo/api.c
drivers/input: Remove some hardcoded values.
[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
584560f1 34static const uint32_t scanopts[] = {
13f2b9d7
DJ
35 SR_CONF_CONN,
36 SR_CONF_SERIALCOMM,
37};
38
55fb76b3
UH
39static const uint32_t drvopts[] = {
40 SR_CONF_OSCILLOSCOPE,
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
dd7a72ea
UH
165static int config_get(uint32_t key, GVariant **data,
166 const struct sr_dev_inst *sdi, 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
dd7a72ea
UH
272static int config_set(uint32_t key, GVariant *data,
273 const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
06a3e78a 274{
61e77667 275 int ret, cg_type;
89280b1a 276 unsigned int i, j;
965b463d 277 char command[MAX_COMMAND_SIZE], float_str[30];
13f2b9d7 278 struct dev_context *devc;
329733d9 279 const struct scope_config *model;
13f2b9d7 280 struct scope_state *state;
89280b1a 281 const char *tmp;
ca9b9f48 282 uint64_t p, q;
89280b1a 283 double tmp_d;
23e1ea7a 284 gboolean update_sample_rate;
06a3e78a 285
b0baddef 286 if (!sdi)
13f2b9d7
DJ
287 return SR_ERR_ARG;
288
b0baddef
UH
289 devc = sdi->priv;
290
61e77667 291 if ((cg_type = check_channel_group(devc, cg)) == CG_INVALID)
13f2b9d7
DJ
292 return SR_ERR;
293
294 model = devc->model_config;
295 state = devc->model_state;
23e1ea7a 296 update_sample_rate = FALSE;
13f2b9d7
DJ
297
298 ret = SR_ERR_NA;
06a3e78a 299
06a3e78a 300 switch (key) {
13f2b9d7
DJ
301 case SR_CONF_LIMIT_FRAMES:
302 devc->frame_limit = g_variant_get_uint64(data);
303 ret = SR_OK;
304 break;
13f2b9d7 305 case SR_CONF_TRIGGER_SOURCE:
13f2b9d7 306 tmp = g_variant_get_string(data, NULL);
13f2b9d7 307 for (i = 0; (*model->trigger_sources)[i]; i++) {
082972e8
UH
308 if (g_strcmp0(tmp, (*model->trigger_sources)[i]) != 0)
309 continue;
310 state->trigger_source = i;
311 g_snprintf(command, sizeof(command),
312 (*model->scpi_dialect)[SCPI_CMD_SET_TRIGGER_SOURCE],
313 (*model->trigger_sources)[i]);
13f2b9d7 314
082972e8
UH
315 ret = sr_scpi_send(sdi->conn, command);
316 break;
13f2b9d7 317 }
89280b1a 318 break;
13f2b9d7 319 case SR_CONF_VDIV:
61e77667 320 if (cg_type == CG_NONE) {
660e398f
UH
321 sr_err("No channel group specified.");
322 return SR_ERR_CHANNEL_GROUP;
13f2b9d7
DJ
323 }
324
325 g_variant_get(data, "(tt)", &p, &q);
326
327 for (i = 0; i < model->num_vdivs; i++) {
082972e8
UH
328 if (p != (*model->vdivs)[i][0] ||
329 q != (*model->vdivs)[i][1])
330 continue;
0a1f7b09 331 for (j = 1; j <= model->analog_channels; j++) {
562b7ae5 332 if (cg != devc->analog_groups[j - 1])
082972e8 333 continue;
8de2dc3b 334 state->analog_channels[j - 1].vdiv = i;
965b463d 335 g_ascii_formatd(float_str, sizeof(float_str), "%E", (float) p / q);
082972e8
UH
336 g_snprintf(command, sizeof(command),
337 (*model->scpi_dialect)[SCPI_CMD_SET_VERTICAL_DIV],
965b463d 338 j, float_str);
082972e8
UH
339
340 if (sr_scpi_send(sdi->conn, command) != SR_OK ||
341 sr_scpi_get_opc(sdi->conn) != SR_OK)
342 return SR_ERR;
13f2b9d7 343
13f2b9d7
DJ
344 break;
345 }
082972e8
UH
346
347 ret = SR_OK;
348 break;
13f2b9d7 349 }
89280b1a 350 break;
13f2b9d7 351 case SR_CONF_TIMEBASE:
13f2b9d7
DJ
352 g_variant_get(data, "(tt)", &p, &q);
353
354 for (i = 0; i < model->num_timebases; i++) {
082972e8
UH
355 if (p != (*model->timebases)[i][0] ||
356 q != (*model->timebases)[i][1])
357 continue;
8de2dc3b 358 state->timebase = i;
965b463d 359 g_ascii_formatd(float_str, sizeof(float_str), "%E", (float) p / q);
082972e8
UH
360 g_snprintf(command, sizeof(command),
361 (*model->scpi_dialect)[SCPI_CMD_SET_TIMEBASE],
965b463d 362 float_str);
13f2b9d7 363
082972e8 364 ret = sr_scpi_send(sdi->conn, command);
23e1ea7a 365 update_sample_rate = TRUE;
082972e8 366 break;
13f2b9d7 367 }
89280b1a 368 break;
13f2b9d7 369 case SR_CONF_HORIZ_TRIGGERPOS:
89280b1a 370 tmp_d = g_variant_get_double(data);
13f2b9d7 371
89280b1a 372 if (tmp_d < 0.0 || tmp_d > 1.0)
13f2b9d7
DJ
373 return SR_ERR;
374
422a1c0d
DJ
375 state->horiz_triggerpos = tmp_d;
376 tmp_d = -(tmp_d - 0.5) *
377 ((double) (*model->timebases)[state->timebase][0] /
378 (*model->timebases)[state->timebase][1])
379 * model->num_xdivs;
380
381 g_ascii_formatd(float_str, sizeof(float_str), "%E", tmp_d);
13f2b9d7
DJ
382 g_snprintf(command, sizeof(command),
383 (*model->scpi_dialect)[SCPI_CMD_SET_HORIZ_TRIGGERPOS],
422a1c0d 384 float_str);
13f2b9d7
DJ
385
386 ret = sr_scpi_send(sdi->conn, command);
89280b1a 387 break;
13f2b9d7 388 case SR_CONF_TRIGGER_SLOPE:
ca9b9f48 389 tmp = g_variant_get_string(data, NULL);
e3abd15d
SB
390 for (i = 0; (*model->trigger_slopes)[i]; i++) {
391 if (g_strcmp0(tmp, (*model->trigger_slopes)[i]) != 0)
392 continue;
393 state->trigger_slope = i;
394 g_snprintf(command, sizeof(command),
395 (*model->scpi_dialect)[SCPI_CMD_SET_TRIGGER_SLOPE],
396 (*model->trigger_slopes)[i]);
13f2b9d7 397
e3abd15d
SB
398 ret = sr_scpi_send(sdi->conn, command);
399 break;
400 }
89280b1a 401 break;
13f2b9d7 402 case SR_CONF_COUPLING:
61e77667 403 if (cg_type == CG_NONE) {
660e398f
UH
404 sr_err("No channel group specified.");
405 return SR_ERR_CHANNEL_GROUP;
13f2b9d7
DJ
406 }
407
408 tmp = g_variant_get_string(data, NULL);
409
410 for (i = 0; (*model->coupling_options)[i]; i++) {
082972e8
UH
411 if (strcmp(tmp, (*model->coupling_options)[i]) != 0)
412 continue;
0a1f7b09 413 for (j = 1; j <= model->analog_channels; j++) {
562b7ae5 414 if (cg != devc->analog_groups[j - 1])
082972e8
UH
415 continue;
416 state->analog_channels[j-1].coupling = i;
13f2b9d7 417
082972e8
UH
418 g_snprintf(command, sizeof(command),
419 (*model->scpi_dialect)[SCPI_CMD_SET_COUPLING],
420 j, tmp);
421
422 if (sr_scpi_send(sdi->conn, command) != SR_OK ||
423 sr_scpi_get_opc(sdi->conn) != SR_OK)
424 return SR_ERR;
13f2b9d7
DJ
425 break;
426 }
082972e8
UH
427
428 ret = SR_OK;
429 break;
13f2b9d7 430 }
89280b1a 431 break;
06a3e78a
DJ
432 default:
433 ret = SR_ERR_NA;
13f2b9d7 434 break;
06a3e78a
DJ
435 }
436
13f2b9d7
DJ
437 if (ret == SR_OK)
438 ret = sr_scpi_get_opc(sdi->conn);
439
23e1ea7a
DJ
440 if (ret == SR_OK && update_sample_rate)
441 ret = hmo_update_sample_rate(sdi);
442
06a3e78a
DJ
443 return ret;
444}
445
dd7a72ea
UH
446static int config_list(uint32_t key, GVariant **data,
447 const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
06a3e78a 448{
6ec3ef9b
AJ
449 int cg_type = CG_NONE;
450 struct dev_context *devc = NULL;
329733d9 451 const struct scope_config *model = NULL;
13f2b9d7 452
4d399734
UH
453 if (sdi) {
454 devc = sdi->priv;
6ec3ef9b
AJ
455 if ((cg_type = check_channel_group(devc, cg)) == CG_INVALID)
456 return SR_ERR;
13f2b9d7 457
6ec3ef9b
AJ
458 model = devc->model_config;
459 }
06a3e78a 460
06a3e78a 461 switch (key) {
226363c4 462 case SR_CONF_SCAN_OPTIONS:
53012da6 463 *data = std_gvar_array_u32(ARRAY_AND_SIZE(scanopts));
226363c4 464 break;
13f2b9d7 465 case SR_CONF_DEVICE_OPTIONS:
61e77667 466 if (cg_type == CG_NONE) {
6ec3ef9b 467 if (model)
105df674 468 *data = std_gvar_array_u32((const uint32_t *)model->devopts, model->num_devopts);
6ec3ef9b 469 else
53012da6 470 *data = std_gvar_array_u32(ARRAY_AND_SIZE(drvopts));
61e77667 471 } else if (cg_type == CG_ANALOG) {
105df674 472 *data = std_gvar_array_u32((const uint32_t *)model->devopts_cg_analog, model->num_devopts_cg_analog);
13f2b9d7 473 } else {
105df674 474 *data = std_gvar_array_u32(NULL, 0);
13f2b9d7
DJ
475 }
476 break;
13f2b9d7 477 case SR_CONF_COUPLING:
61e77667 478 if (cg_type == CG_NONE)
660e398f 479 return SR_ERR_CHANNEL_GROUP;
13f2b9d7 480 *data = g_variant_new_strv(*model->coupling_options,
89280b1a 481 g_strv_length((char **)*model->coupling_options));
13f2b9d7 482 break;
13f2b9d7 483 case SR_CONF_TRIGGER_SOURCE:
6ec3ef9b
AJ
484 if (!model)
485 return SR_ERR_ARG;
13f2b9d7 486 *data = g_variant_new_strv(*model->trigger_sources,
89280b1a 487 g_strv_length((char **)*model->trigger_sources));
13f2b9d7 488 break;
bbabdaf1 489 case SR_CONF_TRIGGER_SLOPE:
6ec3ef9b
AJ
490 if (!model)
491 return SR_ERR_ARG;
bbabdaf1
DJ
492 *data = g_variant_new_strv(*model->trigger_slopes,
493 g_strv_length((char **)*model->trigger_slopes));
494 break;
13f2b9d7 495 case SR_CONF_TIMEBASE:
6ec3ef9b
AJ
496 if (!model)
497 return SR_ERR_ARG;
58ffcf97 498 *data = std_gvar_tuple_array(*model->timebases, model->num_timebases);
13f2b9d7 499 break;
13f2b9d7 500 case SR_CONF_VDIV:
61e77667 501 if (cg_type == CG_NONE)
660e398f 502 return SR_ERR_CHANNEL_GROUP;
58ffcf97 503 *data = std_gvar_tuple_array(*model->vdivs, model->num_vdivs);
13f2b9d7 504 break;
06a3e78a
DJ
505 default:
506 return SR_ERR_NA;
507 }
508
13f2b9d7 509 return SR_OK;
06a3e78a
DJ
510}
511
13f2b9d7 512SR_PRIV int hmo_request_data(const struct sr_dev_inst *sdi)
06a3e78a 513{
13f2b9d7 514 char command[MAX_COMMAND_SIZE];
ba7dd8bb 515 struct sr_channel *ch;
13f2b9d7 516 struct dev_context *devc;
329733d9 517 const struct scope_config *model;
13f2b9d7
DJ
518
519 devc = sdi->priv;
520 model = devc->model_config;
521
ba7dd8bb 522 ch = devc->current_channel->data;
13f2b9d7 523
ba7dd8bb 524 switch (ch->type) {
3f239f08 525 case SR_CHANNEL_ANALOG:
13f2b9d7
DJ
526 g_snprintf(command, sizeof(command),
527 (*model->scpi_dialect)[SCPI_CMD_GET_ANALOG_DATA],
65a6794e
GS
528#ifdef WORDS_BIGENDIAN
529 "MSBF",
530#else
531 "LSBF",
532#endif
ba7dd8bb 533 ch->index + 1);
13f2b9d7 534 break;
3f239f08 535 case SR_CHANNEL_LOGIC:
13f2b9d7
DJ
536 g_snprintf(command, sizeof(command),
537 (*model->scpi_dialect)[SCPI_CMD_GET_DIG_DATA],
ba7dd8bb 538 ch->index < 8 ? 1 : 2);
13f2b9d7
DJ
539 break;
540 default:
ba7dd8bb 541 sr_err("Invalid channel type.");
13f2b9d7
DJ
542 break;
543 }
544
545 return sr_scpi_send(sdi->conn, command);
546}
547
ba7dd8bb 548static int hmo_check_channels(GSList *channels)
13f2b9d7
DJ
549{
550 GSList *l;
ba7dd8bb 551 struct sr_channel *ch;
4889acef
GS
552 gboolean enabled_chan[MAX_ANALOG_CHANNEL_COUNT];
553 gboolean enabled_pod[MAX_DIGITAL_GROUP_COUNT];
554 size_t idx;
555
556 /* Preset "not enabled" for all channels / pods. */
557 for (idx = 0; idx < ARRAY_SIZE(enabled_chan); idx++)
558 enabled_chan[idx] = FALSE;
559 for (idx = 0; idx < ARRAY_SIZE(enabled_pod); idx++)
560 enabled_pod[idx] = FALSE;
561
562 /*
563 * Determine which channels / pods are required for the caller's
564 * specified configuration.
565 */
ba7dd8bb
UH
566 for (l = channels; l; l = l->next) {
567 ch = l->data;
568 switch (ch->type) {
3f239f08 569 case SR_CHANNEL_ANALOG:
4889acef
GS
570 idx = ch->index;
571 if (idx < ARRAY_SIZE(enabled_chan))
572 enabled_chan[idx] = TRUE;
13f2b9d7 573 break;
3f239f08 574 case SR_CHANNEL_LOGIC:
4889acef
GS
575 idx = ch->index / 8;
576 if (idx < ARRAY_SIZE(enabled_pod))
577 enabled_pod[idx] = TRUE;
13f2b9d7 578 break;
13f2b9d7
DJ
579 default:
580 return SR_ERR;
581 }
582 }
583
4889acef
GS
584 /*
585 * Check for resource conflicts. Some channels can be either
586 * analog or digital, but never both at the same time.
587 *
588 * Note that the constraints might depend on the specific model.
589 * These tests might need some adjustment when support for more
590 * models gets added to the driver.
591 */
592 if (enabled_pod[0] && enabled_chan[2])
593 return SR_ERR;
594 if (enabled_pod[1] && enabled_chan[3])
13f2b9d7 595 return SR_ERR;
13f2b9d7
DJ
596 return SR_OK;
597}
598
ba7dd8bb 599static int hmo_setup_channels(const struct sr_dev_inst *sdi)
13f2b9d7
DJ
600{
601 GSList *l;
602 unsigned int i;
23e1ea7a 603 gboolean *pod_enabled, setup_changed;
13f2b9d7 604 char command[MAX_COMMAND_SIZE];
13f2b9d7 605 struct scope_state *state;
329733d9 606 const struct scope_config *model;
ba7dd8bb 607 struct sr_channel *ch;
13f2b9d7 608 struct dev_context *devc;
23f43dff 609 struct sr_scpi_dev_inst *scpi;
13f2b9d7
DJ
610
611 devc = sdi->priv;
23f43dff 612 scpi = sdi->conn;
13f2b9d7
DJ
613 state = devc->model_state;
614 model = devc->model_config;
23e1ea7a 615 setup_changed = FALSE;
13f2b9d7
DJ
616
617 pod_enabled = g_try_malloc0(sizeof(gboolean) * model->digital_pods);
618
ba7dd8bb
UH
619 for (l = sdi->channels; l; l = l->next) {
620 ch = l->data;
621 switch (ch->type) {
3f239f08 622 case SR_CHANNEL_ANALOG:
ba7dd8bb 623 if (ch->enabled == state->analog_channels[ch->index].state)
082972e8
UH
624 break;
625 g_snprintf(command, sizeof(command),
626 (*model->scpi_dialect)[SCPI_CMD_SET_ANALOG_CHAN_STATE],
ba7dd8bb 627 ch->index + 1, ch->enabled);
13f2b9d7 628
23f43dff 629 if (sr_scpi_send(scpi, command) != SR_OK)
082972e8 630 return SR_ERR;
ba7dd8bb 631 state->analog_channels[ch->index].state = ch->enabled;
23e1ea7a 632 setup_changed = TRUE;
89280b1a 633 break;
3f239f08 634 case SR_CHANNEL_LOGIC:
13f2b9d7
DJ
635 /*
636 * A digital POD needs to be enabled for every group of
ba7dd8bb 637 * 8 channels.
13f2b9d7 638 */
ba7dd8bb
UH
639 if (ch->enabled)
640 pod_enabled[ch->index < 8 ? 0 : 1] = TRUE;
13f2b9d7 641
ba7dd8bb 642 if (ch->enabled == state->digital_channels[ch->index])
082972e8
UH
643 break;
644 g_snprintf(command, sizeof(command),
645 (*model->scpi_dialect)[SCPI_CMD_SET_DIG_CHAN_STATE],
ba7dd8bb 646 ch->index, ch->enabled);
13f2b9d7 647
23f43dff 648 if (sr_scpi_send(scpi, command) != SR_OK)
082972e8 649 return SR_ERR;
13f2b9d7 650
ba7dd8bb 651 state->digital_channels[ch->index] = ch->enabled;
23e1ea7a 652 setup_changed = TRUE;
89280b1a 653 break;
13f2b9d7
DJ
654 default:
655 return SR_ERR;
656 }
657 }
658
0a1f7b09 659 for (i = 1; i <= model->digital_pods; i++) {
082972e8
UH
660 if (state->digital_pods[i - 1] == pod_enabled[i - 1])
661 continue;
662 g_snprintf(command, sizeof(command),
663 (*model->scpi_dialect)[SCPI_CMD_SET_DIG_POD_STATE],
664 i, pod_enabled[i - 1]);
23f43dff 665 if (sr_scpi_send(scpi, command) != SR_OK)
082972e8
UH
666 return SR_ERR;
667 state->digital_pods[i - 1] = pod_enabled[i - 1];
23e1ea7a 668 setup_changed = TRUE;
13f2b9d7
DJ
669 }
670
671 g_free(pod_enabled);
672
23e1ea7a
DJ
673 if (setup_changed && hmo_update_sample_rate(sdi) != SR_OK)
674 return SR_ERR;
675
13f2b9d7
DJ
676 return SR_OK;
677}
678
695dc859 679static int dev_acquisition_start(const struct sr_dev_inst *sdi)
13f2b9d7
DJ
680{
681 GSList *l;
1b0f62df 682 gboolean digital_added[MAX_DIGITAL_GROUP_COUNT];
e06875b2 683 size_t group, pod_count;
ba7dd8bb 684 struct sr_channel *ch;
13f2b9d7 685 struct dev_context *devc;
23f43dff 686 struct sr_scpi_dev_inst *scpi;
40a75c8e 687 int ret;
06a3e78a 688
23f43dff 689 scpi = sdi->conn;
13f2b9d7 690 devc = sdi->priv;
13f2b9d7 691
1b0f62df
GS
692 /* Preset empty results. */
693 for (group = 0; group < ARRAY_SIZE(digital_added); group++)
694 digital_added[group] = FALSE;
db81fbb5
SA
695 g_slist_free(devc->enabled_channels);
696 devc->enabled_channels = NULL;
697
1b0f62df 698 /*
e06875b2
GS
699 * Contruct the list of enabled channels. Determine the highest
700 * number of digital pods involved in the acquisition.
1b0f62df 701 */
e06875b2 702 pod_count = 0;
ba7dd8bb
UH
703 for (l = sdi->channels; l; l = l->next) {
704 ch = l->data;
705 if (!ch->enabled)
082972e8 706 continue;
1b0f62df
GS
707 /* Only add a single digital channel per group (pod). */
708 group = ch->index / 8;
709 if (ch->type != SR_CHANNEL_LOGIC || !digital_added[group]) {
ba7dd8bb
UH
710 devc->enabled_channels = g_slist_append(
711 devc->enabled_channels, ch);
e06875b2 712 if (ch->type == SR_CHANNEL_LOGIC) {
1b0f62df 713 digital_added[group] = TRUE;
e06875b2
GS
714 if (pod_count < group + 1)
715 pod_count = group + 1;
716 }
13f2b9d7
DJ
717 }
718 }
ba7dd8bb 719 if (!devc->enabled_channels)
13f2b9d7 720 return SR_ERR;
e06875b2
GS
721 devc->pod_count = pod_count;
722 devc->logic_data = NULL;
13f2b9d7 723
1b0f62df
GS
724 /*
725 * Check constraints. Some channels can be either analog or
726 * digital, but not both at the same time.
727 */
ba7dd8bb
UH
728 if (hmo_check_channels(devc->enabled_channels) != SR_OK) {
729 sr_err("Invalid channel configuration specified!");
40a75c8e
GS
730 ret = SR_ERR_NA;
731 goto free_enabled;
13f2b9d7
DJ
732 }
733
1b0f62df
GS
734 /*
735 * Configure the analog and digital channels and the
736 * corresponding digital pods.
737 */
ba7dd8bb
UH
738 if (hmo_setup_channels(sdi) != SR_OK) {
739 sr_err("Failed to setup channel configuration!");
40a75c8e
GS
740 ret = SR_ERR;
741 goto free_enabled;
13f2b9d7
DJ
742 }
743
1b0f62df
GS
744 /*
745 * Start acquisition on the first enabled channel. The
746 * receive routine will continue driving the acquisition.
747 */
102f1239
BV
748 sr_scpi_source_add(sdi->session, scpi, G_IO_IN, 50,
749 hmo_receive_data, (void *)sdi);
13f2b9d7 750
bee2b016 751 std_session_send_df_header(sdi);
13f2b9d7 752
ba7dd8bb 753 devc->current_channel = devc->enabled_channels;
13f2b9d7
DJ
754
755 return hmo_request_data(sdi);
40a75c8e
GS
756
757free_enabled:
758 g_slist_free(devc->enabled_channels);
759 devc->enabled_channels = NULL;
760 return ret;
06a3e78a
DJ
761}
762
695dc859 763static int dev_acquisition_stop(struct sr_dev_inst *sdi)
06a3e78a 764{
13f2b9d7 765 struct dev_context *devc;
23f43dff 766 struct sr_scpi_dev_inst *scpi;
13f2b9d7 767
bee2b016 768 std_session_send_df_end(sdi);
66e3219d 769
13f2b9d7
DJ
770 devc = sdi->priv;
771
ef1a346b 772 devc->num_frames = 0;
ba7dd8bb
UH
773 g_slist_free(devc->enabled_channels);
774 devc->enabled_channels = NULL;
23f43dff 775 scpi = sdi->conn;
102f1239 776 sr_scpi_source_remove(sdi->session, scpi);
06a3e78a
DJ
777
778 return SR_OK;
779}
780
dd5c48a6 781static struct sr_dev_driver hameg_hmo_driver_info = {
06a3e78a 782 .name = "hameg-hmo",
89280b1a 783 .longname = "Hameg HMO",
06a3e78a 784 .api_version = 1,
c2fdcc25 785 .init = std_init,
700d6b64 786 .cleanup = std_cleanup,
06a3e78a 787 .scan = scan,
c01bf34c 788 .dev_list = std_dev_list,
06a3e78a
DJ
789 .dev_clear = dev_clear,
790 .config_get = config_get,
791 .config_set = config_set,
792 .config_list = config_list,
793 .dev_open = dev_open,
794 .dev_close = dev_close,
795 .dev_acquisition_start = dev_acquisition_start,
796 .dev_acquisition_stop = dev_acquisition_stop,
41812aca 797 .context = NULL,
06a3e78a 798};
dd5c48a6 799SR_REGISTER_DEV_DRIVER(hameg_hmo_driver_info);