]> git.sesse.net Git - vlc/blob - modules/audio_output/packet.c
f245a821345fb38b4b97c474bb3e799909bcb5e2
[vlc] / modules / audio_output / packet.c
1 /*****************************************************************************
2  * packet.c : helper for legacy audio output plugins
3  *****************************************************************************
4  * Copyright (C) 2002-2012 VLC authors and VideoLAN
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU Lesser General Public License as published by
8  * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
19  *****************************************************************************/
20
21 #ifdef HAVE_CONFIG_H
22 # include <config.h>
23 #endif
24 #include <limits.h>
25 #include <assert.h>
26 #include <vlc_common.h>
27 #include <vlc_aout.h>
28
29 /**
30  * Initializes the members of a FIFO.
31  */
32 static void aout_FifoInit( aout_fifo_t *p_fifo, uint32_t i_rate )
33 {
34     assert( i_rate != 0);
35     p_fifo->p_first = NULL;
36     p_fifo->pp_last = &p_fifo->p_first;
37     date_Init( &p_fifo->end_date, i_rate, 1 );
38     date_Set( &p_fifo->end_date, VLC_TS_INVALID );
39 }
40
41 /**
42  * Pushes a packet into the FIFO.
43  */
44 static void aout_FifoPush( aout_fifo_t * p_fifo, block_t * p_buffer )
45 {
46     *p_fifo->pp_last = p_buffer;
47     p_fifo->pp_last = &p_buffer->p_next;
48     *p_fifo->pp_last = NULL;
49     /* Enforce the continuity of the stream. */
50     if( date_Get( &p_fifo->end_date ) != VLC_TS_INVALID )
51     {
52         p_buffer->i_pts = date_Get( &p_fifo->end_date );
53         p_buffer->i_length = date_Increment( &p_fifo->end_date,
54                                              p_buffer->i_nb_samples );
55         p_buffer->i_length -= p_buffer->i_pts;
56     }
57     else
58     {
59         date_Set( &p_fifo->end_date, p_buffer->i_pts + p_buffer->i_length );
60     }
61 }
62
63 /**
64  * Trashes all buffers.
65  */
66 static void aout_FifoReset( aout_fifo_t * p_fifo )
67 {
68     block_t * p_buffer;
69
70     date_Set( &p_fifo->end_date, VLC_TS_INVALID );
71     p_buffer = p_fifo->p_first;
72     while ( p_buffer != NULL )
73     {
74         block_t * p_next = p_buffer->p_next;
75         block_Release( p_buffer );
76         p_buffer = p_next;
77     }
78     p_fifo->p_first = NULL;
79     p_fifo->pp_last = &p_fifo->p_first;
80 }
81
82 /**
83  * Move forwards or backwards all dates in the FIFO.
84  */
85 static void aout_FifoMoveDates( aout_fifo_t *fifo, mtime_t difference )
86 {
87     if( date_Get( &fifo->end_date ) == VLC_TS_INVALID )
88     {
89         assert( fifo->p_first == NULL );
90         return;
91     }
92
93     date_Move( &fifo->end_date, difference );
94     for( block_t *block = fifo->p_first; block != NULL; block = block->p_next )
95         block->i_pts += difference;
96 }
97
98 /**
99  * Gets the next buffer out of the FIFO
100  */
101 static block_t *aout_FifoPop( aout_fifo_t * p_fifo )
102 {
103     block_t *p_buffer = p_fifo->p_first;
104     if( p_buffer != NULL )
105     {
106         p_fifo->p_first = p_buffer->p_next;
107         if( p_fifo->p_first == NULL )
108             p_fifo->pp_last = &p_fifo->p_first;
109     }
110     return p_buffer;
111 }
112
113 /**
114  * Destroys a FIFO and its buffers
115  */
116 static void aout_FifoDestroy( aout_fifo_t * p_fifo )
117 {
118     block_t * p_buffer;
119
120     p_buffer = p_fifo->p_first;
121     while ( p_buffer != NULL )
122     {
123         block_t * p_next = p_buffer->p_next;
124         block_Release( p_buffer );
125         p_buffer = p_next;
126     }
127
128     p_fifo->p_first = NULL;
129     p_fifo->pp_last = &p_fifo->p_first;
130 }
131
132 static inline aout_packet_t *aout_packet (audio_output_t *aout)
133 {
134     return (aout_packet_t *)(aout->sys);
135 }
136
137 void aout_PacketInit (audio_output_t *aout, aout_packet_t *p, unsigned samples,
138                       const audio_sample_format_t *fmt)
139 {
140     assert (p == aout_packet (aout));
141
142     vlc_mutex_init (&p->lock);
143     p->format = *fmt;
144     aout_FifoInit (&p->partial, p->format.i_rate);
145     aout_FifoInit (&p->fifo, p->format.i_rate);
146     p->pause_date = VLC_TS_INVALID;
147     p->samples = samples;
148     p->starving = true;
149 }
150
151 void aout_PacketDestroy (audio_output_t *aout)
152 {
153     aout_packet_t *p = aout_packet (aout);
154
155     aout_FifoDestroy (&p->partial);
156     aout_FifoDestroy (&p->fifo);
157     vlc_mutex_destroy (&p->lock);
158 }
159
160 int aout_PacketTimeGet (audio_output_t *aout,  mtime_t *restrict pts)
161 {
162     aout_packet_t *p = aout_packet (aout);
163     mtime_t time_report;
164
165     vlc_mutex_lock (&p->lock);
166     time_report = date_Get (&p->fifo.end_date);
167     vlc_mutex_unlock (&p->lock);
168
169     if (time_report == VLC_TS_INVALID)
170         return -1;
171     *pts = time_report;
172     return 0;
173 }
174
175 static block_t *aout_OutputSlice (audio_output_t *);
176
177 void aout_PacketPlay (audio_output_t *aout, block_t *block)
178 {
179     aout_packet_t *p = aout_packet (aout);
180
181     vlc_mutex_lock (&p->lock);
182     aout_FifoPush (&p->partial, block);
183     while ((block = aout_OutputSlice (aout)) != NULL)
184         aout_FifoPush (&p->fifo, block);
185     vlc_mutex_unlock (&p->lock);
186 }
187
188 void aout_PacketFlush (audio_output_t *aout, bool drain)
189 {
190     aout_packet_t *p = aout_packet (aout);
191
192     vlc_mutex_lock (&p->lock);
193     if (drain)
194     {
195         mtime_t pts = date_Get (&p->fifo.end_date);
196         vlc_mutex_unlock (&p->lock);
197         if (pts != VLC_TS_INVALID)
198             mwait (pts);
199     }
200     else
201     {
202         aout_FifoReset (&p->partial);
203         aout_FifoReset (&p->fifo);
204         vlc_mutex_unlock (&p->lock);
205     }
206 }
207
208
209 /**
210  * Rearranges audio blocks in correct number of samples.
211  * @note (FIXME) This is left here for historical reasons. It belongs in the
212  * output code. Besides, this operation should be avoided if possible.
213  */
214 static block_t *aout_OutputSlice (audio_output_t *p_aout)
215 {
216     aout_packet_t *p = aout_packet (p_aout);
217     aout_fifo_t *p_fifo = &p->partial;
218     const unsigned samples = p->samples;
219     assert( samples > 0 );
220
221     /* Retrieve the date of the next buffer. */
222     date_t exact_start_date = p->fifo.end_date;
223     mtime_t start_date = date_Get( &exact_start_date );
224
225     /* Check if there is enough data to slice a new buffer. */
226     block_t *p_buffer = p_fifo->p_first;
227     if( p_buffer == NULL )
228         return NULL;
229
230     /* Find the earliest start date available. */
231     if ( start_date == VLC_TS_INVALID )
232     {
233         start_date = p_buffer->i_pts;
234         date_Set( &exact_start_date, start_date );
235     }
236     /* Compute the end date for the new buffer. */
237     mtime_t end_date = date_Increment( &exact_start_date, samples );
238
239     /* Check that we have enough samples (TODO merge with next loop). */
240     for( unsigned available = 0; available < samples; )
241     {
242         p_buffer = p_buffer->p_next;
243         if( p_buffer == NULL )
244             return NULL;
245
246         available += p_buffer->i_nb_samples;
247     }
248
249     if( AOUT_FMT_LINEAR( &p->format ) )
250     {
251         const unsigned framesize = p->format.i_bytes_per_frame;
252         /* Build packet with adequate number of samples */
253         unsigned needed = samples * framesize;
254
255         p_buffer = block_Alloc( needed );
256         if( unlikely(p_buffer == NULL) )
257             /* XXX: should free input buffers */
258             return NULL;
259         p_buffer->i_nb_samples = samples;
260
261         for( uint8_t *p_out = p_buffer->p_buffer; needed > 0; )
262         {
263             block_t *p_inbuf = p_fifo->p_first;
264             if( unlikely(p_inbuf == NULL) )
265             {
266                 msg_Err( p_aout, "packetization error" );
267                 memset( p_out, 0, needed );
268                 break;
269             }
270
271             const uint8_t *p_in = p_inbuf->p_buffer;
272             size_t avail = p_inbuf->i_nb_samples * framesize;
273             if( avail > needed )
274             {
275                 memcpy( p_out, p_in, needed );
276                 p_fifo->p_first->p_buffer += needed;
277                 p_fifo->p_first->i_buffer -= needed;
278                 needed /= framesize;
279                 p_fifo->p_first->i_nb_samples -= needed;
280
281                 mtime_t t = needed * CLOCK_FREQ / p->format.i_rate;
282                 p_fifo->p_first->i_pts += t;
283                 p_fifo->p_first->i_length -= t;
284                 break;
285             }
286
287             memcpy( p_out, p_in, avail );
288             needed -= avail;
289             p_out += avail;
290             /* Next buffer */
291             block_Release( aout_FifoPop( p_fifo ) );
292         }
293     }
294     else
295         p_buffer = aout_FifoPop( p_fifo );
296
297     p_buffer->i_pts = start_date;
298     p_buffer->i_length = end_date - start_date;
299
300     return p_buffer;
301 }
302
303 /**
304  * Dequeues the next audio packet (a.k.a. audio fragment).
305  * The audio output plugin must first call aout_PacketPlay() to queue the
306  * decoded audio samples. Typically, audio_output_t.play is set to, or calls
307  * aout_PacketPlay().
308  * @note This function is considered legacy. Please do not use this function in
309  * new audio output plugins.
310  * @param p_aout audio output instance
311  * @param start_date expected PTS of the audio packet
312  */
313 block_t *aout_PacketNext (audio_output_t *p_aout, mtime_t start_date)
314 {
315     aout_packet_t *p = aout_packet (p_aout);
316     aout_fifo_t *p_fifo = &p->fifo;
317     block_t *p_buffer;
318     const bool b_can_sleek = !AOUT_FMT_LINEAR(&p->format);
319     const mtime_t now = mdate ();
320     const mtime_t threshold =
321         (b_can_sleek ? start_date : now) - AOUT_MAX_PTS_DELAY;
322
323     vlc_mutex_lock( &p->lock );
324     if( p->pause_date != VLC_TS_INVALID )
325         goto out; /* paused: do not dequeue buffers */
326
327     for (;;)
328     {
329         p_buffer = p_fifo->p_first;
330         if (p_buffer == NULL)
331             goto out; /* nothing to play */
332
333         if (p_buffer->i_pts >= threshold)
334             break;
335
336         /* Drop the audio sample if the audio output is really late.
337          * In the case of b_can_sleek, we don't use a resampler so we need to
338          * be a lot more severe. */
339         msg_Dbg (p_aout, "audio output is too slow (%"PRId64" us): "
340                  " trashing %"PRId64" us", threshold - p_buffer->i_pts,
341                  p_buffer->i_length);
342         block_Release (aout_FifoPop (p_fifo));
343     }
344
345     mtime_t delta = start_date - p_buffer->i_pts;
346     /* This assumes that all buffers have the same duration. This is true
347      * since aout_PacketPlay() (aout_OutputSlice()) is used. */
348     if (0 >= delta + p_buffer->i_length)
349     {
350         if (!p->starving)
351         {
352             msg_Dbg (p_aout, "audio output is starving (%"PRId64"), "
353                      "playing silence", delta);
354             p->starving = true;
355         }
356         goto out; /* nothing to play _yet_ */
357     }
358
359     p->starving = false;
360     p_buffer = aout_FifoPop( p_fifo );
361
362     if (!b_can_sleek
363      && (delta < -AOUT_MAX_PTS_ADVANCE || AOUT_MAX_PTS_DELAY < delta))
364     {
365         msg_Warn (p_aout, "audio output out of sync, "
366                           "adjusting dates (%"PRId64" us)", delta);
367         aout_FifoMoveDates (&p->partial, delta);
368         aout_FifoMoveDates (p_fifo, delta);
369     }
370     vlc_mutex_unlock( &p->lock );
371     return p_buffer;
372 out:
373     vlc_mutex_unlock( &p->lock );
374     return NULL;
375 }