svg.c 8.62 KB
Newer Older
Adam Leggett's avatar
Adam Leggett committed
1 2 3 4 5 6
/*****************************************************************************
 * svg.c: svg decoder module making use of librsvg2.
 *****************************************************************************
 * Copyright (C) 2014 VLC authors and VideoLAN
 *
 * Authors: Adam Leggett <adamvleggett@gmail.com>
Jean-Paul Saman's avatar
Jean-Paul Saman committed
7
 *          Jean-Paul Saman <jpsaman@videolan.org>
Adam Leggett's avatar
Adam Leggett committed
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 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
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

#include <vlc_common.h>
#include <vlc_plugin.h>
#include <vlc_codec.h>

#include <glib.h>
#include <glib/gstdio.h>
#include <glib-object.h>                                  /* g_object_unref( ) */

#include <librsvg/rsvg.h>
#include <cairo/cairo.h>

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
static int  OpenDecoder   ( vlc_object_t * );
static void CloseDecoder  ( vlc_object_t * );

static picture_t *DecodeBlock  ( decoder_t *, block_t ** );

Jean-Paul Saman's avatar
Jean-Paul Saman committed
50 51 52 53 54 55 56
#define TEXT_WIDTH       N_("Image width")
#define LONG_TEXT_WIDTH  N_("Specify the width to decode the image too")
#define TEXT_HEIGHT      N_("Image height")
#define LONG_TEXT_HEIGHT N_("Specify the height to decode the image too")
#define TEXT_SCALE       N_("Scale factor")
#define LONG_TEXT_SCALE  N_("Scale factor to apply to image")

Adam Leggett's avatar
Adam Leggett committed
57 58 59 60 61 62 63 64 65 66
/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
vlc_module_begin ()
    set_category( CAT_INPUT )
    set_subcategory( SUBCAT_INPUT_VCODEC )
    set_description( N_("SVG video decoder") )
    set_capability( "decoder", 100 )
    set_callbacks( OpenDecoder, CloseDecoder )
    add_shortcut( "svg" )
Jean-Paul Saman's avatar
Jean-Paul Saman committed
67 68 69 70 71 72 73 74 75 76

    /* svg options */
    add_integer_with_range( "svg-width", -1, 1, 65535,
                            TEXT_WIDTH, LONG_TEXT_WIDTH, false )
        change_safe()
    add_integer_with_range( "svg-height", -1, 1, 65535,
                            TEXT_HEIGHT, LONG_TEXT_HEIGHT, false )
        change_safe()

    add_float( "svg-scale", -1.0, TEXT_SCALE, LONG_TEXT_SCALE, false )
Adam Leggett's avatar
Adam Leggett committed
77 78
vlc_module_end ()

Jean-Paul Saman's avatar
Jean-Paul Saman committed
79 80 81 82 83 84 85
struct decoder_sys_t
{
    int32_t i_width;
    int32_t i_height;
    double  f_scale;
};

Adam Leggett's avatar
Adam Leggett committed
86 87 88 89 90 91 92 93 94 95
/*****************************************************************************
 * OpenDecoder: probe the decoder and return score
 *****************************************************************************/
static int OpenDecoder( vlc_object_t *p_this )
{
    decoder_t *p_dec = (decoder_t*)p_this;

    if( p_dec->fmt_in.i_codec != VLC_CODEC_SVG )
        return VLC_EGENERIC;

Jean-Paul Saman's avatar
Jean-Paul Saman committed
96 97 98 99 100 101 102 103 104
    decoder_sys_t *p_sys = malloc( sizeof(decoder_sys_t) );
    if (!p_sys)
        return VLC_ENOMEM;
    p_dec->p_sys = p_sys;

    p_sys->i_width = var_InheritInteger( p_this, "svg-width" );
    p_sys->i_height = var_InheritInteger( p_this, "svg-height" );
    p_sys->f_scale = var_InheritFloat( p_this, "svg-scale" );

Adam Leggett's avatar
Adam Leggett committed
105 106 107 108 109
    /* Initialize library */
    rsvg_init();

    /* Set output properties */
    p_dec->fmt_out.i_cat = VIDEO_ES;
Jean-Paul Saman's avatar
Jean-Paul Saman committed
110
    p_dec->fmt_out.i_codec = VLC_CODEC_RGB32;
Adam Leggett's avatar
Adam Leggett committed
111 112 113 114 115 116 117 118 119 120 121 122 123 124

    /* Set callbacks */
    p_dec->pf_decode_video = DecodeBlock;

    return VLC_SUCCESS;
}

/****************************************************************************
 * DecodeBlock: the whole thing
 ****************************************************************************
 * This function must be fed with a complete image.
 ****************************************************************************/
static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
{
Jean-Paul Saman's avatar
Jean-Paul Saman committed
125
    decoder_sys_t *p_sys  = (decoder_sys_t *) p_dec->p_sys;
Adam Leggett's avatar
Adam Leggett committed
126 127
    block_t *p_block;
    picture_t *p_pic = NULL;
Jean-Paul Saman's avatar
Jean-Paul Saman committed
128
    int32_t i_width, i_height;
Adam Leggett's avatar
Adam Leggett committed
129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144

    RsvgHandle *rsvg = NULL;
    cairo_surface_t *surface = NULL;
    cairo_t *cr = NULL;

    if( !pp_block || !*pp_block ) return NULL;

    p_block = *pp_block;
    *pp_block = NULL;

    if( p_block->i_flags & BLOCK_FLAG_DISCONTINUITY )
    {
        block_Release( p_block );
        return NULL;
    }

Jean-Paul Saman's avatar
Jean-Paul Saman committed
145
    rsvg = rsvg_handle_new_from_data( p_block->p_buffer, p_block->i_buffer, NULL );
Adam Leggett's avatar
Adam Leggett committed
146 147 148 149
    if( !rsvg )
        goto done;

    RsvgDimensionData dim;
Jean-Paul Saman's avatar
Jean-Paul Saman committed
150
    rsvg_handle_get_dimensions( rsvg, &dim );
Adam Leggett's avatar
Adam Leggett committed
151

Jean-Paul Saman's avatar
Jean-Paul Saman committed
152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
    if( p_sys->f_scale > 0.0 )
    {
        i_width  = (int32_t)(p_sys->f_scale * dim.width);
        i_height = (int32_t)(p_sys->f_scale * dim.height);
    }
    else
    {
        /* Keep aspect */
        if( p_sys->i_width < 0 && p_sys->i_height > 0 )
        {
            i_width  = dim.width * p_sys->i_height / dim.height;
            i_height = p_sys->i_height;
        }
        else if( p_sys->i_width > 0 && p_sys->i_height < 0 )
        {
            i_width  = p_sys->i_width;
            i_height = dim.height * p_sys->i_width / dim.height;
        }
        else if( p_sys->i_width > 0 && p_sys->i_height > 0 )
        {
            i_width  = dim.width * p_sys->i_height / dim.height;
            i_height = p_sys->i_height;
        }
        else
        {
            i_width  = dim.width;
            i_height = dim.height;
        }
    }

Jean-Paul Saman's avatar
Jean-Paul Saman committed
182
    p_dec->fmt_out.video.i_chroma = VLC_CODEC_RGB32;
Jean-Paul Saman's avatar
Jean-Paul Saman committed
183 184 185 186
    p_dec->fmt_out.video.i_width  = i_width;
    p_dec->fmt_out.video.i_height = i_height;
    p_dec->fmt_out.video.i_visible_width  = i_width;
    p_dec->fmt_out.video.i_visible_height = i_height;
Adam Leggett's avatar
Adam Leggett committed
187 188
    p_dec->fmt_out.video.i_sar_num = 1;
    p_dec->fmt_out.video.i_sar_den = 1;
Jean-Paul Saman's avatar
Jean-Paul Saman committed
189
    p_dec->fmt_out.video.i_rmask = 0x80800000; /* Since librsvg v1.0 */
Adam Leggett's avatar
Adam Leggett committed
190
    p_dec->fmt_out.video.i_gmask = 0x0000ff00;
Jean-Paul Saman's avatar
Jean-Paul Saman committed
191 192
    p_dec->fmt_out.video.i_bmask = 0x000000ff;
    video_format_FixRgb(&p_dec->fmt_out.video);
Adam Leggett's avatar
Adam Leggett committed
193 194 195 196 197 198

    /* Get a new picture */
    p_pic = decoder_NewPicture( p_dec );
    if( !p_pic )
        goto done;

Jean-Paul Saman's avatar
Jean-Paul Saman committed
199 200 201 202 203 204
    /* NOTE: Do not use the stride calculation from cairo, because it is wrong:
     * stride = cairo_format_stride_for_width(CAIRO_FORMAT_ARGB32, dim.width);
     * Use the stride from VLC its picture_t::p[0].i_pitch, which is correct.
     */
    surface = cairo_image_surface_create_for_data( p_pic->p->p_pixels,
                                                   CAIRO_FORMAT_ARGB32,
Jean-Paul Saman's avatar
Jean-Paul Saman committed
205
                                                   i_width, i_height,
Jean-Paul Saman's avatar
Jean-Paul Saman committed
206
                                                   p_pic->p[0].i_pitch );
Adam Leggett's avatar
Adam Leggett committed
207 208
    if( !surface )
    {
Jean-Paul Saman's avatar
Jean-Paul Saman committed
209
        picture_Release( p_pic );
Adam Leggett's avatar
Adam Leggett committed
210 211 212 213 214
        p_pic = NULL;
        goto done;
    }

    /* Decode picture */
Jean-Paul Saman's avatar
Jean-Paul Saman committed
215
    cr = cairo_create( surface );
Adam Leggett's avatar
Adam Leggett committed
216 217
    if( !cr )
    {
Jean-Paul Saman's avatar
Jean-Paul Saman committed
218
        picture_Release( p_pic );
Adam Leggett's avatar
Adam Leggett committed
219 220 221 222
        p_pic = NULL;
        goto done;
    }

Jean-Paul Saman's avatar
Jean-Paul Saman committed
223 224 225 226 227 228 229 230 231 232 233 234 235 236 237
    if ( i_width != dim.width || i_height != dim.height )
    {
        double sw, sh;
        if ( p_sys->f_scale > 0.0 && !(p_sys->i_width > 0 || p_sys->i_height > 0) )
            sw = sh = p_sys->f_scale;
        else
        {
            double aspect = (double) (dim.width * p_dec->fmt_out.video.i_sar_num) /
                    (dim.height * p_dec->fmt_out.video.i_sar_den);
            sw = aspect * i_width / dim.width;
            sh = aspect * i_height / dim.height;
        }
        cairo_scale(cr, sw, sh);
    }

Jean-Paul Saman's avatar
Jean-Paul Saman committed
238 239 240 241 242 243
    if( !rsvg_handle_render_cairo( rsvg, cr ) )
    {
        picture_Release( p_pic );
        p_pic = NULL;
        goto done;
    }
Adam Leggett's avatar
Adam Leggett committed
244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266

    p_pic->date = p_block->i_pts > VLC_TS_INVALID ? p_block->i_pts : p_block->i_dts;

done:
    if( rsvg )
        g_object_unref( G_OBJECT( rsvg ) );
    if( cr )
        cairo_destroy( cr );
    if( surface )
        cairo_surface_destroy( surface );

    block_Release( p_block );
    return p_pic;
}

/*****************************************************************************
 * CloseDecoder: png decoder destruction
 *****************************************************************************/
static void CloseDecoder( vlc_object_t *p_this )
{
    VLC_UNUSED( p_this );
    rsvg_term();
}