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:
parent
e2551648b9
commit
77c682e8b3
@ -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_;
|
|
||||||
}
|
|
||||||
|
@ -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;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
/** \} */
|
/** \} */
|
||||||
|
Loading…
x
Reference in New Issue
Block a user