283 lines
11 KiB
C
283 lines
11 KiB
C
|
/*---------------------------------------------------------------------------*\
|
||
|
|
|
||
|
|
FILE........: tfdmdv.c
|
||
|
|
AUTHOR......: David Rowe
|
||
|
|
DATE CREATED: April 16 2012
|
||
|
|
|
||
|
|
Tests for the C version of the FDMDV modem. This program outputs a
|
||
|
|
file of Octave vectors that are loaded and automatically tested
|
||
|
|
against the Octave version of the modem by the Octave script
|
||
|
|
tfmddv.m
|
||
|
|
|
||
|
|
\*---------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
Copyright (C) 2012 David Rowe
|
||
|
|
|
||
|
|
All rights reserved.
|
||
|
|
|
||
|
|
This program is free software; you can redistribute it and/or modify
|
||
|
|
it under the terms of the GNU Lesser General Public License version 2, as
|
||
|
|
published by the Free Software Foundation. This program is
|
||
|
|
distributed in the hope that it will be useful, but WITHOUT ANY
|
||
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||
|
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
|
||
|
|
License for more details.
|
||
|
|
|
||
|
|
You should have received a copy of the GNU Lesser General Public License
|
||
|
|
along with this program; if not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <assert.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <math.h>
|
||
|
|
|
||
|
|
#include "fdmdv_internal.h"
|
||
|
|
#include "fdmdv.h"
|
||
|
|
#include "octave.h"
|
||
|
|
|
||
|
|
#define FRAMES 25
|
||
|
|
#define CHANNEL_BUF_SIZE (10*M)
|
||
|
|
|
||
|
|
int main(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
struct FDMDV *fdmdv;
|
||
|
|
int tx_bits[FDMDV_BITS_PER_FRAME];
|
||
|
|
COMP tx_symbols[NC+1];
|
||
|
|
COMP tx_baseband[NC+1][M];
|
||
|
|
COMP tx_fdm[M];
|
||
|
|
float channel[CHANNEL_BUF_SIZE];
|
||
|
|
int channel_count;
|
||
|
|
COMP rx_fdm[M+M/P];
|
||
|
|
float foff_coarse;
|
||
|
|
int nin, next_nin;
|
||
|
|
COMP rx_fdm_fcorr[M+M/P];
|
||
|
|
COMP rx_baseband[NC+1][M+M/P];
|
||
|
|
COMP rx_filt[NC+1][P+1];
|
||
|
|
float rx_timing;
|
||
|
|
float env[NT*P];
|
||
|
|
COMP rx_symbols[NC+1];
|
||
|
|
int rx_bits[FDMDV_BITS_PER_FRAME];
|
||
|
|
float foff_fine;
|
||
|
|
int sync_bit;
|
||
|
|
int fest_state;
|
||
|
|
|
||
|
|
int tx_bits_log[FDMDV_BITS_PER_FRAME*FRAMES];
|
||
|
|
COMP tx_symbols_log[(NC+1)*FRAMES];
|
||
|
|
COMP tx_baseband_log[(NC+1)][M*FRAMES];
|
||
|
|
COMP tx_fdm_log[M*FRAMES];
|
||
|
|
COMP pilot_baseband1_log[NPILOTBASEBAND*FRAMES];
|
||
|
|
COMP pilot_baseband2_log[NPILOTBASEBAND*FRAMES];
|
||
|
|
COMP pilot_lpf1_log[NPILOTLPF*FRAMES];
|
||
|
|
COMP pilot_lpf2_log[NPILOTLPF*FRAMES];
|
||
|
|
COMP S1_log[MPILOTFFT*FRAMES];
|
||
|
|
COMP S2_log[MPILOTFFT*FRAMES];
|
||
|
|
float foff_coarse_log[FRAMES];
|
||
|
|
float foff_log[FRAMES];
|
||
|
|
COMP rx_baseband_log[(NC+1)][(M+M/P)*FRAMES];
|
||
|
|
int rx_baseband_log_col_index;
|
||
|
|
COMP rx_filt_log[NC+1][(P+1)*FRAMES];
|
||
|
|
int rx_filt_log_col_index;
|
||
|
|
float env_log[NT*P*FRAMES];
|
||
|
|
float rx_timing_log[FRAMES];
|
||
|
|
COMP rx_symbols_log[NC+1][FRAMES];
|
||
|
|
float sig_est_log[NC+1][FRAMES];
|
||
|
|
float noise_est_log[NC+1][FRAMES];
|
||
|
|
int rx_bits_log[FDMDV_BITS_PER_FRAME*FRAMES];
|
||
|
|
float foff_fine_log[FRAMES];
|
||
|
|
int sync_bit_log[FRAMES];
|
||
|
|
int coarse_fine_log[FRAMES];
|
||
|
|
int nin_log[FRAMES];
|
||
|
|
|
||
|
|
FILE *fout;
|
||
|
|
int f,c,i,j;
|
||
|
|
|
||
|
|
fdmdv = fdmdv_create();
|
||
|
|
next_nin = M;
|
||
|
|
channel_count = 0;
|
||
|
|
|
||
|
|
rx_baseband_log_col_index = 0;
|
||
|
|
rx_filt_log_col_index = 0;
|
||
|
|
|
||
|
|
printf("sizeof FDMDV states: %d bytes\n", sizeof(struct FDMDV));
|
||
|
|
|
||
|
|
for(f=0; f<FRAMES; f++) {
|
||
|
|
|
||
|
|
/* --------------------------------------------------------*\
|
||
|
|
Modulator
|
||
|
|
\*---------------------------------------------------------*/
|
||
|
|
|
||
|
|
fdmdv_get_test_bits(fdmdv, tx_bits);
|
||
|
|
bits_to_dqpsk_symbols(tx_symbols, fdmdv->prev_tx_symbols, tx_bits, &fdmdv->tx_pilot_bit);
|
||
|
|
memcpy(fdmdv->prev_tx_symbols, tx_symbols, sizeof(COMP)*(NC+1));
|
||
|
|
tx_filter(tx_baseband, tx_symbols, fdmdv->tx_filter_memory);
|
||
|
|
fdm_upconvert(tx_fdm, tx_baseband, fdmdv->phase_tx, fdmdv->freq);
|
||
|
|
|
||
|
|
/* --------------------------------------------------------*\
|
||
|
|
Channel
|
||
|
|
\*---------------------------------------------------------*/
|
||
|
|
|
||
|
|
nin = next_nin;
|
||
|
|
/*
|
||
|
|
if (f == 2)
|
||
|
|
nin = 120;
|
||
|
|
if (f == 3)
|
||
|
|
nin = 200;
|
||
|
|
if ((f !=2) && (f != 3))
|
||
|
|
nin = M;
|
||
|
|
*/
|
||
|
|
/* add M tx samples to end of buffer */
|
||
|
|
|
||
|
|
assert((channel_count + M) < CHANNEL_BUF_SIZE);
|
||
|
|
for(i=0; i<M; i++)
|
||
|
|
channel[channel_count+i] = tx_fdm[i].real;
|
||
|
|
channel_count += M;
|
||
|
|
|
||
|
|
/* take nin samples from start of buffer */
|
||
|
|
|
||
|
|
for(i=0; i<nin; i++) {
|
||
|
|
rx_fdm[i].real = channel[i];
|
||
|
|
rx_fdm[i].imag = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* shift buffer back */
|
||
|
|
|
||
|
|
for(i=0,j=nin; j<channel_count; i++,j++)
|
||
|
|
channel[i] = channel[j];
|
||
|
|
channel_count -= nin;
|
||
|
|
|
||
|
|
/* --------------------------------------------------------*\
|
||
|
|
Demodulator
|
||
|
|
\*---------------------------------------------------------*/
|
||
|
|
|
||
|
|
/* freq offset estimation and correction */
|
||
|
|
|
||
|
|
foff_coarse = rx_est_freq_offset(fdmdv, rx_fdm, nin);
|
||
|
|
if (fdmdv->coarse_fine == COARSE)
|
||
|
|
fdmdv->foff = foff_coarse;
|
||
|
|
fdmdv_freq_shift(rx_fdm_fcorr, rx_fdm, fdmdv->foff, &fdmdv->foff_rect, &fdmdv->foff_phase_rect, nin);
|
||
|
|
|
||
|
|
/* baseband processing */
|
||
|
|
|
||
|
|
fdm_downconvert(rx_baseband, rx_fdm_fcorr, fdmdv->phase_rx, fdmdv->freq, nin);
|
||
|
|
rx_filter(rx_filt, rx_baseband, fdmdv->rx_filter_memory, nin);
|
||
|
|
rx_timing = rx_est_timing(rx_symbols, rx_filt, rx_baseband, fdmdv->rx_filter_mem_timing, env, fdmdv->rx_baseband_mem_timing, nin);
|
||
|
|
foff_fine = qpsk_to_bits(rx_bits, &sync_bit, fdmdv->phase_difference, fdmdv->prev_rx_symbols, rx_symbols);
|
||
|
|
snr_update(fdmdv->sig_est, fdmdv->noise_est, fdmdv->phase_difference);
|
||
|
|
memcpy(fdmdv->prev_rx_symbols, rx_symbols, sizeof(COMP)*(NC+1));
|
||
|
|
|
||
|
|
next_nin = M;
|
||
|
|
|
||
|
|
if (rx_timing > 2*M/P)
|
||
|
|
next_nin += M/P;
|
||
|
|
|
||
|
|
if (rx_timing < 0)
|
||
|
|
next_nin -= M/P;
|
||
|
|
|
||
|
|
fdmdv->coarse_fine = freq_state(sync_bit, &fdmdv->fest_state);
|
||
|
|
fdmdv->foff -= TRACK_COEFF*foff_fine;
|
||
|
|
|
||
|
|
/* --------------------------------------------------------*\
|
||
|
|
Log each vector
|
||
|
|
\*---------------------------------------------------------*/
|
||
|
|
|
||
|
|
memcpy(&tx_bits_log[FDMDV_BITS_PER_FRAME*f], tx_bits, sizeof(int)*FDMDV_BITS_PER_FRAME);
|
||
|
|
memcpy(&tx_symbols_log[(NC+1)*f], tx_symbols, sizeof(COMP)*(NC+1));
|
||
|
|
for(c=0; c<NC+1; c++)
|
||
|
|
for(i=0; i<M; i++)
|
||
|
|
tx_baseband_log[c][f*M+i] = tx_baseband[c][i];
|
||
|
|
memcpy(&tx_fdm_log[M*f], tx_fdm, sizeof(COMP)*M);
|
||
|
|
|
||
|
|
/* freq offset estimation */
|
||
|
|
|
||
|
|
memcpy(&pilot_baseband1_log[f*NPILOTBASEBAND], fdmdv->pilot_baseband1, sizeof(COMP)*NPILOTBASEBAND);
|
||
|
|
memcpy(&pilot_baseband2_log[f*NPILOTBASEBAND], fdmdv->pilot_baseband2, sizeof(COMP)*NPILOTBASEBAND);
|
||
|
|
memcpy(&pilot_lpf1_log[f*NPILOTLPF], fdmdv->pilot_lpf1, sizeof(COMP)*NPILOTLPF);
|
||
|
|
memcpy(&pilot_lpf2_log[f*NPILOTLPF], fdmdv->pilot_lpf2, sizeof(COMP)*NPILOTLPF);
|
||
|
|
memcpy(&S1_log[f*MPILOTFFT], fdmdv->S1, sizeof(COMP)*MPILOTFFT);
|
||
|
|
memcpy(&S2_log[f*MPILOTFFT], fdmdv->S2, sizeof(COMP)*MPILOTFFT);
|
||
|
|
foff_coarse_log[f] = foff_coarse;
|
||
|
|
foff_log[f] = fdmdv->foff;
|
||
|
|
|
||
|
|
/* rx down conversion */
|
||
|
|
|
||
|
|
for(c=0; c<NC+1; c++) {
|
||
|
|
for(i=0; i<nin; i++)
|
||
|
|
rx_baseband_log[c][rx_baseband_log_col_index + i] = rx_baseband[c][i];
|
||
|
|
}
|
||
|
|
rx_baseband_log_col_index += nin;
|
||
|
|
|
||
|
|
/* rx filtering */
|
||
|
|
|
||
|
|
for(c=0; c<NC+1; c++) {
|
||
|
|
for(i=0; i<(P*nin)/M; i++)
|
||
|
|
rx_filt_log[c][rx_filt_log_col_index + i] = rx_filt[c][i];
|
||
|
|
}
|
||
|
|
rx_filt_log_col_index += (P*nin)/M;
|
||
|
|
|
||
|
|
/* timing estimation */
|
||
|
|
|
||
|
|
memcpy(&env_log[NT*P*f], env, sizeof(float)*NT*P);
|
||
|
|
rx_timing_log[f] = rx_timing;
|
||
|
|
nin_log[f] = nin;
|
||
|
|
for(c=0; c<NC+1; c++)
|
||
|
|
rx_symbols_log[c][f] = rx_symbols[c];
|
||
|
|
|
||
|
|
/* qpsk_to_bits() */
|
||
|
|
|
||
|
|
memcpy(&rx_bits_log[FDMDV_BITS_PER_FRAME*f], rx_bits, sizeof(int)*FDMDV_BITS_PER_FRAME);
|
||
|
|
for(c=0; c<NC+1; c++) {
|
||
|
|
sig_est_log[c][f] = fdmdv->sig_est[c];
|
||
|
|
noise_est_log[c][f] = fdmdv->noise_est[c];
|
||
|
|
}
|
||
|
|
foff_fine_log[f] = foff_fine;
|
||
|
|
sync_bit_log[f] = sync_bit;
|
||
|
|
|
||
|
|
coarse_fine_log[f] = fdmdv->coarse_fine;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*---------------------------------------------------------*\
|
||
|
|
Dump logs to Octave file for evaluation
|
||
|
|
by tfdmdv.m Octave script
|
||
|
|
\*---------------------------------------------------------*/
|
||
|
|
|
||
|
|
fout = fopen("tfdmdv_out.txt","wt");
|
||
|
|
assert(fout != NULL);
|
||
|
|
fprintf(fout, "# Created by tfdmdv.c\n");
|
||
|
|
octave_save_int(fout, "tx_bits_log_c", tx_bits_log, 1, FDMDV_BITS_PER_FRAME*FRAMES);
|
||
|
|
octave_save_complex(fout, "tx_symbols_log_c", tx_symbols_log, 1, (NC+1)*FRAMES, (NC+1)*FRAMES);
|
||
|
|
octave_save_complex(fout, "tx_baseband_log_c", (COMP*)tx_baseband_log, (NC+1), M*FRAMES, M*FRAMES);
|
||
|
|
octave_save_complex(fout, "tx_fdm_log_c", (COMP*)tx_fdm_log, 1, M*FRAMES, M*FRAMES);
|
||
|
|
octave_save_complex(fout, "pilot_lut_c", (COMP*)fdmdv->pilot_lut, 1, NPILOT_LUT, NPILOT_LUT);
|
||
|
|
octave_save_complex(fout, "pilot_baseband1_log_c", pilot_baseband1_log, 1, NPILOTBASEBAND*FRAMES, NPILOTBASEBAND*FRAMES);
|
||
|
|
octave_save_complex(fout, "pilot_baseband2_log_c", pilot_baseband2_log, 1, NPILOTBASEBAND*FRAMES, NPILOTBASEBAND*FRAMES);
|
||
|
|
octave_save_complex(fout, "pilot_lpf1_log_c", pilot_lpf1_log, 1, NPILOTLPF*FRAMES, NPILOTLPF*FRAMES);
|
||
|
|
octave_save_complex(fout, "pilot_lpf2_log_c", pilot_lpf2_log, 1, NPILOTLPF*FRAMES, NPILOTLPF*FRAMES);
|
||
|
|
octave_save_complex(fout, "S1_log_c", S1_log, 1, MPILOTFFT*FRAMES, MPILOTFFT*FRAMES);
|
||
|
|
octave_save_complex(fout, "S2_log_c", S2_log, 1, MPILOTFFT*FRAMES, MPILOTFFT*FRAMES);
|
||
|
|
octave_save_float(fout, "foff_log_c", foff_log, 1, FRAMES, FRAMES);
|
||
|
|
octave_save_float(fout, "foff_coarse_log_c", foff_coarse_log, 1, FRAMES, FRAMES);
|
||
|
|
octave_save_complex(fout, "rx_baseband_log_c", (COMP*)rx_baseband_log, (NC+1), rx_baseband_log_col_index, (M+M/P)*FRAMES);
|
||
|
|
octave_save_complex(fout, "rx_filt_log_c", (COMP*)rx_filt_log, (NC+1), rx_filt_log_col_index, (P+1)*FRAMES);
|
||
|
|
octave_save_float(fout, "env_log_c", env_log, 1, NT*P*FRAMES, NT*P*FRAMES);
|
||
|
|
octave_save_float(fout, "rx_timing_log_c", rx_timing_log, 1, FRAMES, FRAMES);
|
||
|
|
octave_save_complex(fout, "rx_symbols_log_c", (COMP*)rx_symbols_log, (NC+1), FRAMES, FRAMES);
|
||
|
|
octave_save_float(fout, "sig_est_log_c", (float*)sig_est_log, (NC+1), FRAMES, FRAMES);
|
||
|
|
octave_save_float(fout, "noise_est_log_c", (float*)noise_est_log, (NC+1), FRAMES, FRAMES);
|
||
|
|
octave_save_int(fout, "rx_bits_log_c", rx_bits_log, 1, FDMDV_BITS_PER_FRAME*FRAMES);
|
||
|
|
octave_save_float(fout, "foff_fine_log_c", foff_fine_log, 1, FRAMES, FRAMES);
|
||
|
|
octave_save_int(fout, "sync_bit_log_c", sync_bit_log, 1, FRAMES);
|
||
|
|
octave_save_int(fout, "coarse_fine_log_c", coarse_fine_log, 1, FRAMES);
|
||
|
|
octave_save_int(fout, "nin_log_c", nin_log, 1, FRAMES);
|
||
|
|
fclose(fout);
|
||
|
|
|
||
|
|
fdmdv_destroy(fdmdv);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|