aboutsummaryrefslogtreecommitdiff
path: root/talk/app/webrtc/objc/avfoundationvideocapturer.mm
diff options
context:
space:
mode:
authorChih-hung Hsieh <chh@google.com>2015-12-01 17:07:48 +0000
committerandroid-build-merger <android-build-merger@google.com>2015-12-01 17:07:48 +0000
commita4acd9d6bc9b3b033d7d274316e75ee067df8d20 (patch)
tree672a185b294789cf991f385c3e395dd63bea9063 /talk/app/webrtc/objc/avfoundationvideocapturer.mm
parent3681b90ba4fe7a27232dd3e27897d5d7ed9d651c (diff)
parentfe8b4a657979b49e1701bd92f6d5814a99e0b2be (diff)
downloadwebrtc-a4acd9d6bc9b3b033d7d274316e75ee067df8d20.tar.gz
Merge changes I7bbf776e,I1b827825
am: fe8b4a6579 * commit 'fe8b4a657979b49e1701bd92f6d5814a99e0b2be': (7237 commits) WIP: Changes after merge commit 'cb3f9bd' Make the nonlinear beamformer steerable Utilize bitrate above codec max to protect video. Enable VP9 internal resize by default. Filter overlapping RTP header extensions. Make VCMEncodedFrameCallback const. MediaCodecVideoEncoder: Add number of quality resolution downscales to Encoded callback. Remove redudant encoder rate calls. Create isolate files for nonparallel tests. Register header extensions in RtpRtcpObserver to avoid log spam. Make an enum class out of NetEqDecoder, and hide the neteq_decoders_ table ACM: Move NACK functionality inside NetEq Fix chromium-style warnings in webrtc/sound/. Create a 'webrtc_nonparallel_tests' target. Update scalability structure data according to updates in the RTP payload profile. audio_coding: rename interface -> include Rewrote perform_action_on_all_files to be parallell. Update reference indices according to updates in the RTP payload profile. Disable P2PTransport...TestFailoverControlledSide on Memcheck pass clangcl compile options to ignore warnings in gflags.cc ...
Diffstat (limited to 'talk/app/webrtc/objc/avfoundationvideocapturer.mm')
-rw-r--r--talk/app/webrtc/objc/avfoundationvideocapturer.mm447
1 files changed, 447 insertions, 0 deletions
diff --git a/talk/app/webrtc/objc/avfoundationvideocapturer.mm b/talk/app/webrtc/objc/avfoundationvideocapturer.mm
new file mode 100644
index 0000000000..e1b0f88fb6
--- /dev/null
+++ b/talk/app/webrtc/objc/avfoundationvideocapturer.mm
@@ -0,0 +1,447 @@
+/*
+ * libjingle
+ * Copyright 2015 Google Inc.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ * 3. The name of the author may not be used to endorse or promote products
+ * derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
+ * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
+ * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+ * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+ * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+ * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "talk/app/webrtc/objc/avfoundationvideocapturer.h"
+
+#include "webrtc/base/bind.h"
+
+#import <AVFoundation/AVFoundation.h>
+#import <Foundation/Foundation.h>
+#import <UIKit/UIKit.h>
+
+// TODO(tkchin): support other formats.
+static NSString* const kDefaultPreset = AVCaptureSessionPreset640x480;
+static cricket::VideoFormat const kDefaultFormat =
+ cricket::VideoFormat(640,
+ 480,
+ cricket::VideoFormat::FpsToInterval(30),
+ cricket::FOURCC_NV12);
+
+// This queue is used to start and stop the capturer without blocking the
+// calling thread. -[AVCaptureSession startRunning] blocks until the camera is
+// running.
+static dispatch_queue_t kBackgroundQueue = nil;
+
+// This class used to capture frames using AVFoundation APIs on iOS. It is meant
+// to be owned by an instance of AVFoundationVideoCapturer. The reason for this
+// because other webrtc objects own cricket::VideoCapturer, which is not
+// ref counted. To prevent bad behavior we do not expose this class directly.
+@interface RTCAVFoundationVideoCapturerInternal : NSObject
+ <AVCaptureVideoDataOutputSampleBufferDelegate>
+
+@property(nonatomic, readonly) AVCaptureSession* captureSession;
+@property(nonatomic, readonly) BOOL isRunning;
+@property(nonatomic, assign) BOOL useBackCamera; // Defaults to NO.
+
+// We keep a pointer back to AVFoundationVideoCapturer to make callbacks on it
+// when we receive frames. This is safe because this object should be owned by
+// it.
+- (instancetype)initWithCapturer:(webrtc::AVFoundationVideoCapturer*)capturer;
+- (void)startCaptureAsync;
+- (void)stopCaptureAsync;
+
+@end
+
+@implementation RTCAVFoundationVideoCapturerInternal {
+ // Keep pointers to inputs for convenience.
+ AVCaptureDeviceInput* _frontDeviceInput;
+ AVCaptureDeviceInput* _backDeviceInput;
+ AVCaptureVideoDataOutput* _videoOutput;
+ // The cricket::VideoCapturer that owns this class. Should never be NULL.
+ webrtc::AVFoundationVideoCapturer* _capturer;
+ BOOL _orientationHasChanged;
+}
+
+@synthesize captureSession = _captureSession;
+@synthesize useBackCamera = _useBackCamera;
+@synthesize isRunning = _isRunning;
+
++ (void)initialize {
+ static dispatch_once_t onceToken;
+ dispatch_once(&onceToken, ^{
+ kBackgroundQueue = dispatch_queue_create(
+ "com.google.webrtc.RTCAVFoundationCapturerBackground",
+ DISPATCH_QUEUE_SERIAL);
+ });
+}
+
+- (instancetype)initWithCapturer:(webrtc::AVFoundationVideoCapturer*)capturer {
+ NSParameterAssert(capturer);
+ if (self = [super init]) {
+ _capturer = capturer;
+ if (![self setupCaptureSession]) {
+ return nil;
+ }
+ NSNotificationCenter* center = [NSNotificationCenter defaultCenter];
+ [center addObserver:self
+ selector:@selector(deviceOrientationDidChange:)
+ name:UIDeviceOrientationDidChangeNotification
+ object:nil];
+ [center addObserverForName:AVCaptureSessionRuntimeErrorNotification
+ object:nil
+ queue:nil
+ usingBlock:^(NSNotification* notification) {
+ NSLog(@"Capture session error: %@", notification.userInfo);
+ }];
+ }
+ return self;
+}
+
+- (void)dealloc {
+ [self stopCaptureAsync];
+ [[NSNotificationCenter defaultCenter] removeObserver:self];
+ _capturer = nullptr;
+}
+
+- (void)setUseBackCamera:(BOOL)useBackCamera {
+ if (_useBackCamera == useBackCamera) {
+ return;
+ }
+ _useBackCamera = useBackCamera;
+ [self updateSessionInput];
+}
+
+- (void)startCaptureAsync {
+ if (_isRunning) {
+ return;
+ }
+ _orientationHasChanged = NO;
+ [[UIDevice currentDevice] beginGeneratingDeviceOrientationNotifications];
+ AVCaptureSession* session = _captureSession;
+ dispatch_async(kBackgroundQueue, ^{
+ [session startRunning];
+ });
+ _isRunning = YES;
+}
+
+- (void)stopCaptureAsync {
+ if (!_isRunning) {
+ return;
+ }
+ [_videoOutput setSampleBufferDelegate:nil queue:nullptr];
+ AVCaptureSession* session = _captureSession;
+ dispatch_async(kBackgroundQueue, ^{
+ [session stopRunning];
+ });
+ [[UIDevice currentDevice] endGeneratingDeviceOrientationNotifications];
+ _isRunning = NO;
+}
+
+#pragma mark AVCaptureVideoDataOutputSampleBufferDelegate
+
+- (void)captureOutput:(AVCaptureOutput*)captureOutput
+ didOutputSampleBuffer:(CMSampleBufferRef)sampleBuffer
+ fromConnection:(AVCaptureConnection*)connection {
+ NSParameterAssert(captureOutput == _videoOutput);
+ if (!_isRunning) {
+ return;
+ }
+ _capturer->CaptureSampleBuffer(sampleBuffer);
+}
+
+- (void)captureOutput:(AVCaptureOutput*)captureOutput
+ didDropSampleBuffer:(CMSampleBufferRef)sampleBuffer
+ fromConnection:(AVCaptureConnection*)connection {
+ NSLog(@"Dropped sample buffer.");
+}
+
+#pragma mark - Private
+
+- (BOOL)setupCaptureSession {
+ _captureSession = [[AVCaptureSession alloc] init];
+#if defined(__IPHONE_7_0) && __IPHONE_OS_VERSION_MAX_ALLOWED >= __IPHONE_7_0
+ NSString* version = [[UIDevice currentDevice] systemVersion];
+ if ([version integerValue] >= 7) {
+ _captureSession.usesApplicationAudioSession = NO;
+ }
+#endif
+ if (![_captureSession canSetSessionPreset:kDefaultPreset]) {
+ NSLog(@"Default video capture preset unsupported.");
+ return NO;
+ }
+ _captureSession.sessionPreset = kDefaultPreset;
+
+ // Make the capturer output NV12. Ideally we want I420 but that's not
+ // currently supported on iPhone / iPad.
+ _videoOutput = [[AVCaptureVideoDataOutput alloc] init];
+ _videoOutput.videoSettings = @{
+ (NSString*)kCVPixelBufferPixelFormatTypeKey :
+ @(kCVPixelFormatType_420YpCbCr8BiPlanarFullRange)
+ };
+ _videoOutput.alwaysDiscardsLateVideoFrames = NO;
+ [_videoOutput setSampleBufferDelegate:self
+ queue:dispatch_get_main_queue()];
+ if (![_captureSession canAddOutput:_videoOutput]) {
+ NSLog(@"Default video capture output unsupported.");
+ return NO;
+ }
+ [_captureSession addOutput:_videoOutput];
+
+ // Find the capture devices.
+ AVCaptureDevice* frontCaptureDevice = nil;
+ AVCaptureDevice* backCaptureDevice = nil;
+ for (AVCaptureDevice* captureDevice in
+ [AVCaptureDevice devicesWithMediaType:AVMediaTypeVideo]) {
+ if (captureDevice.position == AVCaptureDevicePositionBack) {
+ backCaptureDevice = captureDevice;
+ }
+ if (captureDevice.position == AVCaptureDevicePositionFront) {
+ frontCaptureDevice = captureDevice;
+ }
+ }
+ if (!frontCaptureDevice || !backCaptureDevice) {
+ NSLog(@"Failed to get capture devices.");
+ return NO;
+ }
+
+ // Set up the session inputs.
+ NSError* error = nil;
+ _frontDeviceInput =
+ [AVCaptureDeviceInput deviceInputWithDevice:frontCaptureDevice
+ error:&error];
+ if (!_frontDeviceInput) {
+ NSLog(@"Failed to get capture device input: %@",
+ error.localizedDescription);
+ return NO;
+ }
+ _backDeviceInput =
+ [AVCaptureDeviceInput deviceInputWithDevice:backCaptureDevice
+ error:&error];
+ if (!_backDeviceInput) {
+ NSLog(@"Failed to get capture device input: %@",
+ error.localizedDescription);
+ return NO;
+ }
+
+ // Add the inputs.
+ if (![_captureSession canAddInput:_frontDeviceInput] ||
+ ![_captureSession canAddInput:_backDeviceInput]) {
+ NSLog(@"Session does not support capture inputs.");
+ return NO;
+ }
+ [self updateSessionInput];
+
+ return YES;
+}
+
+- (void)deviceOrientationDidChange:(NSNotification*)notification {
+ _orientationHasChanged = YES;
+ [self updateOrientation];
+}
+
+- (void)updateOrientation {
+ AVCaptureConnection* connection =
+ [_videoOutput connectionWithMediaType:AVMediaTypeVideo];
+ if (!connection.supportsVideoOrientation) {
+ // TODO(tkchin): set rotation bit on frames.
+ return;
+ }
+ AVCaptureVideoOrientation orientation = AVCaptureVideoOrientationPortrait;
+ switch ([UIDevice currentDevice].orientation) {
+ case UIDeviceOrientationPortrait:
+ orientation = AVCaptureVideoOrientationPortrait;
+ break;
+ case UIDeviceOrientationPortraitUpsideDown:
+ orientation = AVCaptureVideoOrientationPortraitUpsideDown;
+ break;
+ case UIDeviceOrientationLandscapeLeft:
+ orientation = AVCaptureVideoOrientationLandscapeRight;
+ break;
+ case UIDeviceOrientationLandscapeRight:
+ orientation = AVCaptureVideoOrientationLandscapeLeft;
+ break;
+ case UIDeviceOrientationFaceUp:
+ case UIDeviceOrientationFaceDown:
+ case UIDeviceOrientationUnknown:
+ if (!_orientationHasChanged) {
+ connection.videoOrientation = orientation;
+ }
+ return;
+ }
+ connection.videoOrientation = orientation;
+}
+
+- (void)updateSessionInput {
+ // Update the current session input to match what's stored in _useBackCamera.
+ [_captureSession beginConfiguration];
+ AVCaptureDeviceInput* oldInput = _backDeviceInput;
+ AVCaptureDeviceInput* newInput = _frontDeviceInput;
+ if (_useBackCamera) {
+ oldInput = _frontDeviceInput;
+ newInput = _backDeviceInput;
+ }
+ // Ok to remove this even if it's not attached. Will be no-op.
+ [_captureSession removeInput:oldInput];
+ [_captureSession addInput:newInput];
+ [self updateOrientation];
+ [_captureSession commitConfiguration];
+}
+
+@end
+
+namespace webrtc {
+
+AVFoundationVideoCapturer::AVFoundationVideoCapturer()
+ : _capturer(nil), _startThread(nullptr) {
+ // Set our supported formats. This matches kDefaultPreset.
+ std::vector<cricket::VideoFormat> supportedFormats;
+ supportedFormats.push_back(cricket::VideoFormat(kDefaultFormat));
+ SetSupportedFormats(supportedFormats);
+ _capturer =
+ [[RTCAVFoundationVideoCapturerInternal alloc] initWithCapturer:this];
+}
+
+AVFoundationVideoCapturer::~AVFoundationVideoCapturer() {
+ _capturer = nil;
+}
+
+cricket::CaptureState AVFoundationVideoCapturer::Start(
+ const cricket::VideoFormat& format) {
+ if (!_capturer) {
+ LOG(LS_ERROR) << "Failed to create AVFoundation capturer.";
+ return cricket::CaptureState::CS_FAILED;
+ }
+ if (_capturer.isRunning) {
+ LOG(LS_ERROR) << "The capturer is already running.";
+ return cricket::CaptureState::CS_FAILED;
+ }
+ if (format != kDefaultFormat) {
+ LOG(LS_ERROR) << "Unsupported format provided.";
+ return cricket::CaptureState::CS_FAILED;
+ }
+
+ // Keep track of which thread capture started on. This is the thread that
+ // frames need to be sent to.
+ RTC_DCHECK(!_startThread);
+ _startThread = rtc::Thread::Current();
+
+ SetCaptureFormat(&format);
+ // This isn't super accurate because it takes a while for the AVCaptureSession
+ // to spin up, and this call returns async.
+ // TODO(tkchin): make this better.
+ [_capturer startCaptureAsync];
+ SetCaptureState(cricket::CaptureState::CS_RUNNING);
+
+ return cricket::CaptureState::CS_STARTING;
+}
+
+void AVFoundationVideoCapturer::Stop() {
+ [_capturer stopCaptureAsync];
+ SetCaptureFormat(NULL);
+ _startThread = nullptr;
+}
+
+bool AVFoundationVideoCapturer::IsRunning() {
+ return _capturer.isRunning;
+}
+
+AVCaptureSession* AVFoundationVideoCapturer::GetCaptureSession() {
+ return _capturer.captureSession;
+}
+
+void AVFoundationVideoCapturer::SetUseBackCamera(bool useBackCamera) {
+ _capturer.useBackCamera = useBackCamera;
+}
+
+bool AVFoundationVideoCapturer::GetUseBackCamera() const {
+ return _capturer.useBackCamera;
+}
+
+void AVFoundationVideoCapturer::CaptureSampleBuffer(
+ CMSampleBufferRef sampleBuffer) {
+ if (CMSampleBufferGetNumSamples(sampleBuffer) != 1 ||
+ !CMSampleBufferIsValid(sampleBuffer) ||
+ !CMSampleBufferDataIsReady(sampleBuffer)) {
+ return;
+ }
+
+ CVImageBufferRef imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer);
+ if (imageBuffer == NULL) {
+ return;
+ }
+
+ // Base address must be unlocked to access frame data.
+ CVOptionFlags lockFlags = kCVPixelBufferLock_ReadOnly;
+ CVReturn ret = CVPixelBufferLockBaseAddress(imageBuffer, lockFlags);
+ if (ret != kCVReturnSuccess) {
+ return;
+ }
+
+ static size_t const kYPlaneIndex = 0;
+ static size_t const kUVPlaneIndex = 1;
+ uint8_t* yPlaneAddress =
+ (uint8_t*)CVPixelBufferGetBaseAddressOfPlane(imageBuffer, kYPlaneIndex);
+ size_t yPlaneHeight =
+ CVPixelBufferGetHeightOfPlane(imageBuffer, kYPlaneIndex);
+ size_t yPlaneWidth =
+ CVPixelBufferGetWidthOfPlane(imageBuffer, kYPlaneIndex);
+ size_t yPlaneBytesPerRow =
+ CVPixelBufferGetBytesPerRowOfPlane(imageBuffer, kYPlaneIndex);
+ size_t uvPlaneHeight =
+ CVPixelBufferGetHeightOfPlane(imageBuffer, kUVPlaneIndex);
+ size_t uvPlaneBytesPerRow =
+ CVPixelBufferGetBytesPerRowOfPlane(imageBuffer, kUVPlaneIndex);
+ size_t frameSize =
+ yPlaneBytesPerRow * yPlaneHeight + uvPlaneBytesPerRow * uvPlaneHeight;
+
+ // Sanity check assumption that planar bytes are contiguous.
+ uint8_t* uvPlaneAddress =
+ (uint8_t*)CVPixelBufferGetBaseAddressOfPlane(imageBuffer, kUVPlaneIndex);
+ RTC_DCHECK(
+ uvPlaneAddress == yPlaneAddress + yPlaneHeight * yPlaneBytesPerRow);
+
+ // Stuff data into a cricket::CapturedFrame.
+ int64_t currentTime = rtc::TimeNanos();
+ cricket::CapturedFrame frame;
+ frame.width = yPlaneWidth;
+ frame.height = yPlaneHeight;
+ frame.pixel_width = 1;
+ frame.pixel_height = 1;
+ frame.fourcc = static_cast<uint32_t>(cricket::FOURCC_NV12);
+ frame.time_stamp = currentTime;
+ frame.data = yPlaneAddress;
+ frame.data_size = frameSize;
+
+ if (_startThread->IsCurrent()) {
+ SignalFrameCaptured(this, &frame);
+ } else {
+ _startThread->Invoke<void>(
+ rtc::Bind(&AVFoundationVideoCapturer::SignalFrameCapturedOnStartThread,
+ this, &frame));
+ }
+ CVPixelBufferUnlockBaseAddress(imageBuffer, lockFlags);
+}
+
+void AVFoundationVideoCapturer::SignalFrameCapturedOnStartThread(
+ const cricket::CapturedFrame* frame) {
+ RTC_DCHECK(_startThread->IsCurrent());
+ // This will call a superclass method that will perform the frame conversion
+ // to I420.
+ SignalFrameCaptured(this, frame);
+}
+
+} // namespace webrtc