From d39ee1493a3982431c0e2b58dd46992ba7b6ed1c Mon Sep 17 00:00:00 2001
From: Ali <>
Date: Wed, 16 Jun 2021 23:11:57 +0400
Subject: [PATCH] Don't generate empty images

---
 submodules/Display/Source/GenerateImage.swift | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/submodules/Display/Source/GenerateImage.swift b/submodules/Display/Source/GenerateImage.swift
index 4241ca4bd7..89cdd49d63 100644
--- a/submodules/Display/Source/GenerateImage.swift
+++ b/submodules/Display/Source/GenerateImage.swift
@@ -94,6 +94,9 @@ public func generateGrayscaleAlphaMaskImage(image: UIImage) -> UIImage? {
 }
 
 public func generateImage(_ size: CGSize, contextGenerator: (CGSize, CGContext) -> Void, opaque: Bool = false, scale: CGFloat? = nil) -> UIImage? {
+    if size.width.isZero || size.height.isZero {
+        return nil
+    }
     let context = DrawingContext(size: size, scale: scale ?? 0.0, opaque: opaque, clear: false)
     context.withFlippedContext { c in
         contextGenerator(context.size, c)
@@ -102,6 +105,9 @@ public func generateImage(_ size: CGSize, contextGenerator: (CGSize, CGContext)
 }
 
 public func generateImage(_ size: CGSize, opaque: Bool = false, scale: CGFloat? = nil, rotatedContext: (CGSize, CGContext) -> Void) -> UIImage? {
+    if size.width.isZero || size.height.isZero {
+        return nil
+    }
     let context = DrawingContext(size: size, scale: scale ?? 0.0, opaque: opaque, clear: false)
     context.withContext { c in
         rotatedContext(context.size, c)