uhd: added a finite acquisition call to usrp source

1 files changed, 19 insertions, 0 deletions

diff --git a/gr-uhd/lib/gr_uhd_usrp_source.cc b/gr-uhd/lib/gr_uhd_usrp_source.cc
index 51a756908..2244238bd 100644
--- a/gr-uhd/lib/gr_uhd_usrp_source.cc
+++ b/gr-uhd/lib/gr_uhd_usrp_source.cc
@@ -409,6 +409,25 @@ public:
         return true;
     }
 
+    std::vector<std::complex<float> > finite_acquisition(const size_t nsamps){
+        #ifdef GR_UHD_USE_STREAM_API
+        uhd::stream_cmd_t cmd(uhd::stream_cmd_t::STREAM_MODE_NUM_SAMPS_AND_DONE);
+        cmd.num_samps = nsamps;
+        cmd.stream_now = true;
+        _dev->issue_stream_cmd(cmd);
+
+        std::vector<std::complex<float> > samps(nsamps);
+        const size_t actual_num_samps = _rx_stream->recv(
+            &samps.front(), nsamps, _metadata, 0.1
+        );
+        samps.resize(actual_num_samps);
+
+        return samps;
+        #else
+        throw std::runtime_error("not implemented in this version");
+        #endif
+    }
+
 private:
     uhd::usrp::multi_usrp::sptr _dev;
     const uhd::stream_args_t _stream_args;