/*
* Check if a device has scan elements and if it is an output device (type = 0)
* or an input device (type = 1).
*/
static bool device_type_get(const struct iio_device *dev, int type)
{
struct iio_channel *ch;
int nb_channels, i;
if (!dev)
return false;
nb_channels = iio_device_get_channels_count(dev);
for (i = 0; i < nb_channels; i++) {
ch = iio_device_get_channel(dev, i);
if (iio_channel_is_scan_element(ch) &&
(type ? !iio_channel_is_output(ch) : iio_channel_is_output(ch)))
return true;
}
return false;
}
int main(int argc, char **argv)
{
unsigned int buffer_size = 1024 * 1024;
int c, option_index = 0, arg_index = 0;
unsigned int n_tx = 0, n_rx = 0;
static struct iio_context *ctx;
unsigned int i, nb_channels;
struct iio_buffer *buffer;
pthread_t monitor_thread;
const char *device_name;
struct iio_device *dev;
char unit;
int ret;
while ((c = getopt_long(argc, argv, "+hs:",
options, &option_index)) != -1) {
switch (c) {
case 'h':
usage(argv);
return EXIT_SUCCESS;
case 's':
arg_index += 2;
ret = sscanf(argv[arg_index], "%u%c", &buffer_size, &unit);
if (ret == 0)
return EXIT_FAILURE;
if (ret == 2) {
if (unit == 'k')
buffer_size *= 1024;
else if (unit == 'M')
buffer_size *= 1024 * 1024;
}
break;
case '?':
return EXIT_FAILURE;
}
}
if (arg_index + 1 >= argc) {
fprintf(stderr, "Incorrect number of arguments.\n\n");
usage(argv);
return EXIT_FAILURE;
}
#ifndef _WIN32
set_handler(SIGHUP, &quit_all);
#endif
set_handler(SIGINT, &quit_all);
set_handler(SIGSEGV, &quit_all);
set_handler(SIGTERM, &quit_all);
ctx = iio_create_default_context();
if (!ctx) {
fprintf(stderr, "Unable to create IIO context\n");
return EXIT_FAILURE;
}
device_name = argv[arg_index + 1];
dev = get_device(ctx, device_name);
if (!dev) {
iio_context_destroy(ctx);
return EXIT_FAILURE;
}
nb_channels = iio_device_get_channels_count(dev);
for (i = 0; i < nb_channels; i++) {
struct iio_channel *ch = iio_device_get_channel(dev, i);
if (!iio_channel_is_scan_element(ch))
continue;
iio_channel_enable(ch);
if (iio_channel_is_output(ch))
n_tx++;
else
n_rx++;
}
if (n_tx >= n_rx)
device_is_tx = true;
else
device_is_tx = false;
printf("Monitoring %s for underflows/overflows\n",
iio_device_get_name(dev));
buffer = iio_device_create_buffer(dev, buffer_size, false);
if (!buffer) {
fprintf(stderr, "Unable to allocate buffer\n");
iio_context_destroy(ctx);
return EXIT_FAILURE;
}
ret = pthread_create(&monitor_thread, NULL, monitor_thread_fn,
(void *)device_name);
if (ret) {
fprintf(stderr, "Failed to create monitor thread: %s\n",
strerror(-ret));
}
while (app_running) {
if (device_is_tx) {
ret = iio_buffer_push(buffer);
if (ret < 0) {
fprintf(stderr, "Unable to push buffer: %s\n",
strerror(-ret));
app_running = false;
break;
}
} else {
ret = iio_buffer_refill(buffer);
if (ret < 0) {
fprintf(stderr, "Unable to refill buffer: %s\n",
strerror(-ret));
app_running = false;
break;
}
}
}
iio_buffer_destroy(buffer);
iio_context_destroy(ctx);
pthread_join(monitor_thread, NULL);
return 0;
}
