1
0
mirror of https://github.com/gnss-sdr/gnss-sdr synced 2025-04-29 22:23:13 +00:00

Removed unnecessary buffer

This commit is contained in:
Stefan van der Linden 2022-02-08 12:51:51 +01:00
parent e2551648b9
commit 77c682e8b3
No known key found for this signature in database
GPG Key ID: C7576734F7DE90D4
2 changed files with 45 additions and 89 deletions

View File

@ -19,8 +19,6 @@
#include <glog/logging.h> #include <glog/logging.h>
#include <volk/volk.h> #include <volk/volk.h>
const int FIFO_SIZE = 1472000; // Value taken from gr_complex_ip_packet_source, is it optimal?
// initial construction; pass to private constructor // initial construction; pass to private constructor
FifoReader::sptr FifoReader::make(const std::string &file_name, const std::string &sample_type) FifoReader::sptr FifoReader::make(const std::string &file_name, const std::string &sample_type)
{ {
@ -33,10 +31,7 @@ FifoReader::FifoReader(const std::string &file_name, const std::string &sample_t
gr::io_signature::make(0, 0, 0), // no input gr::io_signature::make(0, 0, 0), // no input
gr::io_signature::make(1, 1, sizeof(gr_complex))), // <+MIN_OUT+>, <+MAX_OUT+>, sizeof(<+OTYPE+>) gr::io_signature::make(1, 1, sizeof(gr_complex))), // <+MIN_OUT+>, <+MAX_OUT+>, sizeof(<+OTYPE+>)
file_name_(file_name), file_name_(file_name),
sample_type_(sample_type), sample_type_(sample_type)
fifo_buffer_(new char[FIFO_SIZE]),
buffer_idx_(0),
buffer_size_(0)
{ {
DLOG(INFO) << "Starting FifoReader"; DLOG(INFO) << "Starting FifoReader";
} }
@ -67,14 +62,17 @@ int FifoReader::work(int noutput_items,
size_t items_retrieved = 0; size_t items_retrieved = 0;
if (sample_type_ == "ishort") if (sample_type_ == "ishort")
{ {
// ishort == int16_t
items_retrieved = read_interleaved<int16_t>(noutput_items, output_items); items_retrieved = read_interleaved<int16_t>(noutput_items, output_items);
} }
else if (sample_type_ == "ibyte") // Does this also work with cbyte? else if (sample_type_ == "ibyte") // Does this also work with cbyte?
{ {
// ibyte == int8_t
items_retrieved = read_interleaved<int8_t>(noutput_items, output_items); items_retrieved = read_interleaved<int8_t>(noutput_items, output_items);
} }
else if (sample_type_ == "gr_complex") else if (sample_type_ == "gr_complex")
{ {
// gr_complex == complex<float>
items_retrieved = read_gr_complex(noutput_items, output_items); items_retrieved = read_gr_complex(noutput_items, output_items);
} }
else else
@ -89,91 +87,31 @@ int FifoReader::work(int noutput_items,
} }
template <typename Type>
size_t FifoReader::read_interleaved(int noutput_items, gr_vector_void_star &output_items)
{
boost::mutex::scoped_lock lock(d_mutex); // hold mutex for duration of this function
size_t items_retrieved = 0;
int n;
Type real;
Type imag;
if (buffer_idx_ >= buffer_size_)
{
fifo_.read(fifo_buffer_, FIFO_SIZE);
buffer_idx_ = 0;
if (fifo_.good())
{
buffer_size_ = FIFO_SIZE;
}
else if (fifo_.eof())
{
// Although we got an EOF, ensure we don't lose the other samples
buffer_size_ = fifo_.gcount();
fifo_.clear();
}
else
{
fifo_error_output();
return 0;
}
}
for (n = 0; n < noutput_items && buffer_idx_ < buffer_size_; n++)
{
memcpy(&real, &fifo_buffer_[buffer_idx_], sizeof(real));
memcpy(&imag, &fifo_buffer_[buffer_idx_ + sizeof(imag)], sizeof(imag));
static_cast<gr_complex *>(output_items.at(0))[n] = gr_complex(real, imag);
buffer_idx_ += 2 * sizeof(Type);
}
items_retrieved = n;
return items_retrieved;
}
// read gr_complex items from fifo // read gr_complex items from fifo
// this fct has duplicate code with the templated read_interleaved fct above // this fct has duplicate code with the templated read_interleaved fct in header
size_t FifoReader::read_gr_complex(int noutput_items, gr_vector_void_star &output_items) size_t FifoReader::read_gr_complex(int noutput_items, gr_vector_void_star &output_items)
{ {
boost::mutex::scoped_lock lock(d_mutex); // hold mutex for duration of this function
size_t items_retrieved = 0; size_t items_retrieved = 0;
int n; for (int n = 0; n < noutput_items; n++)
gr_complex sample;
if (buffer_idx_ >= buffer_size_)
{ {
fifo_.read(fifo_buffer_, FIFO_SIZE); gr_complex sample;
buffer_idx_ = 0; fifo_.read(reinterpret_cast<char *>(&sample), sizeof(sample));
if (fifo_.good()) if (fifo_.good())
{ {
buffer_size_ = FIFO_SIZE; static_cast<gr_complex *>(output_items.at(0))[n] = sample;
items_retrieved++;
} }
else if (fifo_.eof()) else if (fifo_.eof())
{ {
// Although we got an EOF, ensure we don't lose the other samples
buffer_size_ = fifo_.gcount();
fifo_.clear(); fifo_.clear();
break;
} }
else else
{ {
fifo_error_output(); fifo_error_output();
return 0; break;
} }
} }
for (n = 0; n < noutput_items && buffer_idx_ < buffer_size_; n++)
{
memcpy(&sample, &fifo_buffer_[buffer_idx_], sizeof(gr_complex));
static_cast<gr_complex *>(output_items.at(0))[n] = sample;
buffer_idx_ += sizeof(gr_complex);
}
items_retrieved = n;
return items_retrieved; return items_retrieved;
} }
@ -181,9 +119,3 @@ void FifoReader::fifo_error_output() const
{ {
LOG(ERROR) << "unhandled FIFO event"; LOG(ERROR) << "unhandled FIFO event";
} }
FifoReader::~FifoReader()
{
delete[] fifo_buffer_;
}

View File

@ -19,7 +19,6 @@
#define GNSS_SDR_FIFO_READER_H_ #define GNSS_SDR_FIFO_READER_H_
#include "gnss_block_interface.h" #include "gnss_block_interface.h"
#include <boost/thread.hpp>
#include <gnuradio/sync_block.h> #include <gnuradio/sync_block.h>
#include <fstream> // std::ifstream #include <fstream> // std::ifstream
#include <string> #include <string>
@ -35,6 +34,8 @@ public:
using sptr = gnss_shared_ptr<FifoReader>; using sptr = gnss_shared_ptr<FifoReader>;
static sptr make(const std::string &file_name, const std::string &sample_type); static sptr make(const std::string &file_name, const std::string &sample_type);
~FifoReader() = default;
//! initialize istream resource for FIFO //! initialize istream resource for FIFO
bool start(); bool start();
@ -49,16 +50,42 @@ private:
//! (gr handles this with public and private header pair) //! (gr handles this with public and private header pair)
FifoReader(const std::string &file_name, const std::string &sample_type); FifoReader(const std::string &file_name, const std::string &sample_type);
//! \brief Destructor
//! private destructor
~FifoReader();
size_t read_gr_complex(int noutput_items, gr_vector_void_star &output_items); size_t read_gr_complex(int noutput_items, gr_vector_void_star &output_items);
//! function to read data out of FIFO which is stored as interleaved I/Q stream. //! function to read data out of FIFO which is stored as interleaved I/Q stream.
//! template argument determines sample_type //! template argument determines sample_type
// Note: template definition necessary in header file
// See also: https://stackoverflow.com/questions/495021/why-can-templates-only-be-implemented-in-the-header-file
template <typename Type> template <typename Type>
size_t read_interleaved(int noutput_items, gr_vector_void_star &output_items); size_t read_interleaved(int noutput_items, gr_vector_void_star &output_items)
{
size_t items_retrieved = 0;
for (int n = 0; n < noutput_items; n++)
{
// TODO: try if performance increases if we copy larger chunks to vector.
// how to read from stream: https://en.cppreference.com/w/cpp/io/basic_ifstream
std::array<char, 2 * sizeof(Type)> buffer;
fifo_.read(reinterpret_cast<char *>(buffer.data()), buffer.size());
if (fifo_.good())
{
auto real = reinterpret_cast<Type const *>(&buffer[0]);
auto imag = reinterpret_cast<Type const *>(&buffer[sizeof(Type)]);
static_cast<gr_complex *>(output_items[0])[n] = gr_complex(*real, *imag);
items_retrieved++;
}
else if (fifo_.eof())
{
fifo_.clear();
break;
}
else
{
fifo_error_output();
break;
}
}
return items_retrieved;
}
//! this function moves logging output from this header into the source file //! this function moves logging output from this header into the source file
//! thereby eliminating the need to include glog/logging.h in this header //! thereby eliminating the need to include glog/logging.h in this header
@ -67,9 +94,6 @@ private:
const std::string file_name_; const std::string file_name_;
const std::string sample_type_; const std::string sample_type_;
std::ifstream fifo_; std::ifstream fifo_;
char *fifo_buffer_;
int buffer_idx_, buffer_size_;
boost::mutex d_mutex;
}; };
/** \} */ /** \} */