--- a/DependencyLoader/Tree.lua	Sun Dec 05 03:39:26 2010 -0800
+++ b/DependencyLoader/Tree.lua	Fri Dec 10 00:21:17 2010 -0800
@@ -51,52 +51,38 @@
 end
 
 
-function tree:GetLoadBitfield()
-	local bitfield = self.root:GetLoadBitfield()
-	
+--	Checks if the tree rooted at the specified addon 
+--	can be loaded if all dependencies are enabled 
+--	and the UI is reloaded.
+--	Basically makes sure all dependencies are installed.
+function tree:CanLoad()
+	if not self.root:CanLoad() then
+		return false
+	end
+
+	--	check all the dependencies
 	local dependencies = {self.root:GetDependencies()}
 	for i, depName in pairs(dependencies) do
+		local dep = addonTable.classes.Addon:Get(depName)
+		
+		if not dep:CanLoad() then
+			return false
+		end
+		
 		local depTree = Tree:Get(depName)
 		
-		if depTree == nil then
-			return 0
-		end
-		
-		bitfield = bit.band(bitfield, depTree:GetLoadBitfield())
-	end
-	
-	return bitfield
-end
-
-
----	Checks if this tree can be force-loaded.
---	Does not check user settings nor if force-loading is actually available.
---	@return	canForceLoad	True if this tree can be force loaded, false otherwise
-function tree:CanForceLoad()
-	if self.root:IsLoaded() then
-		return true
-	end
-	
-	if not self.root:CanForceLoad() then
-		return false
-	end
-	
-	--	now check dependencies recursively
-	local dependencies = {self.root:GetDependencies()}
-	for i, depName in pairs(dependencies) do
-		local depTree = Tree:Get(depName)
-		
-		if not depTree:CanForceLoad() then
-			return false
+		if not depTree:CanLoad() then
+			return false		--	missing dependency
 		end
 	end
 	
-	return
+	return true
 end
 
 
---	NOTE: any tree that can be loaded on demand is also eligible for force-loading
---	Checks if the tree can be loaded on demand.
+---	Checks if the tree can be loaded on demand.
+--	Does not consider force-loading; dependencies must 
+--	be enabled and LoD, or loaded
 function tree:CanLoD()
 	--	since this will be used recursively, return true if 
 	--	this is already loaded.
@@ -111,7 +97,7 @@
 	end
 	
 	--	now check dependencies recursively
-	local dependencies = {root:GetDependencies()}
+	local dependencies = {self.root:GetDependencies()}
 	for i, depName in pairs(dependencies) do
 		local depTree = Tree:Get(depName)
 		
@@ -124,35 +110,82 @@
 end
 
 
---	Checks if the tree rooted at the specified addon 
---	can be loaded if all dependencies are enabled 
---	the UI is reloaded.
---	Basically makes sure all dependencies are installed.
-function tree:CanLoad()
-	if self.root:IsLoaded() then
-		return true
-		--	FIXME: 	deps may have been disabled
+---	Checks if this tree can be loaded on demand as long as
+--	force-loading is allowed for preparing dependencies.
+function tree:CanLoDWithForce()
+	if not self.root:CanLoD() then
+		return false
 	end
 	
-	--	TODO: make sure the root isn't incompatible
-	
 	--	now check dependencies recursively
-	--	FIXME: prevent infinite recursion
-
 	local dependencies = {self.root:GetDependencies()}
 	for i, depName in pairs(dependencies) do
+		local dep = addonTable.classes.Addon:Get(depName)
 		local depTree = Tree:Get(depName)
 		
-		if not depTree:CanLoad() then
-			return false		--	missing dependency
+		if not dep:CanLoad() then
+			return false
 		end
-
-		--	TODO: make sure it's compatible
+		
+		if dep:CanLoD() then
+			if not depTree:CanLoDWithForce() then
+				return false
+			end
+		elseif dep:CanForceLoad() then
+			if not depTree:CanForceLoad() then
+				return
+			end
+		end
 	end
 	
 	return true
 end
 
+
+
+---	Checks if this tree can be force-loaded.
+--	Does not check user settings nor if force-loading is actually available.
+--	@return	canForceLoad	True if this tree can be force loaded, false otherwise
+function tree:CanForceLoad()
+	--	TODO: remove redundencies (for now they're here for design flexibility)
+	
+	--	this addon must be loaded, able to load-on-demand, or able to force-load
+	if self.root:IsLoaded() then
+		return true
+	end
+	
+	if not self.root:CanLoad() then
+		return false
+	end
+	
+	if not self.root:CanForceLoad() then
+		return false
+	end
+	
+	--	now check dependencies recursively
+	local dependencies = {self.root:GetDependencies()}
+	for i, depName in pairs(dependencies) do
+		local dep = addonTable.classes.Addon:Get(depName)
+		local depTree = Tree:Get(depName)
+		
+		if not dep:CanLoad() then
+			return false
+		end
+		
+		if not dep:CanLoD() and not dep:CanForceLoad() then
+			return false
+		end
+		
+		if not depTree:CanForceLoad() then
+			return false
+		end
+	end
+	
+	return
+end
+
+
+
 --[[
 --	Loads the tree rooted at the specified addon.
 --	FIXME: load the root too or not?
@@ -172,6 +205,81 @@
 ]]
 
 
+function tree:PrepareForReload()
+		--	if force-loading is enabled then we have to queue this if the addon isn't lod
+	local function callback()
+		self.root:Enable()
+		
+		--	check dependencies
+		local dependencies = {self.root:GetDependencies()}
+		for i, depName in pairs(dependencies) do
+			Tree:Get(depName):PrepareForReload()
+		end
+
+		--	prepare embeds, if they are available separately
+		local embeds = {self.root:GetEmbeds(addon)}
+		for i, embedName in pairs(embeds) do
+			Tree:Get(embedName):PrepareForReload()
+		end
+	end
+	
+	if IsLoggedIn() then
+		callback()
+	else
+		--	TODO: replace with cleaner alternative?
+		local frame = CreateFrame("Frame")
+		frame:SetScript("OnEvent", callback)
+		frame:RegisterEvent("PLAYER_LOGIN")
+	end
+end
+
+
+--	I think the problem this solves is a major issue with 
+--	migrating to separate libs. think about it more and document 
+--	here and in project description
+function tree:PrepareForLoD()
+	--	assume root is LoD
+	
+	--	check dependencies
+	local dependencies = {self.root:GetDependencies(addon)}
+	for i, depName in pairs(dependencies) do
+		local depTree = Tree:Get(depName)
+		depTree:PrepareForLoD()
+		
+		--[[
+		if dep:CanLoD() then
+			--	don't load it now but make sure its dependencies are prepared
+			self:PrepareLoDTree(depName)
+		else
+			--	FIXME: if it's already loaded
+			--	force-load it now so we can load the parent on demand
+			self:ForceLoadTree(depName)
+		end
+		]]
+	end
+
+	--	prepare embeds, if they are available separately
+	local embeds = {self.root:GetEmbeds(addon)}	--	FIXME: addon?
+	for i, embedName in pairs(embeds) do
+		local embedTree = Tree:Get(embedName)
+		embedTree:PrepareForLoD()
+		
+		--[[
+		if embed then
+			if embed:CanLoD() then
+				--	don't load it now but make sure its dependencies are prepared
+				self:PrepareLoDTree(embedName)
+			else
+				--	FIXME: if it's already loaded
+				--	force-load it now so we can load the parent on demand
+				self:ForceLoadTree(depName)
+			end
+		end
+		]]
+	end
+end
+
+
 --	load the root too, since it may actually be a leaf
 function tree:ForceLoad()
 	--	load dependencies
@@ -215,65 +323,3 @@
 end
 ]]
 
-
---	I think the problem this solves is a major issue with 
---	migrating to separate libs. think about it more and document 
---	here and in project description
-function tree:PrepareForLoD()
-	--	assume root is LoD
-	
-	--	check dependencies
-	local dependencies = {root:GetDependencies(addon)}
-	for i, depName in pairs(dependencies) do
-		local depTree = Tree:Get(depName)
-		depTree:PrepareForLoD()
-		
-		--[[
-		if dep:CanLoD() then
-			--	don't load it now but make sure its dependencies are prepared
-			self:PrepareLoDTree(depName)
-		else
-			--	FIXME: if it's already loaded
-			--	force-load it now so we can load the parent on demand
-			self:ForceLoadTree(depName)
-		end
-		]]
-	end
-
-	--	prepare embeds, if they are available separately
-	local embeds = {root:GetEmbeds(addon)}	--	FIXME: addon?
-	for i, embedName in pairs(embeds) do
-		local embedTree = Tree:Get(embedName)
-		embedTree:PrepareForLoD()
-		
-		--[[
-		if embed then
-			if embed:CanLoD() then
-				--	don't load it now but make sure its dependencies are prepared
-				self:PrepareLoDTree(embedName)
-			else
-				--	FIXME: if it's already loaded
-				--	force-load it now so we can load the parent on demand
-				self:ForceLoadTree(depName)
-			end
-		end
-		]]
-	end
-end
-
-
-function tree:PrepareForReload()
-	self.root:Enable()
-	
-	--	check dependencies
-	local dependencies = {self.root:GetDependencies()}
-	for i, depName in pairs(dependencies) do
-		Tree:Get(depName):PrepareForReload()
-	end
-
-	--	prepare embeds, if they are available separately
-	local embeds = {self.root:GetEmbeds(addon)}
-	for i, embedName in pairs(embeds) do
-		Tree:Get(embedName):PrepareForReload()
-	end
-end