modules/codec/gstreamer/gstvlcvideopool.c

   1 /*****************************************************************************
   2  * gstvlcvideopool.c: VLC pictures managed by GstBufferPool
   3  *****************************************************************************
   4  * Copyright (C) 2016 VLC authors and VideoLAN
   5  * $Id:
   6  *
   7  * Author: Vikram Fugro <vikram.fugro@gmail.com>
   8  *
   9  * This library is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU Library General Public
  11  * License as published by the Free Software Foundation; either
  12  * version 2.1 of the License, or (at your option) any later version.
  13  *
  14  * This library 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 GNU
  17  * Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Library General Public
  20  * License along with this library; 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 #include "gstvlcvideopool.h"
  28
  29 #include <vlc_common.h>
  30
  31 /* bufferpool */
  32 static void gst_vlc_video_pool_finalize( GObject *p_object );
  33
  34 #define gst_vlc_video_pool_parent_class parent_class
  35 G_DEFINE_TYPE (GstVlcVideoPool, gst_vlc_video_pool,
  36     GST_TYPE_BUFFER_POOL);
  37
  38 static const gchar** gst_vlc_video_pool_get_options (GstBufferPool *p_pool)
  39 {
  40     static const gchar *options[] = { GST_BUFFER_POOL_OPTION_VIDEO_META,
  41         NULL
  42     };
  43
  44     return options;
  45 }
  46
  47 static gboolean gst_vlc_video_pool_set_config( GstBufferPool *p_pool,
  48         GstStructure *p_config )
  49 {
  50     GstVlcVideoPool *p_vpool = GST_VLC_VIDEO_POOL_CAST( p_pool );
  51     GstCaps *p_caps;
  52     GstVideoInfo info;
  53     guint size, min_buffers, max_buffers;
  54     GstAllocator *p_allocator;
  55     GstAllocationParams params;
  56
  57     if( !gst_buffer_pool_config_get_params( p_config, &p_caps, &size,
  58                 &min_buffers, &max_buffers ))
  59         goto wrong_config;
  60
  61     if( p_caps == NULL )
  62         goto no_caps;
  63
  64     gst_buffer_pool_config_get_allocator( p_config, &p_allocator, &params );
  65     if( p_allocator )
  66     {
  67         if( !GST_IS_VLC_PICTURE_PLANE_ALLOCATOR( p_allocator ))
  68             goto unsupported_allocator;
  69         else
  70         {
  71             if( p_vpool->p_allocator )
  72                 gst_object_unref( p_vpool->p_allocator );
  73             p_vpool->p_allocator = gst_object_ref ( p_allocator );
  74         }
  75     }
  76
  77     /* now parse the caps from the config */
  78     if ( !gst_video_info_from_caps( &info, p_caps ))
  79         goto wrong_caps;
  80
  81     /* enable metadata based on config of the pool */
  82     p_vpool->b_add_metavideo =
  83         gst_buffer_pool_config_has_option( p_config,
  84                 GST_BUFFER_POOL_OPTION_VIDEO_META );
  85
  86     if( !gst_vlc_picture_plane_allocator_query_format( p_vpool->p_allocator,
  87                 &info, p_caps))
  88         goto unknown_format;
  89
  90     if( p_vpool->p_caps )
  91         gst_caps_unref( p_vpool->p_caps );
  92     p_vpool->p_caps = gst_caps_ref( p_caps );
  93     p_vpool->info = info;
  94
  95     gst_buffer_pool_config_set_params( p_config, p_caps, info.size,
  96             min_buffers, max_buffers );
  97
  98     return GST_BUFFER_POOL_CLASS (parent_class)->set_config( p_pool, p_config );
  99
 100     /* ERRORS */
 101 wrong_config:
 102     {
 103         msg_Err(p_vpool->p_dec, "wrong pool config" );
 104         return FALSE;
 105     }
 106 no_caps:
 107     {
 108         msg_Err(p_vpool->p_dec, "no input caps in config" );
 109         return FALSE;
 110     }
 111 wrong_caps:
 112     {
 113         msg_Err(p_vpool->p_dec, "invalid caps" );
 114         return FALSE;
 115     }
 116 unknown_format:
 117     {
 118         msg_Err(p_vpool->p_dec, "format unsupported" );
 119         return FALSE;
 120     }
 121 unsupported_allocator:
 122     {
 123         msg_Err(p_vpool->p_dec, "allocator unsupported" );
 124         return FALSE;
 125     }
 126 }
 127
 128 static GstFlowReturn gst_vlc_video_pool_acquire_buffer( GstBufferPool *p_pool,
 129         GstBuffer **p_buffer, GstBufferPoolAcquireParams *p_params )
 130 {
 131     GstVlcVideoPool *p_vpool = GST_VLC_VIDEO_POOL_CAST( p_pool );
 132     GstVideoInfo *p_info;
 133     GstFlowReturn result;
 134
 135     result = GST_BUFFER_POOL_CLASS( parent_class)->acquire_buffer( p_pool,
 136             p_buffer, p_params );
 137
 138     if( result == GST_FLOW_OK &&
 139             !gst_vlc_picture_plane_allocator_hold( p_vpool->p_allocator,
 140                 *p_buffer ))
 141         result = GST_FLOW_EOS;
 142
 143     return result;
 144 }
 145
 146 static void gst_vlc_video_pool_release_buffer( GstBufferPool *p_pool,
 147         GstBuffer *p_buffer )
 148 {
 149     GstVlcVideoPool* p_vpool = GST_VLC_VIDEO_POOL_CAST( p_pool );
 150
 151     gst_vlc_picture_plane_allocator_release( p_vpool->p_allocator, p_buffer );
 152
 153     GST_BUFFER_POOL_CLASS( parent_class )->release_buffer( p_pool, p_buffer );
 154
 155     return;
 156 }
 157
 158 static void gst_vlc_video_pool_free_buffer( GstBufferPool *p_pool,
 159         GstBuffer *p_buffer )
 160 {
 161     GstVlcVideoPool* p_vpool = GST_VLC_VIDEO_POOL_CAST( p_pool );
 162
 163     gst_vlc_picture_plane_allocator_release( p_vpool->p_allocator, p_buffer );
 164
 165     GST_BUFFER_POOL_CLASS( parent_class )->free_buffer( p_pool, p_buffer );
 166
 167     return;
 168 }
 169
 170 static GstFlowReturn gst_vlc_video_pool_alloc_buffer( GstBufferPool *p_pool,
 171         GstBuffer **p_buffer, GstBufferPoolAcquireParams *p_params)
 172 {
 173     GstVlcVideoPool *p_vpool = GST_VLC_VIDEO_POOL_CAST( p_pool );
 174     GstVideoInfo *p_info = &p_vpool->info;
 175
 176     *p_buffer = gst_buffer_new( );
 177
 178     if( !gst_vlc_picture_plane_allocator_alloc( p_vpool->p_allocator,
 179                 *p_buffer ))
 180         return GST_FLOW_EOS;
 181
 182     if( p_vpool->b_add_metavideo )
 183     {
 184         msg_Dbg( p_vpool->p_dec, "meta video enabled" );
 185         gst_buffer_add_video_meta_full( *p_buffer, GST_VIDEO_FRAME_FLAG_NONE,
 186                 GST_VIDEO_INFO_FORMAT( p_info ), GST_VIDEO_INFO_WIDTH( p_info ),
 187                 GST_VIDEO_INFO_HEIGHT( p_info ),
 188                 GST_VIDEO_INFO_N_PLANES( p_info ),
 189                 p_info->offset, p_info->stride );
 190     }
 191
 192     return GST_FLOW_OK;
 193 }
 194
 195 static void gst_vlc_video_pool_class_init( GstVlcVideoPoolClass *p_klass )
 196 {
 197     GObjectClass *p_gobject_class = ( GObjectClass* )p_klass;
 198     GstBufferPoolClass *p_gstbufferpool_class = ( GstBufferPoolClass* )p_klass;
 199
 200     p_gobject_class->finalize = gst_vlc_video_pool_finalize;
 201
 202     p_gstbufferpool_class->get_options = gst_vlc_video_pool_get_options;
 203     p_gstbufferpool_class->set_config = gst_vlc_video_pool_set_config;
 204     p_gstbufferpool_class->alloc_buffer = gst_vlc_video_pool_alloc_buffer;
 205     p_gstbufferpool_class->free_buffer = gst_vlc_video_pool_free_buffer;
 206     p_gstbufferpool_class->acquire_buffer = gst_vlc_video_pool_acquire_buffer;
 207     p_gstbufferpool_class->release_buffer = gst_vlc_video_pool_release_buffer;
 208 }
 209
 210 static void gst_vlc_video_pool_init( GstVlcVideoPool *p_pool )
 211 {
 212 }
 213
 214 static void gst_vlc_video_pool_finalize( GObject *p_object )
 215 {
 216     GstVlcVideoPool *p_pool = GST_VLC_VIDEO_POOL_CAST( p_object );
 217
 218     gst_object_unref( p_pool->p_allocator );
 219
 220     G_OBJECT_CLASS( parent_class )->finalize( p_object );
 221 }
 222
 223 GstVlcVideoPool* gst_vlc_video_pool_new(
 224     GstAllocator *p_allocator, decoder_t *p_dec )
 225 {
 226     GstVlcVideoPool *p_pool;
 227
 228     if( !GST_IS_VLC_PICTURE_PLANE_ALLOCATOR( p_allocator ))
 229     {
 230         msg_Err( p_pool->p_dec, "unspported allocator for pool" );
 231         return NULL;
 232     }
 233
 234     p_pool = g_object_new( GST_TYPE_VLC_VIDEO_POOL, NULL );
 235     p_pool->p_allocator = gst_object_ref( p_allocator );
 236     p_pool->p_dec = p_dec;
 237
 238     return p_pool;
 239 }