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
|
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- file: unitdefs.lua
-- brief: unitdef parser
-- author: Dave Rodgers
--
-- Copyright (C) 2007.
-- Licensed under the terms of the GNU GPL, v2 or later.
--
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
local unitDefs = {}
local shared = {} -- shared amongst the lua unitdef enviroments
local preProcFile = 'gamedata/unitdefs_pre.lua'
local postProcFile = 'gamedata/unitdefs_post.lua'
local FBI = FBIparser or VFS.Include('gamedata/parse_fbi.lua')
local TDF = TDFparser or VFS.Include('gamedata/parse_tdf.lua')
local DownloadBuilds = VFS.Include('gamedata/download_builds.lua')
local system = VFS.Include('gamedata/system.lua')
VFS.Include('gamedata/VFSUtils.lua')
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- Run a pre-processing script if one exists
--
if (VFS.FileExists(preProcFile)) then
Shared = shared -- make it global
UnitDefs = unitDefs -- make it global
VFS.Include(preProcFile)
UnitDefs = nil
Shared = nil
end
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- Load the FBI unitdef files
--
local fbiFiles = RecursiveFileSearch('units/', '*.fbi')
for _, filename in ipairs(fbiFiles) do
local ud, err = FBI.Parse(filename)
if (ud == nil) then
Spring.Echo('Error parsing ' .. filename .. ': ' .. err)
elseif (ud.unitname == nil) then
Spring.Echo('Missing unitName in ' .. filename)
else
ud.filename = filename
ud.unitname = string.lower(ud.unitname)
unitDefs[ud.unitname] = ud
end
end
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- Load the raw LUA format unitdef files
-- (these will override the FBI/SWU versions)
--
local luaFiles = RecursiveFileSearch('units/', '*.lua')
for _, filename in ipairs(luaFiles) do
local udEnv = {}
udEnv._G = udEnv
udEnv.Shared = shared
udEnv.GetFilename = function() return filename end
setmetatable(udEnv, { __index = system })
local success, uds = pcall(VFS.Include, filename, udEnv)
if (not success) then
Spring.Echo('Error parsing ' .. filename .. ': ' .. uds)
elseif (type(uds) ~= 'table') then
Spring.Echo('Bad return table from: ' .. filename)
else
for udName, ud in pairs(uds) do
if ((type(udName) == 'string') and (type(ud) == 'table')) then
ud.filename = filename
unitDefs[udName] = ud
else
Spring.Echo('Bad return table entry from: ' .. filename)
end
end
end
end
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- Insert the download build entries
--
DownloadBuilds.Execute(unitDefs)
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- Run a post-processing script if one exists
--
if (VFS.FileExists(postProcFile)) then
Shared = shared -- make it global
UnitDefs = unitDefs -- make it global
VFS.Include(postProcFile)
UnitDefs = nil
Shared = nil
end
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
--
-- Basic checks to kill unitDefs that will crash ".give all"
--
for name, def in pairs(unitDefs) do
local cob = 'scripts/' .. name .. '.cob'
local obj = def.objectName or def.objectname
if (obj == nil) then
unitDefs[name] = nil
Spring.Echo('WARNING: removed ' .. name ..
' unitDef, missing objectname param')
for k,v in pairs(def) do print('',k,v) end
else
local objfile = 'objects3d/' .. obj
if ((not VFS.FileExists(objfile)) and
(not VFS.FileExists(objfile .. '.3do')) and
(not VFS.FileExists(objfile .. '.s3o'))) then
unitDefs[name] = nil
Spring.Echo('WARNING: removed ' .. name
.. ' unitDef, missing model file (' .. obj .. ')')
end
end
end
for name, def in pairs(unitDefs) do
local badOptions = {}
local buildOptions = def.buildOptions or def.buildoptions
if (buildOptions) then
for i, option in ipairs(buildOptions) do
if (unitDefs[option] == nil) then
table.insert(badOptions, i)
Spring.Echo('WARNING: removed the "' .. option ..'" entry'
.. ' from the "' .. name .. '" build menu')
end
end
if (#badOptions > 0) then
local removed = 0
for _, badIndex in ipairs(badOptions) do
table.remove(buildOptions, badIndex - removed)
removed = removed + 1
end
end
end
end
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
return unitDefs
--------------------------------------------------------------------------------
--------------------------------------------------------------------------------
|
