From 59be86f54006c301201b1d147ed321c2011cbd5b Mon Sep 17 00:00:00 2001
From: Nadya Post <nadya@bournedigital.com.au>
Date: Sat, 26 Sep 2020 19:46:04 +1000
Subject: [PATCH] Add number of lines

---
 SwiftyPickerPopover/StringPickerPopover.swift       | 13 +++++++++++++
 .../StringPickerPopoverViewController.swift         |  1 +
 2 files changed, 14 insertions(+)

diff --git a/SwiftyPickerPopover/StringPickerPopover.swift b/SwiftyPickerPopover/StringPickerPopover.swift
index 4cdb698..c8eb7b0 100644
--- a/SwiftyPickerPopover/StringPickerPopover.swift
+++ b/SwiftyPickerPopover/StringPickerPopover.swift
@@ -38,6 +38,8 @@ public class StringPickerPopover: AbstractPopover {
     private(set) var fontSize: CGFloat?
     let kDefaultFontSize: CGFloat = 14
     
+    private(set) var numberOfLines: Int = 1
+    
     /// Convert a raw value to the string for displaying it
     private(set) var displayStringFor: DisplayStringForType?
     
@@ -81,6 +83,17 @@ public class StringPickerPopover: AbstractPopover {
         return self
     }
     
+    
+    /// Set numberOfLines
+    ///
+    /// - Parameter numberOfLines: Int how many lines is needed per row
+    /// - Returns: Self
+    public func setNumberOfLines(_ numberOfLines: Int) -> Self {
+        self.numberOfLines = numberOfLines
+        return self
+    }
+    
+    
     public func setFontSize(_ size: CGFloat) -> Self {
         self.fontSize = size
         return self
diff --git a/SwiftyPickerPopover/StringPickerPopoverViewController.swift b/SwiftyPickerPopover/StringPickerPopoverViewController.swift
index 2ce5e8a..69f8545 100644
--- a/SwiftyPickerPopover/StringPickerPopoverViewController.swift
+++ b/SwiftyPickerPopover/StringPickerPopoverViewController.swift
@@ -144,6 +144,7 @@ extension StringPickerPopoverViewController: UIPickerViewDelegate {
         label.text = adjustedValue
         label.attributedText = getAttributedText(image: popover.images?[row], text: adjustedValue)
         label.textAlignment = .center
+        label.numberOfLines = popover.numberOfLines
         return label
     }