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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
/*
* Copyright (c) 2022-2025 The Khronos Group Inc.
* Copyright (c) 2022-2025 RasterGrid Kft.
* Modifications Copyright (C) 2024 Advanced Micro Devices, Inc. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*/
#include "../framework/video_objects.h"
class PositiveVideoEncodeH265 : public VkVideoLayerTest {};
TEST_F(PositiveVideoEncodeH265, Basic) {
TEST_DESCRIPTION("Tests basic H.265/HEVC video encode use case for framework verification purposes");
RETURN_IF_SKIP(Init());
const uint32_t dpb_slots = 3;
const uint32_t active_refs = 2;
VideoConfig config = GetConfig(GetConfigsWithReferences(
GetConfigsWithDpbSlots(GetConfigsWithRateControl(GetConfigsEncodeH265()), dpb_slots), active_refs));
if (!config) {
GTEST_SKIP() << "Test requires H.265 encode support with rate control and 3 DPB slots and 2 active references";
}
config.SessionCreateInfo()->maxDpbSlots = dpb_slots;
config.SessionCreateInfo()->maxActiveReferencePictures = active_refs;
VideoContext context(m_device, config);
context.CreateAndBindSessionMemory();
context.CreateResources();
vkt::CommandBuffer& cb = context.CmdBuffer();
auto rc_info = VideoEncodeRateControlInfo(config).SetAnyMode();
for (uint32_t i = 0; i < config.EncodeCaps()->maxRateControlLayers; ++i) {
auto rc_layer = VideoEncodeRateControlLayerInfo(config);
rc_layer->averageBitrate = 128000;
rc_layer->maxBitrate = 128000;
rc_layer->frameRateNumerator = 30;
rc_layer->frameRateDenominator = 1;
rc_info.AddLayer(rc_layer);
}
cb.Begin();
vk::CmdPipelineBarrier2KHR(cb, context.EncodeInput()->LayoutTransition(VK_IMAGE_LAYOUT_VIDEO_ENCODE_SRC_KHR));
vk::CmdPipelineBarrier2KHR(cb, context.Dpb()->LayoutTransition(VK_IMAGE_LAYOUT_VIDEO_ENCODE_DPB_KHR));
cb.BeginVideoCoding(context.Begin().AddResource(-1, 0).AddResource(-1, 1).AddResource(-1, 2));
cb.ControlVideoCoding(context.Control().Reset().RateControl(rc_info).EncodeQualityLevel(0));
cb.EncodeVideo(context.EncodeReferenceFrame(0));
cb.EncodeVideo(context.EncodeFrame(1).AddReferenceFrame(0));
cb.EncodeVideo(context.EncodeReferenceFrame(1).AddReferenceFrame(0));
cb.EncodeVideo(context.EncodeFrame(2));
cb.EncodeVideo(context.EncodeReferenceFrame(2).AddReferenceFrame(0).AddReferenceFrame(1));
cb.EndVideoCoding(context.End());
cb.End();
context.Queue().Submit(cb);
m_device->Wait();
cb.Begin();
cb.BeginVideoCoding(
context.Begin().RateControl(rc_info).AddResource(0, 0).InvalidateSlot(1).AddResource(-1, 1).AddResource(2, 2));
cb.EncodeVideo(context.EncodeFrame(1));
cb.EncodeVideo(context.EncodeReferenceFrame(1).AddReferenceFrame(0).AddReferenceFrame(2));
cb.EncodeVideo(context.EncodeFrame(2).AddReferenceFrame(1));
cb.EndVideoCoding(context.End());
cb.End();
context.Queue().Submit(cb);
m_device->Wait();
}
TEST_F(PositiveVideoEncodeH265, RateControlLayerCount) {
TEST_DESCRIPTION(
"vkCmdBeginVideoCodingKHR / vkCmdControlVideoCodingKHR - H.265 sub-layer count must only match "
"the layer count if the layer count is greater than 1");
RETURN_IF_SKIP(Init());
VideoConfig config = GetConfig(FilterConfigs(GetConfigsWithRateControl(GetConfigsEncodeH265()), [](const VideoConfig& config) {
return config.EncodeCapsH265()->maxSubLayerCount > 1;
}));
if (!config) {
GTEST_SKIP() << "Test requires H.265 encode support with rate control and sub-layer support";
}
VideoContext context(m_device, config);
context.CreateAndBindSessionMemory();
context.CreateResources();
vkt::CommandBuffer& cb = context.CmdBuffer();
auto rc_info = VideoEncodeRateControlInfo(config, true).SetAnyMode();
rc_info.AddLayer(VideoEncodeRateControlLayerInfo(config));
rc_info.CodecInfo().encode_h265.subLayerCount = 2;
cb.Begin();
cb.BeginVideoCoding(context.Begin().RateControl(rc_info));
cb.ControlVideoCoding(context.Control().RateControl(rc_info));
cb.EndVideoCoding(context.End());
cb.End();
}
TEST_F(PositiveVideoEncodeH265, GetEncodedSessionParams) {
TEST_DESCRIPTION("vkGetEncodedVideoSessionParametersKHR - test basic usage");
RETURN_IF_SKIP(Init());
VideoConfig config = GetConfigEncodeH265();
if (!config) {
GTEST_SKIP() << "Test requires H.265 encode support";
}
VideoContext context(m_device, config);
auto h265_info = vku::InitStruct<VkVideoEncodeH265SessionParametersGetInfoKHR>();
h265_info.writeStdVPS = VK_TRUE;
h265_info.writeStdSPS = VK_TRUE;
h265_info.writeStdPPS = VK_TRUE;
auto get_info = vku::InitStruct<VkVideoEncodeSessionParametersGetInfoKHR>(&h265_info);
get_info.videoSessionParameters = context.SessionParams();
auto feedback_info = vku::InitStruct<VkVideoEncodeSessionParametersFeedbackInfoKHR>();
size_t data_size = 0;
// Calling without feedback info and data pointer is legal
vk::GetEncodedVideoSessionParametersKHR(device(), &get_info, nullptr, &data_size, nullptr);
std::vector<uint8_t> data_buffer(data_size);
// Calling without feedback info but data pointer is legal
vk::GetEncodedVideoSessionParametersKHR(device(), &get_info, nullptr, &data_size, data_buffer.data());
// Calling with feedback info not including codec-specific feedback info
vk::GetEncodedVideoSessionParametersKHR(device(), &get_info, &feedback_info, &data_size, nullptr);
}
|
