jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2022, Gregory Bertilson <zaggy1024@gmail.com>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include <LibTest/TestCase.h>
8
9#include <LibVideo/Containers/Matroska/Reader.h>
10#include <LibVideo/VP9/Decoder.h>
11
12static void decode_video(StringView path, size_t expected_frame_count)
13{
14 auto matroska_reader = MUST(Video::Matroska::Reader::from_file(path));
15 u64 video_track = 0;
16 MUST(matroska_reader.for_each_track_of_type(Video::Matroska::TrackEntry::TrackType::Video, [&](Video::Matroska::TrackEntry const& track_entry) -> Video::DecoderErrorOr<IterationDecision> {
17 video_track = track_entry.track_number();
18 return IterationDecision::Break;
19 }));
20 VERIFY(video_track != 0);
21
22 auto iterator = MUST(matroska_reader.create_sample_iterator(video_track));
23 size_t frame_count = 0;
24 Video::VP9::Decoder vp9_decoder;
25
26 while (frame_count <= expected_frame_count) {
27 auto block_result = iterator.next_block();
28 if (block_result.is_error() && block_result.error().category() == Video::DecoderErrorCategory::EndOfStream) {
29 VERIFY(frame_count == expected_frame_count);
30 return;
31 }
32
33 auto block = block_result.release_value();
34 for (auto const& frame : block.frames()) {
35 MUST(vp9_decoder.receive_sample(frame));
36 frame_count++;
37 }
38 }
39
40 VERIFY_NOT_REACHED();
41}
42
43TEST_CASE(webm_in_vp9)
44{
45 decode_video("./vp9_in_webm.webm"sv, 25);
46}
47
48BENCHMARK_CASE(vp9_4k)
49{
50 decode_video("./vp9_4k.webm"sv, 2);
51}
52
53BENCHMARK_CASE(vp9_clamp_reference_mvs)
54{
55 decode_video("./vp9_clamp_reference_mvs.webm"sv, 92);
56}