Multiple times segment usage in different screen, btn == sender comparison not return true. about customsegmentedcontrol HOT 4 CLOSED

private var textColor: UIColor? = .lightGray
private var selectorViewColor: UIColor? = .red
private var selectorTextColor: UIColor? = .red

private var selectorWidth: Int {
    Int(frame.width) / self.segmentTitles.count
}

var normalTitleFont: UIFont = .light14
var selectedTitleFont: UIFont = .medium16

weak var delegate: LCSegmentedControlDelegate?

private(set) var selectedIndex = 0


private lazy var stackView: UIStackView =  {
    let stackView = UIStackView(arrangedSubviews: buttons)
    stackView.axis = .horizontal
    stackView.alignment = .fill
    stackView.distribution = .fillEqually
    return stackView
}()


//MARK:- Lifecycle
convenience init(titles: [String], textColor: UIColor? = .lightGray, selectedTextColor: UIColor? = .black, lineColor: UIColor? = .red, normalTitleFont: UIFont = .light14, selectedTitleFont: UIFont = .medium16) {
    self.init()
    self.segmentTitles = titles
    self.textColor = textColor
    self.selectorTextColor = selectedTextColor
    self.selectorViewColor = lineColor
    self.normalTitleFont = normalTitleFont
    self.selectedTitleFont = selectedTitleFont
}

//MARK:- Helper methods
override func draw(_ rect: CGRect) {
    super.draw(rect)
    self.backgroundColor = .clear
    updateView()
}

private func setButtonTitles(segmentTitles:[String]) {
    self.segmentTitles = segmentTitles
    self.updateView()
}

private func setIndex(index:Int) {
    buttons.forEach({ $0.setTitleColor(textColor, for: .normal) })
    let button = buttons[index]
    selectedIndex = index
    button.setTitleColor(selectorTextColor, for: .normal)
    let selectorPosition = frame.width/CGFloat(segmentTitles.count) * CGFloat(index)
    UIView.animate(withDuration: 0.2) {
        self.selectorView.frame.origin.x = selectorPosition
    }
}

@objc func buttonAction(sender:UIButton) {
    for (buttonIndex, btn) in buttons.enumerated() {
        btn.setTitleColor(textColor, for: .normal)
        btn.titleLabel?.font = normalTitleFont

        if btn.isEqual(sender) {
            let selectorPosition = frame.width / CGFloat(segmentTitles.count) * CGFloat(buttonIndex)
            selectedIndex = buttonIndex
            delegate?.segmentedControl(self, didIndexChanged: buttonIndex)
            UIView.animate(withDuration: 0.3) {
                self.selectorView.frame.origin.x = selectorPosition
            }
            btn.setTitleColor(selectorTextColor, for: .normal)
            btn.titleLabel?.font = selectedTitleFont

        }
    }
}

private func createButton() {
    buttons.removeAll()
    subviews.forEach({$0.removeFromSuperview()})
    for buttonTitle in segmentTitles {
        let button = UIButton(type: .system)
        button.setTitle(buttonTitle, for: .normal)
        button.addTarget(self, action:#selector(LCSegmentedControl.buttonAction(sender:)), for: .touchUpInside)
        button.setTitleColor(textColor, for: .normal)
        buttons.append(button)
    }
    buttons.first?.setTitleColor(selectorTextColor, for: .normal)
    buttons.first?.titleLabel?.font = selectedTitleFont

}

private func updateView() {
    createButton()
    configSelectorView()
    prepareLayout()
}

private func prepareLayout() {

    addSubviews(stackView, selectorView)

    stackView.snp.makeConstraints { (make) in
        make.edges.equalToSuperview()
    }
    
    selectorView.snp.makeConstraints { (make) in
        make.height.equalTo(2)
        make.top.equalTo(snp.bottom)
        make.width.equalTo(selectorWidth)
    }
}

private func configSelectorView() {
    selectorView.backgroundColor = selectorViewColor
}