Fix DC offset removal, simplify...
This commit is contained in:
parent
5f264e023d
commit
4b9f377bb0
1 changed files with 76 additions and 127 deletions
157
src/main.c
157
src/main.c
|
@ -21,8 +21,8 @@
|
|||
#include <limits.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#define VREG_VOLTAGE VREG_VOLTAGE_1_20
|
||||
#define CLK_SYS_HZ (320 * MHZ)
|
||||
#define VREG_VOLTAGE VREG_VOLTAGE_1_30
|
||||
#define CLK_SYS_HZ (300 * MHZ)
|
||||
|
||||
#define RX_PIN 10
|
||||
#define FB_PIN 6
|
||||
|
@ -55,9 +55,9 @@ static uint32_t lo_sin[LO_WORDS] __attribute__((__aligned__(1 << LO_BITS_DEPTH))
|
|||
static int frequency = INIT_FREQ;
|
||||
static int sample_rate = INIT_SAMPLE_RATE;
|
||||
|
||||
#define BASE_GAIN (1 << 15)
|
||||
#define ATTN_BITS 12
|
||||
#define DECIMATE 4
|
||||
#define ATTN_BITS 16
|
||||
#define BASE_GAIN (1 << 23)
|
||||
#define DC_OFFSET (int)(127.4 * (1 << ATTN_BITS))
|
||||
|
||||
static int gain = BASE_GAIN / (CLK_SYS_HZ / INIT_SAMPLE_RATE);
|
||||
|
||||
|
@ -108,7 +108,7 @@ static void init_rx()
|
|||
pio_gpio_init(PIO, RX_PIN);
|
||||
|
||||
const uint16_t insn[] = {
|
||||
pio_encode_in(pio_pins, 1) | pio_encode_delay(0),
|
||||
pio_encode_in(pio_pins, 1),
|
||||
};
|
||||
|
||||
pio_program_t prog = {
|
||||
|
@ -143,16 +143,8 @@ static void init_bias()
|
|||
gpio_set_slew_rate(FB_PIN, GPIO_SLEW_RATE_SLOW);
|
||||
|
||||
const uint16_t insn[] = {
|
||||
pio_encode_mov(pio_isr, pio_null),
|
||||
pio_encode_in(pio_y, 4),
|
||||
pio_encode_in(pio_pins, 1) | pio_encode_delay(15),
|
||||
pio_encode_in(pio_pins, 1) | pio_encode_delay(15),
|
||||
|
||||
pio_encode_mov(pio_y, pio_isr),
|
||||
pio_encode_mov(pio_x, pio_isr),
|
||||
|
||||
pio_encode_jmp_x_dec(6),
|
||||
pio_encode_mov_not(pio_pins, pio_pins) | pio_encode_sideset(1, 1),
|
||||
pio_encode_mov_not(pio_pins, pio_pins) | pio_encode_sideset(1, 1) |
|
||||
pio_encode_delay(0),
|
||||
};
|
||||
|
||||
pio_program_t prog = {
|
||||
|
@ -426,92 +418,50 @@ inline static void led_set(bool on)
|
|||
gpio_put(PICO_DEFAULT_LED_PIN, on);
|
||||
}
|
||||
|
||||
inline static uint32_t pio_sm_get_blocking_unsafe(pio_hw_t *pio, int sm)
|
||||
{
|
||||
while (pio->fstat & (1u << (PIO_FSTAT_RXEMPTY_LSB + sm)))
|
||||
asm volatile("nop");
|
||||
|
||||
return pio->rxf[sm];
|
||||
}
|
||||
|
||||
static int Ia1, Ia2, Ia3;
|
||||
static int Ib1, Ib2, Ib3;
|
||||
|
||||
static int Qa1, Qa2, Qa3;
|
||||
static int Qb1, Qb2, Qb3;
|
||||
|
||||
inline static void accI()
|
||||
{
|
||||
static uint16_t py, px;
|
||||
|
||||
uint32_t yx = pio_sm_get_blocking_unsafe(PIO, SM_COS);
|
||||
uint16_t y = yx >> 16;
|
||||
uint16_t x = yx;
|
||||
|
||||
uint16_t ny = py - y;
|
||||
uint16_t nx = px - x;
|
||||
|
||||
py = y;
|
||||
px = x;
|
||||
|
||||
uint32_t s = (ny << 1) + nx;
|
||||
|
||||
Ia1 += s;
|
||||
Ia2 += Ia1;
|
||||
Ia3 += Ia2;
|
||||
}
|
||||
|
||||
inline static void accQ()
|
||||
{
|
||||
static uint16_t py, px;
|
||||
|
||||
uint32_t yx = pio_sm_get_blocking_unsafe(PIO, SM_SIN);
|
||||
uint16_t y = yx >> 16;
|
||||
uint16_t x = yx;
|
||||
|
||||
uint16_t ny = py - y;
|
||||
uint16_t nx = px - x;
|
||||
|
||||
py = y;
|
||||
px = x;
|
||||
|
||||
uint32_t s = (ny << 1) + nx;
|
||||
|
||||
Qa1 += s;
|
||||
Qa2 += Qa1;
|
||||
Qa3 += Qa2;
|
||||
}
|
||||
|
||||
inline static int getI()
|
||||
{
|
||||
uint32_t c1, c2, c3;
|
||||
static uint16_t py, px;
|
||||
|
||||
c3 = Ia3 - Ib3;
|
||||
Ib3 = Ia3;
|
||||
uint32_t yx = pio_sm_get_blocking(PIO, SM_COS);
|
||||
uint16_t y = yx >> 16;
|
||||
uint16_t x = yx;
|
||||
|
||||
c2 = c3 - Ib2;
|
||||
Ib2 = c3;
|
||||
uint16_t ny = py - y;
|
||||
uint16_t nx = px - x;
|
||||
|
||||
c1 = c2 - Ib1;
|
||||
Ib1 = c2;
|
||||
py = y;
|
||||
px = x;
|
||||
|
||||
return (int)c1;
|
||||
int s = ((ny << 1) + nx) * gain;
|
||||
|
||||
static int dc;
|
||||
dc += (s - dc) >> ATTN_BITS;
|
||||
s -= dc;
|
||||
|
||||
return (s + DC_OFFSET) >> ATTN_BITS;
|
||||
}
|
||||
|
||||
inline static int getQ()
|
||||
{
|
||||
uint32_t c1, c2, c3;
|
||||
static uint16_t py, px;
|
||||
|
||||
c3 = Qa3 - Qb3;
|
||||
Qb3 = Qa3;
|
||||
uint32_t yx = pio_sm_get_blocking(PIO, SM_SIN);
|
||||
uint16_t y = yx >> 16;
|
||||
uint16_t x = yx;
|
||||
|
||||
c2 = c3 - Qb2;
|
||||
Qb2 = c3;
|
||||
uint16_t ny = py - y;
|
||||
uint16_t nx = px - x;
|
||||
|
||||
c1 = c2 - Qb1;
|
||||
Qb1 = c2;
|
||||
py = y;
|
||||
px = x;
|
||||
|
||||
return (int)c1;
|
||||
int s = ((ny << 1) + nx) * gain;
|
||||
|
||||
static int dc;
|
||||
dc += (s - dc) >> ATTN_BITS;
|
||||
s -= dc;
|
||||
|
||||
return (s + DC_OFFSET) >> ATTN_BITS;
|
||||
}
|
||||
|
||||
static void rf_rx(void)
|
||||
|
@ -527,19 +477,21 @@ static void rf_rx(void)
|
|||
uint8_t *blockptr = block;
|
||||
|
||||
for (int i = 0; i < IQ_SAMPLES; i++) {
|
||||
for (int d = 0; d < DECIMATE; d++) {
|
||||
accI();
|
||||
accQ();
|
||||
}
|
||||
int I = getI();
|
||||
int Q = getQ();
|
||||
|
||||
int I = getI() * gain;
|
||||
int Q = getQ() * gain;
|
||||
if (I < 0)
|
||||
I = 0;
|
||||
else if (I > 255)
|
||||
I = 255;
|
||||
|
||||
I >>= ATTN_BITS;
|
||||
*blockptr++ = I + 128;
|
||||
if (Q < 0)
|
||||
Q = 0;
|
||||
else if (Q > 255)
|
||||
Q = 255;
|
||||
|
||||
Q >>= ATTN_BITS;
|
||||
*blockptr++ = Q + 128;
|
||||
*blockptr++ = I;
|
||||
*blockptr++ = Q;
|
||||
}
|
||||
|
||||
if (queue_try_add(&iq_queue, &block)) {
|
||||
|
@ -561,7 +513,7 @@ static void run_command(uint8_t cmd, uint32_t arg)
|
|||
/* Set the rate at which IQ sample pairs are sent */
|
||||
sample_rate = arg;
|
||||
gain = BASE_GAIN / (CLK_SYS_HZ / sample_rate);
|
||||
dma_timer_set_fraction(dma_t_samp, 1, CLK_SYS_HZ / (sample_rate * DECIMATE));
|
||||
dma_timer_set_fraction(dma_t_samp, 1, CLK_SYS_HZ / sample_rate);
|
||||
rx_lo_init(frequency);
|
||||
}
|
||||
}
|
||||
|
@ -613,10 +565,7 @@ static void do_rx()
|
|||
fwrite(block, IQ_BLOCK_LEN, 1, stdout);
|
||||
fflush(stdout);
|
||||
} else {
|
||||
int wait = xorshift() >> (32 - 15);
|
||||
|
||||
for (int i = 0; i < wait; i++)
|
||||
asm volatile("nop");
|
||||
sleep_us(1);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -657,7 +606,7 @@ int main()
|
|||
|
||||
/* We need to have the sampling timer ready. */
|
||||
dma_t_samp = dma_claim_unused_timer(true);
|
||||
dma_timer_set_fraction(dma_t_samp, 1, CLK_SYS_HZ / (sample_rate * DECIMATE));
|
||||
dma_timer_set_fraction(dma_t_samp, 1, CLK_SYS_HZ / sample_rate);
|
||||
|
||||
while (true) {
|
||||
if (check_command() > 0) {
|
||||
|
|
Loading…
Reference in a new issue