Merge pull request #1 from SDWebImage/feature_cache_cost_function

Adopt SDWebImage's memory cache cost function

Author: dreampiggy

Example/Podfile.lock

@@ -2,14 +2,14 @@ PODS:
   - Expecta (1.0.6)
   - SDWebImage/Core (5.0.0-beta3)
   - SDWebImageYYPlugin (0.1.1):
-    - SDWebImage/Core (>= 5.0.0-beta)
+    - SDWebImage/Core (>= 5.0.0-beta3)
     - SDWebImageYYPlugin/YYCache (= 0.1.1)
     - SDWebImageYYPlugin/YYImage (= 0.1.1)
   - SDWebImageYYPlugin/YYCache (0.1.1):
-    - SDWebImage/Core (>= 5.0.0-beta)
+    - SDWebImage/Core (>= 5.0.0-beta3)
     - YYCache
   - SDWebImageYYPlugin/YYImage (0.1.1):
-    - SDWebImage/Core (>= 5.0.0-beta)
+    - SDWebImage/Core (>= 5.0.0-beta3)
     - YYImage/Core
   - YYCache (1.0.4)
   - YYImage/Core (1.0.4)
@@ -22,12 +22,18 @@ DEPENDENCIES:
   - SDWebImageYYPlugin (from `../`)
   - YYImage/WebP
 
+SPEC REPOS:
+  https://github.com/cocoapods/specs.git:
+    - Expecta
+    - YYCache
+    - YYImage
+
 EXTERNAL SOURCES:
   SDWebImage:
     :branch: 5.x
     :git: https://github.com/rs/SDWebImage.git
   SDWebImageYYPlugin:
-    :path: ../
+    :path: "../"
 
 CHECKOUT OPTIONS:
   SDWebImage:
@@ -37,10 +43,10 @@ CHECKOUT OPTIONS:
 SPEC CHECKSUMS:
   Expecta: 3b6bd90a64b9a1dcb0b70aa0e10a7f8f631667d5
   SDWebImage: e52654ceef9fdc19f4c612d64a5b6d3f05dd81a4
-  SDWebImageYYPlugin: cfaeb2f2834c161d9aa9949d3d7c44aa7c80463c
+  SDWebImageYYPlugin: 3a0cc67eb20702c7db4f66b568eff30ba058e726
   YYCache: 8105b6638f5e849296c71f331ff83891a4942952
   YYImage: 1e1b62a9997399593e4b9c4ecfbbabbf1d3f3b54
 
 PODFILE CHECKSUM: 5900612c85a696261bc740246922b2914976662c
 
-COCOAPODS: 1.4.0
+COCOAPODS: 1.5.3

Example/SDWebImageYYPlugin.xcodeproj/project.pbxproj

@@ -242,7 +242,6 @@
 				6003F587195388D20070C39A /* Frameworks */,
 				6003F588195388D20070C39A /* Resources */,
 				A1C9798AC9A64B3285935ADE /* [CP] Embed Pods Frameworks */,
-				0731EE8A10D058D3EFB2B75A /* [CP] Copy Pods Resources */,
 			);
 			buildRules = (
 			);
@@ -262,7 +261,6 @@
 				6003F5AB195388D20070C39A /* Frameworks */,
 				6003F5AC195388D20070C39A /* Resources */,
 				79112E8232AA29A2C856E6D2 /* [CP] Embed Pods Frameworks */,
-				A1F76FB8CC168A22F62A2894 /* [CP] Copy Pods Resources */,
 			);
 			buildRules = (
 			);
@@ -337,21 +335,6 @@
 /* End PBXResourcesBuildPhase section */
 
 /* Begin PBXShellScriptBuildPhase section */
-		0731EE8A10D058D3EFB2B75A /* [CP] Copy Pods Resources */ = {
-			isa = PBXShellScriptBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-			);
-			inputPaths = (
-			);
-			name = "[CP] Copy Pods Resources";
-			outputPaths = (
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-			shellPath = /bin/sh;
-			shellScript = "\"${SRCROOT}/Pods/Target Support Files/Pods-SDWebImageYYPlugin_Example/Pods-SDWebImageYYPlugin_Example-resources.sh\"\n";
-			showEnvVarsInLog = 0;
-		};
 		0C5606D06E2CBD1FF0BCA1FB /* [CP] Check Pods Manifest.lock */ = {
 			isa = PBXShellScriptBuildPhase;
 			buildActionMask = 2147483647;
@@ -430,21 +413,6 @@
 			shellScript = "\"${SRCROOT}/Pods/Target Support Files/Pods-SDWebImageYYPlugin_Example/Pods-SDWebImageYYPlugin_Example-frameworks.sh\"\n";
 			showEnvVarsInLog = 0;
 		};
-		A1F76FB8CC168A22F62A2894 /* [CP] Copy Pods Resources */ = {
-			isa = PBXShellScriptBuildPhase;
-			buildActionMask = 2147483647;
-			files = (
-			);
-			inputPaths = (
-			);
-			name = "[CP] Copy Pods Resources";
-			outputPaths = (
-			);
-			runOnlyForDeploymentPostprocessing = 0;
-			shellPath = /bin/sh;
-			shellScript = "\"${SRCROOT}/Pods/Target Support Files/Pods-SDWebImageYYPlugin_Tests/Pods-SDWebImageYYPlugin_Tests-resources.sh\"\n";
-			showEnvVarsInLog = 0;
-		};
 /* End PBXShellScriptBuildPhase section */
 
 /* Begin PBXSourcesBuildPhase section */

SDWebImageYYPlugin/Classes/YYImage/YYImageBridge/YYImage+SDAdditions.m

@@ -40,3 +40,28 @@ - (BOOL)isAllFramesLoaded {
 }
 
 @end
+
+@implementation YYImage (MemoryCacheCost)
+
+- (NSUInteger)sd_memoryCost {
+    NSNumber *value = objc_getAssociatedObject(self, @selector(sd_memoryCost));
+    if (value != nil) {
+        return value.unsignedIntegerValue;
+    }
+    
+    CGImageRef imageRef = self.CGImage;
+    if (!imageRef) {
+        return 0;
+    }
+    NSUInteger bytesPerFrame = CGImageGetBytesPerRow(imageRef) * CGImageGetHeight(imageRef);
+    NSUInteger frameCount = 1;
+    if (self.isAllFramesLoaded) {
+        frameCount = self.animatedImageFrameCount;
+    }
+    frameCount = frameCount > 0 ? frameCount : 1;
+    NSUInteger cost = bytesPerFrame * frameCount;
+    return cost;
+}
+
+@end
+