PierMesh
/
direwolf
mirror of https://github.com/wb2osz/direwolf.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
direwolf/atest.c

498 lines
13 KiB
C
Raw Permalink Blame History

//
// This file is part of Dire Wolf, an amateur radio packet TNC.
//
// Copyright (C) 2011,2012,2013 John Langner, WB2OSZ
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 2 of the License, or
// (at your option) any later version.
//
// 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 General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
/*-------------------------------------------------------------------
*
* Name: atest.c
*
* Purpose: Unit test for the AFSK demodulator.
*
* Inputs: Takes audio from a .WAV file insted of the audio device.
*
* Description: This can be used to test the AFSK demodulator under
* controlled and reproducable conditions for tweaking.
*
* For example
*
* (1) Download WA8LMF's TNC Test CD image file from
* http://wa8lmf.net/TNCtest/index.htm
*
* (2) Burn a physical CD.
*
* (3) "Rip" the desired tracks with Windows Media Player.
* This results in .WMA files.
*
* (4) Upload the .WMA file(s) to http://media.io/ and
* convert to .WAV format.
*
*
* Comparison to others:
*
* Here are some other scores from Track 2 of the TNC Test CD:
* http://sites.google.com/site/ki4mcw/Home/arduino-tnc
*
* Without ONE_CHAN defined:
*
* Notice that the number of packets decoded, as reported by
* this test program, will be twice the number expected because
* we are decoding the left and right audio channels separately.
*
*
* With ONE_CHAN defined:
*
* Only process one channel.
*
* Version 0.4 decoded 870 packets.
*
* After a little tweaking, version 0.5 decodes 931 packets.
*
* After more tweaking, version 0.6 gets 965 packets.
* This is without the option to retry after getting a bad FCS.
*
*--------------------------------------------------------------------*/
// #define X 1
#include <stdio.h>
#include <unistd.h>
//#include <fcntl.h>
#include <stdlib.h>
#include <assert.h>
#include <string.h>
#include <time.h>
#include <getopt.h>
#define ATEST_C 1
#include "audio.h"
#include "demod.h"
// #include "fsk_demod_agc.h"
#include "textcolor.h"
#include "ax25_pad.h"
#include "hdlc_rec2.h"
struct wav_header { /* .WAV file header. */
char riff[4]; /* "RIFF" */
int filesize; /* file length - 8 */
char wave[4]; /* "WAVE" */
char fmt[4]; /* "fmt " */
int fmtsize; /* 16. */
short wformattag; /* 1 for PCM. */
short nchannels; /* 1 for mono, 2 for stereo. */
int nsamplespersec; /* sampling freq, Hz. */
int navgbytespersec; /* = nblockalign*nsamplespersec. */
short nblockalign; /* = wbitspersample/8 * nchannels. */
short wbitspersample; /* 16 or 8. */
char data[4]; /* "data" */
int datasize; /* number of bytes following. */
} ;
/* 8 bit samples are unsigned bytes */
/* in range of 0 .. 255. */
/* 16 bit samples are signed short */
/* in range of -32768 .. +32767. */
static struct wav_header header;
static FILE *fp;
static int e_o_f;
static int packets_decoded = 0;
static int decimate = 1; /* Reduce that sampling rate. */
/* 1 = normal, 2 = half, etc. */
int main (int argc, char *argv[])
{
//int err;
int c;
struct audio_s modem;
int channel;
time_t start_time;
text_color_init(1);
text_color_set(DW_COLOR_INFO);
/*
* First apply defaults.
*/
memset (&modem, 0, sizeof(modem));
modem.num_channels = DEFAULT_NUM_CHANNELS;
modem.samples_per_sec = DEFAULT_SAMPLES_PER_SEC;
modem.bits_per_sample = DEFAULT_BITS_PER_SAMPLE;
// Results v0.9:
//
// fix_bits = 0 971 packets, 69 sec
// fix_bits = SINGLE 990 64
// fix_bits = DOUBLE 992 65
// fix_bits = TRIPLE 992 67
// fix_bits = TWO_SEP 1004 476
// Essentially no difference in time for those with order N time.
// Time increases greatly for the one with order N^2 time.
// Results: version 1.1, decoder C, modem.fix_bits = RETRY_MAX - 1
//
// 971 NONE
// +19 SINGLE
// +2 DOUBLE
// +12 TWO_SEP
// +3 REMOVE_MANY
// ----
// 1007 Total in 1008 sec. More than twice as long as earlier version.
// Results: version 1.1, decoders ABC, modem.fix_bits = RETRY_MAX - 1
//
// 976 NONE
// +21 SINGLE
// +1 DOUBLE
// +22 TWO_SEP
// +1 MANY
// +3 REMOVE_MANY
// ----
// 1024 Total in 2042 sec.
// About 34 minutes of CPU time for a roughly 40 minute CD.
// Many computers wouldn't be able to keep up.
// The SINGLE and TWO_SEP techniques are the most effective.
// Should we reorder the enum values so that TWO_SEP
// comes after SINGLE? That way "FIX_BITS 2" would
// use the two most productive techniques and not waste
// time on the others. People with plenty of CPU power
// to spare can still specify larger numbers for the other
// techniques with less return on investment.
// TODO: tabulate results from atest2.c.txt
/* TODO: should have a command line option for this. */
modem.fix_bits = RETRY_NONE;
//modem.fix_bits = RETRY_SWAP_SINGLE;
//modem.fix_bits = RETRY_SWAP_DOUBLE;
//modem.fix_bits = RETRY_SWAP_TRIPLE;
//modem.fix_bits = RETRY_INSERT_DOUBLE;
modem.fix_bits = RETRY_SWAP_TWO_SEP;
//modem.fix_bits = RETRY_REMOVE_MANY;
//modem.fix_bits = RETRY_MAX - 1;
for (channel=0; channel<MAX_CHANS; channel++) {
modem.modem_type[channel] = AFSK;
modem.mark_freq[channel] = DEFAULT_MARK_FREQ;
modem.space_freq[channel] = DEFAULT_SPACE_FREQ;
modem.baud[channel] = DEFAULT_BAUD;
strcpy (modem.profiles[channel], "C");
//strcpy (modem.profiles[channel], "ABC");
// temp
// strcpy (modem.profiles[channel], "F");
modem.num_subchan[channel] = strlen(modem.profiles[channel]);
modem.num_freq[channel] = 1;
modem.offset[channel] = 0;
// temp test
//modem.num_subchan[channel] = modem.num_freq[channel] = 3;
//modem.num_subchan[channel] = modem.num_freq[channel] = 5;
//modem.offset[channel] = 100;
//strcpy (modem.ptt_device[channel], "");
//modem.ptt_line[channel] = PTT_NONE;
//modem.slottime[channel] = DEFAULT_SLOTTIME;
//modem.persist[channel] = DEFAULT_PERSIST;
//modem.txdelay[channel] = DEFAULT_TXDELAY;
//modem.txtail[channel] = DEFAULT_TXTAIL;
}
while (1) {
int this_option_optind = optind ? optind : 1;
int option_index = 0;
static struct option long_options[] = {
{"future1", 1, 0, 0},
{"future2", 0, 0, 0},
{"future3", 1, 0, 'c'},
{0, 0, 0, 0}
};
/* ':' following option character means arg is required. */
c = getopt_long(argc, argv, "B:P:D:",
long_options, &option_index);
if (c == -1)
break;
switch (c) {
case 'B': /* -B for data Bit rate */
/* 300 implies 1600/1800 AFSK. */
/* 1200 implies 1200/2200 AFSK. */
/* 9600 implies scrambled. */
modem.baud[0] = atoi(optarg);
printf ("Data rate set to %d bits / second.\n", modem.baud[0]);
if (modem.baud[0] < 100 || modem.baud[0] > 10000) {
fprintf (stderr, "Use a more reasonable bit rate in range of 100 - 10000.\n");
exit (EXIT_FAILURE);
}
if (modem.baud[0] < 600) {
modem.modem_type[0] = AFSK;
modem.mark_freq[0] = 1600;
modem.space_freq[0] = 1800;
}
else if (modem.baud[0] > 2400) {
modem.modem_type[0] = SCRAMBLE;
modem.mark_freq[0] = 0;
modem.space_freq[0] = 0;
printf ("Using scrambled baseband signal rather than AFSK.\n");
}
else {
modem.modem_type[0] = AFSK;
modem.mark_freq[0] = 1200;
modem.space_freq[0] = 2200;
}
break;
case 'P': /* -P for modem profile. */
printf ("Demodulator profile set to \"%s\"\n", optarg);
strcpy (modem.profiles[0], optarg);
break;
case 'D': /* -D reduce sampling rate for lower CPU usage. */
decimate = atoi(optarg);
printf ("Decimate factor = %d\n", decimate);
modem.decimate[0] = decimate;
break;
case '?':
/* Unknown option message was already printed. */
//usage (argv);
break;
default:
/* Should not be here. */
printf("?? getopt returned character code 0%o ??\n", c);
//usage (argv);
}
}
if (optind >= argc) {
printf ("Specify .WAV file name on command line.\n");
exit (1);
}
fp = fopen(argv[optind], "rb");
if (fp == NULL) {
text_color_set(DW_COLOR_ERROR);
fprintf (stderr, "Couldn't open file for read: %s\n", argv[optind]);
//perror ("more info?");
exit (1);
}
start_time = time(NULL);
/*
* Read the file header.
*/
fread (&header, sizeof(header), (size_t)1, fp);
assert (header.nchannels == 1 || header.nchannels == 2);
assert (header.wbitspersample == 8 || header.wbitspersample == 16);
modem.samples_per_sec = header.nsamplespersec;
modem.samples_per_sec = modem.samples_per_sec;
modem.bits_per_sample = header.wbitspersample;
modem.num_channels = header.nchannels;
text_color_set(DW_COLOR_INFO);
printf ("%d samples per second\n", modem.samples_per_sec);
printf ("%d bits per sample\n", modem.bits_per_sample);
printf ("%d audio channels\n", modem.num_channels);
printf ("%d audio bytes in file\n", (int)(header.datasize));
/*
* Initialize the AFSK demodulator and HDLC decoder.
*/
multi_modem_init (&modem);
e_o_f = 0;
while ( ! e_o_f)
{
int audio_sample;
int c;
for (c=0; c<modem.num_channels; c++)
{
/* This reads either 1 or 2 bytes depending on */
/* bits per sample. */
audio_sample = demod_get_sample ();
if (audio_sample >= 256 * 256)
e_o_f = 1;
#define ONE_CHAN 1 /* only use one audio channel. */
#if ONE_CHAN
if (c != 0) continue;
#endif
multi_modem_process_sample(c,audio_sample);
}
/* When a complete frame is accumulated, */
/* process_rec_frame, below, is called. */
}
text_color_set(DW_COLOR_INFO);
printf ("\n\n");
printf ("%d packets decoded in %d seconds.\n", packets_decoded, (int)(time(NULL) - start_time));
exit (0);
}
/*
* Simulate sample from the audio device.
*/
int audio_get (void)
{
int ch;
ch = getc(fp);
if (ch < 0) e_o_f = 1;
return (ch);
}
/*
* Rather than queuing up frames with bad FCS,
* try to fix them immediately.
*/
void rdq_append (rrbb_t rrbb)
{
int chan;
int alevel;
int subchan;
chan = rrbb_get_chan(rrbb);
subchan = rrbb_get_subchan(rrbb);
alevel = rrbb_get_audio_level(rrbb);
hdlc_rec2_try_to_fix_later (rrbb, chan, subchan, alevel);
}
/*
* This is called when we have a good frame.
*/
void app_process_rec_packet (int chan, int subchan, packet_t pp, int alevel, retry_t retries, char *spectrum)
{
//int err;
//char *p;
char stemp[500];
unsigned char *pinfo;
int info_len;
int h;
char heard[20];
//packet_t pp;
packets_decoded++;
ax25_format_addrs (pp, stemp);
info_len = ax25_get_info (pp, &pinfo);
/* Print so we can see what is going on. */
#if 1
/* Display audio input level. */
/* Who are we hearing? Original station or digipeater. */
h = ax25_get_heard(pp);
ax25_get_addr_with_ssid(pp, h, heard);
text_color_set(DW_COLOR_DEBUG);
printf ("\n");
printf("DECODED[%d] ", packets_decoded );
if (h != AX25_SOURCE) {
printf ("Digipeater ");
}
printf ("%s audio level = %d [%s] %s\n", heard, alevel, retry_text[(int)retries], spectrum);
#endif
// Display non-APRS packets in a different color.
if (ax25_is_aprs(pp)) {
text_color_set(DW_COLOR_REC);
printf ("[%d] ", chan);
}
else {
text_color_set(DW_COLOR_DEBUG);
printf ("[%d] ", chan);
}
printf ("%s", stemp); /* stations followed by : */
ax25_safe_print ((char *)pinfo, info_len, 0);
printf ("\n");
ax25_delete (pp);
} /* end app_process_rec_packet */
/* end atest.c */
Reference in New Issue View Git Blame Copy Permalink