|
| 1 | +/* |
| 2 | + * Copyright (c) 2025 Prevas A/S |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
| 6 | +#include <math.h> |
| 7 | +#include <stdint.h> |
| 8 | + |
| 9 | +#include <zephyr/drivers/adc.h> |
| 10 | +#include <zephyr/drivers/sensor.h> |
| 11 | +#include <zephyr/device.h> |
| 12 | +#include <zephyr/devicetree.h> |
| 13 | +#include <zephyr/logging/log.h> |
| 14 | + |
| 15 | +LOG_MODULE_REGISTER(d6f, CONFIG_SENSOR_LOG_LEVEL); |
| 16 | + |
| 17 | +struct d6f_config { |
| 18 | + const struct adc_dt_spec *adc; |
| 19 | + struct adc_sequence sequence; |
| 20 | + const float *polynomial_coefficients; |
| 21 | + uint8_t polynomial_degree; |
| 22 | +}; |
| 23 | + |
| 24 | +struct d6f_data { |
| 25 | + uint32_t adc_sample; |
| 26 | +}; |
| 27 | + |
| 28 | +static int d6f_sample_fetch(const struct device *dev, enum sensor_channel chan) |
| 29 | +{ |
| 30 | + const struct d6f_config *config = dev->config; |
| 31 | + |
| 32 | + switch (chan) { |
| 33 | + case SENSOR_CHAN_ALL: |
| 34 | + case SENSOR_CHAN_FLOW_RATE: |
| 35 | + return adc_read_dt(config->adc, &config->sequence); |
| 36 | + default: |
| 37 | + return -ENOTSUP; |
| 38 | + } |
| 39 | +} |
| 40 | + |
| 41 | +static int d6f_flow_rate(const struct d6f_config *config, struct d6f_data *data, |
| 42 | + struct sensor_value *val) |
| 43 | +{ |
| 44 | + float flow_rate = config->polynomial_coefficients[0]; |
| 45 | + int32_t uv = data->adc_sample; |
| 46 | + float v; |
| 47 | + int rc; |
| 48 | + |
| 49 | + rc = adc_raw_to_microvolts_dt(config->adc, &uv); |
| 50 | + if (rc != 0) { |
| 51 | + return rc; |
| 52 | + } |
| 53 | + v = uv / 1000000.f; |
| 54 | + |
| 55 | + for (uint8_t i = 1; i <= config->polynomial_degree; ++i) { |
| 56 | + flow_rate += config->polynomial_coefficients[i] * powf(v, i); |
| 57 | + } |
| 58 | + |
| 59 | + rc = sensor_value_from_float(val, flow_rate); |
| 60 | + if (rc != 0) { |
| 61 | + return rc; |
| 62 | + } |
| 63 | + |
| 64 | + return 0; |
| 65 | +} |
| 66 | + |
| 67 | +static int d6f_channel_get(const struct device *dev, enum sensor_channel chan, |
| 68 | + struct sensor_value *val) |
| 69 | +{ |
| 70 | + const struct d6f_config *config = dev->config; |
| 71 | + struct d6f_data *data = dev->data; |
| 72 | + |
| 73 | + switch (chan) { |
| 74 | + case SENSOR_CHAN_FLOW_RATE: |
| 75 | + return d6f_flow_rate(config, data, val); |
| 76 | + default: |
| 77 | + return -ENOTSUP; |
| 78 | + } |
| 79 | + |
| 80 | + return 0; |
| 81 | +} |
| 82 | + |
| 83 | +static int d6f_init(const struct device *dev) |
| 84 | +{ |
| 85 | + const struct d6f_config *config = dev->config; |
| 86 | + int rc; |
| 87 | + |
| 88 | + LOG_DBG("Initializing %s", dev->name); |
| 89 | + |
| 90 | + if (!adc_is_ready_dt(config->adc)) { |
| 91 | + LOG_ERR("%s not ready", dev->name); |
| 92 | + return -ENODEV; |
| 93 | + } |
| 94 | + |
| 95 | + rc = adc_channel_setup_dt(config->adc); |
| 96 | + if (rc != 0) { |
| 97 | + LOG_ERR("%s setup failed: %d", config->adc->dev->name, rc); |
| 98 | + return -ENODEV; |
| 99 | + } |
| 100 | + |
| 101 | + return 0; |
| 102 | +} |
| 103 | + |
| 104 | +static DEVICE_API(sensor, d6f_driver_api) = { |
| 105 | + .sample_fetch = d6f_sample_fetch, |
| 106 | + .channel_get = d6f_channel_get, |
| 107 | +}; |
| 108 | + |
| 109 | +#define D6F_INIT(n, c, p) \ |
| 110 | + static struct d6f_data d6f_data_##c##_##n; \ |
| 111 | + static const struct adc_dt_spec d6f_adc_##c##_##n = ADC_DT_SPEC_INST_GET(n); \ |
| 112 | + static const struct d6f_config d6f_config_##c##_##n = { \ |
| 113 | + .adc = &d6f_adc_##c##_##n, \ |
| 114 | + .sequence = \ |
| 115 | + { \ |
| 116 | + .options = NULL, \ |
| 117 | + .channels = BIT(d6f_adc_##c##_##n.channel_id), \ |
| 118 | + .buffer = &d6f_data_##c##_##n.adc_sample, \ |
| 119 | + .buffer_size = sizeof(d6f_data_##c##_##n.adc_sample), \ |
| 120 | + .resolution = d6f_adc_##c##_##n.resolution, \ |
| 121 | + .oversampling = d6f_adc_##c##_##n.oversampling, \ |
| 122 | + .calibrate = false, \ |
| 123 | + }, \ |
| 124 | + .polynomial_coefficients = (p), \ |
| 125 | + .polynomial_degree = (ARRAY_SIZE(p) - 1), \ |
| 126 | + }; \ |
| 127 | + \ |
| 128 | + SENSOR_DEVICE_DT_INST_DEFINE(n, d6f_init, NULL, &d6f_data_##c##_##n, \ |
| 129 | + &d6f_config_##c##_##n, POST_KERNEL, \ |
| 130 | + CONFIG_SENSOR_INIT_PRIORITY, &d6f_driver_api); |
| 131 | + |
| 132 | +#undef DT_DRV_COMPAT |
| 133 | +#define DT_DRV_COMPAT omron_d6f_p0001 |
| 134 | +static __maybe_unused const float d6f_p0001_polynomial_coefficients[] = {-0.024864, 0.049944}; |
| 135 | +DT_INST_FOREACH_STATUS_OKAY_VARGS(D6F_INIT, DT_DRV_COMPAT, d6f_p0001_polynomial_coefficients) |
| 136 | + |
| 137 | +#undef DT_DRV_COMPAT |
| 138 | +#define DT_DRV_COMPAT omron_d6f_p0010 |
| 139 | +static __maybe_unused const float d6f_p0010_polynomial_coefficients[] = { |
| 140 | + -0.269996, 1.060657, -1.601495, 1.374705, 1.374705, -0.564312, 0.094003}; |
| 141 | +DT_INST_FOREACH_STATUS_OKAY_VARGS(D6F_INIT, DT_DRV_COMPAT, d6f_p0010_polynomial_coefficients) |
0 commit comments