git.sesse.net Git - ffmpeg/blob - libavfilter/video.c

   1 /*
   2  * This file is part of Libav.
   3  *
   4  * Libav 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  * Libav 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 Libav; 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 #include <stdio.h>
  21
  22 #include "libavutil/buffer.h"
  23 #include "libavutil/hwcontext.h"
  24 #include "libavutil/imgutils.h"
  25 #include "libavutil/mem.h"
  26
  27 #include "avfilter.h"
  28 #include "internal.h"
  29 #include "video.h"
  30
  31 AVFrame *ff_null_get_video_buffer(AVFilterLink *link, int w, int h)
  32 {
  33     return ff_get_video_buffer(link->dst->outputs[0], w, h);
  34 }
  35
  36 /* TODO: set the buffer's priv member to a context structure for the whole
  37  * filter chain.  This will allow for a buffer pool instead of the constant
  38  * alloc & free cycle currently implemented. */
  39 AVFrame *ff_default_get_video_buffer(AVFilterLink *link, int w, int h)
  40 {
  41     AVFrame *frame = av_frame_alloc();
  42     int ret;
  43
  44     if (!frame)
  45         return NULL;
  46
  47     if (link->hw_frames_ctx &&
  48         ((AVHWFramesContext*)link->hw_frames_ctx->data)->format == link->format) {
  49         ret = av_hwframe_get_buffer(link->hw_frames_ctx, frame, 0);
  50     } else {
  51         frame->width  = w;
  52         frame->height = h;
  53         frame->format = link->format;
  54
  55         ret = av_frame_get_buffer(frame, 32);
  56     }
  57     if (ret < 0)
  58         av_frame_free(&frame);
  59
  60     return frame;
  61 }
  62
  63 AVFrame *ff_get_video_buffer(AVFilterLink *link, int w, int h)
  64 {
  65     AVFrame *ret = NULL;
  66
  67     FF_DPRINTF_START(NULL, get_video_buffer); ff_dlog_link(NULL, link, 0);
  68
  69     if (link->dstpad->get_video_buffer)
  70         ret = link->dstpad->get_video_buffer(link, w, h);
  71
  72     if (!ret)
  73         ret = ff_default_get_video_buffer(link, w, h);
  74
  75     return ret;
  76 }