Add display options for first and second PickerInlineRow (#2246)

xmartlabs · May 26, 2023 · 9ead011 · 9ead011
1 parent c255e4b
commit 9ead011
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Source/Rows/PickerInlineRow.swift b/Source/Rows/PickerInlineRow.swift
@@ -102,6 +102,12 @@ open class _DoublePickerInlineRow<A, B> : Row<PickerInlineCell<Tuple<A, B>>>, No
 
     /// Options for second component given the selected value from the first component. Will be called often so should be O(1)
     public var secondOptions: ((A) -> [B]) = { _ in [] }
+
+    /// Modify the displayed values for the first picker row.
+    public var displayValueForFirstRow: ((A) -> (String)) = { a in return String(describing: a) }
+
+    /// Modify the displayed values for the second picker row.
+    public var displayValueForSecondRow: ((B) -> (String)) = { b in return String(describing: b) }
 
     public var noValueDisplayText: String?
 
@@ -140,6 +146,8 @@ public final class DoublePickerInlineRow<A, B> : _DoublePickerInlineRow<A, B>, R
     public func setupInlineRow(_ inlineRow: InlineRow) {
         inlineRow.firstOptions = firstOptions
         inlineRow.secondOptions = secondOptions
+        inlineRow.displayValueForFirstRow = displayValueForFirstRow
+        inlineRow.displayValueForSecondRow = displayValueForSecondRow
         inlineRow.displayValueFor = self.displayValueFor
         inlineRow.cell.height = { UITableView.automaticDimension }
     }