Fix image bundle path error if the pod use frameworks.

1and2papa · Mar 17, 2015 · aadc946 · aadc946
1 parent 39861bf
commit aadc946
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/CTAssetsPickerController/UIImage+CTAssetsPickerController.m b/CTAssetsPickerController/UIImage+CTAssetsPickerController.m
@@ -7,12 +7,20 @@
 //
 
 #import "UIImage+CTAssetsPickerController.h"
+#import "NSBundle+CTAssetsPickerController.h"
 
 @implementation UIImage (CTAssetsPickerController)
 
 + (UIImage *)ctassetsPickerControllerImageNamed:(NSString *)name
 {
-    return [UIImage imageNamed:[NSString stringWithFormat:@"CTAssetsPickerController.bundle/%@", name]];
+    if ([UIImage respondsToSelector:@selector(imageNamed:inBundle:compatibleWithTraitCollection:)])
+    {
+        return [UIImage imageNamed:name inBundle:[NSBundle ctassetsPickerControllerBundle] compatibleWithTraitCollection:nil];
+    }
+    else
+    {
+        return [UIImage imageNamed:[NSString stringWithFormat:@"CTAssetsPickerController.bundle/%@", name]];
+    }
 }
 
 @end