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

Adds function to calculate squared error sum to libyuv.

Browse Source 
This will be used for frame matching in a follow up CL.

Bug: webrtc:10365
Change-Id: I57bb743dd10a3327a5befceb98b3539e1138448b
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/130510
Commit-Queue: Sebastian Jansson <srte@webrtc.org>
Reviewed-by: Magnus Jedvert <magjed@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#27398}
This commit is contained in:
Sebastian Jansson
2019-04-01 14:12:08 +02:00
committed by Commit Bot
parent de83d0c81a
commit 8ce89ba820
2 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
common_video/libyuv/include/webrtc_libyuv.h
View File

@ -77,6 +77,9 @@ int ConvertFromI420(const VideoFrame& src_frame,
int dst_sample_size,
uint8_t* dst_frame);
double I420SSE(const I420BufferInterface& ref_buffer,
const I420BufferInterface& test_buffer);
// Compute PSNR for an I420 frame (all planes).
// Returns the PSNR in decibel, to a maximum of kInfinitePSNR.
double I420PSNR(const VideoFrame* ref_frame, const VideoFrame* test_frame);

22
common_video/libyuv/webrtc_libyuv.cc
View File

@ -201,6 +201,28 @@ rtc::scoped_refptr<I420ABufferInterface> ScaleI420ABuffer(
return merged_buffer;
}
double I420SSE(const I420BufferInterface& ref_buffer,
const I420BufferInterface& test_buffer) {
RTC_DCHECK_EQ(ref_buffer.width(), test_buffer.width());
RTC_DCHECK_EQ(ref_buffer.height(), test_buffer.height());
const uint64_t width = test_buffer.width();
const uint64_t height = test_buffer.height();
const uint64_t sse_y = libyuv::ComputeSumSquareErrorPlane(
ref_buffer.DataY(), ref_buffer.StrideY(), test_buffer.DataY(),
test_buffer.StrideY(), width, height);
const int width_uv = (width + 1) >> 1;
const int height_uv = (height + 1) >> 1;
const uint64_t sse_u = libyuv::ComputeSumSquareErrorPlane(
ref_buffer.DataU(), ref_buffer.StrideU(), test_buffer.DataU(),
test_buffer.StrideU(), width_uv, height_uv);
const uint64_t sse_v = libyuv::ComputeSumSquareErrorPlane(
ref_buffer.DataV(), ref_buffer.StrideV(), test_buffer.DataV(),
test_buffer.StrideV(), width_uv, height_uv);
const double samples = width * height + 2 * (width_uv * height_uv);
const double sse = sse_y + sse_u + sse_v;
return sse / (samples * 255.0 * 255.0);
}
// Compute PSNR for an I420A frame (all planes). Can upscale test frame.
double I420APSNR(const I420ABufferInterface& ref_buffer,
const I420ABufferInterface& test_buffer) {