| | 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 | |
| | 20 | #include <stdlib.h> |
| | 21 | #include <glib/gstdio.h> |
| | 22 | #include "protocol.h" |
| | 23 | |
| | 24 | #define SERIALCOMM "115200/8n1/flow=1" |
| | 25 | |
| | 26 | SR_PRIV struct sr_dev_driver hameg_hmo_driver_info; |
| | 27 | static struct sr_dev_driver *di = &hameg_hmo_driver_info; |
| | 28 | |
| | 29 | static const char *manufacturers[] = { |
| | 30 | "HAMEG", |
| | 31 | }; |
| | 32 | |
| | 33 | static const int32_t hwopts[] = { |
| | 34 | SR_CONF_CONN, |
| | 35 | SR_CONF_SERIALCOMM, |
| | 36 | }; |
| | 37 | |
| | 38 | struct usb_id_info { |
| | 39 | uint16_t vendor_id; |
| | 40 | uint16_t product_id; |
| | 41 | }; |
| | 42 | |
| | 43 | static struct usb_id_info ho_models[] = { |
| | 44 | { 0x0403, 0xed72 }, /* HO720 */ |
| | 45 | { 0x0403, 0xed73 }, /* HO730 */ |
| | 46 | }; |
| | 47 | |
| | 48 | enum { |
| | 49 | PG_INVALID = -1, |
| | 50 | PG_NONE, |
| | 51 | PG_ANALOG, |
| | 52 | PG_DIGITAL, |
| | 53 | }; |
| | 54 | |
| | 55 | static int init(struct sr_context *sr_ctx) |
| | 56 | { |
| | 57 | return std_init(sr_ctx, di, LOG_PREFIX); |
| | 58 | } |
| | 59 | |
| | 60 | /** |
| | 61 | * Find USB serial devices via the USB vendor ID and product ID. |
| | 62 | * |
| | 63 | * @param vendor_id Vendor ID of the USB device. |
| | 64 | * @param product_id Product ID of the USB device. |
| | 65 | * |
| | 66 | * @return A GSList of strings containing the path of the serial device or |
| | 67 | * NULL if no serial device is found. The returned list must be freed |
| | 68 | * by the caller. |
| | 69 | */ |
| | 70 | static GSList *auto_find_usb(uint16_t vendor_id, uint16_t product_id) |
| | 71 | { |
| | 72 | #ifdef __linux__ |
| | 73 | const gchar *usb_dev; |
| | 74 | const char device_tree[] = "/sys/bus/usb/devices/"; |
| | 75 | GDir *devices_dir, *device_dir; |
| | 76 | GSList *l = NULL; |
| | 77 | GSList *tty_devs; |
| | 78 | GSList *matched_paths; |
| | 79 | FILE *fd; |
| | 80 | char tmp[5]; |
| | 81 | gchar *vendor_path, *product_path, *path_copy; |
| | 82 | gchar *prefix, *subdir_path, *device_path, *tty_path; |
| | 83 | unsigned long read_vendor_id, read_product_id; |
| | 84 | const char *file; |
| | 85 | |
| | 86 | l = NULL; |
| | 87 | tty_devs = NULL; |
| | 88 | matched_paths = NULL; |
| | 89 | |
| | 90 | if (!(devices_dir = g_dir_open(device_tree, 0, NULL))) |
| | 91 | return NULL; |
| | 92 | |
| | 93 | /* |
| | 94 | * Find potential candidates using the vendor ID and product ID |
| | 95 | * and store them in matched_paths. |
| | 96 | */ |
| | 97 | while ((usb_dev = g_dir_read_name(devices_dir))) { |
| | 98 | vendor_path = g_strconcat(device_tree, |
| | 99 | usb_dev, "/idVendor", NULL); |
| | 100 | product_path = g_strconcat(device_tree, |
| | 101 | usb_dev, "/idProduct", NULL); |
| | 102 | |
| | 103 | if (!g_file_test(vendor_path, G_FILE_TEST_EXISTS) || |
| | 104 | !g_file_test(product_path, G_FILE_TEST_EXISTS)) |
| | 105 | goto skip_device; |
| | 106 | |
| | 107 | if ((fd = g_fopen(vendor_path, "r")) == NULL) |
| | 108 | goto skip_device; |
| | 109 | |
| | 110 | if (fgets(tmp, sizeof(tmp), fd) == NULL) { |
| | 111 | fclose(fd); |
| | 112 | goto skip_device; |
| | 113 | } |
| | 114 | read_vendor_id = strtoul(tmp, NULL, 16); |
| | 115 | |
| | 116 | fclose(fd); |
| | 117 | |
| | 118 | if ((fd = g_fopen(product_path, "r")) == NULL) |
| | 119 | goto skip_device; |
| | 120 | |
| | 121 | if (fgets(tmp, sizeof(tmp), fd) == NULL) { |
| | 122 | fclose(fd); |
| | 123 | goto skip_device; |
| | 124 | } |
| | 125 | read_product_id = strtoul(tmp, NULL, 16); |
| | 126 | |
| | 127 | fclose(fd); |
| | 128 | |
| | 129 | if (vendor_id == read_vendor_id && |
| | 130 | product_id == read_product_id) { |
| | 131 | path_copy = g_strdup(usb_dev); |
| | 132 | matched_paths = g_slist_prepend(matched_paths, |
| | 133 | path_copy); |
| | 134 | } |
| | 135 | |
| | 136 | skip_device: |
| | 137 | g_free(vendor_path); |
| | 138 | g_free(product_path); |
| | 139 | } |
| | 140 | g_dir_close(devices_dir); |
| | 141 | |
| | 142 | /* For every matched device try to find a ttyUSBX subfolder. */ |
| | 143 | for (l = matched_paths; l; l = l->next) { |
| | 144 | subdir_path = NULL; |
| | 145 | |
| | 146 | device_path = g_strconcat(device_tree, l->data, NULL); |
| | 147 | |
| | 148 | if (!(device_dir = g_dir_open(device_path, 0, NULL))) { |
| | 149 | g_free(device_path); |
| | 150 | continue; |
| | 151 | } |
| | 152 | |
| | 153 | prefix = g_strconcat(l->data, ":", NULL); |
| | 154 | |
| | 155 | while ((file = g_dir_read_name(device_dir))) { |
| | 156 | if (g_str_has_prefix(file, prefix)) { |
| | 157 | subdir_path = g_strconcat(device_path, |
| | 158 | "/", file, NULL); |
| | 159 | break; |
| | 160 | } |
| | 161 | } |
| | 162 | g_dir_close(device_dir); |
| | 163 | |
| | 164 | g_free(prefix); |
| | 165 | g_free(device_path); |
| | 166 | |
| | 167 | if (subdir_path) { |
| | 168 | if (!(device_dir = g_dir_open(subdir_path, 0, NULL))) { |
| | 169 | g_free(subdir_path); |
| | 170 | continue; |
| | 171 | } |
| | 172 | g_free(subdir_path); |
| | 173 | |
| | 174 | while ((file = g_dir_read_name(device_dir))) { |
| | 175 | if (g_str_has_prefix(file, "ttyUSB")) { |
| | 176 | tty_path = g_strconcat("/dev/", |
| | 177 | file, NULL); |
| | 178 | sr_dbg("Found USB device %04x:%04x attached to %s.", |
| | 179 | vendor_id, product_id, tty_path); |
| | 180 | tty_devs = g_slist_prepend(tty_devs, |
| | 181 | tty_path); |
| | 182 | break; |
| | 183 | } |
| | 184 | } |
| | 185 | g_dir_close(device_dir); |
| | 186 | } |
| | 187 | } |
| | 188 | g_slist_free_full(matched_paths, g_free); |
| | 189 | |
| | 190 | return tty_devs; |
| | 191 | #else |
| | 192 | return NULL; |
| | 193 | #endif |
| | 194 | } |
| | 195 | |
| | 196 | static int check_manufacturer(const char *manufacturer) |
| | 197 | { |
| | 198 | unsigned int i; |
| | 199 | |
| | 200 | for (i = 0; i < ARRAY_SIZE(manufacturers); ++i) |
| | 201 | if (!strcmp(manufacturer, manufacturers[i])) |
| | 202 | return SR_OK; |
| | 203 | |
| | 204 | return SR_ERR; |
| | 205 | } |
| | 206 | |
| | 207 | static struct sr_dev_inst *hmo_probe_serial_device(const char *serial_device, |
| | 208 | const char *serial_options) |
| | 209 | { |
| | 210 | struct sr_dev_inst *sdi; |
| | 211 | struct dev_context *devc; |
| | 212 | struct sr_scpi_hw_info *hw_info; |
| | 213 | struct sr_scpi_dev_inst *scpi; |
| | 214 | |
| | 215 | sdi = NULL; |
| | 216 | devc = NULL; |
| | 217 | scpi = NULL; |
| | 218 | hw_info = NULL; |
| | 219 | |
| | 220 | if (!(scpi = scpi_dev_inst_new(serial_device, serial_options))) |
| | 221 | goto fail; |
| | 222 | |
| | 223 | sr_info("Probing %s.", serial_device); |
| | 224 | if (sr_scpi_open(scpi) != SR_OK) |
| | 225 | goto fail; |
| | 226 | |
| | 227 | if (sr_scpi_get_hw_id(scpi, &hw_info) != SR_OK) { |
| | 228 | sr_info("Couldn't get IDN response."); |
| | 229 | goto fail; |
| | 230 | } |
| | 231 | |
| | 232 | if (check_manufacturer(hw_info->manufacturer) != SR_OK) |
| | 233 | goto fail; |
| | 234 | |
| | 235 | if (!(sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, |
| | 236 | hw_info->manufacturer, hw_info->model, |
| | 237 | hw_info->firmware_version))) { |
| | 238 | goto fail; |
| | 239 | } |
| | 240 | sr_scpi_hw_info_free(hw_info); |
| | 241 | hw_info = NULL; |
| | 242 | |
| | 243 | if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) |
| | 244 | goto fail; |
| | 245 | |
| | 246 | sdi->driver = di; |
| | 247 | sdi->priv = devc; |
| | 248 | sdi->inst_type = SR_INST_SCPI; |
| | 249 | sdi->conn = scpi; |
| | 250 | |
| | 251 | if (hmo_init_device(sdi) != SR_OK) |
| | 252 | goto fail; |
| | 253 | |
| | 254 | sr_scpi_close(sdi->conn); |
| | 255 | |
| | 256 | sdi->status = SR_ST_INACTIVE; |
| | 257 | |
| | 258 | return sdi; |
| | 259 | |
| | 260 | fail: |
| | 261 | if (hw_info) |
| | 262 | sr_scpi_hw_info_free(hw_info); |
| | 263 | if (scpi) |
| | 264 | sr_scpi_free(scpi); |
| | 265 | if (sdi) |
| | 266 | sr_dev_inst_free(sdi); |
| | 267 | if (devc) |
| | 268 | g_free(devc); |
| | 269 | |
| | 270 | return NULL; |
| | 271 | } |
| | 272 | |
| | 273 | static GSList *scan(GSList *options) |
| | 274 | { |
| | 275 | GSList *devices; |
| | 276 | struct drv_context *drvc; |
| | 277 | struct sr_dev_inst *sdi; |
| | 278 | const char *serial_device, *serial_options; |
| | 279 | GSList *l, *tty_devs; |
| | 280 | unsigned int i; |
| | 281 | |
| | 282 | serial_device = NULL; |
| | 283 | serial_options = SERIALCOMM; |
| | 284 | sdi = NULL; |
| | 285 | devices = NULL; |
| | 286 | drvc = di->priv; |
| | 287 | drvc->instances = NULL; |
| | 288 | |
| | 289 | if (sr_serial_extract_options(options, &serial_device, |
| | 290 | &serial_options) == SR_OK) { |
| | 291 | sdi = hmo_probe_serial_device(serial_device, serial_options); |
| | 292 | if (sdi != NULL) { |
| | 293 | devices = g_slist_append(devices, sdi); |
| | 294 | drvc->instances = g_slist_append(drvc->instances, sdi); |
| | 295 | } |
| | 296 | } else { |
| | 297 | tty_devs = NULL; |
| | 298 | |
| | 299 | for (i = 0; i < ARRAY_SIZE(ho_models); i++) { |
| | 300 | if ((l = auto_find_usb(ho_models[i].vendor_id, |
| | 301 | ho_models[i].product_id)) == NULL) |
| | 302 | continue; |
| | 303 | tty_devs = g_slist_concat(tty_devs, l); |
| | 304 | } |
| | 305 | |
| | 306 | for (l = tty_devs; l; l = l->next) { |
| | 307 | sdi = hmo_probe_serial_device(l->data, serial_options); |
| | 308 | if (sdi != NULL) { |
| | 309 | devices = g_slist_append(devices, sdi); |
| | 310 | drvc->instances = g_slist_append(drvc->instances, sdi); |
| | 311 | } |
| | 312 | } |
| | 313 | |
| | 314 | g_slist_free_full(tty_devs, g_free); |
| | 315 | } |
| | 316 | |
| | 317 | return devices; |
| | 318 | } |
| | 319 | |
| | 320 | static GSList *dev_list(void) |
| | 321 | { |
| | 322 | return ((struct drv_context *)(di->priv))->instances; |
| | 323 | } |
| | 324 | |
| | 325 | static void clear_helper(void *priv) |
| | 326 | { |
| | 327 | unsigned int i; |
| | 328 | struct dev_context *devc; |
| | 329 | struct scope_config *model; |
| | 330 | |
| | 331 | devc = priv; |
| | 332 | model = devc->model_config; |
| | 333 | |
| | 334 | hmo_scope_state_free(devc->model_state); |
| | 335 | |
| | 336 | for (i = 0; i < model->analog_channels; ++i) |
| | 337 | g_slist_free(devc->analog_groups[i].probes); |
| | 338 | |
| | 339 | for (i = 0; i < model->digital_pods; ++i) { |
| | 340 | g_slist_free(devc->digital_groups[i].probes); |
| | 341 | g_free(devc->digital_groups[i].name); |
| | 342 | } |
| | 343 | |
| | 344 | g_free(devc->analog_groups); |
| | 345 | g_free(devc->digital_groups); |
| | 346 | |
| | 347 | g_free(devc); |
| | 348 | } |
| | 349 | |
| | 350 | static int dev_clear(void) |
| | 351 | { |
| | 352 | return std_dev_clear(di, clear_helper); |
| | 353 | } |
| | 354 | |
| | 355 | static int dev_open(struct sr_dev_inst *sdi) |
| | 356 | { |
| | 357 | if (sdi->status != SR_ST_ACTIVE && sr_scpi_open(sdi->conn) != SR_OK) |
| | 358 | return SR_ERR; |
| | 359 | |
| | 360 | if (hmo_scope_state_get(sdi) != SR_OK) |
| | 361 | return SR_ERR; |
| | 362 | |
| | 363 | sdi->status = SR_ST_ACTIVE; |
| | 364 | |
| | 365 | return SR_OK; |
| | 366 | } |
| | 367 | |
| | 368 | static int dev_close(struct sr_dev_inst *sdi) |
| | 369 | { |
| | 370 | if (sdi->status == SR_ST_INACTIVE) |
| | 371 | return SR_OK; |
| | 372 | |
| | 373 | sr_scpi_close(sdi->conn); |
| | 374 | |
| | 375 | sdi->status = SR_ST_INACTIVE; |
| | 376 | |
| | 377 | return SR_OK; |
| | 378 | } |
| | 379 | |
| | 380 | static int cleanup(void) |
| | 381 | { |
| | 382 | dev_clear(); |
| | 383 | |
| | 384 | return SR_OK; |
| | 385 | } |
| | 386 | |
| | 387 | static int check_probe_group(struct dev_context *devc, |
| | 388 | const struct sr_probe_group *probe_group) |
| | 389 | { |
| | 390 | unsigned int i; |
| | 391 | struct scope_config *model; |
| | 392 | |
| | 393 | model = devc->model_config; |
| | 394 | |
| | 395 | if (!probe_group) |
| | 396 | return PG_NONE; |
| | 397 | |
| | 398 | for (i = 0; i < model->analog_channels; ++i) |
| | 399 | if (probe_group == &devc->analog_groups[i]) |
| | 400 | return PG_ANALOG; |
| | 401 | |
| | 402 | for (i = 0; i < model->digital_pods; ++i) |
| | 403 | if (probe_group == &devc->digital_groups[i]) |
| | 404 | return PG_DIGITAL; |
| | 405 | |
| | 406 | sr_err("Invalid probe group specified."); |
| | 407 | |
| | 408 | return PG_INVALID; |
| | 409 | } |
| | 410 | |
| | 411 | static int config_get(int key, GVariant **data, const struct sr_dev_inst *sdi, |
| | 412 | const struct sr_probe_group *probe_group) |
| | 413 | { |
| | 414 | int ret, pg_type; |
| | 415 | unsigned int i; |
| | 416 | struct dev_context *devc; |
| | 417 | struct scope_config *model; |
| | 418 | struct scope_state *state; |
| | 419 | |
| | 420 | if (!sdi || !(devc = sdi->priv)) |
| | 421 | return SR_ERR_ARG; |
| | 422 | |
| | 423 | if ((pg_type = check_probe_group(devc, probe_group)) == PG_INVALID) |
| | 424 | return SR_ERR; |
| | 425 | |
| | 426 | ret = SR_ERR_NA; |
| | 427 | model = devc->model_config; |
| | 428 | state = devc->model_state; |
| | 429 | |
| | 430 | switch (key) { |
| | 431 | case SR_CONF_NUM_TIMEBASE: |
| | 432 | *data = g_variant_new_int32(model->num_xdivs); |
| | 433 | ret = SR_OK; |
| | 434 | break; |
| | 435 | case SR_CONF_NUM_VDIV: |
| | 436 | if (pg_type == PG_NONE) { |
| | 437 | sr_err("No probe group specified."); |
| | 438 | return SR_ERR_PROBE_GROUP; |
| | 439 | } else if (pg_type == PG_ANALOG) { |
| | 440 | for (i = 0; i < model->analog_channels; ++i) { |
| | 441 | if (probe_group != &devc->analog_groups[i]) |
| | 442 | continue; |
| | 443 | *data = g_variant_new_int32(model->num_ydivs); |
| | 444 | ret = SR_OK; |
| | 445 | break; |
| | 446 | } |
| | 447 | |
| | 448 | } else { |
| | 449 | ret = SR_ERR_NA; |
| | 450 | } |
| | 451 | break; |
| | 452 | case SR_CONF_SAMPLERATE: |
| | 453 | *data = g_variant_new_uint64(state->sample_rate); |
| | 454 | ret = SR_OK; |
| | 455 | break; |
| | 456 | default: |
| | 457 | ret = SR_ERR_NA; |
| | 458 | } |
| | 459 | |
| | 460 | return ret; |
| | 461 | } |
| | 462 | |
| | 463 | static GVariant *build_tuples(const uint64_t (*array)[][2], unsigned int n) |
| | 464 | { |
| | 465 | unsigned int i; |
| | 466 | GVariant *rational[2]; |
| | 467 | GVariantBuilder gvb; |
| | 468 | |
| | 469 | g_variant_builder_init(&gvb, G_VARIANT_TYPE_ARRAY); |
| | 470 | |
| | 471 | for (i = 0; i < n; i++) { |
| | 472 | rational[0] = g_variant_new_uint64((*array)[i][0]); |
| | 473 | rational[1] = g_variant_new_uint64((*array)[i][1]); |
| | 474 | |
| | 475 | /* FIXME: Valgrind reports a memory leak here. */ |
| | 476 | g_variant_builder_add_value(&gvb, g_variant_new_tuple(rational, 2)); |
| | 477 | } |
| | 478 | |
| | 479 | return g_variant_builder_end(&gvb); |
| | 480 | } |
| | 481 | |
| | 482 | static int config_set(int key, GVariant *data, const struct sr_dev_inst *sdi, |
| | 483 | const struct sr_probe_group *probe_group) |
| | 484 | { |
| | 485 | int ret, pg_type; |
| | 486 | unsigned int i, j; |
| | 487 | char command[MAX_COMMAND_SIZE], float_str[30]; |
| | 488 | struct dev_context *devc; |
| | 489 | struct scope_config *model; |
| | 490 | struct scope_state *state; |
| | 491 | const char *tmp; |
| | 492 | uint64_t p, q, tmp_u64; |
| | 493 | double tmp_d; |
| | 494 | gboolean update_sample_rate; |
| | 495 | |
| | 496 | if (!sdi || !(devc = sdi->priv)) |
| | 497 | return SR_ERR_ARG; |
| | 498 | |
| | 499 | if ((pg_type = check_probe_group(devc, probe_group)) == PG_INVALID) |
| | 500 | return SR_ERR; |
| | 501 | |
| | 502 | model = devc->model_config; |
| | 503 | state = devc->model_state; |
| | 504 | update_sample_rate = FALSE; |
| | 505 | |
| | 506 | ret = SR_ERR_NA; |
| | 507 | |
| | 508 | switch (key) { |
| | 509 | case SR_CONF_LIMIT_FRAMES: |
| | 510 | devc->frame_limit = g_variant_get_uint64(data); |
| | 511 | ret = SR_OK; |
| | 512 | break; |
| | 513 | case SR_CONF_TRIGGER_SOURCE: |
| | 514 | tmp = g_variant_get_string(data, NULL); |
| | 515 | for (i = 0; (*model->trigger_sources)[i]; i++) { |
| | 516 | if (g_strcmp0(tmp, (*model->trigger_sources)[i]) != 0) |
| | 517 | continue; |
| | 518 | state->trigger_source = i; |
| | 519 | g_snprintf(command, sizeof(command), |
| | 520 | (*model->scpi_dialect)[SCPI_CMD_SET_TRIGGER_SOURCE], |
| | 521 | (*model->trigger_sources)[i]); |
| | 522 | |
| | 523 | ret = sr_scpi_send(sdi->conn, command); |
| | 524 | break; |
| | 525 | } |
| | 526 | break; |
| | 527 | case SR_CONF_VDIV: |
| | 528 | if (pg_type == PG_NONE) { |
| | 529 | sr_err("No probe group specified."); |
| | 530 | return SR_ERR_PROBE_GROUP; |
| | 531 | } |
| | 532 | |
| | 533 | g_variant_get(data, "(tt)", &p, &q); |
| | 534 | |
| | 535 | for (i = 0; i < model->num_vdivs; i++) { |
| | 536 | if (p != (*model->vdivs)[i][0] || |
| | 537 | q != (*model->vdivs)[i][1]) |
| | 538 | continue; |
| | 539 | for (j = 1; j <= model->analog_channels; ++j) { |
| | 540 | if (probe_group != &devc->analog_groups[j - 1]) |
| | 541 | continue; |
| | 542 | state->analog_channels[j - 1].vdiv = i; |
| | 543 | g_ascii_formatd(float_str, sizeof(float_str), "%E", (float) p / q); |
| | 544 | g_snprintf(command, sizeof(command), |
| | 545 | (*model->scpi_dialect)[SCPI_CMD_SET_VERTICAL_DIV], |
| | 546 | j, float_str); |
| | 547 | |
| | 548 | if (sr_scpi_send(sdi->conn, command) != SR_OK || |
| | 549 | sr_scpi_get_opc(sdi->conn) != SR_OK) |
| | 550 | return SR_ERR; |
| | 551 | |
| | 552 | break; |
| | 553 | } |
| | 554 | |
| | 555 | ret = SR_OK; |
| | 556 | break; |
| | 557 | } |
| | 558 | break; |
| | 559 | case SR_CONF_TIMEBASE: |
| | 560 | g_variant_get(data, "(tt)", &p, &q); |
| | 561 | |
| | 562 | for (i = 0; i < model->num_timebases; i++) { |
| | 563 | if (p != (*model->timebases)[i][0] || |
| | 564 | q != (*model->timebases)[i][1]) |
| | 565 | continue; |
| | 566 | state->timebase = i; |
| | 567 | g_ascii_formatd(float_str, sizeof(float_str), "%E", (float) p / q); |
| | 568 | g_snprintf(command, sizeof(command), |
| | 569 | (*model->scpi_dialect)[SCPI_CMD_SET_TIMEBASE], |
| | 570 | float_str); |
| | 571 | |
| | 572 | ret = sr_scpi_send(sdi->conn, command); |
| | 573 | update_sample_rate = TRUE; |
| | 574 | break; |
| | 575 | } |
| | 576 | break; |
| | 577 | case SR_CONF_HORIZ_TRIGGERPOS: |
| | 578 | tmp_d = g_variant_get_double(data); |
| | 579 | |
| | 580 | if (tmp_d < 0.0 || tmp_d > 1.0) |
| | 581 | return SR_ERR; |
| | 582 | |
| | 583 | state->horiz_triggerpos = -(tmp_d - 0.5) * state->timebase * model->num_xdivs; |
| | 584 | g_snprintf(command, sizeof(command), |
| | 585 | (*model->scpi_dialect)[SCPI_CMD_SET_HORIZ_TRIGGERPOS], |
| | 586 | state->horiz_triggerpos); |
| | 587 | |
| | 588 | ret = sr_scpi_send(sdi->conn, command); |
| | 589 | break; |
| | 590 | case SR_CONF_TRIGGER_SLOPE: |
| | 591 | tmp_u64 = g_variant_get_uint64(data); |
| | 592 | |
| | 593 | if (tmp_u64 != 0 && tmp_u64 != 1) |
| | 594 | return SR_ERR; |
| | 595 | |
| | 596 | state->trigger_slope = tmp_u64; |
| | 597 | |
| | 598 | g_snprintf(command, sizeof(command), |
| | 599 | (*model->scpi_dialect)[SCPI_CMD_SET_TRIGGER_SLOPE], |
| | 600 | tmp_u64 ? "POS" : "NEG"); |
| | 601 | |
| | 602 | ret = sr_scpi_send(sdi->conn, command); |
| | 603 | break; |
| | 604 | case SR_CONF_COUPLING: |
| | 605 | if (pg_type == PG_NONE) { |
| | 606 | sr_err("No probe group specified."); |
| | 607 | return SR_ERR_PROBE_GROUP; |
| | 608 | } |
| | 609 | |
| | 610 | tmp = g_variant_get_string(data, NULL); |
| | 611 | |
| | 612 | for (i = 0; (*model->coupling_options)[i]; i++) { |
| | 613 | if (strcmp(tmp, (*model->coupling_options)[i]) != 0) |
| | 614 | continue; |
| | 615 | for (j = 1; j <= model->analog_channels; ++j) { |
| | 616 | if (probe_group != &devc->analog_groups[j - 1]) |
| | 617 | continue; |
| | 618 | state->analog_channels[j-1].coupling = i; |
| | 619 | |
| | 620 | g_snprintf(command, sizeof(command), |
| | 621 | (*model->scpi_dialect)[SCPI_CMD_SET_COUPLING], |
| | 622 | j, tmp); |
| | 623 | |
| | 624 | if (sr_scpi_send(sdi->conn, command) != SR_OK || |
| | 625 | sr_scpi_get_opc(sdi->conn) != SR_OK) |
| | 626 | return SR_ERR; |
| | 627 | break; |
| | 628 | } |
| | 629 | |
| | 630 | ret = SR_OK; |
| | 631 | break; |
| | 632 | } |
| | 633 | break; |
| | 634 | default: |
| | 635 | ret = SR_ERR_NA; |
| | 636 | break; |
| | 637 | } |
| | 638 | |
| | 639 | if (ret == SR_OK) |
| | 640 | ret = sr_scpi_get_opc(sdi->conn); |
| | 641 | |
| | 642 | if (ret == SR_OK && update_sample_rate) |
| | 643 | ret = hmo_update_sample_rate(sdi); |
| | 644 | |
| | 645 | return ret; |
| | 646 | } |
| | 647 | |
| | 648 | static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi, |
| | 649 | const struct sr_probe_group *probe_group) |
| | 650 | { |
| | 651 | int pg_type; |
| | 652 | struct dev_context *devc; |
| | 653 | struct scope_config *model; |
| | 654 | |
| | 655 | if (!sdi || !(devc = sdi->priv)) |
| | 656 | return SR_ERR_ARG; |
| | 657 | |
| | 658 | if ((pg_type = check_probe_group(devc, probe_group)) == PG_INVALID) |
| | 659 | return SR_ERR; |
| | 660 | |
| | 661 | model = devc->model_config; |
| | 662 | |
| | 663 | switch (key) { |
| | 664 | case SR_CONF_DEVICE_OPTIONS: |
| | 665 | if (pg_type == PG_NONE) { |
| | 666 | *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32, |
| | 667 | model->hw_caps, model->num_hwcaps, |
| | 668 | sizeof(int32_t)); |
| | 669 | } else if (pg_type == PG_ANALOG) { |
| | 670 | *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32, |
| | 671 | model->analog_hwcaps, model->num_analog_hwcaps, |
| | 672 | sizeof(int32_t)); |
| | 673 | } else { |
| | 674 | *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32, |
| | 675 | NULL, 0, sizeof(int32_t)); |
| | 676 | } |
| | 677 | break; |
| | 678 | case SR_CONF_COUPLING: |
| | 679 | if (pg_type == PG_NONE) |
| | 680 | return SR_ERR_PROBE_GROUP; |
| | 681 | *data = g_variant_new_strv(*model->coupling_options, |
| | 682 | g_strv_length((char **)*model->coupling_options)); |
| | 683 | break; |
| | 684 | case SR_CONF_TRIGGER_SOURCE: |
| | 685 | *data = g_variant_new_strv(*model->trigger_sources, |
| | 686 | g_strv_length((char **)*model->trigger_sources)); |
| | 687 | break; |
| | 688 | case SR_CONF_TIMEBASE: |
| | 689 | *data = build_tuples(model->timebases, model->num_timebases); |
| | 690 | break; |
| | 691 | case SR_CONF_VDIV: |
| | 692 | if (pg_type == PG_NONE) |
| | 693 | return SR_ERR_PROBE_GROUP; |
| | 694 | *data = build_tuples(model->vdivs, model->num_vdivs); |
| | 695 | break; |
| | 696 | default: |
| | 697 | return SR_ERR_NA; |
| | 698 | } |
| | 699 | |
| | 700 | return SR_OK; |
| | 701 | } |
| | 702 | |
| | 703 | SR_PRIV int hmo_request_data(const struct sr_dev_inst *sdi) |
| | 704 | { |
| | 705 | char command[MAX_COMMAND_SIZE]; |
| | 706 | struct sr_probe *probe; |
| | 707 | struct dev_context *devc; |
| | 708 | struct scope_config *model; |
| | 709 | |
| | 710 | devc = sdi->priv; |
| | 711 | model = devc->model_config; |
| | 712 | |
| | 713 | probe = devc->current_probe->data; |
| | 714 | |
| | 715 | switch (probe->type) { |
| | 716 | case SR_PROBE_ANALOG: |
| | 717 | g_snprintf(command, sizeof(command), |
| | 718 | (*model->scpi_dialect)[SCPI_CMD_GET_ANALOG_DATA], |
| | 719 | probe->index + 1); |
| | 720 | break; |
| | 721 | case SR_PROBE_LOGIC: |
| | 722 | g_snprintf(command, sizeof(command), |
| | 723 | (*model->scpi_dialect)[SCPI_CMD_GET_DIG_DATA], |
| | 724 | probe->index < 8 ? 1 : 2); |
| | 725 | break; |
| | 726 | default: |
| | 727 | sr_err("Invalid probe type."); |
| | 728 | break; |
| | 729 | } |
| | 730 | |
| | 731 | return sr_scpi_send(sdi->conn, command); |
| | 732 | } |
| | 733 | |
| | 734 | static int hmo_check_probes(GSList *probes) |
| | 735 | { |
| | 736 | GSList *l; |
| | 737 | struct sr_probe *probe; |
| | 738 | gboolean enabled_pod1, enabled_pod2, enabled_chan3, enabled_chan4; |
| | 739 | |
| | 740 | enabled_pod1 = enabled_pod2 = enabled_chan3 = enabled_chan4 = FALSE; |
| | 741 | |
| | 742 | for (l = probes; l; l = l->next) { |
| | 743 | probe = l->data; |
| | 744 | switch (probe->type) { |
| | 745 | case SR_PROBE_ANALOG: |
| | 746 | if (probe->index == 2) |
| | 747 | enabled_chan3 = TRUE; |
| | 748 | else if (probe->index == 3) |
| | 749 | enabled_chan4 = TRUE; |
| | 750 | break; |
| | 751 | case SR_PROBE_LOGIC: |
| | 752 | if (probe->index < 8) |
| | 753 | enabled_pod1 = TRUE; |
| | 754 | else |
| | 755 | enabled_pod2 = TRUE; |
| | 756 | break; |
| | 757 | default: |
| | 758 | return SR_ERR; |
| | 759 | } |
| | 760 | } |
| | 761 | |
| | 762 | if ((enabled_pod1 && enabled_chan3) || |
| | 763 | (enabled_pod2 && enabled_chan4)) |
| | 764 | return SR_ERR; |
| | 765 | |
| | 766 | return SR_OK; |
| | 767 | } |
| | 768 | |
| | 769 | static int hmo_setup_probes(const struct sr_dev_inst *sdi) |
| | 770 | { |
| | 771 | GSList *l; |
| | 772 | unsigned int i; |
| | 773 | gboolean *pod_enabled, setup_changed; |
| | 774 | char command[MAX_COMMAND_SIZE]; |
| | 775 | struct scope_state *state; |
| | 776 | struct scope_config *model; |
| | 777 | struct sr_probe *probe; |
| | 778 | struct dev_context *devc; |
| | 779 | struct sr_scpi_dev_inst *scpi; |
| | 780 | |
| | 781 | devc = sdi->priv; |
| | 782 | scpi = sdi->conn; |
| | 783 | state = devc->model_state; |
| | 784 | model = devc->model_config; |
| | 785 | setup_changed = FALSE; |
| | 786 | |
| | 787 | pod_enabled = g_try_malloc0(sizeof(gboolean) * model->digital_pods); |
| | 788 | |
| | 789 | for (l = sdi->probes; l; l = l->next) { |
| | 790 | probe = l->data; |
| | 791 | switch (probe->type) { |
| | 792 | case SR_PROBE_ANALOG: |
| | 793 | if (probe->enabled == state->analog_channels[probe->index].state) |
| | 794 | break; |
| | 795 | g_snprintf(command, sizeof(command), |
| | 796 | (*model->scpi_dialect)[SCPI_CMD_SET_ANALOG_CHAN_STATE], |
| | 797 | probe->index + 1, probe->enabled); |
| | 798 | |
| | 799 | if (sr_scpi_send(scpi, command) != SR_OK) |
| | 800 | return SR_ERR; |
| | 801 | state->analog_channels[probe->index].state = probe->enabled; |
| | 802 | setup_changed = TRUE; |
| | 803 | break; |
| | 804 | case SR_PROBE_LOGIC: |
| | 805 | /* |
| | 806 | * A digital POD needs to be enabled for every group of |
| | 807 | * 8 probes. |
| | 808 | */ |
| | 809 | if (probe->enabled) |
| | 810 | pod_enabled[probe->index < 8 ? 0 : 1] = TRUE; |
| | 811 | |
| | 812 | if (probe->enabled == state->digital_channels[probe->index]) |
| | 813 | break; |
| | 814 | g_snprintf(command, sizeof(command), |
| | 815 | (*model->scpi_dialect)[SCPI_CMD_SET_DIG_CHAN_STATE], |
| | 816 | probe->index, probe->enabled); |
| | 817 | |
| | 818 | if (sr_scpi_send(scpi, command) != SR_OK) |
| | 819 | return SR_ERR; |
| | 820 | |
| | 821 | state->digital_channels[probe->index] = probe->enabled; |
| | 822 | setup_changed = TRUE; |
| | 823 | break; |
| | 824 | default: |
| | 825 | return SR_ERR; |
| | 826 | } |
| | 827 | } |
| | 828 | |
| | 829 | for (i = 1; i <= model->digital_pods; ++i) { |
| | 830 | if (state->digital_pods[i - 1] == pod_enabled[i - 1]) |
| | 831 | continue; |
| | 832 | g_snprintf(command, sizeof(command), |
| | 833 | (*model->scpi_dialect)[SCPI_CMD_SET_DIG_POD_STATE], |
| | 834 | i, pod_enabled[i - 1]); |
| | 835 | if (sr_scpi_send(scpi, command) != SR_OK) |
| | 836 | return SR_ERR; |
| | 837 | state->digital_pods[i - 1] = pod_enabled[i - 1]; |
| | 838 | setup_changed = TRUE; |
| | 839 | } |
| | 840 | |
| | 841 | g_free(pod_enabled); |
| | 842 | |
| | 843 | if (setup_changed && hmo_update_sample_rate(sdi) != SR_OK) |
| | 844 | return SR_ERR; |
| | 845 | |
| | 846 | return SR_OK; |
| | 847 | } |
| | 848 | |
| | 849 | static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data) |
| | 850 | { |
| | 851 | GSList *l; |
| | 852 | gboolean digital_added; |
| | 853 | struct sr_probe *probe; |
| | 854 | struct dev_context *devc; |
| | 855 | struct sr_scpi_dev_inst *scpi; |
| | 856 | |
| | 857 | if (sdi->status != SR_ST_ACTIVE) |
| | 858 | return SR_ERR_DEV_CLOSED; |
| | 859 | |
| | 860 | scpi = sdi->conn; |
| | 861 | devc = sdi->priv; |
| | 862 | digital_added = FALSE; |
| | 863 | |
| | 864 | for (l = sdi->probes; l; l = l->next) { |
| | 865 | probe = l->data; |
| | 866 | if (!probe->enabled) |
| | 867 | continue; |
| | 868 | /* Only add a single digital probe. */ |
| | 869 | if (probe->type != SR_PROBE_LOGIC || !digital_added) { |
| | 870 | devc->enabled_probes = g_slist_append( |
| | 871 | devc->enabled_probes, probe); |
| | 872 | if (probe->type == SR_PROBE_LOGIC) |
| | 873 | digital_added = TRUE; |
| | 874 | } |
| | 875 | } |
| | 876 | |
| | 877 | if (!devc->enabled_probes) |
| | 878 | return SR_ERR; |
| | 879 | |
| | 880 | if (hmo_check_probes(devc->enabled_probes) != SR_OK) { |
| | 881 | sr_err("Invalid probe configuration specified!"); |
| | 882 | return SR_ERR_NA; |
| | 883 | } |
| | 884 | |
| | 885 | if (hmo_setup_probes(sdi) != SR_OK) { |
| | 886 | sr_err("Failed to setup probe configuration!"); |
| | 887 | return SR_ERR; |
| | 888 | } |
| | 889 | |
| | 890 | sr_scpi_source_add(scpi, G_IO_IN, 50, hmo_receive_data, (void *)sdi); |
| | 891 | |
| | 892 | /* Send header packet to the session bus. */ |
| | 893 | std_session_send_df_header(cb_data, LOG_PREFIX); |
| | 894 | |
| | 895 | devc->current_probe = devc->enabled_probes; |
| | 896 | |
| | 897 | return hmo_request_data(sdi); |
| | 898 | } |
| | 899 | |
| | 900 | static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data) |
| | 901 | { |
| | 902 | struct dev_context *devc; |
| | 903 | struct sr_scpi_dev_inst *scpi; |
| | 904 | struct sr_datafeed_packet packet; |
| | 905 | |
| | 906 | (void)cb_data; |
| | 907 | |
| | 908 | packet.type = SR_DF_END; |
| | 909 | packet.payload = NULL; |
| | 910 | sr_session_send(sdi, &packet); |
| | 911 | |
| | 912 | if (sdi->status != SR_ST_ACTIVE) |
| | 913 | return SR_ERR_DEV_CLOSED; |
| | 914 | |
| | 915 | devc = sdi->priv; |
| | 916 | |
| | 917 | g_slist_free(devc->enabled_probes); |
| | 918 | devc->enabled_probes = NULL; |
| | 919 | scpi = sdi->conn; |
| | 920 | sr_scpi_source_remove(scpi); |
| | 921 | |
| | 922 | return SR_OK; |
| | 923 | } |
| | 924 | |
| | 925 | SR_PRIV struct sr_dev_driver hameg_hmo_driver_info = { |
| | 926 | .name = "hameg-hmo", |
| | 927 | .longname = "Hameg HMO", |
| | 928 | .api_version = 1, |
| | 929 | .init = init, |
| | 930 | .cleanup = cleanup, |
| | 931 | .scan = scan, |
| | 932 | .dev_list = dev_list, |
| | 933 | .dev_clear = dev_clear, |
| | 934 | .config_get = config_get, |
| | 935 | .config_set = config_set, |
| | 936 | .config_list = config_list, |
| | 937 | .dev_open = dev_open, |
| | 938 | .dev_close = dev_close, |
| | 939 | .dev_acquisition_start = dev_acquisition_start, |
| | 940 | .dev_acquisition_stop = dev_acquisition_stop, |
| | 941 | .priv = NULL, |
| | 942 | }; |
projects / libsigrok.git / blame_incremental