altos: Make profiling Viterbi decoder more useful
[fw/altos] / src / drivers / ao_packet.c
1 /*
2  * Copyright © 2009 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 #include "ao.h"
19
20 __xdata struct ao_packet_recv ao_rx_packet;
21 __xdata struct ao_packet ao_tx_packet;
22 __pdata uint8_t ao_packet_rx_len, ao_packet_rx_used, ao_packet_tx_used;
23
24 static __xdata char tx_data[AO_PACKET_MAX];
25 static __xdata char rx_data[AO_PACKET_MAX];
26 static __pdata uint8_t rx_seq;
27
28 __xdata struct ao_task  ao_packet_task;
29 __xdata uint8_t ao_packet_enable;
30 __xdata uint8_t ao_packet_master_sleeping;
31
32 void
33 ao_packet_send(void)
34 {
35 #ifdef AO_LED_RED
36         ao_led_on(AO_LED_RED);
37 #endif
38         /* If any tx data is pending then copy it into the tx packet */
39         if (ao_packet_tx_used && ao_tx_packet.len == 0) {
40                 ao_xmemcpy(&ao_tx_packet.d, tx_data, ao_packet_tx_used);
41                 ao_tx_packet.len = ao_packet_tx_used;
42                 ao_tx_packet.seq++;
43                 ao_packet_tx_used = 0;
44                 ao_wakeup(&tx_data);
45         }
46         ao_radio_send(&ao_tx_packet, sizeof (ao_tx_packet));
47 #ifdef AO_LED_RED
48         ao_led_off(AO_LED_RED);
49 #endif
50 }
51
52 uint8_t
53 ao_packet_recv(void)
54 {
55         uint8_t dma_done;
56
57 #ifdef AO_LED_GREEN
58         ao_led_on(AO_LED_GREEN);
59 #endif
60         dma_done = ao_radio_recv(&ao_rx_packet, sizeof (struct ao_packet_recv));
61 #ifdef AO_LED_GREEN
62         ao_led_off(AO_LED_GREEN);
63 #endif
64 #if AO_PROFILE
65                 {
66                         extern uint32_t ao_rx_start_tick, ao_rx_packet_tick, ao_rx_done_tick, ao_rx_last_done_tick;
67                         extern uint32_t ao_fec_decode_start, ao_fec_decode_end;
68
69                         printf ("between packet: %d\n", ao_rx_start_tick - ao_rx_last_done_tick);
70                         printf ("receive start delay: %d\n", ao_rx_packet_tick - ao_rx_start_tick);
71                         printf ("decode time: %d\n", ao_fec_decode_end - ao_fec_decode_start);
72                         printf ("rx cleanup: %d\n\n", ao_rx_done_tick - ao_fec_decode_end);
73                         flush();
74                 }
75 #endif
76
77         /* Check to see if we got a valid packet */
78         if (!dma_done)
79                 return 0;
80         if (!(ao_rx_packet.status & AO_RADIO_STATUS_CRC_OK))
81                 return 0;
82
83         /* Accept packets with matching call signs, or any packet if
84          * our callsign hasn't been configured
85          */
86         if (ao_xmemcmp(ao_rx_packet.packet.callsign,
87                        ao_config.callsign,
88                        AO_MAX_CALLSIGN) != 0 &&
89             ao_xmemcmp(ao_config.callsign, CODE_TO_XDATA("N0CALL"), 7) != 0)
90                 return 0;
91
92         /* SYN packets carry no data */
93         if (ao_rx_packet.packet.len == AO_PACKET_SYN) {
94                 rx_seq = ao_rx_packet.packet.seq;
95                 ao_tx_packet.seq = ao_rx_packet.packet.ack;
96                 ao_tx_packet.ack = rx_seq;
97         } else if (ao_rx_packet.packet.len) {
98
99                 /* Check for incoming data at the next sequence and
100                  * for an empty data buffer
101                  */
102                 if (ao_rx_packet.packet.seq == (uint8_t) (rx_seq + (uint8_t) 1) &&
103                     ao_packet_rx_used == ao_packet_rx_len) {
104
105                         /* Copy data to the receive data buffer and set up the
106                          * offsets
107                          */
108                         ao_xmemcpy(rx_data, ao_rx_packet.packet.d, ao_rx_packet.packet.len);
109                         ao_packet_rx_used = 0;
110                         ao_packet_rx_len = ao_rx_packet.packet.len;
111
112                         /* Mark the sequence that we've received to
113                          * let the sender know when we return a packet
114                          */
115                         rx_seq = ao_rx_packet.packet.seq;
116                         ao_tx_packet.ack = rx_seq;
117
118                         /* Poke anyone looking for received data */
119                         ao_wakeup(&ao_stdin_ready);
120                 }
121         }
122
123         /* If the other side has seen the latest data we queued,
124          * wake up any task waiting to send data and let them go again
125          */
126         if (ao_rx_packet.packet.ack == ao_tx_packet.seq) {
127                 ao_tx_packet.len = 0;
128                 ao_wakeup(&ao_tx_packet);
129         }
130         return 1;
131 }
132
133 #ifndef PACKET_HAS_MASTER
134 #define PACKET_HAS_MASTER 1
135 #endif
136
137 #if PACKET_HAS_MASTER
138 void
139 ao_packet_flush(void)
140 {
141         /* If there is data to send, and this is the master,
142          * then poke the master to send all queued data
143          */
144         if (ao_packet_tx_used && ao_packet_master_sleeping)
145                 ao_wakeup(&ao_packet_master_sleeping);
146 }
147 #endif /* PACKET_HAS_MASTER */
148
149 void
150 ao_packet_putchar(char c) __reentrant
151 {
152         while (ao_packet_tx_used == AO_PACKET_MAX && ao_packet_enable) {
153 #if PACKET_HAS_MASTER
154                 ao_packet_flush();
155 #endif
156                 ao_sleep(&tx_data);
157         }
158
159         if (ao_packet_enable)
160                 tx_data[ao_packet_tx_used++] = c;
161 }
162
163 char
164 ao_packet_pollchar(void) __critical
165 {
166         if (!ao_packet_enable)
167                 return AO_READ_AGAIN;
168
169         if (ao_packet_rx_used == ao_packet_rx_len)
170                 return AO_READ_AGAIN;
171
172         return rx_data[ao_packet_rx_used++];
173 }