Improve bit counting
This commit is contained in:
		
							parent
							
								
									0db366b602
								
							
						
					
					
						commit
						2068d3a01f
					
				
					 1 changed files with 37 additions and 38 deletions
				
			
		
							
								
								
									
										75
									
								
								src/main.c
									
									
									
									
									
								
							
							
						
						
									
										75
									
								
								src/main.c
									
									
									
									
									
								
							|  | @ -22,10 +22,10 @@ | |||
| #include <stdlib.h> | ||||
| 
 | ||||
| #define VREG_VOLTAGE VREG_VOLTAGE_1_20 | ||||
| #define CLK_SYS_HZ (300 * MHZ) | ||||
| #define CLK_SYS_HZ (288 * MHZ) | ||||
| 
 | ||||
| #define LO_PIN 9 | ||||
| #define RX_PIN 8 | ||||
| #define RX_PIN 13 | ||||
| #define FB_PIN 5 | ||||
| #define PSU_PIN 23 | ||||
| 
 | ||||
|  | @ -69,6 +69,7 @@ static int gains[NUM_GAINS] = { 0,   9,	  14,  27,  37,	 77,  87,  125, 144, 157 | |||
| 				166, 197, 207, 229, 254, 280, 297, 328, 338, 364, | ||||
| 				372, 386, 402, 421, 434, 439, 445, 480, 496 }; | ||||
| static int sample_rate = INIT_SAMPLE_RATE; | ||||
| static int dc_level = CLK_SYS_HZ / INIT_SAMPLE_RATE / 2; | ||||
| static int gain = INIT_GAIN; | ||||
| 
 | ||||
| static int dma_ch_rx1 = -1; | ||||
|  | @ -108,8 +109,8 @@ static void init_lo() | |||
| 	gpio_disable_pulls(LO_PIN); | ||||
| 	pio_gpio_init(PIO, LO_PIN); | ||||
| 
 | ||||
| 	gpio_set_drive_strength(LO_PIN, GPIO_DRIVE_STRENGTH_2MA); | ||||
| 	gpio_set_slew_rate(LO_PIN, GPIO_SLEW_RATE_SLOW); | ||||
| 	gpio_set_drive_strength(LO_PIN, GPIO_DRIVE_STRENGTH_12MA); | ||||
| 	gpio_set_slew_rate(LO_PIN, GPIO_SLEW_RATE_FAST); | ||||
| 
 | ||||
| 	const uint16_t insn[] = { | ||||
| 		pio_encode_out(pio_pindirs, 1), | ||||
|  | @ -218,23 +219,34 @@ static void init_rx() | |||
| static void init_ad() | ||||
| { | ||||
| 	const uint16_t insn[] = { | ||||
| 		pio_encode_jmp_y_dec(1), | ||||
| 		pio_encode_out(pio_pc, 2), | ||||
| 		pio_encode_out(pio_pc, 2), | ||||
| 		pio_encode_jmp_x_dec(2), | ||||
| 
 | ||||
| 		/* Avoid Y-- on wrap. */ | ||||
| 		pio_encode_out(pio_pc, 2), | ||||
| 		pio_encode_out(pio_pc, 4), // 0000 +0
 | ||||
| 		pio_encode_jmp_x_dec(0),   // 0001 +1
 | ||||
| 		pio_encode_jmp_x_dec(0),   // 0010 +1
 | ||||
| 		pio_encode_jmp_y_dec(0),   // 0011 +2
 | ||||
| 		pio_encode_jmp_x_dec(0),   // 0100 +1
 | ||||
| 		pio_encode_jmp_y_dec(0),   // 0101 +2
 | ||||
| 		pio_encode_jmp_y_dec(0),   // 0110 +2
 | ||||
| 		pio_encode_jmp_y_dec(1),   // 0111 +2 +1
 | ||||
| 		pio_encode_jmp_x_dec(0),   // 1000 +1
 | ||||
| 		pio_encode_jmp_y_dec(0),   // 1001 +2
 | ||||
| 		pio_encode_jmp_y_dec(0),   // 1010 +2
 | ||||
| 		pio_encode_jmp_y_dec(1),   // 1011 +2 +1
 | ||||
| 		pio_encode_jmp_y_dec(0),   // 1100 +2
 | ||||
| 		pio_encode_jmp_y_dec(1),   // 1101 +2 +1
 | ||||
| 		pio_encode_jmp_y_dec(1),   // 1110 +2 +1
 | ||||
| 		pio_encode_jmp_y_dec(3),   // 1111 +2 +2
 | ||||
| 
 | ||||
| 		/*
 | ||||
| 		 * Should wrap here. | ||||
| 		 * Jump to this portion must be inserted from the outside. | ||||
| 		 */ | ||||
| 		pio_encode_in(pio_x, 32), | ||||
| 		pio_encode_in(pio_y, 32), | ||||
| 		pio_encode_set(pio_x, 0), | ||||
| 		pio_encode_in(pio_x, 32), | ||||
| 		pio_encode_set(pio_y, 0), | ||||
| 		pio_encode_out(pio_pc, 2), | ||||
| 		pio_encode_set(pio_x, 0), | ||||
| 		pio_encode_jmp_y_dec(21), | ||||
| 		pio_encode_jmp_x_dec(22), | ||||
| 		pio_encode_out(pio_pc, 4), | ||||
| 	}; | ||||
| 
 | ||||
| 	pio_program_t prog = { | ||||
|  | @ -250,11 +262,11 @@ static void init_ad() | |||
| 		pio_add_program(PIO, &prog); | ||||
| 
 | ||||
| 	pio_sm_config pc = pio_get_default_sm_config(); | ||||
| 	sm_config_set_wrap(&pc, prog.origin, prog.origin + 5 - 1); | ||||
| 	sm_config_set_wrap(&pc, prog.origin, prog.origin + 15); | ||||
| 	sm_config_set_clkdiv_int_frac(&pc, 1, 0); | ||||
| 	sm_config_set_in_shift(&pc, false, true, 32); | ||||
| 	sm_config_set_out_shift(&pc, false, true, 32); | ||||
| 	pio_sm_init(PIO, AD_SM, prog.origin + 1, &pc); | ||||
| 	pio_sm_init(PIO, AD_SM, prog.origin, &pc); | ||||
| } | ||||
| 
 | ||||
| #define STEP_BASE ((UINT_MAX + 1.0) / CLK_SYS_HZ) | ||||
|  | @ -297,7 +309,7 @@ static void rx_lo_init(double req_freq, bool align) | |||
| 	} | ||||
| } | ||||
| 
 | ||||
| static const uint32_t samp_insn = 5; | ||||
| static const uint32_t samp_insn = 16; | ||||
| 
 | ||||
| static void rf_rx_start() | ||||
| { | ||||
|  | @ -363,10 +375,10 @@ static void rf_rx_start() | |||
| 	dma_channel_configure(dma_ch_samp_cos, &dma_conf, &PIO->sm[AD_SM].instr, &samp_insn, | ||||
| 			      UINT_MAX, false); | ||||
| 
 | ||||
| 	init_ad(); | ||||
| 	init_lo(); | ||||
| 	init_fb(); | ||||
| 	init_rx(); | ||||
| 	init_ad(); | ||||
| 
 | ||||
| 	dma_channel_start(dma_ch_rx1); | ||||
| 	dma_channel_start(dma_ch_mix1); | ||||
|  | @ -424,10 +436,10 @@ inline static struct IQ next_sample(const uint32_t *buf) | |||
| { | ||||
| 	static int h[11]; | ||||
| 
 | ||||
| 	int x3 = buf[0] - buf[1]; | ||||
| 	int x2 = buf[2] - buf[3]; | ||||
| 	int x1 = buf[5] - buf[4]; | ||||
| 	int x0 = buf[7] - buf[6]; | ||||
| 	int x3 = gain * (-2 * buf[0] - buf[1] - dc_level); | ||||
| 	int x2 = gain * (-2 * buf[2] - buf[3] - dc_level); | ||||
| 	int x1 = -(gain * (-2 * buf[4] - buf[5] - dc_level)); | ||||
| 	int x0 = -(gain * (-2 * buf[6] - buf[7] - dc_level)); | ||||
| 
 | ||||
| 	const int c[] = { 4, 2, -8, -9, 19, 55 }; | ||||
| 
 | ||||
|  | @ -449,10 +461,7 @@ inline static struct IQ next_sample(const uint32_t *buf) | |||
| 	I += (c[1] + c[0]) * x1; | ||||
| 	Q += (c[0]) * x0; | ||||
| 
 | ||||
| 	I *= gain; | ||||
| 	I /= 128; | ||||
| 
 | ||||
| 	Q *= gain; | ||||
| 	Q /= 128; | ||||
| 
 | ||||
| 	h[10] = h[6]; | ||||
|  | @ -502,23 +511,13 @@ static void rf_rx(void) | |||
| 		uint8_t *block = iq_queue_buffer[iq_queue_pos]; | ||||
| 		uint8_t *blockptr = block; | ||||
| 
 | ||||
| 		/*
 | ||||
| 		 * Since every 2 samples add to either +1 or -1, | ||||
| 		 * the maximum amplitude in one direction is 1/2. | ||||
| 		 * | ||||
| 		 * We are allowing the counters to only go as high | ||||
| 		 * as sampling rate. | ||||
| 		 */ | ||||
| 		int64_t max_amplitude = CLK_SYS_HZ / 2 / sample_rate; | ||||
| 
 | ||||
| 		for (int i = 0; i < IQ_SAMPLES; i++) { | ||||
| 			struct IQ IQ = next_sample(cos_ptr); | ||||
| 			int64_t I = IQ.I; | ||||
| 			int64_t Q = IQ.Q; | ||||
| 			cos_ptr += 2 * DECIMATE; | ||||
| 
 | ||||
| 			I -= (max_amplitude * 181) / 256; | ||||
| 			I /= max_amplitude; | ||||
| 			I /= dc_level; | ||||
| 
 | ||||
| 			if (I > 127) | ||||
| 				I = 127; | ||||
|  | @ -527,8 +526,7 @@ static void rf_rx(void) | |||
| 
 | ||||
| 			*blockptr++ = (uint8_t)I + 128; | ||||
| 
 | ||||
| 			Q -= (max_amplitude * 181) / 256; | ||||
| 			Q /= max_amplitude; | ||||
| 			Q /= dc_level; | ||||
| 
 | ||||
| 			if (Q > 127) | ||||
| 				Q = 127; | ||||
|  | @ -552,6 +550,7 @@ static void run_command(uint8_t cmd, uint32_t arg) | |||
| 	} else if (0x02 == cmd) { | ||||
| 		/* Set the rate at which IQ sample pairs are sent */ | ||||
| 		sample_rate = arg; | ||||
| 		dc_level = CLK_SYS_HZ / sample_rate / 2; | ||||
| 		dma_timer_set_fraction(dma_t_samp, 1, CLK_SYS_HZ / (sample_rate * DECIMATE)); | ||||
| 		rx_lo_init(arg + sample_rate, true); | ||||
| 	} else if (0x04 == cmd) { | ||||
|  |  | |||
		Loading…
	
		Reference in a new issue