28d71b79a2
Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com>
402 lines
13 KiB
C
402 lines
13 KiB
C
/****************************************************************************
|
|
* apps/testing/sensortest/sensortest.c
|
|
*
|
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
* contributor license agreements. See the NOTICE file distributed with
|
|
* this work for additional information regarding copyright ownership. The
|
|
* ASF licenses this file to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance with the
|
|
* License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Included Files
|
|
****************************************************************************/
|
|
|
|
#include <nuttx/sensors/sensor.h>
|
|
#include <nuttx/config.h>
|
|
#include <sys/ioctl.h>
|
|
#include <inttypes.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <fcntl.h>
|
|
#include <poll.h>
|
|
|
|
/****************************************************************************
|
|
* Pre-processor Definitions
|
|
****************************************************************************/
|
|
|
|
#define ARRAYSIZE(a) (sizeof(a) / sizeof(a)[0])
|
|
#define DEVNAME_FMT "/dev/uorb/sensor_%s"
|
|
#define DEVNAME_MAX 64
|
|
|
|
/****************************************************************************
|
|
* Private Types
|
|
****************************************************************************/
|
|
|
|
typedef void (*data_print)(FAR const char *buffer, FAR const char *name);
|
|
|
|
struct sensor_info
|
|
{
|
|
data_print print;
|
|
const uint8_t esize;
|
|
FAR const char *name;
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Function Prototypes
|
|
****************************************************************************/
|
|
|
|
static void print_vec3(FAR const char *buffer, FAR const char *name);
|
|
static void print_valf3(FAR const char *buffer, FAR const char *name);
|
|
static void print_valf2(FAR const char *buffer, FAR const char *name);
|
|
static void print_valf(FAR const char *buffer, FAR const char *name);
|
|
static void print_valb(FAR const char *buffer, FAR const char *name);
|
|
static void print_vali2(FAR const char *buffer, FAR const char *name);
|
|
static void print_ecg(FAR const char *buffer, FAR const char *name);
|
|
static void print_ppgd(FAR const char *buffer, FAR const char *name);
|
|
static void print_ppgq(FAR const char *buffer, FAR const char *name);
|
|
static void print_cap(FAR const char *buffer, FAR const char *name);
|
|
static void print_gps(FAR const char *buffer, FAR const char *name);
|
|
static void print_gps_satellite(FAR const char *buffer,
|
|
FAR const char *name);
|
|
|
|
/****************************************************************************
|
|
* Private Data
|
|
****************************************************************************/
|
|
|
|
static bool g_should_exit = false;
|
|
|
|
static const struct sensor_info g_sensor_info[] =
|
|
{
|
|
{print_vec3, sizeof(struct sensor_accel), "accel"},
|
|
{print_valf2, sizeof(struct sensor_baro), "baro"},
|
|
{print_cap, sizeof(struct sensor_cap), "cap"},
|
|
{print_valf, sizeof(struct sensor_co2), "co2"},
|
|
{print_valf, sizeof(struct sensor_dust), "dust"},
|
|
{print_ecg, sizeof(struct sensor_ecg), "ecg"},
|
|
{print_gps, sizeof(struct sensor_gps), "gps"},
|
|
{print_gps_satellite,
|
|
sizeof(struct sensor_gps_satellite), "gps_satellite"},
|
|
{print_vec3, sizeof(struct sensor_gyro), "gyro"},
|
|
{print_valb, sizeof(struct sensor_hall), "hall"},
|
|
{print_valf, sizeof(struct sensor_hbeat), "hbeat"},
|
|
{print_valf, sizeof(struct sensor_hcho), "hcho"},
|
|
{print_valf, sizeof(struct sensor_hrate), "hrate"},
|
|
{print_valf, sizeof(struct sensor_humi), "humi"},
|
|
{print_valf2, sizeof(struct sensor_impd), "impd"},
|
|
{print_valf, sizeof(struct sensor_ir), "ir"},
|
|
{print_valf, sizeof(struct sensor_light), "light"},
|
|
{print_vec3, sizeof(struct sensor_mag), "mag"},
|
|
{print_valf, sizeof(struct sensor_noise), "noise"},
|
|
{print_vali2, sizeof(struct sensor_ots), "ots"},
|
|
{print_valf, sizeof(struct sensor_ph), "ph"},
|
|
{print_valf, sizeof(struct sensor_pm10), "pm10"},
|
|
{print_valf, sizeof(struct sensor_pm1p0), "pm1p0"},
|
|
{print_valf, sizeof(struct sensor_pm25), "pm25"},
|
|
{print_ppgd, sizeof(struct sensor_ppgd), "ppgd"},
|
|
{print_ppgq, sizeof(struct sensor_ppgq), "ppgq"},
|
|
{print_valf, sizeof(struct sensor_prox), "prox"},
|
|
{print_valf3, sizeof(struct sensor_rgb), "rgb"},
|
|
{print_valf, sizeof(struct sensor_temp), "temp"},
|
|
{print_valf, sizeof(struct sensor_tvoc), "tvoc"},
|
|
{print_valf, sizeof(struct sensor_uv), "uv"}
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Functions
|
|
****************************************************************************/
|
|
|
|
static void print_vec3(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_accel *event = (FAR struct sensor_accel *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " x:%.2f y:%.2f z:%.2f, "
|
|
"temperature:%.2f\n",
|
|
name, event->timestamp, event->x, event->y,
|
|
event->z, event->temperature);
|
|
}
|
|
|
|
static void print_valb(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_hall *event = (FAR struct sensor_hall *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " value:%" PRIi32 "\n",
|
|
name, event->timestamp, event->hall);
|
|
}
|
|
|
|
static void print_vali2(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_ots *event = (FAR struct sensor_ots *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " value1:% " PRIi32 " value2:% " PRIi32
|
|
"\n", name, event->timestamp, event->x, event->y);
|
|
}
|
|
|
|
static void print_valf(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_prox *event = (FAR struct sensor_prox *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " value:%.2f\n",
|
|
name, event->timestamp, event->proximity);
|
|
}
|
|
|
|
static void print_valf2(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_baro *event = (FAR struct sensor_baro *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " value1:%.2f value2:%.2f\n",
|
|
name, event->timestamp, event->pressure, event->temperature);
|
|
}
|
|
|
|
static void print_valf3(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_rgb *event = (FAR struct sensor_rgb *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " value1:%.2f value2:%.2f, value3:%.2f\n",
|
|
name, event->timestamp, event->r, event->g, event->b);
|
|
}
|
|
|
|
static void print_ecg(const char *buffer, const char *name)
|
|
{
|
|
struct sensor_ecg *event = (struct sensor_ecg *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " ecg:%.4f status:%" PRIu32,
|
|
name, event->timestamp, event->ecg, event->status);
|
|
}
|
|
|
|
static void print_ppgd(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_ppgd *event = (FAR struct sensor_ppgd *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " ppg1:%" PRIu32 " ppg2:%" PRIu32 " "
|
|
"current:%" PRIu32 " gain1:%" PRIu16 " gain2:%" PRIu16 "\n",
|
|
name, event->timestamp, event->ppg[0], event->ppg[1],
|
|
event->current, event->gain[0], event->gain[1]);
|
|
}
|
|
|
|
static void print_ppgq(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_ppgq *event = (FAR struct sensor_ppgq *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " ppg1:%" PRIu32 " ppg2:%" PRIu32 " "
|
|
"ppg3:%" PRIu32 " ppg4:%" PRIu32 " current:%" PRIu32 " "
|
|
"gain1:%" PRIu16 " gain2:%" PRIu16 " gain3:%" PRIu16 " "
|
|
"gain4:%" PRIu16 "\n",
|
|
name, event->timestamp, event->ppg[0], event->ppg[1], event->ppg[2],
|
|
event->ppg[3], event->current, event->gain[0], event->gain[1],
|
|
event->gain[2], event->gain[3]);
|
|
}
|
|
|
|
static void print_cap(FAR const char *buffer, FAR const char *name)
|
|
{
|
|
FAR struct sensor_cap *event = (FAR struct sensor_cap *)buffer;
|
|
printf("%s: timestamp:%" PRIu64 " status:%" PRIu32 " "
|
|
"rawdata0:%" PRIu32 " rawdata1:%" PRIu32 " rawdata2:%" PRIu32 " "
|
|
"rawdata3:%" PRIu32 "\n",
|
|
name, event->timestamp, event->status, event->rawdata[0],
|
|
event->rawdata[1], event->rawdata[2], event->rawdata[3]);
|
|
}
|
|
|
|
static void print_gps(const char *buffer, const char *name)
|
|
{
|
|
FAR struct sensor_gps *event = (FAR struct sensor_gps *)buffer;
|
|
|
|
printf("%s: timestamp:%" PRIu64 " time_utc: %" PRIu64 " latitude: %f "
|
|
"longitude: %f altitude: %f altitude_ellipsoid: %f eph: %f epv: %f "
|
|
"hdop: %f vdop: %f ground_speed: %f course: %f satellites_used: %"
|
|
PRIu32 "\n",
|
|
name, event->timestamp, event->time_utc, event->latitude,
|
|
event->longitude, event->altitude, event->altitude_ellipsoid,
|
|
event->eph, event->epv, event->hdop, event->vdop,
|
|
event->ground_speed, event->course, event->satellites_used);
|
|
}
|
|
|
|
static void print_gps_satellite(FAR const char *buffer, FAR const char *name)
|
|
{
|
|
FAR FAR struct sensor_gps_satellite *event =
|
|
(FAR struct sensor_gps_satellite *)buffer;
|
|
|
|
printf("%s: timestamp: %" PRIu64 " count: %" PRIu32
|
|
" satellites: %" PRIu32 "\n",
|
|
name, event->timestamp, event->count, event->satellites);
|
|
}
|
|
|
|
static void usage(void)
|
|
{
|
|
printf("sensortest [arguments...] <command>\n");
|
|
printf("\t[-h ] sensortest commands help\n");
|
|
printf("\t[-i <val>] The output data period of sensor in us\n");
|
|
printf("\t default: 1000000\n");
|
|
printf("\t[-b <val>] The maximum report latency of sensor in us\n");
|
|
printf("\t default: 0\n");
|
|
printf("\t[-n <val>] The number of output data\n");
|
|
printf("\t default: 0\n");
|
|
|
|
printf(" Commands:\n");
|
|
printf("\t<sensor_node_name> ex, accel0(/dev/uorb/sensor_accel0)\n");
|
|
}
|
|
|
|
static void exit_handler(int signo)
|
|
{
|
|
g_should_exit = true;
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* sensortest_main
|
|
****************************************************************************/
|
|
|
|
int main(int argc, FAR char *argv[])
|
|
{
|
|
unsigned int interval = 1000000;
|
|
unsigned int received = 0;
|
|
unsigned int latency = 0;
|
|
unsigned int count = 0;
|
|
char devname[PATH_MAX];
|
|
struct pollfd fds;
|
|
FAR char *buffer;
|
|
FAR char *name;
|
|
int len = 0;
|
|
int fd;
|
|
int idx;
|
|
int ret;
|
|
|
|
if (argc <= 1)
|
|
{
|
|
usage();
|
|
return -EINVAL;
|
|
}
|
|
|
|
if (signal(SIGINT, exit_handler) == SIG_ERR)
|
|
{
|
|
return -errno;
|
|
}
|
|
|
|
g_should_exit = false;
|
|
while ((ret = getopt(argc, argv, "i:b:n:h")) != EOF)
|
|
{
|
|
switch (ret)
|
|
{
|
|
case 'i':
|
|
interval = strtoul(optarg, NULL, 0);
|
|
break;
|
|
|
|
case 'b':
|
|
latency = strtoul(optarg, NULL, 0);
|
|
break;
|
|
|
|
case 'n':
|
|
count = strtoul(optarg, NULL, 0);
|
|
break;
|
|
|
|
case 'h':
|
|
default:
|
|
usage();
|
|
goto name_err;
|
|
}
|
|
}
|
|
|
|
if (optind < argc)
|
|
{
|
|
name = argv[optind];
|
|
for (idx = 0; idx < ARRAYSIZE(g_sensor_info); idx++)
|
|
{
|
|
if (!strncmp(name, g_sensor_info[idx].name,
|
|
strlen(g_sensor_info[idx].name)))
|
|
{
|
|
len = g_sensor_info[idx].esize;
|
|
buffer = calloc(1, len);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!len)
|
|
{
|
|
printf("The sensor node name:%s is invalid\n", name);
|
|
usage();
|
|
ret = -EINVAL;
|
|
goto name_err;
|
|
}
|
|
|
|
if (!buffer)
|
|
{
|
|
ret = -ENOMEM;
|
|
goto name_err;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
usage();
|
|
ret = -EINVAL;
|
|
goto name_err;
|
|
}
|
|
|
|
snprintf(devname, PATH_MAX, DEVNAME_FMT, name);
|
|
fd = open(devname, O_RDONLY | O_NONBLOCK);
|
|
if (fd < 0)
|
|
{
|
|
ret = -errno;
|
|
printf("Failed to open device:%s, ret:%s\n",
|
|
devname, strerror(errno));
|
|
goto open_err;
|
|
}
|
|
|
|
ret = ioctl(fd, SNIOC_SET_INTERVAL, interval);
|
|
if (ret < 0)
|
|
{
|
|
ret = -errno;
|
|
if (ret != -ENOTSUP)
|
|
{
|
|
printf("Failed to set interval for sensor:%s, ret:%s\n",
|
|
devname, strerror(errno));
|
|
goto ctl_err;
|
|
}
|
|
}
|
|
|
|
ret = ioctl(fd, SNIOC_BATCH, latency);
|
|
if (ret < 0)
|
|
{
|
|
ret = -errno;
|
|
if (ret != -ENOTSUP)
|
|
{
|
|
printf("Failed to batch for sensor:%s, ret:%s\n",
|
|
devname, strerror(errno));
|
|
goto ctl_err;
|
|
}
|
|
}
|
|
|
|
printf("SensorTest: Test %s with interval(%uus), latency(%uus)\n",
|
|
devname, interval, latency);
|
|
|
|
fds.fd = fd;
|
|
fds.events = POLLIN;
|
|
|
|
while ((!count || received < count) && !g_should_exit)
|
|
{
|
|
if (poll(&fds, 1, -1) > 0)
|
|
{
|
|
if (read(fd, buffer, len) >= len)
|
|
{
|
|
received++;
|
|
g_sensor_info[idx].print(buffer, name);
|
|
}
|
|
}
|
|
}
|
|
|
|
printf("SensorTest: Received message: %s, number:%d/%d\n",
|
|
name, received, count);
|
|
|
|
ctl_err:
|
|
close(fd);
|
|
open_err:
|
|
free(buffer);
|
|
name_err:
|
|
optind = 0;
|
|
return ret;
|
|
}
|