X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=src%2Fhardware%2Fbaylibre-acme%2Fapi.c;h=cdf9968b20b94ad33217ebfa1c6f8ff93849d8be;hb=88a0265ebcb265ba839c02cc5bcd39e359c9f60f;hp=b13e2e08c97290ae76cd1b45335eedd80a70feed;hpb=dfaee1de1711cdfba7bd780b448188819691a2db;p=libsigrok.git diff --git a/src/hardware/baylibre-acme/api.c b/src/hardware/baylibre-acme/api.c index b13e2e08..cdf9968b 100644 --- a/src/hardware/baylibre-acme/api.c +++ b/src/hardware/baylibre-acme/api.c @@ -17,49 +17,140 @@ * along with this program. If not, see . */ +#include #include "protocol.h" +#include +#include SR_PRIV struct sr_dev_driver baylibre_acme_driver_info; -static struct sr_dev_driver *di = &baylibre_acme_driver_info; -static int init(struct sr_context *sr_ctx) +static const uint32_t devopts[] = { + SR_CONF_CONTINUOUS | SR_CONF_SET, + SR_CONF_LIMIT_SAMPLES | SR_CONF_GET | SR_CONF_SET, + SR_CONF_LIMIT_MSEC | SR_CONF_GET | SR_CONF_SET, + SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST, +}; + +/* + * Currently there are two channel-group/probe options for ACME: + * - SR_CONF_PROBE_FACTOR - allows to modify current shunt resistance + * calibration + * - SR_CONF_POWER_OFF - allows to remotely cut-off/restore power to + * measured devices + * + * They are not static - we have to check each probe's capabilities in + * config_list(). + */ +#define MAX_DEVOPTS_CG 2 +#define HAS_PROBE_FACTOR (SR_CONF_PROBE_FACTOR | SR_CONF_GET | SR_CONF_SET) +#define HAS_POWER_OFF (SR_CONF_POWER_OFF | SR_CONF_GET | SR_CONF_SET) + +#define MAX_SAMPLE_RATE 500 /* In Hz */ + +static const uint64_t samplerates[] = { + SR_HZ(1), + SR_HZ(MAX_SAMPLE_RATE), + SR_HZ(1), +}; + +static int init(struct sr_dev_driver *di, struct sr_context *sr_ctx) { return std_init(sr_ctx, di, LOG_PREFIX); } -static GSList *scan(GSList *options) +static GSList *scan(struct sr_dev_driver *di, GSList *options) { struct drv_context *drvc; + struct dev_context *devc; + struct sr_dev_inst *sdi; GSList *devices; + gboolean status; + int i; (void)options; + drvc = di->context; devices = NULL; - drvc = di->priv; - drvc->instances = NULL; - /* TODO: scan for devices, either based on a SR_CONF_CONN option - * or on a USB scan. */ + devc = g_malloc0(sizeof(struct dev_context)); + devc->samplerate = SR_HZ(10); + + sdi = g_malloc0(sizeof(struct sr_dev_inst)); + sdi->status = SR_ST_INACTIVE; + sdi->vendor = g_strdup("BayLibre"); + sdi->model = g_strdup("ACME"); + sdi->driver = di; + sdi->priv = devc; + + status = bl_acme_is_sane(); + if (!status) + goto err_out; + + /* + * Iterate over all ACME connectors and check if any probes + * are present. + */ + for (i = 0; i < MAX_PROBES; i++) { + /* + * First check if there's an energy probe on this connector. If + * not, and we're already at the fifth probe - see if we can + * detect a temperature probe. + */ + status = bl_acme_detect_probe(bl_acme_get_enrg_addr(i), + PROBE_NUM(i), ENRG_PROBE_NAME); + if (status) { + /* Energy probe detected. */ + status = bl_acme_register_probe(sdi, PROBE_ENRG, + bl_acme_get_enrg_addr(i), PROBE_NUM(i)); + if (!status) { + sr_err("Error registering power probe %d", + PROBE_NUM(i)); + continue; + } + } else if (i >= TEMP_PRB_START_INDEX) { + status = bl_acme_detect_probe(bl_acme_get_temp_addr(i), + PROBE_NUM(i), TEMP_PROBE_NAME); + if (status) { + /* Temperature probe detected. */ + status = bl_acme_register_probe(sdi,PROBE_TEMP, + bl_acme_get_temp_addr(i), PROBE_NUM(i)); + if (!status) { + sr_err("Error registering temp " + "probe %d", PROBE_NUM(i)); + continue; + } + } + } + } + + /* + * Let's assume there's no ACME device present if no probe + * has been registered. + */ + if (!sdi->channel_groups) + goto err_out; + + devices = g_slist_append(devices, sdi); + drvc->instances = g_slist_append(drvc->instances, sdi); return devices; -} -static GSList *dev_list(void) -{ - return ((struct drv_context *)(di->priv))->instances; +err_out: + g_free(devc); + sr_dev_inst_free(sdi); + + return NULL; } -static int dev_clear(void) +static GSList *dev_list(const struct sr_dev_driver *di) { - return std_dev_clear(di, NULL); + return ((struct drv_context *)(di->context))->instances; } static int dev_open(struct sr_dev_inst *sdi) { (void)sdi; - /* TODO: get handle from sdi->conn and open it. */ - sdi->status = SR_ST_ACTIVE; return SR_OK; @@ -69,34 +160,47 @@ static int dev_close(struct sr_dev_inst *sdi) { (void)sdi; - /* TODO: get handle from sdi->conn and close it. */ - sdi->status = SR_ST_INACTIVE; return SR_OK; } -static int cleanup(void) -{ - dev_clear(); - - /* TODO: free other driver resources, if any. */ - - return SR_OK; -} - -static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi, - const struct sr_channel_group *cg) +static int config_get(uint32_t key, GVariant **data, + const struct sr_dev_inst *sdi, + const struct sr_channel_group *cg) { + struct dev_context *devc; int ret; + uint64_t shunt; + gboolean power_off; - (void)sdi; - (void)data; - (void)cg; + devc = sdi->priv; ret = SR_OK; switch (key) { - /* TODO */ + case SR_CONF_LIMIT_SAMPLES: + *data = g_variant_new_uint64(devc->limit_samples); + break; + case SR_CONF_LIMIT_MSEC: + *data = g_variant_new_uint64(devc->limit_msec); + break; + case SR_CONF_SAMPLERATE: + *data = g_variant_new_uint64(devc->samplerate); + break; + case SR_CONF_PROBE_FACTOR: + if (!cg) + return SR_ERR_CHANNEL_GROUP; + ret = bl_acme_get_shunt(cg, &shunt); + if (ret == SR_OK) + *data = g_variant_new_uint64(shunt); + break; + case SR_CONF_POWER_OFF: + if (!cg) + return SR_ERR_CHANNEL_GROUP; + ret = bl_acme_read_power_state(cg, &power_off); + if (ret == SR_OK) + *data = g_variant_new_boolean(power_off); + break; default: return SR_ERR_NA; } @@ -104,20 +208,49 @@ static int config_get(uint32_t key, GVariant **data, const struct sr_dev_inst *s return ret; } -static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sdi, - const struct sr_channel_group *cg) +static int config_set(uint32_t key, GVariant *data, + const struct sr_dev_inst *sdi, + const struct sr_channel_group *cg) { + struct dev_context *devc; + uint64_t samplerate; int ret; - (void)data; - (void)cg; - if (sdi->status != SR_ST_ACTIVE) return SR_ERR_DEV_CLOSED; + devc = sdi->priv; + ret = SR_OK; switch (key) { - /* TODO */ + case SR_CONF_LIMIT_SAMPLES: + devc->limit_samples = g_variant_get_uint64(data); + devc->limit_msec = 0; + break; + case SR_CONF_LIMIT_MSEC: + devc->limit_msec = g_variant_get_uint64(data) * 1000; + devc->limit_samples = 0; + break; + case SR_CONF_SAMPLERATE: + samplerate = g_variant_get_uint64(data); + if (samplerate > MAX_SAMPLE_RATE) { + sr_err("Maximum sample rate is %d", MAX_SAMPLE_RATE); + ret = SR_ERR_SAMPLERATE; + break; + } + devc->samplerate = samplerate; + bl_acme_maybe_set_update_interval(sdi, samplerate); + break; + case SR_CONF_PROBE_FACTOR: + if (!cg) + return SR_ERR_CHANNEL_GROUP; + ret = bl_acme_set_shunt(cg, g_variant_get_uint64(data)); + break; + case SR_CONF_POWER_OFF: + if (!cg) + return SR_ERR_CHANNEL_GROUP; + ret = bl_acme_set_power_off(cg, g_variant_get_boolean(data)); + break; default: ret = SR_ERR_NA; } @@ -125,61 +258,161 @@ static int config_set(uint32_t key, GVariant *data, const struct sr_dev_inst *sd return ret; } -static int config_list(uint32_t key, GVariant **data, const struct sr_dev_inst *sdi, - const struct sr_channel_group *cg) +static int config_list(uint32_t key, GVariant **data, + const struct sr_dev_inst *sdi, + const struct sr_channel_group *cg) { - int ret; + uint32_t devopts_cg[MAX_DEVOPTS_CG]; + GVariant *gvar; + GVariantBuilder gvb; + int ret, num_devopts_cg = 0; (void)sdi; - (void)data; (void)cg; ret = SR_OK; - switch (key) { - /* TODO */ - default: - return SR_ERR_NA; + if (!cg) { + switch (key) { + case SR_CONF_DEVICE_OPTIONS: + *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32, + devopts, ARRAY_SIZE(devopts), sizeof(uint32_t)); + break; + case SR_CONF_SAMPLERATE: + g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}")); + gvar = g_variant_new_fixed_array(G_VARIANT_TYPE("t"), + samplerates, ARRAY_SIZE(samplerates), sizeof(uint64_t)); + g_variant_builder_add(&gvb, "{sv}", + "samplerate-steps", gvar); + *data = g_variant_builder_end(&gvb); + break; + default: + return SR_ERR_NA; + } + } else { + switch (key) { + case SR_CONF_DEVICE_OPTIONS: + if (bl_acme_get_probe_type(cg) == PROBE_ENRG) + devopts_cg[num_devopts_cg++] = HAS_PROBE_FACTOR; + if (bl_acme_probe_has_pws(cg)) + devopts_cg[num_devopts_cg++] = HAS_POWER_OFF; + + *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32, + devopts_cg, num_devopts_cg, sizeof(uint32_t)); + break; + default: + return SR_ERR_NA; + } } return ret; } -static int dev_acquisition_start(const struct sr_dev_inst *sdi, - void *cb_data) +static void dev_acquisition_close(const struct sr_dev_inst *sdi) { - (void)sdi; - (void)cb_data; + GSList *chl; + struct sr_channel *ch; + + for (chl = sdi->channels; chl; chl = chl->next) { + ch = chl->data; + bl_acme_close_channel(ch); + } +} + +static int dev_acquisition_open(const struct sr_dev_inst *sdi) +{ + GSList *chl; + struct sr_channel *ch; + + for (chl = sdi->channels; chl; chl = chl->next) { + ch = chl->data; + if (bl_acme_open_channel(ch)) { + sr_err("Error opening channel %s", ch->name); + dev_acquisition_close(sdi); + return SR_ERR; + } + } + + return 0; +} + +static int dev_acquisition_start(const struct sr_dev_inst *sdi) +{ + struct dev_context *devc; + struct itimerspec tspec = { + .it_interval = { 0, 0 }, + .it_value = { 0, 0 } + }; if (sdi->status != SR_ST_ACTIVE) return SR_ERR_DEV_CLOSED; - /* TODO: configure hardware, reset acquisition state, set up - * callbacks and send header packet. */ + if (dev_acquisition_open(sdi)) + return SR_ERR; + + devc = sdi->priv; + devc->samples_read = 0; + devc->samples_missed = 0; + devc->timer_fd = timerfd_create(CLOCK_MONOTONIC, 0); + if (devc->timer_fd < 0) { + sr_err("Error creating timer fd"); + return SR_ERR; + } + + tspec.it_interval.tv_sec = 0; + tspec.it_interval.tv_nsec = SR_HZ_TO_NS(devc->samplerate); + tspec.it_value = tspec.it_interval; + + if (timerfd_settime(devc->timer_fd, 0, &tspec, NULL)) { + sr_err("Failed to set timer"); + close(devc->timer_fd); + return SR_ERR; + } + + devc->channel = g_io_channel_unix_new(devc->timer_fd); + g_io_channel_set_flags(devc->channel, G_IO_FLAG_NONBLOCK, NULL); + g_io_channel_set_encoding(devc->channel, NULL, NULL); + g_io_channel_set_buffered(devc->channel, FALSE); + + sr_session_source_add_channel(sdi->session, devc->channel, + G_IO_IN | G_IO_ERR, 1000, bl_acme_receive_data, (void *)sdi); + + std_session_send_df_header(sdi, LOG_PREFIX); + devc->start_time = g_get_monotonic_time(); return SR_OK; } -static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data) +static int dev_acquisition_stop(struct sr_dev_inst *sdi) { - (void)cb_data; + struct dev_context *devc; + + devc = sdi->priv; if (sdi->status != SR_ST_ACTIVE) return SR_ERR_DEV_CLOSED; - /* TODO: stop acquisition. */ + dev_acquisition_close(sdi); + sr_session_source_remove_channel(sdi->session, devc->channel); + g_io_channel_shutdown(devc->channel, FALSE, NULL); + g_io_channel_unref(devc->channel); + devc->channel = NULL; + + std_session_send_df_end(sdi, LOG_PREFIX); + + if (devc->samples_missed > 0) + sr_warn("%" PRIu64 " samples missed", devc->samples_missed); return SR_OK; } SR_PRIV struct sr_dev_driver baylibre_acme_driver_info = { .name = "baylibre-acme", - .longname = "baylibre-acme", + .longname = "BayLibre ACME (Another Cute Measurement Equipment)", .api_version = 1, .init = init, - .cleanup = cleanup, + .cleanup = std_cleanup, .scan = scan, .dev_list = dev_list, - .dev_clear = dev_clear, .config_get = config_get, .config_set = config_set, .config_list = config_list, @@ -187,5 +420,5 @@ SR_PRIV struct sr_dev_driver baylibre_acme_driver_info = { .dev_close = dev_close, .dev_acquisition_start = dev_acquisition_start, .dev_acquisition_stop = dev_acquisition_stop, - .priv = NULL, + .context = NULL, };