Add workaround for the scenarios that has size over the internal size…

… limitation 8192x8192 (#66)
playbook-ui · Feb 16, 2024 · 798f686 · 798f686
1 parent 0a18e71
commit 798f686
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 2 deletions.
diff --git a/Sources/Playbook/SnapshotSupport/SnapshotSupport.swift b/Sources/Playbook/SnapshotSupport/SnapshotSupport.swift
@@ -149,7 +149,31 @@ private extension SnapshotSupport {
                 let actions: UIGraphicsDrawingActions = { context in
                     withoutAnimation {
                         if isEmbedInKeyWindow {
-                            snapshotView.drawHierarchy(in: snapshotView.bounds, afterScreenUpdates: true)
+                            let nativeScale = UIScreen.main.nativeScale
+                            let size = snapshotView.bounds.size
+                            let nativeScaledSize = CGSize(
+                                width: size.width * nativeScale,
+                                height: size.height * nativeScale
+                            )
+                            let sizeLimit: CGFloat = 8192
+                            let limitedScale = max(nativeScaledSize.width / sizeLimit, nativeScaledSize.height / sizeLimit)
+
+                            if limitedScale > 1 {
+                                // `UIView.drawHierarchy` seems to have an internal hard-coded limit size of 8192x8192,
+                                // and specifying a size beyond that resulting in an empty image is drawn.
+                                let limitedRect = CGRect(
+                                    x: .zero,
+                                    y: .zero,
+                                    width: size.width / limitedScale,
+                                    height: size.height / limitedScale
+                                )
+                                context.cgContext.scaleBy(x: limitedScale, y: limitedScale)
+                                snapshotView.drawHierarchy(in: limitedRect, afterScreenUpdates: true)
+                            }
+                            else {
+                                snapshotView.drawHierarchy(in: snapshotView.bounds, afterScreenUpdates: true)
+                            }
+
                             snapshotView.removeFromSuperview()
                         }
                         else {

diff --git a/Tests/ExtraScenarios.swift b/Tests/ExtraScenarios.swift
@@ -4,9 +4,15 @@ import SwiftUI
 
 enum ExtraScenarios: ScenarioProvider {
     static func addScenarios(into playbook: Playbook) {
+        playbook.addScenarios(of: "Extra") {
+            Scenario("Over internal size limitation 8192x8192", layout: .fixed(width: 10000, height: 10000)) {
+                Color.red.edgesIgnoringSafeArea(.all)
+            }
+        }
+
         playbook.addScenarios(of: "Normalizable\u{7FFFF}\u{1D} \n.:/") {
             Scenario("Normalizable\u{7FFFF}\u{1D} \n.:/", layout: .fixed(length: 300)) {
-                Color(.systemBlue).edgesIgnoringSafeArea(.all)
+                Color.blue.edgesIgnoringSafeArea(.all)
             }
         }
     }