2009-11-12 01:50:21 +02:00
/*
2010-11-28 12:22:58 +02:00
* Copyright ( c ) 2007 Bobby Bingham
2009-11-12 01:50:21 +02:00
*
* This file is part of FFmpeg .
*
* FFmpeg 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 .
*
* FFmpeg 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 FFmpeg ; if not , write to the Free Software
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
*/
/**
2010-04-20 17:45:34 +03:00
* @ file
2009-11-12 01:50:21 +02:00
* scale video filter
*/
2012-08-06 16:49:32 +03:00
# include <stdio.h>
# include <string.h>
2009-11-12 01:50:21 +02:00
# include "avfilter.h"
2012-05-30 11:12:55 +03:00
# include "formats.h"
2012-06-12 21:12:42 +03:00
# include "internal.h"
2012-05-30 12:20:32 +03:00
# include "video.h"
2011-03-27 23:58:08 +03:00
# include "libavutil/avstring.h"
# include "libavutil/eval.h"
2012-08-06 16:49:32 +03:00
# include "libavutil/internal.h"
2011-06-04 14:58:23 +03:00
# include "libavutil/mathematics.h"
2012-04-09 07:05:50 +03:00
# include "libavutil/opt.h"
2012-10-29 13:48:53 +03:00
# include "libavutil/parseutils.h"
2010-01-31 18:33:29 +02:00
# include "libavutil/pixdesc.h"
2012-04-08 16:11:13 +03:00
# include "libavutil/imgutils.h"
2011-03-28 01:09:58 +03:00
# include "libavutil/avassert.h"
2009-11-12 01:50:21 +02:00
# include "libswscale/swscale.h"
2012-02-20 11:42:33 +03:00
static const char * const var_names [ ] = {
2011-03-27 23:58:08 +03:00
" in_w " , " iw " ,
" in_h " , " ih " ,
" out_w " , " ow " ,
" out_h " , " oh " ,
2011-07-27 11:58:24 +03:00
" a " ,
2011-07-02 18:27:31 +03:00
" sar " ,
2011-07-27 11:58:24 +03:00
" dar " ,
2011-03-27 23:58:08 +03:00
" hsub " ,
" vsub " ,
2013-11-04 16:18:09 +03:00
" ohsub " ,
" ovsub " ,
2011-03-27 23:58:08 +03:00
NULL
} ;
enum var_name {
VAR_IN_W , VAR_IW ,
VAR_IN_H , VAR_IH ,
VAR_OUT_W , VAR_OW ,
VAR_OUT_H , VAR_OH ,
2011-07-27 11:58:24 +03:00
VAR_A ,
2011-07-02 18:27:31 +03:00
VAR_SAR ,
2011-07-27 11:58:24 +03:00
VAR_DAR ,
2011-03-27 23:58:08 +03:00
VAR_HSUB ,
VAR_VSUB ,
2013-11-04 16:18:09 +03:00
VAR_OHSUB ,
VAR_OVSUB ,
2011-03-27 23:58:08 +03:00
VARS_NB
} ;
2014-04-11 12:54:15 +03:00
typedef struct ScaleContext {
2012-10-06 13:30:26 +03:00
const AVClass * class ;
2009-11-12 01:50:21 +02:00
struct SwsContext * sws ; ///< software scaler context
2011-03-28 01:09:58 +03:00
struct SwsContext * isws [ 2 ] ; ///< software scaler context for interlaced material
2013-08-15 23:42:56 +03:00
AVDictionary * opts ;
2009-11-12 01:50:21 +02:00
/**
* New dimensions . Special values are :
* 0 = original width / height
* - 1 = keep original aspect
2014-01-25 17:02:15 +03:00
* - N = try to keep aspect but make sure it is divisible by N
2009-11-12 01:50:21 +02:00
*/
int w , h ;
2012-10-29 13:48:53 +03:00
char * size_str ;
2010-05-07 12:20:45 +03:00
unsigned int flags ; ///sws flags
2009-11-12 01:50:21 +02:00
int hsub , vsub ; ///< chroma subsampling
int slice_y ; ///< top of current output slice
int input_is_pal ; ///< set to 1 if the input format is paletted
2012-04-08 15:59:53 +03:00
int output_is_pal ; ///< set to 1 if the output format is paletted
2011-03-28 01:09:58 +03:00
int interlaced ;
2011-03-27 23:58:08 +03:00
2012-10-06 13:30:26 +03:00
char * w_expr ; ///< width expression string
char * h_expr ; ///< height expression string
2013-02-25 23:21:29 +03:00
char * flags_str ;
2013-07-20 23:11:50 +03:00
char * in_color_matrix ;
char * out_color_matrix ;
2013-07-20 23:12:42 +03:00
int in_range ;
int out_range ;
2013-07-14 06:03:06 +03:00
int out_h_chr_pos ;
int out_v_chr_pos ;
2013-07-24 19:33:58 +03:00
int in_h_chr_pos ;
int in_v_chr_pos ;
2013-08-04 02:30:00 +03:00
int force_original_aspect_ratio ;
2009-11-12 01:50:21 +02:00
} ScaleContext ;
2013-08-15 23:42:56 +03:00
static av_cold int init_dict ( AVFilterContext * ctx , AVDictionary * * opts )
2009-11-12 01:50:21 +02:00
{
ScaleContext * scale = ctx - > priv ;
2012-10-06 13:30:26 +03:00
int ret ;
2012-10-29 13:48:53 +03:00
if ( scale - > size_str & & ( scale - > w_expr | | scale - > h_expr ) ) {
av_log ( ctx , AV_LOG_ERROR ,
" Size and width/height expressions cannot be set at the same time. \n " ) ;
return AVERROR ( EINVAL ) ;
}
2013-04-12 03:34:40 +03:00
if ( scale - > w_expr & & ! scale - > h_expr )
FFSWAP ( char * , scale - > w_expr , scale - > size_str ) ;
2012-10-29 13:48:53 +03:00
if ( scale - > size_str ) {
char buf [ 32 ] ;
if ( ( ret = av_parse_video_size ( & scale - > w , & scale - > h , scale - > size_str ) ) < 0 ) {
av_log ( ctx , AV_LOG_ERROR ,
" Invalid size '%s' \n " , scale - > size_str ) ;
return ret ;
}
snprintf ( buf , sizeof ( buf ) - 1 , " %d " , scale - > w ) ;
av_opt_set ( scale , " w " , buf , 0 ) ;
snprintf ( buf , sizeof ( buf ) - 1 , " %d " , scale - > h ) ;
av_opt_set ( scale , " h " , buf , 0 ) ;
}
if ( ! scale - > w_expr )
av_opt_set ( scale , " w " , " iw " , 0 ) ;
if ( ! scale - > h_expr )
av_opt_set ( scale , " h " , " ih " , 0 ) ;
2012-10-31 17:45:28 +03:00
av_log ( ctx , AV_LOG_VERBOSE , " w:%s h:%s flags:'%s' interl:%d \n " ,
scale - > w_expr , scale - > h_expr , ( char * ) av_x_if_null ( scale - > flags_str , " " ) , scale - > interlaced ) ;
2011-03-27 23:58:08 +03:00
2013-10-06 13:33:32 +03:00
scale - > flags = 0 ;
2013-04-12 03:34:40 +03:00
2012-10-06 13:30:26 +03:00
if ( scale - > flags_str ) {
const AVClass * class = sws_get_class ( ) ;
const AVOption * o = av_opt_find ( & class , " sws_flags " , NULL , 0 ,
AV_OPT_SEARCH_FAKE_OBJ ) ;
int ret = av_opt_eval_flags ( & class , o , scale - > flags_str , & scale - > flags ) ;
if ( ret < 0 )
return ret ;
2010-05-07 12:20:45 +03:00
}
2013-08-15 23:42:56 +03:00
scale - > opts = * opts ;
* opts = NULL ;
2009-11-12 01:50:21 +02:00
return 0 ;
}
static av_cold void uninit ( AVFilterContext * ctx )
{
ScaleContext * scale = ctx - > priv ;
sws_freeContext ( scale - > sws ) ;
2011-03-28 01:09:58 +03:00
sws_freeContext ( scale - > isws [ 0 ] ) ;
sws_freeContext ( scale - > isws [ 1 ] ) ;
2009-11-12 01:50:21 +02:00
scale - > sws = NULL ;
2013-08-15 23:42:56 +03:00
av_dict_free ( & scale - > opts ) ;
2009-11-12 01:50:21 +02:00
}
static int query_formats ( AVFilterContext * ctx )
{
AVFilterFormats * formats ;
2012-10-06 13:10:34 +03:00
enum AVPixelFormat pix_fmt ;
2010-01-09 01:48:32 +02:00
int ret ;
2009-11-12 01:50:21 +02:00
if ( ctx - > inputs [ 0 ] ) {
2014-05-24 12:15:15 +03:00
const AVPixFmtDescriptor * desc = NULL ;
2010-01-09 01:48:32 +02:00
formats = NULL ;
2014-05-24 12:15:15 +03:00
while ( ( desc = av_pix_fmt_desc_next ( desc ) ) ) {
pix_fmt = av_pix_fmt_desc_get_id ( desc ) ;
2013-04-26 15:49:43 +03:00
if ( ( sws_isSupportedInput ( pix_fmt ) | |
sws_isSupportedEndiannessConversion ( pix_fmt ) )
2012-05-30 11:12:55 +03:00
& & ( ret = ff_add_format ( & formats , pix_fmt ) ) < 0 ) {
ff_formats_unref ( & formats ) ;
2010-01-09 01:48:32 +02:00
return ret ;
}
2014-05-24 12:15:15 +03:00
}
2012-05-30 11:12:55 +03:00
ff_formats_ref ( formats , & ctx - > inputs [ 0 ] - > out_formats ) ;
2009-11-12 01:50:21 +02:00
}
if ( ctx - > outputs [ 0 ] ) {
2014-05-24 12:15:15 +03:00
const AVPixFmtDescriptor * desc = NULL ;
2010-01-09 01:48:32 +02:00
formats = NULL ;
2014-05-24 12:15:15 +03:00
while ( ( desc = av_pix_fmt_desc_next ( desc ) ) ) {
pix_fmt = av_pix_fmt_desc_get_id ( desc ) ;
2013-05-07 12:23:55 +03:00
if ( ( sws_isSupportedOutput ( pix_fmt ) | | pix_fmt = = AV_PIX_FMT_PAL8 | |
2013-04-26 15:49:43 +03:00
sws_isSupportedEndiannessConversion ( pix_fmt ) )
2012-05-30 11:12:55 +03:00
& & ( ret = ff_add_format ( & formats , pix_fmt ) ) < 0 ) {
ff_formats_unref ( & formats ) ;
2010-01-09 01:48:32 +02:00
return ret ;
}
2014-05-24 12:15:15 +03:00
}
2012-05-30 11:12:55 +03:00
ff_formats_ref ( formats , & ctx - > outputs [ 0 ] - > in_formats ) ;
2009-11-12 01:50:21 +02:00
}
return 0 ;
}
2013-07-20 23:11:50 +03:00
static const int * parse_yuv_type ( const char * s , enum AVColorSpace colorspace )
{
if ( ! s )
s = " bt601 " ;
if ( s & & strstr ( s , " bt709 " ) ) {
colorspace = AVCOL_SPC_BT709 ;
} else if ( s & & strstr ( s , " fcc " ) ) {
colorspace = AVCOL_SPC_FCC ;
} else if ( s & & strstr ( s , " smpte240m " ) ) {
colorspace = AVCOL_SPC_SMPTE240M ;
} else if ( s & & ( strstr ( s , " bt601 " ) | | strstr ( s , " bt470 " ) | | strstr ( s , " smpte170m " ) ) ) {
colorspace = AVCOL_SPC_BT470BG ;
}
if ( colorspace < 1 | | colorspace > 7 ) {
colorspace = AVCOL_SPC_BT470BG ;
}
2013-07-27 00:03:58 +03:00
return sws_getCoefficients ( colorspace ) ;
2013-07-20 23:11:50 +03:00
}
2009-11-12 01:50:21 +02:00
static int config_props ( AVFilterLink * outlink )
{
AVFilterContext * ctx = outlink - > src ;
AVFilterLink * inlink = outlink - > src - > inputs [ 0 ] ;
2012-10-08 21:54:00 +03:00
enum AVPixelFormat outfmt = outlink - > format ;
2009-11-12 01:50:21 +02:00
ScaleContext * scale = ctx - > priv ;
2012-10-06 14:29:37 +03:00
const AVPixFmtDescriptor * desc = av_pix_fmt_desc_get ( inlink - > format ) ;
2013-11-04 16:18:09 +03:00
const AVPixFmtDescriptor * out_desc = av_pix_fmt_desc_get ( outlink - > format ) ;
2009-11-12 01:50:21 +02:00
int64_t w , h ;
2011-03-27 23:58:08 +03:00
double var_values [ VARS_NB ] , res ;
char * expr ;
int ret ;
2014-01-25 17:00:02 +03:00
int factor_w , factor_h ;
2011-03-27 23:58:08 +03:00
var_values [ VAR_IN_W ] = var_values [ VAR_IW ] = inlink - > w ;
var_values [ VAR_IN_H ] = var_values [ VAR_IH ] = inlink - > h ;
var_values [ VAR_OUT_W ] = var_values [ VAR_OW ] = NAN ;
var_values [ VAR_OUT_H ] = var_values [ VAR_OH ] = NAN ;
2012-10-06 14:23:05 +03:00
var_values [ VAR_A ] = ( double ) inlink - > w / inlink - > h ;
2011-07-02 18:27:31 +03:00
var_values [ VAR_SAR ] = inlink - > sample_aspect_ratio . num ?
2012-10-05 15:45:30 +03:00
( double ) inlink - > sample_aspect_ratio . num / inlink - > sample_aspect_ratio . den : 1 ;
2011-07-27 11:58:24 +03:00
var_values [ VAR_DAR ] = var_values [ VAR_A ] * var_values [ VAR_SAR ] ;
2012-10-06 14:29:37 +03:00
var_values [ VAR_HSUB ] = 1 < < desc - > log2_chroma_w ;
var_values [ VAR_VSUB ] = 1 < < desc - > log2_chroma_h ;
2013-11-04 16:18:09 +03:00
var_values [ VAR_OHSUB ] = 1 < < out_desc - > log2_chroma_w ;
var_values [ VAR_OVSUB ] = 1 < < out_desc - > log2_chroma_h ;
2011-03-27 23:58:08 +03:00
/* evaluate width and height */
av_expr_parse_and_eval ( & res , ( expr = scale - > w_expr ) ,
var_names , var_values ,
NULL , NULL , NULL , NULL , NULL , 0 , ctx ) ;
2011-04-17 18:23:48 +03:00
scale - > w = var_values [ VAR_OUT_W ] = var_values [ VAR_OW ] = res ;
2011-03-27 23:58:08 +03:00
if ( ( ret = av_expr_parse_and_eval ( & res , ( expr = scale - > h_expr ) ,
var_names , var_values ,
NULL , NULL , NULL , NULL , NULL , 0 , ctx ) ) < 0 )
goto fail ;
2011-04-17 18:23:48 +03:00
scale - > h = var_values [ VAR_OUT_H ] = var_values [ VAR_OH ] = res ;
2011-03-27 23:58:08 +03:00
/* evaluate again the width, as it may depend on the output height */
if ( ( ret = av_expr_parse_and_eval ( & res , ( expr = scale - > w_expr ) ,
var_names , var_values ,
NULL , NULL , NULL , NULL , NULL , 0 , ctx ) ) < 0 )
goto fail ;
scale - > w = res ;
w = scale - > w ;
h = scale - > h ;
2014-01-25 16:40:48 +03:00
/* Check if it is requested that the result has to be divisible by a some
2014-01-25 18:54:24 +03:00
* factor ( w or h = - n with n being the factor ) . */
2014-01-25 17:00:02 +03:00
factor_w = 1 ;
factor_h = 1 ;
2014-01-25 16:40:48 +03:00
if ( w < - 1 ) {
factor_w = - w ;
2011-03-27 23:58:08 +03:00
}
2014-01-25 16:40:48 +03:00
if ( h < - 1 ) {
factor_h = - h ;
}
2014-01-25 18:54:24 +03:00
if ( w < 0 & & h < 0 )
2011-03-27 23:58:08 +03:00
scale - > w = scale - > h = 0 ;
2009-11-12 01:50:21 +02:00
if ( ! ( w = scale - > w ) )
w = inlink - > w ;
if ( ! ( h = scale - > h ) )
h = inlink - > h ;
2014-01-25 16:40:48 +03:00
/* Make sure that the result is divisible by the factor we determined
* earlier . If no factor was set , it is nothing will happen as the default
* factor is 1 */
2014-01-25 18:54:24 +03:00
if ( w < 0 )
2014-01-25 18:48:13 +03:00
w = av_rescale ( h , inlink - > w , inlink - > h * factor_w ) * factor_w ;
2014-01-25 18:54:24 +03:00
if ( h < 0 )
2014-01-25 18:48:13 +03:00
h = av_rescale ( w , inlink - > h , inlink - > w * factor_h ) * factor_h ;
2014-01-25 16:40:48 +03:00
/* Note that force_original_aspect_ratio may overwrite the previous set
* dimensions so that it is not divisible by the set factors anymore . */
2013-08-04 02:30:00 +03:00
if ( scale - > force_original_aspect_ratio ) {
int tmp_w = av_rescale ( h , inlink - > w , inlink - > h ) ;
int tmp_h = av_rescale ( w , inlink - > h , inlink - > w ) ;
if ( scale - > force_original_aspect_ratio = = 1 ) {
w = FFMIN ( tmp_w , w ) ;
h = FFMIN ( tmp_h , h ) ;
} else {
w = FFMAX ( tmp_w , w ) ;
h = FFMAX ( tmp_h , h ) ;
}
}
2009-11-12 01:50:21 +02:00
if ( w > INT_MAX | | h > INT_MAX | |
( h * inlink - > w ) > INT_MAX | |
( w * inlink - > h ) > INT_MAX )
av_log ( ctx , AV_LOG_ERROR , " Rescaled value for width or height is too big. \n " ) ;
outlink - > w = w ;
outlink - > h = h ;
/* TODO: make algorithm configurable */
2013-05-12 16:41:49 +03:00
scale - > input_is_pal = desc - > flags & AV_PIX_FMT_FLAG_PAL | |
desc - > flags & AV_PIX_FMT_FLAG_PSEUDOPAL ;
2012-10-08 21:54:00 +03:00
if ( outfmt = = AV_PIX_FMT_PAL8 ) outfmt = AV_PIX_FMT_BGR8 ;
2013-05-15 11:50:20 +03:00
scale - > output_is_pal = av_pix_fmt_desc_get ( outfmt ) - > flags & AV_PIX_FMT_FLAG_PAL | |
av_pix_fmt_desc_get ( outfmt ) - > flags & AV_PIX_FMT_FLAG_PSEUDOPAL ;
2009-11-12 01:50:21 +02:00
2011-05-02 13:35:39 +03:00
if ( scale - > sws )
2011-03-09 04:30:24 +02:00
sws_freeContext ( scale - > sws ) ;
2013-07-14 06:19:46 +03:00
if ( scale - > isws [ 0 ] )
sws_freeContext ( scale - > isws [ 0 ] ) ;
if ( scale - > isws [ 1 ] )
sws_freeContext ( scale - > isws [ 1 ] ) ;
scale - > isws [ 0 ] = scale - > isws [ 1 ] = scale - > sws = NULL ;
2012-03-29 08:02:27 +03:00
if ( inlink - > w = = outlink - > w & & inlink - > h = = outlink - > h & &
inlink - > format = = outlink - > format )
2013-07-14 06:19:46 +03:00
;
2012-03-29 08:02:27 +03:00
else {
2013-07-14 06:19:46 +03:00
struct SwsContext * * swscs [ 3 ] = { & scale - > sws , & scale - > isws [ 0 ] , & scale - > isws [ 1 ] } ;
int i ;
for ( i = 0 ; i < 3 ; i + + ) {
struct SwsContext * * s = swscs [ i ] ;
* s = sws_alloc_context ( ) ;
if ( ! * s )
return AVERROR ( ENOMEM ) ;
2013-08-15 23:42:56 +03:00
if ( scale - > opts ) {
AVDictionaryEntry * e = NULL ;
while ( ( e = av_dict_get ( scale - > opts , " " , e , AV_DICT_IGNORE_SUFFIX ) ) ) {
2013-08-16 05:25:39 +03:00
if ( ( ret = av_opt_set ( * s , e - > key , e - > value , 0 ) ) < 0 )
2013-08-15 23:42:56 +03:00
return ret ;
}
}
2013-07-14 06:19:46 +03:00
av_opt_set_int ( * s , " srcw " , inlink - > w , 0 ) ;
av_opt_set_int ( * s , " srch " , inlink - > h > > ! ! i , 0 ) ;
av_opt_set_int ( * s , " src_format " , inlink - > format , 0 ) ;
av_opt_set_int ( * s , " dstw " , outlink - > w , 0 ) ;
av_opt_set_int ( * s , " dsth " , outlink - > h > > ! ! i , 0 ) ;
av_opt_set_int ( * s , " dst_format " , outfmt , 0 ) ;
av_opt_set_int ( * s , " sws_flags " , scale - > flags , 0 ) ;
2013-07-24 19:33:58 +03:00
av_opt_set_int ( * s , " src_h_chr_pos " , scale - > in_h_chr_pos , 0 ) ;
av_opt_set_int ( * s , " src_v_chr_pos " , scale - > in_v_chr_pos , 0 ) ;
2013-07-14 06:03:06 +03:00
av_opt_set_int ( * s , " dst_h_chr_pos " , scale - > out_h_chr_pos , 0 ) ;
av_opt_set_int ( * s , " dst_v_chr_pos " , scale - > out_v_chr_pos , 0 ) ;
2013-07-14 06:19:46 +03:00
if ( ( ret = sws_init_context ( * s , NULL , NULL ) ) < 0 )
return ret ;
2013-07-18 12:03:56 +03:00
if ( ! scale - > interlaced )
break ;
2013-07-14 06:19:46 +03:00
}
2012-03-29 08:02:27 +03:00
}
2010-05-08 01:12:13 +03:00
2011-08-27 02:49:55 +03:00
if ( inlink - > sample_aspect_ratio . num ) {
outlink - > sample_aspect_ratio = av_mul_q ( ( AVRational ) { outlink - > h * inlink - > w , outlink - > w * inlink - > h } , inlink - > sample_aspect_ratio ) ;
} else
outlink - > sample_aspect_ratio = inlink - > sample_aspect_ratio ;
2012-07-14 19:26:04 +03:00
av_log ( ctx , AV_LOG_VERBOSE , " w:%d h:%d fmt:%s sar:%d/%d -> w:%d h:%d fmt:%s sar:%d/%d flags:0x%0x \n " ,
2012-10-12 18:04:58 +03:00
inlink - > w , inlink - > h , av_get_pix_fmt_name ( inlink - > format ) ,
2012-03-17 14:28:35 +03:00
inlink - > sample_aspect_ratio . num , inlink - > sample_aspect_ratio . den ,
2012-10-12 18:04:58 +03:00
outlink - > w , outlink - > h , av_get_pix_fmt_name ( outlink - > format ) ,
2012-03-17 14:28:35 +03:00
outlink - > sample_aspect_ratio . num , outlink - > sample_aspect_ratio . den ,
scale - > flags ) ;
2011-01-16 22:11:46 +02:00
return 0 ;
2011-03-27 23:58:08 +03:00
fail :
av_log ( NULL , AV_LOG_ERROR ,
2011-12-05 01:33:40 +03:00
" Error when evaluating the expression '%s'. \n "
" Maybe the expression for out_w:'%s' or for out_h:'%s' is self-referencing. \n " ,
expr , scale - > w_expr , scale - > h_expr ) ;
2011-03-27 23:58:08 +03:00
return ret ;
2009-11-12 01:50:21 +02:00
}
2013-03-10 03:30:30 +03:00
static int scale_slice ( AVFilterLink * link , AVFrame * out_buf , AVFrame * cur_pic , struct SwsContext * sws , int y , int h , int mul , int field )
2009-11-12 01:50:21 +02:00
{
ScaleContext * scale = link - > dst - > priv ;
2011-06-01 21:26:54 +03:00
const uint8_t * in [ 4 ] ;
uint8_t * out [ 4 ] ;
2011-03-28 01:09:58 +03:00
int in_stride [ 4 ] , out_stride [ 4 ] ;
int i ;
for ( i = 0 ; i < 4 ; i + + ) {
int vsub = ( ( i + 1 ) & 2 ) ? scale - > vsub : 0 ;
in_stride [ i ] = cur_pic - > linesize [ i ] * mul ;
out_stride [ i ] = out_buf - > linesize [ i ] * mul ;
in [ i ] = cur_pic - > data [ i ] + ( ( y > > vsub ) + field ) * cur_pic - > linesize [ i ] ;
out [ i ] = out_buf - > data [ i ] + field * out_buf - > linesize [ i ] ;
}
2012-04-08 15:59:53 +03:00
if ( scale - > input_is_pal )
2011-03-28 01:09:58 +03:00
in [ 1 ] = cur_pic - > data [ 1 ] ;
2012-04-08 15:59:53 +03:00
if ( scale - > output_is_pal )
2011-03-28 01:09:58 +03:00
out [ 1 ] = out_buf - > data [ 1 ] ;
return sws_scale ( sws , in , in_stride , y / mul , h ,
out , out_stride ) ;
}
2012-11-28 10:41:07 +03:00
static int filter_frame ( AVFilterLink * link , AVFrame * in )
2009-11-12 01:50:21 +02:00
{
ScaleContext * scale = link - > dst - > priv ;
2012-11-29 03:19:35 +03:00
AVFilterLink * outlink = link - > dst - > outputs [ 0 ] ;
2012-11-28 10:41:07 +03:00
AVFrame * out ;
2012-11-29 03:19:35 +03:00
const AVPixFmtDescriptor * desc = av_pix_fmt_desc_get ( link - > format ) ;
char buf [ 32 ] ;
2013-07-25 02:27:00 +03:00
int in_range ;
2012-11-29 03:19:35 +03:00
2013-03-10 03:30:30 +03:00
if ( in - > width ! = link - > w
| | in - > height ! = link - > h
| | in - > format ! = link - > format ) {
2012-11-29 03:19:35 +03:00
int ret ;
snprintf ( buf , sizeof ( buf ) - 1 , " %d " , outlink - > w ) ;
av_opt_set ( scale , " w " , buf , 0 ) ;
snprintf ( buf , sizeof ( buf ) - 1 , " %d " , outlink - > h ) ;
av_opt_set ( scale , " h " , buf , 0 ) ;
2009-11-12 01:50:21 +02:00
2012-11-29 03:19:35 +03:00
link - > dst - > inputs [ 0 ] - > format = in - > format ;
2013-03-10 03:30:30 +03:00
link - > dst - > inputs [ 0 ] - > w = in - > width ;
link - > dst - > inputs [ 0 ] - > h = in - > height ;
2012-11-29 03:19:35 +03:00
if ( ( ret = config_props ( outlink ) ) < 0 )
return ret ;
2012-03-29 08:02:27 +03:00
}
2012-11-29 03:19:35 +03:00
if ( ! scale - > sws )
return ff_filter_frame ( outlink , in ) ;
scale - > hsub = desc - > log2_chroma_w ;
scale - > vsub = desc - > log2_chroma_h ;
2012-11-28 10:41:07 +03:00
out = ff_get_video_buffer ( outlink , outlink - > w , outlink - > h ) ;
2012-11-29 03:19:35 +03:00
if ( ! out ) {
2012-11-28 10:41:07 +03:00
av_frame_free ( & in ) ;
2012-11-29 03:19:35 +03:00
return AVERROR ( ENOMEM ) ;
}
2012-11-28 10:41:07 +03:00
av_frame_copy_props ( out , in ) ;
out - > width = outlink - > w ;
out - > height = outlink - > h ;
2012-11-29 03:19:35 +03:00
if ( scale - > output_is_pal )
avpriv_set_systematic_pal2 ( ( uint32_t * ) out - > data [ 1 ] , outlink - > format = = AV_PIX_FMT_PAL8 ? AV_PIX_FMT_BGR8 : outlink - > format ) ;
2013-07-25 02:27:00 +03:00
in_range = av_frame_get_color_range ( in ) ;
2013-07-20 23:11:50 +03:00
if ( scale - > in_color_matrix
| | scale - > out_color_matrix
| | scale - > in_range ! = AVCOL_RANGE_UNSPECIFIED
2013-07-25 02:27:00 +03:00
| | in_range ! = AVCOL_RANGE_UNSPECIFIED
2013-07-20 23:11:50 +03:00
| | scale - > out_range ! = AVCOL_RANGE_UNSPECIFIED ) {
int in_full , out_full , brightness , contrast , saturation ;
const int * inv_table , * table ;
sws_getColorspaceDetails ( scale - > sws , ( int * * ) & inv_table , & in_full ,
( int * * ) & table , & out_full ,
& brightness , & contrast , & saturation ) ;
if ( scale - > in_color_matrix )
inv_table = parse_yuv_type ( scale - > in_color_matrix , av_frame_get_colorspace ( in ) ) ;
if ( scale - > out_color_matrix )
table = parse_yuv_type ( scale - > out_color_matrix , AVCOL_SPC_UNSPECIFIED ) ;
2013-07-20 23:12:42 +03:00
if ( scale - > in_range ! = AVCOL_RANGE_UNSPECIFIED )
in_full = ( scale - > in_range = = AVCOL_RANGE_JPEG ) ;
2013-07-25 02:27:00 +03:00
else if ( in_range ! = AVCOL_RANGE_UNSPECIFIED )
in_full = ( in_range = = AVCOL_RANGE_JPEG ) ;
2013-07-20 23:12:42 +03:00
if ( scale - > out_range ! = AVCOL_RANGE_UNSPECIFIED )
out_full = ( scale - > out_range = = AVCOL_RANGE_JPEG ) ;
2013-07-20 23:11:50 +03:00
sws_setColorspaceDetails ( scale - > sws , inv_table , in_full ,
table , out_full ,
brightness , contrast , saturation ) ;
if ( scale - > isws [ 0 ] )
2013-07-20 23:20:25 +03:00
sws_setColorspaceDetails ( scale - > isws [ 0 ] , inv_table , in_full ,
table , out_full ,
brightness , contrast , saturation ) ;
2013-07-20 23:11:50 +03:00
if ( scale - > isws [ 1 ] )
2013-07-20 23:20:25 +03:00
sws_setColorspaceDetails ( scale - > isws [ 1 ] , inv_table , in_full ,
table , out_full ,
brightness , contrast , saturation ) ;
2013-07-20 23:11:50 +03:00
}
2012-11-28 10:41:07 +03:00
av_reduce ( & out - > sample_aspect_ratio . num , & out - > sample_aspect_ratio . den ,
( int64_t ) in - > sample_aspect_ratio . num * outlink - > h * link - > w ,
( int64_t ) in - > sample_aspect_ratio . den * outlink - > w * link - > h ,
2012-11-29 03:19:35 +03:00
INT_MAX ) ;
2009-11-12 01:50:21 +02:00
2013-03-10 03:30:30 +03:00
if ( scale - > interlaced > 0 | | ( scale - > interlaced < 0 & & in - > interlaced_frame ) ) {
2012-11-29 03:19:35 +03:00
scale_slice ( link , out , in , scale - > isws [ 0 ] , 0 , ( link - > h + 1 ) / 2 , 2 , 0 ) ;
scale_slice ( link , out , in , scale - > isws [ 1 ] , 0 , link - > h / 2 , 2 , 1 ) ;
2011-03-28 01:09:58 +03:00
} else {
2012-11-29 03:19:35 +03:00
scale_slice ( link , out , in , scale - > sws , 0 , link - > h , 1 , 0 ) ;
2011-03-28 01:09:58 +03:00
}
2009-11-12 01:50:21 +02:00
2012-11-28 10:41:07 +03:00
av_frame_free ( & in ) ;
2012-11-29 03:19:35 +03:00
return ff_filter_frame ( outlink , out ) ;
2009-11-12 01:50:21 +02:00
}
2013-08-15 23:42:56 +03:00
static const AVClass * child_class_next ( const AVClass * prev )
{
return prev ? NULL : sws_get_class ( ) ;
}
2013-02-25 23:21:29 +03:00
# define OFFSET(x) offsetof(ScaleContext, x)
2013-04-10 21:29:25 +03:00
# define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
static const AVOption scale_options [ ] = {
2013-04-12 03:34:40 +03:00
{ " w " , " Output video width " , OFFSET ( w_expr ) , AV_OPT_TYPE_STRING , . flags = FLAGS } ,
{ " width " , " Output video width " , OFFSET ( w_expr ) , AV_OPT_TYPE_STRING , . flags = FLAGS } ,
{ " h " , " Output video height " , OFFSET ( h_expr ) , AV_OPT_TYPE_STRING , . flags = FLAGS } ,
{ " height " , " Output video height " , OFFSET ( h_expr ) , AV_OPT_TYPE_STRING , . flags = FLAGS } ,
2013-02-25 23:21:29 +03:00
{ " flags " , " Flags to pass to libswscale " , OFFSET ( flags_str ) , AV_OPT_TYPE_STRING , { . str = " bilinear " } , . flags = FLAGS } ,
2013-04-10 21:29:25 +03:00
{ " interl " , " set interlacing " , OFFSET ( interlaced ) , AV_OPT_TYPE_INT , { . i64 = 0 } , - 1 , 1 , FLAGS } ,
{ " size " , " set video size " , OFFSET ( size_str ) , AV_OPT_TYPE_STRING , { . str = NULL } , 0 , FLAGS } ,
{ " s " , " set video size " , OFFSET ( size_str ) , AV_OPT_TYPE_STRING , { . str = NULL } , 0 , FLAGS } ,
2013-07-27 02:16:07 +03:00
{ " in_color_matrix " , " set input YCbCr type " , OFFSET ( in_color_matrix ) , AV_OPT_TYPE_STRING , { . str = " auto " } , . flags = FLAGS } ,
2013-07-20 23:11:50 +03:00
{ " out_color_matrix " , " set output YCbCr type " , OFFSET ( out_color_matrix ) , AV_OPT_TYPE_STRING , { . str = NULL } , . flags = FLAGS } ,
2013-07-20 23:12:42 +03:00
{ " in_range " , " set input color range " , OFFSET ( in_range ) , AV_OPT_TYPE_INT , { . i64 = AVCOL_RANGE_UNSPECIFIED } , 0 , 2 , FLAGS , " range " } ,
{ " out_range " , " set output color range " , OFFSET ( out_range ) , AV_OPT_TYPE_INT , { . i64 = AVCOL_RANGE_UNSPECIFIED } , 0 , 2 , FLAGS , " range " } ,
{ " auto " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = AVCOL_RANGE_UNSPECIFIED } , 0 , 0 , FLAGS , " range " } ,
{ " full " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = AVCOL_RANGE_JPEG } , 0 , 0 , FLAGS , " range " } ,
{ " jpeg " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = AVCOL_RANGE_JPEG } , 0 , 0 , FLAGS , " range " } ,
{ " mpeg " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = AVCOL_RANGE_MPEG } , 0 , 0 , FLAGS , " range " } ,
2013-10-23 13:50:07 +03:00
{ " tv " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = AVCOL_RANGE_MPEG } , 0 , 0 , FLAGS , " range " } ,
{ " pc " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = AVCOL_RANGE_JPEG } , 0 , 0 , FLAGS , " range " } ,
2013-07-24 19:33:58 +03:00
{ " in_v_chr_pos " , " input vertical chroma position in luma grid/256 " , OFFSET ( in_v_chr_pos ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , 512 , FLAGS } ,
{ " in_h_chr_pos " , " input horizontal chroma position in luma grid/256 " , OFFSET ( in_h_chr_pos ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , 512 , FLAGS } ,
2013-07-14 06:03:06 +03:00
{ " out_v_chr_pos " , " output vertical chroma position in luma grid/256 " , OFFSET ( out_v_chr_pos ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , 512 , FLAGS } ,
{ " out_h_chr_pos " , " output horizontal chroma position in luma grid/256 " , OFFSET ( out_h_chr_pos ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , 512 , FLAGS } ,
2013-08-04 02:30:00 +03:00
{ " force_original_aspect_ratio " , " decrease or increase w/h if necessary to keep the original AR " , OFFSET ( force_original_aspect_ratio ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , 2 , FLAGS , " force_oar " } ,
{ " disable " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = 0 } , 0 , 0 , FLAGS , " force_oar " } ,
{ " decrease " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = 1 } , 0 , 0 , FLAGS , " force_oar " } ,
{ " increase " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = 2 } , 0 , 0 , FLAGS , " force_oar " } ,
2013-09-07 15:13:50 +03:00
{ NULL }
2013-02-25 23:21:29 +03:00
} ;
2013-08-15 23:42:56 +03:00
static const AVClass scale_class = {
. class_name = " scale " ,
. item_name = av_default_item_name ,
. option = scale_options ,
. version = LIBAVUTIL_VERSION_INT ,
. child_class_next = child_class_next ,
} ;
2013-02-25 23:21:29 +03:00
2012-07-24 16:14:01 +03:00
static const AVFilterPad avfilter_vf_scale_inputs [ ] = {
{
2013-09-07 15:13:50 +03:00
. name = " default " ,
. type = AVMEDIA_TYPE_VIDEO ,
2012-11-29 03:19:35 +03:00
. filter_frame = filter_frame ,
2012-07-24 16:14:01 +03:00
} ,
{ NULL }
} ;
static const AVFilterPad avfilter_vf_scale_outputs [ ] = {
{
. name = " default " ,
. type = AVMEDIA_TYPE_VIDEO ,
. config_props = config_props ,
} ,
{ NULL }
} ;
2013-10-28 09:44:24 +03:00
AVFilter ff_vf_scale = {
2013-09-07 15:13:50 +03:00
. name = " scale " ,
2013-09-19 20:39:08 +03:00
. description = NULL_IF_CONFIG_SMALL ( " Scale the input video size and/or convert the image format. " ) ,
2013-09-07 15:13:50 +03:00
. init_dict = init_dict ,
. uninit = uninit ,
2009-11-12 01:50:21 +02:00
. query_formats = query_formats ,
2013-09-07 15:13:50 +03:00
. priv_size = sizeof ( ScaleContext ) ,
. priv_class = & scale_class ,
. inputs = avfilter_vf_scale_inputs ,
. outputs = avfilter_vf_scale_outputs ,
2009-11-12 01:50:21 +02:00
} ;