Dataset logic cleanup

Source/Charts/Data/Implementations/ChartBaseDataSet.swift

@@ -24,7 +24,7 @@ open class ChartBaseDataSet: NSObject, ChartDataSetProtocol
         valueColors.append(NSUIColor.black)
     }
 
-    @objc public init(label: String?)
+    @objc public init(label: String)
     {
         super.init()
 

Source/Charts/Data/Implementations/Standard/BarChartDataSet.swift

@@ -29,7 +29,7 @@ open class BarChartDataSet: BarLineScatterCandleBubbleChartDataSet, BarChartData
         initialize()
     }
 
-    public override init(values: [ChartDataEntry]?, label: String?)
+    public override init(values: [ChartDataEntry], label: String)
     {
         super.init(values: values, label: label)
         initialize()
@@ -50,68 +50,36 @@ open class BarChartDataSet: BarLineScatterCandleBubbleChartDataSet, BarChartData
     {
         _entryCountStacks = 0
 
-        for i in 0 ..< entries.count
-        {
-            if let vals = entries[i].yValues
-            {
-                _entryCountStacks += vals.count
-            }
-            else
-            {
-                _entryCountStacks += 1
-            }
-        }
+        entries.forEach { _entryCountStacks += $0.yValues?.count ?? 1 }
     }
 
     /// calculates the maximum stacksize that occurs in the Entries array of this DataSet
     private func calcStackSize(entries: [BarChartDataEntry])
     {
-        for i in 0 ..< entries.count
+        for e in entries where (e.yValues?.count ?? 0) > _stackSize
         {
-            if let vals = entries[i].yValues
-            {
-                if vals.count > _stackSize
-                {
-                    _stackSize = vals.count
-                }
-            }
+            _stackSize = e.yValues!.count
         }
     }
 
     open override func calcMinMax(entry e: ChartDataEntry)
     {
-        guard let e = e as? BarChartDataEntry
+        guard let e = e as? BarChartDataEntry,
+            !e.y.isNaN
             else { return }
 
-        if !e.y.isNaN
+        if e.yValues == nil
         {
-            if e.yValues == nil
-            {
-                if e.y < _yMin
-                {
-                    _yMin = e.y
-                }
-
-                if e.y > _yMax
-                {
-                    _yMax = e.y
-                }
-            }
-            else
-            {
-                if -e.negativeSum < _yMin
-                {
-                    _yMin = -e.negativeSum
-                }
-
-                if e.positiveSum > _yMax
-                {
-                    _yMax = e.positiveSum
-                }
-            }
-
-            calcMinMaxX(entry: e)
+            _yMin = min(e.y, _yMin)
+            _yMax = max(e.y, _yMax)
         }
+        else
+        {
+            _yMin = min(-e.negativeSum, _yMin)
+            _yMax = max(e.positiveSum, _yMax)
+        }
+
+        calcMinMaxX(entry: e)
     }
 
     /// - returns: The maximum number of bars that can be stacked upon another in this DataSet.

Source/Charts/Data/Implementations/Standard/BubbleChartDataSet.swift

@@ -30,12 +30,7 @@ open class BubbleChartDataSet: BarLineScatterCandleBubbleChartDataSet, BubbleCha
 
         super.calcMinMax(entry: e)
 
-        let size = e.size
-
-        if size > _maxSize
-        {
-            _maxSize = size
-        }
+        _maxSize = max(e.size, maxSize)
     }
 
     // MARK: - Styling functions and accessors

Source/Charts/Data/Implementations/Standard/CandleChartDataSet.swift

@@ -21,7 +21,7 @@ open class CandleChartDataSet: LineScatterCandleRadarChartDataSet, CandleChartDa
         super.init()
     }
 
-    public override init(values: [ChartDataEntry]?, label: String?)
+    public override init(values: [ChartDataEntry], label: String)
     {
         super.init(values: values, label: label)
     }
@@ -32,68 +32,42 @@ open class CandleChartDataSet: LineScatterCandleRadarChartDataSet, CandleChartDa
     {
         guard let e = e as? CandleChartDataEntry
             else { return }
-
-        if e.low < _yMin
-        {
-            _yMin = e.low
-        }
-
-        if e.high > _yMax
-        {
-            _yMax = e.high
-        }
-
+
+        _yMin = min(e.low, _yMin)
+        _yMax = max(e.high, _yMax)
+
         calcMinMaxX(entry: e)
     }
 
     open override func calcMinMaxY(entry e: ChartDataEntry)
     {
         guard let e = e as? CandleChartDataEntry
             else { return }
-
-        if e.high < _yMin
-        {
-            _yMin = e.high
-        }
-        if e.high > _yMax
-        {
-            _yMax = e.high
-        }
-
-        if e.low < _yMin
-        {
-            _yMin = e.low
-        }
-        if e.low > _yMax
-        {
-            _yMax = e.low
-        }
+
+        _yMin = min(e.low, _yMin)
+        _yMax = max(e.high, _yMin)
+
+        _yMin = min(e.low, _yMax)
+        _yMax = max(e.high, _yMax)
     }
 
     // MARK: - Styling functions and accessors
 
     /// the space between the candle entries
     ///
     /// **default**: 0.1 (10%)
-    private var _barSpace = CGFloat(0.1)
-    
+    private var _barSpace: CGFloat = 0.1
+
     /// the space that is left out on the left and right side of each candle,
     /// **default**: 0.1 (10%), max 0.45, min 0.0
     open var barSpace: CGFloat
     {
         set
         {
-            if newValue < 0.0
-            {
-                _barSpace = 0.0
-            }
-            else if newValue > 0.45
-            {
-                _barSpace = 0.45
-            }
-            else
-            {
-                _barSpace = newValue
+            switch newValue {
+            case ..<0: _barSpace = 0.0
+            case 0.45...: _barSpace = 0.45
+            default: _barSpace = newValue
             }
         }
         get

Source/Charts/Data/Implementations/Standard/ChartData.swift

@@ -296,7 +296,7 @@ open class ChartData: NSObject, ExpressibleByArrayLiteral
     }
 
     /// - returns: All colors used across all DataSet objects this object represents.
-    @objc open func getColors() -> [NSUIColor]
+    @objc open var colors: [NSUIColor]
     {
         return reduce(into: []) { $0 += $1.colors }
     }