Fix intrinsic size bug

- Animated image class always returned `CGRectZero` as intrinsic content size - Fixes #7
kaishin · Dec 21, 2014 · c51f0f7 · c51f0f7
1 parent b87464f
commit c51f0f7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Source/AnimatedImage.swift b/Source/AnimatedImage.swift
@@ -10,6 +10,10 @@ public class AnimatedImage: UIImage {
   var animatedFrames = [AnimatedFrame]()
   var totalDuration: NSTimeInterval = 0.0
 
+  override public var size: CGSize {
+    return frameAtIndex(0)?.size ?? CGSizeZero
+  }
+
   // MARK: - Private Properties
   private lazy var displayLink: CADisplayLink = CADisplayLink(target: self, selector: "updateCurrentFrame")
   private var currentFrameIndex = 0