FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
vf_hwdownload.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include "libavutil/buffer.h"
20 #include "libavutil/hwcontext.h"
21 #include "libavutil/log.h"
22 #include "libavutil/mem.h"
23 #include "libavutil/opt.h"
24 #include "libavutil/pixdesc.h"
25 
26 #include "avfilter.h"
27 #include "formats.h"
28 #include "internal.h"
29 #include "video.h"
30 
31 typedef struct HWDownloadContext {
32  const AVClass *class;
33 
37 
39 {
40  AVFilterFormats *infmts = NULL;
41  AVFilterFormats *outfmts = NULL;
42  const AVPixFmtDescriptor *desc;
43  int err;
44 
45  for (desc = av_pix_fmt_desc_next(NULL); desc;
46  desc = av_pix_fmt_desc_next(desc)) {
47  if (desc->flags & AV_PIX_FMT_FLAG_HWACCEL)
48  err = ff_add_format(&infmts, av_pix_fmt_desc_get_id(desc));
49  else
50  err = ff_add_format(&outfmts, av_pix_fmt_desc_get_id(desc));
51  if (err) {
52  ff_formats_unref(&infmts);
53  ff_formats_unref(&outfmts);
54  return err;
55  }
56  }
57 
58  if ((err = ff_formats_ref(infmts, &avctx->inputs[0]->out_formats)) < 0 ||
59  (err = ff_formats_ref(outfmts, &avctx->outputs[0]->in_formats)) < 0)
60  return err;
61 
62  return 0;
63 }
64 
66 {
67  AVFilterContext *avctx = inlink->dst;
68  HWDownloadContext *ctx = avctx->priv;
69 
71 
72  if (!inlink->hw_frames_ctx) {
73  av_log(ctx, AV_LOG_ERROR, "The input must have a hardware frame "
74  "reference.\n");
75  return AVERROR(EINVAL);
76  }
77 
79  if (!ctx->hwframes_ref)
80  return AVERROR(ENOMEM);
81 
83 
84  return 0;
85 }
86 
88 {
89  AVFilterContext *avctx = outlink->src;
90  AVFilterLink *inlink = avctx->inputs[0];
91  HWDownloadContext *ctx = avctx->priv;
92  enum AVPixelFormat *formats;
93  int err, i, found;
94 
95  if (!ctx->hwframes_ref)
96  return AVERROR(EINVAL);
97 
100  &formats, 0);
101  if (err < 0)
102  return err;
103 
104  found = 0;
105  for (i = 0; formats[i] != AV_PIX_FMT_NONE; i++) {
106  if (formats[i] == outlink->format) {
107  found = 1;
108  break;
109  }
110  }
111  av_freep(&formats);
112 
113  if (!found) {
114  av_log(ctx, AV_LOG_ERROR, "Invalid output format %s for hwframe "
115  "download.\n", av_get_pix_fmt_name(outlink->format));
116  return AVERROR(EINVAL);
117  }
118 
119  outlink->w = inlink->w;
120  outlink->h = inlink->h;
121 
122  return 0;
123 }
124 
126 {
127  AVFilterContext *avctx = link->dst;
128  AVFilterLink *outlink = avctx->outputs[0];
129  HWDownloadContext *ctx = avctx->priv;
130  AVFrame *output = NULL;
131  int err;
132 
133  if (!ctx->hwframes_ref || !input->hw_frames_ctx) {
134  av_log(ctx, AV_LOG_ERROR, "Input frames must have hardware context.\n");
135  err = AVERROR(EINVAL);
136  goto fail;
137  }
138  if ((void*)ctx->hwframes != input->hw_frames_ctx->data) {
139  av_log(ctx, AV_LOG_ERROR, "Input frame is not the in the configured "
140  "hwframe context.\n");
141  err = AVERROR(EINVAL);
142  goto fail;
143  }
144 
145  output = ff_get_video_buffer(outlink, outlink->w, outlink->h);
146  if (!output) {
147  err = AVERROR(ENOMEM);
148  goto fail;
149  }
150 
151  err = av_hwframe_transfer_data(output, input, 0);
152  if (err < 0) {
153  av_log(ctx, AV_LOG_ERROR, "Failed to download frame: %d.\n", err);
154  goto fail;
155  }
156 
157  err = av_frame_copy_props(output, input);
158  if (err < 0)
159  goto fail;
160 
161  av_frame_free(&input);
162 
163  return ff_filter_frame(avctx->outputs[0], output);
164 
165 fail:
166  av_frame_free(&input);
167  av_frame_free(&output);
168  return err;
169 }
170 
172 {
173  HWDownloadContext *ctx = avctx->priv;
174 
176 }
177 
178 static const AVClass hwdownload_class = {
179  .class_name = "hwdownload",
180  .item_name = av_default_item_name,
181  .option = NULL,
182  .version = LIBAVUTIL_VERSION_INT,
183 };
184 
185 static const AVFilterPad hwdownload_inputs[] = {
186  {
187  .name = "default",
188  .type = AVMEDIA_TYPE_VIDEO,
189  .config_props = hwdownload_config_input,
190  .filter_frame = hwdownload_filter_frame,
191  },
192  { NULL }
193 };
194 
195 static const AVFilterPad hwdownload_outputs[] = {
196  {
197  .name = "default",
198  .type = AVMEDIA_TYPE_VIDEO,
199  .config_props = hwdownload_config_output,
200  },
201  { NULL }
202 };
203 
205  .name = "hwdownload",
206  .description = NULL_IF_CONFIG_SMALL("Download a hardware frame to a normal frame"),
207  .uninit = hwdownload_uninit,
208  .query_formats = hwdownload_query_formats,
209  .priv_size = sizeof(HWDownloadContext),
210  .priv_class = &hwdownload_class,
211  .inputs = hwdownload_inputs,
212  .outputs = hwdownload_outputs,
213 };
static av_cold void hwdownload_uninit(AVFilterContext *avctx)
#define NULL
Definition: coverity.c:32
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it...
Definition: buffer.c:124
This structure describes decoded (raw) audio or video data.
Definition: frame.h:184
#define LIBAVUTIL_VERSION_INT
Definition: version.h:70
Main libavfilter public API header.
memory handling functions
const char * desc
Definition: nvenc.c:89
static enum AVSampleFormat formats[]
Definition: avresample.c:163
static int hwdownload_filter_frame(AVFilterLink *link, AVFrame *input)
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
AVBufferRef * hw_frames_ctx
For hwaccel-format frames, this should be a reference to the AVHWFramesContext describing the frame...
Definition: frame.h:434
const char * name
Pad name.
Definition: internal.h:59
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:72
AVFilterLink ** inputs
array of pointers to input links
Definition: avfilter.h:313
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:1180
static const AVFilterPad hwdownload_inputs[]
#define av_cold
Definition: attributes.h:82
AVOptions.
AVFilter ff_vf_hwdownload
static const AVClass hwdownload_class
#define av_log(a,...)
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
av_default_item_name
#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:153
#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:320
#define AV_PIX_FMT_FLAG_HWACCEL
Pixel format is an HW accelerated format.
Definition: pixdesc.h:140
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 fail()
Definition: checkasm.h:81
int av_hwframe_transfer_data(AVFrame *dst, const AVFrame *src, int flags)
Copy data to or from a hw surface.
Definition: hwcontext.c:361
uint64_t flags
Combination of AV_PIX_FMT_FLAG_...
Definition: pixdesc.h:106
enum AVPixelFormat av_pix_fmt_desc_get_id(const AVPixFmtDescriptor *desc)
Definition: pixdesc.c:2241
Transfer the data from the queried hw frame.
Definition: hwcontext.h:331
static int hwdownload_config_output(AVFilterLink *outlink)
Definition: vf_hwdownload.c:87
int ff_formats_ref(AVFilterFormats *f, AVFilterFormats **ref)
Add *ref as a new reference to formats.
Definition: formats.c:440
AVFormatContext * ctx
Definition: movenc.c:48
AVHWFramesContext * hwframes
Definition: vf_hwdownload.c:35
static const AVFilterPad outputs[]
Definition: af_afftfilt.c:386
static int hwdownload_config_input(AVFilterLink *inlink)
Definition: vf_hwdownload.c:65
static const AVFilterPad inputs[]
Definition: af_afftfilt.c:376
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 * data
The data buffer.
Definition: buffer.h:89
Describe the class of an AVClass context structure.
Definition: log.h:67
Filter definition.
Definition: avfilter.h:142
This struct describes a set or pool of "hardware" frames (i.e.
Definition: hwcontext.h:116
refcounted data buffer API
void ff_formats_unref(AVFilterFormats **ref)
If *ref is non-NULL, remove *ref as a reference to the format list it currently points to...
Definition: formats.c:476
const char * name
Filter name.
Definition: avfilter.h:146
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:317
static const AVFilterPad hwdownload_outputs[]
A reference to a data buffer.
Definition: buffer.h:81
AVBufferRef * av_buffer_ref(AVBufferRef *buf)
Create a new reference to an AVBuffer.
Definition: buffer.c:92
static int hwdownload_query_formats(AVFilterContext *avctx)
Definition: vf_hwdownload.c:38
A list of supported formats for one end of a filter link.
Definition: formats.h:64
An instance of a filter.
Definition: avfilter.h:305
AVBufferRef * hwframes_ref
Definition: vf_hwdownload.c:34
#define av_freep(p)
const char * av_get_pix_fmt_name(enum AVPixelFormat pix_fmt)
Return the short name for a pixel format, NULL in case pix_fmt is unknown.
Definition: pixdesc.c:2138
internal API functions
AVPixelFormat
Pixel format.
Definition: pixfmt.h:60
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:580
int av_hwframe_transfer_get_formats(AVBufferRef *hwframe_ref, enum AVHWFrameTransferDirection dir, enum AVPixelFormat **formats, int flags)
Get a list of possible source or target formats usable in av_hwframe_transfer_data().
Definition: hwcontext.c:307
const AVPixFmtDescriptor * av_pix_fmt_desc_next(const AVPixFmtDescriptor *prev)
Iterate over all pixel format descriptors known to libavutil.
Definition: pixdesc.c:2229