]> git.sesse.net Git - ffmpeg/blob - fftools/ffmpeg_hw.c
Merge commit '508378556631dc18d32247b4a4e35703758e1ca9'
[ffmpeg] / fftools / ffmpeg_hw.c
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 <string.h>
20
21 #include "libavutil/avstring.h"
22
23 #include "ffmpeg.h"
24
25 static int nb_hw_devices;
26 static HWDevice **hw_devices;
27
28 static HWDevice *hw_device_get_by_type(enum AVHWDeviceType type)
29 {
30     HWDevice *found = NULL;
31     int i;
32     for (i = 0; i < nb_hw_devices; i++) {
33         if (hw_devices[i]->type == type) {
34             if (found)
35                 return NULL;
36             found = hw_devices[i];
37         }
38     }
39     return found;
40 }
41
42 HWDevice *hw_device_get_by_name(const char *name)
43 {
44     int i;
45     for (i = 0; i < nb_hw_devices; i++) {
46         if (!strcmp(hw_devices[i]->name, name))
47             return hw_devices[i];
48     }
49     return NULL;
50 }
51
52 static HWDevice *hw_device_add(void)
53 {
54     int err;
55     err = av_reallocp_array(&hw_devices, nb_hw_devices + 1,
56                             sizeof(*hw_devices));
57     if (err) {
58         nb_hw_devices = 0;
59         return NULL;
60     }
61     hw_devices[nb_hw_devices] = av_mallocz(sizeof(HWDevice));
62     if (!hw_devices[nb_hw_devices])
63         return NULL;
64     return hw_devices[nb_hw_devices++];
65 }
66
67 static char *hw_device_default_name(enum AVHWDeviceType type)
68 {
69     // Make an automatic name of the form "type%d".  We arbitrarily
70     // limit at 1000 anonymous devices of the same type - there is
71     // probably something else very wrong if you get to this limit.
72     const char *type_name = av_hwdevice_get_type_name(type);
73     char *name;
74     size_t index_pos;
75     int index, index_limit = 1000;
76     index_pos = strlen(type_name);
77     name = av_malloc(index_pos + 4);
78     if (!name)
79         return NULL;
80     for (index = 0; index < index_limit; index++) {
81         snprintf(name, index_pos + 4, "%s%d", type_name, index);
82         if (!hw_device_get_by_name(name))
83             break;
84     }
85     if (index >= index_limit) {
86         av_freep(&name);
87         return NULL;
88     }
89     return name;
90 }
91
92 int hw_device_init_from_string(const char *arg, HWDevice **dev_out)
93 {
94     // "type=name:device,key=value,key2=value2"
95     // "type:device,key=value,key2=value2"
96     // -> av_hwdevice_ctx_create()
97     // "type=name@name"
98     // "type@name"
99     // -> av_hwdevice_ctx_create_derived()
100
101     AVDictionary *options = NULL;
102     char *type_name = NULL, *name = NULL, *device = NULL;
103     enum AVHWDeviceType type;
104     HWDevice *dev, *src;
105     AVBufferRef *device_ref = NULL;
106     int err;
107     const char *errmsg, *p, *q;
108     size_t k;
109
110     k = strcspn(arg, ":=@");
111     p = arg + k;
112
113     type_name = av_strndup(arg, k);
114     if (!type_name) {
115         err = AVERROR(ENOMEM);
116         goto fail;
117     }
118     type = av_hwdevice_find_type_by_name(type_name);
119     if (type == AV_HWDEVICE_TYPE_NONE) {
120         errmsg = "unknown device type";
121         goto invalid;
122     }
123
124     if (*p == '=') {
125         k = strcspn(p + 1, ":@");
126
127         name = av_strndup(p + 1, k);
128         if (!name) {
129             err = AVERROR(ENOMEM);
130             goto fail;
131         }
132         if (hw_device_get_by_name(name)) {
133             errmsg = "named device already exists";
134             goto invalid;
135         }
136
137         p += 1 + k;
138     } else {
139         name = hw_device_default_name(type);
140         if (!name) {
141             err = AVERROR(ENOMEM);
142             goto fail;
143         }
144     }
145
146     if (!*p) {
147         // New device with no parameters.
148         err = av_hwdevice_ctx_create(&device_ref, type,
149                                      NULL, NULL, 0);
150         if (err < 0)
151             goto fail;
152
153     } else if (*p == ':') {
154         // New device with some parameters.
155         ++p;
156         q = strchr(p, ',');
157         if (q) {
158             device = av_strndup(p, q - p);
159             if (!device) {
160                 err = AVERROR(ENOMEM);
161                 goto fail;
162             }
163             err = av_dict_parse_string(&options, q + 1, "=", ",", 0);
164             if (err < 0) {
165                 errmsg = "failed to parse options";
166                 goto invalid;
167             }
168         }
169
170         err = av_hwdevice_ctx_create(&device_ref, type,
171                                      device ? device : p, options, 0);
172         if (err < 0)
173             goto fail;
174
175     } else if (*p == '@') {
176         // Derive from existing device.
177
178         src = hw_device_get_by_name(p + 1);
179         if (!src) {
180             errmsg = "invalid source device name";
181             goto invalid;
182         }
183
184         err = av_hwdevice_ctx_create_derived(&device_ref, type,
185                                              src->device_ref, 0);
186         if (err < 0)
187             goto fail;
188     } else {
189         errmsg = "parse error";
190         goto invalid;
191     }
192
193     dev = hw_device_add();
194     if (!dev) {
195         err = AVERROR(ENOMEM);
196         goto fail;
197     }
198
199     dev->name = name;
200     dev->type = type;
201     dev->device_ref = device_ref;
202
203     if (dev_out)
204         *dev_out = dev;
205
206     name = NULL;
207     err = 0;
208 done:
209     av_freep(&type_name);
210     av_freep(&name);
211     av_freep(&device);
212     av_dict_free(&options);
213     return err;
214 invalid:
215     av_log(NULL, AV_LOG_ERROR,
216            "Invalid device specification \"%s\": %s\n", arg, errmsg);
217     err = AVERROR(EINVAL);
218     goto done;
219 fail:
220     av_log(NULL, AV_LOG_ERROR,
221            "Device creation failed: %d.\n", err);
222     av_buffer_unref(&device_ref);
223     goto done;
224 }
225
226 static int hw_device_init_from_type(enum AVHWDeviceType type,
227                                     const char *device,
228                                     HWDevice **dev_out)
229 {
230     AVBufferRef *device_ref = NULL;
231     HWDevice *dev;
232     char *name;
233     int err;
234
235     name = hw_device_default_name(type);
236     if (!name) {
237         err = AVERROR(ENOMEM);
238         goto fail;
239     }
240
241     err = av_hwdevice_ctx_create(&device_ref, type, device, NULL, 0);
242     if (err < 0) {
243         av_log(NULL, AV_LOG_ERROR,
244                "Device creation failed: %d.\n", err);
245         goto fail;
246     }
247
248     dev = hw_device_add();
249     if (!dev) {
250         err = AVERROR(ENOMEM);
251         goto fail;
252     }
253
254     dev->name = name;
255     dev->type = type;
256     dev->device_ref = device_ref;
257
258     if (dev_out)
259         *dev_out = dev;
260
261     return 0;
262
263 fail:
264     av_freep(&name);
265     av_buffer_unref(&device_ref);
266     return err;
267 }
268
269 void hw_device_free_all(void)
270 {
271     int i;
272     for (i = 0; i < nb_hw_devices; i++) {
273         av_freep(&hw_devices[i]->name);
274         av_buffer_unref(&hw_devices[i]->device_ref);
275         av_freep(&hw_devices[i]);
276     }
277     av_freep(&hw_devices);
278     nb_hw_devices = 0;
279 }
280
281 static HWDevice *hw_device_match_by_codec(const AVCodec *codec)
282 {
283     const AVCodecHWConfig *config;
284     HWDevice *dev;
285     int i;
286     for (i = 0;; i++) {
287         config = avcodec_get_hw_config(codec, i);
288         if (!config)
289             return NULL;
290         if (!(config->methods & AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX))
291             continue;
292         dev = hw_device_get_by_type(config->device_type);
293         if (dev)
294             return dev;
295     }
296 }
297
298 int hw_device_setup_for_decode(InputStream *ist)
299 {
300     const AVCodecHWConfig *config;
301     enum AVHWDeviceType type;
302     HWDevice *dev = NULL;
303     int err, auto_device = 0;
304
305     if (ist->hwaccel_device) {
306         dev = hw_device_get_by_name(ist->hwaccel_device);
307         if (!dev) {
308             if (ist->hwaccel_id == HWACCEL_AUTO) {
309                 auto_device = 1;
310             } else if (ist->hwaccel_id == HWACCEL_GENERIC) {
311                 type = ist->hwaccel_device_type;
312                 err = hw_device_init_from_type(type, ist->hwaccel_device,
313                                                &dev);
314             } else {
315                 // This will be dealt with by API-specific initialisation
316                 // (using hwaccel_device), so nothing further needed here.
317                 return 0;
318             }
319         } else {
320             if (ist->hwaccel_id == HWACCEL_AUTO) {
321                 ist->hwaccel_device_type = dev->type;
322             } else if (ist->hwaccel_device_type != dev->type) {
323                 av_log(ist->dec_ctx, AV_LOG_ERROR, "Invalid hwaccel device "
324                        "specified for decoder: device %s of type %s is not "
325                        "usable with hwaccel %s.\n", dev->name,
326                        av_hwdevice_get_type_name(dev->type),
327                        av_hwdevice_get_type_name(ist->hwaccel_device_type));
328                 return AVERROR(EINVAL);
329             }
330         }
331     } else {
332         if (ist->hwaccel_id == HWACCEL_AUTO) {
333             auto_device = 1;
334         } else if (ist->hwaccel_id == HWACCEL_GENERIC) {
335             type = ist->hwaccel_device_type;
336             dev = hw_device_get_by_type(type);
337             if (!dev)
338                 err = hw_device_init_from_type(type, NULL, &dev);
339         } else {
340             dev = hw_device_match_by_codec(ist->dec);
341             if (!dev) {
342                 // No device for this codec, but not using generic hwaccel
343                 // and therefore may well not need one - ignore.
344                 return 0;
345             }
346         }
347     }
348
349     if (auto_device) {
350         int i;
351         if (!avcodec_get_hw_config(ist->dec, 0)) {
352             // Decoder does not support any hardware devices.
353             return 0;
354         }
355         for (i = 0; !dev; i++) {
356             config = avcodec_get_hw_config(ist->dec, i);
357             if (!config)
358                 break;
359             type = config->device_type;
360             dev = hw_device_get_by_type(type);
361             if (dev) {
362                 av_log(ist->dec_ctx, AV_LOG_INFO, "Using auto "
363                        "hwaccel type %s with existing device %s.\n",
364                        av_hwdevice_get_type_name(type), dev->name);
365             }
366         }
367         for (i = 0; !dev; i++) {
368             config = avcodec_get_hw_config(ist->dec, i);
369             if (!config)
370                 break;
371             type = config->device_type;
372             // Try to make a new device of this type.
373             err = hw_device_init_from_type(type, ist->hwaccel_device,
374                                            &dev);
375             if (err < 0) {
376                 // Can't make a device of this type.
377                 continue;
378             }
379             if (ist->hwaccel_device) {
380                 av_log(ist->dec_ctx, AV_LOG_INFO, "Using auto "
381                        "hwaccel type %s with new device created "
382                        "from %s.\n", av_hwdevice_get_type_name(type),
383                        ist->hwaccel_device);
384             } else {
385                 av_log(ist->dec_ctx, AV_LOG_INFO, "Using auto "
386                        "hwaccel type %s with new default device.\n",
387                        av_hwdevice_get_type_name(type));
388             }
389         }
390         if (dev) {
391             ist->hwaccel_device_type = type;
392         } else {
393             av_log(ist->dec_ctx, AV_LOG_INFO, "Auto hwaccel "
394                    "disabled: no device found.\n");
395             ist->hwaccel_id = HWACCEL_NONE;
396             return 0;
397         }
398     }
399
400     if (!dev) {
401         av_log(ist->dec_ctx, AV_LOG_ERROR, "No device available "
402                "for decoder: device type %s needed for codec %s.\n",
403                av_hwdevice_get_type_name(type), ist->dec->name);
404         return err;
405     }
406
407     ist->dec_ctx->hw_device_ctx = av_buffer_ref(dev->device_ref);
408     if (!ist->dec_ctx->hw_device_ctx)
409         return AVERROR(ENOMEM);
410
411     return 0;
412 }
413
414 int hw_device_setup_for_encode(OutputStream *ost)
415 {
416     HWDevice *dev;
417
418     dev = hw_device_match_by_codec(ost->enc);
419     if (dev) {
420         ost->enc_ctx->hw_device_ctx = av_buffer_ref(dev->device_ref);
421         if (!ost->enc_ctx->hw_device_ctx)
422             return AVERROR(ENOMEM);
423         return 0;
424     } else {
425         // No device required, or no device available.
426         return 0;
427     }
428 }
429
430 static int hwaccel_retrieve_data(AVCodecContext *avctx, AVFrame *input)
431 {
432     InputStream *ist = avctx->opaque;
433     AVFrame *output = NULL;
434     enum AVPixelFormat output_format = ist->hwaccel_output_format;
435     int err;
436
437     if (input->format == output_format) {
438         // Nothing to do.
439         return 0;
440     }
441
442     output = av_frame_alloc();
443     if (!output)
444         return AVERROR(ENOMEM);
445
446     output->format = output_format;
447
448     err = av_hwframe_transfer_data(output, input, 0);
449     if (err < 0) {
450         av_log(avctx, AV_LOG_ERROR, "Failed to transfer data to "
451                "output frame: %d.\n", err);
452         goto fail;
453     }
454
455     err = av_frame_copy_props(output, input);
456     if (err < 0) {
457         av_frame_unref(output);
458         goto fail;
459     }
460
461     av_frame_unref(input);
462     av_frame_move_ref(input, output);
463     av_frame_free(&output);
464
465     return 0;
466
467 fail:
468     av_frame_free(&output);
469     return err;
470 }
471
472 int hwaccel_decode_init(AVCodecContext *avctx)
473 {
474     InputStream *ist = avctx->opaque;
475
476     ist->hwaccel_retrieve_data = &hwaccel_retrieve_data;
477
478     return 0;
479 }