summaryrefslogtreecommitdiffstats
path: root/src/video_core/host1x/codecs/codec.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/video_core/host1x/codecs/codec.h')
-rw-r--r--src/video_core/host1x/codecs/codec.h81
1 files changed, 81 insertions, 0 deletions
diff --git a/src/video_core/host1x/codecs/codec.h b/src/video_core/host1x/codecs/codec.h
new file mode 100644
index 000000000..117cb3ccd
--- /dev/null
+++ b/src/video_core/host1x/codecs/codec.h
@@ -0,0 +1,81 @@
+// SPDX-FileCopyrightText: Copyright 2020 yuzu Emulator Project
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+#pragma once
+
+#include <memory>
+#include <string_view>
+#include <queue>
+#include "common/common_types.h"
+#include "video_core/host1x/nvdec_common.h"
+
+extern "C" {
+#if defined(__GNUC__) || defined(__clang__)
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wconversion"
+#endif
+#include <libavcodec/avcodec.h>
+#if defined(__GNUC__) || defined(__clang__)
+#pragma GCC diagnostic pop
+#endif
+}
+
+namespace Tegra {
+class GPU;
+
+void AVFrameDeleter(AVFrame* ptr);
+using AVFramePtr = std::unique_ptr<AVFrame, decltype(&AVFrameDeleter)>;
+
+namespace Decoder {
+class H264;
+class VP8;
+class VP9;
+} // namespace Decoder
+
+class Codec {
+public:
+ explicit Codec(GPU& gpu, const Host1x::NvdecCommon::NvdecRegisters& regs);
+ ~Codec();
+
+ /// Initialize the codec, returning success or failure
+ void Initialize();
+
+ /// Sets NVDEC video stream codec
+ void SetTargetCodec(Host1x::NvdecCommon::VideoCodec codec);
+
+ /// Call decoders to construct headers, decode AVFrame with ffmpeg
+ void Decode();
+
+ /// Returns next decoded frame
+ [[nodiscard]] AVFramePtr GetCurrentFrame();
+
+ /// Returns the value of current_codec
+ [[nodiscard]] Host1x::NvdecCommon::VideoCodec GetCurrentCodec() const;
+
+ /// Return name of the current codec
+ [[nodiscard]] std::string_view GetCurrentCodecName() const;
+
+private:
+ void InitializeAvCodecContext();
+
+ void InitializeGpuDecoder();
+
+ bool CreateGpuAvDevice();
+
+ bool initialized{};
+ Host1x::NvdecCommon::VideoCodec current_codec{Host1x::NvdecCommon::VideoCodec::None};
+
+ const AVCodec* av_codec{nullptr};
+ AVCodecContext* av_codec_ctx{nullptr};
+ AVBufferRef* av_gpu_decoder{nullptr};
+
+ GPU& gpu;
+ const Host1x::NvdecCommon::NvdecRegisters& state;
+ std::unique_ptr<Decoder::H264> h264_decoder;
+ std::unique_ptr<Decoder::VP8> vp8_decoder;
+ std::unique_ptr<Decoder::VP9> vp9_decoder;
+
+ std::queue<AVFramePtr> av_frames{};
+};
+
+} // namespace Tegra