UIImage+Metadata.m 7.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221
  1. /*
  2. * This file is part of the SDWebImage package.
  3. * (c) Olivier Poitrey <rs@dailymotion.com>
  4. *
  5. * For the full copyright and license information, please view the LICENSE
  6. * file that was distributed with this source code.
  7. */
  8. #import "UIImage+Metadata.h"
  9. #import "NSImage+Compatibility.h"
  10. #import "SDInternalMacros.h"
  11. #import "objc/runtime.h"
  12. @implementation UIImage (Metadata)
  13. #if SD_UIKIT || SD_WATCH
  14. - (NSUInteger)sd_imageLoopCount {
  15. NSUInteger imageLoopCount = 0;
  16. NSNumber *value = objc_getAssociatedObject(self, @selector(sd_imageLoopCount));
  17. if ([value isKindOfClass:[NSNumber class]]) {
  18. imageLoopCount = value.unsignedIntegerValue;
  19. }
  20. return imageLoopCount;
  21. }
  22. - (void)setSd_imageLoopCount:(NSUInteger)sd_imageLoopCount {
  23. NSNumber *value = @(sd_imageLoopCount);
  24. objc_setAssociatedObject(self, @selector(sd_imageLoopCount), value, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
  25. }
  26. - (NSUInteger)sd_imageFrameCount {
  27. NSArray<UIImage *> *animatedImages = self.images;
  28. if (!animatedImages || animatedImages.count <= 1) {
  29. return 1;
  30. }
  31. NSNumber *value = objc_getAssociatedObject(self, @selector(sd_imageFrameCount));
  32. if ([value isKindOfClass:[NSNumber class]]) {
  33. return [value unsignedIntegerValue];
  34. }
  35. __block NSUInteger frameCount = 1;
  36. __block UIImage *previousImage = animatedImages.firstObject;
  37. [animatedImages enumerateObjectsUsingBlock:^(UIImage * _Nonnull image, NSUInteger idx, BOOL * _Nonnull stop) {
  38. // ignore first
  39. if (idx == 0) {
  40. return;
  41. }
  42. if (![image isEqual:previousImage]) {
  43. frameCount++;
  44. }
  45. previousImage = image;
  46. }];
  47. objc_setAssociatedObject(self, @selector(sd_imageFrameCount), @(frameCount), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
  48. return frameCount;
  49. }
  50. - (BOOL)sd_isAnimated {
  51. return (self.images != nil);
  52. }
  53. #pragma clang diagnostic push
  54. #pragma clang diagnostic ignored "-Warc-performSelector-leaks"
  55. - (BOOL)sd_isVector {
  56. if (@available(iOS 13.0, tvOS 13.0, watchOS 6.0, *)) {
  57. // Xcode 11 supports symbol image, keep Xcode 10 compatible currently
  58. SEL SymbolSelector = NSSelectorFromString(@"isSymbolImage");
  59. if ([self respondsToSelector:SymbolSelector] && [self performSelector:SymbolSelector]) {
  60. return YES;
  61. }
  62. // SVG
  63. SEL SVGSelector = SD_SEL_SPI(CGSVGDocument);
  64. if ([self respondsToSelector:SVGSelector] && [self performSelector:SVGSelector]) {
  65. return YES;
  66. }
  67. }
  68. if (@available(iOS 11.0, tvOS 11.0, watchOS 4.0, *)) {
  69. // PDF
  70. SEL PDFSelector = SD_SEL_SPI(CGPDFPage);
  71. if ([self respondsToSelector:PDFSelector] && [self performSelector:PDFSelector]) {
  72. return YES;
  73. }
  74. }
  75. return NO;
  76. }
  77. #pragma clang diagnostic pop
  78. #else
  79. - (NSUInteger)sd_imageLoopCount {
  80. NSUInteger imageLoopCount = 0;
  81. NSRect imageRect = NSMakeRect(0, 0, self.size.width, self.size.height);
  82. NSImageRep *imageRep = [self bestRepresentationForRect:imageRect context:nil hints:nil];
  83. NSBitmapImageRep *bitmapImageRep;
  84. if ([imageRep isKindOfClass:[NSBitmapImageRep class]]) {
  85. bitmapImageRep = (NSBitmapImageRep *)imageRep;
  86. }
  87. if (bitmapImageRep) {
  88. imageLoopCount = [[bitmapImageRep valueForProperty:NSImageLoopCount] unsignedIntegerValue];
  89. }
  90. return imageLoopCount;
  91. }
  92. - (void)setSd_imageLoopCount:(NSUInteger)sd_imageLoopCount {
  93. NSRect imageRect = NSMakeRect(0, 0, self.size.width, self.size.height);
  94. NSImageRep *imageRep = [self bestRepresentationForRect:imageRect context:nil hints:nil];
  95. NSBitmapImageRep *bitmapImageRep;
  96. if ([imageRep isKindOfClass:[NSBitmapImageRep class]]) {
  97. bitmapImageRep = (NSBitmapImageRep *)imageRep;
  98. }
  99. if (bitmapImageRep) {
  100. [bitmapImageRep setProperty:NSImageLoopCount withValue:@(sd_imageLoopCount)];
  101. }
  102. }
  103. - (NSUInteger)sd_imageFrameCount {
  104. NSRect imageRect = NSMakeRect(0, 0, self.size.width, self.size.height);
  105. NSImageRep *imageRep = [self bestRepresentationForRect:imageRect context:nil hints:nil];
  106. NSBitmapImageRep *bitmapImageRep;
  107. if ([imageRep isKindOfClass:[NSBitmapImageRep class]]) {
  108. bitmapImageRep = (NSBitmapImageRep *)imageRep;
  109. }
  110. if (bitmapImageRep) {
  111. return [[bitmapImageRep valueForProperty:NSImageFrameCount] unsignedIntegerValue];
  112. }
  113. return 1;
  114. }
  115. - (BOOL)sd_isAnimated {
  116. BOOL isAnimated = NO;
  117. NSRect imageRect = NSMakeRect(0, 0, self.size.width, self.size.height);
  118. NSImageRep *imageRep = [self bestRepresentationForRect:imageRect context:nil hints:nil];
  119. NSBitmapImageRep *bitmapImageRep;
  120. if ([imageRep isKindOfClass:[NSBitmapImageRep class]]) {
  121. bitmapImageRep = (NSBitmapImageRep *)imageRep;
  122. }
  123. if (bitmapImageRep) {
  124. NSUInteger frameCount = [[bitmapImageRep valueForProperty:NSImageFrameCount] unsignedIntegerValue];
  125. isAnimated = frameCount > 1 ? YES : NO;
  126. }
  127. return isAnimated;
  128. }
  129. - (BOOL)sd_isVector {
  130. NSRect imageRect = NSMakeRect(0, 0, self.size.width, self.size.height);
  131. NSImageRep *imageRep = [self bestRepresentationForRect:imageRect context:nil hints:nil];
  132. if ([imageRep isKindOfClass:[NSPDFImageRep class]]) {
  133. return YES;
  134. }
  135. if ([imageRep isKindOfClass:[NSEPSImageRep class]]) {
  136. return YES;
  137. }
  138. if ([NSStringFromClass(imageRep.class) hasSuffix:@"NSSVGImageRep"]) {
  139. return YES;
  140. }
  141. return NO;
  142. }
  143. #endif
  144. - (SDImageFormat)sd_imageFormat {
  145. SDImageFormat imageFormat = SDImageFormatUndefined;
  146. NSNumber *value = objc_getAssociatedObject(self, @selector(sd_imageFormat));
  147. if ([value isKindOfClass:[NSNumber class]]) {
  148. imageFormat = value.integerValue;
  149. return imageFormat;
  150. }
  151. // Check CGImage's UTType, may return nil for non-Image/IO based image
  152. CFStringRef uttype = CGImageGetUTType(self.CGImage);
  153. imageFormat = [NSData sd_imageFormatFromUTType:uttype];
  154. return imageFormat;
  155. }
  156. - (void)setSd_imageFormat:(SDImageFormat)sd_imageFormat {
  157. objc_setAssociatedObject(self, @selector(sd_imageFormat), @(sd_imageFormat), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
  158. }
  159. - (void)setSd_isIncremental:(BOOL)sd_isIncremental {
  160. objc_setAssociatedObject(self, @selector(sd_isIncremental), @(sd_isIncremental), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
  161. }
  162. - (BOOL)sd_isIncremental {
  163. NSNumber *value = objc_getAssociatedObject(self, @selector(sd_isIncremental));
  164. return value.boolValue;
  165. }
  166. - (void)setSd_isTransformed:(BOOL)sd_isTransformed {
  167. objc_setAssociatedObject(self, @selector(sd_isTransformed), @(sd_isTransformed), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
  168. }
  169. - (BOOL)sd_isTransformed {
  170. NSNumber *value = objc_getAssociatedObject(self, @selector(sd_isTransformed));
  171. return value.boolValue;
  172. }
  173. - (void)setSd_decodeOptions:(SDImageCoderOptions *)sd_decodeOptions {
  174. objc_setAssociatedObject(self, @selector(sd_decodeOptions), sd_decodeOptions, OBJC_ASSOCIATION_COPY_NONATOMIC);
  175. }
  176. -(BOOL)sd_isThumbnail {
  177. CGSize thumbnailSize = CGSizeZero;
  178. NSValue *thumbnailSizeValue = self.sd_decodeOptions[SDImageCoderDecodeThumbnailPixelSize];
  179. if (thumbnailSizeValue != nil) {
  180. #if SD_MAC
  181. thumbnailSize = thumbnailSizeValue.sizeValue;
  182. #else
  183. thumbnailSize = thumbnailSizeValue.CGSizeValue;
  184. #endif
  185. }
  186. return thumbnailSize.width > 0 && thumbnailSize.height > 0;
  187. }
  188. - (SDImageCoderOptions *)sd_decodeOptions {
  189. SDImageCoderOptions *value = objc_getAssociatedObject(self, @selector(sd_decodeOptions));
  190. if ([value isKindOfClass:NSDictionary.class]) {
  191. return value;
  192. }
  193. return nil;
  194. }
  195. @end