2015-07-28 21:54:17 +00:00
|
|
|
#ifndef INCLUDE_FILESINK_H
|
|
|
|
#define INCLUDE_FILESINK_H
|
|
|
|
|
|
|
|
#include <string>
|
|
|
|
#include <iostream>
|
|
|
|
#include <fstream>
|
|
|
|
|
|
|
|
#include <ctime>
|
|
|
|
#include "dsp/samplesink.h"
|
|
|
|
#include "util/export.h"
|
|
|
|
#include "util/message.h"
|
2015-08-13 06:51:33 +00:00
|
|
|
#include "util/messagequeue.h"
|
2015-07-28 21:54:17 +00:00
|
|
|
|
2015-08-29 23:26:51 +00:00
|
|
|
class SDRANGEL_API FileSink : public SampleSink {
|
2015-07-28 21:54:17 +00:00
|
|
|
public:
|
|
|
|
|
|
|
|
struct Header
|
|
|
|
{
|
|
|
|
int sampleRate;
|
|
|
|
quint64 centerFrequency;
|
|
|
|
std::time_t startTimeStamp;
|
|
|
|
};
|
|
|
|
|
|
|
|
FileSink();
|
2015-08-13 06:51:33 +00:00
|
|
|
virtual ~FileSink();
|
2015-07-28 21:54:17 +00:00
|
|
|
|
|
|
|
quint64 getByteCount() const { return m_byteCount; }
|
|
|
|
|
2015-08-14 03:00:28 +00:00
|
|
|
void configure(MessageQueue* msgQueue, const std::string& filename);
|
2015-07-28 21:54:17 +00:00
|
|
|
|
2015-08-25 06:24:23 +00:00
|
|
|
virtual void feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end, bool positiveOnly);
|
2015-08-13 06:51:33 +00:00
|
|
|
virtual void start();
|
|
|
|
virtual void stop();
|
2015-08-14 03:00:28 +00:00
|
|
|
virtual bool handleMessage(const Message& message);
|
2015-07-28 21:54:17 +00:00
|
|
|
void startRecording();
|
|
|
|
void stopRecording();
|
|
|
|
static void readHeader(std::ifstream& samplefile, Header& header);
|
|
|
|
|
|
|
|
private:
|
|
|
|
class MsgConfigureFileSink : public Message {
|
|
|
|
MESSAGE_CLASS_DECLARATION
|
|
|
|
|
|
|
|
public:
|
|
|
|
const std::string& getFileName() const { return m_fileName; }
|
|
|
|
|
2015-08-13 06:51:33 +00:00
|
|
|
static MsgConfigureFileSink* create(const std::string& fileName)
|
2015-07-28 21:54:17 +00:00
|
|
|
{
|
2015-08-13 06:51:33 +00:00
|
|
|
return new MsgConfigureFileSink(fileName);
|
2015-07-28 21:54:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
std::string m_fileName;
|
|
|
|
|
2015-08-13 06:51:33 +00:00
|
|
|
MsgConfigureFileSink(const std::string& fileName) :
|
2015-07-28 21:54:17 +00:00
|
|
|
Message(),
|
2015-08-13 06:51:33 +00:00
|
|
|
m_fileName(fileName)
|
2015-07-28 21:54:17 +00:00
|
|
|
{ }
|
|
|
|
};
|
|
|
|
|
|
|
|
std::string m_fileName;
|
|
|
|
int m_sampleRate;
|
|
|
|
quint64 m_centerFrequency;
|
|
|
|
bool m_recordOn;
|
|
|
|
bool m_recordStart;
|
|
|
|
std::ofstream m_sampleFile;
|
|
|
|
quint64 m_byteCount;
|
|
|
|
|
2015-08-14 03:00:28 +00:00
|
|
|
void handleConfigure(const std::string& fileName);
|
2015-07-28 21:54:17 +00:00
|
|
|
void writeHeader();
|
|
|
|
};
|
|
|
|
|
|
|
|
#endif // INCLUDE_FILESINK_H
|