iOS截图和保存图片

//
//  ViewController.m
//  Test
//
//  Created by york on 16/3/30.
//  Copyright © 2016年 裴杨威. All rights reserved.
//

#define DEVICE_WIDTH [UIScreen mainScreen].bounds.size.width
#define DEVICE_HEIGHT [UIScreen mainScreen].bounds.size.height
#import "ViewController.h"

@interface ViewController ()
{
UIImageView *topImageView;
UIImageView *bottomImageView;
}
@end

@implementation ViewController

- (void)viewDidLoad {
[super viewDidLoad];
// Do any additional setup after loading the view, typically from a nib.
self.view.backgroundColor = [UIColor grayColor];

topImageView = [[UIImageView alloc] initWithFrame:CGRectMake((DEVICE_WIDTH-252)/2, 20, 252, 220)];
topImageView.image = [UIImage imageNamed:@"1.jpg"];
[self.view addSubview:topImageView];

bottomImageView = [[UIImageView alloc] initWithFrame:CGRectMake((DEVICE_WIDTH-140)/2, CGRectGetMaxY(topImageView.frame), 140, 220)];
bottomImageView.image = [UIImage imageNamed:@"2.jpg"];
[self.view addSubview:bottomImageView];

UIButton *saveTopImageBtn = [UIButton buttonWithType:UIButtonTypeSystem];
saveTopImageBtn.frame = CGRectMake(10, CGRectGetMaxY(bottomImageView.frame)+20, (DEVICE_WIDTH-30)/2, 30);
[saveTopImageBtn setTitle:@"截图上图" forState:0];
saveTopImageBtn.tag = 100;
[saveTopImageBtn addTarget:self action:@selector(saveBtnClick:) forControlEvents:UIControlEventTouchUpInside];
[self.view addSubview:saveTopImageBtn];

UIButton *saveBottomImageBtn = [UIButton buttonWithType:UIButtonTypeSystem];
saveBottomImageBtn.frame = CGRectMake(CGRectGetMaxX(saveTopImageBtn.frame)+10, CGRectGetMaxY(bottomImageView.frame)+20, (DEVICE_WIDTH-30)/2, 30);
[saveBottomImageBtn setTitle:@"截图下图" forState:0];
saveBottomImageBtn.tag = 101;
[saveBottomImageBtn addTarget:self action:@selector(saveBtnClick:) forControlEvents:UIControlEventTouchUpInside];
[self.view addSubview:saveBottomImageBtn];

UIButton *saveAllImageBtn = [UIButton buttonWithType:UIButtonTypeSystem];
saveAllImageBtn.frame = CGRectMake(10, CGRectGetMaxY(saveTopImageBtn.frame)+20, (DEVICE_WIDTH-30)/2, 30);
[saveAllImageBtn setTitle:@"截图整屏" forState:0];
saveAllImageBtn.tag = 102;
[saveAllImageBtn addTarget:self action:@selector(saveBtnClick:) forControlEvents:UIControlEventTouchUpInside];
[self.view addSubview:saveAllImageBtn];

UIButton *SaveNetImageBtn = [UIButton buttonWithType:UIButtonTypeSystem];
SaveNetImageBtn.frame = CGRectMake(CGRectGetMaxX(saveTopImageBtn.frame)+10, CGRectGetMaxY(saveBottomImageBtn.frame)+20, (DEVICE_WIDTH-30)/2, 30);
[SaveNetImageBtn setTitle:@"保存网络图片" forState:0];
SaveNetImageBtn.tag = 103;
[SaveNetImageBtn addTarget:self action:@selector(saveBtnClick:) forControlEvents:UIControlEventTouchUpInside];
[self.view addSubview:SaveNetImageBtn];

}

- (BOOL)isOpenAlbum//判断是否能打开相册
{
if ([UIImagePickerController isSourceTypeAvailable:
UIImagePickerControllerSourceTypePhotoLibrary])
{
return YES;
}
else
{
UIAlertView *alert = [[UIAlertView alloc] initWithTitle:nil message:@"无法打开相册" delegate:nil cancelButtonTitle:@"确定" otherButtonTitles:nil, nil];
[alert show];
return NO;
}
}

#pragma mark -
#pragma mark --保存通过url获取的图片--
- (void)saveNetImage
{
NSData *dateImg = [NSData dataWithContentsOfURL:[NSURL URLWithString:@"http://d.hiphotos.baidu.com/image/h%3D200/sign=8663264274f082023292963f7bfbfb8a/f3d3572c11dfa9eca13b947665d0f703918fc1be.jpg"]];
UIImage *image = [UIImage imageWithData:dateImg];
UIImageWriteToSavedPhotosAlbum(image, self, @selector(imageSavedToPhotosAlbum:didFinishSavingWithError:contextInfo:), nil);
}

#pragma mark -
#pragma mark --截图--
- (UIImage *)getRealImage:(UIView *)view
{
/*
UIGraphicsBeginImageContextWithOptions(<#CGSize size#>, <#BOOL opaque#>, <#CGFloat scale#>)
size ----> 新创建的位图上下文的大小
opaque ----> 透明度开关
scale ----> 缩放因子 0指的是系统会自动设置正确的比例
*/
UIGraphicsBeginImageContextWithOptions(view.bounds.size,YES,0);//创建一个基于位图的上下文
[view drawViewHierarchyInRect:view.bounds afterScreenUpdates:YES];
UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
UIGraphicsEndImageContext();
return image;
}

#pragma mark -
#pragma mark --系统保存图片的方法--
- (void)imageSavedToPhotosAlbum:(UIImage *)image didFinishSavingWithError:(NSError *)error contextInfo:(void *)contextInfo
{
if (!error) {
UIAlertView *alert = [[UIAlertView alloc] initWithTitle:nil message:@"保存成功" delegate:nil cancelButtonTitle:@"确定" otherButtonTitles:nil, nil];
[alert show];
}else
{
UIAlertView *alert = [[UIAlertView alloc] initWithTitle:nil message:@"保存失败" delegate:nil cancelButtonTitle:@"确定" otherButtonTitles:nil, nil];
[alert show];
}
}

#pragma mark -
#pragma mark --按钮点击事件--
- (void)saveBtnClick:(UIButton *)sender
{
switch (sender.tag) {
case 100:
{
if ([self isOpenAlbum]) {
UIImageWriteToSavedPhotosAlbum([self getRealImage:topImageView], self, @selector(imageSavedToPhotosAlbum:didFinishSavingWithError:contextInfo:), nil);
}
}
break;
case 101:
{
if ([self isOpenAlbum]) {
UIImageWriteToSavedPhotosAlbum([self getRealImage:bottomImageView], self, @selector(imageSavedToPhotosAlbum:didFinishSavingWithError:contextInfo:), nil);
}
}
break;
case 102:
{
if ([self isOpenAlbum]) {
UIImageWriteToSavedPhotosAlbum([self getRealImage:self.view], self, @selector(imageSavedToPhotosAlbum:didFinishSavingWithError:contextInfo:), nil);
}
}
break;
case 103:
{
if ([self isOpenAlbum]) {
[self saveNetImage];
}
}
break;
default:
break;
}
}

- (void)didReceiveMemoryWarning {
[super didReceiveMemoryWarning];
// Dispose of any resources that can be recreated.
}

@end