diff options
Diffstat (limited to 'plugins/radio/radio-rtlsdr.c')
-rw-r--r-- | plugins/radio/radio-rtlsdr.c | 194 |
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); |