UIViewのExtensionでUIImage変換、破線、斜線など実装する

個人アプリ開発で利用したUIViewのExtension

UIViewをUIImageに変換したり、周りを破線で囲んだり、斜めの打ち消し線を引いたりしたかったので以下のようなコードを書きました。

枠線を書くメソッドについては先日書いたStoryboardからUIViewの枠線、角丸、影を設定するで解決する方がいいかもしれません。

もっとスマートな実装があればコメントいただけると助かります。

enum DashedLineType {
    case All,Top,Down,Right,Left
}

extension UIView {

    // ViewをUIImageに変換して返却
    func castImage() -> UIImage? {
        UIGraphicsBeginImageContextWithOptions(self.frame.size, false, 0.0)

        guard let context = UIGraphicsGetCurrentContext() else {
            return nil
        }

        self.layer.render(in: context)

        guard let image = UIGraphicsGetImageFromCurrentImageContext() else {
            return nil
        }

        UIGraphicsEndImageContext()
        return image
    }

    // 枠線を描くメソッド
    func drawLine(color: UIColor, lineWidth: CGFloat) {
        let lineLayer = CAShapeLayer()
        lineLayer.frame = self.bounds
        lineLayer.strokeColor = color.cgColor
        lineLayer.lineWidth = lineWidth
        lineLayer.fillColor = nil
        lineLayer.path = UIBezierPath(rect: lineLayer.frame).cgPath

        self.layer.addSublayer(lineLayer)
    }

    // 斜めの打ち消し線を描くメソッド
    func drawCancelLine(color: UIColor, lineWidth: CGFloat) {
        let lineLayer = CAShapeLayer()
        let path: CGMutablePath = CGMutablePath()

        lineLayer.frame = self.bounds
        lineLayer.strokeColor = color.cgColor
        lineLayer.lineWidth = lineWidth
        lineLayer.fillColor = nil

        path.move(to: CGPoint(x: 0.0, y: 0.0))
        path.addLine(to: CGPoint.init(x: self.frame.size.width, y: self.frame.size.height))
        lineLayer.path = path

        self.layer.addSublayer(lineLayer)
    }

    // 破線を描くメソッド
    func drawDashedLine(color: UIColor, lineWidth: CGFloat, lineSize: NSNumber, spaceSize: NSNumber, type: DashedLineType) {
        let dashedLineLayer: CAShapeLayer = CAShapeLayer()
        dashedLineLayer.frame = self.bounds
        dashedLineLayer.strokeColor = color.cgColor
        dashedLineLayer.lineWidth = lineWidth
        dashedLineLayer.lineDashPattern = [lineSize, spaceSize]
        let path: CGMutablePath = CGMutablePath()

        switch type {

        case .All:
            dashedLineLayer.fillColor = nil
            dashedLineLayer.path = UIBezierPath(rect: dashedLineLayer.frame).cgPath
        case .Top:
            path.move(to: CGPoint(x: 0.0, y: 0.0))
            path.addLine(to: CGPoint(x: self.frame.size.width, y: 0.0))
            dashedLineLayer.path = path
        case .Down:
            path.move(to: CGPoint(x: 0.0, y: self.frame.size.height))
            path.addLine(to: CGPoint(x: self.frame.size.width, y: self.frame.size.height))
            dashedLineLayer.path = path
        case .Right:
            path.move(to: CGPoint(x: self.frame.size.width, y: 0.0))
            path.addLine(to: CGPoint(x: self.frame.size.width, y: self.frame.size.height))
            dashedLineLayer.path = path
        case .Left:
            path.move(to: CGPoint(x: 0.0, y: 0.0))
            path.addLine(to: CGPoint(x: 0.0, y: self.frame.size.height))
            dashedLineLayer.path = path
        }
        self.layer.addSublayer(dashedLineLayer)
    }

    // サブレイヤーが指定した数以上存在するかを返す
    func isSubLayer(count: Int) -> Bool {

        guard let sublayers = self.layer.sublayers else {
            // サブレイヤーがnilだったらfalseを返す
            return false
        }

        // サブレイヤーの有無を返却
        return sublayers.count >= count
    }

}