1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- /*
- * This file is part of the SDWebImage package.
- * (c) Olivier Poitrey <rs@dailymotion.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- #import "UIImage+MultiFormat.h"
- #import "NSImage+WebCache.h"
- #import "SDWebImageCodersManager.h"
- #import "objc/runtime.h"
- @implementation UIImage (MultiFormat)
- #if SD_MAC
- - (NSUInteger)sd_imageLoopCount {
- NSUInteger imageLoopCount = 0;
- for (NSImageRep *rep in self.representations) {
- if ([rep isKindOfClass:[NSBitmapImageRep class]]) {
- NSBitmapImageRep *bitmapRep = (NSBitmapImageRep *)rep;
- imageLoopCount = [[bitmapRep valueForProperty:NSImageLoopCount] unsignedIntegerValue];
- break;
- }
- }
- return imageLoopCount;
- }
- - (void)setSd_imageLoopCount:(NSUInteger)sd_imageLoopCount {
- for (NSImageRep *rep in self.representations) {
- if ([rep isKindOfClass:[NSBitmapImageRep class]]) {
- NSBitmapImageRep *bitmapRep = (NSBitmapImageRep *)rep;
- [bitmapRep setProperty:NSImageLoopCount withValue:@(sd_imageLoopCount)];
- break;
- }
- }
- }
- #else
- - (NSUInteger)sd_imageLoopCount {
- NSUInteger imageLoopCount = 0;
- NSNumber *value = objc_getAssociatedObject(self, @selector(sd_imageLoopCount));
- if ([value isKindOfClass:[NSNumber class]]) {
- imageLoopCount = value.unsignedIntegerValue;
- }
- return imageLoopCount;
- }
- - (void)setSd_imageLoopCount:(NSUInteger)sd_imageLoopCount {
- NSNumber *value = @(sd_imageLoopCount);
- objc_setAssociatedObject(self, @selector(sd_imageLoopCount), value, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
- }
- #endif
- - (SDImageFormat)sd_imageFormat {
- SDImageFormat imageFormat = SDImageFormatUndefined;
- NSNumber *value = objc_getAssociatedObject(self, @selector(sd_imageFormat));
- if ([value isKindOfClass:[NSNumber class]]) {
- imageFormat = value.integerValue;
- return imageFormat;
- }
- // Check CGImage's UTType, may return nil for non-Image/IO based image
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wunguarded-availability"
- if (&CGImageGetUTType != NULL) {
- CFStringRef uttype = CGImageGetUTType(self.CGImage);
- imageFormat = [NSData sd_imageFormatFromUTType:uttype];
- }
- #pragma clang diagnostic pop
- return imageFormat;
- }
- - (void)setSd_imageFormat:(SDImageFormat)sd_imageFormat {
- objc_setAssociatedObject(self, @selector(sd_imageFormat), @(sd_imageFormat), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
- }
- + (nullable UIImage *)sd_imageWithData:(nullable NSData *)data {
- return [[SDWebImageCodersManager sharedInstance] decodedImageWithData:data];
- }
- - (nullable NSData *)sd_imageData {
- return [self sd_imageDataAsFormat:SDImageFormatUndefined];
- }
- - (nullable NSData *)sd_imageDataAsFormat:(SDImageFormat)imageFormat {
- NSData *imageData = nil;
- if (self) {
- imageData = [[SDWebImageCodersManager sharedInstance] encodedDataWithImage:self format:imageFormat];
- }
- return imageData;
- }
- @end
|