ios / ytwatch

//
//  UIView+Extension.swift
//  taiji
//
//  Created by 泰极科技 on 2019/5/15.
//  Copyright © 2019 泰极科技. All rights reserved.
//

import UIKit



public extension UIView {
    func screenSnapScreen() -> UIImage? {
        //截取整个屏幕
        guard let window = UIApplication.shared.keyWindow else { return nil }
        // 用下面这行而不是UIGraphicsBeginImageContext()，因为前者支持Retina
        UIGraphicsBeginImageContextWithOptions(window.bounds.size, false, 0.0)
        window.layer.render(in: UIGraphicsGetCurrentContext()!)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
       
        return UIImage.init(data: image?.jpegData(compressionQuality: 0.5) ?? Data()) //image
    }
    
    func asImage() -> UIImage? {
        if #available(iOS 10.0, *) {
            let renderer = UIGraphicsImageRenderer(bounds: bounds)
            return renderer.image { rendererContext in
                layer.render(in: rendererContext.cgContext)
            }
        } else {
            UIGraphicsBeginImageContextWithOptions(frame.size, true, UIScreen.main.scale)
            layer.render(in: UIGraphicsGetCurrentContext()!)
            let image = UIGraphicsGetImageFromCurrentImageContext()
            
            UIGraphicsEndImageContext()
            return image
            // Fallback on earlier versions
        }
     
     }
    /// 截屏Image
    var captureImage: UIImage? {
        
        // 参数①：截屏区域  参数②：是否透明  参数③：清晰度
        UIGraphicsBeginImageContextWithOptions(frame.size, true, UIScreen.main.scale)
        layer.render(in: UIGraphicsGetCurrentContext()!)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        
        UIGraphicsEndImageContext()
        return image
    }
    
    
    func setCorners(corners: UIRectCorner, radio: CGFloat) {
        let maskPath = UIBezierPath(roundedRect: self.bounds, byRoundingCorners: corners, cornerRadii: CGSize(width: radio, height: radio))
        let maskLayer = CAShapeLayer()
        maskLayer.frame = self.bounds
        maskLayer.path = maskPath.cgPath
        self.layer.mask = maskLayer
    }
    
    func setRadio(radio: CGFloat) {
        self.layer.cornerRadius = radio
        //        self.layer.mask = maskLayer
    }
    func drawFertileView(type: Int) {
        backgroundColor = UIColor.rgbColorFromHex(0xD8E9FA)
        var corners: UIRectCorner = []
        if type == 1 {
            corners = [.topLeft, .bottomLeft]
        } else if type == 3 {
            corners = [.topRight, .bottomRight]
        } else if type == 4 {
            corners = .allCorners
        }
        
        let maskPath = UIBezierPath(roundedRect: CGRect(x: 0, y: bounds.height/2-11, width: bounds.width, height: 22), byRoundingCorners: corners, cornerRadii: CGSize(width: 11, height: 11))
        let maskLayer = CAShapeLayer()
        maskLayer.frame = self.bounds
        maskLayer.path = maskPath.cgPath
        self.layer.mask = maskLayer
    }
    
    func drawMenstrualView(type: Int) {
//        left = 1, middle, right, full, none
        backgroundColor = UIColor.rgbColorFromHex(0xFFDFEA)
        let bezierPath = UIBezierPath()
        let circlePath = UIBezierPath(ovalIn: CGRect(x: bounds.width/2-16, y: bounds.height/2-16, width: 32, height: 32))
        var squarePath = UIBezierPath()
        if type == 1 {
            squarePath = UIBezierPath(roundedRect: CGRect(x: bounds.width/2, y: bounds.height/2-6.5, width: bounds.width, height: 13), cornerRadius: 0)
        } else if type == 3 {
            squarePath = UIBezierPath(roundedRect: CGRect(x: 0, y: bounds.height/2-6.5, width: bounds.width/2, height: 13), cornerRadius: 0)
        } else if type == 2 {
            squarePath = UIBezierPath(roundedRect: CGRect(x: 0, y: bounds.height/2-6.5, width: bounds.width, height: 13), cornerRadius: 0)
        } else {
            squarePath = UIBezierPath(roundedRect: CGRect(x: bounds.width/2-1, y: bounds.height/2-1, width: 2, height: 2), cornerRadius: 0)
        }
        squarePath.append(circlePath)
        bezierPath.append(squarePath)
        let maskLayer = CAShapeLayer()
        maskLayer.frame = self.bounds
        maskLayer.path = bezierPath.cgPath
        self.layer.mask = maskLayer
    }
    
    func setDefault() {
        self.layer.shadowColor = UIColor.rgbaColorFromHex(0x3D5A80, alpha: 0.1).cgColor
        self.layer.shadowOffset = CGSize.zero
        self.layer.shadowRadius = 6
        self.layer.shadowOpacity = 1
        self.layer.cornerRadius = 6
    }
    
    
    
    
    
    
    // MARK: 添加渐变色图层
    func gradientColor(_ text: String, _ font: UIFont, offsetY: CGFloat = 0, textAlignment: NSTextAlignment = NSTextAlignment.center) {
        //        layoutIfNeeded()
        var b: CGRect!
        if offsetY == 0 {
            b = self.bounds
        } else {
            b = CGRect(x: 0, y: offsetY, width: self.bounds.size.width, height: self.bounds.size.height)
        }
        
        let label = UILabel()
        label.text = text
        label.font = font
        label.textAlignment = textAlignment
        label.frame = b
        
        
        let gradient = CAGradientLayer()
        gradient.colors = [UIColor(red: 0.23, green: 0.98, blue: 0.99, alpha: 1).cgColor, UIColor(red: 0.21, green: 0.71, blue: 1, alpha: 1).cgColor]
        gradient.startPoint = CGPoint(x: 0, y: 0)
        gradient.endPoint = CGPoint(x: 1.1155598958333333, y: 1)
        gradient.frame = b
        self.layer.insertSublayer(gradient, at: 0)
        self.mask = label
        
        
        //        self.layer.masksToBounds = false
        //        guard startPoint.x >= 0, startPoint.x <= 1, startPoint.y >= 0, startPoint.y <= 1, endPoint.x >= 0, endPoint.x <= 1, endPoint.y >= 0, endPoint.y <= 1 else {
        //            return
        //        }
        //        // 外界如果改变了self的大小，需要先刷新
        //        layoutIfNeeded()
        //
        //        var gradientLayer: CAGradientLayer!
        //        removeGradientLayer()
        //
        //        gradientLayer = CAGradientLayer()
        //        gradientLayer.frame = self.layer.bounds
        //        gradientLayer.startPoint = startPoint
        //        gradientLayer.endPoint = endPoint
        //        gradientLayer.colors = colors
        //        gradientLayer.cornerRadius = self.layer.cornerRadius
        //        gradientLayer.masksToBounds = true
        //        // 渐变图层插入到最底层，避免在uibutton上遮盖文字图片
        //        self.layer.insertSublayer(gradientLayer, at: 0)
        //        self.backgroundColor = UIColor.clear
        //        // self如果是UILabel，masksToBounds设为true会导致文字消失
        //        self.layer.masksToBounds = false
    }
    
    func gradientColor1(_ text: String, _ font: UIFont, cornerRadius: CGFloat = 4) {
        
        layoutIfNeeded()
        
        
        let button = UIButton()
        button.setTitle(text, for: .normal)
        button.frame = self.bounds
        
        button.layer.cornerRadius = cornerRadius
        button.layer.borderWidth = 1
        button.layer.masksToBounds = true
        
        button.titleLabel?.font = font
        button.titleLabel?.textAlignment = .center
        let gradient = CAGradientLayer()
        gradient.colors = [UIColor(red: 0.23, green: 0.98, blue: 0.99, alpha: 1).cgColor, UIColor(red: 0.21, green: 0.71, blue: 1, alpha: 1).cgColor]
        gradient.startPoint = CGPoint(x: 0, y: 0)
        gradient.endPoint = CGPoint(x: 1.1155598958333333, y: 1)
        gradient.frame = self.bounds
        self.layer.insertSublayer(gradient, at: 0)
        self.mask = button
        
    }
    
    
    // MARK: 移除渐变图层
    // （当希望只使用backgroundColor的颜色时，需要先移除之前加过的渐变图层）
    func removeGradientLayer() {
        self.mask = nil
        if let sl = self.layer.sublayers {
            for layer in sl {
                if layer.isKind(of: CAGradientLayer.self) {
                    layer.removeFromSuperlayer()
                }
            }
        }
    }
    func removeShapeLayer() {
        self.mask = nil
        if let sl = self.layer.sublayers {
            for layer in sl {
                if layer.isKind(of: CAShapeLayer.self) {
                    layer.removeFromSuperlayer()
                }
            }
        }
    }
    
}

extension UIScrollView {
    
    /// 截长屏Image
    var captureLongImage: UIImage? {
        
        var image: UIImage? = nil
        UIGraphicsBeginImageContext(contentSize)
        
        let formerContentOffset = contentOffset
        let formerFrame = frame
        self.contentOffset = .zero
        self.frame = CGRect(origin: CGPoint.zero, size: contentSize)
        
        UIGraphicsBeginImageContextWithOptions(contentSize, false, UIScreen.main.scale)
        if let currentContext = UIGraphicsGetCurrentContext() {
            self.layer.render(in: currentContext)
            image = UIGraphicsGetImageFromCurrentImageContext()
        }
        UIGraphicsEndImageContext()
        
        self.contentOffset = formerContentOffset
        self.frame = formerFrame
        
        return image
        
    }
    
    
}

private var SwViewCaptureKey_IsCapturing: String = "SwViewCapture_AssoKey_isCapturing"

public extension UIView {
    
    @objc func swSetFrame(_ frame: CGRect) {
        // Do nothing, use for swizzling
    }
    
    var isCapturing:Bool! {
        get {
            let num = objc_getAssociatedObject(self, &SwViewCaptureKey_IsCapturing)
            if num == nil {
                return false
            }
            
            if let numObj = num as? NSNumber {
                return numObj.boolValue
            }else {
                return false
            }
        }
        set(newValue) {
            let num = NSNumber(value: newValue as Bool)
            objc_setAssociatedObject(self, &SwViewCaptureKey_IsCapturing, num, objc_AssociationPolicy.OBJC_ASSOCIATION_RETAIN_NONATOMIC)
        }
    }
    

    
    func swCapture(_ completionHandler: (_ capturedImage: UIImage?) -> Void) {
        
        self.isCapturing = true
        
        let bounds = self.bounds
        
        UIGraphicsBeginImageContextWithOptions(bounds.size, false, UIScreen.main.scale)
        
        let context = UIGraphicsGetCurrentContext()
        context?.saveGState()
        context?.translateBy(x: -self.frame.origin.x, y: -self.frame.origin.y);
        
//        if (swContainsWKWebView()) {
//            self.drawHierarchy(in: bounds, afterScreenUpdates: true)
//        }else{
            self.layer.render(in: context!)
//        }
        let capturedImage = UIGraphicsGetImageFromCurrentImageContext()
        
        context?.restoreGState();
        UIGraphicsEndImageContext()
        
        self.isCapturing = false
        
        completionHandler(capturedImage)
    }
}


extension UIScrollView {
    
    public func swContentCapture (_ completionHandler: @escaping (_ capturedImage: UIImage?) -> Void) {
        
        self.isCapturing = true
        
        // Put a fake Cover of View
        let snapShotView = self.snapshotView(afterScreenUpdates: false)
        snapShotView?.frame = CGRect(x: self.frame.origin.x, y: self.frame.origin.y, width: (snapShotView?.frame.size.width)!, height: (snapShotView?.frame.size.height)!)
        self.superview?.addSubview(snapShotView!)
        
        // Backup all properties of scrollview if needed
        let bakFrame     = self.frame
        let bakOffset    = self.contentOffset
        let bakSuperView = self.superview
        let bakIndex     = self.superview?.subviews.firstIndex(of: self)
        
        // Scroll To Bottom show all cached view
        if self.frame.size.height < self.contentSize.height {
            self.contentOffset = CGPoint(x: 0, y: self.contentSize.height - self.frame.size.height)
        }
        
        self.swRenderImageView({ [weak self] (capturedImage) -> Void in
            // Recover View
            
            let strongSelf = self!
            
            strongSelf.removeFromSuperview()
            strongSelf.frame = bakFrame
            strongSelf.contentOffset = bakOffset
            bakSuperView?.insertSubview(strongSelf, at: bakIndex!)
            
            snapShotView?.removeFromSuperview()
            
            strongSelf.isCapturing = false
            
            completionHandler(capturedImage)
        })

        
    }
    
    fileprivate func swRenderImageView(_ completionHandler: @escaping (_ capturedImage: UIImage?) -> Void) {
        // Rebuild scrollView superView and their hold relationship
        let swTempRenderView = UIView(frame: CGRect(x: 0, y: 0, width: self.contentSize.width, height: self.contentSize.height))
        self.removeFromSuperview()
        swTempRenderView.addSubview(self)
        
        self.contentOffset = CGPoint.zero
        self.frame         = swTempRenderView.bounds
        
        // Swizzling setFrame
        let method: Method = class_getInstanceMethod(object_getClass(self), #selector(setter: UIView.frame))!
        let swizzledMethod: Method = class_getInstanceMethod(object_getClass(self), #selector(self.swSetFrame(_:)))!
        method_exchangeImplementations(method, swizzledMethod)
        
        // Sometimes ScrollView will Capture nothing without defer;
        DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + Double(Int64(0.3 * Double(NSEC_PER_SEC))) / Double(NSEC_PER_SEC)) { () -> Void in
            let bounds = self.bounds
            UIGraphicsBeginImageContextWithOptions(bounds.size, false, UIScreen.main.scale)
            
//            if (self.swContainsWKWebView()) {
//                self.drawHierarchy(in: bounds, afterScreenUpdates: true)
//            }else{
                self.layer.render(in: UIGraphicsGetCurrentContext()!)
//            }
            let capturedImage = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()
            
            method_exchangeImplementations(swizzledMethod, method)
            
            completionHandler(capturedImage)
        }
    }
    
    
    // Simulate People Action, all the `fixed` element will be repeate
    // SwContentCapture will capture all content without simulate people action, more perfect.
    public func swContentScrollCapture (_ completionHandler: @escaping (_ capturedImage: UIImage?) -> Void) {
        
        self.isCapturing = true
        
        // Put a fake Cover of View
        let snapShotView = self.snapshotView(afterScreenUpdates: true)
        snapShotView?.frame = CGRect(x: self.frame.origin.x, y: self.frame.origin.y, width: (snapShotView?.frame.size.width)!, height: (snapShotView?.frame.size.height)!)
        self.superview?.addSubview(snapShotView!)
        
        // Backup
        let bakOffset    = self.contentOffset
        
        // Divide
        let page  = floorf(Float(self.contentSize.height / self.bounds.height))
        
        UIGraphicsBeginImageContextWithOptions(self.contentSize, false, UIScreen.main.scale)
        
        self.swContentScrollPageDraw(0, maxIndex: Int(page), drawCallback: { [weak self] () -> Void in
            let strongSelf = self
            
            let capturedImage = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()
            
            // Recover
            strongSelf?.setContentOffset(bakOffset, animated: false)
            snapShotView?.removeFromSuperview()
            
            strongSelf?.isCapturing = false
            
            completionHandler(capturedImage)
            })
        
    }
    
    fileprivate func swContentScrollPageDraw (_ index: Int, maxIndex: Int, drawCallback: @escaping () -> Void) {
        
        self.setContentOffset(CGPoint(x: 0, y: CGFloat(index) * self.frame.size.height), animated: false)
        let splitFrame = CGRect(x: 0, y: CGFloat(index) * self.frame.size.height, width: bounds.size.width, height: bounds.size.height)
        
        DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + Double(Int64(0.3 * Double(NSEC_PER_SEC))) / Double(NSEC_PER_SEC)) { () -> Void in
            self.drawHierarchy(in: splitFrame, afterScreenUpdates: true)
            
            if index < maxIndex {
                self.swContentScrollPageDraw(index + 1, maxIndex: maxIndex, drawCallback: drawCallback)
            }else{
                drawCallback()
            }
        }
    }
}