UIView+RoundCorners.swift

//
//  UIView+RoundCorners.swift
//  PullableView
//
//  Created by Dylan Gyesbreghs on 09/06/2017.
//  Copyright © 2017 Dylan Gyesbreghs. All rights reserved.
//

import UIKit

extension UIView {
    func roundCorners(_ corners: UIRectCorner, radius: CGFloat) {
        let path = UIBezierPath(roundedRect: self.bounds,
                                byRoundingCorners: corners,
                                cornerRadii: CGSize(width: radius, height: radius))
        let mask = CAShapeLayer()
        mask.path = path.cgPath
        self.layer.mask = mask
    }
}