Gifu/Source/AnimatedImage.swift

121 lines
3.7 KiB
Swift
Executable File

import UIKit
import ImageIO
public class AnimatedImage: UIImage {
// MARK: - Constants
let maxTimeStep = 1.0
// MARK: - Public Properties
var delegate: UIImageView?
var animatedFrames = [AnimatedFrame]()
var totalDuration: NSTimeInterval = 0.0
// MARK: - Private Properties
private lazy var displayLink: CADisplayLink = CADisplayLink(target: self, selector: "updateCurrentFrame")
private var currentFrameIndex = 0
private var timeSinceLastFrameChange: NSTimeInterval = 0.0
// MARK: - Computed Properties
var currentFrame: UIImage? {
return frameAtIndex(currentFrameIndex)
}
private var isAnimated: Bool {
return totalDuration != 0.0
}
// MARK: - Initializers
required public init(coder aDecoder: NSCoder) {
super.init(coder: aDecoder)
}
public override convenience init(data: NSData) {
self.init(data: data, size: CGSizeZero)
}
required public init(data: NSData, size: CGSize) {
super.init()
let imageSource = CGImageSourceCreateWithData(data, nil)
attachDisplayLink()
curry(prepareFrames) <^> imageSource <*> size
pauseAnimation()
}
// MARK: - Factories
public class func animatedImageWithName(name: String) -> AnimatedImage? {
let path = NSBundle.mainBundle().bundlePath.stringByAppendingPathComponent(name)
return animatedImageWithData <^> NSData(contentsOfFile: path)
}
public class func animatedImageWithData(data: NSData) -> AnimatedImage {
let size = UIImage.sizeForImageData(data) ?? CGSizeZero
return self(data: data, size: size)
}
public class func animatedImageWithName(name: String, size: CGSize) -> AnimatedImage? {
let path = NSBundle.mainBundle().bundlePath.stringByAppendingPathComponent(name)
return curry(animatedImageWithData) <^> NSData(contentsOfFile: path) <*> size
}
public class func animatedImageWithData(data: NSData, size: CGSize) -> AnimatedImage {
return self(data: data, size: size)
}
// MARK: - Display Link Helpers
func attachDisplayLink() {
displayLink.addToRunLoop(NSRunLoop.mainRunLoop(), forMode: NSRunLoopCommonModes)
}
// MARK: - Frame Methods
private func prepareFrames(imageSource: CGImageSourceRef, size: CGSize) {
let numberOfFrames = Int(CGImageSourceGetCount(imageSource))
animatedFrames.reserveCapacity(numberOfFrames)
(animatedFrames, totalDuration) = reduce(0..<numberOfFrames, ([AnimatedFrame](), 0.0)) { accumulator, index in
let accumulatedFrames = accumulator.0
let accumulatedDuration = accumulator.1
let frameDuration = CGImageSourceGIFFrameDuration(imageSource, index)
let frameImageRef = CGImageSourceCreateImageAtIndex(imageSource, UInt(index), nil)
let frame = UIImage(CGImage: frameImageRef)?.resize(size)
let animatedFrame = AnimatedFrame(image: frame, duration: frameDuration)
return (accumulatedFrames + [animatedFrame], accumulatedDuration + frameDuration)
}
}
func frameAtIndex(index: Int) -> UIImage? {
if index >= animatedFrames.count { return .None }
return animatedFrames[index].image
}
func updateCurrentFrame() {
if !isAnimated { return }
timeSinceLastFrameChange += min(maxTimeStep, displayLink.duration)
var frameDuration = animatedFrames[currentFrameIndex].duration
if timeSinceLastFrameChange >= frameDuration {
timeSinceLastFrameChange -= frameDuration
currentFrameIndex = ++currentFrameIndex % animatedFrames.count
delegate?.layer.setNeedsDisplay()
}
}
// MARK: - Animation
func pauseAnimation() {
displayLink.paused = true
}
func resumeAnimation() {
if isAnimated {
displayLink.paused = false
}
}
func isAnimating() -> Bool {
return !displayLink.paused
}
}