From a78c21ce52e8203ddc27a958af29996fffecded4 Mon Sep 17 00:00:00 2001 From: wasim Date: Sun, 19 Sep 2021 16:59:01 +0530 Subject: [PATCH] image of the first selection is not appearing issue resolved --- WMSegmentControl/Source/WMSegment.swift | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/WMSegmentControl/Source/WMSegment.swift b/WMSegmentControl/Source/WMSegment.swift index 3529d33..5236872 100644 --- a/WMSegmentControl/Source/WMSegment.swift +++ b/WMSegmentControl/Source/WMSegment.swift @@ -258,7 +258,7 @@ open class WMSegment: UIControl { } if btn == sender { - if let imageSelected = imagesSelected[safe: buttonIndex] { + if let imageSelected = imagesSelected[safe: buttonIndex], !imageSelected.isEmpty { btn.setImage(UIImage(named: imageSelected), for: .normal) } selectedSegmentIndex = buttonIndex @@ -291,7 +291,7 @@ open class WMSegment: UIControl { } if btn.tag == index { - if let imageSelected = imagesSelected[safe: buttonIndex] { + if let imageSelected = imagesSelected[safe: buttonIndex], !imageSelected.isEmpty { btn.setImage(UIImage(named: imageSelected), for: .normal) } selectedSegmentIndex = buttonIndex