]> git.sesse.net Git - vlc/blobdiff - modules/codec/tarkin.c
Fix memleak in lua module.
[vlc] / modules / codec / tarkin.c
index f4224ba1b63812ae678d5fad6efc1fc7cc261a31..a408ba9cc7e5666571a0f11b0078a2e15bbfd862 100644 (file)
@@ -1,8 +1,8 @@
 /*****************************************************************************
  * tarkin.c: tarkin decoder module making use of libtarkin.
  *****************************************************************************
- * Copyright (C) 1999-2001 VideoLAN
- * $Id: tarkin.c,v 1.3 2002/11/28 17:34:59 sam Exp $
+ * Copyright (C) 2001-2003 the VideoLAN team
+ * $Id$
  *
  * Authors: Gildas Bazin <gbazin@netcourrier.com>
  *
@@ -10,7 +10,7 @@
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation; either version 2 of the License, or
  * (at your option) any later version.
- * 
+ *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  *****************************************************************************/
 
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
-#include <vlc/vlc.h>
-#include <vlc/vout.h>
-#include <vlc/input.h>
-#include <vlc/decoder.h>
-
-#include <stdlib.h>                                      /* malloc(), free() */
-#include <string.h>                                    /* memcpy(), memset() */
-
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_codec.h>
+#include <vlc_vout.h>
 #include <ogg/ogg.h>
 
 /* FIXME */
 #include <tarkin.h>
 
 /*****************************************************************************
- * dec_thread_t : tarkin decoder thread descriptor
+ * decoder_sys_t : tarkin decoder descriptor
  *****************************************************************************/
-typedef struct dec_thread_t
+struct decoder_sys_t
 {
-    /*
-     * Thread properties
-     */
-    vlc_thread_t        thread_id;                /* id for thread functions */
-
     /*
      * Tarkin properties
      */
@@ -61,40 +56,32 @@ typedef struct dec_thread_t
 
     TarkinInfo       ti;                        /* tarkin bitstream settings */
     TarkinComment    tc;                   /* tarkin bitstream user comments */
-    TarkinTime           tarkdate;
-
-    /*
-     * Input properties
-     */
-    decoder_fifo_t         *p_fifo;            /* stores the PES stream data */
-    pes_packet_t           *p_pes;            /* current PES we are decoding */
-
-    /*
-     * Output properties
-     */
-    vout_thread_t *p_vout;
+    TarkinTime       tarkdate;
 
-} dec_thread_t;
+    int i_headers;
+    mtime_t i_pts;
+};
 
 /*****************************************************************************
  * Local prototypes
  *****************************************************************************/
 static int  OpenDecoder  ( vlc_object_t * );
-static int  RunDecoder   ( decoder_fifo_t * );
-static void CloseDecoder ( dec_thread_t * );
+static void CloseDecoder ( vlc_object_t * );
 
-static void DecodePacket ( dec_thread_t * );
-static int  GetOggPacket ( dec_thread_t *, ogg_packet *, mtime_t * );
+static void *DecodeBlock ( decoder_t *, block_t ** );
+static picture_t *DecodePacket ( decoder_t *, block_t **, ogg_packet * );
 
-static void tarkin_CopyPicture( dec_thread_t *, picture_t *, uint8_t * );
+static void tarkin_CopyPicture( decoder_t *, picture_t *, uint8_t *, int );
 
 /*****************************************************************************
  * Module descriptor
  *****************************************************************************/
 vlc_module_begin();
-    set_description( _("Tarkin decoder module") );
+    set_description( N_("Tarkin decoder module") );
     set_capability( "decoder", 100 );
-    set_callbacks( OpenDecoder, NULL );
+    set_category( CAT_INPUT );
+    set_subcategory( SUBCAT_INPUT_VCODEC );
+    set_callbacks( OpenDecoder, CloseDecoder );
     add_shortcut( "tarkin" );
 vlc_module_end();
 
@@ -103,245 +90,231 @@ vlc_module_end();
  *****************************************************************************/
 static int OpenDecoder( vlc_object_t *p_this )
 {
-    decoder_fifo_t *p_fifo = (decoder_fifo_t*) p_this;
+    decoder_t *p_dec = (decoder_t*)p_this;
+    decoder_sys_t *p_sys;
 
-    if( p_fifo->i_fourcc != VLC_FOURCC('t','a','r','k') )
+    if( p_dec->fmt_in.i_codec != VLC_FOURCC('t','a','r','k') )
     {
         return VLC_EGENERIC;
     }
 
-    p_fifo->pf_run = RunDecoder;
+    /* Allocate the memory needed to store the decoder's structure */
+    if( ( p_dec->p_sys = p_sys =
+          (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL )
+        return VLC_ENOMEM;
+
+    /* Set output properties */
+    p_dec->fmt_out.i_cat = VIDEO_ES;
+    p_sys->i_headers = 0;
+
+    /* Set callbacks */
+    p_dec->pf_decode_video = (picture_t *(*)(decoder_t *, block_t **))
+        DecodeBlock;
+    p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))
+        DecodeBlock;
+
+    /* Init supporting Tarkin structures needed in header parsing */
+    p_sys->tarkin_stream = tarkin_stream_new();
+    tarkin_info_init( &p_sys->ti );
+    tarkin_comment_init( &p_sys->tc );
+
     return VLC_SUCCESS;
 }
 
-/*****************************************************************************
- * RunDecoder: the tarkin decoder
- *****************************************************************************/
-static int RunDecoder( decoder_fifo_t *p_fifo )
+/****************************************************************************
+ * DecodeBlock: the whole thing
+ ****************************************************************************
+ * This function must be fed with ogg packets.
+ ****************************************************************************/
+static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 {
-    dec_thread_t *p_dec;
+    decoder_sys_t *p_sys = p_dec->p_sys;
+    block_t *p_block;
     ogg_packet oggpacket;
-    mtime_t i_pts;
-
-    /* Allocate the memory needed to store the thread's structure */
-    if( (p_dec = (dec_thread_t *)malloc (sizeof(dec_thread_t)) )
-            == NULL)
-    {
-        msg_Err( p_fifo, "out of memory" );
-        goto error;
-    }
-
-    /* Initialize the thread properties */
-    memset( p_dec, 0, sizeof(dec_thread_t) );
-    p_dec->p_fifo = p_fifo;
-    p_dec->p_pes  = NULL;
-    p_dec->p_vout = NULL;
 
-    /* Take care of the initial Tarkin header */
-    p_dec->tarkin_stream = tarkin_stream_new();
-    tarkin_info_init(&p_dec->ti);
-    tarkin_comment_init(&p_dec->tc);
+    if( !pp_block ) return NULL;
 
-    if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
-        goto error;
-
-    oggpacket.b_o_s = 1; /* yes this actually is a b_o_s packet :) */
-    if( tarkin_synthesis_headerin( &p_dec->ti, &p_dec->tc, &oggpacket ) < 0 )
+    if( *pp_block )
     {
-        msg_Err( p_dec->p_fifo, "This bitstream does not contain Tarkin "
-                 "video data");
-        goto error;
+        /* Block to Ogg packet */
+        oggpacket.packet = (*pp_block)->p_buffer;
+        oggpacket.bytes = (*pp_block)->i_buffer;
     }
-
-    /* The next two packets in order are the comment and codebook headers.
-       We need to watch out that these packets are not missing as a
-       missing or corrupted header is fatal. */
-    if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
-        goto error;
-
-    if( tarkin_synthesis_headerin( &p_dec->ti, &p_dec->tc, &oggpacket ) < 0 )
+    else
     {
-        msg_Err( p_dec->p_fifo, "2nd Tarkin header is corrupted" );
-        goto error;
+        /* Block to Ogg packet */
+        oggpacket.packet = NULL;
+        oggpacket.bytes = 0;
     }
 
-    if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
-        goto error;
+    p_block = *pp_block;
+
+    oggpacket.granulepos = -1;
+    oggpacket.b_o_s = 0;
+    oggpacket.e_o_s = 0;
+    oggpacket.packetno = 0;
 
-    if( tarkin_synthesis_headerin( &p_dec->ti, &p_dec->tc, &oggpacket ) < 0 )
+    if( p_sys->i_headers == 0 )
     {
-        msg_Err( p_dec->p_fifo, "3rd Tarkin header is corrupted" );
-        goto error;
-    }
+        /* Take care of the initial Tarkin header */
 
-    /* Initialize the tarkin decoder */
-    tarkin_synthesis_init( p_dec->tarkin_stream, &p_dec->ti );
+        oggpacket.b_o_s = 1; /* yes this actually is a b_o_s packet :) */
+        if( tarkin_synthesis_headerin( &p_sys->ti, &p_sys->tc, &oggpacket )
+            < 0 )
+        {
+            msg_Err( p_dec, "this bitstream does not contain Tarkin "
+                     "video data.");
+            block_Release( p_block );
+            return NULL;
+        }
+        p_sys->i_headers++;
 
-    /* tarkin decoder thread's main loop */
-    while( (!p_dec->p_fifo->b_die) && (!p_dec->p_fifo->b_error) )
-    {
-        DecodePacket( p_dec );
+        block_Release( p_block );
+        return NULL;
     }
 
-    /* If b_error is set, the tarkin decoder thread enters the error loop */
-    if( p_dec->p_fifo->b_error )
+    if( p_sys->i_headers == 1 )
     {
-        DecoderError( p_dec->p_fifo );
+        if( tarkin_synthesis_headerin( &p_sys->ti, &p_sys->tc, &oggpacket )
+            < 0 )
+        {
+            msg_Err( p_dec, "2nd Tarkin header is corrupted." );
+            block_Release( p_block );
+            return NULL;
+        }
+        p_sys->i_headers++;
+        block_Release( p_block );
+        return NULL;
     }
 
-    /* End of the tarkin decoder thread */
-    CloseDecoder( p_dec );
+    if( p_sys->i_headers == 2 )
+    {
+        if( tarkin_synthesis_headerin( &p_sys->ti, &p_sys->tc, &oggpacket )
+            < 0 )
+        {
+            msg_Err( p_dec, "3rd Tarkin header is corrupted." );
+            block_Release( p_block );
+            return NULL;
+        }
+        p_sys->i_headers++;
 
-    return 0;
+        /* Initialize the tarkin decoder */
+        tarkin_synthesis_init( p_sys->tarkin_stream, &p_sys->ti );
 
- error:
-    DecoderError( p_fifo );
-    if( p_dec )
-    {
-        if( p_dec->p_fifo )
-            p_dec->p_fifo->b_error = 1;
+        msg_Err( p_dec, "Tarkin codec initialized");
 
-        /* End of the tarkin decoder thread */
-        CloseDecoder( p_dec );
+        block_Release( p_block );
+        return NULL;
     }
 
-    return -1;
+    return DecodePacket( p_dec, pp_block, &oggpacket );
 }
 
 /*****************************************************************************
  * DecodePacket: decodes a Tarkin packet.
  *****************************************************************************/
-static void DecodePacket( dec_thread_t *p_dec )
+static picture_t *DecodePacket( decoder_t *p_dec, block_t **pp_block,
+                                ogg_packet *p_oggpacket )
 {
-    ogg_packet oggpacket;
-    picture_t *p_pic;
-    mtime_t i_pts;
-    int i_width, i_height, i_chroma, i_aspect;
-    uint32_t frame = 0;
+    decoder_sys_t *p_sys = p_dec->p_sys;
     uint8_t *rgb;
 
-    if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
+    if( p_oggpacket->bytes )
     {
-        /* This should mean an eos */
-        return;
+        tarkin_synthesis_packetin( p_sys->tarkin_stream, p_oggpacket );
+        //block_Release( *pp_block ); /* FIXME duplicate packet */
+        *pp_block = NULL;
     }
 
-    tarkin_synthesis_packetin( p_dec->tarkin_stream, &oggpacket );
-
-    while( tarkin_synthesis_frameout( p_dec->tarkin_stream,
-                                      &rgb, 0, &p_dec->tarkdate ) == 0 )
+    if( tarkin_synthesis_frameout( p_sys->tarkin_stream,
+                                   &rgb, 0, &p_sys->tarkdate ) == 0 )
     {
+        int i_width, i_height, i_chroma, i_stride;
+        picture_t *p_pic;
+
+        msg_Err( p_dec, "Tarkin frame decoded" );
 
-        i_width = p_dec->tarkin_stream->layer->desc.width;
-        i_height = p_dec->tarkin_stream->layer->desc.height;
-        switch( p_dec->tarkin_stream->layer->desc.format )
+        i_width = p_sys->tarkin_stream->layer->desc.width;
+        i_height = p_sys->tarkin_stream->layer->desc.height;
+
+        switch( p_sys->tarkin_stream->layer->desc.format )
         {
         case TARKIN_RGB24:
             i_chroma = VLC_FOURCC('R','V','2','4');
+            i_stride = i_width * 3;
             break;
         case TARKIN_RGB32:
             i_chroma = VLC_FOURCC('R','V','3','2');
+            i_stride = i_width * 4;
             break;
         case TARKIN_RGBA:
             i_chroma = VLC_FOURCC('R','G','B','A');
+            i_stride = i_width * 4;
             break;
         default:
-            i_chroma = VLC_FOURCC('Y','V','1','2');
+            i_chroma = VLC_FOURCC('I','4','2','0');
+            i_stride = i_width;
             break;
         }
-        i_aspect = VOUT_ASPECT_FACTOR * i_width / i_height;
-        p_dec->p_vout = vout_Request( p_dec->p_fifo, p_dec->p_vout,
-                                      i_width, i_height, i_aspect, i_chroma );
+
+        /* Set output properties */
+        p_dec->fmt_out.video.i_width = i_width;
+        p_dec->fmt_out.video.i_height = i_height;
+
+        p_dec->fmt_out.video.i_aspect =
+            VOUT_ASPECT_FACTOR * i_width / i_height;
+        p_dec->fmt_out.i_codec = i_chroma;
 
         /* Get a new picture */
-        while( !(p_pic = vout_CreatePicture( p_dec->p_vout, 0, 0, 0 ) ) )
+        if( (p_pic = p_dec->pf_vout_buffer_new( p_dec )) )
         {
-            if( p_dec->p_fifo->b_die || p_dec->p_fifo->b_error )
-            {
-                return;
-            }
-            msleep( VOUT_OUTMEM_SLEEP );
-        }
-        if( !p_pic )
-            break;
-
-        tarkin_CopyPicture( p_dec, p_pic, rgb );
+            tarkin_CopyPicture( p_dec, p_pic, rgb, i_stride );
 
-        tarkin_synthesis_freeframe( p_dec->tarkin_stream, rgb );
+            tarkin_synthesis_freeframe( p_sys->tarkin_stream, rgb );
 
-        vout_DatePicture( p_dec->p_vout, p_pic, mdate()+DEFAULT_PTS_DELAY/*i_pts*/ );
-        vout_DisplayPicture( p_dec->p_vout, p_pic );
+            p_pic->date = mdate() + DEFAULT_PTS_DELAY/*i_pts*/;
 
+            return p_pic;
+        }
     }
-}
-
-/*****************************************************************************
- * GetOggPacket: get the following tarkin packet from the stream and send back
- *               the result in an ogg packet (for easy decoding by libtarkin).
- *****************************************************************************
- * Returns VLC_EGENERIC in case of eof.
- *****************************************************************************/
-static int GetOggPacket( dec_thread_t *p_dec, ogg_packet *p_oggpacket,
-                         mtime_t *p_pts )
-{
-    if( p_dec->p_pes ) input_DeletePES( p_dec->p_fifo->p_packets_mgt,
-                                        p_dec->p_pes );
-
-    input_ExtractPES( p_dec->p_fifo, &p_dec->p_pes );
-    if( !p_dec->p_pes ) return VLC_EGENERIC;
 
-    p_oggpacket->packet = p_dec->p_pes->p_first->p_payload_start;
-    p_oggpacket->bytes = p_dec->p_pes->i_pes_size;
-    p_oggpacket->granulepos = p_dec->p_pes->i_dts;
-    p_oggpacket->b_o_s = 0;
-    p_oggpacket->e_o_s = 0;
-    p_oggpacket->packetno = 0;
-
-    *p_pts = p_dec->p_pes->i_pts;
-
-    return VLC_SUCCESS;
+    return NULL;
 }
 
 /*****************************************************************************
  * CloseDecoder: tarkin decoder destruction
  *****************************************************************************/
-static void CloseDecoder( dec_thread_t * p_dec )
+static void CloseDecoder( vlc_object_t *p_this )
 {
+    decoder_t *p_dec = (decoder_t *)p_this;
+    decoder_sys_t *p_sys = p_dec->p_sys;
 
-    if( p_dec )
-    {
-        if( p_dec->p_pes )
-            input_DeletePES( p_dec->p_fifo->p_packets_mgt, p_dec->p_pes );
-
-        vout_Request( p_dec, p_dec->p_vout, 0, 0, 0, 0 );
-
-        if( p_dec->tarkin_stream )
-            tarkin_stream_destroy( p_dec->tarkin_stream );
+    tarkin_stream_destroy( p_sys->tarkin_stream );
 
-        free( p_dec );
-    }
+    free( p_sys );
 }
 
 /*****************************************************************************
  * tarkin_CopyPicture: copy a picture from tarkin internal buffers to a
  *                     picture_t structure.
  *****************************************************************************/
-static void tarkin_CopyPicture( dec_thread_t *p_dec, picture_t *p_pic,
-                                uint8_t *p_src )
+static void tarkin_CopyPicture( decoder_t *p_dec, picture_t *p_pic,
+                                uint8_t *p_src, int i_pitch )
 {
-    int i_plane, i_line, i_width, i_dst_stride;
-    u *p_dst;
+    int i_plane, i_line, i_src_stride, i_dst_stride;
+    uint8_t *p_dst;
 
     for( i_plane = 0; i_plane < p_pic->i_planes; i_plane++ )
     {
         p_dst = p_pic->p[i_plane].p_pixels;
-        i_width = p_pic->p[i_plane].i_visible_pitch;
         i_dst_stride = p_pic->p[i_plane].i_pitch;
+        i_src_stride = i_pitch;
 
-        for( i_line = 0; i_line < p_pic->p[i_plane].i_lines; i_line++ )
+        for( i_line = 0; i_line < p_pic->p[i_plane].i_visible_lines; i_line++ )
         {
-            p_dec->p_fifo->p_vlc->pf_memcpy( p_dst, p_src, i_width );
-            p_src += i_width;
+            vlc_memcpy( p_dst, p_src, i_src_stride );
+
+            p_src += i_src_stride;
             p_dst += i_dst_stride;
         }
     }