Fix image issue

Author: LeoMobileDeveloper

PullToRefreshKit/MainViewController.swift

@@ -9,6 +9,7 @@
 import UIKit
 import AudioToolbox
 
+// 如果你喜欢这个库，一个★就是对我最好的支持，项目地址 https://github.com/LeoMobileDeveloper/PullToRefreshKit
 class MainViewController: UITableViewController {
     var models = [SectionModel]()
     override func viewDidLoad() {

Source/Classes/Header.swift

@@ -30,7 +30,8 @@ public class DefaultRefreshHeader:UIView,RefreshableHeader{
         addSubview(spinner)
         addSubview(textLabel)
         addSubview(imageView);
-        imageView.image = UIImage(named: "arrow_down");
+        let image = UIImage(named: "arrow_down", inBundle: NSBundle(forClass: self.dynamicType), compatibleWithTraitCollection: nil)
+        imageView.image = image
         imageView.sizeToFit()
         imageView.frame = CGRectMake(0, 0, 24, 24)
         imageView.center = CGPointMake(frame.width/2 - 60 - 20, frame.size.height/2)

Source/Classes/Left.swift

@@ -14,9 +14,7 @@ public enum RefreshKitLeftRightText{
     case releaseToAction
 }
 public class DefaultRefreshLeft:UIView,RefreshableLeftRight{
-    public let imageView:UIImageView = UIImageView().SetUp {
-        $0.image = UIImage(named: "arrow_right")
-    }
+    public let imageView:UIImageView = UIImageView()
     public let textLabel:UILabel  = UILabel().SetUp {
         $0.font = UIFont.systemFontOfSize(14)
     }
@@ -39,6 +37,8 @@ public class DefaultRefreshLeft:UIView,RefreshableLeftRight{
         textLabel.numberOfLines = 0
         imageView.frame = CGRectMake(0, 0,20, 20)
         imageView.center = CGPointMake(40,frame.size.height/2)
+        let image = UIImage(named: "arrow_right", inBundle: NSBundle(forClass: self.dynamicType), compatibleWithTraitCollection: nil)
+        imageView.image = image
         textDic[.scrollToAction] = PullToRefreshKitLeftString.scrollToAction
         textDic[.releaseToAction] = PullToRefreshKitLeftString.releaseToAction
         textLabel.text = textDic[.scrollToAction]

Source/Classes/Right.swift

@@ -10,9 +10,7 @@ import UIKit
 
 
 public class DefaultRefreshRight:UIView,RefreshableLeftRight{
-    public let imageView:UIImageView = UIImageView().SetUp {
-        $0.image = UIImage(named: "arrow_left")
-    }
+    public let imageView:UIImageView = UIImageView()
     public  let textLabel:UILabel  = UILabel().SetUp {
         $0.font = UIFont.systemFontOfSize(14)
     }
@@ -34,6 +32,8 @@ public class DefaultRefreshRight:UIView,RefreshableLeftRight{
         textLabel.numberOfLines = 0
         imageView.frame = CGRectMake(0, 0,20, 20)
         imageView.center = CGPointMake(10,frame.size.height/2)
+        let image = UIImage(named: "arrow_left", inBundle: NSBundle(forClass: self.dynamicType), compatibleWithTraitCollection: nil)
+        imageView.image = image
         textDic[.scrollToAction] = PullToRefreshKitRightString.scrollToAction
         textDic[.releaseToAction] = PullToRefreshKitRightString.releaseToAction
         textLabel.text = textDic[.scrollToAction]