yading@10: /* yading@10: * Copyright (c) 2010 Bobby Bingham yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * aspect ratio modification video filters yading@10: */ yading@10: yading@10: #include yading@10: yading@10: #include "libavutil/common.h" yading@10: #include "libavutil/eval.h" yading@10: #include "libavutil/mathematics.h" yading@10: #include "libavutil/opt.h" yading@10: #include "libavutil/parseutils.h" yading@10: yading@10: #include "avfilter.h" yading@10: #include "internal.h" yading@10: #include "video.h" yading@10: yading@10: typedef struct { yading@10: const AVClass *class; yading@10: AVRational aspect; yading@10: int max; yading@10: #if FF_API_OLD_FILTER_OPTS yading@10: float aspect_den; yading@10: #endif yading@10: char *ratio_str; yading@10: } AspectContext; yading@10: yading@10: static av_cold int init(AVFilterContext *ctx) yading@10: { yading@10: AspectContext *s = ctx->priv; yading@10: int ret; yading@10: yading@10: #if FF_API_OLD_FILTER_OPTS yading@10: if (s->ratio_str && s->aspect_den > 0) { yading@10: double num; yading@10: av_log(ctx, AV_LOG_WARNING, yading@10: "num:den syntax is deprecated, please use num/den or named options instead\n"); yading@10: ret = av_expr_parse_and_eval(&num, s->ratio_str, NULL, NULL, yading@10: NULL, NULL, NULL, NULL, NULL, 0, ctx); yading@10: if (ret < 0) { yading@10: av_log(ctx, AV_LOG_ERROR, "Unable to parse ratio numerator \"%s\"\n", s->ratio_str); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: s->aspect = av_d2q(num / s->aspect_den, s->max); yading@10: } else yading@10: #endif yading@10: if (s->ratio_str) { yading@10: ret = av_parse_ratio(&s->aspect, s->ratio_str, s->max, 0, ctx); yading@10: if (ret < 0 || s->aspect.num < 0 || s->aspect.den <= 0) { yading@10: av_log(ctx, AV_LOG_ERROR, yading@10: "Invalid string '%s' for aspect ratio\n", s->ratio_str); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static int filter_frame(AVFilterLink *link, AVFrame *frame) yading@10: { yading@10: AspectContext *aspect = link->dst->priv; yading@10: yading@10: frame->sample_aspect_ratio = aspect->aspect; yading@10: return ff_filter_frame(link->dst->outputs[0], frame); yading@10: } yading@10: yading@10: #define OFFSET(x) offsetof(AspectContext, x) yading@10: #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM yading@10: yading@10: static inline void compute_dar(AVRational *dar, AVRational sar, int w, int h) yading@10: { yading@10: if (sar.num && sar.den) { yading@10: av_reduce(&dar->num, &dar->den, sar.num * w, sar.den * h, INT_MAX); yading@10: } else { yading@10: av_reduce(&dar->num, &dar->den, w, h, INT_MAX); yading@10: } yading@10: } yading@10: yading@10: #if CONFIG_SETDAR_FILTER yading@10: yading@10: static int setdar_config_props(AVFilterLink *inlink) yading@10: { yading@10: AspectContext *aspect = inlink->dst->priv; yading@10: AVRational dar = aspect->aspect, old_dar; yading@10: AVRational old_sar = inlink->sample_aspect_ratio; yading@10: yading@10: if (aspect->aspect.num && aspect->aspect.den) { yading@10: av_reduce(&aspect->aspect.num, &aspect->aspect.den, yading@10: aspect->aspect.num * inlink->h, yading@10: aspect->aspect.den * inlink->w, INT_MAX); yading@10: inlink->sample_aspect_ratio = aspect->aspect; yading@10: } else { yading@10: inlink->sample_aspect_ratio = (AVRational){ 1, 1 }; yading@10: dar = (AVRational){ inlink->w, inlink->h }; yading@10: } yading@10: yading@10: compute_dar(&old_dar, old_sar, inlink->w, inlink->h); yading@10: av_log(inlink->dst, AV_LOG_VERBOSE, "w:%d h:%d dar:%d/%d sar:%d/%d -> dar:%d/%d sar:%d/%d\n", yading@10: inlink->w, inlink->h, old_dar.num, old_dar.den, old_sar.num, old_sar.den, yading@10: dar.num, dar.den, inlink->sample_aspect_ratio.num, inlink->sample_aspect_ratio.den); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static const AVOption setdar_options[] = { yading@10: { "dar", "set display aspect ratio", OFFSET(ratio_str), AV_OPT_TYPE_STRING, {.str="0"}, .flags=FLAGS }, yading@10: { "ratio", "set display aspect ratio", OFFSET(ratio_str), AV_OPT_TYPE_STRING, {.str="0"}, .flags=FLAGS }, yading@10: { "r", "set display aspect ratio", OFFSET(ratio_str), AV_OPT_TYPE_STRING, {.str="0"}, .flags=FLAGS }, yading@10: #if FF_API_OLD_FILTER_OPTS yading@10: { "dar_den", NULL, OFFSET(aspect_den), AV_OPT_TYPE_FLOAT, { .dbl = 0 }, 0, FLT_MAX, FLAGS }, yading@10: #endif yading@10: { "max", "set max value for nominator or denominator in the ratio", OFFSET(max), AV_OPT_TYPE_INT, {.i64=100}, 1, INT_MAX, FLAGS }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFILTER_DEFINE_CLASS(setdar); yading@10: yading@10: static const AVFilterPad avfilter_vf_setdar_inputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .config_props = setdar_config_props, yading@10: .get_video_buffer = ff_null_get_video_buffer, yading@10: .filter_frame = filter_frame, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: static const AVFilterPad avfilter_vf_setdar_outputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFilter avfilter_vf_setdar = { yading@10: .name = "setdar", yading@10: .description = NULL_IF_CONFIG_SMALL("Set the frame display aspect ratio."), yading@10: .init = init, yading@10: .priv_size = sizeof(AspectContext), yading@10: .priv_class = &setdar_class, yading@10: yading@10: .inputs = avfilter_vf_setdar_inputs, yading@10: yading@10: .outputs = avfilter_vf_setdar_outputs, yading@10: }; yading@10: yading@10: #endif /* CONFIG_SETDAR_FILTER */ yading@10: yading@10: #if CONFIG_SETSAR_FILTER yading@10: yading@10: static int setsar_config_props(AVFilterLink *inlink) yading@10: { yading@10: AspectContext *aspect = inlink->dst->priv; yading@10: AVRational old_sar = inlink->sample_aspect_ratio; yading@10: AVRational old_dar, dar; yading@10: yading@10: inlink->sample_aspect_ratio = aspect->aspect; yading@10: yading@10: compute_dar(&old_dar, old_sar, inlink->w, inlink->h); yading@10: compute_dar(&dar, aspect->aspect, inlink->w, inlink->h); yading@10: av_log(inlink->dst, AV_LOG_VERBOSE, "w:%d h:%d sar:%d/%d dar:%d/%d -> sar:%d/%d dar:%d/%d\n", yading@10: inlink->w, inlink->h, old_sar.num, old_sar.den, old_dar.num, old_dar.den, yading@10: inlink->sample_aspect_ratio.num, inlink->sample_aspect_ratio.den, dar.num, dar.den); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static const AVOption setsar_options[] = { yading@10: { "sar", "set sample (pixel) aspect ratio", OFFSET(ratio_str), AV_OPT_TYPE_STRING, {.str="0"}, .flags=FLAGS }, yading@10: { "ratio", "set sample (pixel) aspect ratio", OFFSET(ratio_str), AV_OPT_TYPE_STRING, {.str="0"}, .flags=FLAGS }, yading@10: { "r", "set sample (pixel) aspect ratio", OFFSET(ratio_str), AV_OPT_TYPE_STRING, {.str="0"}, .flags=FLAGS }, yading@10: #if FF_API_OLD_FILTER_OPTS yading@10: { "sar_den", NULL, OFFSET(aspect_den), AV_OPT_TYPE_FLOAT, { .dbl = 0 }, 0, FLT_MAX, FLAGS }, yading@10: #endif yading@10: { "max", "set max value for nominator or denominator in the ratio", OFFSET(max), AV_OPT_TYPE_INT, {.i64=100}, 1, INT_MAX, FLAGS }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFILTER_DEFINE_CLASS(setsar); yading@10: yading@10: static const AVFilterPad avfilter_vf_setsar_inputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .config_props = setsar_config_props, yading@10: .get_video_buffer = ff_null_get_video_buffer, yading@10: .filter_frame = filter_frame, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: static const AVFilterPad avfilter_vf_setsar_outputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFilter avfilter_vf_setsar = { yading@10: .name = "setsar", yading@10: .description = NULL_IF_CONFIG_SMALL("Set the pixel sample aspect ratio."), yading@10: .init = init, yading@10: .priv_size = sizeof(AspectContext), yading@10: .priv_class = &setsar_class, yading@10: yading@10: .inputs = avfilter_vf_setsar_inputs, yading@10: yading@10: .outputs = avfilter_vf_setsar_outputs, yading@10: }; yading@10: yading@10: #endif /* CONFIG_SETSAR_FILTER */