2 // ScreenshotEditorViewController.swift
5 // Created by Hori,Masaki on 2016/12/29.
6 // Copyright © 2016年 Hori,Masaki. All rights reserved.
11 private struct EditedImage {
13 var editedImage: NSImage
16 init(image: NSImage, url: URL) {
23 final class TrimRectInformation: NSObject {
25 @objc private(set) var name: String
26 private(set) var rect: NSRect
28 fileprivate init(name: String, rect: NSRect) {
35 final class ScreenshotEditorViewController: BridgeViewController {
37 @objc let trimInfo: [TrimRectInformation]
39 override init(nibName: NSNib.Name?, bundle: Bundle?) {
42 TrimRectInformation(name: "Status", rect: NSRect(x: 328, y: 13, width: 470, height: 365)),
43 TrimRectInformation(name: "List", rect: NSRect(x: 362, y: 15, width: 438, height: 368)),
44 TrimRectInformation(name: "AirplaneBase", rect: NSRect(x: 575, y: 13, width: 225, height: 358))
46 currentTrimInfo = trimInfo[0]
48 super.init(nibName: ScreenshotEditorViewController.nibName, bundle: nil)
51 required init?(coder: NSCoder) {
53 fatalError("init(coder:) has not been implemented")
57 @IBOutlet weak var tiledImageView: TiledImageView!
58 @IBOutlet weak var doneButton: NSButton!
60 @objc var columnCount: Int {
62 get { return tiledImageView.columnCount }
64 tiledImageView.columnCount = newValue
65 UserDefaults.standard[.screenshotEditorColumnCount] = newValue
71 return tiledImageView.image
74 @objc dynamic var currentTrimInfoIndex: Int {
76 get { return realiesCurrentTrimInforIndex }
78 guard 0..<trimInfo.count ~= newValue else { return }
80 realiesCurrentTrimInforIndex = newValue
81 currentTrimInfo = trimInfo[newValue]
85 private var editedImage: NSImage?
86 private var currentSelection: [ScreenshotInformation] = []
87 private var editedImages: [EditedImage] = []
88 private var realiesCurrentTrimInforIndex = UserDefaults.standard[.scrennshotEditorType]
89 private var currentTrimInfo: TrimRectInformation {
96 if $0.name != currentTrimInfo.name { return false }
97 return $0.rect == currentTrimInfo.rect
99 .map { UserDefaults.standard[.scrennshotEditorType] = $0 }
103 private var selectionObservation: NSKeyValueObservation?
105 var completeHandler: ((NSImage?) -> Void)?
107 override func viewDidLoad() {
111 selectionObservation = arrayController.observe(\NSArrayController.selectionIndexes) { [weak self] (_, _) in
113 self?.updateSelections()
116 currentTrimInfoIndex = UserDefaults.standard[.scrennshotEditorType]
120 override func viewWillAppear() {
122 doneButton.action = #selector(ScreenshotEditorViewController.done(_:))
125 private func updateSelections() {
127 guard let selection = arrayController.selectedObjects as? [ScreenshotInformation] else { return }
129 if selection == currentSelection { return }
131 let removed: [ScreenshotInformation] = currentSelection.flatMap {
133 selection.contains($0) ? nil : $0
136 let appended: [ScreenshotInformation] = selection.flatMap {
138 currentSelection.contains($0) ? nil : $0
143 removeEditedImage(url: $0.url)
148 appendEditedImage(url: $0.url)
151 currentSelection = selection
155 private func removeEditedImage(url: URL) {
158 .index { $0.url == url }
159 .map { editedImages.remove(at: $0) }
162 private func appendEditedImage(url: URL) {
164 NSImage(contentsOf: url)
165 .flatMap { EditedImage(image: $0, url: url) }
166 .map { editedImages.append($0) }
169 private func makeEditedImage() {
171 guard !editedImages.isEmpty else {
173 tiledImageView.images = []
177 DispatchQueue(label: "makeTrimedImage queue").async {
179 let images: [NSImage] = self.editedImages.flatMap {
181 guard let originalImage = NSImage(contentsOf: $0.url) else { return nil }
183 let trimedImage = NSImage(size: self.currentTrimInfo.rect.size)
185 trimedImage.lockFocus()
186 originalImage.draw(at: .zero,
187 from: self.currentTrimInfo.rect,
190 trimedImage.unlockFocus()
195 DispatchQueue.main.async {
197 self.tiledImageView.images = images
202 @IBAction func done(_ sender: AnyObject?) {
204 completeHandler?(self.image)
208 extension ScreenshotEditorViewController: NibLoadable {}