From 2235ec2648258bdf51c75280b2289cea0c970d0f Mon Sep 17 00:00:00 2001
From: Aki <nthirtyone@gmail.com>
Date: Fri, 22 Sep 2017 18:02:23 +0200
Subject: Added getIcon for Selector

---
 not/Selector.lua | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/not/Selector.lua b/not/Selector.lua
index ee6f0e3..5536b44 100644
--- a/not/Selector.lua
+++ b/not/Selector.lua
@@ -100,6 +100,12 @@ function Selector:getText ()
 	return tostring(self:getSelected())
 end
 
+function Selector:getIcon ()
+	if self.icons then
+		return self.icons[self.index]
+	end
+end
+
 function Selector:focus ()
 	self.focused = true
 	return true
@@ -134,8 +140,9 @@ function Selector:draw (scale)
 	end
 	love.graphics.draw(self.atlas, self.quads[self:getShapeString()][boxType], x*scale, y*scale, 0, scale, scale)
 	-- TODO: That is one way to draw icon for selected value. Find better one. See: `config/menus/host`.
-	if self.icons_atlas and self.icons_quads then
-		love.graphics.draw(self.icons_atlas, self.icons_quads[self.index], (x+2)*scale, (y+3)*scale, 0, scale, scale)
+	local icon = self:getIcon()
+	if icon then
+		love.graphics.draw(icon, (x+2)*scale, (y+3)*scale, 0, scale, scale)
 	end
 
 	love.graphics.setColor(255, 255, 255, 255)
-- 
cgit v1.1