SDImageCodersManager.m 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
  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 "SDImageCodersManager.h"
  9. #import "SDImageIOCoder.h"
  10. #import "SDImageGIFCoder.h"
  11. #import "SDImageAPNGCoder.h"
  12. #import "SDImageHEICCoder.h"
  13. #import "SDInternalMacros.h"
  14. @interface SDImageCodersManager ()
  15. @property (nonatomic, strong, nonnull) NSMutableArray<id<SDImageCoder>> *imageCoders;
  16. @end
  17. @implementation SDImageCodersManager {
  18. SD_LOCK_DECLARE(_codersLock);
  19. }
  20. + (nonnull instancetype)sharedManager {
  21. static dispatch_once_t once;
  22. static id instance;
  23. dispatch_once(&once, ^{
  24. instance = [self new];
  25. });
  26. return instance;
  27. }
  28. - (instancetype)init {
  29. if (self = [super init]) {
  30. // initialize with default coders
  31. _imageCoders = [NSMutableArray arrayWithArray:@[[SDImageIOCoder sharedCoder], [SDImageGIFCoder sharedCoder], [SDImageAPNGCoder sharedCoder]]];
  32. SD_LOCK_INIT(_codersLock);
  33. }
  34. return self;
  35. }
  36. - (NSArray<id<SDImageCoder>> *)coders {
  37. SD_LOCK(_codersLock);
  38. NSArray<id<SDImageCoder>> *coders = [_imageCoders copy];
  39. SD_UNLOCK(_codersLock);
  40. return coders;
  41. }
  42. - (void)setCoders:(NSArray<id<SDImageCoder>> *)coders {
  43. SD_LOCK(_codersLock);
  44. [_imageCoders removeAllObjects];
  45. if (coders.count) {
  46. [_imageCoders addObjectsFromArray:coders];
  47. }
  48. SD_UNLOCK(_codersLock);
  49. }
  50. #pragma mark - Coder IO operations
  51. - (void)addCoder:(nonnull id<SDImageCoder>)coder {
  52. if (![coder conformsToProtocol:@protocol(SDImageCoder)]) {
  53. return;
  54. }
  55. SD_LOCK(_codersLock);
  56. [_imageCoders addObject:coder];
  57. SD_UNLOCK(_codersLock);
  58. }
  59. - (void)removeCoder:(nonnull id<SDImageCoder>)coder {
  60. if (![coder conformsToProtocol:@protocol(SDImageCoder)]) {
  61. return;
  62. }
  63. SD_LOCK(_codersLock);
  64. [_imageCoders removeObject:coder];
  65. SD_UNLOCK(_codersLock);
  66. }
  67. #pragma mark - SDImageCoder
  68. - (BOOL)canDecodeFromData:(NSData *)data {
  69. NSArray<id<SDImageCoder>> *coders = self.coders;
  70. for (id<SDImageCoder> coder in coders.reverseObjectEnumerator) {
  71. if ([coder canDecodeFromData:data]) {
  72. return YES;
  73. }
  74. }
  75. return NO;
  76. }
  77. - (BOOL)canEncodeToFormat:(SDImageFormat)format {
  78. NSArray<id<SDImageCoder>> *coders = self.coders;
  79. for (id<SDImageCoder> coder in coders.reverseObjectEnumerator) {
  80. if ([coder canEncodeToFormat:format]) {
  81. return YES;
  82. }
  83. }
  84. return NO;
  85. }
  86. - (UIImage *)decodedImageWithData:(NSData *)data options:(nullable SDImageCoderOptions *)options {
  87. if (!data) {
  88. return nil;
  89. }
  90. UIImage *image;
  91. NSArray<id<SDImageCoder>> *coders = self.coders;
  92. for (id<SDImageCoder> coder in coders.reverseObjectEnumerator) {
  93. if ([coder canDecodeFromData:data]) {
  94. image = [coder decodedImageWithData:data options:options];
  95. break;
  96. }
  97. }
  98. return image;
  99. }
  100. - (NSData *)encodedDataWithImage:(UIImage *)image format:(SDImageFormat)format options:(nullable SDImageCoderOptions *)options {
  101. if (!image) {
  102. return nil;
  103. }
  104. NSArray<id<SDImageCoder>> *coders = self.coders;
  105. for (id<SDImageCoder> coder in coders.reverseObjectEnumerator) {
  106. if ([coder canEncodeToFormat:format]) {
  107. return [coder encodedDataWithImage:image format:format options:options];
  108. }
  109. }
  110. return nil;
  111. }
  112. - (NSData *)encodedDataWithFrames:(NSArray<SDImageFrame *> *)frames loopCount:(NSUInteger)loopCount format:(SDImageFormat)format options:(SDImageCoderOptions *)options {
  113. if (!frames || frames.count < 1) {
  114. return nil;
  115. }
  116. NSArray<id<SDImageCoder>> *coders = self.coders;
  117. for (id<SDImageCoder> coder in coders.reverseObjectEnumerator) {
  118. if ([coder canEncodeToFormat:format]) {
  119. if ([coder respondsToSelector:@selector(encodedDataWithFrames:loopCount:format:options:)]) {
  120. return [coder encodedDataWithFrames:frames loopCount:loopCount format:format options:options];
  121. }
  122. }
  123. }
  124. return nil;
  125. }
  126. @end