Spring RTS Engine

Code: Select all

CMD_WEAPON = 33319
desc = {
	name="Change Weapon", --button label
	type=CMDTYPE.ICON, --a button that activates when clicked on
	id=CMD_WEAPON,
	action="changeweapon"
}

--For new units
function gadget:UnitCreated(u,ud,defs)
	if UnitDefs[ud].customParams.canchangeweapon == "1" then
		Spring.InsertCmdDesc(u, desc) --add the command to the unit's list
	end
end

--Blocks commands that are invalid
function gadget:AllowCommand(u,ud,team,cmd,param,opt)
	if cmd == CMD_WEAPON then
		return UnitDefs[ud].customParams.canchangeweapon == "1" --makes sure CMD_WEAPPON can only be given to units that can change their weapon
	end
	return true
end

--Handles custom commands
function gadget:CommandFallback(u,ud,team,cmd,param,opt)
	if cmd == CMD_WEAPON then
		Spring.CallCOBScript(u, "ChangeWeapon", 0) --calls this COB function
		return true, true
	end
	return false
end

Code: Select all

[customParams] {
    canChangeWeapon=1;
}