diff options
Diffstat (limited to 'src/freedv_rx.c')
| -rw-r--r-- | src/freedv_rx.c | 546 |
1 files changed, 280 insertions, 266 deletions
diff --git a/src/freedv_rx.c b/src/freedv_rx.c index 8cb6019..c4df02c 100644 --- a/src/freedv_rx.c +++ b/src/freedv_rx.c @@ -9,7 +9,8 @@ Example usage (all one line): $ cd codec2/build_linux/src - $ ./freedv_tx 1600 ../../raw/ve9qrp_10s.raw - | ./freedv_rx 1600 - - | aplay -f S16 + $ ./freedv_tx 1600 ../../raw/ve9qrp_10s.raw - | ./freedv_rx 1600 - - | aplay +-f S16 \*---------------------------------------------------------------------------*/ @@ -31,299 +32,312 @@ */ #include <assert.h> -#include <stdlib.h> -#include <stdio.h> -#include <string.h> #include <errno.h> #include <getopt.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> -#include "reliable_text.h" #include "freedv_api.h" #include "modem_stats.h" +#include "reliable_text.h" -#define NDISCARD 5 /* BER measure optionally discards first few frames after sync */ +#define NDISCARD \ + 5 /* BER measure optionally discards first few frames after sync */ /* optioal call-back function for received txt characters */ -void my_put_next_rx_char(void *states, char c) { fprintf((FILE*)states, "%c", c); } +void my_put_next_rx_char(void *states, char c) { + fprintf((FILE *)states, "%c", c); +} -static FILE* reliable_tx_fp; +static FILE *reliable_tx_fp; reliable_text_t reliable_text_obj; -void on_reliable_text_rx(reliable_text_t rt, const char* txt_ptr, int length, void* state) -{ - fprintf(reliable_tx_fp, "%s\n", txt_ptr); - reliable_text_reset(reliable_text_obj); +void on_reliable_text_rx(reliable_text_t rt, const char *txt_ptr, int length, + void *state) { + fprintf(reliable_tx_fp, "%s\n", txt_ptr); + reliable_text_reset(reliable_text_obj); } int main(int argc, char *argv[]) { - FILE *fin, *fout, *ftxt_rx = NULL; - int nin, nout, nout_total = 0, frame = 0; - struct MODEM_STATS stats = {0}; - int mode; - int sync; - float snr_est; - float clock_offset, foff; - int use_testframes, verbose, discard, use_complex, use_reliabletext; - int use_squelch; - float squelch = 0; - struct freedv *freedv; - int use_passthroughgain; - float passthroughgain = 0.0; - - char f2020[80] = {0}; + FILE *fin, *fout, *ftxt_rx = NULL; + int nin, nout, nout_total = 0, frame = 0; + struct MODEM_STATS stats = {0}; + int mode; + int sync; + float snr_est; + float clock_offset, foff; + int use_testframes, verbose, discard, use_complex, use_reliabletext; + int use_squelch; + float squelch = 0; + struct freedv *freedv; + int use_passthroughgain; + float passthroughgain = 0.0; + + char f2020[80] = {0}; #ifdef __LPCNET__ - sprintf(f2020,"|2020|2020B|2020C"); + sprintf(f2020, "|2020|2020B"); #endif - - if (argc < 4) { - helpmsg: - fprintf(stderr, "usage: %s [options] 1600|700C|700D|700E|2400A|2400B|800XA%s InputModemSpeechFile OutputSpeechRawFile\n" - "\n" - " --discard Reset BER stats on loss of sync, helps us get sensible BER results\n" - " --reliabletext txt Send 'txt' using reliable text protocol\n" - " --txtrx filename Store reliable text output to filename\n" - " --squelch leveldB Set squelch level\n" - " --testframes testframes assumed to be received instead of coded speech, measure BER/PER\n" - " --usecomplex Complex int16 input samples (default real int16)\n" - " -v Verbose level 1\n" - " --vv Verbose level 2\n" - "\n", argv[0], f2020); - fprintf(stderr, "example: $ %s 1600 hts1a_fdmdv.raw hts1a_out.raw \n", argv[0]); - exit(1); - } - use_testframes = verbose = discard = use_complex = use_squelch = 0; use_reliabletext = 0; - use_passthroughgain = 0; - - int o = 0; - int opt_idx = 0; - while( o != -1 ){ - static struct option long_opts[] = { - {"discard", no_argument, 0, 'i'}, - {"help", no_argument, 0, 'h'}, - {"reliabletext", no_argument, 0, 'r'}, - {"squelch", required_argument, 0, 's'}, - {"txtrx", required_argument, 0, 'x'}, - {"testframes", no_argument, 0, 't'}, - {"usecomplex", no_argument, 0, 'c'}, - {"verbose1", no_argument, 0, 'v'}, - {"vv", no_argument, 0, 'w'}, - {"passthroughgain", required_argument, 0, 'p'}, - {0, 0, 0, 0} - }; - - o = getopt_long(argc,argv,"idhr:s:x:tcvwp:",long_opts,&opt_idx); - - switch(o) { - case 'i': - discard = 1; - break; - case 'c': - use_complex = 1; - break; - case 'p': - use_passthroughgain = 1; - passthroughgain = atof(optarg); - break; - case 'r': - use_reliabletext = 1; - break; - case 's': - use_squelch = 1; - squelch = atof(optarg); - break; - case 't': - use_testframes = 1; - break; - case 'x': - ftxt_rx = fopen(optarg, "wt"); - assert(ftxt_rx != NULL); - break; - case 'v': - verbose = 1; - break; - case 'w': - verbose = 2; - break; - case 'h': - case '?': - goto helpmsg; - break; - } - } - int dx = optind; - - if( (argc - dx) < 3) { - fprintf(stderr, "too few arguments.\n"); + if (argc < 4) { + helpmsg: + fprintf( + stderr, + "usage: %s [options] 1600|700C|700D|700E|2400A|2400B|800XA%s " + "InputModemSpeechFile OutputSpeechRawFile\n" + "\n" + " --discard Reset BER stats on loss of sync, helps us " + "get sensible BER results\n" + " --reliabletext txt Send 'txt' using reliable text protocol\n" + " --txtrx filename Store reliable text output to filename\n" + " --squelch leveldB Set squelch level\n" + " --testframes testframes assumed to be received instead " + "of coded speech, measure BER/PER\n" + " --usecomplex Complex int16 input samples (default real " + "int16)\n" + " -v Verbose level 1\n" + " --vv Verbose level 2\n" + "\n", + argv[0], f2020); + fprintf(stderr, "example: $ %s 1600 hts1a_fdmdv.raw hts1a_out.raw \n", + argv[0]); + exit(1); + } + + use_testframes = verbose = discard = use_complex = use_squelch = 0; + use_reliabletext = 0; + use_passthroughgain = 0; + + int o = 0; + int opt_idx = 0; + while (o != -1) { + static struct option long_opts[] = { + {"discard", no_argument, 0, 'i'}, + {"help", no_argument, 0, 'h'}, + {"reliabletext", no_argument, 0, 'r'}, + {"squelch", required_argument, 0, 's'}, + {"txtrx", required_argument, 0, 'x'}, + {"testframes", no_argument, 0, 't'}, + {"usecomplex", no_argument, 0, 'c'}, + {"verbose1", no_argument, 0, 'v'}, + {"vv", no_argument, 0, 'w'}, + {"passthroughgain", required_argument, 0, 'p'}, + {0, 0, 0, 0}}; + + o = getopt_long(argc, argv, "idhr:s:x:tcvwp:", long_opts, &opt_idx); + + switch (o) { + case 'i': + discard = 1; + break; + case 'c': + use_complex = 1; + break; + case 'p': + use_passthroughgain = 1; + passthroughgain = atof(optarg); + break; + case 'r': + use_reliabletext = 1; + break; + case 's': + use_squelch = 1; + squelch = atof(optarg); + break; + case 't': + use_testframes = 1; + break; + case 'x': + ftxt_rx = fopen(optarg, "wt"); + assert(ftxt_rx != NULL); + break; + case 'v': + verbose = 1; + break; + case 'w': + verbose = 2; + break; + case 'h': + case '?': goto helpmsg; + break; } - - mode = -1; - if (!strcmp(argv[dx],"1600")) mode = FREEDV_MODE_1600; - if (!strcmp(argv[dx],"700C")) mode = FREEDV_MODE_700C; - if (!strcmp(argv[dx],"700D")) mode = FREEDV_MODE_700D; - if (!strcmp(argv[dx],"700E")) mode = FREEDV_MODE_700E; - if (!strcmp(argv[dx],"2400A")) mode = FREEDV_MODE_2400A; - if (!strcmp(argv[dx],"2400B")) mode = FREEDV_MODE_2400B; - if (!strcmp(argv[dx],"800XA")) mode = FREEDV_MODE_800XA; - #ifdef __LPCNET__ - if (!strcmp(argv[dx],"2020")) mode = FREEDV_MODE_2020; - if (!strcmp(argv[dx],"2020B")) mode = FREEDV_MODE_2020B; - if (!strcmp(argv[dx],"2020C")) mode = FREEDV_MODE_2020C; - #endif - if (mode == -1) { - fprintf(stderr, "Error in mode: %s\n", argv[dx]); - exit(1); - } - - if (strcmp(argv[dx+1], "-") == 0) fin = stdin; - else if ( (fin = fopen(argv[dx+1],"rb")) == NULL ) { - fprintf(stderr, "Error opening input raw modem sample file: %s: %s.\n", - argv[dx+1], strerror(errno)); - exit(1); + } + int dx = optind; + + if ((argc - dx) < 3) { + fprintf(stderr, "too few arguments.\n"); + goto helpmsg; + } + + mode = -1; + if (!strcmp(argv[dx], "1600")) mode = FREEDV_MODE_1600; + if (!strcmp(argv[dx], "700C")) mode = FREEDV_MODE_700C; + if (!strcmp(argv[dx], "700D")) mode = FREEDV_MODE_700D; + if (!strcmp(argv[dx], "700E")) mode = FREEDV_MODE_700E; + if (!strcmp(argv[dx], "2400A")) mode = FREEDV_MODE_2400A; + if (!strcmp(argv[dx], "2400B")) mode = FREEDV_MODE_2400B; + if (!strcmp(argv[dx], "800XA")) mode = FREEDV_MODE_800XA; +#ifdef __LPCNET__ + if (!strcmp(argv[dx], "2020")) mode = FREEDV_MODE_2020; + if (!strcmp(argv[dx], "2020B")) mode = FREEDV_MODE_2020B; +#endif + if (mode == -1) { + fprintf(stderr, "Error in mode: %s\n", argv[dx]); + exit(1); + } + + if (strcmp(argv[dx + 1], "-") == 0) + fin = stdin; + else if ((fin = fopen(argv[dx + 1], "rb")) == NULL) { + fprintf(stderr, "Error opening input raw modem sample file: %s: %s.\n", + argv[dx + 1], strerror(errno)); + exit(1); + } + + if (strcmp(argv[dx + 2], "-") == 0) + fout = stdout; + else if ((fout = fopen(argv[dx + 2], "wb")) == NULL) { + fprintf(stderr, "Error opening output speech sample file: %s: %s.\n", + argv[dx + 2], strerror(errno)); + exit(1); + } + + freedv = freedv_open(mode); + assert(freedv != NULL); + + /* set up a few options, calling these is optional -------------------------*/ + + freedv_set_test_frames(freedv, use_testframes); + if (verbose == 2) freedv_set_verbose(freedv, verbose); + + if (use_squelch) { + freedv_set_snr_squelch_thresh(freedv, squelch); + freedv_set_squelch_en(freedv, true); + } + if (use_passthroughgain) freedv_passthrough_gain(freedv, passthroughgain); + + /* install optional handler for received txt characters */ + if (ftxt_rx != NULL) { + if (use_reliabletext) { + reliable_tx_fp = ftxt_rx; + + reliable_text_obj = reliable_text_create(); + assert(reliable_text_obj != NULL); + reliable_text_set_string(reliable_text_obj, "AB1CDEF", 7); // not used + reliable_text_use_with_freedv(reliable_text_obj, freedv, + on_reliable_text_rx, NULL); + } else { + freedv_set_callback_txt(freedv, my_put_next_rx_char, NULL, ftxt_rx); } - - if (strcmp(argv[dx+2], "-") == 0) fout = stdout; - else if ( (fout = fopen(argv[dx+2],"wb")) == NULL ) { - fprintf(stderr, "Error opening output speech sample file: %s: %s.\n", - argv[dx+2], strerror(errno)); - exit(1); + } + + /* note use of API functions to tell us how big our buffers need to be -----*/ + + short speech_out[freedv_get_n_max_speech_samples(freedv)]; + short demod_in[freedv_get_n_max_modem_samples(freedv)]; + + /* We need to work out how many samples the demod needs on each + call (nin). This is used to adjust for differences in the tx + and rx sample clock frequencies. Note also the number of + output speech samples "nout" is time varying. */ + + nin = freedv_nin(freedv); + while (fread(demod_in, sizeof(short), nin, fin) == nin) { + frame++; + + if (use_complex) { + /* exercise the complex version of the API (useful + for testing 700D which has a different code path for + short samples) */ + COMP demod_in_complex[nin]; + + for (int i = 0; i < nin; i++) { + demod_in_complex[i].real = (float)demod_in[i]; + demod_in_complex[i].imag = 0.0f; + } + nout = freedv_comprx(freedv, speech_out, demod_in_complex); + } else { + // most common interface - real shorts in, real shorts out + nout = freedv_rx(freedv, speech_out, demod_in); } - freedv = freedv_open(mode); - assert(freedv != NULL); - - /* set up a few options, calling these is optional -------------------------*/ + /* IMPORTANT: don't forget to do this in the while loop to + ensure we fread the correct number of samples: ie update + "nin" before every call to freedv_rx()/freedv_comprx() */ + nin = freedv_nin(freedv); - freedv_set_test_frames(freedv, use_testframes); - if (verbose == 2) freedv_set_verbose(freedv, verbose); - - if (use_squelch) { - freedv_set_snr_squelch_thresh(freedv, squelch); - freedv_set_squelch_en(freedv, true); - } - if (use_passthroughgain) freedv_passthrough_gain(freedv, passthroughgain); - - /* install optional handler for received txt characters */ - if (ftxt_rx != NULL) - { - if (use_reliabletext) - { - reliable_tx_fp = ftxt_rx; - - reliable_text_obj = reliable_text_create(); - assert(reliable_text_obj != NULL); - reliable_text_set_string(reliable_text_obj, "AB1CDEF", 7); // not used - reliable_text_use_with_freedv(reliable_text_obj, freedv, on_reliable_text_rx, NULL); - } - else - { - freedv_set_callback_txt(freedv, my_put_next_rx_char, NULL, ftxt_rx); - } + /* optionally read some stats */ + freedv_get_modem_stats(freedv, &sync, &snr_est); + freedv_get_modem_extended_stats(freedv, &stats); + int total_bit_errors = freedv_get_total_bit_errors(freedv); + clock_offset = stats.clock_offset; + foff = stats.foff; + + if (discard && (sync == 0)) { + // discard BER results if we get out of sync, helps us get sensible BER + // results + freedv_set_total_bits(freedv, 0); + freedv_set_total_bit_errors(freedv, 0); + freedv_set_total_bits_coded(freedv, 0); + freedv_set_total_bit_errors_coded(freedv, 0); } - /* note use of API functions to tell us how big our buffers need to be -----*/ + fwrite(speech_out, sizeof(short), nout, fout); + nout_total += nout; - short speech_out[freedv_get_n_max_speech_samples(freedv)]; - short demod_in[freedv_get_n_max_modem_samples(freedv)]; - - /* We need to work out how many samples the demod needs on each - call (nin). This is used to adjust for differences in the tx - and rx sample clock frequencies. Note also the number of - output speech samples "nout" is time varying. */ - - nin = freedv_nin(freedv); - while(fread(demod_in, sizeof(short), nin, fin) == nin) { - frame++; - - if (use_complex) { - /* exercise the complex version of the API (useful - for testing 700D which has a different code path for - short samples) */ - COMP demod_in_complex[nin]; - - for(int i=0; i<nin; i++) { - demod_in_complex[i].real = (float)demod_in[i]; - demod_in_complex[i].imag = 0.0f; - } - nout = freedv_comprx(freedv, speech_out, demod_in_complex); - } else { - // most common interface - real shorts in, real shorts out - nout = freedv_rx(freedv, speech_out, demod_in); - } - - /* IMPORTANT: don't forget to do this in the while loop to - ensure we fread the correct number of samples: ie update - "nin" before every call to freedv_rx()/freedv_comprx() */ - nin = freedv_nin(freedv); - - /* optionally read some stats */ - freedv_get_modem_stats(freedv, &sync, &snr_est); - freedv_get_modem_extended_stats(freedv, &stats); - int total_bit_errors = freedv_get_total_bit_errors(freedv); - clock_offset = stats.clock_offset; - foff = stats.foff; - - if (discard && (sync == 0)) { - // discard BER results if we get out of sync, helps us get sensible BER results - freedv_set_total_bits(freedv, 0); freedv_set_total_bit_errors(freedv, 0); - freedv_set_total_bits_coded(freedv, 0); freedv_set_total_bit_errors_coded(freedv, 0); - } - - fwrite(speech_out, sizeof(short), nout, fout); - nout_total += nout; - - if (verbose == 1) { - fprintf(stderr, "frame: %d sync: %d nin: %d snr: %3.2f dB bit errors: %d clock_off: %6.2f foff: %5.2f\n", - frame, sync, nin, snr_est, total_bit_errors, clock_offset, foff); - } - - /* if using pipes we probably don't want the usual buffering - to occur */ - if (fout == stdout) fflush(stdout); + if (verbose == 1) { + fprintf(stderr, + "frame: %d sync: %d nin: %d snr: %3.2f dB bit errors: %d " + "clock_off: %6.2f foff: %5.2f\n", + frame, sync, nin, snr_est, total_bit_errors, clock_offset, foff); } - if (ftxt_rx != NULL) fclose(ftxt_rx); - fclose(fin); - fclose(fout); - fprintf(stderr, "frames decoded: %d output speech samples: %d\n", frame, nout_total); - - /* finish up with some stats */ - - if (freedv_get_test_frames(freedv)) { - int Tbits = freedv_get_total_bits(freedv); - int Terrs = freedv_get_total_bit_errors(freedv); - float uncoded_ber = (float)Terrs/Tbits; - fprintf(stderr, "BER......: %5.4f Tbits: %8d Terrs: %8d\n", - (double)uncoded_ber, Tbits, Terrs); - if ((mode == FREEDV_MODE_700D) || (mode == FREEDV_MODE_700E) || - (mode == FREEDV_MODE_2020) || (mode == FREEDV_MODE_2020B) || - (mode == FREEDV_MODE_2020C)) { - int Tbits_coded = freedv_get_total_bits_coded(freedv); - int Terrs_coded = freedv_get_total_bit_errors_coded(freedv); - float coded_ber = (float)Terrs_coded/Tbits_coded; - fprintf(stderr, "Coded BER: %5.4f Tbits: %8d Terrs: %8d\n", - (double)coded_ber, Tbits_coded, Terrs_coded); - int Tpackets = freedv_get_total_packets(freedv); - int Tpacket_errors = freedv_get_total_packet_errors(freedv); - float per = (float)Tpacket_errors/Tpackets; - fprintf(stderr, "Coded PER: %5.4f Tpkts: %8d Tpers: %8d\n", - per, Tpackets, Tpacket_errors); - - /* set return code for Ctest */ - if ((uncoded_ber < 0.1f) && (coded_ber < 0.01f)) - return 0; - else - return 1; - } + /* if using pipes we probably don't want the usual buffering + to occur */ + if (fout == stdout) fflush(stdout); + } + + if (ftxt_rx != NULL) fclose(ftxt_rx); + fclose(fin); + fclose(fout); + fprintf(stderr, "frames decoded: %d output speech samples: %d\n", frame, + nout_total); + + /* finish up with some stats */ + + if (freedv_get_test_frames(freedv)) { + int Tbits = freedv_get_total_bits(freedv); + int Terrs = freedv_get_total_bit_errors(freedv); + float uncoded_ber = (float)Terrs / Tbits; + fprintf(stderr, "BER......: %5.4f Tbits: %8d Terrs: %8d\n", + (double)uncoded_ber, Tbits, Terrs); + if ((mode == FREEDV_MODE_700D) || (mode == FREEDV_MODE_700E) || + (mode == FREEDV_MODE_2020) || (mode == FREEDV_MODE_2020B)) { + int Tbits_coded = freedv_get_total_bits_coded(freedv); + int Terrs_coded = freedv_get_total_bit_errors_coded(freedv); + float coded_ber = (float)Terrs_coded / Tbits_coded; + fprintf(stderr, "Coded BER: %5.4f Tbits: %8d Terrs: %8d\n", + (double)coded_ber, Tbits_coded, Terrs_coded); + int Tpackets = freedv_get_total_packets(freedv); + int Tpacket_errors = freedv_get_total_packet_errors(freedv); + float per = (float)Tpacket_errors / Tpackets; + fprintf(stderr, "Coded PER: %5.4f Tpkts: %8d Tpers: %8d\n", per, + Tpackets, Tpacket_errors); + + /* set return code for Ctest */ + if ((uncoded_ber < 0.1f) && (coded_ber < 0.01f)) + return 0; + else + return 1; } + } - if (use_reliabletext) - { - reliable_text_destroy(reliable_text_obj); - } - - freedv_close(freedv); - - return 0; + if (use_reliabletext) { + reliable_text_destroy(reliable_text_obj); + } + + freedv_close(freedv); + + return 0; } |