1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
|
#include "complex_to_mag.h"
#include <volk/volk.h>
namespace dsp
{
ComplexToMagBlock::ComplexToMagBlock(std::shared_ptr<dsp::stream<complex_t>> input) : Block(input)
{
}
ComplexToMagBlock::~ComplexToMagBlock()
{
}
void ComplexToMagBlock::work()
{
int nsamples = input_stream->read();
if (nsamples <= 0)
{
input_stream->flush();
return;
}
volk_32fc_magnitude_32f_u(output_stream->writeBuf, (lv_32fc_t *)input_stream->readBuf, nsamples);
input_stream->flush();
output_stream->swap(nsamples);
}
}
|