From 03222655fad08d050fdefce5c131a94d300cbf51 Mon Sep 17 00:00:00 2001
From: Eugene Mozharovsky <mozharovsky@live.com>
Date: Thu, 9 Jun 2016 03:15:13 +0300
Subject: [PATCH] Fast view snapshotting & improved UX

---
 Sources/FusumaViewController.swift | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/Sources/FusumaViewController.swift b/Sources/FusumaViewController.swift
index cdd89a75..1214b2cf 100644
--- a/Sources/FusumaViewController.swift
+++ b/Sources/FusumaViewController.swift
@@ -200,7 +200,11 @@ public final class FusumaViewController: UIViewController {
     
     public override func viewWillDisappear(animated: Bool) {
         super.viewWillDisappear(animated)
-        self.stopAll()
+    }
+  
+    public override func viewDidDisappear(animated: Bool) {
+      super.viewDidDisappear(animated)
+      stopAll()
     }
 
     override public func prefersStatusBarHidden() -> Bool {
@@ -233,11 +237,11 @@ public final class FusumaViewController: UIViewController {
     @IBAction func doneButtonPressed(sender: UIButton) {
         
         let view = albumView.imageCropView
-        
+      
         UIGraphicsBeginImageContextWithOptions(view.frame.size, true, 0)
         let context = UIGraphicsGetCurrentContext()
         CGContextTranslateCTM(context, -albumView.imageCropView.contentOffset.x, -albumView.imageCropView.contentOffset.y)
-        view.layer.renderInContext(context!)
+        view.drawViewHierarchyInRect(view.bounds, afterScreenUpdates: false)
         let image = UIGraphicsGetImageFromCurrentImageContext()
         UIGraphicsEndImageContext()