From 2e8c33d43d9736ba997b3e37451bbcd82f39ee9f Mon Sep 17 00:00:00 2001
From: Ilya Laktyushin <ilya@laktyush.in>
Date: Wed, 13 Feb 2019 01:05:14 +0400
Subject: [PATCH] Fixed mask copying on layer snapshot

---
 Display/UIKitUtils.swift | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/Display/UIKitUtils.swift b/Display/UIKitUtils.swift
index eaaaccadc8..25d1d222a4 100644
--- a/Display/UIKitUtils.swift
+++ b/Display/UIKitUtils.swift
@@ -244,10 +244,11 @@ private func makeSubtreeSnapshot(layer: CALayer, keepTransform: Bool = false) ->
     view.layer.masksToBounds = layer.masksToBounds
     if let mask = layer.mask {
         let maskLayer = CALayer()
-        maskLayer.bounds = mask.bounds
         maskLayer.contents = mask.contents
+        maskLayer.contentsRect = mask.contentsRect
         maskLayer.contentsScale = mask.contentsScale
         maskLayer.contentsCenter = mask.contentsCenter
+        maskLayer.contentsGravity = mask.contentsGravity
         view.layer.mask = maskLayer
     }
     view.layer.cornerRadius = layer.cornerRadius
@@ -261,6 +262,9 @@ private func makeSubtreeSnapshot(layer: CALayer, keepTransform: Bool = false) ->
                 }
                 subtree.frame = sublayer.frame
                 subtree.bounds = sublayer.bounds
+                if let maskLayer = subtree.layer.mask {
+                    maskLayer.frame = sublayer.bounds
+                }
                 view.addSubview(subtree)
             } else {
                 return nil