Fix crash with max count.

spromicky · Jan 18, 2017 · dd1825a · dd1825a
1 parent fa6f175
commit dd1825a
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/StepSlider/source/StepSlider/StepSlider.m b/StepSlider/source/StepSlider/StepSlider.m
@@ -109,6 +109,7 @@ - (void)generalSetup
     _sliderCircleColor  = [UIColor whiteColor];
     _labelOffset        = 20.f;
     _labelColor         = [UIColor whiteColor];
+    [self updateMaxRadius];
 
     [self setNeedsLayout];
 }
@@ -508,16 +509,24 @@ - (void)setLabels:(NSArray<NSString *> *)labels
         _labels = labels;
 
         if (_labels.count > 0) {
-            self.maxCount = _labels.count;
-        } else {
-            [self updateIndex];
-            [self setNeedsLayout];
+            _maxCount = _labels.count;
         }
+
+        [self updateIndex];
+        [self setNeedsLayout];
+    }
+}
+
+- (void)setMaxCount:(NSUInteger)maxCount
+{
+    if (_maxCount != maxCount && !self.labels.count) {
+        _maxCount = maxCount;
+        [self updateIndex];
+        [self setNeedsLayout];
     }
 }
 
 GENERATE_SETTER(index, NSUInteger, setIndex, [self updateIndex]; [self sendActionsForControlEvents:UIControlEventValueChanged];);
-GENERATE_SETTER(maxCount, NSUInteger, setMaxCount, [self updateIndex];);
 
 GENERATE_SETTER(trackHeight, CGFloat, setTrackHeight, [self updateDiff];);
 GENERATE_SETTER(trackCircleRadius, CGFloat, setTrackCircleRadius, [self updateDiff]; [self updateMaxRadius];);