srsLTE/lib/examples/usrp_capture.c

183 lines
4.6 KiB
C
Raw Normal View History

2015-03-24 08:11:14 -07:00
/**
*
* \section COPYRIGHT
*
2015-11-13 04:22:33 -08:00
* Copyright 2013-2015 Software Radio Systems Limited
2015-03-24 08:11:14 -07:00
*
* \section LICENSE
*
* This file is part of the srsLTE library.
*
* srsLTE is free software: you can redistribute it and/or modify
2015-05-08 08:05:40 -07:00
* it under the terms of the GNU Affero General Public License as
2015-03-24 08:11:14 -07:00
* published by the Free Software Foundation, either version 3 of
* the License, or (at your option) any later version.
*
* srsLTE is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2015-05-08 08:05:40 -07:00
* GNU Affero General Public License for more details.
2015-03-24 08:11:14 -07:00
*
2015-05-08 08:05:40 -07:00
* A copy of the GNU Affero General Public License can be found in
2015-03-24 08:11:14 -07:00
* the LICENSE file in the top-level directory of this distribution
* and at http://www.gnu.org/licenses/.
*
*/
#include <stdio.h>
#include <signal.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <unistd.h>
#include <math.h>
#include <time.h>
#include <stdbool.h>
#include "srslte/srslte.h"
2017-05-18 00:48:24 -07:00
#include "srslte/phy/rf/rf.h"
#include "srslte/phy/io/filesink.h"
2015-03-24 08:11:14 -07:00
static bool keep_running = true;
char *output_file_name;
char *rf_args="";
float rf_gain=40.0, rf_freq=-1.0, rf_rate=0.96e6;
2015-03-24 08:11:14 -07:00
int nof_samples = -1;
2017-07-25 04:17:36 -07:00
int nof_rx_antennas = 1;
2015-03-24 08:11:14 -07:00
void int_handler(int dummy) {
keep_running = false;
}
void usage(char *prog) {
printf("Usage: %s [agrnv] -f rx_frequency_hz -o output_file\n", prog);
printf("\t-a RF args [Default %s]\n", rf_args);
printf("\t-g RF Gain [Default %.2f dB]\n", rf_gain);
printf("\t-r RF Rate [Default %.6f Hz]\n", rf_rate);
2015-03-24 08:11:14 -07:00
printf("\t-n nof_samples [Default %d]\n", nof_samples);
2017-07-25 04:17:36 -07:00
printf("\t-A nof_rx_antennas [Default %d]\n", nof_rx_antennas);
2015-03-24 08:11:14 -07:00
printf("\t-v srslte_verbose\n");
}
void parse_args(int argc, char **argv) {
int opt;
2017-07-25 04:17:36 -07:00
while ((opt = getopt(argc, argv, "agrnvfoA")) != -1) {
2015-03-24 08:11:14 -07:00
switch (opt) {
case 'o':
output_file_name = argv[optind];
break;
case 'a':
rf_args = argv[optind];
2015-03-24 08:11:14 -07:00
break;
case 'g':
rf_gain = atof(argv[optind]);
2015-03-24 08:11:14 -07:00
break;
case 'r':
rf_rate = atof(argv[optind]);
2015-03-24 08:11:14 -07:00
break;
case 'f':
rf_freq = atof(argv[optind]);
2015-03-24 08:11:14 -07:00
break;
case 'n':
nof_samples = atoi(argv[optind]);
break;
2017-07-25 04:17:36 -07:00
case 'A':
nof_rx_antennas = atoi(argv[optind]);
break;
2015-03-24 08:11:14 -07:00
case 'v':
srslte_verbose++;
break;
default:
usage(argv[0]);
exit(-1);
}
}
if (rf_freq < 0) {
2015-03-24 08:11:14 -07:00
usage(argv[0]);
exit(-1);
}
}
int main(int argc, char **argv) {
2017-07-25 04:17:36 -07:00
cf_t *buffer[SRSLTE_MAX_PORTS];
2015-03-24 08:11:14 -07:00
int sample_count, n;
2015-12-16 09:02:25 -08:00
srslte_rf_t rf;
2015-03-24 08:11:14 -07:00
srslte_filesink_t sink;
2017-07-25 04:17:36 -07:00
uint32_t buflen;
2015-03-24 08:11:14 -07:00
signal(SIGINT, int_handler);
parse_args(argc, argv);
buflen = 4800;
sample_count = 0;
2017-07-25 04:17:36 -07:00
for (int i = 0; i < nof_rx_antennas; i++) {
buffer[i] = malloc(sizeof(cf_t) * buflen);
if (!buffer[i]) {
perror("malloc");
exit(-1);
}
2015-03-24 08:11:14 -07:00
}
srslte_filesink_init(&sink, output_file_name, SRSLTE_COMPLEX_FLOAT_BIN);
printf("Opening RF device...\n");
2017-07-25 04:17:36 -07:00
if (srslte_rf_open_multi(&rf, rf_args, nof_rx_antennas)) {
fprintf(stderr, "Error opening rf\n");
2015-03-24 08:11:14 -07:00
exit(-1);
}
2015-12-16 09:02:25 -08:00
srslte_rf_set_master_clock_rate(&rf, 30.72e6);
sigset_t sigset;
sigemptyset(&sigset);
sigaddset(&sigset, SIGINT);
sigprocmask(SIG_UNBLOCK, &sigset, NULL);
2015-12-16 09:02:25 -08:00
printf("Set RX freq: %.2f MHz\n", srslte_rf_set_rx_freq(&rf, rf_freq) / 1000000);
printf("Set RX gain: %.2f dB\n", srslte_rf_set_rx_gain(&rf, rf_gain));
float srate = srslte_rf_set_rx_srate(&rf, rf_rate);
if (srate != rf_rate) {
if (srate < 10e6) {
2015-12-16 09:02:25 -08:00
srslte_rf_set_master_clock_rate(&rf, 4*rf_rate);
} else {
2015-12-16 09:02:25 -08:00
srslte_rf_set_master_clock_rate(&rf, rf_rate);
}
2015-12-16 09:02:25 -08:00
srate = srslte_rf_set_rx_srate(&rf, rf_rate);
if (srate != rf_rate) {
fprintf(stderr, "Errror setting samplign frequency %.2f MHz\n", rf_rate*1e-6);
exit(-1);
}
}
printf("Correctly RX rate: %.2f MHz\n", srate*1e-6);
2015-12-16 09:02:25 -08:00
srslte_rf_rx_wait_lo_locked(&rf);
2017-11-23 10:46:34 -08:00
srslte_rf_start_rx_stream(&rf, false);
2015-03-24 08:11:14 -07:00
while((sample_count < nof_samples || nof_samples == -1)
&& keep_running){
2017-07-25 04:17:36 -07:00
n = srslte_rf_recv_with_time_multi(&rf, (void**) buffer, buflen, true, NULL, NULL);
2015-03-24 08:11:14 -07:00
if (n < 0) {
fprintf(stderr, "Error receiving samples\n");
exit(-1);
}
2017-07-25 04:17:36 -07:00
srslte_filesink_write_multi(&sink, (void**) buffer, buflen, nof_rx_antennas);
2015-03-24 08:11:14 -07:00
sample_count += buflen;
}
2017-07-25 04:17:36 -07:00
for (int i = 0; i < nof_rx_antennas; i++) {
if (buffer[i]) {
free(buffer[i]);
}
}
2015-03-24 08:11:14 -07:00
srslte_filesink_free(&sink);
2015-12-16 09:02:25 -08:00
srslte_rf_close(&rf);
2015-03-24 08:11:14 -07:00
printf("Ok - wrote %d samples\n", sample_count);
exit(0);
}