mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
3097008de0
These fields will be used for bitstream validation in upcoming CLs. A new vp9_constants.h file is also added, containing common constants defined by the bitstream spec. Bug: webrtc:12354 Change-Id: If04256d83409069c8bee43ad41aed41c3707dfd3 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/226060 Commit-Queue: Erik Språng <sprang@webrtc.org> Reviewed-by: Philip Eliasson <philipel@webrtc.org> Cr-Commit-Position: refs/heads/master@{#34476}
77 lines
3.3 KiB
C++
77 lines
3.3 KiB
C++
/*
|
|
* Copyright (c) 2021 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "modules/video_coding/utility/vp9_uncompressed_header_parser.h"
|
|
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace vp9 {
|
|
using ::testing::AllOf;
|
|
using ::testing::ElementsAreArray;
|
|
using ::testing::Eq;
|
|
using ::testing::Field;
|
|
using ::testing::Optional;
|
|
|
|
TEST(Vp9UncompressedHeaderParserTest, FrameWithSegmentation) {
|
|
// Uncompressed header from a frame generated with libvpx.
|
|
// Encoded QVGA frame (SL0 of a VGA frame) that includes a segmentation.
|
|
const uint8_t kHeader[] = {
|
|
0x87, 0x01, 0x00, 0x00, 0x02, 0x7e, 0x01, 0xdf, 0x02, 0x7f, 0x01, 0xdf,
|
|
0xc6, 0x87, 0x04, 0x83, 0x83, 0x2e, 0x46, 0x60, 0x20, 0x38, 0x0c, 0x06,
|
|
0x03, 0xcd, 0x80, 0xc0, 0x60, 0x9f, 0xc5, 0x46, 0x00, 0x00, 0x00, 0x00,
|
|
0x2e, 0x73, 0xb7, 0xee, 0x22, 0x06, 0x81, 0x82, 0xd4, 0xef, 0xc3, 0x58,
|
|
0x1f, 0x12, 0xd2, 0x7b, 0x28, 0x1f, 0x80, 0xfc, 0x07, 0xe0, 0x00, 0x00};
|
|
|
|
absl::optional<UncompressedHeader> frame_info =
|
|
ParseUncompressedHeader(kHeader, sizeof(kHeader));
|
|
ASSERT_TRUE(frame_info.has_value());
|
|
|
|
EXPECT_FALSE(frame_info->is_keyframe);
|
|
EXPECT_TRUE(frame_info->error_resilient);
|
|
EXPECT_TRUE(frame_info->show_frame);
|
|
EXPECT_FALSE(frame_info->show_existing_frame);
|
|
EXPECT_EQ(frame_info->base_qp, 185);
|
|
EXPECT_EQ(frame_info->frame_width, 320);
|
|
EXPECT_EQ(frame_info->frame_height, 240);
|
|
EXPECT_EQ(frame_info->render_width, 640);
|
|
EXPECT_EQ(frame_info->render_height, 480);
|
|
EXPECT_TRUE(frame_info->allow_high_precision_mv);
|
|
EXPECT_EQ(frame_info->frame_context_idx, 0u);
|
|
EXPECT_EQ(frame_info->interpolation_filter, InterpolationFilter::kSwitchable);
|
|
EXPECT_EQ(frame_info->is_lossless, false);
|
|
EXPECT_EQ(frame_info->profile, 0);
|
|
EXPECT_THAT(frame_info->reference_buffers, ElementsAreArray({0, 0, 0}));
|
|
EXPECT_THAT(frame_info->reference_buffers_sign_bias,
|
|
ElementsAreArray({false, false, false, false}));
|
|
EXPECT_EQ(frame_info->updated_buffers, 0b10000000);
|
|
EXPECT_EQ(frame_info->tile_cols_log2, 0u);
|
|
EXPECT_EQ(frame_info->tile_rows_log2, 0u);
|
|
EXPECT_THAT(
|
|
frame_info->render_size_position,
|
|
::testing::Optional(AllOf(
|
|
Field(&UncompressedHeader::BitstreamPosition::byte_offset, 8u),
|
|
Field(&UncompressedHeader::BitstreamPosition::bit_offset, 0u))));
|
|
EXPECT_EQ(frame_info->compressed_header_size, 23u);
|
|
EXPECT_EQ(frame_info->uncompressed_header_size, 37u);
|
|
|
|
EXPECT_TRUE(frame_info->segmentation_enabled);
|
|
EXPECT_FALSE(frame_info->segmentation_is_delta);
|
|
EXPECT_THAT(frame_info->segmentation_pred_prob,
|
|
Optional(ElementsAreArray({205, 1, 1})));
|
|
EXPECT_THAT(frame_info->segmentation_tree_probs,
|
|
Optional(ElementsAreArray({255, 255, 128, 1, 128, 128, 128})));
|
|
EXPECT_THAT(frame_info->segmentation_features[1][kSegLvlAlt_Q], Eq(-63));
|
|
EXPECT_THAT(frame_info->segmentation_features[2][kSegLvlAlt_Q], Eq(-81));
|
|
}
|
|
|
|
} // namespace vp9
|
|
} // namespace webrtc
|