From a03bf0ac98c836f94c72e829984aa1b151baed61 Mon Sep 17 00:00:00 2001
From: Kevin Lyles <kevinlyles@gmail.com>
Date: Sat, 19 Jun 2010 12:49:37 -0500
Subject: [PATCH] Converted matchesSocket to use a lookup table

---
 WeightsWatcher.lua |   55 +++++++++++++++++++++++++++++++++++-----------------
 1 file changed, 37 insertions(+), 18 deletions(-)

diff --git a/WeightsWatcher.lua b/WeightsWatcher.lua
index 613fe92..e133c6a 100644
--- a/WeightsWatcher.lua
+++ b/WeightsWatcher.lua
@@ -818,6 +818,39 @@ function WeightsWatcher.bestGemForSocket(socketColor, weightScale, qualityLimit)
 	return bestGem, bestWeight
 end
 
+local socketColors = {
+	["blue"] = {
+		["blue"] = true,
+		["green"] = true,
+		["prismatic"] = true,
+		["purple"] = true,
+	},
+	["red"] = {
+		["orange"] = true,
+		["prismatic"] = true,
+		["purple"] = true,
+		["red"] = true,
+	},
+	["meta"] = {
+		["meta"] = true,
+	},
+	["prismatic"] = {
+		["blue"] = true,
+		["green"] = true,
+		["orange"] = true,
+		["prismatic"] = true,
+		["purple"] = true,
+		["red"] = true,
+		["yellow"] = true,
+	},
+	["yellow"] = {
+		["green"] = true,
+		["orange"] = true,
+		["prismatic"] = true,
+		["yellow"] = true,
+	},
+}
+
 function WeightsWatcher.matchesSocket(gemId, socketColor)
 	local gemColor
 
@@ -833,26 +866,12 @@ function WeightsWatcher.matchesSocket(gemId, socketColor)
 		gemColor = gemId
 	end
 
-	if socketColor == "red" then
-		if gemColor == "red" or gemColor == "orange" or gemColor == "purple" or gemColor == "prismatic" then
-			return true
-		end
-	elseif socketColor == "blue" then
-		if gemColor == "blue" or gemColor == "green" or gemColor == "purple" or gemColor == "prismatic" then
-			return true
-		end
-	elseif socketColor == "yellow" then
-		if gemColor == "yellow" or gemColor == "orange" or gemColor == "green" or gemColor == "prismatic" then
-			return true
-		end
-	elseif socketColor == "meta" then
-		if gemColor == "meta" then
-			return true
-		end
-	else
+	if not socketColors[socketColor] then
 		print("Warning: Unrecognized socket color: " .. socketColor)
+		return false
 	end
-	return false
+
+	return socketColors[socketColor][gemColor] == true
 end
 
 function WeightsWatcher.calculateWeight(bareItemStats, itemStats, weightsScale)
-- 
1.7.9.5