Add an property to YYAnimatedImageView for convenience

levenon · Apr 11, 2016 · 6105ed7 · 6105ed7
1 parent 51fe3ef
commit 6105ed7
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/YYWebImage/Image/YYAnimatedImageView.h b/YYWebImage/Image/YYAnimatedImageView.h
@@ -11,6 +11,8 @@
 
 #import <UIKit/UIKit.h>
 
+@protocol YYAnimatedImage;
+
 NS_ASSUME_NONNULL_BEGIN
 
 /**
@@ -80,6 +82,12 @@ NS_ASSUME_NONNULL_BEGIN
  */
 @property (nonatomic) NSUInteger maxBufferSize;
 
+/**
+ *  A convenience accessor for the getting the animated image that the image view is currently displaying
+ *  If the `image` property does not contain an animated image, this property returns nil.
+ */
+@property (nonatomic, strong, nullable) UIImage<YYAnimatedImage> *animatedImage;
+
 @end
 
 

diff --git a/YYWebImage/Image/YYAnimatedImageView.m b/YYWebImage/Image/YYAnimatedImageView.m
@@ -301,6 +301,20 @@ - (void)setHighlighted:(BOOL)highlighted {
     [self imageChanged];
 }
 
+- (void)setAnimatedImage:(UIImage<YYAnimatedImage> *)animatedImage {
+    if (self.image == animatedImage) {
+        return;
+    }
+    [self setImage:animatedImage withType:YYAnimatedImageTypeImage];
+}
+
+- (UIImage<YYAnimatedImage> *)animatedImage {
+    if (![self.image conformsToProtocol:@protocol(YYAnimatedImage)]) {
+        return nil;
+    }
+    return (UIImage<YYAnimatedImage> *)self.image;
+}
+
 - (id)imageForType:(YYAnimatedImageType)type {
     switch (type) {
         case YYAnimatedImageTypeNone: return nil;