Spring RTS Engine

Code: Select all

	local exceptionMap  = {}
38	for _, unitName in pairs(exceptionList) do
39	  exceptionMap[UnitDefNames[unitName].id] = true
40	end

Code: Select all

--------------------------------------------------------------------------------
	--------------------------------------------------------------------------------
	
	function gadget:GetInfo()
	  return {
	    name      = "No Self Pwn",
	    desc      = "Prevents units from damaging themselves.",
	    author    = "quantum, edited by Pxtl",
	    date      = "May 26, 2008",
	    license   = "GNU GPL, v2 or later",
	    layer     = 0,
	    enabled   = true  --  loaded by default?
	  }
	end
	
	--------------------------------------------------------------------------------
	--------------------------------------------------------------------------------
	
	local Spring = Spring
	
	local GetUnitHealth = Spring.GetUnitHealth
	local SetUnitHealth = Spring.SetUnitHealth
	  
	if (not gadgetHandler:IsSyncedCode()) then
	  return false  --  silent removal
	end
	  
	local inclusionMap  = {}
	for unitName, ud in pairs(UnitDefs) do
	  if(ud["customParams"]["noselfdamage"] == "1") then
		inclusionMap[UnitDefNames[unitName].id] = true
	  end
	end
	
	
	--------------------------------------------------------------------------------
	--------------------------------------------------------------------------------
	
	function gadget:UnitDamaged(unitID, unitDefID, unitTeam, damage, paralyzer, 
	                            weaponID, attackerID, attackerDefID, attackerTeam)
	  if (unitID == attackerID and inclusionMap[unitDefID]) then
	    local health, _, paralyzeDamage = GetUnitHealth(unitID)
	    if (paralyzer) then
	      SetUnitHealth(unitID, {paralyze = paralyzeDamage + damage})
	    else
	      SetUnitHealth(unitID, health + damage)
	    end
	  end
	end
	
	--------------------------------------------------------------------------------
	--------------------------------------------------------------------------------