add gain.

IQTLabs · Apr 3, 2024 · 6f39df6 · 6f39df6
1 parent f767fde
commit 6f39df6
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/lib/sample_pipeline.cpp b/lib/sample_pipeline.cpp
@@ -261,12 +261,12 @@ void sample_pipeline_start(const std::string &file, const std::string &fft_file,
 }
 
 void sample_pipeline_stop(size_t overflows, const std::string &file,
-                          size_t rate, size_t freq, double timestamp) {
+                          size_t rate, size_t freq, double timestamp, double gain) {
   samples_input_done = true;
   writer_threads->join_all();
   sample_writer->close(overflows);
   fft_sample_writer->close(overflows);
-  sample_writer->write_sigmf(file + ".sigmf-meta", timestamp, "s16", rate, freq, 0);
+  sample_writer->write_sigmf(file + ".sigmf-meta", timestamp, "s16", rate, freq, gain);
   if (useVkFFT) {
     free_vkfft();
   }

diff --git a/lib/sample_pipeline.h b/lib/sample_pipeline.h
@@ -11,6 +11,6 @@ void sample_pipeline_start(const std::string &file, const std::string &fft_file,
                            size_t sample_id);
 size_t get_samp_size();
 void sample_pipeline_stop(size_t overflows, const std::string &file,
-                          size_t rate, size_t freq, double timestamp);
+                          size_t rate, size_t freq, double timestamp, double gain);
 void set_sample_pipeline_types(const std::string &type,
                                std::string &cpu_format);
diff --git a/lib/sample_pipeline_test.cpp b/lib/sample_pipeline_test.cpp
@@ -10,7 +10,7 @@ BOOST_AUTO_TEST_CASE(SmokeTest) {
   set_sample_pipeline_types("short", cpu_format);
   BOOST_TEST(cpu_format == "sc16");
   sample_pipeline_start("", "", 1e6, 1, false, 0, 0, 1, 1, 1e6, 0, 0);
-  sample_pipeline_stop(0, "", 1e6, 1e6, 1.1);
+  sample_pipeline_stop(0, "", 1e6, 1e6, 1.1, -1);
 }
 
 BOOST_AUTO_TEST_CASE(RandomFFTTest) {
@@ -33,7 +33,7 @@ BOOST_AUTO_TEST_CASE(RandomFFTTest) {
   memcpy(buffer_p, samples.memptr(),
          samples.size() * sizeof(std::complex<float>));
   enqueue_samples(write_ptr);
-  sample_pipeline_stop(0, file, 1e3 * 1024, 100e6, 1.1);
+  sample_pipeline_stop(0, file, 1e3 * 1024, 100e6, 1.1, -1);
   arma::Col<std::complex<float>> disk_samples;
   disk_samples.copy_size(samples);
   FILE *samples_fp = fopen(file.c_str(), "rb");

diff --git a/lib/uhd_sample_recorder.cpp b/lib/uhd_sample_recorder.cpp
@@ -205,7 +205,7 @@ void sample_record(uhd::usrp::multi_usrp::sptr usrp, const std::string &type,
   stream_cmd.stream_mode = uhd::stream_cmd_t::STREAM_MODE_STOP_CONTINUOUS;
   rx_stream->issue_stream_cmd(stream_cmd);
   std::cerr << "stream stopped" << std::endl;
-  sample_pipeline_stop(overflows, file, rate, freq, timestamp);
+  sample_pipeline_stop(overflows, file, rate, freq, timestamp, gain);
   std::cerr << "pipeline stopped" << std::endl;
 }