bgayman#6 Fix rendering an image from view

abekert · Jun 1, 2022 · 5c82571 · 5c82571
1 parent bd9f29c
commit 5c82571
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/Sources/FlipBook/View.swift b/Sources/FlipBook/View.swift
@@ -61,12 +61,9 @@ extension View {
     }
 
     func fb_makeViewSnapshot() -> Image? {
-        UIGraphicsBeginImageContextWithOptions(frame.size, true, 0)
-        guard let context = UIGraphicsGetCurrentContext() else { return nil }
-        layer.presentation()?.render(in: context)
-        let rasterizedView = UIGraphicsGetImageFromCurrentImageContext()
-        UIGraphicsEndImageContext()
-        return rasterizedView
+        UIGraphicsImageRenderer(size: bounds.size).image { rendererContext in
+            drawHierarchy(in: bounds, afterScreenUpdates: false)
+        }
     }
 }
 #endif