| Index: webrtc/sdk/objc/Framework/Classes/corevideo_frame_buffer.cc
|
| diff --git a/webrtc/sdk/objc/Framework/Classes/corevideo_frame_buffer.cc b/webrtc/sdk/objc/Framework/Classes/corevideo_frame_buffer.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..3e98cb478616c4e400a3ae79b824442f1814ab8c
|
| --- /dev/null
|
| +++ b/webrtc/sdk/objc/Framework/Classes/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/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
|
|
|