|
| 1 | +/* |
| 2 | + * Copyright (c) 2025 Croxel, Inc. |
| 3 | + * Copyright (c) 2025 CogniPilot Foundation |
| 4 | + * |
| 5 | + * SPDX-License-Identifier: Apache-2.0 |
| 6 | + */ |
| 7 | + |
| 8 | +#define DT_DRV_COMPAT pni_rm3100 |
| 9 | + |
| 10 | +#include <zephyr/drivers/sensor.h> |
| 11 | +#include <zephyr/drivers/i2c.h> |
| 12 | +#include <zephyr/rtio/rtio.h> |
| 13 | +#include <zephyr/rtio/work.h> |
| 14 | +#include <zephyr/sys/check.h> |
| 15 | + |
| 16 | +#include "rm3100.h" |
| 17 | +#include "rm3100_reg.h" |
| 18 | +#include "rm3100_bus.h" |
| 19 | +#include "rm3100_decoder.h" |
| 20 | + |
| 21 | +#include <zephyr/logging/log.h> |
| 22 | +LOG_MODULE_REGISTER(RM3100, CONFIG_SENSOR_LOG_LEVEL); |
| 23 | + |
| 24 | +static void rm3100_complete_result(struct rtio *ctx, const struct rtio_sqe *sqe, void *arg) |
| 25 | +{ |
| 26 | + struct rtio_iodev_sqe *iodev_sqe = (struct rtio_iodev_sqe *)sqe->userdata; |
| 27 | + struct rtio_cqe *cqe; |
| 28 | + int err = 0; |
| 29 | + |
| 30 | + do { |
| 31 | + cqe = rtio_cqe_consume(ctx); |
| 32 | + if (cqe != NULL) { |
| 33 | + err = cqe->result; |
| 34 | + rtio_cqe_release(ctx, cqe); |
| 35 | + } |
| 36 | + } while (cqe != NULL); |
| 37 | + |
| 38 | + if (err) { |
| 39 | + rtio_iodev_sqe_err(iodev_sqe, err); |
| 40 | + } else { |
| 41 | + rtio_iodev_sqe_ok(iodev_sqe, 0); |
| 42 | + } |
| 43 | + |
| 44 | + LOG_DBG("One-shot fetch completed"); |
| 45 | +} |
| 46 | + |
| 47 | +static void rm3100_submit_one_shot(const struct device *dev, struct rtio_iodev_sqe *iodev_sqe) |
| 48 | +{ |
| 49 | + const struct sensor_read_config *cfg = iodev_sqe->sqe.iodev->data; |
| 50 | + const struct sensor_chan_spec *const channels = cfg->channels; |
| 51 | + const size_t num_channels = cfg->count; |
| 52 | + uint32_t min_buf_len = sizeof(struct rm3100_encoded_data); |
| 53 | + int err; |
| 54 | + uint8_t *buf; |
| 55 | + uint32_t buf_len; |
| 56 | + struct rm3100_encoded_data *edata; |
| 57 | + struct rm3100_data *data = dev->data; |
| 58 | + |
| 59 | + err = rtio_sqe_rx_buf(iodev_sqe, min_buf_len, min_buf_len, &buf, &buf_len); |
| 60 | + if (err) { |
| 61 | + LOG_ERR("Failed to get a read buffer of size %u bytes", min_buf_len); |
| 62 | + rtio_iodev_sqe_err(iodev_sqe, err); |
| 63 | + return; |
| 64 | + } |
| 65 | + |
| 66 | + edata = (struct rm3100_encoded_data *)buf; |
| 67 | + |
| 68 | + err = rm3100_encode(dev, channels, num_channels, buf); |
| 69 | + if (err != 0) { |
| 70 | + LOG_ERR("Failed to encode sensor data"); |
| 71 | + rtio_iodev_sqe_err(iodev_sqe, err); |
| 72 | + return; |
| 73 | + } |
| 74 | + |
| 75 | + struct rtio_sqe *write_sqe = rtio_sqe_acquire(data->rtio.ctx); |
| 76 | + struct rtio_sqe *read_sqe = rtio_sqe_acquire(data->rtio.ctx); |
| 77 | + struct rtio_sqe *complete_sqe = rtio_sqe_acquire(data->rtio.ctx); |
| 78 | + |
| 79 | + if (!write_sqe || !read_sqe || !complete_sqe) { |
| 80 | + LOG_ERR("Failed to acquire RTIO SQEs"); |
| 81 | + rtio_iodev_sqe_err(iodev_sqe, -ENOMEM); |
| 82 | + return; |
| 83 | + } |
| 84 | + |
| 85 | + uint8_t val = RM3100_REG_MX; |
| 86 | + |
| 87 | + rtio_sqe_prep_tiny_write(write_sqe, |
| 88 | + data->rtio.iodev, |
| 89 | + RTIO_PRIO_HIGH, |
| 90 | + &val, |
| 91 | + 1, |
| 92 | + NULL); |
| 93 | + write_sqe->flags |= RTIO_SQE_TRANSACTION; |
| 94 | + |
| 95 | + rtio_sqe_prep_read(read_sqe, |
| 96 | + data->rtio.iodev, |
| 97 | + RTIO_PRIO_HIGH, |
| 98 | + edata->payload, |
| 99 | + sizeof(edata->payload), |
| 100 | + NULL); |
| 101 | + read_sqe->iodev_flags |= RTIO_IODEV_I2C_STOP | RTIO_IODEV_I2C_RESTART; |
| 102 | + read_sqe->flags |= RTIO_SQE_CHAINED; |
| 103 | + |
| 104 | + rtio_sqe_prep_callback_no_cqe(complete_sqe, |
| 105 | + rm3100_complete_result, |
| 106 | + (void *)dev, |
| 107 | + iodev_sqe); |
| 108 | + |
| 109 | + rtio_submit(data->rtio.ctx, 0); |
| 110 | +} |
| 111 | + |
| 112 | +static void rm3100_submit(const struct device *dev, struct rtio_iodev_sqe *iodev_sqe) |
| 113 | +{ |
| 114 | + const struct sensor_read_config *cfg = iodev_sqe->sqe.iodev->data; |
| 115 | + |
| 116 | + if (!cfg->is_streaming) { |
| 117 | + rm3100_submit_one_shot(dev, iodev_sqe); |
| 118 | + } else { |
| 119 | + LOG_ERR("Streaming not supported"); |
| 120 | + rtio_iodev_sqe_err(iodev_sqe, -ENOTSUP); |
| 121 | + } |
| 122 | +} |
| 123 | + |
| 124 | +/* This will be implemented later */ |
| 125 | +static DEVICE_API(sensor, rm3100_driver_api) = { |
| 126 | + /* API functions will be added here */ |
| 127 | + .submit = rm3100_submit, |
| 128 | + .get_decoder = rm3100_get_decoder, |
| 129 | +}; |
| 130 | + |
| 131 | +static int rm3100_init(const struct device *dev) |
| 132 | +{ |
| 133 | + uint8_t val; |
| 134 | + int err; |
| 135 | + |
| 136 | + /* Check device ID to make sure we can talk to the sensor */ |
| 137 | + err = rm3100_bus_read(dev, RM3100_REG_REVID, &val, 1); |
| 138 | + if (err < 0) { |
| 139 | + LOG_ERR("Failed to read chip ID"); |
| 140 | + return err; |
| 141 | + } else if (val != RM3100_REVID_VALUE) { |
| 142 | + LOG_ERR("Invalid chip ID: 0x%02x, expected 0x%02x", |
| 143 | + val, RM3100_REVID_VALUE); |
| 144 | + return -ENODEV; |
| 145 | + } |
| 146 | + LOG_DBG("RM3100 chip ID confirmed: 0x%02x", val); |
| 147 | + |
| 148 | + /** Enable Continuous measurement on all axis */ |
| 149 | + val = RM3100_CMM_ALL_AXIS; |
| 150 | + |
| 151 | + err = rm3100_bus_write(dev, RM3100_REG_CMM, &val, 1); |
| 152 | + if (err < 0) { |
| 153 | + LOG_ERR("Failed to set sensor in Continuous Measurement Mode: %d", err); |
| 154 | + return err; |
| 155 | + } |
| 156 | + |
| 157 | + return 0; |
| 158 | +} |
| 159 | + |
| 160 | +#define RM3100_DEFINE(inst) \ |
| 161 | + \ |
| 162 | + RTIO_DEFINE(rm3100_rtio_ctx_##inst, 8, 8); \ |
| 163 | + I2C_DT_IODEV_DEFINE(rm3100_bus_##inst, DT_DRV_INST(inst)); \ |
| 164 | + \ |
| 165 | + static const struct rm3100_config rm3100_cfg_##inst; \ |
| 166 | + \ |
| 167 | + static struct rm3100_data rm3100_data_##inst = { \ |
| 168 | + .rtio = { \ |
| 169 | + .iodev = &rm3100_bus_##inst, \ |
| 170 | + .ctx = &rm3100_rtio_ctx_##inst, \ |
| 171 | + }, \ |
| 172 | + }; \ |
| 173 | + \ |
| 174 | + SENSOR_DEVICE_DT_INST_DEFINE(inst, rm3100_init, NULL, \ |
| 175 | + &rm3100_data_##inst, \ |
| 176 | + &rm3100_cfg_##inst, \ |
| 177 | + POST_KERNEL, \ |
| 178 | + CONFIG_SENSOR_INIT_PRIORITY, \ |
| 179 | + &rm3100_driver_api); |
| 180 | + |
| 181 | +DT_INST_FOREACH_STATUS_OKAY(RM3100_DEFINE) |
0 commit comments