mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-11-02 16:01:14 -04:00
127 lines
3.2 KiB
C++
127 lines
3.2 KiB
C++
#include "dsp/filesink.h"
|
|
#include "util/simpleserializer.h"
|
|
#include "util/messagequeue.h"
|
|
|
|
#include <iostream>
|
|
|
|
MESSAGE_CLASS_DEFINITION(FileSink::MsgConfigureFileSink, Message)
|
|
|
|
FileSink::FileSink() :
|
|
SampleSink(),
|
|
m_fileName(std::string("test.sdriq")),
|
|
m_sampleRate(0),
|
|
m_centerFrequency(0),
|
|
m_recordOn(false),
|
|
m_recordStart(false),
|
|
m_byteCount(0)
|
|
{
|
|
}
|
|
|
|
FileSink::~FileSink()
|
|
{
|
|
stopRecording();
|
|
}
|
|
|
|
void FileSink::configure(MessageQueue* msgQueue, const std::string& filename, int sampleRate, quint64 centerFrequency)
|
|
{
|
|
Message* cmd = MsgConfigureFileSink::create(filename, sampleRate, centerFrequency);
|
|
cmd->submit(msgQueue, this);
|
|
}
|
|
|
|
void FileSink::feed(SampleVector::const_iterator begin, SampleVector::const_iterator end, bool positiveOnly)
|
|
{
|
|
// if no recording is active, send the samples to /dev/null
|
|
if(!m_recordOn)
|
|
return;
|
|
|
|
if (begin < end) // if there is something to put out
|
|
{
|
|
if (m_recordStart)
|
|
{
|
|
writeHeader();
|
|
m_recordStart = false;
|
|
}
|
|
|
|
m_sampleFile.write(reinterpret_cast<const char*>(&*(begin)), (end - begin)*sizeof(Sample));
|
|
m_byteCount += end - begin;
|
|
}
|
|
}
|
|
|
|
void FileSink::start()
|
|
{
|
|
}
|
|
|
|
void FileSink::stop()
|
|
{
|
|
stopRecording();
|
|
}
|
|
|
|
void FileSink::startRecording()
|
|
{
|
|
if (!m_sampleFile.is_open())
|
|
{
|
|
std::cerr << "FileSink::startRecording" << std::endl;
|
|
m_sampleFile.open(m_fileName.c_str(), std::ios::binary);
|
|
m_recordOn = true;
|
|
m_recordStart = true;
|
|
m_byteCount = 0;
|
|
}
|
|
}
|
|
|
|
void FileSink::stopRecording()
|
|
{
|
|
if (m_sampleFile.is_open())
|
|
{
|
|
std::cerr << "FileSink::stopRecording" << std::endl;
|
|
m_sampleFile.close();
|
|
m_recordOn = false;
|
|
m_recordStart = false;
|
|
}
|
|
}
|
|
|
|
bool FileSink::handleMessage(Message* message)
|
|
{
|
|
if (MsgConfigureFileSink::match(message))
|
|
{
|
|
MsgConfigureFileSink* conf = (MsgConfigureFileSink*) message;
|
|
handleConfigure(conf->getFileName(), conf->getSampleRate(), conf->getCenterFrequency());
|
|
std::cerr << "FileSink::handleMessage:"
|
|
<< " fileName: " << m_fileName
|
|
<< " sampleRate: " << m_sampleRate
|
|
<< " centerFrequency: " << m_centerFrequency << std::endl;
|
|
message->completed();
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void FileSink::handleConfigure(const std::string& fileName, int sampleRate, quint64 centerFrequency)
|
|
{
|
|
if ((fileName != m_fileName) || (m_sampleRate != sampleRate) || (m_centerFrequency != centerFrequency))
|
|
{
|
|
stopRecording();
|
|
}
|
|
|
|
m_fileName = fileName;
|
|
m_sampleRate = sampleRate;
|
|
m_centerFrequency = centerFrequency;
|
|
}
|
|
|
|
void FileSink::writeHeader()
|
|
{
|
|
m_sampleFile.write((const char *) &m_sampleRate, sizeof(int));
|
|
m_sampleFile.write((const char *) &m_centerFrequency, sizeof(quint64));
|
|
std::time_t ts = time(0);
|
|
m_sampleFile.write((const char *) &ts, sizeof(std::time_t));
|
|
}
|
|
|
|
void FileSink::readHeader(std::ifstream& sampleFile, Header& header)
|
|
{
|
|
sampleFile.read((char *) &(header.sampleRate), sizeof(int));
|
|
sampleFile.read((char *) &(header.centerFrequency), sizeof(quint64));
|
|
sampleFile.read((char *) &(header.startTimeStamp), sizeof(std::time_t));
|
|
}
|