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
|
--!A cross-platform build utility based on Lua
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
-- http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
-- Copyright (C) 2015-present, TBOOX Open Source Group.
--
-- @author ruki
-- @file unity_build.lua
--
-- imports
import("core.project.depend")
function _merge_unityfile(target, sourcefile_unity, sourcefiles, opt)
local dependfile = target:dependfile(sourcefile_unity)
depend.on_changed(function ()
-- trace
vprint("generating.unityfile %s", sourcefile_unity)
-- do merge
local unityfile = io.open(sourcefile_unity, "w")
for _, sourcefile in ipairs(sourcefiles) do
sourcefile = path.absolute(sourcefile)
sourcefile_unity = path.absolute(sourcefile_unity)
sourcefile = path.relative(sourcefile, path.directory(sourcefile_unity))
unityfile:print("#include \"%s\"", sourcefile)
end
unityfile:close()
end, {dependfile = dependfile, files = sourcefiles})
end
function generate_unityfiles(target, sourcebatch, opt)
local unity_batch = target:data("unity_build.unity_batch." .. sourcebatch.rulename)
if unity_batch then
for _, sourcefile_unity in ipairs(sourcebatch.sourcefiles) do
local sourceinfo = unity_batch[sourcefile_unity]
if sourceinfo then
local sourcefiles = sourceinfo.sourcefiles
if sourcefiles then
_merge_unityfile(target, sourcefile_unity, sourcefiles, opt)
end
end
end
end
end
-- use unity build
--
-- e.g.
-- add_rules("c++.unity_build", {batchsize = 2})
-- add_files("src/*.c", "src/*.cpp", {unity_ignored = true})
-- add_files("src/foo/*.c", {unity_group = "foo"})
-- add_files("src/bar/*.c", {unity_group = "bar"})
--
function main(target, sourcebatch)
-- get unit batch sources
local extraconf = target:extraconf("rules", "c++.unity_build")
local batchsize = extraconf and extraconf.batchsize
local id = 1
local count = 0
local unity_batch = {}
local sourcefiles = {}
local objectfiles = {}
local dependfiles = {}
local sourcedir = path.join(target:autogendir({root = true}), "unity_build")
for idx, sourcefile in pairs(sourcebatch.sourcefiles) do
local sourcefile_unity
local objectfile = sourcebatch.objectfiles[idx]
local dependfile = sourcebatch.dependfiles[idx]
local fileconfig = target:fileconfig(sourcefile)
if fileconfig and fileconfig.unity_group then
sourcefile_unity = path.join(sourcedir, "unity_" .. fileconfig.unity_group .. path.extension(sourcefile))
elseif (fileconfig and fileconfig.unity_ignored) or (batchsize == 0) then
-- we do not add these files to unity file
table.insert(sourcefiles, sourcefile)
table.insert(objectfiles, objectfile)
table.insert(dependfiles, dependfile)
else
if batchsize and count > batchsize then
id = id + 1
end
sourcefile_unity = path.join(sourcedir, "unity_" .. hash.uuid(tostring(id)):split("-", {plain = true})[1] .. path.extension(sourcefile))
count = count + 1
end
if sourcefile_unity then
local sourceinfo = unity_batch[sourcefile_unity]
if not sourceinfo then
sourceinfo = {}
sourceinfo.objectfile = target:objectfile(sourcefile_unity)
sourceinfo.dependfile = target:dependfile(sourceinfo.objectfile)
unity_batch[sourcefile_unity] = sourceinfo
end
sourceinfo.sourcefiles = sourceinfo.sourcefiles or {}
table.insert(sourceinfo.sourcefiles, sourcefile)
end
end
-- use unit batch
for sourcefile_unity, sourceinfo in pairs(unity_batch) do
table.insert(sourcefiles, sourcefile_unity)
table.insert(objectfiles, sourceinfo.objectfile)
table.insert(dependfiles, sourceinfo.dependfile)
end
sourcebatch.sourcefiles = sourcefiles
sourcebatch.objectfiles = objectfiles
sourcebatch.dependfiles = dependfiles
-- save unit batch
target:data_set("unity_build.unity_batch." .. sourcebatch.rulename, unity_batch)
end
|