File: share_levels.lua

package info (click to toggle)
spring 0.81.2.1%2Bdfsg1-6
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 28,496 kB
  • ctags: 37,096
  • sloc: cpp: 238,659; ansic: 13,784; java: 12,175; awk: 3,428; python: 1,159; xml: 738; perl: 405; sh: 297; makefile: 267; pascal: 228; objc: 192
file content (231 lines) | stat: -rw-r--r-- 5,526 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
--  file:    share_levels.lua
--  brief:   control resource sharing
--  author:  Dave Rodgers
--
--  Copyright (C) 2007.
--  Licensed under the terms of the GNU GPL, v2 or later.
--
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------

function gadget:GetInfo()
  return {
    name      = "ShareLevels",
    desc      = "Control resource sharing",
    author    = "trepan",
    date      = "Apr 22, 2007",
    license   = "GNU GPL, v2 or later",
    layer     = -5,
    enabled   = true  --  loaded by default?
  }
end

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

local metalForce  = false
local metalLevel  = 0.5

local energyForce = false
local energyLevel = 0.5

-- for toggling
local prevMetal = false
local prevEnery = false

local cmdName = "sharelevels"
local help    = " <on|off|list|toggle|<<e|m|*> [<on|off|%>]>]>"
                .. ":  control automatic team resource sharing"


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

local function AllowAction(playerID)
  if (playerID ~= 0) then
    Spring.SendMessageToPlayer(playerID, "Must be the host player")
    return false
  end
  if (not Spring.IsCheatingEnabled()) then
    Spring.SendMessageToPlayer(playerID, "Cheating must be enabled")
    return false
  end
  return true
end



local function PrintHelp()
  Spring.Echo(help)
  return true
end


local function PrintState()
  if (not metalForce) then
    Spring.Echo('forced metal sharing is disabled')
  else
    Spring.Echo(string.format(
      'forced metal sharing level set to: %i%%',
      ((1 - metalLevel) * 100)
    ))
  end
  if (not energyForce) then
    Spring.Echo('forced energy sharing is disabled')
  else
    Spring.Echo(string.format(
      'forced energy sharing level set to: %i%%',
      ((1 - energyLevel) * 100)
    ))
  end
  return true
end


local function ToggleBoth()
  local hadPrev = (prevMetal or prevEnergy)
  if (metalForce or energyForce) then
    prevMetal  = metalForce
    prevEnergy = energyForce
    metalForce  = false
    energyForce = false
  elseif (hadPrev) then
    metalForce  = prevMetal
    energyForce = prevEnergy
  else
    metalForce  = true
    energyForce = true
  end
end


local function ChatControl(cmd, line, words, playerID)
  if (not AllowAction(playerID)) then
    PrintState()
    return true
  end
  
  local w1, w2 = words[1], words[2]
  
  if ((w1 == 'list') or (w1 == 'l')) then
    return PrintState()
  elseif ((w1 == 'toggle') or (w1 == 't')) then
    ToggleBoth()
    return PrintState()
  elseif (w1 == 'on') then
    metalForce  = true
    energyForce = true
    return PrintState()
  elseif (w1 == 'off') then
    metalForce  = false
    energyForce = false
    return PrintState()
  end

  if (not (w1 == 'e') and
      not (w1 == 'm') and
      not (w1 == '*')) then
    return PrintHelp()
  end

  if (w2 == nil) then
    if (w1 == 'm') then
      metalForce = not metalForce
      return PrintState()
    elseif (w1 == 'e') then
      energyForce = not energyForce
      return PrintState()
    else
      ToggleBoth()
      return PrintState()
    end
  end

  if ((w2 == 'on') or (w2 == 'off')) then
    local state = (w2 == 'on')
    if (w1 == 'm') then
      metalForce = state
      return PrintState()
    elseif (w1 == 'e') then
      energyForce = state
      return PrintState()
    else
      metalForce  = state
      energyForce = state
      return PrintState()
    end
    return PrintHelp()
  end

  local level = tonumber(w2)
  if (level == nil) then
    PrintHelp()
    Spring.Echo("bad sharing level: " .. w2)
    return true
  end
  level = (100 - level) * 0.01
  if (level < 0) then level = 0 end
  if (level > 1) then level = 1 end

  if ((w1 == 'm') or (w1 == '*')) then
    metalLevel = level
    metalForce = true
  end
  if ((w1 == 'e') or (w1 == '*')) then
    energyLevel = level
    energyForce = true
  end

  return PrintState()
end


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

function gadget:Initialize()
  if (not gadgetHandler:IsSyncedCode()) then
    gadgetHandler:RemoveGadget()
    return
  end
  gadgetHandler:AddChatAction(cmdName, ChatControl, help)
  Script.AddActionFallback(cmdName .. ' ', help)
end


function gadget:Shutdown()
  gadgetHandler:RemoveChatAction(cmdName, ChatControl)
end


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

local GetTeamList       = Spring.GetTeamList
local SetTeamShareLevel = Spring.SetTeamShareLevel


function gadget:GameFrame()
  if (not (metalForce or energyForce)) then
    return
  end
  for _,team in ipairs(GetTeamList()) do
    if (metalForce)  then SetTeamShareLevel(team, 'm', metalLevel)  end
    if (energyForce) then SetTeamShareLevel(team, 'e', energyLevel) end
  end
  return false
end


function gadget:AllowResourceLevel(teamID, type, level)
  if (type == 'm') then
    return not metalForce
  elseif (type == 'e') then
    return not energyForce
  end
  return true
end


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