swift4 数字输入框

demo地址：
https://github.com/weiman152/InputNum

实现效果：

11.png

22.gif

使用：
默认：

inputNumView.setup()
inputNumView.delegate = self
inputNumView.beFirstResponder()

自定义：

testInput.setup(numOfRect: 6, margin: 10, pointWidth: 5, pointLocation: 4)
        testInput.delegate = self
        testInput.beFirstResponder()

实现：

//
//  InputNumView.swift
//  GuessTest
//
//  Created by iOS on 2018/6/13.
//  Copyright © 2018年 weiman. All rights reserved.
//

import UIKit

protocol InputNumViewDelegate: NSObjectProtocol {
    /// 输入是否完成
    func inputFinish(complete: Bool)
}

class InputNumView: UIView {
    
    var delegate: InputNumViewDelegate?
    
    lazy var textField: UITextField = {
        return $0
    }( UITextField() )
    
    private var labelArray = [UILabel]()
    private var pointLoaction = 0

    override func awakeFromNib() {
        super.awakeFromNib()
        setup()
    }
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        setup()
    }
    
    required init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
    }
    
    private func setup() {
        addSubview(textField)
        textField.keyboardType = .numberPad
        textField.delegate = self
    }
    
    private func setupLayout() {
        textField.frame = CGRect(x: 0, y: 0, width: 1, height: 1)
    }

}

extension InputNumView {
    
     /// 创建框框
     ///
     /// - Parameters:
     ///   - numOfRect: 框框个数
     ///   - margin: 框框之间的间距
     ///   - pointWidth: 中间小数点的宽度
     ///   - pointLocation: 小数点的位置
     func setup(numOfRect: Int = 4 ,
                margin: CGFloat = 10,
                pointWidth: CGFloat = 5,
                pointLocation: Int = 2) {
        
        self.pointLoaction = pointLocation
        let tempWidth: CGFloat = CGFloat(numOfRect + 1) * margin
        let width: CGFloat = (bounds.size.width - pointWidth - tempWidth) / CGFloat(numOfRect)
        let height: CGFloat = bounds.size.height
        
        for i in 0...numOfRect {
            
            if i < pointLocation {
                // 点的左边
                let frame = CGRect(x: CGFloat(i) * width + margin * CGFloat(i+1),
                                   y: 0,
                                   width: width,
                                   height: height)
                let label = createLabel(frame: frame)
                addSubview(label)
                labelArray.append(label)
            } else if i == pointLocation {
                // 点
                let frame = CGRect(x: CGFloat(i) * width + margin * CGFloat(i+1),
                                   y: 0,
                                   width: pointWidth,
                                   height: height)
                let label = createLabel(frame: frame, needBorder: false)
                label.text = "."
                addSubview(label)
            } else {
                // 点的右边
                let frame = CGRect(x: CGFloat(i-1) * width + pointWidth + margin * CGFloat(i+1),
                                   y: 0,
                                   width: width,
                                   height: height)
                let label = createLabel(frame: frame)
                addSubview(label)
                labelArray.append(label)
            }
        }
        // 第一个框变成选中状态
        if let firstLabel = labelArray.first {
            setSelectBorder(label: firstLabel)
        }
    }
    
    /// 弹出键盘
    func beFirstResponder() {
        textField.becomeFirstResponder()
    }
    
    /// 获取最终的输入结果
    func getInputText() -> String {
        var text = textField.text ?? ""
        // 插入小数点
        if text.count > pointLoaction {
            text.insert(".", at: text.index(text.startIndex, offsetBy: pointLoaction))
        }
        return text
    }
    
}

extension InputNumView {
    
    private func createLabel(frame: CGRect, needBorder: Bool = true) -> UILabel {
        let label = UILabel()
        label.frame = frame
        label.text = ""
        label.textColor = .darkGray
        label.font = .systemFont(ofSize: 22)
        label.textAlignment = .center
        if needBorder {
            setNormalBorder(label: label)
        }
        return label
    }
    
    /// 设置选中颜色
    private func setSelectBorder(label: UILabel) {
        label.layer.borderColor = UIColor.red.cgColor
        label.layer.borderWidth = 1
    }
    
    private func setNormalBorder(label: UILabel) {
        label.layer.borderColor = UIColor.blue.cgColor
        label.layer.borderWidth = 1
    }
    
}

extension InputNumView: UITextFieldDelegate {
    
    func textField(_ textField: UITextField, shouldChangeCharactersIn range: NSRange, replacementString string: String) -> Bool {
        
        guard range.location < labelArray.count else {
            print("输入无效")
            return false
        }
        
        for label in labelArray {
            setNormalBorder(label: label)
        }
        
        let showLabel = labelArray[range.location]
        showLabel.text = string
        
        if string.count > 0 {
            // 输入的时候
            if range.location < labelArray.count - 1 {
                let nextLabel = labelArray[range.location + 1]
                setSelectBorder(label: nextLabel)
            } else {
                setSelectBorder(label: labelArray[labelArray.count-1])
                delegate?.inputFinish(complete: true)
            }
        } else {
            // 删除的时候
            delegate?.inputFinish(complete: false)
            if range.location > 0 {
                let previousLabel = labelArray[range.location - 1]
                setSelectBorder(label: previousLabel)
            } else {
                setSelectBorder(label: labelArray[0])
            }
        }
        return true
    }
    
}