summaryrefslogtreecommitdiffstats
path: root/plugins/radio/radio-rtlsdr.c
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/radio/radio-rtlsdr.c')
-rw-r--r--plugins/radio/radio-rtlsdr.c194
1 files changed, 99 insertions, 95 deletions
diff --git a/plugins/radio/radio-rtlsdr.c b/plugins/radio/radio-rtlsdr.c
index 45db7f7b..49efd2b9 100644
--- a/plugins/radio/radio-rtlsdr.c
+++ b/plugins/radio/radio-rtlsdr.c
@@ -16,25 +16,30 @@
*/
#include <stdio.h>
+#include <stdlib.h>
#include <string.h>
+#include <unistd.h>
#include "radio-api.h"
#include "radio-rtlsdr.h"
+static unsigned int init_dev_count = 0;
+static struct dev_ctx **dev_ctx = NULL;
+
/* ------------- RADIO RTLSDR IMPLEMENTATION ---------------- */
/* --- PUBLIC FUNCTIONS --- */
/* Radio initialization should be done only when user start the radio and not at plugin initialization
Making this call too early would impose to restart the binder to detect a radio */
-PUBLIC unsigned char _radio_on (unsigned int num, radioCtxHandleT *ctx) {
+unsigned char _radio_on (unsigned int num, radioCtxHandleT *ctx) {
if (num >= _radio_dev_count())
return 0;
if (init_dev_count < _radio_dev_count()) {
init_dev_count = _radio_dev_count();
- dev_ctx = (dev_ctx_T**) realloc (dev_ctx, init_dev_count * sizeof(dev_ctx_T));
+ dev_ctx = (dev_ctx_T**) realloc (dev_ctx, init_dev_count * sizeof(dev_ctx_T*));
}
dev_ctx[num] = (dev_ctx_T*) malloc (sizeof(dev_ctx_T));
@@ -46,95 +51,95 @@ PUBLIC unsigned char _radio_on (unsigned int num, radioCtxHandleT *ctx) {
dev_ctx[num]->dongle = NULL;
dev_ctx[num]->demod = NULL;
dev_ctx[num]->output = NULL;
- _radio_dev_init(dev_ctx[num], num);
+ _radio_dev_init (dev_ctx[num], num);
return 1;
}
-PUBLIC void _radio_off (unsigned int num) {
+void _radio_off (unsigned int num) {
if (num >= _radio_dev_count())
return;
if (dev_ctx[num]) {
- _radio_dev_free(dev_ctx[num]);
- free(dev_ctx[num]);
+ _radio_dev_free (dev_ctx[num]);
+ free (dev_ctx[num]);
}
/* free(dev_ctx); */
}
-PUBLIC void _radio_set_mode (unsigned int num, Mode mode) {
+void _radio_set_mode (unsigned int num, Mode mode) {
if (!dev_ctx || !dev_ctx[num])
return;
dev_ctx[num]->mode = mode;
- _radio_apply_params(dev_ctx[num]);
+ _radio_apply_params (dev_ctx[num]);
}
-PUBLIC void _radio_set_freq (unsigned int num, double freq) {
+void _radio_set_freq (unsigned int num, double freq) {
if (!dev_ctx || !dev_ctx[num])
return;
dev_ctx[num]->freq = (float)freq;
- _radio_apply_params(dev_ctx[num]);
+ _radio_apply_params (dev_ctx[num]);
}
-PUBLIC void _radio_set_mute (unsigned int num, unsigned char mute) {
+void _radio_set_mute (unsigned int num, unsigned char mute) {
if (!dev_ctx || !dev_ctx[num])
return;
dev_ctx[num]->mute = mute;
- _radio_apply_params(dev_ctx[num]);
+ _radio_apply_params (dev_ctx[num]);
}
-PUBLIC void _radio_play (unsigned int num) {
+void _radio_play (unsigned int num) {
if (!dev_ctx || !dev_ctx[num])
return;
- _radio_start_threads(dev_ctx[num]);
+ _radio_start_threads (dev_ctx[num]);
}
-PUBLIC void _radio_stop (unsigned int num) {
+void _radio_stop (unsigned int num) {
if (!dev_ctx || !dev_ctx[num])
return;
- _radio_stop_threads(dev_ctx[num]);
+ _radio_stop_threads (dev_ctx[num]);
}
-PUBLIC unsigned int _radio_dev_count () {
+unsigned int _radio_dev_count () {
return rtlsdr_get_device_count();
}
-PUBLIC const char* _radio_dev_name (unsigned int num) {
- return rtlsdr_get_device_name(num);
+const char* _radio_dev_name (unsigned int num) {
+ return rtlsdr_get_device_name (num);
}
/* --- LOCAL HELPER FUNCTIONS --- */
-STATIC unsigned char _radio_dev_init (dev_ctx_T *dev_ctx, unsigned int num) {
+unsigned char _radio_dev_init (dev_ctx_T *dev_ctx, unsigned int num) {
rtlsdr_dev_t *dev = dev_ctx->dev;
- if (rtlsdr_open(&dev, num) < 0)
+ if (rtlsdr_open (&dev, num) < 0)
return 0;
- rtlsdr_set_tuner_gain_mode(dev, 0);
+ rtlsdr_set_tuner_gain_mode (dev, 0);
- if (rtlsdr_reset_buffer(dev) < 0)
+ if (rtlsdr_reset_buffer (dev) < 0)
return 0;
dev_ctx->dev = dev;
- _radio_apply_params(dev_ctx);
+ _radio_apply_params (dev_ctx);
return 1;
}
-STATIC unsigned char _radio_dev_free (dev_ctx_T *dev_ctx) {
+unsigned char _radio_dev_free (dev_ctx_T *dev_ctx) {
rtlsdr_dev_t *dev = dev_ctx->dev;
- if (rtlsdr_close(dev) < 0)
+ if (rtlsdr_close (dev) < 0)
return 0;
dev = NULL;
@@ -143,7 +148,7 @@ STATIC unsigned char _radio_dev_free (dev_ctx_T *dev_ctx) {
return 1;
}
-STATIC void _radio_apply_params (dev_ctx_T *dev_ctx) {
+void _radio_apply_params (dev_ctx_T *dev_ctx) {
rtlsdr_dev_t *dev = dev_ctx->dev;
Mode mode = dev_ctx->mode;
float freq = dev_ctx->freq;
@@ -156,48 +161,47 @@ STATIC void _radio_apply_params (dev_ctx_T *dev_ctx) {
freq += 16000;
freq += rate / 4;
- rtlsdr_set_center_freq(dev, freq);
- rtlsdr_set_sample_rate(dev, rate);
+ rtlsdr_set_center_freq (dev, freq);
+ rtlsdr_set_sample_rate (dev, rate);
dev_ctx->dev = dev;
}
-STATIC void _radio_start_threads (dev_ctx_T *dev_ctx) {
- rtlsdr_dev_t *dev = dev_ctx->dev;
- dev_ctx->dongle = (dongle_ctx*) malloc(sizeof(dongle_ctx));
- dev_ctx->demod = (demod_ctx*) malloc(sizeof(demod_ctx));
- dev_ctx->output = (output_ctx*) malloc(sizeof(output_ctx));
+void _radio_start_threads (dev_ctx_T *dev_ctx) {
+ dev_ctx->dongle = (dongle_ctx*) malloc (sizeof(dongle_ctx));
+ dev_ctx->demod = (demod_ctx*) malloc (sizeof(demod_ctx));
+ dev_ctx->output = (output_ctx*) malloc (sizeof(output_ctx));
dongle_ctx *dongle = dev_ctx->dongle;
demod_ctx *demod = dev_ctx->demod;
output_ctx *output = dev_ctx->output;
- pthread_rwlock_init(&demod->lck, NULL);
- pthread_cond_init(&demod->ok, NULL);
- pthread_mutex_init(&demod->ok_m, NULL);
- pthread_rwlock_init(&output->lck, NULL);
- pthread_cond_init(&output->ok, NULL);
- pthread_mutex_init(&output->ok_m, NULL);
+ pthread_rwlock_init (&demod->lck, NULL);
+ pthread_cond_init (&demod->ok, NULL);
+ pthread_mutex_init (&demod->ok_m, NULL);
+ pthread_rwlock_init (&output->lck, NULL);
+ pthread_cond_init (&output->ok, NULL);
+ pthread_mutex_init (&output->ok_m, NULL);
dev_ctx->should_run = 1;
/* dongle thread */
dongle->thr_finished = 0;
- pthread_create(&dongle->thr, NULL, _dongle_thread_fn, (void*)dev_ctx);
+ pthread_create (&dongle->thr, NULL, _dongle_thread_fn, (void*)dev_ctx);
/* demod thread */
demod->pre_r = demod->pre_j = 0;
demod->now_r = demod->now_j = 0;
demod->index = demod->pre_index = demod->now_index = 0;
demod->thr_finished = 0;
- pthread_create(&demod->thr, NULL, _demod_thread_fn, (void*)dev_ctx);
+ pthread_create (&demod->thr, NULL, _demod_thread_fn, (void*)dev_ctx);
/* output thread */
output->thr_finished = 0;
- pthread_create(&output->thr, NULL, _output_thread_fn, (void*)dev_ctx);
+ pthread_create (&output->thr, NULL, _output_thread_fn, (void*)dev_ctx);
}
-STATIC void _radio_stop_threads (dev_ctx_T *dev_ctx) {
+void _radio_stop_threads (dev_ctx_T *dev_ctx) {
rtlsdr_dev_t *dev = dev_ctx->dev;
dongle_ctx *dongle = dev_ctx->dongle;
demod_ctx *demod = dev_ctx->demod;
@@ -209,33 +213,33 @@ STATIC void _radio_stop_threads (dev_ctx_T *dev_ctx) {
/* stop each "while" loop in threads */
dev_ctx->should_run = 0;
- rtlsdr_cancel_async(dev);
- pthread_signal(&demod->ok, &demod->ok_m);
- pthread_signal(&output->ok, &output->ok_m);
+ rtlsdr_cancel_async (dev);
+ pthread_signal (&demod->ok, &demod->ok_m);
+ pthread_signal (&output->ok, &output->ok_m);
while (!dongle->thr_finished ||
!demod->thr_finished ||
!output->thr_finished)
- usleep(100000);
-
- pthread_join(dongle->thr, NULL);
- pthread_join(demod->thr, NULL);
- pthread_join(output->thr, NULL);
- pthread_rwlock_destroy(&demod->lck);
- pthread_cond_destroy(&demod->ok);
- pthread_mutex_destroy(&demod->ok_m);
- pthread_rwlock_destroy(&output->lck);
- pthread_cond_destroy(&output->ok);
- pthread_mutex_destroy(&output->ok_m);
-
- free(dongle); dev_ctx->dongle = NULL;
- free(demod); dev_ctx->demod = NULL;
- free(output); dev_ctx->output = NULL;
+ usleep (100000);
+
+ pthread_join (dongle->thr, NULL);
+ pthread_join (demod->thr, NULL);
+ pthread_join (output->thr, NULL);
+ pthread_rwlock_destroy (&demod->lck);
+ pthread_cond_destroy (&demod->ok);
+ pthread_mutex_destroy (&demod->ok_m);
+ pthread_rwlock_destroy (&output->lck);
+ pthread_cond_destroy (&output->ok);
+ pthread_mutex_destroy (&output->ok_m);
+
+ free (dongle); dev_ctx->dongle = NULL;
+ free (demod); dev_ctx->demod = NULL;
+ free (output); dev_ctx->output = NULL;
}
/* ---- LOCAL THREADED FUNCTIONS ---- */
-STATIC void _rtlsdr_callback (unsigned char *buf, uint32_t len, void *ctx) {
+static void _rtlsdr_callback (unsigned char *buf, uint32_t len, void *ctx) {
dev_ctx_T *dev_ctx = (dev_ctx_T *)ctx;
dongle_ctx *dongle = dev_ctx->dongle;
demod_ctx *demod = dev_ctx->demod;
@@ -264,24 +268,24 @@ STATIC void _rtlsdr_callback (unsigned char *buf, uint32_t len, void *ctx) {
dongle->buf[i] = (int16_t)buf[i] - 127;
/* lock demod thread, write to it, unlock */
- pthread_rwlock_wrlock(&demod->lck);
- memcpy(demod->buf, dongle->buf, 2 * len);
+ pthread_rwlock_wrlock (&demod->lck);
+ memcpy (demod->buf, dongle->buf, 2 * len);
demod->buf_len = len;
- pthread_rwlock_unlock(&demod->lck);
- pthread_signal(&demod->ok, &demod->ok_m);
+ pthread_rwlock_unlock (&demod->lck);
+ pthread_signal (&demod->ok, &demod->ok_m);
}
/**/
-STATIC void* _dongle_thread_fn (void *ctx) {
+static void* _dongle_thread_fn (void *ctx) {
dev_ctx_T *dev_ctx = (dev_ctx_T *)ctx;
dongle_ctx *dongle = dev_ctx->dongle;
- rtlsdr_read_async(dev_ctx->dev, _rtlsdr_callback, dev_ctx, 0, 0);
+ rtlsdr_read_async (dev_ctx->dev, _rtlsdr_callback, dev_ctx, 0, 0);
dongle->thr_finished = 1;
return 0;
}
-STATIC void _lowpass_demod (void *ctx) {
+static void _lowpass_demod (void *ctx) {
demod_ctx *demod = (demod_ctx *)ctx;
int i=0, i2=0;
@@ -301,7 +305,7 @@ STATIC void _lowpass_demod (void *ctx) {
demod->buf_len = i2;
}
/**/
-STATIC void _lowpassreal_demod (void *ctx) {
+static void _lowpassreal_demod (void *ctx) {
demod_ctx *demod = (demod_ctx *)ctx;
int i=0, i2=0;
int fast = 200000;
@@ -321,30 +325,30 @@ STATIC void _lowpassreal_demod (void *ctx) {
demod->res_len = i2;
}
/**/
-STATIC void _multiply (int ar, int aj, int br, int bj, int *cr, int *cj) {
+static void _multiply (int ar, int aj, int br, int bj, int *cr, int *cj) {
*cr = ar*br - aj*bj;
*cj = aj*br + ar*bj;
}
/**/
-STATIC int _polar_discriminant (int ar, int aj, int br, int bj) {
+static int _polar_discriminant (int ar, int aj, int br, int bj) {
int cr, cj;
double angle;
- _multiply(ar, aj, br, -bj, &cr, &cj);
- angle = atan2((double)cj, (double)cr);
+ _multiply (ar, aj, br, -bj, &cr, &cj);
+ angle = atan2 ((double)cj, (double)cr);
return (int)(angle / 3.14159 * (1<<14));
}
/**/
-STATIC void _fm_demod (void *ctx) {
+static void _fm_demod (void *ctx) {
demod_ctx *demod = (demod_ctx *)ctx;
int16_t *buf = demod->buf;
int buf_len = demod->buf_len;
int pcm, i;
- pcm = _polar_discriminant(buf[0], buf[1], demod->pre_r, demod->pre_j);
+ pcm = _polar_discriminant (buf[0], buf[1], demod->pre_r, demod->pre_j);
demod->res[0] = (int16_t)pcm;
for (i = 2; i < (buf_len-1); i += 2) {
- pcm = _polar_discriminant(buf[i], buf[i+1], buf[i-2], buf[i-1]);
+ pcm = _polar_discriminant (buf[i], buf[i+1], buf[i-2], buf[i-1]);
demod->res[i/2] = (int16_t)pcm;
}
demod->pre_r = buf[buf_len - 2];
@@ -352,7 +356,7 @@ STATIC void _fm_demod (void *ctx) {
demod->res_len = buf_len/2;
}
/**/
-STATIC void _am_demod (void *ctx) {
+static void _am_demod (void *ctx) {
demod_ctx *demod = (demod_ctx *)ctx;
int16_t *buf = demod->buf;
int buf_len = demod->buf_len;
@@ -366,35 +370,35 @@ STATIC void _am_demod (void *ctx) {
demod->res_len = buf_len/2;
}
/**/
-STATIC void* _demod_thread_fn (void *ctx) {
+static void* _demod_thread_fn (void *ctx) {
dev_ctx_T *dev_ctx = (dev_ctx_T *)ctx;
demod_ctx *demod = dev_ctx->demod;
output_ctx *output = dev_ctx->output;
while(dev_ctx->should_run) {
- pthread_wait(&demod->ok, &demod->ok_m);
- pthread_rwlock_wrlock(&demod->lck);
- _lowpass_demod(demod);
+ pthread_wait (&demod->ok, &demod->ok_m);
+ pthread_rwlock_wrlock (&demod->lck);
+ _lowpass_demod (demod);
if (dev_ctx->mode == FM)
- _fm_demod(demod);
+ _fm_demod (demod);
else
- _am_demod(demod);
- _lowpassreal_demod(demod);
- pthread_rwlock_unlock(&demod->lck);
+ _am_demod (demod);
+ _lowpassreal_demod (demod);
+ pthread_rwlock_unlock (&demod->lck);
/* lock demod thread, write to it, unlock */
- pthread_rwlock_wrlock(&output->lck);
- memcpy(output->buf, demod->res, 2 * demod->res_len);
+ pthread_rwlock_wrlock (&output->lck);
+ memcpy (output->buf, demod->res, 2 * demod->res_len);
output->buf_len = demod->res_len;
- pthread_rwlock_unlock(&output->lck);
- pthread_signal(&output->ok, &output->ok_m);
+ pthread_rwlock_unlock (&output->lck);
+ pthread_signal (&output->ok, &output->ok_m);
}
demod->thr_finished = 1;
return 0;
}
-STATIC void* _output_thread_fn (void *ctx) {
+static void* _output_thread_fn (void *ctx) {
dev_ctx_T *dev_ctx = (dev_ctx_T *)ctx;
output_ctx *output = dev_ctx->output;
FILE *file;
@@ -402,14 +406,14 @@ STATIC void* _output_thread_fn (void *ctx) {
file = fopen (AUDIO_BUFFER, "wb");
while (dev_ctx->should_run) {
- pthread_wait(&output->ok, &output->ok_m);
- pthread_rwlock_rdlock(&output->lck);
+ pthread_wait (&output->ok, &output->ok_m);
+ pthread_rwlock_rdlock (&output->lck);
if (!dev_ctx->mute && file) {
fwrite (output->buf, 2, output->buf_len, file);
fflush (file);
fseek (file, 0, SEEK_SET);
}
- pthread_rwlock_unlock(&output->lck);
+ pthread_rwlock_unlock (&output->lck);
}
if (file) fclose(file);
unlink (AUDIO_BUFFER);