IOS 之使用闪光灯当手电筒

悠悠 2022-08-24 11:47 158阅读 0赞

首先导入

\#import <AVFoundation/AVFoundation.h> 头文件和引入AVFoundation库文件

头文件声明变量

@interface UTTorchViewController : UIViewController
    
    @property (strong, nonatomic) AVCaptureDevice *device;
    @property (assign, nonatomic) BOOL isLighOn;
    @property (strong, nonatomic) UIButton *btn;
    
    
    - (void)press:(id)pSender;
    
    @end

代码文件如下

- (void)viewDidLoad
    {
        [super viewDidLoad];
        
        _btn = [[UIButton alloc] initWithFrame:CGRectMake(10, 50, 300, 300)];
        [_btn setBackgroundImage:[UIImage imageNamed:@"torch_off"] forState:UIControlStateNormal];
        [_btn addTarget:self action:@selector(press:) forControlEvents:UIControlEventTouchUpInside];
        [self.view addSubview:_btn];
        
        _device = [AVCaptureDevice defaultDeviceWithMediaType:AVMediaTypeVideo];
        if ( ![_device hasTorch] )
        {
            UIAlertView *av = [[UIAlertView alloc] initWithTitle:@"提示" message:@"当前设备没有闪光灯,不支持手电筒功能!" delegate:nil cancelButtonTitle:@"确定" otherButtonTitles:nil, nil];
            [av show];
        }
        
        [self setIsLighOn:NO];
        // Do any additional setup after loading the view from its nib.
    }
    
    
    #pragma mark - Custom method
    - (void)press:(id)pSender
    {
        if ( _isLighOn )
        {
            [_device lockForConfiguration:nil];
            [_device setTorchMode:AVCaptureTorchModeOff];
            [_device unlockForConfiguration];
            [_btn setBackgroundImage:[UIImage imageNamed:@"torch_off"] forState:UIControlStateNormal];
            _isLighOn = NO;
        }else
        {
            [_device lockForConfiguration:nil];
            [_device setTorchMode:AVCaptureTorchModeOn];
            [_device unlockForConfiguration];
            [_btn setBackgroundImage:[UIImage imageNamed:@"torch_on"] forState:UIControlStateNormal];
            _isLighOn = YES;
        }
    }

就是这么简单