FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
vf_fieldhint.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016 Paul B Mahol
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (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 GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 #include "libavutil/avassert.h"
22 #include "libavutil/imgutils.h"
23 #include "libavutil/internal.h"
24 #include "libavutil/opt.h"
25 #include "libavutil/pixdesc.h"
26 #include "avfilter.h"
27 #include "internal.h"
28 #include "video.h"
29 
30 typedef struct FieldHintContext {
31  const AVClass *class;
32 
34  FILE *hint;
35  int mode;
36 
38 
39  int64_t line;
40  int nb_planes;
41  int eof;
42  int planewidth[4];
43  int planeheight[4];
45 
46 #define OFFSET(x) offsetof(FieldHintContext, x)
47 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
48 
49 static const AVOption fieldhint_options[] = {
50  { "hint", "set hint file", OFFSET(hint_file_str), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS },
51  { "mode", "set hint mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "mode" },
52  { "absolute", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "mode" },
53  { "relative", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "mode" },
54  { NULL }
55 };
56 
57 AVFILTER_DEFINE_CLASS(fieldhint);
58 
60 {
61  FieldHintContext *s = ctx->priv;
62  int ret;
63 
64  if (!s->hint_file_str) {
65  av_log(ctx, AV_LOG_ERROR, "Hint file must be set.\n");
66  return AVERROR(EINVAL);
67  }
68  s->hint = fopen(s->hint_file_str, "r");
69  if (!s->hint) {
70  ret = AVERROR(errno);
71  av_log(ctx, AV_LOG_ERROR, "%s: %s\n", s->hint_file_str, av_err2str(ret));
72  return ret;
73  }
74 
75  return 0;
76 }
77 
79 {
81  int fmt, ret;
82 
83  for (fmt = 0; av_pix_fmt_desc_get(fmt); fmt++) {
85  if (!(desc->flags & AV_PIX_FMT_FLAG_HWACCEL ||
86  desc->flags & AV_PIX_FMT_FLAG_PAL ||
88  (ret = ff_add_format(&pix_fmts, fmt)) < 0)
89  return ret;
90  }
91 
92  return ff_set_common_formats(ctx, pix_fmts);
93 }
94 
95 static int config_input(AVFilterLink *inlink)
96 {
97  FieldHintContext *s = inlink->dst->priv;
99  int ret;
100 
101  if ((ret = av_image_fill_linesizes(s->planewidth, inlink->format, inlink->w)) < 0)
102  return ret;
103 
104  s->planeheight[1] = s->planeheight[2] = AV_CEIL_RSHIFT(inlink->h, desc->log2_chroma_h);
105  s->planeheight[0] = s->planeheight[3] = inlink->h;
106 
108 
109  return 0;
110 }
111 
112 static int filter_frame(AVFilterLink *inlink, AVFrame *in)
113 {
114  AVFilterContext *ctx = inlink->dst;
115  AVFilterLink *outlink = ctx->outputs[0];
116  FieldHintContext *s = ctx->priv;
117  AVFrame *out, *top, *bottom;
118  char buf[1024] = { 0 };
119  int64_t tf, bf;
120  char hint = '=';
121  int p;
122 
123  av_frame_free(&s->frame[0]);
124  s->frame[0] = s->frame[1];
125  s->frame[1] = s->frame[2];
126  s->frame[2] = in;
127  if (!s->frame[1])
128  return 0;
129  else if (!s->frame[0]) {
130  s->frame[0] = av_frame_clone(s->frame[1]);
131  if (!s->frame[0])
132  return AVERROR(ENOMEM);
133  }
134 
135  while (1) {
136  if (fgets(buf, sizeof(buf)-1, s->hint)) {
137  s->line++;
138  if (buf[0] == '#' || buf[0] == ';') {
139  continue;
140  } else if (sscanf(buf, "%"PRId64",%"PRId64" %c", &tf, &bf, &hint) == 3) {
141  ;
142  } else if (sscanf(buf, "%"PRId64",%"PRId64"", &tf, &bf) == 2) {
143  ;
144  } else {
145  av_log(ctx, AV_LOG_ERROR, "Invalid entry at line %"PRId64".\n", s->line);
146  return AVERROR_INVALIDDATA;
147  }
148  switch (s->mode) {
149  case 0:
150  if (tf > outlink->frame_count + 1 || tf < FFMAX(0, outlink->frame_count - 1) ||
151  bf > outlink->frame_count + 1 || bf < FFMAX(0, outlink->frame_count - 1)) {
152  av_log(ctx, AV_LOG_ERROR, "Out of range frames %"PRId64" and/or %"PRId64" on line %"PRId64" for %"PRId64". input frame.\n", tf, bf, s->line, inlink->frame_count);
153  return AVERROR_INVALIDDATA;
154  }
155  break;
156  case 1:
157  if (tf > 1 || tf < -1 ||
158  bf > 1 || bf < -1) {
159  av_log(ctx, AV_LOG_ERROR, "Out of range %"PRId64" and/or %"PRId64" on line %"PRId64" for %"PRId64". input frame.\n", tf, bf, s->line, inlink->frame_count);
160  return AVERROR_INVALIDDATA;
161  }
162  };
163  break;
164  } else {
165  av_log(ctx, AV_LOG_ERROR, "Missing entry for %"PRId64". input frame.\n", inlink->frame_count);
166  return AVERROR_INVALIDDATA;
167  }
168  }
169 
170  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
171  if (!out)
172  return AVERROR(ENOMEM);
173  av_frame_copy_props(out, s->frame[1]);
174 
175  switch (s->mode) {
176  case 0:
177  top = s->frame[tf - outlink->frame_count + 1];
178  bottom = s->frame[bf - outlink->frame_count + 1];
179  break;
180  case 1:
181  top = s->frame[1 + tf];
182  bottom = s->frame[1 + bf];
183  break;
184  default:
185  av_assert0(0);
186  }
187 
188  switch (hint) {
189  case '+':
190  out->interlaced_frame = 1;
191  break;
192  case '-':
193  out->interlaced_frame = 0;
194  break;
195  case '=':
196  break;
197  default:
198  av_log(ctx, AV_LOG_ERROR, "Invalid hint: %c.\n", hint);
199  return AVERROR(EINVAL);
200  }
201 
202  for (p = 0; p < s->nb_planes; p++) {
203  av_image_copy_plane(out->data[p],
204  out->linesize[p] * 2,
205  top->data[p],
206  top->linesize[p] * 2,
207  s->planewidth[p],
208  (s->planeheight[p] + 1) / 2);
209  av_image_copy_plane(out->data[p] + out->linesize[p],
210  out->linesize[p] * 2,
211  bottom->data[p] + bottom->linesize[p],
212  bottom->linesize[p] * 2,
213  s->planewidth[p],
214  (s->planeheight[p] + 1) / 2);
215  }
216 
217  return ff_filter_frame(outlink, out);
218 }
219 
220 static int request_frame(AVFilterLink *outlink)
221 {
222  AVFilterContext *ctx = outlink->src;
223  FieldHintContext *s = ctx->priv;
224  int ret;
225 
226  if (s->eof)
227  return AVERROR_EOF;
228 
229  ret = ff_request_frame(ctx->inputs[0]);
230  if (ret == AVERROR_EOF && s->frame[2]) {
231  AVFrame *next = av_frame_clone(s->frame[2]);
232  if (!next)
233  return AVERROR(ENOMEM);
234  ret = filter_frame(ctx->inputs[0], next);
235  s->eof = 1;
236  }
237 
238  return ret;
239 }
240 
242 {
243  FieldHintContext *s = ctx->priv;
244 
245  if (s->hint)
246  fclose(s->hint);
247  s->hint = NULL;
248 
249  av_frame_free(&s->frame[0]);
250  av_frame_free(&s->frame[1]);
251  av_frame_free(&s->frame[2]);
252 }
253 
254 static const AVFilterPad inputs[] = {
255  {
256  .name = "default",
257  .type = AVMEDIA_TYPE_VIDEO,
258  .config_props = config_input,
259  .filter_frame = filter_frame,
260  },
261  { NULL }
262 };
263 
264 static const AVFilterPad outputs[] = {
265  {
266  .name = "default",
267  .type = AVMEDIA_TYPE_VIDEO,
268  .request_frame = request_frame,
269  },
270  { NULL }
271 };
272 
274  .name = "fieldhint",
275  .description = NULL_IF_CONFIG_SMALL("Field matching using hints."),
276  .priv_size = sizeof(FieldHintContext),
277  .priv_class = &fieldhint_class,
278  .init = init,
279  .uninit = uninit,
281  .inputs = inputs,
282  .outputs = outputs,
283 };
#define AV_PIX_FMT_FLAG_PAL
Pixel format has a palette in data[1], values are indexes in this palette.
Definition: pixdesc.h:132
#define NULL
Definition: coverity.c:32
const char * s
Definition: avisynth_c.h:768
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2266
This structure describes decoded (raw) audio or video data.
Definition: frame.h:184
AVOption.
Definition: opt.h:245
const char * fmt
Definition: avisynth_c.h:769
#define FLAGS
Definition: vf_fieldhint.c:47
misc image utilities
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2306
Main libavfilter public API header.
static const AVFilterPad inputs[]
Definition: vf_fieldhint.c:254
const char * desc
Definition: nvenc.c:101
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:76
#define OFFSET(x)
Definition: vf_fieldhint.c:46
#define tf
Definition: regdef.h:73
const char * name
Pad name.
Definition: internal.h:59
static av_cold int init(AVFilterContext *ctx)
Definition: vf_fieldhint.c:59
AVFilterLink ** inputs
array of pointers to input links
Definition: avfilter.h:315
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1189
#define av_cold
Definition: attributes.h:82
mode
Definition: f_perms.c:27
AVOptions.
#define AVERROR_EOF
End of file.
Definition: error.h:55
int interlaced_frame
The content of the picture is interlaced.
Definition: frame.h:322
#define av_log(a,...)
static int request_frame(AVFilterLink *outlink)
Definition: vf_fieldhint.c:220
AVFilter ff_vf_fieldhint
Definition: vf_fieldhint.c:273
A filter pad used for either input or output.
Definition: internal.h:53
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
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
uint8_t log2_chroma_h
Amount to shift the luma height right to find the chroma height.
Definition: pixdesc.h:101
#define AVERROR(e)
Definition: error.h:43
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:158
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:176
void * priv
private data for use by the filter
Definition: avfilter.h:322
static int query_formats(AVFilterContext *ctx)
Definition: vf_fieldhint.c:78
#define AV_PIX_FMT_FLAG_HWACCEL
Pixel format is an HW accelerated format.
Definition: pixdesc.h:140
simple assert() macros that are a bit more flexible than ISO C assert().
int ff_add_format(AVFilterFormats **avff, int64_t fmt)
Add fmt to the list of media formats contained in *avff.
Definition: formats.c:337
#define FFMAX(a, b)
Definition: common.h:94
AVFILTER_DEFINE_CLASS(fieldhint)
common internal API header
uint64_t flags
Combination of AV_PIX_FMT_FLAG_...
Definition: pixdesc.h:106
#define av_err2str(errnum)
Convenience macro, the return value should be used only directly in function arguments but never stan...
Definition: error.h:119
AVFormatContext * ctx
Definition: movenc.c:48
AVFrame * av_frame_clone(const AVFrame *src)
Create a new frame that references the same data as src.
Definition: frame.c:480
char * hint_file_str
Definition: vf_fieldhint.c:33
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:215
static const AVOption fieldhint_options[]
Definition: vf_fieldhint.c:49
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:81
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
void * buf
Definition: avisynth_c.h:690
Describe the class of an AVClass context structure.
Definition: log.h:67
Filter definition.
Definition: avfilter.h:144
int av_image_fill_linesizes(int linesizes[4], enum AVPixelFormat pix_fmt, int width)
Fill plane linesizes for an image with pixel format pix_fmt and width width.
Definition: imgutils.c:88
const char * name
Filter name.
Definition: avfilter.h:148
#define AV_PIX_FMT_FLAG_BITSTREAM
All values of a component are bit-wise packed end to end.
Definition: pixdesc.h:136
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:319
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:262
static const AVFilterPad outputs[]
Definition: vf_fieldhint.c:264
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:198
static int config_input(AVFilterLink *inlink)
Definition: vf_fieldhint.c:95
AVFrame * frame[3]
Definition: vf_fieldhint.c:37
A list of supported formats for one end of a filter link.
Definition: formats.h:64
An instance of a filter.
Definition: avfilter.h:307
FILE * out
Definition: movenc.c:54
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_fieldhint.c:241
int ff_request_frame(AVFilterLink *link)
Request an input frame from the filter at the other end of the link.
Definition: avfilter.c:369
void av_image_copy_plane(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize, int bytewidth, int height)
Copy image plane from src to dst.
Definition: imgutils.c:287
internal API functions
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_fieldhint.c:112
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:589
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58