rtlsource.hh

#ifndef __SDR_RTLSOURCE_HH__
#define __SDR_RTLSOURCE_HH__

#include <rtl-sdr.h>
#include <pthread.h>
#include "node.hh"


namespace sdr {

class RTLSource: public Source
{
public:
  RTLSource(double frequency, double sample_rate=1e6, size_t device_idx=0);

  virtual ~RTLSource();

  inline double frequency() const { return _frequency; }
  void setFrequency(double frequency);

  inline double freqCorrection() const { return rtlsdr_get_freq_correction(_device); }
  void setFreqCorrection(double ppm);

  inline double sampleRate() const { return _sample_rate; }
  void setSampleRate(double sample_rate);

  inline bool agcEnabled() const { return _agc_enabled; }
  void enableAGC(bool enable);

  inline double gain() const { return rtlsdr_get_tuner_gain(_device); }
  void setGain(double gain);
  inline const std::vector<double> & gainFactors() const { return _gains; }

  void start();
  void stop();

  static size_t numDevices();
  static std::string deviceName(size_t idx);

protected:
  static void *__rtl_sdr_parallel_main(void *ctx);
  static void __rtl_sdr_callback(unsigned char *buffer, uint32_t len, void *ctx);

protected:
  double _frequency;
  double _sample_rate;
  bool _agc_enabled;
  std::vector<double> _gains;
  size_t _buffer_size;
  rtlsdr_dev_t *_device;
  pthread_t _thread;
};


}

#endif // __SDR_RTLSOURCE_HH__