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

Unified Diff: webrtc/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.cc

Issue 2890733003: Reland of Split iOS sdk in to separate targets (Closed)
Patch Set: Created 3 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/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.cc
diff --git a/webrtc/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.cc b/webrtc/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.cc
new file mode 100644
index 0000000000000000000000000000000000000000..9593d13b341223a3f1ca6de95a54cc495fc48a43
--- /dev/null
+++ b/webrtc/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.cc
@@ -0,0 +1,149 @@
+/*
+ * Copyright 2016 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/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.h"
+
+#include "libyuv/convert.h"
+#include "webrtc/api/video/i420_buffer.h"
+#include "webrtc/base/checks.h"
+#include "webrtc/base/logging.h"
+#include "webrtc/common_video/libyuv/include/webrtc_libyuv.h"
+
+namespace webrtc {
+
+CoreVideoFrameBuffer::CoreVideoFrameBuffer(CVPixelBufferRef pixel_buffer,
+ int adapted_width,
+ int adapted_height,
+ int crop_width,
+ int crop_height,
+ int crop_x,
+ int crop_y)
+ : NativeHandleBuffer(pixel_buffer, adapted_width, adapted_height),
+ pixel_buffer_(pixel_buffer),
+ buffer_width_(CVPixelBufferGetWidth(pixel_buffer)),
+ buffer_height_(CVPixelBufferGetHeight(pixel_buffer)),
+ crop_width_(crop_width), crop_height_(crop_height),
+ // Can only crop at even pixels.
+ crop_x_(crop_x & ~1), crop_y_(crop_y & ~1) {
+ CVBufferRetain(pixel_buffer_);
+}
+
+CoreVideoFrameBuffer::CoreVideoFrameBuffer(CVPixelBufferRef pixel_buffer)
+ : NativeHandleBuffer(pixel_buffer,
+ CVPixelBufferGetWidth(pixel_buffer),
+ CVPixelBufferGetHeight(pixel_buffer)),
+ pixel_buffer_(pixel_buffer),
+ buffer_width_(width_), buffer_height_(height_),
+ crop_width_(width_), crop_height_(height_),
+ crop_x_(0), crop_y_(0) {
+ CVBufferRetain(pixel_buffer_);
+}
+
+CoreVideoFrameBuffer::~CoreVideoFrameBuffer() {
+ CVBufferRelease(pixel_buffer_);
+}
+
+rtc::scoped_refptr<VideoFrameBuffer>
+CoreVideoFrameBuffer::NativeToI420Buffer() {
+ const OSType pixel_format = CVPixelBufferGetPixelFormatType(pixel_buffer_);
+ RTC_DCHECK(pixel_format == kCVPixelFormatType_420YpCbCr8BiPlanarFullRange ||
+ pixel_format == kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange);
+
+ CVPixelBufferLockBaseAddress(pixel_buffer_, kCVPixelBufferLock_ReadOnly);
+ const uint8_t* src_y = static_cast<const uint8_t*>(
+ CVPixelBufferGetBaseAddressOfPlane(pixel_buffer_, 0));
+ const int src_y_stride = CVPixelBufferGetBytesPerRowOfPlane(pixel_buffer_, 0);
+ const uint8_t* src_uv = static_cast<const uint8_t*>(
+ CVPixelBufferGetBaseAddressOfPlane(pixel_buffer_, 1));
+ const int src_uv_stride =
+ CVPixelBufferGetBytesPerRowOfPlane(pixel_buffer_, 1);
+
+ // Crop just by modifying pointers.
+ src_y += src_y_stride * crop_y_ + crop_x_;
+ src_uv += src_uv_stride * (crop_y_ / 2) + crop_x_;
+
+ // TODO(magjed): Use a frame buffer pool.
+ NV12ToI420Scaler nv12_to_i420_scaler;
+ rtc::scoped_refptr<I420Buffer> buffer =
+ new rtc::RefCountedObject<I420Buffer>(width_, height_);
+ nv12_to_i420_scaler.NV12ToI420Scale(
+ src_y, src_y_stride,
+ src_uv, src_uv_stride,
+ crop_width_, crop_height_,
+ buffer->MutableDataY(), buffer->StrideY(),
+ buffer->MutableDataU(), buffer->StrideU(),
+ buffer->MutableDataV(), buffer->StrideV(),
+ buffer->width(), buffer->height());
+
+ CVPixelBufferUnlockBaseAddress(pixel_buffer_, kCVPixelBufferLock_ReadOnly);
+
+ return buffer;
+}
+
+bool CoreVideoFrameBuffer::RequiresCropping() const {
+ return crop_width_ != buffer_width_ || crop_height_ != buffer_height_;
+}
+
+bool CoreVideoFrameBuffer::CropAndScaleTo(
+ std::vector<uint8_t>* tmp_buffer,
+ CVPixelBufferRef output_pixel_buffer) const {
+ // Prepare output pointers.
+ RTC_DCHECK_EQ(CVPixelBufferGetPixelFormatType(output_pixel_buffer),
+ kCVPixelFormatType_420YpCbCr8BiPlanarFullRange);
+ CVReturn cv_ret = CVPixelBufferLockBaseAddress(output_pixel_buffer, 0);
+ if (cv_ret != kCVReturnSuccess) {
+ LOG(LS_ERROR) << "Failed to lock base address: " << cv_ret;
+ return false;
+ }
+ const int dst_width = CVPixelBufferGetWidth(output_pixel_buffer);
+ const int dst_height = CVPixelBufferGetHeight(output_pixel_buffer);
+ uint8_t* dst_y = reinterpret_cast<uint8_t*>(
+ CVPixelBufferGetBaseAddressOfPlane(output_pixel_buffer, 0));
+ const int dst_y_stride =
+ CVPixelBufferGetBytesPerRowOfPlane(output_pixel_buffer, 0);
+ uint8_t* dst_uv = reinterpret_cast<uint8_t*>(
+ CVPixelBufferGetBaseAddressOfPlane(output_pixel_buffer, 1));
+ const int dst_uv_stride =
+ CVPixelBufferGetBytesPerRowOfPlane(output_pixel_buffer, 1);
+
+ // Prepare source pointers.
+ const OSType src_pixel_format =
+ CVPixelBufferGetPixelFormatType(pixel_buffer_);
+ RTC_DCHECK(
+ src_pixel_format == kCVPixelFormatType_420YpCbCr8BiPlanarFullRange ||
+ src_pixel_format == kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange);
+ CVPixelBufferLockBaseAddress(pixel_buffer_, kCVPixelBufferLock_ReadOnly);
+ const uint8_t* src_y = static_cast<const uint8_t*>(
+ CVPixelBufferGetBaseAddressOfPlane(pixel_buffer_, 0));
+ const int src_y_stride = CVPixelBufferGetBytesPerRowOfPlane(pixel_buffer_, 0);
+ const uint8_t* src_uv = static_cast<const uint8_t*>(
+ CVPixelBufferGetBaseAddressOfPlane(pixel_buffer_, 1));
+ const int src_uv_stride =
+ CVPixelBufferGetBytesPerRowOfPlane(pixel_buffer_, 1);
+
+ // Crop just by modifying pointers.
+ src_y += src_y_stride * crop_y_ + crop_x_;
+ src_uv += src_uv_stride * (crop_y_ / 2) + crop_x_;
+
+ NV12Scale(tmp_buffer,
+ src_y, src_y_stride,
+ src_uv, src_uv_stride,
+ crop_width_, crop_height_,
+ dst_y, dst_y_stride,
+ dst_uv, dst_uv_stride,
+ dst_width, dst_height);
+
+ CVPixelBufferUnlockBaseAddress(pixel_buffer_, kCVPixelBufferLock_ReadOnly);
+ CVPixelBufferUnlockBaseAddress(output_pixel_buffer, 0);
+
+ return true;
+}
+
+} // namespace webrtc

Powered by Google App Engine
This is Rietveld 408576698