Added image retrieval

Author: TimOliver

TOSegmentedControl/TOSegmentedControl.m

@@ -245,6 +245,21 @@ - (void)removeAllItems
     self.items = nil;
 }
 
+- (nullable UIImage *)imageForItemAtIndex:(NSInteger)index
+{
+    // Make sure the index provided is valid
+    if (index < 0 || index >= self.items.count) { return nil; }
+
+    // Return the item only if it is an image
+    id item = self.items[index];
+    if ([item isKindOfClass:[UIImage class]]) {
+        return item;
+    }
+    
+    // Return nil if a label or anything else
+    return nil;
+}
+
 #pragma mark - View Layout -
 
 - (void)layoutSubviews