fixed issue #4

okmr-d · okmr-d · commit 0b12848d9446 · 2015-07-22T01:28:00.000+09:00
diff --git a/DOFavoriteButton/DOFavoriteButton.swift b/DOFavoriteButton/DOFavoriteButton.swift
@@ -17,7 +17,7 @@ public class DOFavoriteButton: UIButton {
     private var imageShape: CAShapeLayer!
     @IBInspectable public var image: UIImage! {
         didSet {
-            imageShape.mask.contents = image.CGImage
+            createLayers(image: image)
         }
     }
     @IBInspectable public var imageColorOn: UIColor! = UIColor(red: 255/255, green: 172/255, blue: 51/255, alpha: 1.0) {
@@ -104,6 +104,7 @@ public class DOFavoriteButton: UIButton {
     }
     
     private func createLayers(#image: UIImage!) {
+        self.layer.sublayers = nil
         
         let imageFrame = CGRectMake(frame.size.width / 2 - frame.size.width / 4, frame.size.height / 2 - frame.size.height / 4, frame.size.width / 2, frame.size.height / 2)
         let imgCenterPoint = CGPointMake(imageFrame.origin.x + imageFrame.width / 2, imageFrame.origin.y + imageFrame.height / 2)

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ public class DOFavoriteButton: UIButton {`
`17`	`17`	`private var imageShape: CAShapeLayer!`
`18`	`18`	`@IBInspectable public var image: UIImage! {`
`19`	`19`	`didSet {`
`20`		`- imageShape.mask.contents = image.CGImage`
	`20`	`+ createLayers(image: image)`
`21`	`21`	`}`
`22`	`22`	`}`
`23`	`23`	`@IBInspectable public var imageColorOn: UIColor! = UIColor(red: 255/255, green: 172/255, blue: 51/255, alpha: 1.0) {`
`@@ -104,6 +104,7 @@ public class DOFavoriteButton: UIButton {`
`104`	`104`	`}`
`105`	`105`
`106`	`106`	`private func createLayers(#image: UIImage!) {`
	`107`	`+ self.layer.sublayers = nil`
`107`	`108`
`108`	`109`	`let imageFrame = CGRectMake(frame.size.width / 2 - frame.size.width / 4, frame.size.height / 2 - frame.size.height / 4, frame.size.width / 2, frame.size.height / 2)`
`109`	`110`	`let imgCenterPoint = CGPointMake(imageFrame.origin.x + imageFrame.width / 2, imageFrame.origin.y + imageFrame.height / 2)`