vout_pictures.c 14.1 KB
Newer Older
Sam Hocevar's avatar
 
Sam Hocevar committed
1 2 3
/*****************************************************************************
 * vout_pictures.c : picture management functions
 *****************************************************************************
4
 * Copyright (C) 2000-2004 the VideoLAN team
5
 * $Id$
Sam Hocevar's avatar
 
Sam Hocevar committed
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * Authors: Vincent Seguin <seguin@via.ecp.fr>
 *          Samuel Hocevar <sam@zoy.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * 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
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
dionoea's avatar
dionoea committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
Sam Hocevar's avatar
 
Sam Hocevar committed
23 24 25 26 27 28
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/

29 30 31
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
32
#include <assert.h>
33

34
#include <vlc_common.h>
35
#include <libvlc.h>
zorglub's avatar
zorglub committed
36
#include <vlc_vout.h>
37
#include <vlc_filter.h>
38 39
#include <vlc_image.h>
#include <vlc_block.h>
Laurent Aimar's avatar
Laurent Aimar committed
40
#include <vlc_picture_fifo.h>
41
#include <vlc_picture_pool.h>
42

43
#include "vout_internal.h"
44

45
/**
Laurent Aimar's avatar
Laurent Aimar committed
46 47
 * It retreives a picture from the vout or NULL if no pictures are
 * available yet.
48
 *
Laurent Aimar's avatar
Laurent Aimar committed
49
 * You MUST call vout_PutPicture or vout_ReleasePicture on it.
50
 *
Laurent Aimar's avatar
Laurent Aimar committed
51 52
 * You may use vout_HoldPicture(paired with vout_ReleasePicture) to keep a
 * read-only reference.
53
 */
Laurent Aimar's avatar
Laurent Aimar committed
54
picture_t *vout_GetPicture( vout_thread_t *p_vout )
Sam Hocevar's avatar
 
Sam Hocevar committed
55 56
{
    /* Get lock */
57
    vlc_mutex_lock( &p_vout->p->picture_lock );
58 59 60
    picture_t *p_pic = picture_pool_Get(p_vout->p->decoder_pool);
    if (p_pic) {
        picture_Reset(p_pic);
61
        p_pic->p_next = NULL;
Sam Hocevar's avatar
 
Sam Hocevar committed
62
    }
63
    vlc_mutex_unlock( &p_vout->p->picture_lock );
Sam Hocevar's avatar
 
Sam Hocevar committed
64

65
    return p_pic;
Sam Hocevar's avatar
 
Sam Hocevar committed
66 67
}

Laurent Aimar's avatar
Laurent Aimar committed
68 69 70 71 72 73 74 75 76
/**
 * It gives to the vout a picture to be displayed.
 *
 * The given picture MUST comes from vout_GetPicture.
 *
 * Becareful, after vout_PutPicture is called, picture_t::p_next cannot be
 * read/used.
 */
void vout_PutPicture( vout_thread_t *p_vout, picture_t *p_pic )
77
{
78
    vlc_mutex_lock( &p_vout->p->picture_lock );
79

Laurent Aimar's avatar
Laurent Aimar committed
80 81
    p_pic->p_next = NULL;
    picture_fifo_Push(p_vout->p->decoder_fifo, p_pic);
82

83
    vlc_mutex_unlock( &p_vout->p->picture_lock );
84 85

    vout_control_Wake( &p_vout->p->control);
86 87
}

88
/**
Laurent Aimar's avatar
Laurent Aimar committed
89
 * It releases a picture retreived by vout_GetPicture.
90
 */
Laurent Aimar's avatar
Laurent Aimar committed
91
void vout_ReleasePicture( vout_thread_t *p_vout, picture_t *p_pic  )
Sam Hocevar's avatar
 
Sam Hocevar committed
92
{
93
    vlc_mutex_lock( &p_vout->p->picture_lock );
Laurent Aimar's avatar
Laurent Aimar committed
94 95 96

    picture_Release( p_pic );

97
    vlc_mutex_unlock( &p_vout->p->picture_lock );
98 99

    vout_control_Wake( &p_vout->p->control);
Sam Hocevar's avatar
 
Sam Hocevar committed
100 101
}

102
/**
Laurent Aimar's avatar
Laurent Aimar committed
103 104
 * It increment the reference counter of a picture retreived by
 * vout_GetPicture.
105
 */
Laurent Aimar's avatar
Laurent Aimar committed
106
void vout_HoldPicture( vout_thread_t *p_vout, picture_t *p_pic )
Sam Hocevar's avatar
 
Sam Hocevar committed
107
{
108
    vlc_mutex_lock( &p_vout->p->picture_lock );
Sam Hocevar's avatar
 
Sam Hocevar committed
109

Laurent Aimar's avatar
Laurent Aimar committed
110 111
    picture_Hold( p_pic );

112
    vlc_mutex_unlock( &p_vout->p->picture_lock );
Sam Hocevar's avatar
 
Sam Hocevar committed
113 114
}

115 116 117
/**
 * Allocate a new picture in the heap.
 *
Sam Hocevar's avatar
 
Sam Hocevar committed
118 119 120
 * This function allocates a fake direct buffer in memory, which can be
 * used exactly like a video buffer. The video output thread then manages
 * how it gets displayed.
121
 */
122 123 124 125
static int vout_AllocatePicture( picture_t *p_pic,
                                 vlc_fourcc_t i_chroma,
                                 int i_width, int i_height,
                                 int i_sar_num, int i_sar_den )
Sam Hocevar's avatar
 
Sam Hocevar committed
126
{
127
    /* Make sure the real dimensions are a multiple of 16 */
128 129
    if( picture_Setup( p_pic, i_chroma, i_width, i_height,
                       i_sar_num, i_sar_den ) != VLC_SUCCESS )
130
        return VLC_EGENERIC;
gbazin's avatar
 
gbazin committed
131 132

    /* Calculate how big the new image should be */
133
    size_t i_bytes = 0;
134
    for( int i = 0; i < p_pic->i_planes; i++ )
135
    {
136 137 138 139
        const plane_t *p = &p_pic->p[i];

        if( p->i_pitch <= 0 || p->i_lines <= 0 ||
            p->i_pitch > (SIZE_MAX - i_bytes)/p->i_lines )
140
        {
141 142
            p_pic->i_planes = 0;
            return VLC_ENOMEM;
143
        }
144
        i_bytes += p->i_pitch * p->i_lines;
145
    }
gbazin's avatar
 
gbazin committed
146

147 148
    uint8_t *p_data = vlc_memalign( &p_pic->p_data_orig, 16, i_bytes );
    if( !p_data )
gbazin's avatar
 
gbazin committed
149 150
    {
        p_pic->i_planes = 0;
151
        return VLC_EGENERIC;
gbazin's avatar
 
gbazin committed
152 153 154
    }

    /* Fill the p_pixels field for each plane */
155
    p_pic->p[0].p_pixels = p_data;
156
    for( int i = 1; i < p_pic->i_planes; i++ )
gbazin's avatar
 
gbazin committed
157
    {
158 159
        p_pic->p[i].p_pixels = &p_pic->p[i-1].p_pixels[ p_pic->p[i-1].i_lines *
                                                        p_pic->p[i-1].i_pitch ];
gbazin's avatar
 
gbazin committed
160
    }
161 162

    return VLC_SUCCESS;
gbazin's avatar
 
gbazin committed
163 164
}

165 166 167 168 169 170 171 172 173
/*****************************************************************************
 *
 *****************************************************************************/
static void PictureReleaseCallback( picture_t *p_picture )
{
    if( --p_picture->i_refcount > 0 )
        return;
    picture_Delete( p_picture );
}
174

Laurent Aimar's avatar
Laurent Aimar committed
175 176 177 178 179 180 181 182 183 184 185 186 187 188
/*****************************************************************************
 *
 *****************************************************************************/
void picture_Reset( picture_t *p_picture )
{
    /* */
    p_picture->date = VLC_TS_INVALID;
    p_picture->b_force = false;
    p_picture->b_progressive = false;
    p_picture->i_nb_fields = 0;
    p_picture->b_top_field_first = false;
    picture_CleanupQuant( p_picture );
}

189 190 191
/*****************************************************************************
 *
 *****************************************************************************/
192 193 194 195 196
static int LCM( int a, int b )
{
    return a * b / GCD( a, b );
}

197 198
int picture_Setup( picture_t *p_picture, vlc_fourcc_t i_chroma,
                   int i_width, int i_height, int i_sar_num, int i_sar_den )
199 200
{
    /* Store default values */
201 202
    p_picture->i_planes = 0;
    for( unsigned i = 0; i < VOUT_MAX_PLANES; i++ )
203
    {
204 205 206
        plane_t *p = &p_picture->p[i];
        p->p_pixels = NULL;
        p->i_pixel_pitch = 0;
207 208 209
    }

    p_picture->pf_release = NULL;
Laurent Aimar's avatar
Laurent Aimar committed
210
    p_picture->p_release_sys = NULL;
211 212 213 214 215 216
    p_picture->i_refcount = 0;

    p_picture->i_qtype = QTYPE_NONE;
    p_picture->i_qstride = 0;
    p_picture->p_q = NULL;

217
    video_format_Setup( &p_picture->format, i_chroma, i_width, i_height,
218
                        i_sar_num, i_sar_den );
219

220
    const vlc_chroma_description_t *p_dsc =
221 222 223
        vlc_fourcc_GetChromaDescription( p_picture->format.i_chroma );
    if( !p_dsc )
        return VLC_EGENERIC;
224

225 226 227 228 229 230 231 232 233
    /* We want V (width/height) to respect:
        (V * p_dsc->p[i].w.i_num) % p_dsc->p[i].w.i_den == 0
        (V * p_dsc->p[i].w.i_num/p_dsc->p[i].w.i_den * p_dsc->i_pixel_size) % 16 == 0
       Which is respected if you have
       V % lcm( p_dsc->p[0..planes].w.i_den * 16) == 0
    */
    int i_modulo_w = 1;
    int i_modulo_h = 1;
    int i_ratio_h  = 1;
234
    for( unsigned i = 0; i < p_dsc->plane_count; i++ )
235
    {
236 237 238 239
        i_modulo_w = LCM( i_modulo_w, 16 * p_dsc->p[i].w.den );
        i_modulo_h = LCM( i_modulo_h, 16 * p_dsc->p[i].h.den );
        if( i_ratio_h < p_dsc->p[i].h.den )
            i_ratio_h = p_dsc->p[i].h.den;
240
    }
241

242 243 244
    const int i_width_aligned  = ( i_width  + i_modulo_w - 1 ) / i_modulo_w * i_modulo_w;
    const int i_height_aligned = ( i_height + i_modulo_h - 1 ) / i_modulo_h * i_modulo_h;
    const int i_height_extra   = 2 * i_ratio_h; /* This one is a hack for some ASM functions */
245
    for( unsigned i = 0; i < p_dsc->plane_count; i++ )
246 247
    {
        plane_t *p = &p_picture->p[i];
248

249 250 251 252 253
        p->i_lines         = (i_height_aligned + i_height_extra ) * p_dsc->p[i].h.num / p_dsc->p[i].h.den;
        p->i_visible_lines = i_height * p_dsc->p[i].h.num / p_dsc->p[i].h.den;
        p->i_pitch         = i_width_aligned * p_dsc->p[i].w.num / p_dsc->p[i].w.den * p_dsc->pixel_size;
        p->i_visible_pitch = i_width * p_dsc->p[i].w.num / p_dsc->p[i].w.den * p_dsc->pixel_size;
        p->i_pixel_pitch   = p_dsc->pixel_size;
254

255
        assert( (p->i_pitch % 16) == 0 );
256
    }
257
    p_picture->i_planes  = p_dsc->plane_count;
258 259 260 261

    return VLC_SUCCESS;
}

262 263 264
/*****************************************************************************
 *
 *****************************************************************************/
265
picture_t *picture_NewFromResource( const video_format_t *p_fmt, const picture_resource_t *p_resource )
266
{
267 268
    video_format_t fmt = *p_fmt;

ivoire's avatar
ivoire committed
269
    /* It is needed to be sure all information are filled */
270
    video_format_Setup( &fmt, p_fmt->i_chroma,
271
                              p_fmt->i_width, p_fmt->i_height,
Laurent Aimar's avatar
Laurent Aimar committed
272
                              p_fmt->i_sar_num, p_fmt->i_sar_den );
273 274

    /* */
ivoire's avatar
ivoire committed
275
    picture_t *p_picture = calloc( 1, sizeof(*p_picture) );
276 277 278
    if( !p_picture )
        return NULL;

279
    if( p_resource )
280
    {
281 282
        if( picture_Setup( p_picture, fmt.i_chroma, fmt.i_width, fmt.i_height,
                           fmt.i_sar_num, fmt.i_sar_den ) )
283 284 285 286 287
        {
            free( p_picture );
            return NULL;
        }
        p_picture->p_sys = p_resource->p_sys;
288

289 290 291 292 293 294 295 296 297
        for( int i = 0; i < p_picture->i_planes; i++ )
        {
            p_picture->p[i].p_pixels = p_resource->p[i].p_pixels;
            p_picture->p[i].i_lines  = p_resource->p[i].i_lines;
            p_picture->p[i].i_pitch  = p_resource->p[i].i_pitch;
        }
    }
    else
    {
298
        if( vout_AllocatePicture( p_picture,
299 300
                                  fmt.i_chroma, fmt.i_width, fmt.i_height,
                                  fmt.i_sar_num, fmt.i_sar_den ) )
301 302 303 304 305 306 307
        {
            free( p_picture );
            return NULL;
        }
    }
    /* */
    p_picture->format = fmt;
308 309 310 311 312
    p_picture->i_refcount = 1;
    p_picture->pf_release = PictureReleaseCallback;

    return p_picture;
}
313 314 315 316
picture_t *picture_NewFromFormat( const video_format_t *p_fmt )
{
    return picture_NewFromResource( p_fmt, NULL );
}
317
picture_t *picture_New( vlc_fourcc_t i_chroma, int i_width, int i_height, int i_sar_num, int i_sar_den )
318 319 320 321
{
    video_format_t fmt;

    memset( &fmt, 0, sizeof(fmt) );
322
    video_format_Setup( &fmt, i_chroma, i_width, i_height,
323
                        i_sar_num, i_sar_den );
324 325 326

    return picture_NewFromFormat( &fmt );
}
327 328 329 330 331 332 333

/*****************************************************************************
 *
 *****************************************************************************/
void picture_Delete( picture_t *p_picture )
{
    assert( p_picture && p_picture->i_refcount == 0 );
Laurent Aimar's avatar
Laurent Aimar committed
334
    assert( p_picture->p_release_sys == NULL );
335

336
    free( p_picture->p_q );
337 338 339 340 341 342 343 344 345 346
    free( p_picture->p_data_orig );
    free( p_picture->p_sys );
    free( p_picture );
}

/*****************************************************************************
 *
 *****************************************************************************/
void picture_CopyPixels( picture_t *p_dst, const picture_t *p_src )
{
347 348 349
    int i;

    for( i = 0; i < p_src->i_planes ; i++ )
350 351
        plane_CopyPixels( p_dst->p+i, p_src->p+i );
}
352

353 354 355 356 357 358
void plane_CopyPixels( plane_t *p_dst, const plane_t *p_src )
{
    const unsigned i_width  = __MIN( p_dst->i_visible_pitch,
                                     p_src->i_visible_pitch );
    const unsigned i_height = __MIN( p_dst->i_visible_lines,
                                     p_src->i_visible_lines );
359

360 361 362 363 364 365 366 367 368 369 370 371
    if( p_src->i_pitch == p_dst->i_pitch )
    {
        /* There are margins, but with the same width : perfect ! */
        vlc_memcpy( p_dst->p_pixels, p_src->p_pixels,
                    p_src->i_pitch * i_height );
    }
    else
    {
        /* We need to proceed line by line */
        uint8_t *p_in = p_src->p_pixels;
        uint8_t *p_out = p_dst->p_pixels;
        int i_line;
372

373 374 375 376 377 378 379 380
        assert( p_in );
        assert( p_out );

        for( i_line = i_height; i_line--; )
        {
            vlc_memcpy( p_out, p_in, i_width );
            p_in += p_src->i_pitch;
            p_out += p_dst->i_pitch;
381 382 383
        }
    }
}
Laurent Aimar's avatar
Laurent Aimar committed
384

385 386 387
/*****************************************************************************
 *
 *****************************************************************************/
388 389 390 391 392
int picture_Export( vlc_object_t *p_obj,
                    block_t **pp_image,
                    video_format_t *p_fmt,
                    picture_t *p_picture,
                    vlc_fourcc_t i_format,
393
                    int i_override_width, int i_override_height )
394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409
{
    /* */
    video_format_t fmt_in = p_picture->format;
    if( fmt_in.i_sar_num <= 0 || fmt_in.i_sar_den <= 0 )
    {
        fmt_in.i_sar_num =
        fmt_in.i_sar_den = 1;
    }

    /* */
    video_format_t fmt_out;
    memset( &fmt_out, 0, sizeof(fmt_out) );
    fmt_out.i_sar_num =
    fmt_out.i_sar_den = 1;
    fmt_out.i_chroma  = i_format;

410 411 412 413
    /* compute original width/height */
    unsigned int i_original_width;
    unsigned int i_original_height;
    if( fmt_in.i_sar_num >= fmt_in.i_sar_den )
414
    {
415
        i_original_width = (int64_t)fmt_in.i_width * fmt_in.i_sar_num / fmt_in.i_sar_den;
416
        i_original_height = fmt_in.i_height;
417 418 419
    }
    else
    {
420
        i_original_width =  fmt_in.i_width;
421
        i_original_height = (int64_t)fmt_in.i_height * fmt_in.i_sar_den / fmt_in.i_sar_num;
422 423 424
    }

    /* */
425 426 427 428
    fmt_out.i_width  = ( i_override_width < 0 ) ?
                       i_original_width : i_override_width;
    fmt_out.i_height = ( i_override_height < 0 ) ?
                       i_original_height : i_override_height;
429 430 431 432 433 434 435 436 437 438 439

    /* scale if only one direction is provided */
    if( fmt_out.i_height == 0 && fmt_out.i_width > 0 )
    {
        fmt_out.i_height = fmt_in.i_height * fmt_out.i_width
                     * fmt_in.i_sar_den / fmt_in.i_width / fmt_in.i_sar_num;
    }
    else if( fmt_out.i_width == 0 && fmt_out.i_height > 0 )
    {
        fmt_out.i_width  = fmt_in.i_width * fmt_out.i_height
                     * fmt_in.i_sar_num / fmt_in.i_height / fmt_in.i_sar_den;
440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
    }

    image_handler_t *p_image = image_HandlerCreate( p_obj );

    block_t *p_block = image_Write( p_image, p_picture, &fmt_in, &fmt_out );

    image_HandlerDelete( p_image );

    if( !p_block )
        return VLC_EGENERIC;

    p_block->i_pts =
    p_block->i_dts = p_picture->date;

    if( p_fmt )
        *p_fmt = fmt_out;
    *pp_image = p_block;

    return VLC_SUCCESS;
}