git.sesse.net Git - vlc/blob - src/audio_output/mixer.c

   1 /*****************************************************************************
   2  * mixer.c : audio output mixing operations
   3  *****************************************************************************
   4  * Copyright (C) 2002-2004 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #ifdef HAVE_CONFIG_H
  28 # include "config.h"
  29 #endif
  30 #include <assert.h>
  31
  32 #include <stddef.h>
  33 #include <vlc_common.h>
  34 #include <libvlc.h>
  35
  36 #ifdef HAVE_ALLOCA_H
  37 #   include <alloca.h>
  38 #endif
  39 #include <vlc_aout.h>
  40 #include "aout_internal.h"
  41 /*****************************************************************************
  42  * aout_MixerNew: prepare a mixer plug-in
  43  *****************************************************************************
  44  * Please note that you must hold the mixer lock.
  45  *****************************************************************************/
  46 int aout_MixerNew( aout_instance_t * p_aout )
  47 {
  48     assert( !p_aout->p_mixer );
  49     vlc_assert_locked( &p_aout->input_fifos_lock );
  50
  51     aout_mixer_t *p_mixer = vlc_object_create( p_aout, sizeof(*p_mixer) );
  52     if( !p_mixer )
  53         return VLC_EGENERIC;
  54
  55     p_mixer->fmt = p_aout->mixer_format;
  56     p_mixer->allocation = p_aout->mixer_allocation;
  57     p_mixer->multiplier = p_aout->mixer_multiplier;
  58     p_mixer->input_count = p_aout->i_nb_inputs;
  59     p_mixer->input = calloc( p_mixer->input_count, sizeof(*p_mixer->input) );
  60     for( int i = 0; i < p_aout->i_nb_inputs; i++ )
  61         p_mixer->input[i] = &p_aout->pp_inputs[i]->mixer;
  62     p_mixer->mix = NULL;
  63     p_mixer->sys = NULL;
  64
  65     vlc_object_attach( p_mixer, p_aout );
  66
  67     p_mixer->module = module_need( p_mixer, "audio mixer", NULL, false );
  68     if( !p_mixer->module )
  69     {
  70         msg_Err( p_aout, "no suitable audio mixer" );
  71         vlc_object_detach( p_mixer );
  72         free( p_mixer->input );
  73         vlc_object_release( p_mixer );
  74         return VLC_EGENERIC;
  75     }
  76
  77     /* */
  78     p_aout->p_mixer = p_mixer;
  79     return VLC_SUCCESS;
  80 }
  81
  82 /*****************************************************************************
  83  * aout_MixerDelete: delete the mixer
  84  *****************************************************************************
  85  * Please note that you must hold the mixer lock.
  86  *****************************************************************************/
  87 void aout_MixerDelete( aout_instance_t * p_aout )
  88 {
  89     if( !p_aout->p_mixer )
  90         return;
  91
  92     vlc_object_detach( p_aout->p_mixer );
  93
  94     module_unneed( p_aout->p_mixer, p_aout->p_mixer->module );
  95
  96     free( p_aout->p_mixer->input );
  97     vlc_object_release( p_aout->p_mixer );
  98
  99     /* */
 100     p_aout->p_mixer = NULL;
 101 }
 102
 103 /*****************************************************************************
 104  * MixBuffer: try to prepare one output buffer
 105  *****************************************************************************
 106  * Please note that you must hold the mixer lock.
 107  *****************************************************************************/
 108 static int MixBuffer( aout_instance_t * p_aout )
 109 {
 110     int             i, i_first_input = 0;
 111     aout_buffer_t * p_output_buffer;
 112     mtime_t start_date, end_date;
 113     date_t  exact_start_date;
 114
 115     if( !p_aout->p_mixer )
 116     {
 117         /* Free all incoming buffers. */
 118         aout_lock_input_fifos( p_aout );
 119         for ( i = 0; i < p_aout->i_nb_inputs; i++ )
 120         {
 121             aout_input_t * p_input = p_aout->pp_inputs[i];
 122             aout_buffer_t * p_buffer = p_input->mixer.fifo.p_first;
 123             if ( p_input->b_error ) continue;
 124             while ( p_buffer != NULL )
 125             {
 126                 aout_buffer_t * p_next = p_buffer->p_next;
 127                 aout_BufferFree( p_buffer );
 128                 p_buffer = p_next;
 129             }
 130         }
 131         aout_unlock_input_fifos( p_aout );
 132         return -1;
 133     }
 134
 135
 136     aout_lock_output_fifo( p_aout );
 137     aout_lock_input_fifos( p_aout );
 138
 139     /* Retrieve the date of the next buffer. */
 140     exact_start_date = p_aout->output.fifo.end_date;
 141     start_date = date_Get( &exact_start_date );
 142
 143     if ( start_date != 0 && start_date < mdate() )
 144     {
 145         /* The output is _very_ late. This can only happen if the user
 146          * pauses the stream (or if the decoder is buggy, which cannot
 147          * happen :). */
 148         msg_Warn( p_aout, "output PTS is out of range (%"PRId64"), clearing out",
 149                   mdate() - start_date );
 150         aout_FifoSet( p_aout, &p_aout->output.fifo, 0 );
 151         date_Set( &exact_start_date, 0 );
 152         start_date = 0;
 153     }
 154
 155     aout_unlock_output_fifo( p_aout );
 156
 157     /* See if we have enough data to prepare a new buffer for the audio
 158      * output. First : start date. */
 159     if ( !start_date )
 160     {
 161         /* Find the latest start date available. */
 162         for ( i = 0; i < p_aout->i_nb_inputs; i++ )
 163         {
 164             aout_input_t * p_input = p_aout->pp_inputs[i];
 165             aout_fifo_t * p_fifo = &p_input->mixer.fifo;
 166             aout_buffer_t * p_buffer;
 167
 168             if ( p_input->b_error || p_input->b_paused )
 169                 continue;
 170
 171             p_buffer = p_fifo->p_first;
 172             while ( p_buffer != NULL && p_buffer->i_pts < mdate() )
 173             {
 174                 msg_Warn( p_aout, "input PTS is out of range (%"PRId64"), "
 175                           "trashing", mdate() - p_buffer->i_pts );
 176                 p_buffer = aout_FifoPop( p_aout, p_fifo );
 177                 aout_BufferFree( p_buffer );
 178                 p_buffer = p_fifo->p_first;
 179                 p_input->mixer.begin = NULL;
 180             }
 181
 182             if ( p_buffer == NULL )
 183             {
 184                 break;
 185             }
 186
 187             if ( !start_date || start_date < p_buffer->i_pts )
 188             {
 189                 date_Set( &exact_start_date, p_buffer->i_pts );
 190                 start_date = p_buffer->i_pts;
 191             }
 192         }
 193
 194         if ( i < p_aout->i_nb_inputs )
 195         {
 196             /* Interrupted before the end... We can't run. */
 197             aout_unlock_input_fifos( p_aout );
 198             return -1;
 199         }
 200     }
 201     date_Increment( &exact_start_date, p_aout->output.i_nb_samples );
 202     end_date = date_Get( &exact_start_date );
 203
 204     /* Check that start_date and end_date are available for all input
 205      * streams. */
 206     for ( i = 0; i < p_aout->i_nb_inputs; i++ )
 207     {
 208         aout_input_t * p_input = p_aout->pp_inputs[i];
 209         aout_fifo_t * p_fifo = &p_input->mixer.fifo;
 210         aout_buffer_t * p_buffer;
 211         mtime_t prev_date;
 212         bool b_drop_buffers;
 213
 214         p_input->mixer.is_invalid = p_input->b_error || p_input->b_paused;
 215         if ( p_input->mixer.is_invalid )
 216         {
 217             if ( i_first_input == i ) i_first_input++;
 218             continue;
 219         }
 220
 221         p_buffer = p_fifo->p_first;
 222         if ( p_buffer == NULL )
 223         {
 224             break;
 225         }
 226
 227         /* Check for the continuity of start_date */
 228         while ( p_buffer != NULL && p_buffer->end_date < start_date - 1 )
 229         {
 230             /* We authorize a +-1 because rounding errors get compensated
 231              * regularly. */
 232             aout_buffer_t * p_next = p_buffer->p_next;
 233             msg_Warn( p_aout, "the mixer got a packet in the past (%"PRId64")",
 234                       start_date - p_buffer->end_date );
 235             aout_BufferFree( p_buffer );
 236             p_fifo->p_first = p_buffer = p_next;
 237             p_input->mixer.begin = NULL;
 238         }
 239         if ( p_buffer == NULL )
 240         {
 241             p_fifo->pp_last = &p_fifo->p_first;
 242             break;
 243         }
 244
 245         /* Check that we have enough samples. */
 246         for ( ; ; )
 247         {
 248             p_buffer = p_fifo->p_first;
 249             if ( p_buffer == NULL ) break;
 250             if ( p_buffer->end_date >= end_date ) break;
 251
 252             /* Check that all buffers are contiguous. */
 253             prev_date = p_fifo->p_first->end_date;
 254             p_buffer = p_buffer->p_next;
 255             b_drop_buffers = 0;
 256             for ( ; p_buffer != NULL; p_buffer = p_buffer->p_next )
 257             {
 258                 if ( prev_date != p_buffer->i_pts )
 259                 {
 260                     msg_Warn( p_aout,
 261                               "buffer hole, dropping packets (%"PRId64")",
 262                               p_buffer->i_pts - prev_date );
 263                     b_drop_buffers = 1;
 264                     break;
 265                 }
 266                 if ( p_buffer->end_date >= end_date ) break;
 267                 prev_date = p_buffer->end_date;
 268             }
 269             if ( b_drop_buffers )
 270             {
 271                 aout_buffer_t * p_deleted = p_fifo->p_first;
 272                 while ( p_deleted != NULL && p_deleted != p_buffer )
 273                 {
 274                     aout_buffer_t * p_next = p_deleted->p_next;
 275                     aout_BufferFree( p_deleted );
 276                     p_deleted = p_next;
 277                 }
 278                 p_fifo->p_first = p_deleted; /* == p_buffer */
 279             }
 280             else break;
 281         }
 282         if ( p_buffer == NULL ) break;
 283
 284         p_buffer = p_fifo->p_first;
 285         if ( !AOUT_FMT_NON_LINEAR( &p_aout->p_mixer->fmt ) )
 286         {
 287             /* Additionally check that p_first_byte_to_mix is well
 288              * located. */
 289             mtime_t i_nb_bytes = (start_date - p_buffer->i_pts)
 290                             * p_aout->p_mixer->fmt.i_bytes_per_frame
 291                             * p_aout->p_mixer->fmt.i_rate
 292                             / p_aout->p_mixer->fmt.i_frame_length
 293                             / 1000000;
 294             ptrdiff_t mixer_nb_bytes;
 295
 296             if ( p_input->mixer.begin == NULL )
 297             {
 298                 p_input->mixer.begin = p_buffer->p_buffer;
 299             }
 300             mixer_nb_bytes = p_input->mixer.begin - p_buffer->p_buffer;
 301
 302             if ( !((i_nb_bytes + p_aout->p_mixer->fmt.i_bytes_per_frame
 303                      > mixer_nb_bytes) &&
 304                    (i_nb_bytes < p_aout->p_mixer->fmt.i_bytes_per_frame
 305                      + mixer_nb_bytes)) )
 306             {
 307                 msg_Warn( p_aout, "mixer start isn't output start (%"PRId64")",
 308                           i_nb_bytes - mixer_nb_bytes );
 309
 310                 /* Round to the nearest multiple */
 311                 i_nb_bytes /= p_aout->p_mixer->fmt.i_bytes_per_frame;
 312                 i_nb_bytes *= p_aout->p_mixer->fmt.i_bytes_per_frame;
 313                 if( i_nb_bytes < 0 )
 314                 {
 315                     /* Is it really the best way to do it ? */
 316                     aout_lock_output_fifo( p_aout );
 317                     aout_FifoSet( p_aout, &p_aout->output.fifo, 0 );
 318                     date_Set( &exact_start_date, 0 );
 319                     aout_unlock_output_fifo( p_aout );
 320                     break;
 321                 }
 322
 323                 p_input->mixer.begin = p_buffer->p_buffer + i_nb_bytes;
 324             }
 325         }
 326     }
 327
 328     if ( i < p_aout->i_nb_inputs || i_first_input == p_aout->i_nb_inputs )
 329     {
 330         /* Interrupted before the end... We can't run. */
 331         aout_unlock_input_fifos( p_aout );
 332         return -1;
 333     }
 334
 335     /* Run the mixer. */
 336     p_output_buffer = aout_BufferAlloc( &p_aout->p_mixer->allocation,
 337                           ((uint64_t)p_aout->output.i_nb_samples * 1000000)
 338                             / p_aout->output.output.i_rate,
 339                           /* This is a bit kludgy, but is actually only used
 340                            * for the S/PDIF dummy mixer : */
 341                           p_aout->pp_inputs[i_first_input]->mixer.fifo.p_first);
 342     if ( p_output_buffer == NULL )
 343     {
 344         aout_unlock_input_fifos( p_aout );
 345         return -1;
 346     }
 347     /* This is again a bit kludgy - for the S/PDIF mixer. */
 348     if ( p_aout->p_mixer->allocation.b_alloc )
 349     {
 350         p_output_buffer->i_nb_samples = p_aout->output.i_nb_samples;
 351         p_output_buffer->i_nb_bytes = p_aout->output.i_nb_samples
 352                               * p_aout->p_mixer->fmt.i_bytes_per_frame
 353                               / p_aout->p_mixer->fmt.i_frame_length;
 354     }
 355     p_output_buffer->i_pts = start_date;
 356     p_output_buffer->end_date = end_date;
 357
 358     p_aout->p_mixer->mix( p_aout->p_mixer, p_output_buffer );
 359
 360     aout_unlock_input_fifos( p_aout );
 361
 362     aout_OutputPlay( p_aout, p_output_buffer );
 363
 364     return 0;
 365 }
 366
 367 /*****************************************************************************
 368  * aout_MixerRun: entry point for the mixer & post-filters processing
 369  *****************************************************************************
 370  * Please note that you must hold the mixer lock.
 371  *****************************************************************************/
 372 void aout_MixerRun( aout_instance_t * p_aout )
 373 {
 374     while( MixBuffer( p_aout ) != -1 );
 375 }
 376
 377 /*****************************************************************************
 378  * aout_MixerMultiplierSet: set p_aout->mixer.f_multiplier
 379  *****************************************************************************
 380  * Please note that we assume that you own the mixer lock when entering this
 381  * function. This function returns -1 on error.
 382  *****************************************************************************/
 383 int aout_MixerMultiplierSet( aout_instance_t * p_aout, float f_multiplier )
 384 {
 385     float f_old = p_aout->mixer_multiplier;
 386     bool b_new_mixer = false;
 387
 388     if ( p_aout->p_mixer )
 389     {
 390         aout_MixerDelete( p_aout );
 391         b_new_mixer = true;
 392     }
 393
 394     p_aout->mixer_multiplier = f_multiplier;
 395
 396     if ( b_new_mixer && aout_MixerNew( p_aout ) )
 397     {
 398         p_aout->mixer_multiplier = f_old;
 399         aout_MixerNew( p_aout );
 400         return -1;
 401     }
 402
 403     return 0;
 404 }
 405
 406 /*****************************************************************************
 407  * aout_MixerMultiplierGet: get p_aout->mixer.f_multiplier
 408  *****************************************************************************
 409  * Please note that we assume that you own the mixer lock when entering this
 410  * function. This function returns -1 on error.
 411  *****************************************************************************/
 412 int aout_MixerMultiplierGet( aout_instance_t * p_aout, float * pf_multiplier )
 413 {
 414     *pf_multiplier = p_aout->mixer_multiplier;
 415     return 0;
 416 }
 417