Update DLTiledImageView.m

darling0825 · Dec 19, 2017 · 05b4110 · 05b4110
1 parent 0031d27
commit 05b4110
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/DLPhotoPicker/View/TiledImageView/DLTiledImageView.m b/DLPhotoPicker/View/TiledImageView/DLTiledImageView.m
@@ -47,19 +47,24 @@ - (void)setImage:(UIImage *)image {
 - (void)drawRect:(CGRect)rect {
     if (self.image) {
 
-        dispatch_async(dispatch_get_main_queue(), ^{
-            CGContextRef ctx = UIGraphicsGetCurrentContext();
-            CGContextSaveGState(ctx);
+        __block CGRect rect = CGRectZero;
+        if (![NSThread isMainThread]) {
+            dispatch_sync(dispatch_get_main_queue(), ^{
+                rect = self.bounds;
+            });
+        }else {
+            rect = self.bounds;
+        }
 
-            CGRect rect = self.bounds;
-
-            CGContextTranslateCTM(ctx, rect.origin.x, rect.origin.y);
-            CGContextTranslateCTM(ctx, 0.0, rect.size.height);
-            CGContextScaleCTM(ctx, 1.0, -1.0);
-            CGContextTranslateCTM(ctx, -rect.origin.x, -rect.origin.y);
-            CGContextDrawImage(ctx, rect, self.image.CGImage);
-            CGContextRestoreGState(ctx);
-        });
+        //
+        CGContextRef ctx = UIGraphicsGetCurrentContext();
+        CGContextSaveGState(ctx);
+        CGContextTranslateCTM(ctx, rect.origin.x, rect.origin.y);
+        CGContextTranslateCTM(ctx, 0.0, rect.size.height);
+        CGContextScaleCTM(ctx, 1.0, -1.0);
+        CGContextTranslateCTM(ctx, -rect.origin.x, -rect.origin.y);
+        CGContextDrawImage(ctx, rect, self.image.CGImage);
+        CGContextRestoreGState(ctx);
     }
 }