FFmpeg: libavfilter/vf_cover_rect.c Source File

FFmpeg
vf_cover_rect.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2014-2015 Michael Niedermayer <michaelni@gmx.at>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License along
17  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19  */
20 
21 /**
22  * @todo switch to dualinput
23  */
24 
25 #include "libavutil/avassert.h"
26 #include "libavutil/imgutils.h"
27 #include "libavutil/opt.h"
28 #include "internal.h"
29 
30 #include "lavfutils.h"
31 
32  enum mode {
33   MODE_COVER,
34   MODE_BLUR,
35   NB_MODES
36 };
37 
38  typedef struct CoverContext {
39   AVClass *class;
40   int mode;
41   char *cover_filename;
42   AVFrame *cover_frame;
43   int width, height;
44 } CoverContext;
45 
46  #define OFFSET(x) offsetof(CoverContext, x)
47  #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
48  static const AVOption cover_rect_options[] = {
49  { "cover", "cover bitmap filename", OFFSET(cover_filename), AV_OPT_TYPE_STRING, {.str = NULL}, .flags = FLAGS },
50  { "mode", "set removal mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64 = MODE_BLUR}, 0, NB_MODES - 1, FLAGS, "mode" },
51  { "cover", "cover area with bitmap", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_COVER}, INT_MIN, INT_MAX, FLAGS, "mode" },
52  { "blur", "blur area", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_BLUR}, INT_MIN, INT_MAX, FLAGS, "mode" },
53  { NULL }
54 };
55 
56 AVFILTER_DEFINE_CLASS(cover_rect);
57 
58  static int query_formats(AVFilterContext *ctx)
59 {
60  static const enum AVPixelFormat pix_fmts[] = {
61  AV_PIX_FMT_YUV420P,
62  AV_PIX_FMT_YUVJ420P,
63  AV_PIX_FMT_NONE
64  };
65 
66  return ff_set_common_formats(ctx, ff_make_format_list(pix_fmts));
67 }
68 
69  static int config_input(AVFilterLink *inlink)
70 {
71  return 0;
72 }
73 
74  static void cover_rect(CoverContext *cover, AVFrame *in, int offx, int offy)
75 {
76  int x, y, p;
77 
78  for (p = 0; p < 3; p++) {
79  uint8_t *data = in->data[p] + (offx>>!!p) + (offy>>!!p) * in->linesize[p];
80  const uint8_t *src = cover->cover_frame->data[p];
81  int w = AV_CEIL_RSHIFT(cover->cover_frame->width , !!p);
82  int h = AV_CEIL_RSHIFT(cover->cover_frame->height, !!p);
83  for (y = 0; y < h; y++) {
84  for (x = 0; x < w; x++) {
85  data[x] = src[x];
86  }
87  data += in->linesize[p];
88  src += cover->cover_frame->linesize[p];
89  }
90  }
91 }
92  static void blur(CoverContext *cover, AVFrame *in, int offx, int offy)
93 {
94  int x, y, p;
95 
96  for (p=0; p<3; p++) {
97  int ox = offx>>!!p;
98  int oy = offy>>!!p;
99  int stride = in->linesize[p];
100  uint8_t *data = in->data[p] + ox + oy * stride;
101  int w = AV_CEIL_RSHIFT(cover->width , !!p);
102  int h = AV_CEIL_RSHIFT(cover->height, !!p);
103  int iw = AV_CEIL_RSHIFT(in->width , !!p);
104  int ih = AV_CEIL_RSHIFT(in->height, !!p);
105  for (y = 0; y < h; y++) {
106  for (x = 0; x < w; x++) {
107  int c = 0;
108  int s = 0;
109  if (ox) {
110  int scale = 65536 / (x + 1);
111  s += data[-1 + y*stride] * scale;
112  c += scale;
113  }
114  if (oy) {
115  int scale = 65536 / (y + 1);
116  s += data[x - stride] * scale;
117  c += scale;
118  }
119  if (ox + w < iw) {
120  int scale = 65536 / (w - x);
121  s += data[w + y*stride] * scale;
122  c += scale;
123  }
124  if (oy + h < ih) {
125  int scale = 65536 / (h - y);
126  s += data[x + h*stride] * scale;
127  c += scale;
128  }
129  data[x + y*stride] = c ? (s + (c>>1)) / c : 0;
130  }
131  }
132  }
133 }
134 
135  static int filter_frame(AVFilterLink *inlink, AVFrame *in)
136 {
137  AVFilterContext *ctx = inlink->dst;
138  CoverContext *cover = ctx->priv;
139  AVDictionaryEntry *ex, *ey, *ew, *eh;
140  int x = -1, y = -1, w = -1, h = -1;
141  char *xendptr = NULL, *yendptr = NULL, *wendptr = NULL, *hendptr = NULL;
142 
143  ex = av_dict_get(in->metadata, "lavfi.rect.x", NULL, AV_DICT_MATCH_CASE);
144  ey = av_dict_get(in->metadata, "lavfi.rect.y", NULL, AV_DICT_MATCH_CASE);
145  ew = av_dict_get(in->metadata, "lavfi.rect.w", NULL, AV_DICT_MATCH_CASE);
146  eh = av_dict_get(in->metadata, "lavfi.rect.h", NULL, AV_DICT_MATCH_CASE);
147  if (ex && ey && ew && eh) {
148  x = strtol(ex->value, &xendptr, 10);
149  y = strtol(ey->value, &yendptr, 10);
150  w = strtol(ew->value, &wendptr, 10);
151  h = strtol(eh->value, &hendptr, 10);
152  }
153 
154  if (!xendptr || *xendptr || !yendptr || *yendptr ||
155  !wendptr || *wendptr || !hendptr || !hendptr
156  ) {
157  return ff_filter_frame(ctx->outputs[0], in);
158  }
159 
160  if (x < 0) {
161  w += x;
162  x = 0;
163  }
164  if (y < 0) {
165  h += y;
166  y = 0;
167  }
168  w = FFMIN(w, in->width - x);
169  h = FFMIN(h, in->height - y);
170 
171  if (w > in->width || h > in->height || w <= 0 || h <= 0)
172  return AVERROR(EINVAL);
173 
174  if (cover->cover_frame) {
175  if (w != cover->cover_frame->width || h != cover->cover_frame->height)
176  return AVERROR(EINVAL);
177  }
178 
179  cover->width = w;
180  cover->height = h;
181 
182  x = av_clip(x, 0, in->width - w);
183  y = av_clip(y, 0, in->height - h);
184 
185  av_frame_make_writable(in);
186 
187  if (cover->mode == MODE_BLUR) {
188  blur (cover, in, x, y);
189  } else {
190  cover_rect(cover, in, x, y);
191  }
192  return ff_filter_frame(ctx->outputs[0], in);
193 }
194 
195  static av_cold void uninit(AVFilterContext *ctx)
196 {
197  CoverContext *cover = ctx->priv;
198 
199  if (cover->cover_frame)
200  av_freep(&cover->cover_frame->data[0]);
201 }
202 
203  static av_cold int init(AVFilterContext *ctx)
204 {
205  CoverContext *cover = ctx->priv;
206  int ret;
207 
208  if (cover->mode == MODE_COVER) {
209  if (!cover->cover_filename) {
210  av_log(ctx, AV_LOG_ERROR, "cover filename not set\n");
211  return AVERROR(EINVAL);
212  }
213 
214  cover->cover_frame = av_frame_alloc();
215  if (!cover->cover_frame)
216  return AVERROR(ENOMEM);
217 
218  if ((ret = ff_load_image(cover->cover_frame->data, cover->cover_frame->linesize,
219  &cover->cover_frame->width, &cover->cover_frame->height,
220  &cover->cover_frame->format, cover->cover_filename, ctx)) < 0)
221  return ret;
222 
223  if (cover->cover_frame->format != AV_PIX_FMT_YUV420P && cover->cover_frame->format != AV_PIX_FMT_YUVJ420P) {
224  av_log(ctx, AV_LOG_ERROR, "cover image is not a YUV420 image\n");
225  return AVERROR(EINVAL);
226  }
227  }
228 
229  return 0;
230 }
231 
232  static const AVFilterPad cover_rect_inputs[] = {
233  {
234  .name = "default",
235  .type = AVMEDIA_TYPE_VIDEO,
236  .config_props = config_input,
237  .filter_frame = filter_frame,
238  },
239  { NULL }
240 };
241 
242  static const AVFilterPad cover_rect_outputs[] = {
243  {
244  .name = "default",
245  .type = AVMEDIA_TYPE_VIDEO,
246  },
247  { NULL }
248 };
249 
250  AVFilter ff_vf_cover_rect = {
251  .name = "cover_rect",
252  .description = NULL_IF_CONFIG_SMALL("Find and cover a user specified object."),
253  .priv_size = sizeof(CoverContext),
254  .init = init,
255  .uninit = uninit,
256  .query_formats = query_formats,
257  .inputs = cover_rect_inputs,
258  .outputs = cover_rect_outputs,
259  .priv_class = &cover_rect_class,
260 };
NULL
#define NULL
Definition: coverity.c:32
s
const char * s
Definition: avisynth_c.h:631
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:181
h
h
Definition: vp9dsp_template.c:2093
AVOption
AVOption.
Definition: opt.h:245
data
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:101
ctx
AVFormatContext * ctx
Definition: movenc-test.c:48
imgutils.h
misc image utilities
AV_OPT_TYPE_INT
Definition: opt.h:223
CoverContext::cover_frame
AVFrame * cover_frame
Definition: vf_cover_rect.c:42
query_formats
static int query_formats(AVFilterContext *ctx)
Definition: vf_cover_rect.c:58
config_input
static int config_input(AVFilterLink *inlink)
Definition: vf_cover_rect.c:69
MODE_BLUR
Definition: vf_cover_rect.c:34
init
static av_cold int init(AVFilterContext *ctx)
Definition: vf_cover_rect.c:203
ff_make_format_list
AVFilterFormats * ff_make_format_list(const int *fmts)
Create a list of supported formats.
Definition: formats.c:283
AVFilterPad::name
const char * name
Pad name.
Definition: internal.h:59
ff_vf_cover_rect
AVFilter ff_vf_cover_rect
Definition: vf_cover_rect.c:250
AV_OPT_TYPE_CONST
Definition: opt.h:231
ff_filter_frame
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1163
uint8_t
uint8_t
Definition: audio_convert.c:194
av_cold
#define av_cold
Definition: attributes.h:82
av_frame_alloc
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:141
mode
mode
Definition: f_perms.c:27
opt.h
AVOptions.
av_dict_get
AVDictionaryEntry * av_dict_get(const AVDictionary *m, const char *key, const AVDictionaryEntry *prev, int flags)
Get a dictionary entry with matching key.
Definition: dict.c:39
AV_OPT_TYPE_STRING
Definition: opt.h:227
AVFrame::metadata
AVDictionary * metadata
metadata.
Definition: frame.h:459
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
AVFilterPad
A filter pad used for either input or output.
Definition: internal.h:53
AVFrame::width
int width
width and height of the video frame
Definition: frame.h:230
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
AV_DICT_MATCH_CASE
#define AV_DICT_MATCH_CASE
Only get an entry with exact-case key match.
Definition: dict.h:71
ff_set_common_formats
int ff_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
A helper for query_formats() which sets all links to the same list of formats.
Definition: formats.c:568
blur
static void blur(CoverContext *cover, AVFrame *in, int offx, int offy)
Definition: vf_cover_rect.c:92
AVERROR
#define AVERROR(e)
Definition: error.h:43
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:176
AVFilterContext::priv
void * priv
private data for use by the filter
Definition: avfilter.h:319
avassert.h
simple assert() macros that are a bit more flexible than ISO C assert().
FFMIN
#define FFMIN(a, b)
Definition: common.h:96
AV_PIX_FMT_YUVJ420P
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:75
cover_rect
static void cover_rect(CoverContext *cover, AVFrame *in, int offx, int offy)
Definition: vf_cover_rect.c:74
ff_load_image
int ff_load_image(uint8_t *data[4], int linesize[4], int *w, int *h, enum AVPixelFormat *pix_fmt, const char *filename, void *log_ctx)
Load image from filename and put the resulting image in data.
Definition: lavfutils.c:24
NB_MODES
Definition: vf_cover_rect.c:35
outputs
static const AVFilterPad outputs[]
Definition: af_afftfilt.c:385
src
#define src
Definition: vp9dsp.c:530
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_cover_rect.c:195
AVFrame::format
int format
format of the frame, -1 if unknown or unset Values correspond to enum AVPixelFormat for video frames...
Definition: frame.h:242
inputs
static const AVFilterPad inputs[]
Definition: af_afftfilt.c:375
cover_rect_outputs
static const AVFilterPad cover_rect_outputs[]
Definition: vf_cover_rect.c:242
AVFrame::linesize
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:209
in
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_log(ac->avr, AV_LOG_TRACE,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> in
Definition: audio_convert.c:194
cover_rect_options
static const AVOption cover_rect_options[]
Definition: vf_cover_rect.c:48
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:67
AVFilter
Filter definition.
Definition: avfilter.h:141
OFFSET
#define OFFSET(x)
Definition: vf_cover_rect.c:46
AVFilter::name
const char * name
Filter name.
Definition: avfilter.h:145
AVFILTER_DEFINE_CLASS
AVFILTER_DEFINE_CLASS(cover_rect)
AVFilterContext::outputs
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:316
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:262
av_frame_make_writable
int av_frame_make_writable(AVFrame *frame)
Ensure that the frame data is writable, avoiding data copy if possible.
Definition: frame.c:522
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:192
cover_rect_inputs
static const AVFilterPad cover_rect_inputs[]
Definition: vf_cover_rect.c:232
stride
GLint GLenum GLboolean GLsizei stride
Definition: opengl_enc.c:105
AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:63
c
static double c[64]
Definition: vsrc_mptestsrc.c:87
MODE_COVER
Definition: vf_cover_rect.c:33
AVDictionaryEntry::value
char * value
Definition: dict.h:88
FLAGS
#define FLAGS
Definition: vf_cover_rect.c:47
AVFilterContext
An instance of a filter.
Definition: avfilter.h:304
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_cover_rect.c:135
AVFrame::height
int height
Definition: frame.h:230
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:35
CoverContext::cover_filename
char * cover_filename
Definition: vf_cover_rect.c:41
AVMEDIA_TYPE_VIDEO
Definition: avutil.h:193
stride
#define stride
internal.h
internal API functions
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:61
lavfutils.h
Miscellaneous utilities which make use of the libavformat library.
AV_CEIL_RSHIFT
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58
Generated on Mon Feb 15 2016 15:20:44 for FFmpeg by   doxygen 1.8.6

AltStyle によって変換されたページ (->オリジナル) /