xref: /btstack/example/le_audio_demo_util_sink.c (revision 7a6944e61752feee97128a898495f932798eaf2a)
124b69d49SMatthias Ringwald /*
224b69d49SMatthias Ringwald  * Copyright (C) {copyright_year} BlueKitchen GmbH
324b69d49SMatthias Ringwald  *
424b69d49SMatthias Ringwald  * Redistribution and use in source and binary forms, with or without
524b69d49SMatthias Ringwald  * modification, are permitted provided that the following conditions
624b69d49SMatthias Ringwald  * are met:
724b69d49SMatthias Ringwald  *
824b69d49SMatthias Ringwald  * 1. Redistributions of source code must retain the above copyright
924b69d49SMatthias Ringwald  *    notice, this list of conditions and the following disclaimer.
1024b69d49SMatthias Ringwald  * 2. Redistributions in binary form must reproduce the above copyright
1124b69d49SMatthias Ringwald  *    notice, this list of conditions and the following disclaimer in the
1224b69d49SMatthias Ringwald  *    documentation and/or other materials provided with the distribution.
1324b69d49SMatthias Ringwald  * 3. Neither the name of the copyright holders nor the names of
1424b69d49SMatthias Ringwald  *    contributors may be used to endorse or promote products derived
1524b69d49SMatthias Ringwald  *    from this software without specific prior written permission.
1624b69d49SMatthias Ringwald  * 4. Any redistribution, use, or modification is done solely for
1724b69d49SMatthias Ringwald  *    personal benefit and not for any commercial purpose or for
1824b69d49SMatthias Ringwald  *    monetary gain.
1924b69d49SMatthias Ringwald  *
2024b69d49SMatthias Ringwald  * THIS SOFTWARE IS PROVIDED BY BLUEKITCHEN GMBH AND CONTRIBUTORS
2124b69d49SMatthias Ringwald  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
2224b69d49SMatthias Ringwald  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
2324b69d49SMatthias Ringwald  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL BLUEKITCHEN
2424b69d49SMatthias Ringwald  * GMBH OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
2524b69d49SMatthias Ringwald  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
2624b69d49SMatthias Ringwald  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
2724b69d49SMatthias Ringwald  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
2824b69d49SMatthias Ringwald  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
2924b69d49SMatthias Ringwald  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
3024b69d49SMatthias Ringwald  * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3124b69d49SMatthias Ringwald  * SUCH DAMAGE.
3224b69d49SMatthias Ringwald  *
3324b69d49SMatthias Ringwald  * Please inquire about commercial licensing options at
3424b69d49SMatthias Ringwald  * [email protected]
3524b69d49SMatthias Ringwald  *
3624b69d49SMatthias Ringwald  */
3724b69d49SMatthias Ringwald 
3824b69d49SMatthias Ringwald #define BTSTACK_FILE__ "le_audio_demo_util_sink.c"
3924b69d49SMatthias Ringwald 
4078d87b81SDirk Helbig #include <stdio.h>
41a18990d3SMatthias Ringwald #include <inttypes.h>
4278d87b81SDirk Helbig 
4324b69d49SMatthias Ringwald #include "le_audio_demo_util_sink.h"
4424b69d49SMatthias Ringwald 
4524b69d49SMatthias Ringwald #include "btstack_bool.h"
4624b69d49SMatthias Ringwald #include "btstack_config.h"
4724b69d49SMatthias Ringwald #include <btstack_debug.h>
48a18990d3SMatthias Ringwald #include <stdio.h>
4924b69d49SMatthias Ringwald 
5024b69d49SMatthias Ringwald #include "hci.h"
5124b69d49SMatthias Ringwald #include "btstack_audio.h"
5224b69d49SMatthias Ringwald #include "btstack_lc3_google.h"
5324b69d49SMatthias Ringwald #include "btstack_lc3plus_fraunhofer.h"
5424b69d49SMatthias Ringwald 
5578d87b81SDirk Helbig #include "btstack_sample_rate_compensation.h"
5678d87b81SDirk Helbig #include "btstack_resample.h"
57c18c19deSDirk Helbig #include "btstack_fsm.h"
5878d87b81SDirk Helbig 
5924b69d49SMatthias Ringwald #include "hxcmod.h"
6024b69d49SMatthias Ringwald #include "mods/mod.h"
6124b69d49SMatthias Ringwald 
62a18990d3SMatthias Ringwald #include "btstack_ring_buffer.h"
6324b69d49SMatthias Ringwald #ifdef HAVE_POSIX_FILE_IO
6424b69d49SMatthias Ringwald #include "wav_util.h"
6524b69d49SMatthias Ringwald #endif
6624b69d49SMatthias Ringwald 
6724b69d49SMatthias Ringwald #define MAX_CHANNELS 2
6824b69d49SMatthias Ringwald #define MAX_SAMPLES_PER_FRAME 480
6924b69d49SMatthias Ringwald #define MAX_LC3_FRAME_BYTES   155
7024b69d49SMatthias Ringwald 
7124b69d49SMatthias Ringwald // playback
72c18c19deSDirk Helbig #define MAX_NUM_LC3_FRAMES   (15*2)
7324b69d49SMatthias Ringwald #define MAX_BYTES_PER_SAMPLE 4
748d446e2bSDirk Helbig #define PLAYBACK_BUFFER_SIZE (MAX_NUM_LC3_FRAMES * MAX_SAMPLES_PER_FRAME * MAX_CHANNELS * MAX_BYTES_PER_SAMPLE)
7524b69d49SMatthias Ringwald #define PLAYBACK_START_MS (MAX_NUM_LC3_FRAMES * 20 / 3)
7624b69d49SMatthias Ringwald 
77a18990d3SMatthias Ringwald // analysis
78a18990d3SMatthias Ringwald #define PACKET_PREFIX_LEN 10
79a18990d3SMatthias Ringwald 
8024b69d49SMatthias Ringwald #define ANSI_COLOR_RED     "\x1b[31m"
8124b69d49SMatthias Ringwald #define ANSI_COLOR_GREEN   "\x1b[32m"
8224b69d49SMatthias Ringwald #define ANSI_COLOR_YELLOW  "\x1b[33m"
8324b69d49SMatthias Ringwald #define ANSI_COLOR_BLUE    "\x1b[34m"
8424b69d49SMatthias Ringwald #define ANSI_COLOR_MAGENTA "\x1b[35m"
8524b69d49SMatthias Ringwald #define ANSI_COLOR_CYAN    "\x1b[36m"
8624b69d49SMatthias Ringwald #define ANSI_COLOR_RESET   "\x1b[0m"
8724b69d49SMatthias Ringwald 
88a18990d3SMatthias Ringwald // statistics
89a18990d3SMatthias Ringwald static uint16_t last_packet_sequence[MAX_CHANNELS];
90a18990d3SMatthias Ringwald static uint32_t last_packet_time_ms[MAX_CHANNELS];
91a18990d3SMatthias Ringwald static uint8_t  last_packet_prefix[MAX_CHANNELS * PACKET_PREFIX_LEN];
92a18990d3SMatthias Ringwald 
9324b69d49SMatthias Ringwald // SINK
9424b69d49SMatthias Ringwald 
950f01c821SMatthias Ringwald static enum {
960f01c821SMatthias Ringwald     LE_AUDIO_SINK_IDLE,
970f01c821SMatthias Ringwald     LE_AUDIO_SINK_INIT,
980f01c821SMatthias Ringwald     LE_AUDIO_SINK_CONFIGURED,
990f01c821SMatthias Ringwald } le_audio_demo_util_sink_state = LE_AUDIO_SINK_IDLE;
1000f01c821SMatthias Ringwald 
10124b69d49SMatthias Ringwald static const char * le_audio_demo_sink_filename_wav;
10278d87b81SDirk Helbig static btstack_sample_rate_compensation_t sample_rate_compensation;
1038dee1629SMatthias Ringwald static uint32_t le_audio_demo_sink_received_samples;
10478d87b81SDirk Helbig static btstack_resample_t resample_instance;
10578d87b81SDirk Helbig static bool sink_receive_streaming;
10678d87b81SDirk Helbig 
10778d87b81SDirk Helbig static int16_t pcm_resample[MAX_CHANNELS * MAX_SAMPLES_PER_FRAME * 2];
10878d87b81SDirk Helbig 
10924b69d49SMatthias Ringwald 
11024b69d49SMatthias Ringwald static btstack_lc3_frame_duration_t le_audio_demo_sink_frame_duration;
11124b69d49SMatthias Ringwald static hci_iso_type_t               le_audio_demo_sink_type;
11224b69d49SMatthias Ringwald 
11324b69d49SMatthias Ringwald static uint32_t le_audio_demo_sink_sampling_frequency_hz;
11424b69d49SMatthias Ringwald static uint16_t le_audio_demo_sink_num_samples_per_frame;
11524b69d49SMatthias Ringwald static uint8_t  le_audio_demo_sink_num_streams;
11624b69d49SMatthias Ringwald static uint8_t  le_audio_demo_sink_num_channels_per_stream;
11724b69d49SMatthias Ringwald static uint8_t  le_audio_demo_sink_num_channels;
11824b69d49SMatthias Ringwald static uint16_t le_audio_demo_sink_octets_per_frame;
11924b69d49SMatthias Ringwald static uint16_t le_audio_demo_sink_iso_interval_1250us;
12024b69d49SMatthias Ringwald static uint8_t  le_audio_demo_sink_flush_timeout;
12124b69d49SMatthias Ringwald static uint8_t  le_audio_demo_sink_pre_transmission_offset;
12224b69d49SMatthias Ringwald 
12324b69d49SMatthias Ringwald // playback
12424b69d49SMatthias Ringwald static uint16_t              playback_start_threshold_bytes;
12524b69d49SMatthias Ringwald static bool                  playback_active;
12624b69d49SMatthias Ringwald static uint8_t               playback_buffer_storage[PLAYBACK_BUFFER_SIZE];
12724b69d49SMatthias Ringwald static btstack_ring_buffer_t playback_buffer;
12824b69d49SMatthias Ringwald 
12924b69d49SMatthias Ringwald // PLC
13024b69d49SMatthias Ringwald static uint32_t le_audio_demo_sink_lc3_frames;
13124b69d49SMatthias Ringwald static uint32_t samples_received;
13224b69d49SMatthias Ringwald static uint32_t samples_played;
13324b69d49SMatthias Ringwald static uint32_t samples_dropped;
13424b69d49SMatthias Ringwald 
135c18c19deSDirk Helbig // Audio FSM
136c18c19deSDirk Helbig #define TRAN( target ) btstack_fsm_transit( &me->super, (btstack_fsm_state_handler_t)target )
137c18c19deSDirk Helbig 
138c18c19deSDirk Helbig typedef struct {
139c18c19deSDirk Helbig     btstack_fsm_t super;
140c18c19deSDirk Helbig     uint32_t receive_time_ms;
141c18c19deSDirk Helbig     uint32_t last_receive_time_ms;
142c18c19deSDirk Helbig     uint32_t zero_frames;
143c18c19deSDirk Helbig     uint32_t have_pcm;
144c18c19deSDirk Helbig     uint32_t received_samples;
145c18c19deSDirk Helbig } audio_processing_t;
146c18c19deSDirk Helbig 
147c18c19deSDirk Helbig typedef struct {
148c18c19deSDirk Helbig     btstack_fsm_event_t super;
149c18c19deSDirk Helbig     uint16_t sequence_number;
150c18c19deSDirk Helbig     uint16_t size;
151c18c19deSDirk Helbig     uint32_t receive_time_ms;
152c18c19deSDirk Helbig     uint8_t *data;
153c18c19deSDirk Helbig     uint8_t stream;
154c18c19deSDirk Helbig } data_event_t;
155c18c19deSDirk Helbig 
156c18c19deSDirk Helbig typedef struct {
157c18c19deSDirk Helbig     btstack_fsm_event_t super;
158c18c19deSDirk Helbig     uint32_t time_ms;
159c18c19deSDirk Helbig } time_event_t;
160c18c19deSDirk Helbig 
161c18c19deSDirk Helbig audio_processing_t audio_processing;
162c18c19deSDirk Helbig 
163c18c19deSDirk Helbig enum EventSignals {
164c18c19deSDirk Helbig     DATA_SIG = BTSTACK_FSM_USER_SIG,
165c18c19deSDirk Helbig     TIME_SIG
166c18c19deSDirk Helbig };
167c18c19deSDirk Helbig 
1684b3f8617SDirk Helbig #define AUDIO_FSM_DEBUGx
169c18c19deSDirk Helbig #ifdef AUDIO_FSM_DEBUG
170c18c19deSDirk Helbig #define ENUM_TO_TEXT(sig) [sig] = #sig
171c18c19deSDirk Helbig #define audio_fsm_debug(format, ...) \
172c18c19deSDirk Helbig   printf( format __VA_OPT__(,) __VA_ARGS__)
173c18c19deSDirk Helbig 
174c18c19deSDirk Helbig const char * const sigToString[] = {
175c18c19deSDirk Helbig         ENUM_TO_TEXT(BTSTACK_FSM_INIT_SIG),
176c18c19deSDirk Helbig         ENUM_TO_TEXT(BTSTACK_FSM_ENTRY_SIG),
177c18c19deSDirk Helbig         ENUM_TO_TEXT(BTSTACK_FSM_EXIT_SIG),
178c18c19deSDirk Helbig         ENUM_TO_TEXT(DATA_SIG),
179c18c19deSDirk Helbig         ENUM_TO_TEXT(TIME_SIG),
180c18c19deSDirk Helbig };
181c18c19deSDirk Helbig #else
182c18c19deSDirk Helbig #define audio_fsm_debug(...)
183c18c19deSDirk Helbig #endif
184c18c19deSDirk Helbig 
185c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_initial( audio_processing_t * const me, btstack_fsm_event_t const * const e );
186c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_waiting( audio_processing_t * const me, btstack_fsm_event_t const * const e );
187c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_streaming( audio_processing_t * const me, btstack_fsm_event_t const * const e );
188c18c19deSDirk Helbig static bool audio_processing_is_streaming( audio_processing_t * const me );
189c18c19deSDirk Helbig 
19024b69d49SMatthias Ringwald static btstack_timer_source_t next_packet_timer;
19124b69d49SMatthias Ringwald 
19224b69d49SMatthias Ringwald // lc3 decoder
19324b69d49SMatthias Ringwald static bool le_audio_demo_lc3plus_decoder_requested = false;
19424b69d49SMatthias Ringwald static const btstack_lc3_decoder_t * lc3_decoder;
19524b69d49SMatthias Ringwald static int16_t pcm[MAX_CHANNELS * MAX_SAMPLES_PER_FRAME];
19624b69d49SMatthias Ringwald 
19724b69d49SMatthias Ringwald static btstack_lc3_decoder_google_t google_decoder_contexts[MAX_CHANNELS];
19824b69d49SMatthias Ringwald #ifdef HAVE_LC3PLUS
19924b69d49SMatthias Ringwald static btstack_lc3plus_fraunhofer_decoder_t fraunhofer_decoder_contexts[MAX_CHANNELS];
20024b69d49SMatthias Ringwald #endif
20124b69d49SMatthias Ringwald static void * decoder_contexts[MAX_CHANNELS];
20224b69d49SMatthias Ringwald 
le_audio_connection_sink_playback(int16_t * buffer,uint16_t num_samples)20324b69d49SMatthias Ringwald static void le_audio_connection_sink_playback(int16_t * buffer, uint16_t num_samples){
20424b69d49SMatthias Ringwald     // called from lower-layer but guaranteed to be on main thread
205*7a6944e6SMatthias Ringwald     log_info("Playback: need %u, have %" PRIu32 "", num_samples, btstack_ring_buffer_bytes_available(&playback_buffer) / (le_audio_demo_sink_num_channels * 2));
20624b69d49SMatthias Ringwald 
20724b69d49SMatthias Ringwald     samples_played += num_samples;
20824b69d49SMatthias Ringwald 
20924b69d49SMatthias Ringwald     uint32_t bytes_needed = num_samples * le_audio_demo_sink_num_channels * 2;
21024b69d49SMatthias Ringwald     if (playback_active == false){
21124b69d49SMatthias Ringwald         if (btstack_ring_buffer_bytes_available(&playback_buffer) >= playback_start_threshold_bytes) {
21224b69d49SMatthias Ringwald             log_info("Playback started");
213c18c19deSDirk Helbig             printf("Playback started\n");
21424b69d49SMatthias Ringwald             playback_active = true;
21524b69d49SMatthias Ringwald         }
21624b69d49SMatthias Ringwald     } else {
21724b69d49SMatthias Ringwald         if (bytes_needed > btstack_ring_buffer_bytes_available(&playback_buffer)) {
218c18c19deSDirk Helbig             if( audio_processing_is_streaming( &audio_processing ) ) {
21924b69d49SMatthias Ringwald                 log_info("Playback underrun");
22024b69d49SMatthias Ringwald                 printf("Playback Underrun\n");
221c18c19deSDirk Helbig             } else {
222c18c19deSDirk Helbig                 log_info("Playback stopped");
223c18c19deSDirk Helbig                 printf("Playback stopped\n");
224c18c19deSDirk Helbig             }
22524b69d49SMatthias Ringwald             // empty buffer
22624b69d49SMatthias Ringwald             uint32_t bytes_read;
22724b69d49SMatthias Ringwald             btstack_ring_buffer_read(&playback_buffer, (uint8_t *) buffer, bytes_needed, &bytes_read);
22824b69d49SMatthias Ringwald             playback_active = false;
22924b69d49SMatthias Ringwald         }
23024b69d49SMatthias Ringwald     }
23124b69d49SMatthias Ringwald 
23224b69d49SMatthias Ringwald     if (playback_active){
23324b69d49SMatthias Ringwald         uint32_t bytes_read;
23424b69d49SMatthias Ringwald         btstack_ring_buffer_read(&playback_buffer, (uint8_t *) buffer, bytes_needed, &bytes_read);
23524b69d49SMatthias Ringwald         btstack_assert(bytes_read == bytes_needed);
23624b69d49SMatthias Ringwald     } else {
23724b69d49SMatthias Ringwald         memset(buffer, 0, bytes_needed);
23824b69d49SMatthias Ringwald     }
23924b69d49SMatthias Ringwald }
24024b69d49SMatthias Ringwald 
le_audio_demo_util_sink_enable_lc3plus(bool enable)24124b69d49SMatthias Ringwald void le_audio_demo_util_sink_enable_lc3plus(bool enable){
24224b69d49SMatthias Ringwald     le_audio_demo_lc3plus_decoder_requested = enable;
24324b69d49SMatthias Ringwald }
24424b69d49SMatthias Ringwald 
setup_lc3_decoder(bool use_lc3plus_decoder)245bc66a7deSMatthias Ringwald static void setup_lc3_decoder(bool use_lc3plus_decoder){
246*7a6944e6SMatthias Ringwald     UNUSED(use_lc3plus_decoder);
247*7a6944e6SMatthias Ringwald 
248f4c19309SMatthias Ringwald     uint8_t channel;
249f4c19309SMatthias Ringwald     for (channel = 0 ; channel < le_audio_demo_sink_num_channels ; channel++){
250f4c19309SMatthias Ringwald         // pick decoder
251f4c19309SMatthias Ringwald         void * decoder_context = NULL;
252f4c19309SMatthias Ringwald #ifdef HAVE_LC3PLUS
253f4c19309SMatthias Ringwald         if (use_lc3plus_decoder){
254f4c19309SMatthias Ringwald             decoder_context = &fraunhofer_decoder_contexts[channel];
255f4c19309SMatthias Ringwald             lc3_decoder = btstack_lc3plus_fraunhofer_decoder_init_instance(decoder_context);
256f4c19309SMatthias Ringwald         }
257f4c19309SMatthias Ringwald         else
258f4c19309SMatthias Ringwald #endif
259f4c19309SMatthias Ringwald         {
260f4c19309SMatthias Ringwald             decoder_context = &google_decoder_contexts[channel];
261f4c19309SMatthias Ringwald             lc3_decoder = btstack_lc3_decoder_google_init_instance(decoder_context);
262f4c19309SMatthias Ringwald         }
263f4c19309SMatthias Ringwald         decoder_contexts[channel] = decoder_context;
264f4c19309SMatthias Ringwald         lc3_decoder->configure(decoder_context, le_audio_demo_sink_sampling_frequency_hz, le_audio_demo_sink_frame_duration, le_audio_demo_sink_octets_per_frame);
265f4c19309SMatthias Ringwald     }
266f4c19309SMatthias Ringwald     btstack_assert(le_audio_demo_sink_num_samples_per_frame <= MAX_SAMPLES_PER_FRAME);
267f4c19309SMatthias Ringwald }
268f4c19309SMatthias Ringwald 
le_audio_demo_util_sink_configure_general(uint8_t num_streams,uint8_t num_channels_per_stream,uint32_t sampling_frequency_hz,btstack_lc3_frame_duration_t frame_duration,uint16_t octets_per_frame,uint32_t iso_interval_1250us)26924b69d49SMatthias Ringwald void le_audio_demo_util_sink_configure_general(uint8_t num_streams, uint8_t num_channels_per_stream,
27024b69d49SMatthias Ringwald                                                uint32_t sampling_frequency_hz,
27124b69d49SMatthias Ringwald                                                btstack_lc3_frame_duration_t frame_duration, uint16_t octets_per_frame,
27224b69d49SMatthias Ringwald                                                uint32_t iso_interval_1250us) {
27324b69d49SMatthias Ringwald     le_audio_demo_sink_sampling_frequency_hz = sampling_frequency_hz;
27424b69d49SMatthias Ringwald     le_audio_demo_sink_frame_duration = frame_duration;
27524b69d49SMatthias Ringwald     le_audio_demo_sink_octets_per_frame = octets_per_frame;
27624b69d49SMatthias Ringwald     le_audio_demo_sink_iso_interval_1250us = iso_interval_1250us;
27724b69d49SMatthias Ringwald     le_audio_demo_sink_num_streams = num_streams;
27824b69d49SMatthias Ringwald     le_audio_demo_sink_num_channels_per_stream = num_channels_per_stream;
27924b69d49SMatthias Ringwald 
28078d87b81SDirk Helbig     sink_receive_streaming = false;
2810f01c821SMatthias Ringwald     le_audio_demo_util_sink_state = LE_AUDIO_SINK_CONFIGURED;
28278d87b81SDirk Helbig 
28324b69d49SMatthias Ringwald     le_audio_demo_sink_num_channels = le_audio_demo_sink_num_streams * le_audio_demo_sink_num_channels_per_stream;
28424b69d49SMatthias Ringwald     btstack_assert((le_audio_demo_sink_num_channels == 1) || (le_audio_demo_sink_num_channels == 2));
28524b69d49SMatthias Ringwald 
28624b69d49SMatthias Ringwald     le_audio_demo_sink_lc3_frames = 0;
28724b69d49SMatthias Ringwald 
28824b69d49SMatthias Ringwald     le_audio_demo_sink_num_samples_per_frame = btstack_lc3_samples_per_frame(le_audio_demo_sink_sampling_frequency_hz, le_audio_demo_sink_frame_duration);
28924b69d49SMatthias Ringwald 
29024b69d49SMatthias Ringwald     // switch to lc3plus if requested and possible
29124b69d49SMatthias Ringwald     bool use_lc3plus_decoder = le_audio_demo_lc3plus_decoder_requested && (frame_duration == BTSTACK_LC3_FRAME_DURATION_10000US);
29224b69d49SMatthias Ringwald 
29324b69d49SMatthias Ringwald     // init decoder
294bc66a7deSMatthias Ringwald     setup_lc3_decoder(use_lc3plus_decoder);
29524b69d49SMatthias Ringwald 
296*7a6944e6SMatthias Ringwald     printf("Configure: %u streams, %u channels per stream, sampling rate %" PRIu32 ", samples per frame %u, lc3plus %u\n",
29724b69d49SMatthias Ringwald            num_streams, num_channels_per_stream, sampling_frequency_hz, le_audio_demo_sink_num_samples_per_frame, use_lc3plus_decoder);
29824b69d49SMatthias Ringwald 
29924b69d49SMatthias Ringwald #ifdef HAVE_POSIX_FILE_IO
30024b69d49SMatthias Ringwald     // create wav file
30124b69d49SMatthias Ringwald     printf("WAV file: %s\n", le_audio_demo_sink_filename_wav);
30224b69d49SMatthias Ringwald     wav_writer_open(le_audio_demo_sink_filename_wav, le_audio_demo_sink_num_channels, le_audio_demo_sink_sampling_frequency_hz);
30324b69d49SMatthias Ringwald #endif
30424b69d49SMatthias Ringwald 
30524b69d49SMatthias Ringwald     // init playback buffer
30624b69d49SMatthias Ringwald     btstack_ring_buffer_init(&playback_buffer, playback_buffer_storage, PLAYBACK_BUFFER_SIZE);
30724b69d49SMatthias Ringwald 
30824b69d49SMatthias Ringwald     // calc start threshold in bytes for PLAYBACK_START_MS
30924b69d49SMatthias Ringwald     playback_start_threshold_bytes = (sampling_frequency_hz / 1000 * PLAYBACK_START_MS) * le_audio_demo_sink_num_channels * 2;
31024b69d49SMatthias Ringwald 
3118dee1629SMatthias Ringwald     // sample rate compensation
3128dee1629SMatthias Ringwald     le_audio_demo_sink_received_samples = 0;
3138dee1629SMatthias Ringwald 
31424b69d49SMatthias Ringwald     // start playback
31524b69d49SMatthias Ringwald     const btstack_audio_sink_t * sink = btstack_audio_sink_get_instance();
31624b69d49SMatthias Ringwald     if (sink != NULL){
31778d87b81SDirk Helbig         btstack_sample_rate_compensation_reset( &sample_rate_compensation, btstack_run_loop_get_time_ms() );
3180b541edfSMatthias Ringwald         btstack_resample_init(&resample_instance, le_audio_demo_sink_num_channels);
31924b69d49SMatthias Ringwald         sink->init(le_audio_demo_sink_num_channels, le_audio_demo_sink_sampling_frequency_hz, le_audio_connection_sink_playback);
32024b69d49SMatthias Ringwald         sink->start_stream();
32124b69d49SMatthias Ringwald     }
32224b69d49SMatthias Ringwald }
32324b69d49SMatthias Ringwald 
le_audio_demo_util_sink_configure_unicast(uint8_t num_streams,uint8_t num_channels_per_stream,uint32_t sampling_frequency_hz,btstack_lc3_frame_duration_t frame_duration,uint16_t octets_per_frame,uint32_t iso_interval_1250us,uint8_t flush_timeout)32424b69d49SMatthias Ringwald void le_audio_demo_util_sink_configure_unicast(uint8_t num_streams, uint8_t num_channels_per_stream, uint32_t sampling_frequency_hz,
32524b69d49SMatthias Ringwald                                                btstack_lc3_frame_duration_t frame_duration, uint16_t octets_per_frame,
32624b69d49SMatthias Ringwald                                                uint32_t iso_interval_1250us, uint8_t flush_timeout){
32724b69d49SMatthias Ringwald     le_audio_demo_sink_type = HCI_ISO_TYPE_CIS;
32824b69d49SMatthias Ringwald     le_audio_demo_sink_flush_timeout = flush_timeout;
32979d6b605SMatthias Ringwald 
330f4c19309SMatthias Ringwald     // set playback start: FT * ISO Interval + max(10 ms, 1/2 ISO Interval)
3310388b643SMatthias Ringwald     uint16_t playback_start_ms = flush_timeout * (iso_interval_1250us * 5 / 4) + btstack_max(10, iso_interval_1250us * 5 / 8);
33279d6b605SMatthias Ringwald     uint16_t playback_start_samples = sampling_frequency_hz / 1000 * playback_start_ms;
33379d6b605SMatthias Ringwald     playback_start_threshold_bytes = playback_start_samples * num_streams * num_channels_per_stream * 2;
33479d6b605SMatthias Ringwald     printf("Playback: start %u ms (%u samples, %u bytes)\n", playback_start_ms, playback_start_samples, playback_start_threshold_bytes);
33579d6b605SMatthias Ringwald 
33624b69d49SMatthias Ringwald     le_audio_demo_util_sink_configure_general(num_streams, num_channels_per_stream, sampling_frequency_hz,
33724b69d49SMatthias Ringwald                                               frame_duration, octets_per_frame, iso_interval_1250us);
33824b69d49SMatthias Ringwald }
33924b69d49SMatthias Ringwald 
le_audio_demo_util_sink_configure_broadcast(uint8_t num_streams,uint8_t num_channels_per_stream,uint32_t sampling_frequency_hz,btstack_lc3_frame_duration_t frame_duration,uint16_t octets_per_frame,uint32_t iso_interval_1250us,uint8_t pre_transmission_offset)34024b69d49SMatthias Ringwald void le_audio_demo_util_sink_configure_broadcast(uint8_t num_streams, uint8_t num_channels_per_stream, uint32_t sampling_frequency_hz,
34124b69d49SMatthias Ringwald                                                btstack_lc3_frame_duration_t frame_duration, uint16_t octets_per_frame,
34224b69d49SMatthias Ringwald                                                uint32_t iso_interval_1250us, uint8_t pre_transmission_offset) {
34324b69d49SMatthias Ringwald     le_audio_demo_sink_type = HCI_ISO_TYPE_BIS;
34424b69d49SMatthias Ringwald     le_audio_demo_sink_pre_transmission_offset = pre_transmission_offset;
34579d6b605SMatthias Ringwald 
34679d6b605SMatthias Ringwald     // set playback start: ISO Interval + 10 ms
34779d6b605SMatthias Ringwald     uint16_t playback_start_ms = (iso_interval_1250us * 5 / 4) + 10;
34879d6b605SMatthias Ringwald     uint16_t playback_start_samples = sampling_frequency_hz / 1000 * playback_start_ms;
34979d6b605SMatthias Ringwald     playback_start_threshold_bytes = playback_start_samples * num_streams * num_channels_per_stream * 2;
35079d6b605SMatthias Ringwald     printf("Playback: start %u ms (%u samples, %u bytes)\n", playback_start_ms, playback_start_samples, playback_start_threshold_bytes);
35179d6b605SMatthias Ringwald 
352a8325db8SMatthias Ringwald     le_audio_demo_util_sink_configure_general(num_streams, num_channels_per_stream, sampling_frequency_hz, frame_duration, octets_per_frame, iso_interval_1250us);
35324b69d49SMatthias Ringwald }
35424b69d49SMatthias Ringwald 
le_audio_demo_util_sink_count(uint8_t stream_index,uint8_t * packet,uint16_t size)355a18990d3SMatthias Ringwald void le_audio_demo_util_sink_count(uint8_t stream_index, uint8_t *packet, uint16_t size) {
356*7a6944e6SMatthias Ringwald     UNUSED(size);
357a18990d3SMatthias Ringwald     // check for missing packet
358a18990d3SMatthias Ringwald     uint16_t header = little_endian_read_16(packet, 0);
359a18990d3SMatthias Ringwald     uint8_t ts_flag = (header >> 14) & 1;
360a18990d3SMatthias Ringwald 
361a18990d3SMatthias Ringwald     uint16_t offset = 4;
362a18990d3SMatthias Ringwald     uint32_t time_stamp = 0;
363a18990d3SMatthias Ringwald     if (ts_flag){
364a18990d3SMatthias Ringwald         time_stamp = little_endian_read_32(packet, offset);
365a18990d3SMatthias Ringwald         offset += 4;
366a18990d3SMatthias Ringwald     }
367*7a6944e6SMatthias Ringwald 
368*7a6944e6SMatthias Ringwald     UNUSED(time_stamp);
369a18990d3SMatthias Ringwald     uint32_t receive_time_ms = btstack_run_loop_get_time_ms();
370a18990d3SMatthias Ringwald 
371a18990d3SMatthias Ringwald     uint16_t packet_sequence_number = little_endian_read_16(packet, offset);
372a18990d3SMatthias Ringwald     offset += 4;
373a18990d3SMatthias Ringwald 
374a18990d3SMatthias Ringwald     uint16_t last_seq_no = last_packet_sequence[stream_index];
375a18990d3SMatthias Ringwald     bool packet_missed = (last_seq_no != 0) && ((last_seq_no + 1) != packet_sequence_number);
376a18990d3SMatthias Ringwald     if (packet_missed){
377a18990d3SMatthias Ringwald         // print last packet
378a18990d3SMatthias Ringwald         printf("\n");
379a18990d3SMatthias Ringwald         printf("%04x %10"PRIu32" %u ", last_seq_no, last_packet_time_ms[stream_index], stream_index);
380a18990d3SMatthias Ringwald         printf_hexdump(&last_packet_prefix[le_audio_demo_sink_num_streams*PACKET_PREFIX_LEN], PACKET_PREFIX_LEN);
381a18990d3SMatthias Ringwald         last_seq_no++;
382a18990d3SMatthias Ringwald 
383a18990d3SMatthias Ringwald         printf(ANSI_COLOR_RED);
384a18990d3SMatthias Ringwald         while (last_seq_no < packet_sequence_number){
385a18990d3SMatthias Ringwald             printf("%04x            %u MISSING\n", last_seq_no, stream_index);
386a18990d3SMatthias Ringwald             last_seq_no++;
387a18990d3SMatthias Ringwald         }
388a18990d3SMatthias Ringwald         printf(ANSI_COLOR_RESET);
389a18990d3SMatthias Ringwald 
390a18990d3SMatthias Ringwald         // print current packet
391a18990d3SMatthias Ringwald         printf("%04x %10"PRIu32" %u ", packet_sequence_number, receive_time_ms, stream_index);
392a18990d3SMatthias Ringwald         printf_hexdump(&packet[offset], PACKET_PREFIX_LEN);
393a18990d3SMatthias Ringwald     }
394a18990d3SMatthias Ringwald 
395a18990d3SMatthias Ringwald     // cache current packet
396a18990d3SMatthias Ringwald     memcpy(&last_packet_prefix[le_audio_demo_sink_num_streams*PACKET_PREFIX_LEN], &packet[offset], PACKET_PREFIX_LEN);
397a18990d3SMatthias Ringwald }
398a18990d3SMatthias Ringwald 
audio_processing_initial(audio_processing_t * const me,btstack_fsm_event_t const * const e)399c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_initial( audio_processing_t * const me, btstack_fsm_event_t const * const e ) {
400*7a6944e6SMatthias Ringwald     UNUSED(e);
401c18c19deSDirk Helbig     audio_fsm_debug("%s\n", __FUNCTION__ );
402c18c19deSDirk Helbig     return TRAN(audio_processing_waiting);
403c18c19deSDirk Helbig }
4040f01c821SMatthias Ringwald 
audio_processing_waiting(audio_processing_t * const me,btstack_fsm_event_t const * const e)405c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_waiting( audio_processing_t * const me, btstack_fsm_event_t const * const e ) {
406c18c19deSDirk Helbig     audio_fsm_debug("%s - %s\n", __FUNCTION__, sigToString[e->sig]);
407c18c19deSDirk Helbig     btstack_fsm_state_t status;
408c18c19deSDirk Helbig     switch(e->sig) {
409c18c19deSDirk Helbig         case BTSTACK_FSM_ENTRY_SIG: {
410c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
411c18c19deSDirk Helbig             break;
412c18c19deSDirk Helbig         }
413c18c19deSDirk Helbig         case BTSTACK_FSM_EXIT_SIG: {
414c18c19deSDirk Helbig             btstack_ring_buffer_init(&playback_buffer, playback_buffer_storage, PLAYBACK_BUFFER_SIZE);
415c18c19deSDirk Helbig 
416c18c19deSDirk Helbig             btstack_sample_rate_compensation_init(&sample_rate_compensation, me->last_receive_time_ms,
417c18c19deSDirk Helbig                                                   le_audio_demo_sink_sampling_frequency_hz, FLOAT_TO_Q15(1.f));
418c18c19deSDirk Helbig             me->zero_frames = 0;
419c18c19deSDirk Helbig             me->received_samples = 0;
420c18c19deSDirk Helbig             btstack_resample_init( &resample_instance, le_audio_demo_sink_num_channels );
421c18c19deSDirk Helbig             me->have_pcm = 0;
422c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
423c18c19deSDirk Helbig             break;
424c18c19deSDirk Helbig         }
425c18c19deSDirk Helbig         case DATA_SIG: {
426c18c19deSDirk Helbig             data_event_t *data_event = (data_event_t*)e;
427c18c19deSDirk Helbig             // nothing to do here
428c18c19deSDirk Helbig             if( data_event->data == NULL ) {
429c18c19deSDirk Helbig                 status = BTSTACK_FSM_IGNORED_STATUS;
430c18c19deSDirk Helbig                 break;
431c18c19deSDirk Helbig             }
432c18c19deSDirk Helbig 
433c18c19deSDirk Helbig             // ignore empty data at start
434c18c19deSDirk Helbig             if( data_event->size == 0 ) {
435c18c19deSDirk Helbig                 status = BTSTACK_FSM_IGNORED_STATUS;
436c18c19deSDirk Helbig                 break;
437c18c19deSDirk Helbig             }
438c18c19deSDirk Helbig 
439c18c19deSDirk Helbig             // always start at first stream
440c18c19deSDirk Helbig             if( data_event->stream > 0 ) {
441c18c19deSDirk Helbig                 status = BTSTACK_FSM_IGNORED_STATUS;
442c18c19deSDirk Helbig                 break;
443c18c19deSDirk Helbig             }
444c18c19deSDirk Helbig 
445c18c19deSDirk Helbig             me->last_receive_time_ms = data_event->receive_time_ms;
446c18c19deSDirk Helbig             status = TRAN(audio_processing_streaming);
447c18c19deSDirk Helbig             break;
448c18c19deSDirk Helbig         }
449c18c19deSDirk Helbig         default: {
450c18c19deSDirk Helbig             status = BTSTACK_FSM_IGNORED_STATUS;
451c18c19deSDirk Helbig             break;
452c18c19deSDirk Helbig         }
453c18c19deSDirk Helbig     }
454c18c19deSDirk Helbig     return status;
455c18c19deSDirk Helbig }
456c18c19deSDirk Helbig 
audio_processing_resample(audio_processing_t * const me,data_event_t * e)457c18c19deSDirk Helbig static void audio_processing_resample( audio_processing_t * const me, data_event_t *e ) {
4584b3f8617SDirk Helbig     // mark current packet as handled
4594b3f8617SDirk Helbig     e->data = NULL;
460*7a6944e6SMatthias Ringwald     if( me->have_pcm != (uint32_t)((1<<(le_audio_demo_sink_num_streams*le_audio_demo_sink_num_channels_per_stream))-1) ) {
461c18c19deSDirk Helbig         return;
462c18c19deSDirk Helbig     }
463c18c19deSDirk Helbig 
464c18c19deSDirk Helbig     int16_t *data_in = pcm;
465c18c19deSDirk Helbig     int16_t *data_out = pcm_resample;
466c18c19deSDirk Helbig #ifdef HAVE_POSIX_FILE_IO
467c18c19deSDirk Helbig     // write wav samples
468c18c19deSDirk Helbig     wav_writer_write_int16(le_audio_demo_sink_num_channels * le_audio_demo_sink_num_samples_per_frame, data_in);
469c18c19deSDirk Helbig #endif
470c18c19deSDirk Helbig 
471c18c19deSDirk Helbig     // count for samplerate compensation
472c18c19deSDirk Helbig     me->received_samples += le_audio_demo_sink_num_samples_per_frame;
473c18c19deSDirk Helbig 
474c18c19deSDirk Helbig     // store samples in playback buffer
475c18c19deSDirk Helbig     samples_received += le_audio_demo_sink_num_samples_per_frame;
476c18c19deSDirk Helbig     uint32_t resampled_frames = btstack_resample_block(&resample_instance, data_in, le_audio_demo_sink_num_samples_per_frame, data_out);
477c18c19deSDirk Helbig     uint32_t bytes_to_store = resampled_frames * le_audio_demo_sink_num_channels * 2;
478c18c19deSDirk Helbig 
479c18c19deSDirk Helbig     if (btstack_ring_buffer_bytes_free(&playback_buffer) >= bytes_to_store) {
480c18c19deSDirk Helbig         btstack_ring_buffer_write(&playback_buffer, (uint8_t *)data_out, bytes_to_store);
481*7a6944e6SMatthias Ringwald         log_info("Samples in playback buffer %5" PRIu32 "", btstack_ring_buffer_bytes_available(&playback_buffer) / (le_audio_demo_sink_num_channels * 2));
482c18c19deSDirk Helbig     } else {
483c18c19deSDirk Helbig         printf("Samples dropped\n");
484c18c19deSDirk Helbig         samples_dropped += le_audio_demo_sink_num_samples_per_frame;
485c18c19deSDirk Helbig     }
486c18c19deSDirk Helbig     me->have_pcm = 0;
487c18c19deSDirk Helbig }
488c18c19deSDirk Helbig 
audio_processing_decode(audio_processing_t * const me,btstack_fsm_event_t const * const e)489c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_decode( audio_processing_t * const me, btstack_fsm_event_t const * const e ) {
490c18c19deSDirk Helbig     audio_fsm_debug("%s - %s\n", __FUNCTION__, sigToString[e->sig]);
491c18c19deSDirk Helbig     btstack_fsm_state_t status;
492c18c19deSDirk Helbig     switch(e->sig) {
493c18c19deSDirk Helbig         case BTSTACK_FSM_ENTRY_SIG: {
494c18c19deSDirk Helbig             btstack_assert( (le_audio_demo_sink_num_streams*le_audio_demo_sink_num_channels_per_stream) < (sizeof(me->have_pcm)*8));
495c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
496c18c19deSDirk Helbig             break;
497c18c19deSDirk Helbig         }
498c18c19deSDirk Helbig         case BTSTACK_FSM_EXIT_SIG: {
499c18c19deSDirk Helbig             const btstack_audio_sink_t * sink = btstack_audio_sink_get_instance();
500c18c19deSDirk Helbig             if( sink == NULL ) {
501c18c19deSDirk Helbig 
502c18c19deSDirk Helbig                 status = BTSTACK_FSM_HANDLED_STATUS;
503c18c19deSDirk Helbig                 break;
504c18c19deSDirk Helbig             }
505c18c19deSDirk Helbig             uint32_t resampling_factor = btstack_sample_rate_compensation_update( &sample_rate_compensation, me->receive_time_ms,
506c18c19deSDirk Helbig                                                                                   me->received_samples, sink->get_samplerate() );
507c18c19deSDirk Helbig             btstack_resample_set_factor(&resample_instance, resampling_factor);
508c18c19deSDirk Helbig             me->received_samples = 0;
509c18c19deSDirk Helbig 
510c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
511c18c19deSDirk Helbig             break;
512c18c19deSDirk Helbig         }
513c18c19deSDirk Helbig         case DATA_SIG: {
514c18c19deSDirk Helbig             data_event_t *data_event = (data_event_t*)e;
515c18c19deSDirk Helbig             uint8_t *data_in = data_event->data;
516c18c19deSDirk Helbig             int16_t *data_out = pcm;
517c18c19deSDirk Helbig             uint16_t offset = 0;
518c18c19deSDirk Helbig             uint8_t BFI = 0;
519c18c19deSDirk Helbig             if (data_event->size != le_audio_demo_sink_num_channels_per_stream * le_audio_demo_sink_octets_per_frame) {
520c18c19deSDirk Helbig                 // incorrect size. we assume that we received this packet on time but cannot decode it, so we use PLC
521c18c19deSDirk Helbig                 BFI = 1;
522c18c19deSDirk Helbig                 printf("predict audio\n");
523c18c19deSDirk Helbig             }
524c18c19deSDirk Helbig             uint8_t i;
525c18c19deSDirk Helbig             for (i = 0 ; i < le_audio_demo_sink_num_channels_per_stream ; i++){
526c18c19deSDirk Helbig                 uint8_t tmp_BEC_detect;
527c18c19deSDirk Helbig                 uint8_t effective_channel = (data_event->stream * le_audio_demo_sink_num_channels_per_stream) + i;
528c18c19deSDirk Helbig                 (void) lc3_decoder->decode_signed_16(decoder_contexts[effective_channel], &data_in[offset], BFI,
529c18c19deSDirk Helbig                                                      &data_out[effective_channel], le_audio_demo_sink_num_channels,
530c18c19deSDirk Helbig                                                      &tmp_BEC_detect);
531c18c19deSDirk Helbig                 offset += le_audio_demo_sink_octets_per_frame;
5324b3f8617SDirk Helbig                 audio_fsm_debug("effective_channel: %d\n", effective_channel );
5334b3f8617SDirk Helbig                 if( (me->have_pcm & (1<<effective_channel)) ) {
5344b3f8617SDirk Helbig                     audio_fsm_debug("de-syncroniced, resync\n");
5354b3f8617SDirk Helbig                     status = TRAN(audio_processing_waiting);
5364b3f8617SDirk Helbig                     break;
5374b3f8617SDirk Helbig                 }
538c18c19deSDirk Helbig                 me->have_pcm |= (1<<effective_channel);
539c18c19deSDirk Helbig             }
540c18c19deSDirk Helbig             audio_processing_resample( me, data_event );
541c18c19deSDirk Helbig             status = TRAN(audio_processing_streaming);
542c18c19deSDirk Helbig             break;
543c18c19deSDirk Helbig         }
544c18c19deSDirk Helbig         default: {
545c18c19deSDirk Helbig             status = BTSTACK_FSM_IGNORED_STATUS;
546c18c19deSDirk Helbig             break;
547c18c19deSDirk Helbig         }
548c18c19deSDirk Helbig     }
549c18c19deSDirk Helbig     return status;
550c18c19deSDirk Helbig }
551c18c19deSDirk Helbig 
audio_processing_streaming(audio_processing_t * const me,btstack_fsm_event_t const * const e)552c18c19deSDirk Helbig static btstack_fsm_state_t audio_processing_streaming( audio_processing_t * const me, btstack_fsm_event_t const * const e ) {
553c18c19deSDirk Helbig     audio_fsm_debug("%s - %s\n", __FUNCTION__, sigToString[e->sig]);
554c18c19deSDirk Helbig 
555c18c19deSDirk Helbig     btstack_fsm_state_t status;
556c18c19deSDirk Helbig     switch(e->sig) {
557c18c19deSDirk Helbig         case BTSTACK_FSM_ENTRY_SIG: {
558c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
559c18c19deSDirk Helbig             break;
560c18c19deSDirk Helbig         }
561c18c19deSDirk Helbig         case BTSTACK_FSM_EXIT_SIG: {
562c18c19deSDirk Helbig             me->last_receive_time_ms = me->receive_time_ms;
563c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
564c18c19deSDirk Helbig             break;
565c18c19deSDirk Helbig         }
566c18c19deSDirk Helbig         case TIME_SIG: {
567c18c19deSDirk Helbig             time_event_t *time = (time_event_t*)e;
568*7a6944e6SMatthias Ringwald             printf("time: %" PRId32 " - %" PRId32 " %" PRId32 "\n", time->time_ms, me->last_receive_time_ms, time->time_ms-me->last_receive_time_ms );
569c18c19deSDirk Helbig             // we were last called ages ago, so just start waiting again
570c18c19deSDirk Helbig             if( btstack_time_delta( time->time_ms, me->last_receive_time_ms ) > 100) {
571c18c19deSDirk Helbig                 status = TRAN(audio_processing_waiting);
572c18c19deSDirk Helbig                 break;
573c18c19deSDirk Helbig             }
574c18c19deSDirk Helbig             status = BTSTACK_FSM_HANDLED_STATUS;
575c18c19deSDirk Helbig             break;
576c18c19deSDirk Helbig         }
577c18c19deSDirk Helbig         case DATA_SIG: {
578c18c19deSDirk Helbig             data_event_t *data_event = (data_event_t*)e;
579c18c19deSDirk Helbig             me->receive_time_ms = data_event->receive_time_ms;
580c18c19deSDirk Helbig 
581c18c19deSDirk Helbig             // done processing this data
582c18c19deSDirk Helbig             if( data_event->data == NULL ) {
583c18c19deSDirk Helbig                 status = BTSTACK_FSM_HANDLED_STATUS;
584c18c19deSDirk Helbig                 break;
585c18c19deSDirk Helbig             }
586c18c19deSDirk Helbig 
587c18c19deSDirk Helbig             if( btstack_time_delta( data_event->receive_time_ms, me->last_receive_time_ms ) > 100) {
588c18c19deSDirk Helbig                 status = TRAN(audio_processing_waiting);
589c18c19deSDirk Helbig                 break;
590c18c19deSDirk Helbig             }
591c18c19deSDirk Helbig 
592c18c19deSDirk Helbig             if( me->zero_frames > 10 ) {
593c18c19deSDirk Helbig                 status = TRAN(audio_processing_waiting);
594c18c19deSDirk Helbig                 break;
595c18c19deSDirk Helbig             }
596c18c19deSDirk Helbig 
597c18c19deSDirk Helbig             // track consecutive audio frames without data
598c18c19deSDirk Helbig             if( data_event->size == 0 ) {
599c18c19deSDirk Helbig                 me->zero_frames++;
600c18c19deSDirk Helbig             } else {
601c18c19deSDirk Helbig                 me->zero_frames = 0;
602c18c19deSDirk Helbig             }
603c18c19deSDirk Helbig 
604c18c19deSDirk Helbig             // will decode and/or predict missing data
605c18c19deSDirk Helbig             status = TRAN(audio_processing_decode);
606c18c19deSDirk Helbig             break;
607c18c19deSDirk Helbig         }
608c18c19deSDirk Helbig         default: {
609c18c19deSDirk Helbig             status = BTSTACK_FSM_IGNORED_STATUS;
610c18c19deSDirk Helbig             break;
611c18c19deSDirk Helbig         }
612c18c19deSDirk Helbig     }
613c18c19deSDirk Helbig     return status;
614c18c19deSDirk Helbig }
615c18c19deSDirk Helbig 
audio_processing_constructor(audio_processing_t * me)616c18c19deSDirk Helbig static void audio_processing_constructor( audio_processing_t *me) {
617c18c19deSDirk Helbig     btstack_fsm_constructor(&me->super, (btstack_fsm_state_handler_t)&audio_processing_initial);
618c18c19deSDirk Helbig     btstack_fsm_init(&me->super, NULL);
619c18c19deSDirk Helbig }
620c18c19deSDirk Helbig 
audio_processing_task(audio_processing_t * me,btstack_fsm_event_t const * e)621c18c19deSDirk Helbig static void audio_processing_task( audio_processing_t *me, btstack_fsm_event_t const *e ) {
622c18c19deSDirk Helbig     btstack_fsm_dispatch_until(&me->super, e);
623c18c19deSDirk Helbig }
624c18c19deSDirk Helbig 
audio_processing_is_streaming(audio_processing_t * me)625c18c19deSDirk Helbig static bool audio_processing_is_streaming( audio_processing_t *me ) {
626c18c19deSDirk Helbig     btstack_fsm_t *fsm = &me->super;
627*7a6944e6SMatthias Ringwald     time_event_t const time_event = { { TIME_SIG }, btstack_run_loop_get_time_ms() };
628c18c19deSDirk Helbig     audio_processing_task( me, &time_event.super );
629c18c19deSDirk Helbig     return fsm->state == (btstack_fsm_state_handler_t)&audio_processing_streaming;
630c18c19deSDirk Helbig }
631c18c19deSDirk Helbig 
le_audio_demo_util_sink_receive(uint8_t stream_index,uint8_t * packet,uint16_t size)632c18c19deSDirk Helbig void le_audio_demo_util_sink_receive(uint8_t stream_index, uint8_t *packet, uint16_t size) {
633*7a6944e6SMatthias Ringwald     UNUSED(size);
634*7a6944e6SMatthias Ringwald 
6350f01c821SMatthias Ringwald     if (le_audio_demo_util_sink_state != LE_AUDIO_SINK_CONFIGURED) return;
6360f01c821SMatthias Ringwald 
63724b69d49SMatthias Ringwald     uint16_t header = little_endian_read_16(packet, 0);
63824b69d49SMatthias Ringwald     hci_con_handle_t con_handle = header & 0x0fff;
63924b69d49SMatthias Ringwald     uint8_t pb_flag = (header >> 12) & 3;
64024b69d49SMatthias Ringwald     uint8_t ts_flag = (header >> 14) & 1;
64124b69d49SMatthias Ringwald     uint16_t iso_load_len = little_endian_read_16(packet, 2);
64224b69d49SMatthias Ringwald 
64324b69d49SMatthias Ringwald     uint16_t offset = 4;
64424b69d49SMatthias Ringwald     uint32_t time_stamp = 0;
64524b69d49SMatthias Ringwald     if (ts_flag){
646ed07a8bdSMatthias Ringwald         time_stamp = little_endian_read_32(packet, offset);
64724b69d49SMatthias Ringwald         offset += 4;
64824b69d49SMatthias Ringwald     }
64924b69d49SMatthias Ringwald 
65024b69d49SMatthias Ringwald     uint32_t receive_time_ms = btstack_run_loop_get_time_ms();
65124b69d49SMatthias Ringwald 
65224b69d49SMatthias Ringwald     uint16_t packet_sequence_number = little_endian_read_16(packet, offset);
65324b69d49SMatthias Ringwald     offset += 2;
65424b69d49SMatthias Ringwald 
65524b69d49SMatthias Ringwald     uint16_t header_2 = little_endian_read_16(packet, offset);
65624b69d49SMatthias Ringwald     uint16_t iso_sdu_length = header_2 & 0x3fff;
65724b69d49SMatthias Ringwald     uint8_t packet_status_flag = (uint8_t) (header_2 >> 14);
65824b69d49SMatthias Ringwald     offset += 2;
65924b69d49SMatthias Ringwald 
660ed07a8bdSMatthias Ringwald     // avoid warning for (yet) unused fields
661ed07a8bdSMatthias Ringwald     UNUSED(con_handle);
662ed07a8bdSMatthias Ringwald     UNUSED(pb_flag);
663ed07a8bdSMatthias Ringwald     UNUSED(iso_load_len);
664ed07a8bdSMatthias Ringwald     UNUSED(packet_status_flag);
665f020be6aSMatthias Ringwald     UNUSED(time_stamp);
666ed07a8bdSMatthias Ringwald 
667c18c19deSDirk Helbig     data_event_t const data_event = {
668c18c19deSDirk Helbig             .super.sig = DATA_SIG,
669c18c19deSDirk Helbig             .sequence_number = packet_sequence_number,
670c18c19deSDirk Helbig             .stream = stream_index,
671c18c19deSDirk Helbig             .data = &packet[offset],
672c18c19deSDirk Helbig             .size = iso_sdu_length,
673c18c19deSDirk Helbig             .receive_time_ms = receive_time_ms,
674c18c19deSDirk Helbig     };
6754b3f8617SDirk Helbig 
6764b3f8617SDirk Helbig     audio_fsm_debug("new data\n stream_index: %d\n", stream_index);
677c18c19deSDirk Helbig     audio_processing_task( &audio_processing, &data_event.super );
6788dee1629SMatthias Ringwald 
67924b69d49SMatthias Ringwald     le_audio_demo_sink_lc3_frames++;
68024b69d49SMatthias Ringwald 
6818cb805e6SMatthias Ringwald     if (samples_received >= 10 * le_audio_demo_sink_sampling_frequency_hz){
682*7a6944e6SMatthias Ringwald         printf("LC3 Frames: %4" PRIu32 " - samples received %5" PRIu32 ", played %5" PRIu32 ", dropped %5" PRIu32 "\n", le_audio_demo_sink_lc3_frames, samples_received, samples_played, samples_dropped);
68324b69d49SMatthias Ringwald         samples_received = 0;
68424b69d49SMatthias Ringwald         samples_dropped  =  0;
68524b69d49SMatthias Ringwald         samples_played = 0;
68624b69d49SMatthias Ringwald     }
687c18c19deSDirk Helbig }
68824b69d49SMatthias Ringwald 
le_audio_demo_util_sink_init(const char * filename_wav)689c18c19deSDirk Helbig void le_audio_demo_util_sink_init(const char * filename_wav){
690c18c19deSDirk Helbig     le_audio_demo_sink_filename_wav = filename_wav;
691c18c19deSDirk Helbig     le_audio_demo_util_sink_state = LE_AUDIO_SINK_INIT;
692c18c19deSDirk Helbig     audio_processing_constructor( &audio_processing );
69324b69d49SMatthias Ringwald }
69424b69d49SMatthias Ringwald 
69524b69d49SMatthias Ringwald /**
69624b69d49SMatthias Ringwald  * @brief Close sink: close wav file, stop playback
69724b69d49SMatthias Ringwald  */
le_audio_demo_util_sink_close(void)69824b69d49SMatthias Ringwald void le_audio_demo_util_sink_close(void){
69924b69d49SMatthias Ringwald #ifdef HAVE_POSIX_FILE_IO
70024b69d49SMatthias Ringwald     printf("Close WAV file\n");
70124b69d49SMatthias Ringwald     wav_writer_close();
70224b69d49SMatthias Ringwald #endif
70324b69d49SMatthias Ringwald     // stop playback
70424b69d49SMatthias Ringwald     const btstack_audio_sink_t * sink = btstack_audio_sink_get_instance();
70524b69d49SMatthias Ringwald     if (sink != NULL){
70624b69d49SMatthias Ringwald         sink->stop_stream();
70724b69d49SMatthias Ringwald     }
7080f01c821SMatthias Ringwald     le_audio_demo_util_sink_state = LE_AUDIO_SINK_INIT;
70978d87b81SDirk Helbig     sink_receive_streaming = false;
71024b69d49SMatthias Ringwald     // stop timer
71124b69d49SMatthias Ringwald     btstack_run_loop_remove_timer(&next_packet_timer);
71224b69d49SMatthias Ringwald }
713