Use custom snapshot for views that implements HeroCustomSnapshotView

HeroTransitions · Dec 18, 2018 · be8cc65 · be8cc65
1 parent d0a0ef8
commit be8cc65
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/Sources/HeroContext.swift b/Sources/HeroContext.swift
@@ -172,7 +172,9 @@ extension HeroContext {
       #if os(tvOS)
         snapshot = view.snapshotView(afterScreenUpdates: true)!
       #else
-        if #available(iOS 9.0, *), let stackView = view as? UIStackView {
+        if let customSnapshotView = view as? HeroCustomSnapshotView, let snapshotView = customSnapshotView.heroSnapshot {
+          snapshot = snapshotView
+        } else if #available(iOS 9.0, *), let stackView = view as? UIStackView {
           snapshot = stackView.slowSnapshotView()
         } else if let imageView = view as? UIImageView, view.subviews.isEmpty {
           let contentView = UIImageView(image: imageView.image)
@@ -388,3 +390,8 @@ extension HeroContext {
     }
   }
 }
+
+/// Allows a view to create their own custom snapshot when using **Optimized** snapshot
+public protocol HeroCustomSnapshotView {
+	var heroSnapshot: UIView? { get }
+}