itsmattkc
7bfdf8af7c
Includes splitting off core functionality to a base MemoryPool class that can be re-used later if necessary.
96 lines
2.2 KiB
C++
96 lines
2.2 KiB
C++
/***
|
|
|
|
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
|