block.c
上传用户:riyaled888
上传日期:2009-03-27
资源大小:7338k
文件大小:7k
源码类别:

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * block.c: Data blocks management functions
  3.  *****************************************************************************
  4.  * Copyright (C) 2003-2004 VideoLAN
  5.  * $Id: block.c 8699 2004-09-13 13:28:10Z gbazin $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@videolan.org>
  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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #include <stdlib.h>
  27. #include <vlc/vlc.h>
  28. #include "vlc_block.h"
  29. /*****************************************************************************
  30.  * Block functions.
  31.  *****************************************************************************/
  32. /* private */
  33. struct block_sys_t
  34. {
  35.     uint8_t     *p_allocated_buffer;
  36.     int         i_allocated_buffer;
  37. };
  38. #define BLOCK_PADDING_SIZE 32
  39. static void BlockRelease( block_t * );
  40. block_t *__block_New( vlc_object_t *p_obj, int i_size )
  41. {
  42.     /* We do only one malloc
  43.      * TODO bench if doing 2 malloc but keeping a pool of buffer is better
  44.      * 16 -> align on 16
  45.      * 2 * BLOCK_PADDING_SIZE -> pre + post padding
  46.      */
  47.     block_sys_t *p_sys;
  48.     const int i_alloc = i_size + 2 * BLOCK_PADDING_SIZE + 16;
  49.     block_t *p_block =
  50.         malloc( sizeof( block_t ) + sizeof( block_sys_t ) + i_alloc );
  51.     if( p_block == NULL ) return NULL;
  52.     /* Fill opaque data */
  53.     p_sys = (block_sys_t*)( (uint8_t*)p_block + sizeof( block_t ) );
  54.     p_sys->i_allocated_buffer = i_alloc;
  55.     p_sys->p_allocated_buffer = (uint8_t*)p_block + sizeof( block_t ) +
  56.         sizeof( block_sys_t );
  57.     /* Fill all fields */
  58.     p_block->p_next         = NULL;
  59.     p_block->i_flags        = 0;
  60.     p_block->i_pts          = 0;
  61.     p_block->i_dts          = 0;
  62.     p_block->i_length       = 0;
  63.     p_block->i_rate         = 0;
  64.     p_block->i_buffer       = i_size;
  65.     p_block->p_buffer       =
  66.         &p_sys->p_allocated_buffer[BLOCK_PADDING_SIZE +
  67.             16 - ((uintptr_t)p_sys->p_allocated_buffer % 16 )];
  68.     p_block->pf_release     = BlockRelease;
  69.     /* Is ok, as no comunication between p_vlc */
  70.     p_block->p_manager      = VLC_OBJECT( p_obj->p_vlc );
  71.     p_block->p_sys          = p_sys;
  72.     return p_block;
  73. }
  74. block_t *block_Realloc( block_t *p_block, int i_prebody, int i_body )
  75. {
  76.     int i_buffer_size;
  77.     if( p_block->pf_release != BlockRelease )
  78.     {
  79.         /* Special case when pf_release if overloaded
  80.          * TODO if used one day, them implement it in a smarter way */
  81.         block_t *p_dup = block_Duplicate( p_block );
  82.         block_Release( p_block );
  83.         p_block = p_dup;
  84.     }
  85.     i_buffer_size = i_prebody + i_body;
  86.     if( i_body < 0 || i_buffer_size <= 0 ) return NULL;
  87.     if( i_prebody < ( p_block->p_buffer - p_block->p_sys->p_allocated_buffer +
  88.                       p_block->p_sys->i_allocated_buffer ) ||
  89.         p_block->p_buffer - i_prebody > p_block->p_sys->p_allocated_buffer )
  90.     {
  91.         p_block->p_buffer -= i_prebody;
  92.         p_block->i_buffer += i_prebody;
  93.         i_prebody = 0;
  94.     }
  95.     if( p_block->p_buffer + i_body < p_block->p_sys->p_allocated_buffer +
  96.         p_block->p_sys->i_allocated_buffer )
  97.     {
  98.         p_block->i_buffer = i_buffer_size;
  99.         i_body = 0;
  100.     }
  101.     if( i_body > 0 || i_prebody > 0 )
  102.     {
  103.         block_t *p_rea = block_New( p_block->p_manager, i_buffer_size );
  104.         p_rea->i_dts     = p_block->i_dts;
  105.         p_rea->i_pts     = p_block->i_pts;
  106.         p_rea->i_flags   = p_block->i_flags;
  107.         p_rea->i_length  = p_block->i_length;
  108.         p_rea->i_rate    = p_block->i_rate;
  109.         p_rea->i_samples = p_block->i_samples;
  110.         memcpy( p_rea->p_buffer + i_prebody, p_block->p_buffer,
  111.                 __MIN( p_block->i_buffer, p_rea->i_buffer - i_prebody ) );
  112.         block_Release( p_block );
  113.         return p_rea;
  114.     }
  115.     return p_block;
  116. }
  117. static void BlockRelease( block_t *p_block )
  118. {
  119.     free( p_block );
  120. }
  121. /*****************************************************************************
  122.  * block_fifo_t management
  123.  *****************************************************************************/
  124. block_fifo_t *__block_FifoNew( vlc_object_t *p_obj )
  125. {
  126.     block_fifo_t *p_fifo;
  127.     p_fifo = malloc( sizeof( vlc_object_t ) );
  128.     vlc_mutex_init( p_obj, &p_fifo->lock );
  129.     vlc_cond_init( p_obj, &p_fifo->wait );
  130.     p_fifo->i_depth = p_fifo->i_size = 0;
  131.     p_fifo->p_first = NULL;
  132.     p_fifo->pp_last = &p_fifo->p_first;
  133.     return p_fifo;
  134. }
  135. void block_FifoRelease( block_fifo_t *p_fifo )
  136. {
  137.     block_FifoEmpty( p_fifo );
  138.     vlc_cond_destroy( &p_fifo->wait );
  139.     vlc_mutex_destroy( &p_fifo->lock );
  140.     free( p_fifo );
  141. }
  142. void block_FifoEmpty( block_fifo_t *p_fifo )
  143. {
  144.     block_t *b;
  145.     vlc_mutex_lock( &p_fifo->lock );
  146.     for( b = p_fifo->p_first; b != NULL; )
  147.     {
  148.         block_t *p_next;
  149.         p_next = b->p_next;
  150.         block_Release( b );
  151.         b = p_next;
  152.     }
  153.     p_fifo->i_depth = p_fifo->i_size = 0;
  154.     p_fifo->p_first = NULL;
  155.     p_fifo->pp_last = &p_fifo->p_first;
  156.     vlc_mutex_unlock( &p_fifo->lock );
  157. }
  158. int block_FifoPut( block_fifo_t *p_fifo, block_t *p_block )
  159. {
  160.     int i_size = 0;
  161.     vlc_mutex_lock( &p_fifo->lock );
  162.     do
  163.     {
  164.         i_size += p_block->i_buffer;
  165.         *p_fifo->pp_last = p_block;
  166.         p_fifo->pp_last = &p_block->p_next;
  167.         p_fifo->i_depth++;
  168.         p_fifo->i_size += p_block->i_buffer;
  169.         p_block = p_block->p_next;
  170.     } while( p_block );
  171.     /* warn there is data in this fifo */
  172.     vlc_cond_signal( &p_fifo->wait );
  173.     vlc_mutex_unlock( &p_fifo->lock );
  174.     return i_size;
  175. }
  176. block_t *block_FifoGet( block_fifo_t *p_fifo )
  177. {
  178.     block_t *b;
  179.     vlc_mutex_lock( &p_fifo->lock );
  180.     /* We do a while here because there is a race condition in the
  181.      * win32 implementation of vlc_cond_wait() (We can't be sure the fifo
  182.      * hasn't been emptied again since we were signaled). */
  183.     while( p_fifo->p_first == NULL )
  184.     {
  185.         vlc_cond_wait( &p_fifo->wait, &p_fifo->lock );
  186.     }
  187.     b = p_fifo->p_first;
  188.     p_fifo->p_first = b->p_next;
  189.     p_fifo->i_depth--;
  190.     p_fifo->i_size -= b->i_buffer;
  191.     if( p_fifo->p_first == NULL )
  192.     {
  193.         p_fifo->pp_last = &p_fifo->p_first;
  194.     }
  195.     vlc_mutex_unlock( &p_fifo->lock );
  196.     b->p_next = NULL;
  197.     return b;
  198. }
  199. block_t *block_FifoShow( block_fifo_t *p_fifo )
  200. {
  201.     block_t *b;
  202.     vlc_mutex_lock( &p_fifo->lock );
  203.     if( p_fifo->p_first == NULL )
  204.     {
  205.         vlc_cond_wait( &p_fifo->wait, &p_fifo->lock );
  206.     }
  207.     b = p_fifo->p_first;
  208.     vlc_mutex_unlock( &p_fifo->lock );
  209.     return( b );
  210. }