modules/video_chroma/swscale.c

   1 /*****************************************************************************
   2  * swscale.c: scaling and chroma conversion using libswscale
   3  *****************************************************************************
   4  * Copyright (C) 1999-2008 VLC authors and VideoLAN
   5  *
   6  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   7  *          Gildas Bazin <gbazin@videolan.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify it
  10  * under the terms of the GNU Lesser General Public License as published by
  11  * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public License
  20  * along with this program; if not, write to the Free Software Foundation,
  21  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #ifdef HAVE_CONFIG_H
  28 # include "config.h"
  29 #endif
  30 #include <assert.h>
  31
  32 #include <vlc_common.h>
  33 #include <vlc_plugin.h>
  34 #include <vlc_filter.h>
  35 #include <vlc_picture.h>
  36 #include <vlc_cpu.h>
  37
  38 #include <libswscale/swscale.h>
  39 #include <libswscale/version.h>
  40
  41 #ifdef __APPLE__
  42 # include <TargetConditionals.h>
  43 #endif
  44
  45 #include "../codec/avcodec/chroma.h" // Chroma Avutil <-> VLC conversion
  46
  47 /* Gruikkkkkkkkkk!!!!! */
  48 #undef AVPALETTE_SIZE
  49 #define AVPALETTE_SIZE (256 * sizeof(uint32_t))
  50
  51 /*****************************************************************************
  52  * Module descriptor
  53  *****************************************************************************/
  54 static int  OpenScaler( filter_t * );
  55 static void CloseScaler( filter_t * );
  56
  57 #define SCALEMODE_TEXT N_("Scaling mode")
  58 #define SCALEMODE_LONGTEXT N_("Scaling mode to use.")
  59
  60 static const int pi_mode_values[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 };
  61 static const char *const ppsz_mode_descriptions[] =
  62 { N_("Fast bilinear"), N_("Bilinear"), N_("Bicubic (good quality)"),
  63   N_("Experimental"), N_("Nearest neighbor (bad quality)"),
  64   N_("Area"), N_("Luma bicubic / chroma bilinear"), N_("Gauss"),
  65   N_("SincR"), N_("Lanczos"), N_("Bicubic spline") };
  66
  67 vlc_module_begin ()
  68     set_description( N_("Video scaling filter") )
  69     set_shortname( N_("Swscale" ) )
  70     set_category( CAT_VIDEO )
  71     set_subcategory( SUBCAT_VIDEO_VFILTER )
  72     set_callback_video_converter( OpenScaler, 150 )
  73     add_integer( "swscale-mode", 2, SCALEMODE_TEXT, SCALEMODE_LONGTEXT, true )
  74         change_integer_list( pi_mode_values, ppsz_mode_descriptions )
  75 vlc_module_end ()
  76
  77 /* Version checking */
  78 /****************************************************************************
  79  * Local prototypes
  80  ****************************************************************************/
  81
  82 /**
  83  * Internal swscale filter structure.
  84  */
  85 typedef struct
  86 {
  87     SwsFilter *p_filter;
  88     int i_cpu_mask, i_sws_flags;
  89
  90     video_format_t fmt_in;
  91     video_format_t fmt_out;
  92     const vlc_chroma_description_t *desc_in;
  93     const vlc_chroma_description_t *desc_out;
  94
  95     struct SwsContext *ctx;
  96     struct SwsContext *ctxA;
  97     picture_t *p_src_a;
  98     picture_t *p_dst_a;
  99     int i_extend_factor;
 100     picture_t *p_src_e;
 101     picture_t *p_dst_e;
 102     bool b_add_a;
 103     bool b_copy;
 104     bool b_swap_uvi;
 105     bool b_swap_uvo;
 106 } filter_sys_t;
 107
 108 static picture_t *Filter( filter_t *, picture_t * );
 109 static int  Init( filter_t * );
 110 static void Clean( filter_t * );
 111
 112 typedef struct
 113 {
 114     int  i_fmti;
 115     int  i_fmto;
 116     bool b_has_a;
 117     bool b_add_a;
 118     int  i_sws_flags;
 119     bool b_copy;
 120     bool b_swap_uvi;
 121     bool b_swap_uvo;
 122 } ScalerConfiguration;
 123
 124 static int GetParameters( ScalerConfiguration *,
 125                           const video_format_t *p_fmti,
 126                           const video_format_t *p_fmto,
 127                           int i_sws_flags_default );
 128
 129 static int GetSwsCpuMask(void);
 130
 131 /* SwScaler point resize quality seems really bad, let our scale module do it
 132  * (change it to true to try) */
 133 #define ALLOW_YUVP (false)
 134 /* SwScaler does not like too small picture */
 135 #define MINIMUM_WIDTH (32)
 136
 137 /* XXX is it always 3 even for BIG_ENDIAN (blend.c seems to think so) ? */
 138 #define OFFSET_A (3)
 139
 140 static const struct vlc_filter_operations filter_ops = {
 141     .filter_video = Filter, .close = CloseScaler,
 142 };
 143
 144 /*****************************************************************************
 145  * OpenScaler: probe the filter and return score
 146  *****************************************************************************/
 147 static int OpenScaler( filter_t *p_filter )
 148 {
 149     filter_sys_t *p_sys;
 150
 151     int i_sws_mode;
 152
 153     if( GetParameters( NULL,
 154                        &p_filter->fmt_in.video,
 155                        &p_filter->fmt_out.video, 0 ) )
 156         return VLC_EGENERIC;
 157
 158     /* Allocate the memory needed to store the decoder's structure */
 159     if( ( p_filter->p_sys = p_sys = calloc(1, sizeof(filter_sys_t)) ) == NULL )
 160         return VLC_ENOMEM;
 161
 162     /* Set CPU capabilities */
 163     p_sys->i_cpu_mask = GetSwsCpuMask();
 164
 165     /* */
 166     i_sws_mode = var_CreateGetInteger( p_filter, "swscale-mode" );
 167     switch( i_sws_mode )
 168     {
 169     case 0:  p_sys->i_sws_flags = SWS_FAST_BILINEAR; break;
 170     case 1:  p_sys->i_sws_flags = SWS_BILINEAR; break;
 171     case 2:  p_sys->i_sws_flags = SWS_BICUBIC; break;
 172     case 3:  p_sys->i_sws_flags = SWS_X; break;
 173     case 4:  p_sys->i_sws_flags = SWS_POINT; break;
 174     case 5:  p_sys->i_sws_flags = SWS_AREA; break;
 175     case 6:  p_sys->i_sws_flags = SWS_BICUBLIN; break;
 176     case 7:  p_sys->i_sws_flags = SWS_GAUSS; break;
 177     case 8:  p_sys->i_sws_flags = SWS_SINC; break;
 178     case 9:  p_sys->i_sws_flags = SWS_LANCZOS; break;
 179     case 10: p_sys->i_sws_flags = SWS_SPLINE; break;
 180     default: p_sys->i_sws_flags = SWS_BICUBIC; i_sws_mode = 2; break;
 181     }
 182
 183     /* Misc init */
 184     memset( &p_sys->fmt_in,  0, sizeof(p_sys->fmt_in) );
 185     memset( &p_sys->fmt_out, 0, sizeof(p_sys->fmt_out) );
 186
 187     if( Init( p_filter ) )
 188     {
 189         if( p_sys->p_filter )
 190             sws_freeFilter( p_sys->p_filter );
 191         free( p_sys );
 192         return VLC_EGENERIC;
 193     }
 194
 195     /* */
 196     p_filter->ops = &filter_ops;
 197
 198     msg_Dbg( p_filter, "%ix%i (%ix%i) chroma: %4.4s -> %ix%i (%ix%i) chroma: %4.4s with scaling using %s",
 199              p_filter->fmt_in.video.i_visible_width, p_filter->fmt_in.video.i_visible_height,
 200              p_filter->fmt_in.video.i_width, p_filter->fmt_in.video.i_height,
 201              (char *)&p_filter->fmt_in.video.i_chroma,
 202              p_filter->fmt_out.video.i_visible_width, p_filter->fmt_out.video.i_visible_height,
 203              p_filter->fmt_out.video.i_width, p_filter->fmt_out.video.i_height,
 204              (char *)&p_filter->fmt_out.video.i_chroma,
 205              ppsz_mode_descriptions[i_sws_mode] );
 206
 207     return VLC_SUCCESS;
 208 }
 209
 210 /*****************************************************************************
 211  * CloseFilter: clean up the filter
 212  *****************************************************************************/
 213 static void CloseScaler( filter_t *p_filter )
 214 {
 215     filter_sys_t *p_sys = p_filter->p_sys;
 216
 217     Clean( p_filter );
 218     if( p_sys->p_filter )
 219         sws_freeFilter( p_sys->p_filter );
 220     free( p_sys );
 221 }
 222
 223 /*****************************************************************************
 224  * Helpers
 225  *****************************************************************************/
 226 static int GetSwsCpuMask(void)
 227 {
 228     int i_sws_cpu = 0;
 229
 230 #if LIBSWSCALE_VERSION_MAJOR < 4
 231 #if defined(__i386__) || defined(__x86_64__)
 232     if( vlc_CPU_MMX() )
 233         i_sws_cpu |= SWS_CPU_CAPS_MMX;
 234     if( vlc_CPU_MMXEXT() )
 235         i_sws_cpu |= SWS_CPU_CAPS_MMX2;
 236     if( vlc_CPU_3dNOW() )
 237         i_sws_cpu |= SWS_CPU_CAPS_3DNOW;
 238 #elif defined(__ppc__) || defined(__ppc64__) || defined(__powerpc__)
 239     if( vlc_CPU_ALTIVEC() )
 240         i_sws_cpu |= SWS_CPU_CAPS_ALTIVEC;
 241 #endif
 242 #endif
 243
 244     return i_sws_cpu;
 245 }
 246
 247 static void FixParameters( int *pi_fmt, bool *pb_has_a, bool *pb_swap_uv, vlc_fourcc_t fmt )
 248 {
 249     switch( fmt )
 250     {
 251     case VLC_CODEC_YUV422A:
 252         *pi_fmt = AV_PIX_FMT_YUV422P;
 253         *pb_has_a = true;
 254         break;
 255     case VLC_CODEC_YUV420A:
 256         *pi_fmt = AV_PIX_FMT_YUV420P;
 257         *pb_has_a = true;
 258         break;
 259     case VLC_CODEC_YUVA:
 260         *pi_fmt = AV_PIX_FMT_YUV444P;
 261         *pb_has_a = true;
 262         break;
 263     case VLC_CODEC_RGBA:
 264         *pi_fmt = AV_PIX_FMT_BGR32;
 265         *pb_has_a = true;
 266         break;
 267     case VLC_CODEC_ARGB:
 268         *pi_fmt = AV_PIX_FMT_BGR32_1;
 269         *pb_has_a = true;
 270         break;
 271     case VLC_CODEC_BGRA:
 272         *pi_fmt = AV_PIX_FMT_RGB32;
 273         *pb_has_a = true;
 274         break;
 275     case VLC_CODEC_YV12:
 276         *pi_fmt = AV_PIX_FMT_YUV420P;
 277         *pb_swap_uv = true;
 278         break;
 279     case VLC_CODEC_YV9:
 280         *pi_fmt = AV_PIX_FMT_YUV410P;
 281         *pb_swap_uv = true;
 282         break;
 283     default:
 284         break;
 285     }
 286 }
 287
 288 static int GetParameters( ScalerConfiguration *p_cfg,
 289                           const video_format_t *p_fmti,
 290                           const video_format_t *p_fmto,
 291                           int i_sws_flags_default )
 292 {
 293     int i_fmti = -1;
 294     int i_fmto = -1;
 295
 296     bool b_has_ai = false;
 297     bool b_has_ao = false;
 298     int i_sws_flags = i_sws_flags_default;
 299     bool b_swap_uvi = false;
 300     bool b_swap_uvo = false;
 301
 302     GetFfmpegChroma( &i_fmti, p_fmti );
 303     GetFfmpegChroma( &i_fmto, p_fmto );
 304
 305     if( p_fmti->i_chroma == p_fmto->i_chroma )
 306     {
 307         if( p_fmti->i_chroma == VLC_CODEC_YUVP && ALLOW_YUVP )
 308         {
 309             i_fmti = i_fmto = AV_PIX_FMT_GRAY8;
 310             i_sws_flags = SWS_POINT;
 311         }
 312     }
 313
 314     FixParameters( &i_fmti, &b_has_ai, &b_swap_uvi, p_fmti->i_chroma );
 315     FixParameters( &i_fmto, &b_has_ao, &b_swap_uvo, p_fmto->i_chroma );
 316
 317 #if !defined (__ANDROID__) && !defined(TARGET_OS_IPHONE)
 318     /* FIXME TODO removed when ffmpeg is fixed
 319      * Without SWS_ACCURATE_RND the quality is really bad for some conversions */
 320     switch( i_fmto )
 321     {
 322     case AV_PIX_FMT_ARGB:
 323     case AV_PIX_FMT_RGBA:
 324     case AV_PIX_FMT_ABGR:
 325         i_sws_flags |= SWS_ACCURATE_RND;
 326         break;
 327     }
 328 #endif
 329
 330     if( p_cfg )
 331     {
 332         p_cfg->i_fmti = i_fmti;
 333         p_cfg->i_fmto = i_fmto;
 334         p_cfg->b_has_a = b_has_ai && b_has_ao;
 335         p_cfg->b_add_a = (!b_has_ai) && b_has_ao;
 336         p_cfg->b_copy = i_fmti == i_fmto &&
 337                         p_fmti->i_visible_width == p_fmto->i_visible_width &&
 338                         p_fmti->i_visible_height == p_fmto->i_visible_height;
 339         p_cfg->b_swap_uvi = b_swap_uvi;
 340         p_cfg->b_swap_uvo = b_swap_uvo;
 341         p_cfg->i_sws_flags = i_sws_flags;
 342     }
 343
 344     if( i_fmti < 0 || i_fmto < 0 )
 345         return VLC_EGENERIC;
 346
 347     return VLC_SUCCESS;
 348 }
 349
 350 static int Init( filter_t *p_filter )
 351 {
 352     filter_sys_t *p_sys = p_filter->p_sys;
 353     const video_format_t *p_fmti = &p_filter->fmt_in.video;
 354     video_format_t       *p_fmto = &p_filter->fmt_out.video;
 355
 356     if( p_fmti->orientation != p_fmto->orientation )
 357         return VLC_EGENERIC;
 358
 359     if( video_format_IsSimilar( p_fmti, &p_sys->fmt_in ) &&
 360         video_format_IsSimilar( p_fmto, &p_sys->fmt_out ) &&
 361         p_sys->ctx )
 362     {
 363         return VLC_SUCCESS;
 364     }
 365
 366     Clean( p_filter );
 367
 368     /* Init with new parameters */
 369     ScalerConfiguration cfg;
 370     if( GetParameters( &cfg, p_fmti, p_fmto, p_sys->i_sws_flags ) )
 371     {
 372         msg_Err( p_filter, "format not supported '%4.4s' %ux%u -> '%4.4s' %ux%u",
 373                  (const char *)&p_fmti->i_chroma, p_fmti->i_visible_width, p_fmti->i_visible_height,
 374                  (const char *)&p_fmto->i_chroma, p_fmto->i_visible_width, p_fmto->i_visible_height );
 375         return VLC_EGENERIC;
 376     }
 377     if( p_fmti->i_visible_width == 0 || p_fmti->i_visible_height == 0 ||
 378         p_fmto->i_visible_width == 0 || p_fmto->i_visible_height == 0 )
 379     {
 380         msg_Err( p_filter, "invalid scaling: %ix%i -> %ix%i",
 381                  p_fmti->i_visible_width, p_fmti->i_visible_height,
 382                  p_fmto->i_visible_width, p_fmto->i_visible_height);
 383         return VLC_EGENERIC;
 384     }
 385
 386     p_sys->desc_in = vlc_fourcc_GetChromaDescription( p_fmti->i_chroma );
 387     p_sys->desc_out = vlc_fourcc_GetChromaDescription( p_fmto->i_chroma );
 388     if( p_sys->desc_in == NULL || p_sys->desc_out == NULL )
 389         return VLC_EGENERIC;
 390
 391     /* swscale does not like too small width */
 392     p_sys->i_extend_factor = 1;
 393     while( __MIN( p_fmti->i_visible_width, p_fmto->i_visible_width ) * p_sys->i_extend_factor < MINIMUM_WIDTH)
 394         p_sys->i_extend_factor++;
 395
 396     const unsigned i_fmti_visible_width = p_fmti->i_visible_width * p_sys->i_extend_factor;
 397     const unsigned i_fmto_visible_width = p_fmto->i_visible_width * p_sys->i_extend_factor;
 398     for( int n = 0; n < (cfg.b_has_a ? 2 : 1); n++ )
 399     {
 400         const int i_fmti = n == 0 ? cfg.i_fmti : AV_PIX_FMT_GRAY8;
 401         const int i_fmto = n == 0 ? cfg.i_fmto : AV_PIX_FMT_GRAY8;
 402         struct SwsContext *ctx;
 403
 404         ctx = sws_getContext( i_fmti_visible_width, p_fmti->i_visible_height, i_fmti,
 405                               i_fmto_visible_width, p_fmto->i_visible_height, i_fmto,
 406                               cfg.i_sws_flags | p_sys->i_cpu_mask,
 407                               p_sys->p_filter, NULL, 0 );
 408         if( n == 0 )
 409             p_sys->ctx = ctx;
 410         else
 411             p_sys->ctxA = ctx;
 412     }
 413     if( p_sys->ctxA )
 414     {
 415         p_sys->p_src_a = picture_New( VLC_CODEC_GREY, i_fmti_visible_width, p_fmti->i_visible_height, 0, 1 );
 416         p_sys->p_dst_a = picture_New( VLC_CODEC_GREY, i_fmto_visible_width, p_fmto->i_visible_height, 0, 1 );
 417     }
 418     if( p_sys->i_extend_factor != 1 )
 419     {
 420         p_sys->p_src_e = picture_New( p_fmti->i_chroma, i_fmti_visible_width, p_fmti->i_visible_height, 0, 1 );
 421         p_sys->p_dst_e = picture_New( p_fmto->i_chroma, i_fmto_visible_width, p_fmto->i_visible_height, 0, 1 );
 422
 423         if( p_sys->p_src_e )
 424             memset( p_sys->p_src_e->p[0].p_pixels, 0, p_sys->p_src_e->p[0].i_pitch * p_sys->p_src_e->p[0].i_lines );
 425         if( p_sys->p_dst_e )
 426             memset( p_sys->p_dst_e->p[0].p_pixels, 0, p_sys->p_dst_e->p[0].i_pitch * p_sys->p_dst_e->p[0].i_lines );
 427     }
 428
 429     if( !p_sys->ctx ||
 430         ( cfg.b_has_a && ( !p_sys->ctxA || !p_sys->p_src_a || !p_sys->p_dst_a ) ) ||
 431         ( p_sys->i_extend_factor != 1 && ( !p_sys->p_src_e || !p_sys->p_dst_e ) ) )
 432     {
 433         msg_Err( p_filter, "could not init SwScaler and/or allocate memory" );
 434         Clean( p_filter );
 435         return VLC_EGENERIC;
 436     }
 437
 438     if (p_filter->b_allow_fmt_out_change)
 439     {
 440         /*
 441          * If the transformation is not homothetic we must modify the
 442          * aspect ratio of the output format in order to have the
 443          * output picture displayed correctly and not stretched
 444          * horizontally or vertically.
 445          * WARNING: this is a hack, ideally this should not be needed
 446          * and the vout should update its video format instead.
 447          */
 448         unsigned i_sar_num = p_fmti->i_sar_num * p_fmti->i_visible_width;
 449         unsigned i_sar_den = p_fmti->i_sar_den * p_fmto->i_visible_width;
 450         vlc_ureduce(&i_sar_num, &i_sar_den, i_sar_num, i_sar_den, 65536);
 451         i_sar_num *= p_fmto->i_visible_height;
 452         i_sar_den *= p_fmti->i_visible_height;
 453         vlc_ureduce(&i_sar_num, &i_sar_den, i_sar_num, i_sar_den, 65536);
 454         p_fmto->i_sar_num = i_sar_num;
 455         p_fmto->i_sar_den = i_sar_den;
 456     }
 457
 458     p_sys->b_add_a = cfg.b_add_a;
 459     p_sys->b_copy = cfg.b_copy;
 460     p_sys->fmt_in  = *p_fmti;
 461     p_sys->fmt_out = *p_fmto;
 462     p_sys->b_swap_uvi = cfg.b_swap_uvi;
 463     p_sys->b_swap_uvo = cfg.b_swap_uvo;
 464
 465     return VLC_SUCCESS;
 466 }
 467
 468 static void Clean( filter_t *p_filter )
 469 {
 470     filter_sys_t *p_sys = p_filter->p_sys;
 471
 472     if( p_sys->p_src_e )
 473         picture_Release( p_sys->p_src_e );
 474     if( p_sys->p_dst_e )
 475         picture_Release( p_sys->p_dst_e );
 476
 477     if( p_sys->p_src_a )
 478         picture_Release( p_sys->p_src_a );
 479     if( p_sys->p_dst_a )
 480         picture_Release( p_sys->p_dst_a );
 481
 482     if( p_sys->ctxA )
 483         sws_freeContext( p_sys->ctxA );
 484
 485     if( p_sys->ctx )
 486         sws_freeContext( p_sys->ctx );
 487
 488     /* We have to set it to null has we call be called again :( */
 489     p_sys->ctx = NULL;
 490     p_sys->ctxA = NULL;
 491     p_sys->p_src_a = NULL;
 492     p_sys->p_dst_a = NULL;
 493     p_sys->p_src_e = NULL;
 494     p_sys->p_dst_e = NULL;
 495 }
 496
 497 static void GetPixels( uint8_t *pp_pixel[4], int pi_pitch[4],
 498                        const vlc_chroma_description_t *desc,
 499                        const video_format_t *fmt,
 500                        const picture_t *p_picture, unsigned planes,
 501                        bool b_swap_uv )
 502 {
 503     unsigned i = 0;
 504
 505     if( planes > (unsigned)p_picture->i_planes )
 506         planes = p_picture->i_planes;
 507     assert( !b_swap_uv || planes >= 3 );
 508
 509     for( ; i < planes; i++ )
 510     {
 511         const plane_t *p = p_picture->p + i;
 512         if( b_swap_uv && (i == 1 || i== 2) )
 513             p = p_picture->p + 3 - i;
 514
 515         pp_pixel[i] = p->p_pixels
 516             + (((fmt->i_x_offset * desc->p[i].w.num) / desc->p[i].w.den)
 517                 * p->i_pixel_pitch)
 518             + (((fmt->i_y_offset * desc->p[i].h.num) / desc->p[i].h.den)
 519                 * p->i_pitch);
 520         pi_pitch[i] = p->i_pitch;
 521     }
 522
 523     for( ; i < 4; i++ )
 524     {
 525         pp_pixel[i] = NULL;
 526         pi_pitch[i] = 0;
 527     }
 528 }
 529
 530 static void ExtractA( picture_t *p_dst, const picture_t *restrict p_src,
 531                       unsigned offset )
 532 {
 533     plane_t *d = &p_dst->p[0];
 534     const plane_t *s = &p_src->p[0];
 535
 536     for( unsigned y = 0; y < p_dst->format.i_height; y++ )
 537         for( unsigned x = 0; x < p_dst->format.i_width; x++ )
 538             d->p_pixels[y*d->i_pitch+x] = s->p_pixels[y*s->i_pitch+4*x+offset];
 539 }
 540
 541 static void InjectA( picture_t *p_dst, const picture_t *restrict p_src,
 542                      unsigned offset )
 543 {
 544     plane_t *d = &p_dst->p[0];
 545     const plane_t *s = &p_src->p[0];
 546
 547     for( unsigned y = 0; y < p_src->format.i_height; y++ )
 548         for( unsigned x = 0; x < p_src->format.i_width; x++ )
 549             d->p_pixels[y*d->i_pitch+4*x+offset] = s->p_pixels[y*s->i_pitch+x];
 550 }
 551
 552 static void FillA( plane_t *d, unsigned i_offset )
 553 {
 554     for( int y = 0; y < d->i_visible_lines; y++ )
 555         for( int x = 0; x < d->i_visible_pitch; x += d->i_pixel_pitch )
 556             d->p_pixels[y*d->i_pitch+x+i_offset] = 0xff;
 557 }
 558
 559 static void CopyPad( picture_t *p_dst, const picture_t *p_src )
 560 {
 561     picture_Copy( p_dst, p_src );
 562     for( int n = 0; n < p_dst->i_planes; n++ )
 563     {
 564         const plane_t *s = &p_src->p[n];
 565         plane_t *d = &p_dst->p[n];
 566
 567         for( int y = 0; y < s->i_lines && y < d->i_lines; y++ )
 568         {
 569             for( int x = s->i_visible_pitch; x < d->i_visible_pitch; x += s->i_pixel_pitch )
 570                 memcpy( &d->p_pixels[y*d->i_pitch + x], &d->p_pixels[y*d->i_pitch + s->i_visible_pitch - s->i_pixel_pitch], s->i_pixel_pitch );
 571         }
 572     }
 573 }
 574
 575 static void SwapUV( picture_t *p_dst, const picture_t *p_src )
 576 {
 577     picture_t tmp = *p_src;
 578     tmp.p[1] = p_src->p[2];
 579     tmp.p[2] = p_src->p[1];
 580
 581     picture_CopyPixels( p_dst, &tmp );
 582 }
 583
 584 static void Convert( filter_t *p_filter, struct SwsContext *ctx,
 585                      picture_t *p_dst, picture_t *p_src, int i_height,
 586                      int i_plane_count, bool b_swap_uvi, bool b_swap_uvo )
 587 {
 588     filter_sys_t *p_sys = p_filter->p_sys;
 589     uint8_t palette[AVPALETTE_SIZE];
 590     uint8_t *src[4], *dst[4];
 591     const uint8_t *csrc[4];
 592     int src_stride[4], dst_stride[4];
 593
 594     GetPixels( src, src_stride, p_sys->desc_in, &p_filter->fmt_in.video,
 595                p_src, i_plane_count, b_swap_uvi );
 596     if( p_filter->fmt_in.video.i_chroma == VLC_CODEC_RGBP )
 597     {
 598         memset( palette, 0, sizeof(palette) );
 599         if( p_filter->fmt_in.video.p_palette )
 600             memcpy( palette, p_filter->fmt_in.video.p_palette->palette,
 601                     __MIN( sizeof(video_palette_t), AVPALETTE_SIZE ) );
 602         src[1] = palette;
 603         src_stride[1] = 4;
 604     }
 605
 606     GetPixels( dst, dst_stride, p_sys->desc_out, &p_filter->fmt_out.video,
 607                p_dst, i_plane_count, b_swap_uvo );
 608
 609     for (size_t i = 0; i < ARRAY_SIZE(src); i++)
 610         csrc[i] = src[i];
 611
 612 #if LIBSWSCALE_VERSION_INT  >= ((0<<16)+(5<<8)+0)
 613     sws_scale( ctx, csrc, src_stride, 0, i_height,
 614                dst, dst_stride );
 615 #else
 616     sws_scale_ordered( ctx, csrc, src_stride, 0, i_height,
 617                        dst, dst_stride );
 618 #endif
 619 }
 620
 621 /****************************************************************************
 622  * Filter: the whole thing
 623  ****************************************************************************
 624  * This function is called just after the thread is launched.
 625  ****************************************************************************/
 626 static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
 627 {
 628     filter_sys_t *p_sys = p_filter->p_sys;
 629     const video_format_t *p_fmti = &p_filter->fmt_in.video;
 630     const video_format_t *p_fmto = &p_filter->fmt_out.video;
 631     picture_t *p_pic_dst;
 632
 633     /* Check if format properties changed */
 634     if( Init( p_filter ) )
 635     {
 636         picture_Release( p_pic );
 637         return NULL;
 638     }
 639
 640     /* Request output picture */
 641     p_pic_dst = filter_NewPicture( p_filter );
 642     if( !p_pic_dst )
 643     {
 644         picture_Release( p_pic );
 645         return NULL;
 646     }
 647
 648     /* */
 649     picture_t *p_src = p_pic;
 650     picture_t *p_dst = p_pic_dst;
 651     if( p_sys->i_extend_factor != 1 )
 652     {
 653         p_src = p_sys->p_src_e;
 654         p_dst = p_sys->p_dst_e;
 655
 656         CopyPad( p_src, p_pic );
 657     }
 658
 659     if( p_sys->b_copy && p_sys->b_swap_uvi == p_sys->b_swap_uvo )
 660         picture_CopyPixels( p_dst, p_src );
 661     else if( p_sys->b_copy )
 662         SwapUV( p_dst, p_src );
 663     else
 664     {
 665         /* Even if alpha is unused, swscale expects the pointer to be set */
 666         const int n_planes = !p_sys->ctxA && (p_src->i_planes == 4 ||
 667                              p_dst->i_planes == 4) ? 4 : 3;
 668         Convert( p_filter, p_sys->ctx, p_dst, p_src, p_fmti->i_visible_height,
 669                  n_planes, p_sys->b_swap_uvi, p_sys->b_swap_uvo );
 670     }
 671     if( p_sys->ctxA )
 672     {
 673         /* We extract the A plane to rescale it, and then we reinject it. */
 674         if( p_fmti->i_chroma == VLC_CODEC_RGBA || p_fmti->i_chroma == VLC_CODEC_BGRA )
 675             ExtractA( p_sys->p_src_a, p_src, OFFSET_A );
 676         else if( p_fmti->i_chroma == VLC_CODEC_ARGB )
 677             ExtractA( p_sys->p_src_a, p_src, 0 );
 678         else
 679             plane_CopyPixels( p_sys->p_src_a->p, p_src->p+A_PLANE );
 680
 681         Convert( p_filter, p_sys->ctxA, p_sys->p_dst_a, p_sys->p_src_a,
 682                  p_fmti->i_visible_height, 1, false, false );
 683         if( p_fmto->i_chroma == VLC_CODEC_RGBA || p_fmto->i_chroma == VLC_CODEC_BGRA )
 684             InjectA( p_dst, p_sys->p_dst_a, OFFSET_A );
 685         else if( p_fmto->i_chroma == VLC_CODEC_ARGB )
 686             InjectA( p_dst, p_sys->p_dst_a, 0 );
 687         else
 688             plane_CopyPixels( p_dst->p+A_PLANE, p_sys->p_dst_a->p );
 689     }
 690     else if( p_sys->b_add_a )
 691     {
 692         /* We inject a complete opaque alpha plane */
 693         if( p_fmto->i_chroma == VLC_CODEC_RGBA || p_fmto->i_chroma == VLC_CODEC_BGRA )
 694             FillA( &p_dst->p[0], OFFSET_A );
 695         else if( p_fmto->i_chroma == VLC_CODEC_ARGB )
 696             FillA( &p_dst->p[0], 0 );
 697         else
 698             FillA( &p_dst->p[A_PLANE], 0 );
 699     }
 700
 701     if( p_sys->i_extend_factor != 1 )
 702     {
 703         picture_CopyPixels( p_pic_dst, p_dst );
 704     }
 705
 706     picture_CopyProperties( p_pic_dst, p_pic );
 707     picture_Release( p_pic );
 708     return p_pic_dst;
 709 }