replace checkmarkImage and render mode

support change theme color

Author: YanAn

Images.xcassets/checkmark_selected.imageset/checkmark_select.pdf

@@ -43,7 +43,7 @@ class AssetCell: UICollectionViewCell {
     }
 
     func setChecked(_ checked: Bool, animation: Bool) {
-        checkMarkImageView.image = checked ? UIImage(named: selectedCheckMarkImageName, in: currentBundle, compatibleWith: nil) : UIImage(named: unselectedCheckMarkImageName, in: currentBundle, compatibleWith: nil)
+        checkMarkImageView.image = checked ? UIImage(named: selectedCheckMarkImageName, in: currentBundle, compatibleWith: nil)?.withRenderingMode(.alwaysTemplate) : UIImage(named: unselectedCheckMarkImageName, in: currentBundle, compatibleWith: nil)
         if !self.checked && checked && animation {
             UIView.animateKeyframes(withDuration: 0.33, delay: 0.0, options: [], animations: { () -> Void in
                 UIView.addKeyframe(withRelativeStartTime: 0.0, relativeDuration: 0.5, animations: { [unowned self]() -> Void in

PhotoPicker/AssetCell.swift

@@ -224,8 +224,7 @@ extension AssetsViewController {
         let numberBarItem = UIBarButtonItem(customView: toolbarNumberView)
         let leftSpace = UIBarButtonItem(barButtonSystemItem: .flexibleSpace, target: nil, action: nil)
         toolbarItems = [highqualityBarItem, leftSpace, numberBarItem, sendBarItem]
-        
-        navigationController?.toolbar.tintColor = toolBarTintColor
+        navigationController?.toolbar.tintColor = themeToolBarTintColor
         navigationController?.setToolbarHidden(false, animated: true)
     }
 

PhotoPicker/AssetsViewController.swift

@@ -24,15 +24,19 @@ open class PhotoPickerController: UIViewController {
     //MARK: - private property
     fileprivate var albumsNavigationController: UINavigationController!
 
-    public init(localizedStrings: [String: String]) {
+    public init(localizedStrings: [String: String], themeColor: UIColor? = nil) {
         super.init(nibName:nil, bundle:nil)
 
         self.assetCollectionSubtypes = [.smartAlbumUserLibrary,
                                         .smartAlbumFavorites,
                                         .albumMyPhotoStream,
                                         .albumCloudShared,
                                         .albumRegular]
-        
+
+        if let color = themeColor {
+            themeToolBarTintColor = color
+            themeTextColor = color
+        }
         setupAlbumsViewController()
         localizedString = localizedStrings
 
@@ -54,7 +58,6 @@ open class PhotoPickerController: UIViewController {
 
     open override func viewDidLoad() {
         super.viewDidLoad()
-
         // Do any additional setup after loading the view.
     }
 

PhotoPicker/PhotoPickerController.swift

@@ -15,11 +15,12 @@ class ToolBarHighQualityButton: UIView {
     weak var assetsViewController: AssetsViewController!
 
     //MARK: - public property
+    var themeColor: UIColor?
     var checked: Bool = false {
         didSet {
             if checked {
-                imageView.image = UIImage(named: toolbarHighQualityImageCheckedImageName, in: currentBundle, compatibleWith: nil)
-                titleLabel.textColor = blueTextColor
+                imageView.image = UIImage(named: toolbarHighQualityImageCheckedImageName, in: currentBundle, compatibleWith: nil)?.withRenderingMode(.alwaysTemplate)
+                titleLabel.textColor = themeTextColor
                 assetsViewController.updateHighQualityImageSize()
             } else {
                 imageView.image = UIImage(named: toolbarHighQualityImageUnCheckedImageName, in: currentBundle, compatibleWith: nil)

PhotoPicker/ToolBarHighQualityButton.swift

@@ -36,7 +36,7 @@ class ToolBarNumberView: UIView {
 
     func setupProperty() {
         imageView = UIImageView(frame: bounds)
-        imageView.image = UIImage(named: toolbarNumberViewBackgroundImageName, in: currentBundle, compatibleWith: nil)
+        imageView.image = UIImage(named: toolbarNumberViewBackgroundImageName, in: currentBundle, compatibleWith: nil)?.withRenderingMode(.alwaysTemplate)
         imageView.isHidden = true
 
         numberLabel = UILabel(frame: bounds)

PhotoPicker/ToolBarNumberView.swift

@@ -8,9 +8,9 @@
 
 import UIKit
 
-let toolBarTintColor: UIColor = UIColor(red: 3/255.0, green: 169/255.0, blue: 244/255.0, alpha: 1.0)
+var themeToolBarTintColor: UIColor = UIColor(red: 3/255.0, green: 169/255.0, blue: 244/255.0, alpha: 1.0)
 let greyTextColor: UIColor = UIColor(red: 191/255.0, green: 191/255.0, blue: 191/255.0, alpha: 1.0)
-let blueTextColor: UIColor = UIColor(red: 3/255.0, green: 169/255.0, blue: 244/255.0, alpha: 1.0)
+var themeTextColor: UIColor = UIColor(red: 3/255.0, green: 169/255.0, blue: 244/255.0, alpha: 1.0)
 let AlbumCoverBorderColor: CGColor = UIColor.white.cgColor
 let currentBundle: Bundle = Bundle(for: PhotoPickerController.self)
 let selectedCheckMarkImageName = "checkmark_selected"