Merge pull request #18 from lkzhao/stack-view-fix

Temp fix for stack view
HeroTransitions · Jan 6, 2017 · 44b6c3b · 44b6c3b
2 parents e83e19c + 7058500
commit 44b6c3b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/Hero/HeroContext.swift b/Hero/HeroContext.swift
@@ -100,7 +100,9 @@ extension HeroContext{
     let oldCornerRadius = view.layer.cornerRadius
     view.layer.cornerRadius = 0
     let snapshot:UIView
-    if let imageView = view as? UIImageView{
+    if #available(iOS 9.0, *), let stackView = view as? UIStackView{
+      snapshot = stackView.slowSnapshotView()
+    } else if let imageView = view as? UIImageView{
       let contentView = UIImageView(image: imageView.image)
       contentView.frame = imageView.bounds
       contentView.contentMode = imageView.contentMode

diff --git a/Hero/UIKit+Hero.swift b/Hero/UIKit+Hero.swift
@@ -56,6 +56,20 @@ public extension UIView{
       )
     }
   }
+
+  func slowSnapshotView() -> UIView{
+    UIGraphicsBeginImageContextWithOptions(bounds.size, isOpaque, 0)
+    layer.render(in: UIGraphicsGetCurrentContext()!)
+//    drawHierarchy(in: bounds, afterScreenUpdates: true)
+    let image = UIGraphicsGetImageFromCurrentImageContext()
+    UIGraphicsEndImageContext()
+
+    let imageView = UIImageView(image: image)
+    imageView.frame = bounds
+    let snapshotView = UIView(frame:bounds)
+    snapshotView.addSubview(imageView)
+    return snapshotView
+  }
 }
 
 public extension UIViewController{