mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 22:30:40 +01:00
2efb665e27
Also fix rendering of certain i420 buffers in debug quicklook. Bug: None Change-Id: I793915c3a5a1fcb4cd7b24383d1579655e9a7c28 Reviewed-on: https://webrtc-review.googlesource.com/72080 Reviewed-by: Kári Helgason <kthelgason@webrtc.org> Commit-Queue: Anders Carlsson <andersc@webrtc.org> Cr-Commit-Position: refs/heads/master@{#23005}
159 lines
3.8 KiB
Text
159 lines
3.8 KiB
Text
/*
|
|
* Copyright 2017 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.
|
|
*/
|
|
|
|
#import "RTCI420Buffer+Private.h"
|
|
|
|
#include "api/video/i420_buffer.h"
|
|
|
|
#if !defined(NDEBUG) && defined(WEBRTC_IOS)
|
|
#import <UIKit/UIKit.h>
|
|
#include "third_party/libyuv/include/libyuv.h"
|
|
#endif
|
|
|
|
@implementation RTCI420Buffer {
|
|
@protected
|
|
rtc::scoped_refptr<webrtc::I420BufferInterface> _i420Buffer;
|
|
}
|
|
|
|
- (instancetype)initWithWidth:(int)width height:(int)height {
|
|
if (self = [super init]) {
|
|
_i420Buffer = webrtc::I420Buffer::Create(width, height);
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithWidth:(int)width
|
|
height:(int)height
|
|
dataY:(const uint8_t *)dataY
|
|
dataU:(const uint8_t *)dataU
|
|
dataV:(const uint8_t *)dataV {
|
|
if (self = [super init]) {
|
|
_i420Buffer = webrtc::I420Buffer::Copy(
|
|
width, height, dataY, width, dataU, (width + 1) / 2, dataV, (width + 1) / 2);
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithWidth:(int)width
|
|
height:(int)height
|
|
strideY:(int)strideY
|
|
strideU:(int)strideU
|
|
strideV:(int)strideV {
|
|
if (self = [super init]) {
|
|
_i420Buffer = webrtc::I420Buffer::Create(width, height, strideY, strideU, strideV);
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithFrameBuffer:(rtc::scoped_refptr<webrtc::I420BufferInterface>)i420Buffer {
|
|
if (self = [super init]) {
|
|
_i420Buffer = i420Buffer;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (int)width {
|
|
return _i420Buffer->width();
|
|
}
|
|
|
|
- (int)height {
|
|
return _i420Buffer->height();
|
|
}
|
|
|
|
- (int)strideY {
|
|
return _i420Buffer->StrideY();
|
|
}
|
|
|
|
- (int)strideU {
|
|
return _i420Buffer->StrideU();
|
|
}
|
|
|
|
- (int)strideV {
|
|
return _i420Buffer->StrideV();
|
|
}
|
|
|
|
- (int)chromaWidth {
|
|
return _i420Buffer->ChromaWidth();
|
|
}
|
|
|
|
- (int)chromaHeight {
|
|
return _i420Buffer->ChromaHeight();
|
|
}
|
|
|
|
- (const uint8_t *)dataY {
|
|
return _i420Buffer->DataY();
|
|
}
|
|
|
|
- (const uint8_t *)dataU {
|
|
return _i420Buffer->DataU();
|
|
}
|
|
|
|
- (const uint8_t *)dataV {
|
|
return _i420Buffer->DataV();
|
|
}
|
|
|
|
- (id<RTCI420Buffer>)toI420 {
|
|
return self;
|
|
}
|
|
|
|
#pragma mark - Private
|
|
|
|
- (rtc::scoped_refptr<webrtc::I420BufferInterface>)nativeI420Buffer {
|
|
return _i420Buffer;
|
|
}
|
|
|
|
#pragma mark - Debugging
|
|
|
|
#if !defined(NDEBUG) && defined(WEBRTC_IOS)
|
|
- (id)debugQuickLookObject {
|
|
UIGraphicsBeginImageContext(CGSizeMake(_i420Buffer->width(), _i420Buffer->height()));
|
|
CGContextRef c = UIGraphicsGetCurrentContext();
|
|
uint8_t *ctxData = (uint8_t *)CGBitmapContextGetData(c);
|
|
|
|
libyuv::I420ToARGB(_i420Buffer->DataY(),
|
|
_i420Buffer->StrideY(),
|
|
_i420Buffer->DataU(),
|
|
_i420Buffer->StrideU(),
|
|
_i420Buffer->DataV(),
|
|
_i420Buffer->StrideV(),
|
|
ctxData,
|
|
CGBitmapContextGetBytesPerRow(c),
|
|
CGBitmapContextGetWidth(c),
|
|
CGBitmapContextGetHeight(c));
|
|
|
|
UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
|
|
UIGraphicsEndImageContext();
|
|
|
|
return image;
|
|
}
|
|
#endif
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation RTCMutableI420Buffer
|
|
|
|
- (uint8_t *)mutableDataY {
|
|
return static_cast<webrtc::I420Buffer *>(_i420Buffer.get())->MutableDataY();
|
|
}
|
|
|
|
- (uint8_t *)mutableDataU {
|
|
return static_cast<webrtc::I420Buffer *>(_i420Buffer.get())->MutableDataU();
|
|
}
|
|
|
|
- (uint8_t *)mutableDataV {
|
|
return static_cast<webrtc::I420Buffer *>(_i420Buffer.get())->MutableDataV();
|
|
}
|
|
|
|
@end
|