Optimize hot paths: reduce allocations and improve loop efficiency

Co-authored-by: CapsAdmin <204157+CapsAdmin@users.noreply.github.com>

Author: Copilot

nattlua/parser/node.lua

@@ -1034,31 +1034,37 @@ local NAN_TYPE = {}
 local ANY_TYPE = {}
 
 function META:AssociateType(obj)
-	self.inferred_types_done = self.inferred_types_done or {}
-	self.inferred_types = self.inferred_types or {}
-
-	do
-		local t = obj.Type
-		local hash = obj
-
-		if hash.Type == "symbol" then
-			hash = obj.Data
-		elseif obj.Type == "string" then
-			hash = obj.Data or STRING_TYPE
-		elseif obj.Type == "number" then
-			hash = obj.Data or NUMBER_TYPE
-
-			if hash ~= hash then hash = NAN_TYPE end
-		elseif obj.Type == "any" then
-			hash = ANY_TYPE
-		end
+	local done = self.inferred_types_done
+	local types = self.inferred_types
+	
+	if not done then
+		done = {}
+		self.inferred_types_done = done
+	end
+	
+	if not types then
+		types = {}
+		self.inferred_types = types
+	end
 
-		if self.inferred_types_done[hash] then return end
+	local obj_type = obj.Type
+	local hash = obj
 
-		self.inferred_types_done[hash] = true
+	if obj_type == "symbol" then
+		hash = obj.Data
+	elseif obj_type == "string" then
+		hash = obj.Data or STRING_TYPE
+	elseif obj_type == "number" then
+		hash = obj.Data or NUMBER_TYPE
+		if hash ~= hash then hash = NAN_TYPE end
+	elseif obj_type == "any" then
+		hash = ANY_TYPE
 	end
 
-	self.inferred_types[#self.inferred_types + 1] = obj
+	if done[hash] then return end
+
+	done[hash] = true
+	types[#types + 1] = obj
 end
 
 function META:GetAssociatedTypes()

nattlua/types/number.lua

@@ -92,21 +92,7 @@ end
 function META.Equal(a--[[#: TNumber]], b--[[#: TBaseType]])
 	if a.Type ~= b.Type then return false, "types differ" end
 
-	do
-		return a.Hash == b.Hash
-	end
-
-	if a.Data == b.Data then return true, "max values are equal" end
-
-	if not a.Data and not b.Data then
-		return true, "no literal data in either value"
-	else
-		if a:IsNan() and b:IsNan() then return true, "both values are nan" end
-
-		return a.Data == b.Data, "literal values are equal"
-	end
-
-	return false, "values are not equal"
+	return a.Hash == b.Hash, "hash values are equal"
 end
 
 function META:IsLiteral()

nattlua/types/table.lua

@@ -675,8 +675,11 @@ function META:FindKeyValWide(key--[[#: TBaseType]], reverse--[[#: boolean | nil]
 	if keyval then return keyval end
 
 	local reasons = {}
+	local data = self.Data
+	local len = #data
 
-	for i, keyval in ipairs(self.Data) do
+	for i = 1, len do
+		local keyval = data[i]
 		if key:Equal(keyval.key) then return keyval end
 
 		local ok, reason

nattlua/types/union.lua

@@ -81,8 +81,9 @@ function META.Equal(
 end
 
 function META:GetHash(visited)--[[#: string]]
-	if self:GetCardinality() == 1 then
-		return (self.Data[1]--[[# as any]]):GetHash()
+	local data = self.Data
+	if #data == 1 then
+		return (data[1]--[[# as any]]):GetHash()
 	end
 
 	visited = visited or {}
@@ -91,14 +92,16 @@ function META:GetHash(visited)--[[#: string]]
 
 	visited[self] = "*circular*"
 	local types = {}
+	local len = #data
 
-	for i, v in ipairs(self.Data) do
-		types[i] = v:GetHash(visited)
+	for i = 1, len do
+		types[i] = data[i]:GetHash(visited)
 	end
 
 	table_sort(types)
-	visited[self] = table.concat(types, "|")
-	return visited[self]--[[# as string]]
+	local hash = table_concat(types, "|")
+	visited[self] = hash
+	return hash
 end
 
 local sort = function(a--[[#: string]], b--[[#: string]])
@@ -110,9 +113,12 @@ function META:__tostring()
 
 	local s = {}
 	self.suppress = true
+	
+	local data = self.Data
+	local len = #data
 
-	for i, v in ipairs(self.Data) do
-		s[i] = tostring(v)
+	for i = 1, len do
+		s[i] = tostring(data[i])
 	end
 
 	if not s[1] then
@@ -122,7 +128,7 @@ function META:__tostring()
 
 	self.suppress = false
 
-	if #s == 1 then return (s[1]--[[# as string]]) .. "|" end
+	if len == 1 then return (s[1]--[[# as string]]) .. "|" end
 
 	table_sort(s, sort)
 	return table_concat(s, " | ")
@@ -155,11 +161,16 @@ local function remove(self--[[#: TUnion]], index--[[#: number]])
 end
 
 local function find_index(self--[[#: TUnion]], obj--[[#: any]])
-	for i = 1, #self.Data do
-		local v = self.Data[i]--[[# as any]]
+	local data = self.Data
+	local len = #data
+	local obj_type = obj.Type
 
-		if v:Equal(obj) then
-			if v.Type ~= "function" or v:GetFunctionBodyNode() == obj:GetFunctionBodyNode() then
+	for i = 1, len do
+		local v = data[i]--[[# as any]]
+		
+		-- Early exit if types don't match
+		if v.Type == obj_type and v:Equal(obj) then
+			if obj_type ~= "function" or v:GetFunctionBodyNode() == obj:GetFunctionBodyNode() then
 				return i
 			end
 		end