屏幕截图功能实现

//

//  ViewController.m

//  1030屏幕截图1

//

//  Created by weibiao on 15/10/30.

//  Copyright © 2015年 weibiao. All rights reserved.

//

#import "ViewController.h"

#import "MBProgressHUD+NJ.h"

@interface ViewController ()

- (IBAction)captureScreen;

@property (weak, nonatomic) IBOutlet UIView *customView;

- (IBAction)changeColor:(UISwitch *)sender;

@end

@implementation ViewController

- (void)viewDidLoad {

    [super viewDidLoad];

    // Do any additional setup after loading the view, typically from a nib.

}

- (IBAction)captureScreen {

    

    // 延时截屏,

    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(2.0 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{

        // 1.创建一个bitmap的上下文

        UIGraphicsBeginImageContext(self.view.frame.size);

        //    UIGraphicsBeginImageContext(self.customView.frame.size);

        

        // 2.将屏幕绘制到上下文中,render提交,只要把要截屏的layer调用方法renderInContext:即可

        //    [self.customView.layer renderInContext:UIGraphicsGetCurrentContext()];

        [self.view.layer renderInContext:UIGraphicsGetCurrentContext()] ;

        

        // 根据图形上下文得到新图片

        UIImage *newImage = UIGraphicsGetImageFromCurrentImageContext();

        

        UIImageWriteToSavedPhotosAlbum(newImage, self,@selector(image:didFinishSavingWithError:contextInfo:) , nil);

    });

    

    

}

- (void)image:(UIImage *)image didFinishSavingWithError:(NSError *)error contextInfo:(void *)contextInfo

{

    if (error) {

        [MBProgressHUD showError:@"保存失败"];

    }else

    {

        [MBProgressHUD showSuccess:@"保存成功"];

    }

}

- (IBAction)changeColor:(UISwitch *)sender {

    NSLog(@"%d",sender.isOn);

    if (sender.isOn) {

        [self.view setBackgroundColor:[UIColor orangeColor]];

    }else

    {

        [self.view setBackgroundColor:[UIColor greenColor]];

    }

}

@end