前言：

最近項目有個需求是對試圖對手機密碼進(jìn)行強破解的人進(jìn)行拍照(通過攝像頭截圖)，因為之前沒做過，所以一堆坑。現(xiàn)在就把我的經(jīng)驗都分享出來，希望后來人不用再踏上坑途中。

?

直接上代碼：

　 // 創(chuàng)建會話

　 self.session = [[AVCaptureSession alloc] init];

? ?// 獲取攝像頭的權(quán)限信息，判斷是否有開啟權(quán)限

? ? AVAuthorizationStatus status? ? = [AVCaptureDevice authorizationStatusForMediaType:AVMediaTypeVideo];

? ? if (status == AVAuthorizationStatusAuthorized)

? ? {

? ? ? ? self.setupResult = AVCamSetupResultSuccess;

? ? }

? ??if ( self.setupResult != AVCamSetupResultSuccess )

? ? ? ???return;

? ? self.backgroundRecordingID = UIBackgroundTaskInvalid;

? ??NSError *error = nil;

? ? // 創(chuàng)建輸入設(shè)備

? ? AVCaptureDevice *videoDevice = [self deviceWithMediaType:AVMediaTypeVideo preferringPosition:AVCaptureDevicePositionFront];

? ??AVCaptureDeviceInput *videoDeviceInput = [AVCaptureDeviceInput deviceInputWithDevice:videoDevice error:&error];

?

? ? //?beginConfiguration這個很重要，在addInput或者addOutput的時候一定要調(diào)用這個，add之后再調(diào)用commitConfiguration

? ? [self.session beginConfiguration];

?? ?

? ? if ([self.session canAddInput:videoDeviceInput])

? ? ? {

? ? ? ? [self.session addInput:videoDeviceInput]；

? ? ? ? self.videoDeviceInput = videoDeviceInput;

? ? ? ?}

　 ?// 為會話加入output設(shè)備

? ? dispatch_queue_t videoDataOutputQueue = dispatch_queue_create("VideoDataOutputQueue", DISPATCH_QUEUE_SERIAL);

? ? ? ? AVCaptureVideoDataOutput *videoDataOutput = [[AVCaptureVideoDataOutput alloc] init];

? ? ? ? videoDataOutput.videoSettings =[NSDictionary dictionaryWithObject:[NSNumber numberWithInt:kCVPixelFormatType_32BGRA]forKey: ? ? ? ? ? ? ? (id)kCVPixelBufferPixelFormatTypeKey];

? ? // 設(shè)置self的AVCaptureVideoDataOutputSampleBufferDelegate

? ? ? ? [videoDataOutput setSampleBufferDelegate:self queue:videoDataOutputQueue];

? ? ? ? if ([self.session canAddOutput:videoDataOutput])

? ? ? ? {

? ? ? ? ? ? [self.session addOutput:videoDataOutput];

?

? ? ? ? }

? ? ? ? [self.session commitConfiguration];

? ? ? ?

?

? ? ?// 獲取前置攝像頭

- (AVCaptureDevice *)deviceWithMediaType:(NSString *)mediaType preferringPosition:(AVCaptureDevicePosition)position{

? ? NSArray *devices = [AVCaptureDevice devicesWithMediaType:mediaType];

? ? AVCaptureDevice *captureDevice = devices.firstObject;

?? ?

? ? for ( AVCaptureDevice *device in devices ) {

? ? ? ? if ( device.position == position ) {

? ? ? ? ? ? captureDevice = device;

? ? ? ? ? ? break;

? ? ? ? }

? ? }

?? ??return captureDevice;

}

? ? // 實現(xiàn)代理方法并獲取圖片

- (void)captureOutput:(AVCaptureOutput *)captureOutput didOutputSampleBuffer:(CMSampleBufferRef)sampleBuffer fromConnection:(AVCaptureConnection *)connection{

? ? if (takephoto.isShutScreen)

? ? {

? ? ? ? NSLog(@"didOutputSampleBuffer");

? ? ? ? UIImage *image? = [self getImageBySampleBufferref:sampleBuffer];

? ? }

}

? ? // 將獲取的數(shù)據(jù)轉(zhuǎn)換成圖片，網(wǎng)上很多轉(zhuǎn)換方式轉(zhuǎn)出來的圖片都是錯誤的，最后找到這個方法

- (UIImage *)getImageBySampleBufferref:(CMSampleBufferRef)sampleBuffer

{

? ? CVImageBufferRef imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer);

? ? /*Lock the image buffer*/

? ? CVPixelBufferLockBaseAddress(imageBuffer,0);

? ? /*Get information about the image*/

? ? uint8_t *baseAddress = (uint8_t *)CVPixelBufferGetBaseAddress(imageBuffer);

? ? size_t bytesPerRow = CVPixelBufferGetBytesPerRow(imageBuffer);

? ? size_t width = CVPixelBufferGetWidth(imageBuffer);

? ? size_t height = CVPixelBufferGetHeight(imageBuffer);

?? ?

? ? /*We unlock the? image buffer*/

? ? CVPixelBufferUnlockBaseAddress(imageBuffer,0);

?? ?

? ? /*Create a CGImageRef from the CVImageBufferRef*/

? ? CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();

? ? CGContextRef newContext = CGBitmapContextCreate(baseAddress, width, height, 8, bytesPerRow, colorSpace, kCGBitmapByteOrder32Little | kCGImageAlphaPremultipliedFirst);

? ? CGImageRef newImage = CGBitmapContextCreateImage(newContext);

?? ?

? ? /*We release some components*/

? ? CGContextRelease(newContext);

? ? CGColorSpaceRelease(colorSpace);

?? ?

? ? UIImage *image= [UIImage imageWithCGImage:newImage scale:1.0 orientation:UIImageOrientationRight];

? ? NSLog(@"%@", image);

?? ?

? ? /*We relase the CGImageRef*/

? ? CGImageRelease(newImage);

?

? ? return image;

}

??

? ? 只要照著我的思路走，就可以達(dá)到攝像機截屏的效果，希望能幫到需要的人，如有其他意見，請留言。

?

轉(zhuǎn)載于:https://www.cnblogs.com/destiLaugh/p/6552569.html

總結(jié)

以上是生活随笔為你收集整理的AVFoundation之如何从摄像头获取图像的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。