-----------------
-----------------
-- DISPLAYERS  --
-----------------
-----------------

-- Displayer are objects attached to instance in the scenario
-- They react to modification events (creations of new objects such as nps, groups ...)
-- and update their display accordingly
-- There is zero or one displayer attached per category of display for each instance in the map
-- For now this include :
-- UI displayers       :  - They update the scenario window to display new things added to the map
-- Property displayers :  - They update the property sheet for an instance when one is displayed
-- Visual displayers   :  - For now they are only implemented in C++. Their fonction is to update the display of a instance
--                        -  in the 3D scene
--
-- Displayer at attached at creation time by the C++ code
-- The displayers to add to a specific object are given its the class definition
-- (see r2_base_class.lua for details)

-- helper : update the context toolbar for the given instance if it is the current selected instance
local function updateContextToolbar(instance)
	if r2:getSelectedInstance() == instance then
		r2.ContextualCommands:update()
	end
end




-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------

-- The following code describes how to create a basic displayer that just
-- output the events it handles in the log
-- Mots of the time, when creating a new displayer, one will
-- just construct an existing displayer, and redefine methods of interest, 
-- possibly calling the parent one

function r2:exampleUIDisplayer()	
	local handler = {}
	local ct = colorTag(0, 0, 255)
	------------------------------------------------
	-- Called by C++ at creation
	function handler:onCreate(instance)
		debugInfo(ct .. "Instance " .. instance.InstanceId .."  was created")
	end
	------------------------------------------------
	-- Called by C++ just before object is removed (so properties are still readable)
	function handler:onErase(instance)	
		debugInfo(ct .. "Instance " .. instance.InstanceId .."  was erased")
	end
	------------------------------------------------
	-- Called by C++ just before object is moved in the object hierarchy
	function handler:onPreHrcMove(instance)
		updateContextToolbar(instance)
		debugInfo(ct .. "Instance " .. instance.InstanceId .."  is about to move")
	end
	------------------------------------------------
	-- Called by C++ just after object is move in the object hierarchy
	function handler:onPostHrcMove(instance)
		updateContextToolbar(instance)
		debugInfo(ct .. "Instance " .. instance.InstanceId .."  has moved")
	end
	------------------------------------------------
	-- Called by C++ just after object is highlighted by mouse
	function handler:onFocus(instance, hasFocus)
		if (instance.User.HasFocus ~= hasFocus) then
			if hasFocus == true then
				debugInfo(ct .. "Instance " .. instance.InstanceId .."  has gained focus")
			else
				debugInfo(ct .. "Instance " .. instance.InstanceId .."  has lost focus")
			end
			instance.User.HasFocus = hasFocus
		end		
	end
	------------------------------------------------
	-- Called by C++ just after object has been selected
	function handler:onSelect(instance, isSelected)
		if (isSelected == true) then
			debugInfo(ct .. "Instance " .. instance.InstanceId .."  is selected")
		else
			debugInfo(ct .. "Instance " .. instance.InstanceId .."  is unselected")
		end
	end
	------------------------------------------------
	-- Called by C++ when an attribute of this object has been modified
	-- An attribute inside this object has been modified
	-- attributeName  :Name of the attribute inside this object, as given by its class definition. If the attribute
    --                 is an array, then an additionnal parameter gives the index of the element being modified in the array (or -1 if the whole array is set)	
	function handler:onAttrModified(instance, attributeName, indexInArray)
		updateContextToolbar(instance)
		debugInfo(ct .. "Instance " .. instance.InstanceId .." has an attribute modified : " .. attributeName)
	end		
end


function r2:onInstanceSelectedInTree(id)
	-- is there's an active pick tool then
	local currTool = r2:getCurrentTool()
	if currTool and currTool:isPickTool() then
		local tree = getUICaller()		
		tree:cancelNextSelectLine() -- don't want real selection, actually ...
		if currTool:canPick() then
			currTool:pick()
		end		
		-- no-op else ...	
		return
	end
	--debugInfo("Seleting instance with id = " .. tostring(id) )
	r2:setSelectedInstanceId(id)
end

function r2:onInstanceRightClickInTree(id)	
	r2:setSelectedInstanceId(id)
	r2:displayContextMenu()
end


r2.VerboseEvents = false;

-- before to go to "test mode", store opened/closed nodes in scenario window tree
-- to correctly initialize tree when go back in edition mode
r2.storedClosedTreeNodes = {}

-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------

-- displayer that update the tree control (scenario window)
function r2:defaultUIDisplayer()
	local function eventDebugInfo(msg)
		if r2.VerboseEvents == true then
			debugInfo(msg)
		end
	end

	local handler = {}
	local ct = colorTag(255, 0, 255)
	------------------------------------------------
	-- helper function : notify current act ui displayer that its quota has been modified
	function handler:updateCurrentActQuota()
		-- defer update to the next frame (many element can be added at once)
		r2.UIMainLoop.LeftQuotaModified = true
	end
	------------------------------------------------
	function handler:onCut(instance, cutted)				
		-- NOT IMPLEMENTED
		-- debugInfo("On cut " .. tostring(cutted))
		-- local tree = getUI(r2.InstanceTreePath)
		-- debugInfo(tostring(select(cutted, 127, 255)))
		-- instance.User.TreeNode.Color.G = select(cutted, 0, 255)		
		-- tree:forceRebuild()
	end	
	------------------------------------------------
	function handler:onCreate(instance)		
		--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  was created")		
		self:addTreeNode(instance)		
		-- if my quota is not 0, then we should update
		-- the current act quota ..		
		--if instance:getUsedQuota() ~= 0 then
		--	self:updateCurrentActQuota()
		--end
		if instance:hasScenarioCost() ~= false then
			self:updateCurrentActQuota()
		end
		
	end
	------------------------------------------------
	function handler:onPostCreate(instance)

		-- Special : if the cookie 'AskName' is set (by C++ or lua), then show property and ask name
		-- to user for that object		    
		if instance.User.AskName then			
			if instance.User.ShowProps then
				r2:showProperties(instance)
				instance.User.ShowProps = nil
			end			
			if instance.User.Select then
				r2:setSelectedInstanceId(instance.InstanceId)
			end			
			local propWindow = r2.CurrentPropertyWindow

			-- tmp : quick & dirty access to the widget ...
			if propWindow and propWindow.active then
				local editBox = propWindow:find("Name"):find("eb")
				if editBox then
					setCaptureKeyboard(editBox)
					editBox:setSelectionAll()
				end
			end
			instance.User.AskName = nil -- get rid of cookie
		end	
		-- Special : if the cookie 'Select' is set (by C++ or lua), then the object should be selected after creation
		if instance.User.Select then			
			r2:setSelectedInstanceId(instance.InstanceId)				
		end
		if type(instance.User.CreateFunc) == "function" then
			instance.User.CreateFunc(instance)
		end
	end
	------------------------------------------------
	function handler:onErase(instance)						
		--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  was erased")
		self:removeTreeNode(instance)
		-- if my quota is not 0, then we should update
		-- the current act quota ..
		if instance:hasScenarioCost() ~= false then
			self:updateCurrentActQuota()
		end		
	end
	------------------------------------------------
	function handler:onPreHrcMove(instance)
		updateContextToolbar(instance)
		--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  is about to move")
		self:removeTreeNode(instance)
	end
	------------------------------------------------
	function handler:onPostHrcMove(instance)

		-- if parent is a group, for its creation you don't know category of children : people or creature
		-- you check it for first child
		local parent = instance.ParentInstance
		if instance:isGrouped() and parent.Components.Size==1 then
			self:onErase(parent)
			self:onCreate(parent)
			self:onPostCreate(parent)
		end

		updateContextToolbar(instance)
		--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  has moved")
		--eventDebugInfo(ct .. "New parent is " .. instance.ParentInstance.InstanceId)		
		local nodes = self:addTreeNode(instance)
		if (r2:getSelectedInstance() == instance) and nodes then
			for k, node in pairs(nodes) do
				assert(node)			
				assert(node:getParentTree())
				assert(node:getParentTree().selectNodeById)
				node:getParentTree():selectNodeById(node.Id, false)
			end
		end
		-- if my quota is not 0, then we should update
		-- the current act quota ..
		if instance:hasScenarioCost() ~= false then
			self:updateCurrentActQuota()
		end

		-- if instance has Components, its children's nodes have been deleted at onPreHrcMove call
		if instance.Components then	
			for i=0, instance.Components.Size-1 do
				local child = instance.Components[i]
				self:onCreate(child)
			end

			self:onPostCreate(instance)
		end
	end
	------------------------------------------------
	function handler:onFocus(instance, hasFocus)
		if (instance.User.HasFocus ~= hasFocus) then
			if hasFocus == true then
				--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  has gained focus")
			else
				--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  has lost focus")
			end
			instance.User.HasFocus = hasFocus
		end		
	end
	------------------------------------------------
	function handler:onSelect(instance, isSelected)		
		if not instance.User.TreeNodes then			
			return
		end	
		for k, treeNode in pairs(instance.User.TreeNodes) do

			if not (treeNode == nil or treeNode.isNil == true) then 		

				local tree = treeNode:getParentTree()	
				if (isSelected == true) then
					--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  is selected")			
					tree:selectNodeById(instance.InstanceId, false)
				else
					--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .."  is unselected")
					tree:unselect()
				end
			end
		end
	end
	------------------------------------------------
	function handler:onAttrModified(instance, attributeName, indexInArray)
		if attributeName == "Position" or attributeName == "Angle" then
			return
		end
		if attributeName == "Selectable" then
			self:removeTreeNode(instance)
			self:addTreeNode(instance)
		end

		updateContextToolbar(instance)
		if not instance.User.TreeNodes then			
			return
		end	
		local nodes = instance.User.TreeNodes
		for k, node in pairs(nodes) do
			local tree = node:getParentTree()		
			if attributeName == 'Name' then
				setupNodeName(instance)
				if node:getFather() then
					node:getFather():sortByBitmap()
				end
				tree:forceRebuild()			
				tree:selectNodeById(node.Id, false) -- reforce the selection
			end
		end

		if attr == "Ghost" then
			if instance.Ghost then
				self:removeTreeNode(instance)
			end
		end
		--eventDebugInfo(ct .. "Instance " .. instance.InstanceId .." has an attribute modified : " .. attributeName)
	end

	function setupNodeName(instance)				
		local treeNodes = instance.User.TreeNodes		
		if not treeNodes then return end

		for k, treeNode in pairs(treeNodes) do
			if not (treeNode == nil or treeNode.isNil == true) then 
				local tree = treeNode:getParentTree()				
						
				treeNode.Text = instance:getDisplayName()		
				if tree then -- nb : tree may be nil if node is setupped before being attached to its parent tree
					tree:forceRebuild()
				end
			end
		end
	end

	function handler:storeClosedTreeNodes()

		function downInTree(node, nodeTable)

			for i=0, node:getNumChildren()-1 do
				local child = node:getChild(i)
				assert(child)

				nodeTable[child.Id] = child.Opened

				if child:getNumChildren()>0 then
					downInTree(child, nodeTable)
				end
			end
		end

		r2.storedClosedTreeNodes = {}

		-- scenary objects
		r2.storedClosedTreeNodes[r2.Scenario:getBaseAct().InstanceId] = {}
		local objectNodes = r2.storedClosedTreeNodes[r2.Scenario:getBaseAct().InstanceId]

		local container = getUI("ui:interface:r2ed_scenario")
		--local objectsRoot = container:find("content_tree_list"):getRootNode():getNodeFromId("scenery_objects")
		local objectsRoot = container:find("content_tree_list"):getRootNode()
		assert(objectsRoot)
		downInTree(objectsRoot, objectNodes)

		-- entities and components
		if r2.Scenario.Acts.Size>1 then
			for i=1, r2.Scenario.Acts.Size-1 do
				local act = r2.Scenario.Acts[i]
				local peopleRoot = act:getContentTree():getRootNode():getNodeFromId("people")
				assert(peopleRoot)
				local creatureRoot = act:getContentTree():getRootNode():getNodeFromId("creatures")
				assert(creatureRoot)
				--local componentRoot = act:getMacroContentTree():getRootNode():getNodeFromId("macro_components")
				local componentRoot = act:getMacroContentTree():getRootNode()
				assert(componentRoot)

				r2.storedClosedTreeNodes[act.InstanceId] = {}
				local actNodes = r2.storedClosedTreeNodes[act.InstanceId]

				downInTree(peopleRoot, actNodes)
				downInTree(creatureRoot, actNodes)
				downInTree(componentRoot, actNodes)
			end
		end
	end

	function handler:addPermanentNodes()

		if r2.ScenarioInstanceId then
			local scenario = r2:getInstanceFromId(r2.ScenarioInstanceId)
			if scenario and scenario.Acts.Size>0 then
				local addToTreesTable = {}
				scenario:getBaseAct():appendInstancesByType(addToTreesTable, "LogicEntity")
				for k, instance in pairs(addToTreesTable) do
					self:addTreeNode(instance)
				end
			end
		end
	end

	-- private
	function handler:addTreeNode(instance)	

		if instance.Ghost then return nil end

		local parentNodes = instance:getParentTreeNode()

		if parentNodes==nil then return nil end

		if instance.User.TreeNodes==nil then instance.User.TreeNodes = {} end

		for actId,parentNode in pairs(parentNodes) do

			local alreadyAdded = false		

			for k2, treeNode in pairs(instance.User.TreeNodes) do
				if not (treeNode==nil or treeNode.isNil==true) then
					
					local father = treeNode:getFather()
					if father==parentNode then 
						alreadyAdded=true 
						break
					end
				end
			end

			if not alreadyAdded then
				
				if parentNode == nil then
					return nil -- one of the ancestors may be unselectable			
				end
				if not instance.SelectableFromRoot then
					return nil
				end
				local tree = parentNode:getParentTree()		
				local treeNode = SNode()
					
				-- store reference in object
				table.insert(instance.User.TreeNodes, treeNode)

				treeNode.Bitmap = instance:getPermanentStatutIcon()
				local openTree = true
				if r2.storedClosedTreeNodes[actId] then
					openTree = (r2.storedClosedTreeNodes[actId][instance.InstanceId]==true)
				end
				treeNode.Opened = openTree
				
				treeNode.Id = instance.InstanceId
				treeNode.AHName = "lua"
				local ahParams = "r2:onInstanceSelectedInTree('" .. instance.InstanceId .. "')"
				--eventDebugInfo(ahParams)
				treeNode.AHParams = ahParams 
				treeNode.AHNameRight = "lua"
				treeNode.AHParamsRight = "r2:onInstanceRightClickInTree('" .. instance.InstanceId .. "')"
				treeNode.AHNameClose = "lua"
				treeNode.AHParamsClose = "r2.storedClosedTreeNodes = {}"

				setupNodeName(instance)

				assert(parentNode)
				parentNode:addChildSortedByBitmap(treeNode)
				parentNode.Show = (parentNode:getNumChildren() ~= 0)
						
				tree:forceRebuild()
			end
		end

		return instance.User.TreeNodes						
	end
	function handler:removeTreeNode(instance)
			
		local nodes = instance.User.TreeNodes		
		if  nodes == nil or nodes.isNil then 			
			return
		end
		for k, node in pairs(nodes) do
			if not (node == nil or node.isNil == true) then
				local tree = node:getParentTree()
				if node:getFather().isNil then			
					if (node == node:getParentTree():getRootNode()) then
						--debugInfo("ROOT NODE")
						node:getParentTree():setRootNode(nil)
					else
						--debugInfo("ISOLATED NODE")
						deleteReflectable(node) -- isolated node (the tree was never built ?)
					end
				else
				 -- update parent node visibility only if a direct son of the root node
				 if node:getFather() then
					if (node:getFather():getFather() == tree:getRootNode()) then
						node:getFather().Show = (node:getFather():getNumChildren() > 1)
					end
					node:getFather():deleteChild(node)
				 end
					
				end
				tree:forceRebuild()
			end
		end
		instance.User.TreeNodes = nil
	end	

	return handler
	
end

-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------

-- special display for groups in scenario window
function r2:groupUIDisplayer()
	local handler = self:defaultUIDisplayer()	
   function handler:updateLeaderColor(instance)
	  if not instance.User.TreeNodes then 
		return
	  end
	  for k, node in pairs(instance.User.TreeNodes) do
		  local tree = node:getParentTree()
		  for i = 0, instance.Components.Size - 1 do
			  --debugInfo("I = " .. tostring(i))
			 local treeNodes = instance.Components[i].User.TreeNodes
			 if treeNodes then
				for k2, treeNode in pairs(treeNodes) do
					if i == 0 then
					   treeNode.Color = CRGBA(255, 0, 0) -- mark leader in red
					else
					   treeNode.Color = CRGBA(255, 255, 255)
					end
				end
			 end
		  end						
		  tree:forceRebuild()
	  end
   end
   --
   local oldOnAttrModified = handler.onAttrModified
	function handler:onAttrModified(instance, attrName, indexInArray)
		if attrName == "Components" then				
         self:updateLeaderColor(instance)
		end
		oldOnAttrModified(self, instance, attrName, indexInArray)
	end
   --
   -- local oldOnCreate = handler.onCreate
   -- function handler:onCreate(instance)      
   -- debugInfo("On create group")
   -- oldOnCreate(self, instance)      
   -- end
   --
   local oldOnPostCreate = handler.onPostCreate
   function handler:onPostCreate(instance)      
      oldOnPostCreate(self, instance)
      self:updateLeaderColor(instance)
   end
   --
	return handler
end



-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------

-- Displayer for ACTS. In the ui, acts are added into the act combo box --
-- in the environment of the container we store a table that gives the act Instance id,
-- and the index of the tree control for each line in the combo box
-- Table has the following look
-- ActTable = { { Act = ..., TreeIndex = ... }, -- combo box line 1
--              { Act = ..., TreeIndex = ... }, -- combo box line 2
--              { Act = ..., TreeIndex = ... }, -- combo box line 3 etc.
--            }


r2.ActUIDisplayer = {}
r2.ActUIDisplayer.ActTable = {} -- table that map each line of the combo box to an act
r2.ActUIDisplayer.LastSelfCreatedActInstanceId = nil -- id of the last act created by the pionner (not by another pionner)
                                                     -- When created, an act change will automatically occur

	------------------------------------------------
	-- helper function : notify current act ui displayer that its quota has been modified
	function r2.ActUIDisplayer:updateCurrentActQuota()
		-- defer update to the next frame (many element can be added at once)
		r2.UIMainLoop.LeftQuotaModified = true
	end

	------------------------------------------------
	function r2.ActUIDisplayer:updateActName(act)

		if act and not act:isBaseAct() then
				
			local actTable = self:getActTable()	
			for index, entry in pairs(actTable) do
				if entry.Act == act then
					local comboBox = self:getActComboBox() 

					local actTitle = act:getName()	
					if act==r2.Scenario:getCurrentAct() then
						actTitle = actTitle .. "  [" .. i18n.get("uiR2EDCurrentActComboBox"):toUtf8() .."]"	
					end
					local text = ucstring()
					text:fromUtf8(actTitle)
					comboBox:setText(index - 1, text)
					return
				end
			end 
		end
	end
	
	------------------------------------------------
	function r2.ActUIDisplayer:onAttrModified(instance, attributeName, indexInArray)

		-- if title is modified, then must update names of all entities in the scene
		if attributeName == "Name" then
			local npcs = {}
			r2:getCurrentAct():appendInstancesByType(npcs, "Npc")
			for k, npc in pairs(npcs) do
				npc.DisplayerVisual:updateName()
			end
			
			self:updateActName(instance)
		end
	end


		
	------------------------------------------------
	function r2.ActUIDisplayer:onCreate(act)
			
		local container = self:getContainer() 
		local comboBox = self:getActComboBox() 

		local tree, macroTree 
		if not act:isBaseAct() then	
			local text = ucstring()	
			local index = r2.logicComponents:searchElementIndex(act)-2
			local actTitle = act:getName()
			if type(actTitle) ~= "string" then
				text:fromUtf8("bad type for title : " .. type(actTitle))
				comboBox:insertText(index, text)
			else		
				text:fromUtf8(actTitle)
				comboBox:insertText(index, text)
			end

			tree = self:findFreeTreeCtrl()	
			macroTree = self:findFreeTreeCtrl(true)	
			local actTable = self:getActTable()
			table.insert(actTable, index+1, { Act = act, Tree = tree , MacroTree = macroTree})		
		end

		-- store tree in the act for future insertion of items
		act.User.ContentTree = tree
		act.User.MacroContentTree = macroTree
		self:updateCurrentActQuota()

		-- add permanent nodes to act node
		r2:defaultUIDisplayer():addPermanentNodes()
	end

	------------------------------------------------
	function r2.ActUIDisplayer:onPostCreate(act)
		-- when a new act is created, select this act as the default
		if act.InstanceId == self.LastSelfCreatedActInstanceId then
			-- the act was just created by pionner on that computer, so change right now		
			r2.ScenarioWindow:setAct(act)
			self.LastSelfCreatedActInstanceId = nil
		end
		r2.ScenarioWindow:updateUIFromCurrentAct()
		self:updateCurrentActQuota()
	end

	------------------------------------------------
	function r2.ActUIDisplayer:onErase(erasedAct)
		-- clean tree content
		local tree = erasedAct.User.ContentTree
		local macroTree = erasedAct.User.MacroContentTree
		if tree then
			r2:cleanTreeNode(tree, "people")
			r2:cleanTreeNode(tree, "creatures")
		end
		if macroTree then
			--r2:cleanTreeNode(macroTree, "macro_components")
			r2:cleanTreeRootNode(macroTree)
		end
		local actTable = self:getActTable()	
		for index, entry in pairs(actTable) do
			if entry.Act == erasedAct then
				self:getActComboBox():removeTextByIndex(index - 1)
				table.remove(actTable, index)
				return
			end
		end 
	
      self:updateCurrentActQuota()
	end
	------------------------------------------------
	function r2.ActUIDisplayer:getActTable()	
		return self.ActTable
	end

	------------------------------------------------
	function r2.ActUIDisplayer:getContainer()
		return getUI("ui:interface:r2ed_scenario")
	end

	------------------------------------------------
	function r2.ActUIDisplayer:getActComboBox()
		return self:getContainer():find("act_combo_box")
	end

	-----------------------------------------------	
	function r2.ActUIDisplayer:findFreeTreeCtrl(macroTree)	
	
		local treeName = "act_tree_"
		if macroTree==true then treeName="macro_act_tree_" end	
		for i = 0, r2:getMaxNumberOfAdditionnalActs() - 1 do
			local tree = self:getContainer():find(treeName .. tostring(i))
			local used = false
			for index, entry in pairs(self:getActTable()) do
				local entryTree = entry.Tree
				if macroTree==true then entryTree = entry.MacroTree end	
				if entryTree == tree then
					used = true
					break
				end
			end
			if not used then
				return tree
			end
		end
		return nil
	end


function r2:createActUIDisplayer()
	return r2.ActUIDisplayer		
end