X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Ffilter_graph.cc;h=fc6b9033ac97805e01b4ea45aaeca6dc6fb41208;hp=7af247f669f64f7fb022843f9af994a3a6d5887f;hb=2d4e8c5f69cc694625ad95dcee554499605f823b;hpb=b67a3505b58ffd204ee1a2560be3026688be8f10 diff --git a/src/lib/filter_graph.cc b/src/lib/filter_graph.cc index 7af247f66..fc6b9033a 100644 --- a/src/lib/filter_graph.cc +++ b/src/lib/filter_graph.cc @@ -1,37 +1,36 @@ /* - Copyright (C) 2012 Carl Hetherington + Copyright (C) 2012-2021 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ + /** @file src/lib/filter_graph.cc * @brief A graph of FFmpeg filters. */ + #include "filter_graph.h" #include "filter.h" #include "exceptions.h" #include "image.h" -#include "ffmpeg_content.h" -#include "safe_stringstream.h" #include "compose.hpp" extern "C" { -#include #include -#include #include #include } @@ -39,136 +38,91 @@ extern "C" { #include "i18n.h" + using std::string; using std::list; using std::pair; using std::make_pair; using std::cout; -using boost::shared_ptr; -using boost::weak_ptr; +using std::vector; +using std::shared_ptr; +using std::weak_ptr; using dcp::Size; -/** Construct a FilterGraph for the settings in a piece of content. - * @param content Content. - * @param s Size of the images to process. - * @param p Pixel format of the images to process. - */ -FilterGraph::FilterGraph (shared_ptr content, dcp::Size s, AVPixelFormat p) - : _copy (false) - , _buffer_src_context (0) - , _buffer_sink_context (0) - , _size (s) - , _pixel_format (p) - , _frame (0) + +void +FilterGraph::setup (vector filters) { - string const filters = Filter::ffmpeg_string (content->filters()); - if (filters.empty ()) { + auto const filters_string = Filter::ffmpeg_string (filters); + if (filters.empty()) { _copy = true; return; } _frame = av_frame_alloc (); - AVFilterGraph* graph = avfilter_graph_alloc(); - if (graph == 0) { + _graph = avfilter_graph_alloc(); + if (!_graph) { throw DecodeError (N_("could not create filter graph.")); } - AVFilter* buffer_src = avfilter_get_by_name(N_("buffer")); - if (buffer_src == 0) { + auto const buffer_src = avfilter_get_by_name (src_name().c_str()); + if (!buffer_src) { throw DecodeError (N_("could not find buffer src filter")); } - AVFilter* buffer_sink = avfilter_get_by_name(N_("buffersink")); - if (buffer_sink == 0) { + auto const buffer_sink = avfilter_get_by_name (sink_name().c_str()); + if (!buffer_sink) { throw DecodeError (N_("Could not create buffer sink filter")); } - SafeStringStream a; - a << "video_size=" << _size.width << "x" << _size.height << ":" - << "pix_fmt=" << _pixel_format << ":" - << "time_base=1/1:" - << "pixel_aspect=1/1"; - - if (avfilter_graph_create_filter (&_buffer_src_context, buffer_src, "in", a.str().c_str(), 0, graph) < 0) { + if (avfilter_graph_create_filter (&_buffer_src_context, buffer_src, N_("in"), src_parameters().c_str(), 0, _graph) < 0) { throw DecodeError (N_("could not create buffer source")); } - AVBufferSinkParams* sink_params = av_buffersink_params_alloc (); - AVPixelFormat* pixel_fmts = new AVPixelFormat[2]; - pixel_fmts[0] = _pixel_format; - pixel_fmts[1] = AV_PIX_FMT_NONE; - sink_params->pixel_fmts = pixel_fmts; - - if (avfilter_graph_create_filter (&_buffer_sink_context, buffer_sink, N_("out"), 0, sink_params, graph) < 0) { + if (avfilter_graph_create_filter (&_buffer_sink_context, buffer_sink, N_("out"), nullptr, nullptr, _graph) < 0) { throw DecodeError (N_("could not create buffer sink.")); } - av_free (sink_params); + set_parameters (_buffer_sink_context); - AVFilterInOut* outputs = avfilter_inout_alloc (); + auto outputs = avfilter_inout_alloc (); outputs->name = av_strdup(N_("in")); outputs->filter_ctx = _buffer_src_context; outputs->pad_idx = 0; outputs->next = 0; - AVFilterInOut* inputs = avfilter_inout_alloc (); + auto inputs = avfilter_inout_alloc (); inputs->name = av_strdup(N_("out")); inputs->filter_ctx = _buffer_sink_context; inputs->pad_idx = 0; inputs->next = 0; - if (avfilter_graph_parse (graph, filters.c_str(), inputs, outputs, 0) < 0) { + if (avfilter_graph_parse (_graph, filters_string.c_str(), inputs, outputs, 0) < 0) { throw DecodeError (N_("could not set up filter graph.")); } - if (avfilter_graph_config (graph, 0) < 0) { - throw DecodeError (N_("could not configure filter graph.")); + int e = avfilter_graph_config (_graph, 0); + if (e < 0) { + throw DecodeError (String::compose(N_("could not configure filter graph (%1)"), e)); } } + FilterGraph::~FilterGraph () { if (_frame) { av_frame_free (&_frame); } -} -/** Take an AVFrame and process it using our configured filters, returning a - * set of Images. Caller handles memory management of the input frame. - */ -list, int64_t> > -FilterGraph::process (AVFrame* frame) -{ - list, int64_t> > images; - - if (_copy) { - images.push_back (make_pair (shared_ptr (new Image (frame)), av_frame_get_best_effort_timestamp (frame))); - } else { - int r = av_buffersrc_write_frame (_buffer_src_context, frame); - if (r < 0) { - throw DecodeError (String::compose (N_("could not push buffer into filter chain (%1)."), r)); - } - - while (true) { - if (av_buffersink_get_frame (_buffer_sink_context, _frame) < 0) { - break; - } - - images.push_back (make_pair (shared_ptr (new Image (_frame)), av_frame_get_best_effort_timestamp (_frame))); - av_frame_unref (_frame); - } + if (_graph) { + avfilter_graph_free (&_graph); } - - return images; } -/** @param s Image size. - * @param p Pixel format. - * @return true if this chain can process images with `s' and `p', otherwise false. - */ -bool -FilterGraph::can_process (dcp::Size s, AVPixelFormat p) const + +AVFilterContext * +FilterGraph::get (string name) { - return (_size == s && _pixel_format == p); + return avfilter_graph_get_filter (_graph, name.c_str()); }