hrtl2832reader.h

#ifndef __HRTL2832READER_H
#define __HRTL2832READER_H
 
#include <mutex>
#include <condition_variable>
#include <thread>
#include "rtl-sdr.h"
 
#include "hardt.h"
#include "hobject.h"
#include "hreader.h"
#include "hfft.h"
#include "hprobe.h"
#include "hcommand.h"
#include "hrtl2832.h"
 
#define NUMBER_OF_RTL_BUFFERS 15
 
template <class T>
class HRtl2832Reader : public HReader<T>
{
    public:
 
    private:
 
        HRtl2832::MODE _mode;
        int _buflen;
        HFft<T>* _fft;
        T* _buffer;
        std::thread* _current;
        rtlsdr_dev_t *_dev;
        size_t _blocksize;
        HProbe<T>* _probe;
 
        // Data accessed by the (static) callback function
        struct CallbackData
        {
            int wrloc;
            int rdloc;
            unsigned char* buffer;
            std::mutex mtx;
            std::condition_variable lock;
        } _cbd;
 
        // Indicate if the device has been started
        bool _isStarted;
 
        // Indicate if the device has been initialize
        bool _isInitialized;
 
        bool Start();
 
        bool Stop();
 
        void CopyBlock(T* dest, unsigned char* src, size_t blocksize);
 
    public:
 
        HRtl2832Reader(int device, H_SAMPLE_RATE rate, HRtl2832::MODE mode, int gain, int32_t frequency, int blocksize, bool offset = 0, int correction = 0, HProbe<T>* probe = nullptr);
 
        ~HRtl2832Reader();
 
        int Read(T* dest, size_t blocksize);
 
        static void callback( unsigned char *buffer, uint32_t length, void* context);
 
        bool Command(HCommand* command) {
 
            // Handle known commands
            switch( command->Class ) {
                case H_COMMAND_CLASS::TUNER: {
                    switch( command->Opcode ) {
                        case H_COMMAND_OPCODE::SET_FREQUENCY: {
                            return SetCenterFrequency(command->Data.Value);
                        }
                        case H_COMMAND_OPCODE::SET_GAIN: {
                            return SetGain(command->Data.Value);
                        }
                    }
                }
                case H_COMMAND_CLASS::ANY: {
                    switch( command->Opcode ) {
                        case H_COMMAND_OPCODE::SET_GAIN: {
                            return SetGain(command->Data.Value);
                        }
                    }
                }
            }
 
            // Any other command returns true
            return true;
        }
 
        bool EnableBiasTee(int gpio = 0);
 
        bool DisableBiasTee(int gpio = 0);
 
        bool SetCenterFrequency(uint32_t frequency);
 
        uint32_t GetCenterFrequency();
 
        bool SetGain(int gain);
 
        int GetGain();
};
 
#endif