//
// DrawPath.h
// 畫板
//
// Created by xgao on 16/4/13.
// Copyright © 2016年 xgao. All rights reserved.
//
#import <UIKit/UIKit.h>
@interface DrawPath : UIBezierPath
// 畫筆顏色
@property(nonatomic,retain)UIColor* pathColor;
@end

//
// DrawPath.m
// 畫板
//
// Created by xgao on 16/4/13.
// Copyright © 2016年 xgao. All rights reserved.
//
#import "DrawPath.h"
@implementation DrawPath
@end

//
// DrawView.h
// 畫板
//
// Created by xgao on 16/4/13.
// Copyright © 2016年 xgao. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface DrawView : UIView

// 畫線的寬度
@property(nonatomic,assign)CGFloat lineWidth;

// 線條顏色
@property(nonatomic,retain)UIColor* pathColor;

// 加載背景圖片
@property(nonatomic,strong)UIImage* image;

// 清屏
- (void)clear;

// 撤銷
- (void)undo;

// 橡皮擦
- (void)eraser;

// 保存
- (void)save;
@end

//
// DrawView.m
// 畫板
//
// Created by xgao on 16/4/13.
// Copyright © 2016年 xgao. All rights reserved.
//

#import "DrawView.h"
#import "DrawPath.h"

@interface DrawView()

@property(nonatomic,strong) DrawPath* path;

// 線的數(shù)組
@property(nonatomic,strong) NSMutableArray* paths;

@end

@implementation DrawView

- (void)awakeFromNib{

 [self setUp];
}

- (instancetype)initWithFrame:(CGRect)frame
{
 self = [super initWithFrame:frame];
 if (self) {
 [self setUp];
 }
 return self;
}

// 重繪UI
- (void)drawRect:(CGRect)rect {

 for (DrawPath* path in self.paths) {

 if ([path isKindOfClass:[UIImage class]]) {
  // 畫圖片
  UIImage* image = (UIImage*)path;
  [image drawInRect:rect];
 }else{
  // 畫線

  // 設(shè)置畫筆顏色
  [path.pathColor set];

  // 繪制
  [path stroke];
 }
 }
}

// 懶加載屬性
- (NSMutableArray*)paths{

 if (_paths == nil) {
 _paths = [NSMutableArray array];
 }
 return _paths;
}

// 重寫image屬性
- (void)setImage:(UIImage *)image{

 _image = image;

 // 將圖片加入到線條數(shù)組中
 [self.paths addObject:image];

 [self setNeedsDisplay];
}

#pragma mark - Init

// 初始化
- (void)setUp{

 // 添加平移手勢
 UIPanGestureRecognizer* panGes = [[UIPanGestureRecognizer alloc]initWithTarget:self action:@selector(panGes:)];
 [self addGestureRecognizer:panGes];

 // 默認(rèn)值
 self.lineWidth = 1;
 self.pathColor = [UIColor blackColor];
}

#pragma mark - Event

// 平移事件
- (void)panGes:(UIPanGestureRecognizer*)ges{

 // 獲取當(dāng)前點(diǎn)
 CGPoint curPoint = [ges locationInView:self];

 if (ges.state == UIGestureRecognizerStateBegan) { // 開始移動(dòng)

 // 創(chuàng)建貝塞爾曲線
 _path = [[DrawPath alloc]init];

 // 設(shè)置線條寬度
 _path.lineWidth = _lineWidth;

 // 線條默認(rèn)顏色
 _path.pathColor = _pathColor;

 // 設(shè)置起始點(diǎn)
 [_path moveToPoint:curPoint];

 [self.paths addObject:_path];
 }

 // 連線
 [_path addLineToPoint:curPoint];

 // 重繪
 [self setNeedsDisplay];
}

#pragma mark - Method

// 清屏
- (void)clear{

 [self.paths removeAllObjects];

 [self setNeedsDisplay];
}

// 撤銷
- (void)undo{

 [self.paths removeLastObject];

 [self setNeedsDisplay];
}

// 橡皮擦
- (void)eraser{

 self.pathColor = [UIColor whiteColor];

 [self setNeedsDisplay];
}

// 保存
- (void)save{

 // ---- 截圖操作
 // 開啟上下文
 UIGraphicsBeginImageContextWithOptions(self.frame.size, NO, 0);

 // 獲取當(dāng)前上下文
 CGContextRef context = UIGraphicsGetCurrentContext();

 // 渲染圖層到上下文
 [self.layer renderInContext:context];

 // 從上下文中獲取圖片
 UIImage* image = UIGraphicsGetImageFromCurrentImageContext();

 // 關(guān)閉上下文
 UIGraphicsEndImageContext();

 // ---- 保存圖片
 UIImageWriteToSavedPhotosAlbum(image, self, @selector(image:didFinishSavingWithError:contextInfo:), nil);

}

// 圖片保存方法，必需寫這個(gè)方法體，不能會(huì)保存不了圖片
- (void)image:(UIImage *)image didFinishSavingWithError:(NSError *)error contextInfo:(void *)contextInfo{

 // 提示
 UIAlertView* alert = [[UIAlertView alloc]initWithTitle:@"保存成功" message:nil delegate:nil cancelButtonTitle:@"ok" otherButtonTitles:nil, nil];
 [alert show];
}
@end

//
// ViewController.m
// 畫板
//
// Created by xgao on 16/4/13.
// Copyright © 2016年 xgao. All rights reserved.
//

#import "ViewController.h"
#import "DrawView.h"

@interface ViewController ()<UIImagePickerControllerDelegate,UINavigationControllerDelegate>

// 畫板
@property (weak, nonatomic) IBOutlet DrawView *drawView;

@end

@implementation ViewController

- (void)viewDidLoad {
 [super viewDidLoad];

}

#pragma mark - Event

// 線條寬度變化
- (IBAction)lineWidthChange:(UISlider*)sender {

 _drawView.lineWidth = sender.value;
}

// 線條顏色變化
- (IBAction)pathColorChange:(UIButton*)sender {

 _drawView.pathColor = sender.backgroundColor;
}

// 清屏
- (IBAction)clearAction:(id)sender {

 [_drawView clear];
}

// 撤銷
- (IBAction)undoAction:(id)sender {

 [_drawView undo];
}

// 橡皮擦
- (IBAction)eraserAction:(id)sender {

 [_drawView eraser];
}

// 照片
- (IBAction)pickerPhotoAction:(id)sender {

 // 照片選擇控制器
 UIImagePickerController* picVC = [[UIImagePickerController alloc]init];
 // 照片源
 picVC.sourceType = UIImagePickerControllerSourceTypePhotoLibrary;
 // 委托
 picVC.delegate = self;

 [self presentViewController:picVC animated:YES completion:nil];
}

// 保存
- (IBAction)saveAction:(id)sender {

 [_drawView save];
}

#pragma mark - UIImagePickerControllerDelegate

- (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingImage:(UIImage *)image editingInfo:(nullable NSDictionary<NSString *,id> *)editingInfo{

 // 設(shè)置圖片
 _drawView.image = image;

 // 關(guān)閉窗口
 [self dismissViewControllerAnimated:YES completion:nil];
}
@end