Chromium Code Reviews| 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..d0ea55145f4364c7daf0c4dde71446ebacb1fabe |
| --- /dev/null |
| +++ b/webrtc/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.cc |
| @@ -0,0 +1,148 @@ |
| +/* |
| + * 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. |
| + */ |
| + |
|
magjed_webrtc
2017/05/08 15:57:36
Remove webrtc/sdk/objc/Framework/Classes/corevideo
|
| +#include "webrtc/sdk/objc/Framework/Classes/Video/corevideo_frame_buffer.h" |
| + |
| +#include "libyuv/convert.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 |