You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

200 lines
6.2KB

  1. /*
  2. * Copyright (c) 2008 Vitor Sessak
  3. *
  4. * This file is part of Libav.
  5. *
  6. * Libav 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. * Libav 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 Libav; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. /**
  21. * @file
  22. * memory buffer source filter
  23. */
  24. #include "avfilter.h"
  25. #include "buffersrc.h"
  26. #include "vsrc_buffer.h"
  27. #include "libavutil/fifo.h"
  28. #include "libavutil/imgutils.h"
  29. typedef struct {
  30. AVFifoBuffer *fifo;
  31. int h, w;
  32. enum PixelFormat pix_fmt;
  33. AVRational time_base; ///< time_base to set in the output link
  34. AVRational pixel_aspect;
  35. } BufferSourceContext;
  36. #define CHECK_PARAM_CHANGE(s, c, width, height, format)\
  37. if (c->w != width || c->h != height || c->pix_fmt != format) {\
  38. av_log(s, AV_LOG_ERROR, "Changing frame properties on the fly is not supported.\n");\
  39. return AVERROR(EINVAL);\
  40. }
  41. int av_vsrc_buffer_add_frame(AVFilterContext *buffer_filter, AVFrame *frame,
  42. int64_t pts, AVRational pixel_aspect)
  43. {
  44. BufferSourceContext *c = buffer_filter->priv;
  45. AVFilterBufferRef *buf;
  46. int ret;
  47. if (!av_fifo_space(c->fifo) &&
  48. (ret = av_fifo_realloc2(c->fifo, av_fifo_size(c->fifo) +
  49. sizeof(buf))) < 0)
  50. return ret;
  51. CHECK_PARAM_CHANGE(buffer_filter, c, frame->width, frame->height, frame->format);
  52. buf = avfilter_get_video_buffer(buffer_filter->outputs[0], AV_PERM_WRITE,
  53. c->w, c->h);
  54. av_image_copy(buf->data, buf->linesize, frame->data, frame->linesize,
  55. c->pix_fmt, c->w, c->h);
  56. avfilter_copy_frame_props(buf, frame);
  57. buf->pts = pts;
  58. buf->video->pixel_aspect = pixel_aspect;
  59. if ((ret = av_fifo_generic_write(c->fifo, &buf, sizeof(buf), NULL)) < 0) {
  60. avfilter_unref_buffer(buf);
  61. return ret;
  62. }
  63. return 0;
  64. }
  65. int av_buffersrc_buffer(AVFilterContext *s, AVFilterBufferRef *buf)
  66. {
  67. BufferSourceContext *c = s->priv;
  68. int ret;
  69. if (!av_fifo_space(c->fifo) &&
  70. (ret = av_fifo_realloc2(c->fifo, av_fifo_size(c->fifo) +
  71. sizeof(buf))) < 0)
  72. return ret;
  73. CHECK_PARAM_CHANGE(s, c, buf->video->w, buf->video->h, buf->format);
  74. if ((ret = av_fifo_generic_write(c->fifo, &buf, sizeof(buf), NULL)) < 0)
  75. return ret;
  76. return 0;
  77. }
  78. static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
  79. {
  80. BufferSourceContext *c = ctx->priv;
  81. char pix_fmt_str[128];
  82. int n = 0;
  83. if (!args ||
  84. (n = sscanf(args, "%d:%d:%127[^:]:%d:%d:%d:%d", &c->w, &c->h, pix_fmt_str,
  85. &c->time_base.num, &c->time_base.den,
  86. &c->pixel_aspect.num, &c->pixel_aspect.den)) != 7) {
  87. av_log(ctx, AV_LOG_ERROR, "Expected 7 arguments, but %d found in '%s'\n", n, args);
  88. return AVERROR(EINVAL);
  89. }
  90. if ((c->pix_fmt = av_get_pix_fmt(pix_fmt_str)) == PIX_FMT_NONE) {
  91. char *tail;
  92. c->pix_fmt = strtol(pix_fmt_str, &tail, 10);
  93. if (*tail || c->pix_fmt < 0 || c->pix_fmt >= PIX_FMT_NB) {
  94. av_log(ctx, AV_LOG_ERROR, "Invalid pixel format string '%s'\n", pix_fmt_str);
  95. return AVERROR(EINVAL);
  96. }
  97. }
  98. if (!(c->fifo = av_fifo_alloc(sizeof(AVFilterBufferRef*))))
  99. return AVERROR(ENOMEM);
  100. av_log(ctx, AV_LOG_INFO, "w:%d h:%d pixfmt:%s\n", c->w, c->h, av_pix_fmt_descriptors[c->pix_fmt].name);
  101. return 0;
  102. }
  103. static av_cold void uninit(AVFilterContext *ctx)
  104. {
  105. BufferSourceContext *s = ctx->priv;
  106. while (av_fifo_size(s->fifo)) {
  107. AVFilterBufferRef *buf;
  108. av_fifo_generic_read(s->fifo, &buf, sizeof(buf), NULL);
  109. avfilter_unref_buffer(buf);
  110. }
  111. av_fifo_free(s->fifo);
  112. s->fifo = NULL;
  113. }
  114. static int query_formats(AVFilterContext *ctx)
  115. {
  116. BufferSourceContext *c = ctx->priv;
  117. enum PixelFormat pix_fmts[] = { c->pix_fmt, PIX_FMT_NONE };
  118. avfilter_set_common_formats(ctx, avfilter_make_format_list(pix_fmts));
  119. return 0;
  120. }
  121. static int config_props(AVFilterLink *link)
  122. {
  123. BufferSourceContext *c = link->src->priv;
  124. link->w = c->w;
  125. link->h = c->h;
  126. link->sample_aspect_ratio = c->pixel_aspect;
  127. link->time_base = c->time_base;
  128. return 0;
  129. }
  130. static int request_frame(AVFilterLink *link)
  131. {
  132. BufferSourceContext *c = link->src->priv;
  133. AVFilterBufferRef *buf;
  134. if (!av_fifo_size(c->fifo)) {
  135. av_log(link->src, AV_LOG_ERROR,
  136. "request_frame() called with no available frame!\n");
  137. return AVERROR(EINVAL);
  138. }
  139. av_fifo_generic_read(c->fifo, &buf, sizeof(buf), NULL);
  140. avfilter_start_frame(link, avfilter_ref_buffer(buf, ~0));
  141. avfilter_draw_slice(link, 0, link->h, 1);
  142. avfilter_end_frame(link);
  143. avfilter_unref_buffer(buf);
  144. return 0;
  145. }
  146. static int poll_frame(AVFilterLink *link)
  147. {
  148. BufferSourceContext *c = link->src->priv;
  149. return !!av_fifo_size(c->fifo);
  150. }
  151. AVFilter avfilter_vsrc_buffer = {
  152. .name = "buffer",
  153. .description = NULL_IF_CONFIG_SMALL("Buffer video frames, and make them accessible to the filterchain."),
  154. .priv_size = sizeof(BufferSourceContext),
  155. .query_formats = query_formats,
  156. .init = init,
  157. .uninit = uninit,
  158. .inputs = (AVFilterPad[]) {{ .name = NULL }},
  159. .outputs = (AVFilterPad[]) {{ .name = "default",
  160. .type = AVMEDIA_TYPE_VIDEO,
  161. .request_frame = request_frame,
  162. .poll_frame = poll_frame,
  163. .config_props = config_props, },
  164. { .name = NULL}},
  165. };