Update Mask.cs

Author: yanjinhuagood

src/WPFDevelopers.Shared/Controls/Mask/Mask.cs

@@ -50,18 +50,20 @@ private static void Parent_Loaded(object sender, RoutedEventArgs e)
 
         private static void CreateMask(UIElement uIElement, bool isRemove = false)
         {
+            if (uIElement == null) return;
             var layer = AdornerLayer.GetAdornerLayer(uIElement);
             if (layer == null) return;
-            if (isRemove && uIElement != null)
+            var adorners = layer.GetAdorners(uIElement);
+            if (adorners != null)
             {
-                var adorners = layer.GetAdorners(uIElement);
-                if (adorners != null)
-                    foreach (var item in adorners)
-                        if (item is AdornerContainer container)
-                            layer.Remove(container);
-                return;
+                foreach (var item in adorners)
+                {
+                    if (item is AdornerContainer container)
+                        layer.Remove(container);
+                }
             }
-
+            if (isRemove)
+                return;
             var adornerContainer = new AdornerContainer(uIElement);
             var value = GetChild(uIElement);
             if (value != null)