| Anders Carlsson | 7bca8ca | 2018-08-30 09:30:29 +0200 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2017 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
| 11 | #import "RTCEncodedImage+Private.h" |
| 12 | |
| 13 | #include "rtc_base/numerics/safe_conversions.h" |
| 14 | |
| 15 | @implementation RTCEncodedImage (Private) |
| 16 | |
| 17 | - (instancetype)initWithNativeEncodedImage:(webrtc::EncodedImage)encodedImage { |
| 18 | if (self = [super init]) { |
| 19 | // Wrap the buffer in NSData without copying, do not take ownership. |
| 20 | self.buffer = [NSData dataWithBytesNoCopy:encodedImage._buffer |
| 21 | length:encodedImage._length |
| 22 | freeWhenDone:NO]; |
| 23 | self.encodedWidth = rtc::dchecked_cast<int32_t>(encodedImage._encodedWidth); |
| 24 | self.encodedHeight = rtc::dchecked_cast<int32_t>(encodedImage._encodedHeight); |
| 25 | self.timeStamp = encodedImage.Timestamp(); |
| 26 | self.captureTimeMs = encodedImage.capture_time_ms_; |
| 27 | self.ntpTimeMs = encodedImage.ntp_time_ms_; |
| 28 | self.flags = encodedImage.timing_.flags; |
| 29 | self.encodeStartMs = encodedImage.timing_.encode_start_ms; |
| 30 | self.encodeFinishMs = encodedImage.timing_.encode_finish_ms; |
| 31 | self.frameType = static_cast<RTCFrameType>(encodedImage._frameType); |
| 32 | self.rotation = static_cast<RTCVideoRotation>(encodedImage.rotation_); |
| 33 | self.completeFrame = encodedImage._completeFrame; |
| 34 | self.qp = @(encodedImage.qp_); |
| 35 | self.contentType = (encodedImage.content_type_ == webrtc::VideoContentType::SCREENSHARE) ? |
| 36 | RTCVideoContentTypeScreenshare : |
| 37 | RTCVideoContentTypeUnspecified; |
| 38 | } |
| 39 | |
| 40 | return self; |
| 41 | } |
| 42 | |
| 43 | - (webrtc::EncodedImage)nativeEncodedImage { |
| 44 | // Return the pointer without copying. |
| 45 | webrtc::EncodedImage encodedImage( |
| 46 | (uint8_t *)self.buffer.bytes, (size_t)self.buffer.length, (size_t)self.buffer.length); |
| 47 | encodedImage._encodedWidth = rtc::dchecked_cast<uint32_t>(self.encodedWidth); |
| 48 | encodedImage._encodedHeight = rtc::dchecked_cast<uint32_t>(self.encodedHeight); |
| 49 | encodedImage.SetTimestamp(self.timeStamp); |
| 50 | encodedImage.capture_time_ms_ = self.captureTimeMs; |
| 51 | encodedImage.ntp_time_ms_ = self.ntpTimeMs; |
| 52 | encodedImage.timing_.flags = self.flags; |
| 53 | encodedImage.timing_.encode_start_ms = self.encodeStartMs; |
| 54 | encodedImage.timing_.encode_finish_ms = self.encodeFinishMs; |
| 55 | encodedImage._frameType = webrtc::FrameType(self.frameType); |
| 56 | encodedImage.rotation_ = webrtc::VideoRotation(self.rotation); |
| 57 | encodedImage._completeFrame = self.completeFrame; |
| 58 | encodedImage.qp_ = self.qp ? self.qp.intValue : -1; |
| 59 | encodedImage.content_type_ = (self.contentType == RTCVideoContentTypeScreenshare) ? |
| 60 | webrtc::VideoContentType::SCREENSHARE : |
| 61 | webrtc::VideoContentType::UNSPECIFIED; |
| 62 | |
| 63 | return encodedImage; |
| 64 | } |
| 65 | |
| 66 | @end |