ffmpegframepool: implemented functional frame pool

Includes splitting off core functionality to a base MemoryPool class that can be re-used later if necessary.
2020-04-14 02:43:12 +10:00
parent 2eb03bf386
commit 7bfdf8af7c
10 changed files with 396 additions and 260 deletions
@@ -0,0 +1,95 @@
+/***
+
+  Olive - Non-Linear Video Editor
+  Copyright (C) 2019 Olive Team
+
+  This program 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 3 of the License, or
+  (at your option) any later version.
+
+  This program 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, see <http://www.gnu.org/licenses/>.
+
+***/
+
+#include "ffmpegframepool.h"
+
+extern "C" {
+#include <libavutil/imgutils.h>
+}
+
+OLIVE_NAMESPACE_ENTER
+
+FFmpegFramePool::FFmpegFramePool() :
+  width_(0),
+  height_(0),
+  format_(AV_PIX_FMT_NONE)
+{
+}
+
+FFmpegFramePool::ElementPtr FFmpegFramePool::Get(AVFrame *copy)
+{
+  ElementPtr ele = MemoryPool::Get();
+
+  if (ele) {
+    av_image_copy_to_buffer(ele->data(),
+                            GetElementSize(),
+                            copy->data,
+                            copy->linesize,
+                            format_,
+                            width_,
+                            height_,
+                            1);
+  }
+
+  return ele;
+}
+
+void FFmpegFramePool::SetParams(int width, int height, AVPixelFormat format)
+{
+  int old_nb_elements;
+
+  if (IsAllocated()) {
+    old_nb_elements = GetElementCount();
+
+    Destroy();
+  } else {
+    old_nb_elements = 0;
+  }
+
+  width_ = width;
+  height_ = height;
+  format_ = format;
+
+  if (old_nb_elements) {
+    // Re-allocate automatically
+    Allocate(old_nb_elements);
+  }
+}
+
+size_t FFmpegFramePool::GetElementSize()
+{
+  if (width_ == 0 || height_ == 0 || format_ == AV_PIX_FMT_NONE) {
+    return 0;
+  }
+
+  int buf_sz = av_image_get_buffer_size(static_cast<AVPixelFormat>(format_),
+                                        width_,
+                                        height_,
+                                        1);
+
+  if (buf_sz < 0) {
+    qDebug() << "Failed to find buffer size:" << buf_sz;
+    return 0;
+  }
+
+  return buf_sz;
+}
+
+OLIVE_NAMESPACE_EXIT