如何在UIView和UIImageView中重用相同的IBDesignable代碼

Question

我有一個IBDesignable類，用於制作邊界線和角半徑。 問題是我必須為UIView和UIImageview編寫單獨的類。 是否有一個簡單的解決方案。 提前致謝。

 @IBDesignable
    class View: UIView {
        //MARK:- Corner Radius
        @IBInspectable
        var cornerRadius : CGFloat =  0{
          didSet {
                self.layer.masksToBounds = true
                self.layer.cornerRadius = cornerRadius   
            } 
        }
    }

影像檢視

 @IBDesignable
        class ImageView: UIImageView {
            //MARK:- Corner Radius
            @IBInspectable
            var cornerRadius : CGFloat =  0{
              didSet {
                    self.layer.masksToBounds = true
                    self.layer.cornerRadius = cornerRadius   
                } 
            }
        }

Answer 1

如下創建擴展

extension UIView
{
    @IBInspectable var cornerRadius: CGFloat {
        get {
            return layer.cornerRadius;
        }
        set(value) {
            layer.cornerRadius = value;
        }
    }

    var rounded: Bool {
        return layer.cornerRadius > 0.0;
    }
}