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

Author: Musholic

src/Classes/ImportTab.lua

@@ -824,7 +824,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()
@@ -1111,17 +1115,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
-		end
-	end
-	return largestGroupIndex
+	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 bestSocketGroup
 end
 
 function HexToChar(x)