nuttx-apps/testing/sensortest/sensortest.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;
}