zqz/platform-external-webrtc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add vp9 QP parser.

Browse Source 
BUG=webrtc:7662

Review-Url: https://codereview.webrtc.org/2891803003
Cr-Commit-Position: refs/heads/master@{#18260}
This commit is contained in:
jianj
2017-05-24 10:00:16 -07:00
committed by Commit bot
parent a0c5d40a9e
commit 20acdf2443
4 changed files with 303 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
webrtc/modules/video_coding/BUILD.gn
View File

@ -122,6 +122,8 @@ rtc_static_library("video_coding_utility") {
"utility/quality_scaler.h",
"utility/vp8_header_parser.cc",
"utility/vp8_header_parser.h",
"utility/vp9_uncompressed_header_parser.cc",
"utility/vp9_uncompressed_header_parser.h",
]
if (!build_with_chromium && is_clang) {

4
webrtc/modules/video_coding/qp_parser.cc
View File

@ -12,6 +12,7 @@
#include "webrtc/common_types.h"
#include "webrtc/modules/video_coding/utility/vp8_header_parser.h"
#include "webrtc/modules/video_coding/utility/vp9_uncompressed_header_parser.h"
namespace webrtc {
@ -20,6 +21,9 @@ bool QpParser::GetQp(const VCMEncodedFrame& frame, int* qp) {
case kVideoCodecVP8:
// QP range: [0, 127].
return vp8::GetQp(frame.Buffer(), frame.Length(), qp);
case kVideoCodecVP9:
// QP range: [0, 255].
return vp9::GetQp(frame.Buffer(), frame.Length(), qp);
default:
return false;
}

236
webrtc/modules/video_coding/utility/vp9_uncompressed_header_parser.cc Normal file
View File

@ -0,0 +1,236 @@
/*
* Copyright (c) 2017 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 "webrtc/modules/video_coding/utility/vp9_uncompressed_header_parser.h"
namespace webrtc {
namespace vp9 {
namespace {
const size_t kVp9MaxProfile = 4;
const size_t kVp9NumRefsPerFrame = 3;
const size_t kVp9MaxRefLFDeltas = 4;
const size_t kVp9MaxModeLFDeltas = 2;
} // namespace
static uint8_t VP9ReadProfile(VP9BitReader* br) {
uint8_t profile = 0;
if (br->GetBit())
profile |= 1;
if (br->GetBit())
profile |= 2;
if (profile > 2 && br->GetBit())
profile += 1;
return profile;
}
static bool VP9ReadColorConfig(VP9BitReader* br, uint8_t profile) {
if (profile == 2 || profile == 3) {
// Bitdepth.
br->GetBit();
}
uint8_t color_space = br->GetValue(3);
// SRGB is 7.
if (color_space != 7) {
// YUV range flag.
br->GetBit();
if (profile == 1 || profile == 3) {
// Subsampling x.
br->GetBit();
// Subsampling y.
br->GetBit();
// Reserved.
if (br->GetBit()) {
LOG(LS_WARNING) << "Failed to get QP. Reserved bit set.";
return false;
}
}
} else {
if (profile == 1 || profile == 3) {
// Reserved.
if (br->GetBit()) {
LOG(LS_WARNING) << "Failed to get QP. Reserved bit set.";
return false;
}
} else {
LOG(LS_WARNING) << "Failed to get QP. 4:4:4 color not supported in "
"profile 0 or 2.";
return false;
}
}
return true;
}
static void VP9ReadFrameSize(VP9BitReader* br) {
// Frame width.
br->GetValue(16);
// Frame height.
br->GetValue(16);
}
static void VP9ReadRenderSize(VP9BitReader* br) {
// Scaling.
if (br->GetBit()) {
// Render width.
br->GetValue(16);
// Render height.
br->GetValue(16);
}
}
static void VP9ReadFrameSizeFromRefs(VP9BitReader* br) {
int found_ref = 0;
for (size_t i = 0; i < kVp9NumRefsPerFrame; i++) {
// Size in refs.
found_ref = br->GetBit();
if (found_ref)
break;
}
if (!found_ref)
VP9ReadFrameSize(br);
VP9ReadRenderSize(br);
}
static void VP9ReadInterpolationFilter(VP9BitReader* br) {
if (br->GetBit())
return;
br->GetValue(2);
}
static void VP9ReadLoopfilter(VP9BitReader* br) {
// Filter level.
br->GetValue(6);
// Sharpness level.
br->GetValue(3);
uint32_t mode_ref_delta_enabled = br->GetBit();
if (mode_ref_delta_enabled) {
uint32_t mode_ref_delta_update = br->GetBit();
if (mode_ref_delta_update) {
for (size_t i = 0; i < kVp9MaxRefLFDeltas; i++) {
if (br->GetBit())
br->GetSignedValue(6);
}
for (size_t i = 0; i < kVp9MaxModeLFDeltas; i++) {
if (br->GetBit())
br->GetSignedValue(6);
}
}
}
}
bool GetQp(const uint8_t* buf, size_t length, int* qp) {
VP9BitReader br(buf, length);
// Frame marker.
if (br.GetValue(2) != 0x2) {
LOG(LS_WARNING) << "Failed to get QP. Frame marker should be 2.";
return false;
}
// Profile.
uint8_t profile = VP9ReadProfile(&br);
if (profile > kVp9MaxProfile) {
LOG(LS_WARNING) << "Failed to get QP. Unsupported bitstream profile: "
<< profile;
return false;
}
// Show existing frame.
if (br.GetBit())
return false;
// Frame type: KEY_FRAME(0), INTER_FRAME(1).
uint8_t frame_type = br.GetBit();
// Show frame.
uint8_t show_frame = br.GetBit();
// Error resilient.
uint8_t error_resilient = br.GetBit();
if (!frame_type) {
// Sync code.
uint32_t sync_code = br.GetValue(24);
if (sync_code != 0x498342) {
LOG(LS_WARNING) << "Failed to get QP. Invalid sync code.";
return false;
}
if (!VP9ReadColorConfig(&br, profile))
return false;
VP9ReadFrameSize(&br);
VP9ReadRenderSize(&br);
} else {
uint8_t intra_only = 0;
if (!show_frame)
intra_only = br.GetBit();
if (!error_resilient)
// Reset frame context.
br.GetValue(2);
if (intra_only) {
// Sync code.
if (br.GetValue(24) != 0x498342) {
LOG(LS_WARNING) << "Failed to get QP. Invalid sync code.";
return false;
}
if (profile > 0) {
if (!VP9ReadColorConfig(&br, profile))
return false;
}
// Refresh frame flags.
br.GetValue(8);
VP9ReadFrameSize(&br);
VP9ReadRenderSize(&br);
} else {
// Refresh frame flags.
br.GetValue(8);
for (size_t i = 0; i < kVp9NumRefsPerFrame; i++) {
// Ref frame index.
br.GetValue(3);
// Ref frame sign biases.
br.GetBit();
}
VP9ReadFrameSizeFromRefs(&br);
// Allow high precision mv.
br.GetBit();
// Interpolation filter.
VP9ReadInterpolationFilter(&br);
}
}
if (!error_resilient) {
// Refresh frame context.
br.GetBit();
// Frame parallel decoding mode.
br.GetBit();
}
// Frame context index.
br.GetValue(2);
VP9ReadLoopfilter(&br);
// Base QP.
const int base_q0 = br.GetValue(8);
*qp = base_q0;
return true;
}
} // namespace vp9
} // namespace webrtc

61
webrtc/modules/video_coding/utility/vp9_uncompressed_header_parser.h Normal file
View File

@ -0,0 +1,61 @@
/*
* Copyright (c) 2017 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.
*/
#ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_VP9_UNCOMPRESSED_HEADER_PARSER_H_
#define WEBRTC_MODULES_VIDEO_CODING_UTILITY_VP9_UNCOMPRESSED_HEADER_PARSER_H_
#include <stddef.h>
#include <stdint.h>
#include "webrtc/base/bitbuffer.h"
#include "webrtc/base/logging.h"
namespace webrtc {
namespace vp9 {
class VP9BitReader : public ::rtc::BitBuffer {
public:
VP9BitReader(const uint8_t* buffer, size_t length_)
: BitBuffer(buffer, length_) {}
uint32_t GetBit() {
uint32_t bit = 0;
if (ReadBits(&bit, 1))
return bit;
LOG(LS_WARNING) << "Failed to get bit. Reached EOF.";
return 0;
}
uint32_t GetValue(int bits) {
uint32_t value = 0;
if (ReadBits(&value, bits))
return value;
LOG(LS_WARNING) << "Failed to get bit. Reached EOF.";
return 0;
}
int32_t GetSignedValue(int bits) {
const int32_t value = static_cast<int>(GetValue(bits));
return GetBit() ? -value : value;
}
};
// Gets the QP, QP range: [0, 255].
// Returns true on success, false otherwise.
bool GetQp(const uint8_t* buf, size_t length, int* qp);
} // namespace vp9
} // namespace webrtc
#endif // WEBRTC_MODULES_VIDEO_CODING_UTILITY_VP9_UNCOMPRESSED_HEADER_PARSER_H_