smem: try not to re-create the variables too many times.
[vlc.git] / modules / stream_out / smem.c
index 5d9b79e..236639d 100644 (file)
@@ -134,17 +134,16 @@ static int SendAudio( sout_stream_t *p_stream, sout_stream_id_t *id,
 struct sout_stream_id_t
 {
     es_format_t* format;
+    void *p_data;
 };
 
 struct sout_stream_sys_t
 {
     vlc_mutex_t *p_lock;
-    void ( *pf_video_prerender_callback ) ( void* p_video_data , uint8_t** pp_pixel_buffer , int size );
-    void ( *pf_audio_prerender_callback ) ( void* p_audio_data , uint8_t** pp_pcm_buffer , unsigned int );
+    void ( *pf_video_prerender_callback ) ( void* p_video_data, uint8_t** pp_pixel_buffer , int size );
+    void ( *pf_audio_prerender_callback ) ( void* p_audio_data, uint8_t** pp_pcm_buffer , unsigned int size );
     void ( *pf_video_postrender_callback ) ( void* p_video_data, uint8_t* p_pixel_buffer, int width, int height, int pixel_pitch, int size, int pts );
-    void ( *pf_audio_postrender_callback ) ( void*, unsigned int, unsigned int, unsigned int, unsigned int );
-    void *p_audio_data;
-    void *p_video_data;
+    void ( *pf_audio_postrender_callback ) ( void* p_audio_data, uint8_t* p_pcm_buffer, unsigned int channels, unsigned int rate, unsigned int nb_samples, unsigned int bits_per_sample, unsigned int size, int pts );
     bool time_sync;
 };
 
@@ -177,16 +176,12 @@ static int Open( vlc_object_t *p_this )
     free( psz_tmp );
 
     psz_tmp = var_CreateGetString( p_stream, SOUT_PREFIX_AUDIO "postrender-callback" );
-    p_sys->pf_audio_postrender_callback = (void (*) (void*, unsigned int, unsigned int, unsigned int, unsigned int))(intptr_t)atoll( psz_tmp );
+    p_sys->pf_audio_postrender_callback = (void (*) (void*, uint8_t*, unsigned int, unsigned int, unsigned int, unsigned int, unsigned int, int))(intptr_t)atoll( psz_tmp );
     free( psz_tmp );
 
-    psz_tmp = var_CreateGetString( p_stream, SOUT_PREFIX_VIDEO "data" );
-    p_sys->p_video_data = (void *)( intptr_t )atoll( psz_tmp );
-    free( psz_tmp );
-
-    psz_tmp = var_CreateGetString( p_stream, SOUT_PREFIX_AUDIO "data" );
-    p_sys->p_audio_data = (void *)( intptr_t )atoll( psz_tmp );
-    free( psz_tmp );
+    /* Create the remaining variables for a later use */
+    var_Create( p_stream, SOUT_PREFIX_VIDEO "data", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
+    var_Create( p_stream, SOUT_PREFIX_AUDIO "data", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
 
     /* Setting stream out module callbacks */
     p_stream->pf_add    = Add;
@@ -224,6 +219,7 @@ static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
 
 static sout_stream_id_t *AddVideo( sout_stream_t *p_stream, es_format_t *p_fmt )
 {
+    char* psz_tmp;
     sout_stream_id_t    *id;
     int i_bits_per_pixel;
 
@@ -258,6 +254,10 @@ static sout_stream_id_t *AddVideo( sout_stream_t *p_stream, es_format_t *p_fmt )
     if( !id )
         return NULL;
 
+    psz_tmp = var_GetString( p_stream, SOUT_PREFIX_VIDEO "data" );
+    id->p_data = (void *)( intptr_t )atoll( psz_tmp );
+    free( psz_tmp );
+
     id->format = p_fmt;
     id->format->video.i_bits_per_pixel = i_bits_per_pixel;
     return id;
@@ -265,9 +265,53 @@ static sout_stream_id_t *AddVideo( sout_stream_t *p_stream, es_format_t *p_fmt )
 
 static sout_stream_id_t *AddAudio( sout_stream_t *p_stream, es_format_t *p_fmt )
 {
-    VLC_UNUSED( p_stream );
-    VLC_UNUSED( p_fmt );
-    return NULL;
+    char* psz_tmp;
+    sout_stream_id_t* id;
+    int i_bits_per_sample;
+
+    switch( p_fmt->i_codec )
+    {
+    case VLC_CODEC_U8:
+    case VLC_CODEC_S8:
+        i_bits_per_sample = 8;
+        break;
+    case VLC_CODEC_U16L:
+    case VLC_CODEC_S16L:
+    case VLC_CODEC_U16B:
+    case VLC_CODEC_S16B:
+        i_bits_per_sample =  16;
+        break;
+    case VLC_CODEC_U24L:
+    case VLC_CODEC_S24L:
+    case VLC_CODEC_U24B:
+    case VLC_CODEC_S24B:
+        i_bits_per_sample = 24;
+        break;
+    case VLC_CODEC_S32L:
+    case VLC_CODEC_S32B:
+    case VLC_CODEC_FL32:
+    case VLC_CODEC_FI32:
+        i_bits_per_sample = 32;
+        break;
+    case VLC_CODEC_FL64:
+        i_bits_per_sample = 64;
+        break;
+    default:
+        msg_Err( p_stream, "Smem does only support raw audio format" );
+        return NULL;
+    }
+
+    id = calloc( 1, sizeof( sout_stream_id_t ) );
+    if( !id )
+        return NULL;
+
+    psz_tmp = var_GetString( p_stream, SOUT_PREFIX_AUDIO "data" );
+    id->p_data = (void *)( intptr_t )atoll( psz_tmp );
+    free( psz_tmp );
+
+    id->format = p_fmt;
+    id->format->audio.i_bitspersample = i_bits_per_sample;
+    return id;
 }
 
 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
@@ -300,22 +344,36 @@ static int SendVideo( sout_stream_t *p_stream, sout_stream_id_t *id,
     i_line_size = i_pixel_pitch * id->format->video.i_width;
     i_size = i_line * i_line_size;
     /* Calling the prerender callback to get user buffer */
-    p_sys->pf_video_prerender_callback( p_sys->p_video_data, &p_pixels , i_size );
+    p_sys->pf_video_prerender_callback( id->p_data, &p_pixels , i_size );
     /* Copying data into user buffer */
     for ( int line = 0; line < i_line; line++, p_pixels += i_line_size )
         vlc_memcpy( p_pixels, p_buffer->p_buffer + i_line_size * line , i_line_size );
     /* Calling the postrender callback to tell the user his buffer is ready */
-    p_sys->pf_video_postrender_callback( p_sys->p_video_data, p_pixels,
+    p_sys->pf_video_postrender_callback( id->p_data, p_pixels,
                                          id->format->video.i_width, id->format->video.i_height,
                                          id->format->video.i_bits_per_pixel, i_size, p_buffer->i_pts );
+    block_ChainRelease( p_buffer );
     return VLC_SUCCESS;
 }
 
 static int SendAudio( sout_stream_t *p_stream, sout_stream_id_t *id,
                       block_t *p_buffer )
 {
-    VLC_UNUSED( p_stream );
-    VLC_UNUSED( id );
-    VLC_UNUSED( p_buffer );
+    sout_stream_sys_t *p_sys = p_stream->p_sys;
+    int i_size;
+    uint8_t* p_pcm_buffer;
+    int i_samples = 0;
+
+    i_size = p_buffer->i_buffer;
+    i_samples = i_size / ( ( id->format->audio.i_bitspersample / 8 ) * id->format->audio.i_channels );
+    /* Calling the prerender callback to get user buffer */
+    p_sys->pf_audio_prerender_callback( id->p_data, &p_pcm_buffer, i_size );
+    /* Copying data into user buffer */
+    vlc_memcpy( p_pcm_buffer, p_buffer->p_buffer, i_size );
+    /* Calling the postrender callback to tell the user his buffer is ready */
+    p_sys->pf_audio_postrender_callback( id->p_data, p_pcm_buffer,
+                                         id->format->audio.i_channels, id->format->audio.i_rate, p_buffer->i_nb_samples,
+                                         id->format->audio.i_bitspersample, i_size, p_buffer->i_pts );
+    block_ChainRelease( p_buffer );
     return VLC_SUCCESS;
 }