T hacks: add ant0 utility to dump IQ data as float complex
Or as int16.
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
Reference in New Issue
Block a user