T hacks: add ant0 utility to dump IQ data as float complex

Or as int16.
This commit is contained in:
Cedric Roux
2025-07-11 10:51:59 +02:00
parent e82fde247a
commit 6a2997ec9b
2 changed files with 269 additions and 2 deletions
+10 -2
View File
@@ -3,7 +3,7 @@ CFLAGS=-Wall -g -pthread -DT_TRACER -I. -I..
LIBS=-lX11 -lm -lpng -lXft
all: dump_nack_signal time_meas timeplot multi-rru-clean
all: dump_nack_signal time_meas timeplot multi-rru-clean ant0
dump_nack_signal: ../utils.o ../database.o ../configuration.o ../event.o \
dump_nack_signal.o
@@ -19,10 +19,18 @@ timplot: timeplot.o
multi-rru-clean: ../utils.o ../database.o ../event.o ../configuration.o multi-rru-clean.o
$(CC) $(CFLAGS) -o $@ $^ $(LIBS)
ant0: ant0.o ../utils.o ../database.o ../configuration.o ../event.o
$(CC) $(CFLAGS) -o $@ $^ -lm
#ant0 needs good compilation, conversion to float is heavy
ant0.o: ant0.c
$(CC) $(CFLAGS) -c -o $@ $< -march=native -ffast-math -O3
.PHONY: all
%.o: %.c
$(CC) $(CFLAGS) -c -o $@ $<
clean:
rm -f *.o core dump_nack_signal time_meas timeplot multi-rru-clean
rm -f *.o core dump_nack_signal time_meas timeplot multi-rru-clean \
ant0
+259
View File
@@ -0,0 +1,259 @@
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include <unistd.h>
#include "database.h"
//#include "utils.h"
#include "configuration.h"
#include "event.h"
void usage(void)
{
fprintf(stderr,
"options:\n"
" -d <database file> this option is mandatory\n"
" -o <output file> this option is mandatory (use - to dump to stdout)\n"
" -ip <host> connect to given IP address (default %s)\n"
" -p <port> connect to given port (default %d)\n"
" -n <max> save maximum 'max' IQ points (may be a bit more)\n"
" -nosig don't intercept ctrl+c, ctrl+z (it may mess up when piping to another tool)\n"
" -e <event name> trace given event (default 'USRP_TX_ANT0')\n"
" -b <buffer name> trace given buffer (default 'data')\n"
" -no-timestamp don't use 'timestamp' to fill missing transmission\n"
" -scale <scale value> divide 16bits input by given value for float output (default 32768.)\n"
" -s16 output short instead of float\n",
DEFAULT_REMOTE_IP,
DEFAULT_REMOTE_PORT
);
exit(1);
}
volatile int run = 1;
static int socket = -1;
void force_stop(int x)
{
fprintf(stderr, "\ngently quit...\n");
close(socket);
socket = -1;
run = 0;
}
int main(int n, char **v)
{
char *database_filename = NULL;
char *output_filename = NULL;
FILE *out;
void *database;
char *ip = DEFAULT_REMOTE_IP;
int port = DEFAULT_REMOTE_PORT;
int *is_on;
int number_of_events;
int i;
char mt;
long max = -1;
long dumped = 0;
int nosig = 0;
char *event_name = "USRP_TX_ANT0";
char *buffer_name = "data";
int no_timestamp = 0;
float scale = 32768;
int do_float = 1;
/* write on a socket fails if the other end is closed and we get SIGPIPE */
if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) abort();
for (i = 1; i < n; i++) {
if (!strcmp(v[i], "-h") || !strcmp(v[i], "--help")) usage();
if (!strcmp(v[i], "-d")) { if (i > n-2) usage(); database_filename = v[++i]; continue; }
if (!strcmp(v[i], "-o")) { if (i > n-2) usage(); output_filename = v[++i]; continue; }
if (!strcmp(v[i], "-ip")) { if (i > n-2) usage(); ip = v[++i]; continue; }
if (!strcmp(v[i], "-p")) { if (i > n-2) usage(); port = atoi(v[++i]); continue; }
if (!strcmp(v[i], "-n")) { if (i > n-2) usage(); max = atol(v[++i]); continue; }
if (!strcmp(v[i], "-nosig")) { nosig = 1; continue; }
if (!strcmp(v[i], "-e")) { if (i > n-2) usage(); event_name = v[++i]; continue; }
if (!strcmp(v[i], "-b")) { if (i > n-2) usage(); buffer_name = v[++i]; continue; }
if (!strcmp(v[i], "-no-timestamp")) { no_timestamp = 1; continue; }
if (!strcmp(v[i], "-scale")) { if (i > n-2) usage(); scale = atof(v[++i]); continue; }
if (!strcmp(v[i], "-s16")) { do_float = 0; continue; }
usage();
}
if (database_filename == NULL) {
fprintf(stderr, "ERROR: provide a database file (-d)\n");
exit(1);
}
if (output_filename == NULL) {
fprintf(stderr, "ERROR: provide an output file (-o)\n");
exit(1);
}
database = parse_database(database_filename);
load_config_file(database_filename);
/* get the event and fields from database */
int tx_ant0_id = event_id_from_name(database, event_name);
database_event_format f = get_format(database, tx_ant0_id);
#define G(var_name, var_type, var) \
if (!strcmp(f.name[i], var_name)) { \
if (strcmp(f.type[i], var_type)) abort(); \
var = i; \
continue; \
}
int timestamp_id = -1;
int data_id = -1;
for (i = 0; i < f.count; i++) {
/* there is a typo in T_messages.txt, "timestap" instead of "timestamp" */
if (!no_timestamp)
G("timestap", "int", timestamp_id);
G(buffer_name, "buffer", data_id);
}
if ((!no_timestamp && timestamp_id == -1) || data_id == -1) {
if (data_id == -1)
fprintf(stderr, "fatal: field '%s' of event '%s' does not exist\n",
buffer_name, event_name);
if (!no_timestamp && timestamp_id == -1)
fprintf(stderr, "fatal: field 'timestamp' of event '%s' does not exist\n",
event_name);
exit(1);
}
number_of_events = number_of_ids(database);
is_on = calloc(number_of_events, sizeof(int));
if (is_on == NULL) abort();
on_off(database, event_name, is_on, 1);
/* connect to tracee */
socket = connect_to(ip, port);
/* activate trace */
mt = 1;
if (socket_send(socket, &mt, 1) == -1 ||
socket_send(socket, &number_of_events, sizeof(int)) == -1 ||
socket_send(socket, is_on, number_of_events * sizeof(int)) == -1)
abort();
/* - means to output to stdout */
if (!strcmp(output_filename, "-")) {
out = stdout;
} else {
out = fopen(output_filename, "w");
if (out == NULL) {
perror(output_filename);
exit(1);
}
}
if (!nosig) {
/* exit on ctrl+c and ctrl+z */
if (signal(SIGQUIT, force_stop) == SIG_ERR) abort();
if (signal(SIGINT, force_stop) == SIG_ERR) abort();
if (signal(SIGTSTP, force_stop) == SIG_ERR) abort();
}
OBUF ebuf = {.osize = 0, .omaxsize = 0, .obuf = NULL};
/* read messages */
unsigned int last_timestamp = 0;
int last_count = 0;
int first = 1;
/* stop if max reached */
while ((max == -1 || dumped < max) && run) {
event e;
e = get_event(socket, &ebuf, database);
if (e.type == -1) break;
if (e.type != tx_ant0_id) continue;
unsigned int timestamp = 0;
if (!no_timestamp)
timestamp = e.e[timestamp_id].i;
int count = e.e[data_id].bsize / 4;
#if 0
fprintf(stderr, "timestamp %u buffer %p count %d not transmitted %d delta %d\n",
timestamp,
e.e[data_id].b,
count,
timestamp - (last_timestamp + last_count),
timestamp - last_timestamp);
#endif
/* rfsimulator may send 1 sample with old time when a new client connects
* let's reject this packet
*/
if (!first && timestamp < last_timestamp + last_count) {
/* more than 1 sample is not accepted */
if (count != 1) {
fprintf(stderr, "fatal: past transmission detected (%d samples)\n", count);
abort();
}
fprintf(stderr, "past transmission detected (%d samples), ignore\n", count);
continue;
}
/* insert missing samples */
if (!no_timestamp && !first) {
int missing = timestamp - (last_timestamp + last_count);
if (missing < 0) abort();
while (missing) {
/* limit size of array miss[], it's allocated on the stack */
int count = missing > 65536 ? 65536 : missing;
float miss[count * 2];
memset(miss, 0, count * 2 * sizeof(float));
if (fwrite(miss, count * 2 * sizeof(float), 1, out) != 1) {
fprintf(stderr, "error writing to %s\n", output_filename);
if (strcmp(output_filename, "-"))
fclose(out);
close(socket);
exit(1);
}
dumped += count;
missing -= count;
}
}
first = 0;
if (do_float) {
/* transform to float */
float iq[count * 2];
short *iq_short = (short *)e.e[data_id].b;
for (int i = 0; i < count * 2; i++)
iq[i] = iq_short[i] / scale;
/* save */
if (fwrite(iq, count * 2 * sizeof(float), 1, out) != 1) {
fprintf(stderr, "error writing to %s\n", output_filename);
if (strcmp(output_filename, "-"))
fclose(out);
exit(1);
}
} else {
/* save directly as short */
if (fwrite(e.e[data_id].b, count * 2 * 2, 1, out) != 1) {
fprintf(stderr, "error writing to %s\n", output_filename);
if (strcmp(output_filename, "-"))
fclose(out);
exit(1);
}
}
dumped += count;
last_timestamp = timestamp;
last_count = count;
}
if (strcmp(output_filename, "-"))
fclose(out);
close(socket);
return 0;
}