From ee22af7a379404072461e5e636155d306bb6ae3f Mon Sep 17 00:00:00 2001 From: manjaro Date: Tue, 19 Jan 2021 11:03:14 +0000 Subject: [PATCH] demo save to file --- uhd_cpp/uhd_io_cpp.cpp | 6 +- uhd_cpp/uhd_io_save.cpp | 283 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 287 insertions(+), 2 deletions(-) create mode 100644 uhd_cpp/uhd_io_save.cpp diff --git a/uhd_cpp/uhd_io_cpp.cpp b/uhd_cpp/uhd_io_cpp.cpp index acea90b..b6042b9 100644 --- a/uhd_cpp/uhd_io_cpp.cpp +++ b/uhd_cpp/uhd_io_cpp.cpp @@ -17,6 +17,7 @@ #include #include #include +#include using uhd::tune_request_t; using uhd::tx_streamer; using uhd::usrp::multi_usrp; @@ -28,7 +29,8 @@ using std::string; using std::vector; using std::shared_ptr; -static bool stop_signal_called = false; + +static std::atomic stop_signal_called (false); void sigint_handler(int code){ (void)code; stop_signal_called = true; @@ -82,7 +84,7 @@ void do_io( vec_buffersz.push_back(0); } //收发计数 - long long rx_count = 0, tx_count = 0; + std::atomic rx_count (0), tx_count (0); //接收线程 auto thcall_rx = [&]()->void{ uhd::rx_metadata_t md_rx; diff --git a/uhd_cpp/uhd_io_save.cpp b/uhd_cpp/uhd_io_save.cpp new file mode 100644 index 0000000..cff4dbb --- /dev/null +++ b/uhd_cpp/uhd_io_save.cpp @@ -0,0 +1,283 @@ +/* + * Copyright 2015 Ettus Research LLC + * Copyright 2018 Ettus Research, a National Instruments Company + * + * SPDX-License-Identifier: GPL-3.0-or-later + * 丁劲犇修改 2021 + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +using uhd::tune_request_t; +using uhd::tx_streamer; +using uhd::usrp::multi_usrp; +using uhd::rx_streamer; +using std::thread; +using std::cerr; +using std::endl; +using std::string; +using std::vector; +using std::shared_ptr; + +static std::atomic stop_signal_called (false); +void sigint_handler(int code){ + (void)code; + stop_signal_called = true; +} +/*! + * \brief The tag_channelOptions struct + * 通道配置参数 + */ +struct tag_channelOptions{ + string type = "sc16"; //样点类型，为上位机上类型: fc64, fc32, or sc16 + string subdev=""; //通道，A:A, A:B两个通道，对应B210左侧、右侧接口.一般默认配置，用信道号channels选取接口。 + string ant = "TX/RX"; //天线选取，B210 有 TX/RX 或者 RX2两个 + string wirefmt = "sc16"; //内部类型 (sc8 or sc16)，是片上处理的类型 + vector channels {0};//通道号,可以设置0,1之类的。默认subdev时，0=A:A，1=A:B，subdev被修改，则采取修改后的编号 + size_t spb = 20000; //缓冲大小，太小会丢包，太大会超时 + double rate = 50e6; //采样率，单位Hz + double freq = 1.0e9; //射频频率，单位Hz + double gain = 55; //射频增益，单位dB + double bw = 56e6; //滤波带宽，默认为采样窗口 + double lo_offset = 0; //LO偏移，单位 Hz (缺省) + bool int_n_mod = false; //int-n 模式（本例不配置） + bool docheck = true; //在开始前执行检查 + double setup_time = 1.0; //rx配置检查时间，可选。 +}; +//通道检查函数 +bool check_rx_status(const string & ref, multi_usrp::sptr usrp,const tag_channelOptions & op); + + + +/*! + * 范例落盘函数,使用环形队列保持跟随收发 + */ +template +void do_save( + const tag_channelOptions & oprx, + rx_streamer::sptr rx) +{ + if (oprx.channels.size()>1) + { + cerr << "multi channels IO is not suitable for this simple demo."< > vec_buffer; + vector< int > vec_buffersz; + const int bufsz = 65536; + for (int i=0;i (new FMT[oprx.spb * 2])); + vec_buffersz.push_back(0); + } + //收发计数 + std::atomic rx_count (0), tx_count (0); + //接收线程 + auto thcall_rx = [&]()->void{ + uhd::rx_metadata_t md_rx; + uhd::stream_cmd_t stream_cmd(uhd::stream_cmd_t::STREAM_MODE_START_CONTINUOUS); + stream_cmd.num_samps = size_t(oprx.spb); + stream_cmd.stream_now = true; + stream_cmd.time_spec = uhd::time_spec_t(); + rx->issue_stream_cmd(stream_cmd); + while (!stop_signal_called) + { + vec_buffersz[rx_count % bufsz] = rx->recv((void *)(vec_buffer[rx_count % bufsz].get()),oprx.spb,md_rx,0.1,false); + //md_rx可以读取时戳 + //auto tm_first = md_rx.time_spec; + ++rx_count; + if (md_rx.error_code == uhd::rx_metadata_t::ERROR_CODE_TIMEOUT) + fputs("Time out.",stderr); + else if (md_rx.error_code == uhd::rx_metadata_t::ERROR_CODE_OVERFLOW) + { + fputs("Over flow",stderr); + stop_signal_called = true; + } + else if (md_rx.error_code != uhd::rx_metadata_t::ERROR_CODE_NONE) + { + cerr << "Receiver error: "<< md_rx.strerror() << endl ; + stop_signal_called = true; + } + } + + }; + //存储线程 + auto th_save = [&]()->void{ + + FILE * fp = fopen ("/run/media/user/SYSTEM/56Msps_IQ_F2450MHz.pcm","wb"); + while (!stop_signal_called) + { + //等待一会数据，以便TX可以保持跟随。也可以不设置，这样开始时会争夺。 + if (tx_count + 10 >=rx_count) + continue; + if (tx_count + 65500 <=rx_count) + stop_signal_called = true; + //fwrite((void *)(vec_buffer[tx_count % bufsz].get()),sizeof(FMT)*2,vec_buffersz[tx_count % bufsz],fp); + ++tx_count; + } + fclose(fp); + }; + + //启动线程 + thread rx_thread(thcall_rx); + thread tx_thread(th_save); + cerr<<"Press ^C to Stop."<set_clock_source(ref,multi_usrp::ALL_MBOARDS); + + + //3.配置接收 + tag_channelOptions rx_op; + rx_op.ant = "RX2"; + rx_op.rate = 50e6; + rx_op.bw = 50e6; + rx_op.freq = 2450e6; + rx_op.gain = 35; + rx_op.channels[0] = 0; + //3.1 子设备 + if (rx_op.subdev.size()) + usrp->set_rx_subdev_spec(rx_op.subdev,multi_usrp::ALL_MBOARDS); + cerr << "RX Using Device: " << usrp->get_pp_string() << endl; + //3.2 采样率 + cerr << "Setting RX Rate: " << (rx_op.rate / 1e6) << "Msps..." << endl; + usrp->set_rx_rate(rx_op.rate,multi_usrp::ALL_CHANS); + cerr << "Actual RX Rate: " << usrp->get_rx_rate(rx_op.channels[0]) / 1e6 << "Msps..." << endl; + //3.3 中心频率 + cerr << "Setting RX Freq: " << (rx_op.freq / 1e6) <<"MHz..." << endl; + cerr << "Setting RX LO Offset: " << (rx_op.lo_offset / 1e6) << "MHz..." <set_rx_freq(tune_request_rx,rx_op.channels[0]); + cerr << "Actual RX Freq: " << (usrp->get_rx_freq(rx_op.channels[0]) / 1e6) << "MHz..." << endl; + //3.4 增益 + cerr << "Setting RX Gain: " << rx_op.gain <<" dB..." << endl; + usrp->set_rx_gain(rx_op.gain,rx_op.channels[0]); + cerr << "Actual RX Gain: " << usrp->get_rx_gain(rx_op.channels[0]) << " dB..." << endl; + //3.5 前端模拟滤波带宽 + cerr << "Setting RX Bandwidth: " << (rx_op.bw / 1e6) << "MHz..." << endl; + usrp->set_rx_bandwidth(rx_op.bw,rx_op.channels[0]); + cerr << "Actual RX Bandwidth: " << usrp->get_rx_bandwidth(rx_op.channels[0]) / 1e6 << "MHz..." << endl; + //3.6 选择天线 + if (rx_op.ant.size()) + usrp->set_rx_antenna(rx_op.ant,rx_op.channels[0]); + + //4 检查状态 + if (rx_op.docheck) check_rx_status(ref,usrp,rx_op); + + //5.创建流对象实例 + uhd::stream_args_t stream_args_rx(rx_op.type, rx_op.wirefmt); + stream_args_rx.channels = rx_op.channels; + rx_streamer::sptr rx_stream = usrp->get_rx_stream(stream_args_rx); + + //开始收发循环 + do_save(rx_op,rx_stream); + + // finished + cerr << endl << "Done!" << endl << endl; + + return EXIT_SUCCESS; +} + + +typedef std::function get_sensor_fn_t; +bool check_locked_sensor(vector sensor_names, + const char* sensor_name, + get_sensor_fn_t get_sensor_fn, + double setup_time) +{ + if (std::find(sensor_names.begin(), sensor_names.end(), sensor_name) + == sensor_names.end()) + return false; + + auto setup_timeout = std::chrono::steady_clock::now() + + std::chrono::milliseconds(int64_t(setup_time * 1000)); + bool lock_detected = false; + + std::cerr << "Checking RX Waiting for: " << sensor_name; + std::cerr.flush(); + + while (true) { + if (lock_detected and (std::chrono::steady_clock::now() > setup_timeout)) { + std::cerr << " locked." << std::endl; + break; + } + if (get_sensor_fn(sensor_name).to_bool()) { + std::cerr << "+"; + std::cerr.flush(); + lock_detected = true; + } else { + if (std::chrono::steady_clock::now() > setup_timeout) { + std::cerr << std::endl; + std::cerr << "timed out waiting for consecutive locks on sensor : "<get_rx_sensor_names(c), + "lo_locked", + [usrp, op, c](const string& sensor_name) { + return usrp->get_rx_sensor(sensor_name, c); + }, + op.setup_time); + } + if (ref == "mimo") { + check_locked_sensor(usrp->get_mboard_sensor_names(0), + "mimo_locked", + [usrp](const string& sensor_name) { + return usrp->get_mboard_sensor(sensor_name); + }, + op.setup_time); + } + if (ref == "external") { + check_locked_sensor(usrp->get_mboard_sensor_names(0), + "ref_locked", + [usrp](const string& sensor_name) { + return usrp->get_mboard_sensor(sensor_name); + }, + op.setup_time); + } + return true; +} -- GitLab