summaryrefslogtreecommitdiffstats
path: root/src/video_core/command_classes/nvdec.cpp
blob: ede9466ebe0b71f2990b7ad382f2c3317bfce34a (plain) (blame)
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
// Copyright 2020 yuzu Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#include <bitset>
#include "common/assert.h"
#include "common/bit_util.h"
#include "core/memory.h"
#include "video_core/command_classes/nvdec.h"
#include "video_core/gpu.h"
#include "video_core/memory_manager.h"

namespace Tegra {

Nvdec::Nvdec(GPU& gpu_) : gpu(gpu_), codec(std::make_unique<Codec>(gpu)) {}

Nvdec::~Nvdec() = default;

void Nvdec::ProcessMethod(Nvdec::Method method, const std::vector<u32>& arguments) {
    if (method == Method::SetVideoCodec) {
        codec->StateWrite(static_cast<u32>(method), arguments[0]);
    } else {
        codec->StateWrite(static_cast<u32>(method), static_cast<u64>(arguments[0]) << 8);
    }

    switch (method) {
    case Method::SetVideoCodec:
        codec->SetTargetCodec(static_cast<NvdecCommon::VideoCodec>(arguments[0]));
        break;
    case Method::Execute:
        Execute();
        break;
    }
}

AVFrame* Nvdec::GetFrame() {
    return codec->GetCurrentFrame();
}

const AVFrame* Nvdec::GetFrame() const {
    return codec->GetCurrentFrame();
}

void Nvdec::Execute() {
    switch (codec->GetCurrentCodec()) {
    case NvdecCommon::VideoCodec::H264:
    case NvdecCommon::VideoCodec::Vp9:
        codec->Decode();
        break;
    default:
        UNIMPLEMENTED_MSG("Unknown codec {}", static_cast<u32>(codec->GetCurrentCodec()));
        break;
    }
}

} // namespace Tegra