blob: ab3b835aa103865a967b1981d80c3e9f7af35a2e [file] [log] [blame]
/*
* Copyright (c) 2023 Google LLC
*
* SPDX-License-Identifier: Apache-2.0
*/
#include <stdlib.h>
#include <string.h>
#include <zephyr/drivers/sensor.h>
#include <zephyr/kernel.h>
#include "sensor_shell.h"
/* Create a single common config for streaming */
static struct sensor_stream_trigger iodev_sensor_shell_trigger;
static struct sensor_read_config iodev_sensor_shell_stream_config = {
.sensor = NULL,
.is_streaming = true,
.triggers = &iodev_sensor_shell_trigger,
.count = 0,
.max = 1,
};
RTIO_IODEV_DEFINE(iodev_sensor_shell_stream, &__sensor_iodev_api,
&iodev_sensor_shell_stream_config);
static void sensor_shell_processing_entry_point(void *a, void *b, void *c)
{
ARG_UNUSED(a);
ARG_UNUSED(b);
ARG_UNUSED(c);
while (true) {
sensor_processing_with_callback(&sensor_read_rtio,
sensor_shell_processing_callback);
}
}
K_THREAD_DEFINE(sensor_shell_processing_tid, CONFIG_SENSOR_SHELL_THREAD_STACK_SIZE,
sensor_shell_processing_entry_point, NULL, NULL, NULL, 0, 0, 0);
int cmd_sensor_stream(const struct shell *shell_ptr, size_t argc, char *argv[])
{
static struct rtio_sqe *current_streaming_handle;
static struct sensor_shell_processing_context ctx;
const struct device *dev = device_get_binding(argv[1]);
if (argc != 5 && argc != 3) {
shell_error(shell_ptr, "Wrong number of arguments (%zu)", argc);
return -EINVAL;
}
if (dev == NULL) {
shell_error(shell_ptr, "Device unknown (%s)", argv[1]);
return -ENODEV;
}
if (current_streaming_handle != NULL) {
shell_info(shell_ptr, "Disabling existing stream");
rtio_sqe_cancel(current_streaming_handle);
}
if (strcmp("off", argv[2]) == 0) {
return 0;
}
if (strcmp("on", argv[2]) != 0) {
shell_error(shell_ptr, "Unknown streaming operation (%s)", argv[2]);
return -EINVAL;
}
if (strcmp("double_tap", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_DOUBLE_TAP;
} else if (strcmp("data_ready", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_DATA_READY;
} else if (strcmp("delta", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_DELTA;
} else if (strcmp("freefall", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_FREEFALL;
} else if (strcmp("motion", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_MOTION;
} else if (strcmp("near_far", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_NEAR_FAR;
} else if (strcmp("stationary", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_STATIONARY;
} else if (strcmp("threshold", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_THRESHOLD;
} else if (strcmp("fifo_wm", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_FIFO_WATERMARK;
} else if (strcmp("fifo_full", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_FIFO_FULL;
} else if (strcmp("tap", argv[3]) == 0) {
iodev_sensor_shell_trigger.trigger = SENSOR_TRIG_TAP;
} else {
shell_error(shell_ptr, "Invalid trigger (%s)", argv[3]);
return -EINVAL;
}
if (strcmp("incl", argv[4]) == 0) {
iodev_sensor_shell_trigger.opt = SENSOR_STREAM_DATA_INCLUDE;
} else if (strcmp("drop", argv[4]) == 0) {
iodev_sensor_shell_trigger.opt = SENSOR_STREAM_DATA_DROP;
} else if (strcmp("nop", argv[4]) == 0) {
iodev_sensor_shell_trigger.opt = SENSOR_STREAM_DATA_NOP;
} else {
shell_error(shell_ptr, "Unknown trigger op (%s)", argv[4]);
return -EINVAL;
}
shell_print(shell_ptr, "Enabling stream...");
iodev_sensor_shell_stream_config.sensor = dev;
iodev_sensor_shell_stream_config.count = 1;
ctx.dev = dev;
ctx.sh = shell_ptr;
int rc = sensor_stream(&iodev_sensor_shell_stream, &sensor_read_rtio, &ctx,
&current_streaming_handle);
if (rc != 0) {
shell_error(shell_ptr, "Failed to start stream");
}
return rc;
}