summaryrefslogtreecommitdiff
path: root/src/video_core/host1x/codecs/codec.h
blob: c3622af57e1dec999977cff2731301977fb9209d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// SPDX-FileCopyrightText: Copyright 2020 yuzu Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#pragma once

#include <memory>
#include <optional>
#include <string_view>
#include <queue>
#include "common/common_types.h"
#include "video_core/host1x/ffmpeg/ffmpeg.h"
#include "video_core/host1x/nvdec_common.h"

namespace Tegra {

    namespace Decoder {
        class H264;
        class VP8;
        class VP9;
    } // namespace Decoder

    namespace Host1x {
        class Host1x;
    } // namespace Host1x

    class Codec {
    public:
        explicit Codec(Host1x::Host1x& host1x, 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]] std::unique_ptr<FFmpeg::Frame> 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:
        bool initialized{};
        Host1x::NvdecCommon::VideoCodec current_codec{Host1x::NvdecCommon::VideoCodec::None};
        FFmpeg::DecodeApi decode_api;

        Host1x::Host1x& host1x;
        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<std::unique_ptr<FFmpeg::Frame>> frames{};
    };

} // namespace Tegra