Update preset.
[bertos.git] / bertos / net / afsk_test.c
1 /**
2  * \file
3  * <!--
4  * This file is part of BeRTOS.
5  *
6  * Bertos is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19  *
20  * As a special exception, you may use this file as part of a free software
21  * library without restriction.  Specifically, if other files instantiate
22  * templates or use macros or inline functions from this file, or you compile
23  * this file and link it with other files to produce an executable, this
24  * file does not by itself cause the resulting executable to be covered by
25  * the GNU General Public License.  This exception does not however
26  * invalidate any other reasons why the executable file might be covered by
27  * the GNU General Public License.
28  *
29  * Copyright 2009 Develer S.r.l. (http://www.develer.com/)
30  *
31  * -->
32  *
33  * \brief AFSK demodulator test.
34  *
35  * \author Francesco Sacchi <batt@develer.com>
36  * $test$: cp bertos/cfg/cfg_ax25.h $cfgdir/
37  * $test$: echo "#undef AX25_LOG_LEVEL" >> $cfgdir/cfg_ax25.h
38  * $test$: echo "#define AX25_LOG_LEVEL LOG_LVL_INFO" >> $cfgdir/cfg_ax25.h
39  * $test$: cp bertos/cfg/cfg_afsk.h $cfgdir/
40  * $test$: echo "#undef CONFIG_AFSK_TX_BUFLEN" >> $cfgdir/cfg_afsk.h
41  * $test$: echo "#define CONFIG_AFSK_TX_BUFLEN 512" >> $cfgdir/cfg_afsk.h
42  */
43
44
45 #include "afsk.h"
46 #include "cfg/cfg_afsk.h"
47
48 #include <drv/timer.h>
49 #include <net/ax25.h>
50
51 #include <cfg/test.h>
52 #include <cfg/debug.h>
53 #include <cfg/kfile_debug.h>
54
55 #include <cpu/byteorder.h>
56
57 #include <stdio.h>
58 #include <string.h>
59
60 FILE *fp_adc;
61 FILE *fp_dac;
62 uint32_t data_size;
63 uint32_t data_written;
64 Afsk afsk_fd;
65 AX25Ctx ax25;
66 KFileDebug dbg;
67
68 int msg_cnt;
69 static void message_hook(struct AX25Msg *msg)
70 {
71         msg_cnt++;
72         ax25_print(&dbg.fd, msg);
73 }
74
75 static FILE *afsk_fileOpen(const char *name)
76 {
77         FILE *fp = 0;
78         #if CPU_AVR
79                 (void)name;
80                 #warning TODO: open the file?
81         #else
82                 fp = fopen(name, "rb");
83         #endif
84         ASSERT(fp);
85
86         char snd[5];
87         ASSERT(fread(snd, 1, 4, fp) == 4);
88         snd[4] = 0;
89         ASSERT(strcmp(snd, ".snd") == 0);
90
91         uint32_t offset;
92         ASSERT(fread(&offset, 1, sizeof(offset), fp) == sizeof(offset));
93         offset = be32_to_cpu(offset);
94         kprintf("AU file offset: %ld\n", (long)offset);
95         ASSERT(offset >= 24);
96
97         ASSERT(fread(&data_size, 1, sizeof(data_size), fp) == sizeof(data_size));
98         data_size = be32_to_cpu(data_size);
99         kprintf("AU file data_size: %ld\n", (long)data_size);
100         ASSERT(data_size);
101
102         uint32_t encoding;
103         ASSERT(fread(&encoding, 1, sizeof(encoding), fp) == sizeof(encoding));
104         encoding = be32_to_cpu(encoding);
105         kprintf("AU file encoding: %ld\n", (long)encoding);
106         ASSERT(encoding == 2); // 8 bit linear PCM
107
108         uint32_t sample_rate;
109         ASSERT(fread(&sample_rate, 1, sizeof(sample_rate), fp) == sizeof(sample_rate));
110         sample_rate = be32_to_cpu(sample_rate);
111         kprintf("AU file sample_rate: %ld\n", (long)sample_rate);
112         ASSERT(sample_rate == 9600);
113
114         uint32_t channels;
115         ASSERT(fread(&channels, 1, sizeof(channels), fp) == sizeof(channels));
116         channels = be32_to_cpu(channels);
117         kprintf("AU file channels: %ld\n", (long)channels);
118         ASSERT(channels == 1);
119
120         #if CPU_AVR
121                 #warning TODO: fseek?
122         #else
123                 ASSERT(fseek(fp, offset, SEEK_SET) == 0);
124         #endif
125         return fp;
126 }
127
128 int afsk_testSetup(void)
129 {
130         kdbg_init();
131         kfiledebug_init(&dbg);
132         fp_adc = afsk_fileOpen("test/afsk_test.au");
133         #if CPU_AVR
134                 #warning TODO: open the file?
135         #else
136                 fp_dac = fopen("test/afsk_test_out.au", "w+b");
137         #endif
138         ASSERT(fp_dac);
139         #define FS_HH (((uint32_t)CONFIG_AFSK_DAC_SAMPLERATE) >> 24)
140         #define FS_HL ((((uint32_t)CONFIG_AFSK_DAC_SAMPLERATE) >> 16) & 0xff)
141         #define FS_LH ((((uint32_t)CONFIG_AFSK_DAC_SAMPLERATE) >> 8) & 0xff)
142         #define FS_LL (((uint32_t)CONFIG_AFSK_DAC_SAMPLERATE) & 0xff)
143
144         uint8_t snd_header[] = { '.','s','n','d', 0,0,0,24, 0,0,0,0, 0,0,0,2, FS_HH,FS_HL,FS_LH,FS_LL, 0,0,0,1};
145
146         ASSERT(fwrite(snd_header, 1, sizeof(snd_header), fp_dac) == sizeof(snd_header));
147
148         timer_init();
149         afsk_init(&afsk_fd, 0 ,0);
150         ax25_init(&ax25, &afsk_fd.fd, message_hook);
151         return 0;
152 }
153
154
155 static void messageout_hook(struct AX25Msg *msg)
156 {
157         ASSERT(strncmp(msg->dst.call, "ABCDEF", 6) == 0);
158         ASSERT(strncmp(msg->src.call, "123456", 6) == 0);
159         ASSERT(msg->src.ssid == 1);
160         ASSERT(msg->dst.ssid == 0);
161         ASSERT(msg->ctrl == AX25_CTRL_UI);
162         ASSERT(msg->pid == AX25_PID_NOLAYER3);
163         ASSERT(msg->len == 256);
164         for (int i = 0; i < 256; i++)
165                 ASSERT(msg->info[i] == i);
166 }
167
168 int afsk_testRun(void)
169 {
170         int c;
171         while ((c = fgetc(fp_adc)) != EOF)
172         {
173                 afsk_adc_isr(&afsk_fd, (int8_t)c);
174
175                 ax25_poll(&ax25);
176         }
177         kprintf("Messages correctly received: %d\n", msg_cnt);
178         ASSERT(msg_cnt >= 15);
179
180         char buf[256];
181         for (unsigned i = 0; i < sizeof(buf); i++)
182                 buf[i] = i;
183
184         ax25_send(&ax25, AX25_CALL("abcdef", 0), AX25_CALL("123456", 1), buf, sizeof(buf));
185
186         do
187         {
188                 int8_t val = afsk_dac_isr(&afsk_fd) - 128;
189                 ASSERT(fwrite(&val, 1, sizeof(val), fp_dac) == sizeof(val));
190                 data_written++;
191         }
192         while (afsk_fd.sending);
193
194         #define SND_DATASIZE_OFF 8
195         #if CPU_AVR
196                 #warning TODO: fseek?
197         #else
198                 ASSERT(fseek(fp_dac, SND_DATASIZE_OFF, SEEK_SET) == 0);
199         #endif
200         data_written = cpu_to_be32(data_written);
201         ASSERT(fwrite(&data_written, 1, sizeof(data_written), fp_dac) == sizeof(data_written));
202         ASSERT(fclose(fp_adc) + fclose(fp_dac) == 0);
203
204         fp_adc = afsk_fileOpen("test/afsk_test_out.au");
205         ax25_init(&ax25, &afsk_fd.fd, messageout_hook);
206
207         while ((c = fgetc(fp_adc)) != EOF)
208         {
209                 afsk_adc_isr(&afsk_fd, (int8_t)c);
210
211                 ax25_poll(&ax25);
212         }
213
214         return 0;
215 }
216
217 int afsk_testTearDown(void)
218 {
219         return fclose(fp_adc);
220 }
221
222 TEST_MAIN(afsk);