sdrangel/sdrbase/dsp/sigmffilerecord.h

108 wiersze
4.3 KiB
C++

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2015-2016, 2018-2021 Edouard Griffiths, F4EXB <f4exb06@gmail.com> //
// Copyright (C) 2018 beta-tester <alpha-beta-release@gmx.net> //
// Copyright (C) 2021 Jon Beniston, M7RCE <jon@beniston.com> //
// //
// File recorder in SigMF format single channel for SI plugins //
// //
// This program is free software; you can redistribute it and/or modify //
// it under the terms of the GNU General Public License as published by //
// the Free Software Foundation as version 3 of the License, or //
// (at your option) any later version. //
// //
// This program is distributed in the hope that it will be useful, //
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
// GNU General Public License V3 for more details. //
// //
// You should have received a copy of the GNU General Public License //
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
///////////////////////////////////////////////////////////////////////////////////
#ifndef INCLUDE_SIGMF_FILERECORD_H
#define INCLUDE_SIGMF_FILERECORD_H
#include <string>
#include <iostream>
#include <fstream>
#include <ctime>
#include <QDateTime>
#include "dsp/sigmf_forward.h"
#include "dsp/filerecordinterface.h"
#include "export.h"
class Message;
class SDRBASE_API SigMFFileRecord : public FileRecordInterface {
public:
SigMFFileRecord();
SigMFFileRecord(const QString& filename, const QString& hardwareId);
virtual ~SigMFFileRecord();
virtual void feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end, bool positiveOnly) override;
virtual void start() override;
virtual void stop() override;
virtual bool handleMessage(const Message& message) override;
virtual void setFileName(const QString& filename) override;
virtual const QString& getCurrentFileName() override { return m_fileName; }
virtual bool startRecording() override;
virtual bool stopRecording() override;
virtual bool isRecording() const override { return m_recordOn; }
void setHardwareId(const QString& hardwareId) { m_hardwareId = hardwareId; }
void setMsShift(qint64 msShift) override { m_msShift = msShift; }
unsigned int getNbCaptures() const;
uint64_t getInitialMsCount() const { return m_initialMsCount; }
uint64_t getInitialBytesCount() const { return m_initialBytesCount; }
void setLog2RecordSampleSize(uint32_t log2RecordSampleSize) { m_log2RecordSampleSize = log2RecordSampleSize; }
private:
struct Sample8 {
qint8 m_real;
qint8 m_imag;
};
struct Sample16 {
qint16 m_real;
qint16 m_imag;
};
struct Sample32 {
qint32 m_real;
qint32 m_imag;
};
QString m_hardwareId;
QString m_fileName;
QString m_sampleFileName;
QString m_metaFileName;
quint32 m_sampleRate;
quint64 m_centerFrequency;
qint64 m_msShift;
bool m_recordOn;
bool m_recordStart;
QDateTime m_captureStartDT;
std::ofstream m_metaFile;
std::ofstream m_sampleFile;
quint64 m_sampleStart;
quint64 m_sampleCount;
quint64 m_initialMsCount;
quint64 m_initialBytesCount;
uint32_t m_log2RecordSampleSize;
sigmf::SigMF<sigmf::Global<core::DescrT, sdrangel::DescrT>,
sigmf::Capture<core::DescrT, sdrangel::DescrT>,
sigmf::Annotation<core::DescrT> > *m_metaRecord;
std::vector<Sample8> m_samples8;
std::vector<Sample16> m_samples16;
std::vector<Sample32> m_samples32;
void makeHeader();
void makeCapture();
void clearMeta();
void feedConv(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end);
};
#endif // INCLUDE_SIGMF_FILERECORD_H