Guess the main socket group based on DPS instead of number of gems

Author: Musholic

src/Classes/ImportTab.lua

@@ -827,7 +827,11 @@ function ImportTabClass:ImportItemsAndSkills(charData)
 		end)
 	end
 	if mainSkillEmpty then
-		self.build.mainSocketGroup = self:GuessMainSocketGroup()
+		local mainSocketGroup = self:GuessMainSocketGroup()
+		self.build.mainSocketGroup = mainSocketGroup
+		if mainSocketGroup ~= nil then
+			self.build.skillsTab.socketGroupList[mainSocketGroup].includeInFullDPS = true
+		end
 	end
 	self.build.itemsTab:PopulateSlots()
 	self.build.itemsTab:AddUndoState()
@@ -1114,17 +1118,22 @@ function ImportTabClass:ImportSocketedItems(item, socketedItems, slotName)
 	end
 end
 
--- Return the index of the group with the most gems
+-- Return the index of the group with the most DPS
 function ImportTabClass:GuessMainSocketGroup()
-	local largestGroupSize = 0
-	local largestGroupIndex = 1
-	for i, socketGroup in ipairs(self.build.skillsTab.socketGroupList) do
-		if #socketGroup.gemList > largestGroupSize then
-			largestGroupSize = #socketGroup.gemList
-			largestGroupIndex = i
+	local bestDps = 0
+	local bestSocketGroup = nil
+	for i, socketGroup in pairs(self.build.skillsTab.socketGroupList) do
+		self.build.mainSocketGroup = i
+		socketGroup.includeInFullDPS = true
+		local mainOutput = self.build.calcsTab.calcs.buildOutput(self.build, "MAIN").player.output
+		socketGroup.includeInFullDPS = false
+		local dps = mainOutput.FullDPS
+		if dps > bestDps then
+			bestDps = dps
+			bestSocketGroup = i
 		end
 	end
-	return largestGroupIndex
+	return bestSocketGroup
 end
 
 function HexToChar(x)