AvatarGenerator: add sanity check

RolePlayGateway · Mar 8, 2017 · de4e489 · de4e489
1 parent 70ab051
commit de4e489
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/Vector/Utils/AvatarGenerator.m b/Vector/Utils/AvatarGenerator.m
@@ -103,9 +103,14 @@ + (UIImage *)imageFromText:(NSString*)text withBackgroundColor:(UIColor*)color
 
     // set to the top quality
     CGContextRef context = UIGraphicsGetCurrentContext();
-    CGContextSetInterpolationQuality(context, kCGInterpolationHigh);
-    [[backgroundLabel layer] renderInContext: UIGraphicsGetCurrentContext()];
-    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
+    UIImage *image;
+    if (context)
+    {
+        CGContextSetInterpolationQuality(context, kCGInterpolationHigh);
+        [[backgroundLabel layer] renderInContext: context];
+        image = UIGraphicsGetImageFromCurrentImageContext();
+    }
+
     UIGraphicsEndImageContext();
 
     // Return the image.
@@ -127,9 +132,14 @@ + (UIImage *)imageFromText:(NSString*)text withBackgroundColor:(UIColor*)color s
 
     // set to the top quality
     CGContextRef context = UIGraphicsGetCurrentContext();
-    CGContextSetInterpolationQuality(context, kCGInterpolationHigh);
-    [[bgLabel layer] renderInContext: UIGraphicsGetCurrentContext()];
-    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
+    UIImage *image;
+    if (context)
+    {
+        CGContextSetInterpolationQuality(context, kCGInterpolationHigh);
+        [[bgLabel layer] renderInContext: context];
+        image = UIGraphicsGetImageFromCurrentImageContext();
+    }   
+
     UIGraphicsEndImageContext();
 
     // Return the image.