Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(152)

Unified Diff: webrtc/common_video/video_frame_buffer.cc

Issue 2020593002: Refactor scaling. (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Include <algorithm>, needed for std::min. Created 4 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: webrtc/common_video/video_frame_buffer.cc
diff --git a/webrtc/common_video/video_frame_buffer.cc b/webrtc/common_video/video_frame_buffer.cc
index 34213e4656ce766f5cb6f2fdc24e566c02cf152e..2013596fcbf2dfcc933431c982118c730b0608c5 100644
--- a/webrtc/common_video/video_frame_buffer.cc
+++ b/webrtc/common_video/video_frame_buffer.cc
@@ -8,11 +8,14 @@
* be found in the AUTHORS file in the root of the source tree.
*/
+#include <algorithm>
+
#include "webrtc/common_video/include/video_frame_buffer.h"
#include "webrtc/base/checks.h"
#include "webrtc/base/keep_ref_until_done.h"
#include "libyuv/convert.h"
+#include "libyuv/scale.h"
// Aligning pointer to 64 bytes for improved performance, e.g. use SIMD.
static const int kBufferAlignment = 64;
@@ -200,6 +203,57 @@ rtc::scoped_refptr<I420Buffer> I420Buffer::Copy(
return copy;
}
+void I420Buffer::CropAndScale(
+ const rtc::scoped_refptr<VideoFrameBuffer>& buffer,
+ int offset_x,
+ int offset_y,
+ int crop_width,
+ int crop_height) {
+ RTC_CHECK_LE(crop_width, buffer->width());
+ RTC_CHECK_LE(crop_height, buffer->height());
+ RTC_CHECK_LE(crop_width + offset_x, buffer->width());
+ RTC_CHECK_LE(crop_height + offset_y, buffer->height());
+ RTC_CHECK_GE(offset_x, 0);
+ RTC_CHECK_GE(offset_y, 0);
+
+ // Make sure offset is even so that u/v plane becomes aligned.
+ const int uv_offset_x = offset_x / 2;
+ const int uv_offset_y = offset_y / 2;
+ offset_x = uv_offset_x * 2;
+ offset_y = uv_offset_y * 2;
+
+ const uint8_t* y_plane =
+ buffer->DataY() + buffer->StrideY() * offset_y + offset_x;
+ const uint8_t* u_plane =
+ buffer->DataU() + buffer->StrideU() * uv_offset_y + uv_offset_x;
+ const uint8_t* v_plane =
+ buffer->DataV() + buffer->StrideV() * uv_offset_y + uv_offset_x;
+ int res = libyuv::I420Scale(y_plane, buffer->StrideY(),
+ u_plane, buffer->StrideU(),
+ v_plane, buffer->StrideV(),
+ crop_width, crop_height,
+ MutableDataY(), StrideY(),
+ MutableDataU(), StrideU(),
+ MutableDataV(), StrideV(),
+ width(), height(), libyuv::kFilterBox);
+
+ RTC_DCHECK(res == 0);
magjed_webrtc 2016/05/30 11:58:26 Use RTC_DCHECK_EQ instead
nisse-webrtc 2016/05/30 12:58:22 Done.
+}
+
+void I420Buffer::CenterCropAndScale(
+ const rtc::scoped_refptr<I420Buffer>& dst,
+ const rtc::scoped_refptr<VideoFrameBuffer>& src) {
+ const int crop_width =
+ std::min(src->width(), dst->width() * src->height() / dst->height());
+ const int crop_height =
+ std::min(src->height(), dst->height() * src->width() / dst->width());
+
+ dst->CropAndScale(
+ src,
+ (src->width() - crop_width) / 2, (src->height() - crop_height) / 2,
+ crop_width, crop_height);
+}
+
NativeHandleBuffer::NativeHandleBuffer(void* native_handle,
int width,
int height)

Powered by Google App Engine
This is Rietveld 408576698