diff options
Diffstat (limited to 'src/main.c')
| -rw-r--r-- | src/main.c | 358 |
1 files changed, 37 insertions, 321 deletions
@@ -24,303 +24,12 @@ #include <math.h> #include <unistd.h> -#include <libusb.h> - -#include "tuner_e4000.h" -#include "tuner_fc0013.h" +#include "rtl-sdr.h" #define READLEN (16 * 16384) -#define CTRL_IN (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_IN) -#define CTRL_OUT (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT) - -/* ezcap USB 2.0 DVB-T/DAB/FM stick */ -#define EZCAP_VID 0x0bda -#define EZCAP_PID 0x2838 - -/* Terratec NOXON DAB/DAB+ USB-Stick */ -#define NOXON_VID 0x0ccd -#define NOXON_PID 0x00b3 - -#define CRYSTAL_FREQ 28800000 -static struct libusb_device_handle *devh = NULL; static int do_exit = 0; -enum TUNER_TYPE { - TUNER_E4000, - TUNER_FC0013 -} tuner_type; - -static int find_device(void) -{ - devh = libusb_open_device_with_vid_pid(NULL, EZCAP_VID, EZCAP_PID); - if (devh > 0) { - tuner_type = TUNER_E4000; - printf("Found ezcap stick with E4000 tuner\n"); - return 0; - } - - devh = libusb_open_device_with_vid_pid(NULL, NOXON_VID, NOXON_PID); - if (devh > 0) { - tuner_type = TUNER_FC0013; - printf("Found Terratec NOXON stick with FC0013 tuner\n"); - return 0; - } - - return -EIO; -} - -enum usb_reg { - USB_SYSCTL = 0x2000, - USB_CTRL = 0x2010, - USB_STAT = 0x2014, - USB_EPA_CFG = 0x2144, - USB_EPA_CTL = 0x2148, - USB_EPA_MAXPKT = 0x2158, - USB_EPA_MAXPKT_2 = 0x215a, - USB_EPA_FIFO_CFG = 0x2160, -}; - -enum sys_reg { - DEMOD_CTL = 0x3000, - GPO = 0x3001, - GPI = 0x3002, - GPOE = 0x3003, - GPD = 0x3004, - SYSINTE = 0x3005, - SYSINTS = 0x3006, - GP_CFG0 = 0x3007, - GP_CFG1 = 0x3008, - SYSINTE_1 = 0x3009, - SYSINTS_1 = 0x300a, - DEMOD_CTL_1 = 0x300b, - IR_SUSPEND = 0x300c, -}; - -enum blocks { - DEMODB = 0, - USBB = 1, - SYSB = 2, - TUNB = 3, - ROMB = 4, - IRB = 5, - IICB = 6, -}; - -int rtl_read_array(uint8_t block, uint16_t addr, uint8_t *array, uint8_t len) -{ - int r; - uint16_t index = (block << 8); - - r = libusb_control_transfer(devh, CTRL_IN, 0, addr, index, array, len, 0); - - return r; -} - -int rtl_write_array(uint8_t block, uint16_t addr, uint8_t *array, uint8_t len) -{ - int r; - uint16_t index = (block << 8) | 0x10; - - r = libusb_control_transfer(devh, CTRL_OUT, 0, addr, index, array, len, 0); - - return r; -} - -int rtl_i2c_write(uint8_t i2c_addr, uint8_t *buffer, int len) -{ - uint16_t addr = i2c_addr; - return rtl_write_array(IICB, addr, buffer, len); -} - -int rtl_i2c_read(uint8_t i2c_addr, uint8_t *buffer, int len) -{ - uint16_t addr = i2c_addr; - return rtl_read_array(IICB, addr, buffer, len); -} - -uint16_t rtl_read_reg(uint8_t block, uint16_t addr, uint8_t len) -{ - int r; - unsigned char data[2]; - uint16_t index = (block << 8); - uint16_t reg; - - r = libusb_control_transfer(devh, CTRL_IN, 0, addr, index, data, len, 0); - - if (r < 0) - printf("%s failed\n", __FUNCTION__); - - reg = (data[1] << 8) | data[0]; - - return reg; -} - -void rtl_write_reg(uint8_t block, uint16_t addr, uint16_t val, uint8_t len) -{ - int r; - unsigned char data[2]; - - uint16_t index = (block << 8) | 0x10; - - if (len == 1) - data[0] = val & 0xff; - else - data[0] = val >> 8; - - data[1] = val & 0xff; - - r = libusb_control_transfer(devh, CTRL_OUT, 0, addr, index, data, len, 0); - - if (r < 0) - printf("%s failed\n", __FUNCTION__); -} - -uint16_t demod_read_reg(uint8_t page, uint8_t addr, uint8_t len) -{ - int r; - unsigned char data[2]; - - uint16_t index = page; - uint16_t reg; - addr = (addr << 8) | 0x20; - - r = libusb_control_transfer(devh, CTRL_IN, 0, addr, index, data, len, 0); - - if (r < 0) - printf("%s failed\n", __FUNCTION__); - - reg = (data[1] << 8) | data[0]; - - return reg; -} - -void demod_write_reg(uint8_t page, uint16_t addr, uint16_t val, uint8_t len) -{ - int r; - unsigned char data[2]; - uint16_t index = 0x10 | page; - addr = (addr << 8) | 0x20; - - if (len == 1) - data[0] = val & 0xff; - else - data[0] = val >> 8; - - data[1] = val & 0xff; - - r = libusb_control_transfer(devh, CTRL_OUT, 0, addr, index, data, len, 0); - - if (r < 0) - printf("%s failed\n", __FUNCTION__); - - demod_read_reg(0x0a, 0x01, 1); -} - -void set_samp_rate(uint32_t samp_rate) -{ - uint16_t tmp; - uint32_t rsamp_ratio; - double real_rate; - - /* check for the maximum rate the resampler supports */ - if (samp_rate > 3200000) - samp_rate = 3200000; - - rsamp_ratio = (CRYSTAL_FREQ * pow(2, 22)) / samp_rate; - rsamp_ratio &= ~3; - - real_rate = (CRYSTAL_FREQ * pow(2, 22)) / rsamp_ratio; - printf("Setting sample rate: %.3f Hz\n", real_rate); - - tmp = (rsamp_ratio >> 16); - demod_write_reg(1, 0x9f, tmp, 2); - tmp = rsamp_ratio & 0xffff; - demod_write_reg(1, 0xa1, tmp, 2); -} - -void set_i2c_repeater(int on) -{ - demod_write_reg(1, 0x01, on ? 0x18 : 0x10, 1); -} - -void rtl_init(void) -{ - unsigned int i; - - /* default FIR coefficients used for DAB/FM by the Windows driver, - * the DVB driver uses different ones */ - uint8_t fir_coeff[] = { - 0xca, 0xdc, 0xd7, 0xd8, 0xe0, 0xf2, 0x0e, 0x35, 0x06, 0x50, - 0x9c, 0x0d, 0x71, 0x11, 0x14, 0x71, 0x74, 0x19, 0x41, 0x00, - }; - - /* initialize USB */ - rtl_write_reg(USBB, USB_SYSCTL, 0x09, 1); - rtl_write_reg(USBB, USB_EPA_MAXPKT, 0x0002, 2); - rtl_write_reg(USBB, USB_EPA_CTL, 0x1002, 2); - - /* poweron demod */ - rtl_write_reg(SYSB, DEMOD_CTL_1, 0x22, 1); - rtl_write_reg(SYSB, DEMOD_CTL, 0xe8, 1); - - /* reset demod (bit 3, soft_rst) */ - demod_write_reg(1, 0x01, 0x14, 1); - demod_write_reg(1, 0x01, 0x10, 1); - - /* disable spectrum inversion and adjacent channel rejection */ - demod_write_reg(1, 0x15, 0x00, 1); - demod_write_reg(1, 0x16, 0x0000, 2); - - /* set IF-frequency to 0 Hz */ - demod_write_reg(1, 0x19, 0x0000, 2); - - /* set FIR coefficients */ - for (i = 0; i < sizeof (fir_coeff); i++) - demod_write_reg(1, 0x1c + i, fir_coeff[i], 1); - - demod_write_reg(0, 0x19, 0x25, 1); - - /* init FSM state-holding register */ - demod_write_reg(1, 0x93, 0xf0, 1); - - /* disable AGC (en_dagc, bit 0) */ - demod_write_reg(1, 0x11, 0x00, 1); - - /* disable PID filter (enable_PID = 0) */ - demod_write_reg(0, 0x61, 0x60, 1); - - /* opt_adc_iq = 0, default ADC_I/ADC_Q datapath */ - demod_write_reg(0, 0x06, 0x80, 1); - - /* Enable Zero-IF mode (en_bbin bit), DC cancellation (en_dc_est), - * IQ estimation/compensation (en_iq_comp, en_iq_est) */ - demod_write_reg(1, 0xb1, 0x1b, 1); -} - -void tuner_init(int frequency) -{ - set_i2c_repeater(1); - - switch (tuner_type) { - case TUNER_E4000: - e4000_Initialize(1); - e4000_SetBandwidthHz(1, 8000000); - e4000_SetRfFreqHz(1, frequency); - break; - case TUNER_FC0013: - FC0013_Open(); - FC0013_SetFrequency(frequency/1000, 8); - break; - default: - printf("No valid tuner available!"); - break; - } - - printf("Tuned to %i Hz\n", frequency); - set_i2c_repeater(0); -} - void usage(void) { printf("rtl-sdr, an I/Q recorder for RTL2832 based USB-sticks\n\n" @@ -342,8 +51,9 @@ int main(int argc, char **argv) char *filename; uint32_t frequency = 0, samp_rate = 2048000; uint8_t buffer[READLEN]; - int n_read; + uint32_t n_read; FILE *file; + rtlsdr_dev_t *dev = NULL; while ((opt = getopt(argc, argv, "f:s:")) != -1) { switch (opt) { @@ -365,22 +75,20 @@ int main(int argc, char **argv) filename = argv[optind]; } - r = libusb_init(NULL); - if (r < 0) { - fprintf(stderr, "Failed to initialize libusb\n"); - exit(1); - } + rtlsdr_init(); - r = find_device(); - if (r < 0) { - fprintf(stderr, "Could not find/open device\n"); - goto out; - } + int device_count = rtlsdr_get_device_count(); + if (!device_count) { + fprintf(stderr, "No supported devices found.\n"); + exit(1); + } - r = libusb_claim_interface(devh, 0); - if (r < 0) { - fprintf(stderr, "usb_claim_interface error %d\n", r); - goto out; + printf("Found %d device(s).\n", device_count); + + dev = rtlsdr_open(0); /* open the first device */ + if (NULL == dev) { + fprintf(stderr, "Failed to open rtlsdr device.\n"); + exit(1); } sigact.sa_handler = sighandler; @@ -390,12 +98,17 @@ int main(int argc, char **argv) sigaction(SIGTERM, &sigact, NULL); sigaction(SIGQUIT, &sigact, NULL); - /* Initialize the RTL2832 */ - rtl_init(); - set_samp_rate(samp_rate); + /* Set the sample rate */ + r = rtlsdr_set_sample_rate(dev, samp_rate); + if (r < 0) { + fprintf(stderr, "WARNING: Failed to set sample rate.\n"); + } - /* Initialize tuner & set frequency */ - tuner_init(frequency); + /* Set the frequency */ + r = rtlsdr_set_center_freq(dev, frequency); + if (r < 0) { + fprintf(stderr, "WARNING: Failed to set center freq.\n"); + } file = fopen(filename, "wb"); @@ -404,27 +117,30 @@ int main(int argc, char **argv) goto out; } - /* reset endpoint before we start reading */ - rtl_write_reg(USBB, USB_EPA_CTL, 0x1002, 2); - rtl_write_reg(USBB, USB_EPA_CTL, 0x0000, 2); + /* Reset endpoint before we start reading from it */ + r = rtlsdr_reset_buffer(dev); + if (r < 0) { + fprintf(stderr, "WARNING: Failed to reset buffers.\n"); + } printf("Reading samples...\n"); while (!do_exit) { - libusb_bulk_transfer(devh, 0x81, buffer, READLEN, &n_read, 3000); + r = rtlsdr_read_sync(dev, buffer, READLEN, &n_read); + if (r < 0) { + fprintf(stderr, "WARNING: sync read failed.\n"); + } + fwrite(buffer, n_read, 1, file); if (n_read < READLEN) { - printf("Short bulk read, samples lost, exiting!\n"); + printf("Short read, samples lost, exiting!\n"); break; } } fclose(file); - libusb_release_interface(devh, 0); + rtlsdr_exit(); out: - libusb_close(devh); - libusb_exit(NULL); return r >= 0 ? r : -r; } - |