mirror of https://gitlab.com/ita1024/waf.git
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
#! /usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
"""
|
|
Take a group of C++ files and compile them at once.
|
|
|
|
def options(opt):
|
|
opt.load('compiler_cxx unity')
|
|
"""
|
|
|
|
import re
|
|
from waflib import Task, Options, Logs
|
|
from waflib.Tools import ccroot, cxx, c_preproc
|
|
from waflib.TaskGen import extension, taskgen_method
|
|
|
|
MAX_BATCH = 20
|
|
|
|
def options(opt):
|
|
global MAX_BATCH
|
|
opt.add_option('--batchsize', action='store', dest='batchsize', type='int', default=MAX_BATCH, help='batch size (0 for no batch)')
|
|
|
|
class unity(Task.Task):
|
|
color = 'BLUE'
|
|
scan = c_preproc.scan
|
|
def run(self):
|
|
lst = ['#include "%s"\n' % node.abspath() for node in self.inputs]
|
|
txt = ''.join(lst)
|
|
self.outputs[0].write(txt)
|
|
|
|
@taskgen_method
|
|
def batch_size(self):
|
|
return Options.options.batchsize
|
|
|
|
@extension('.cpp', '.cc', '.cxx', '.C', '.c++')
|
|
def make_cxx_batch(self, node):
|
|
cnt = self.batch_size()
|
|
if cnt <= 1:
|
|
tsk = self.create_compiled_task('cxx', node)
|
|
return tsk
|
|
|
|
try:
|
|
self.cnt_cxx
|
|
except AttributeError:
|
|
self.cnt_cxx = 0
|
|
|
|
x = getattr(self, 'master_cxx', None)
|
|
if not x or len(x.inputs) >= cnt:
|
|
x = self.master_cxx = self.create_task('unity')
|
|
cxxnode = node.parent.find_or_declare('union_%s_%d_%d.cxx' % (self.idx, self.cnt_cxx, cnt))
|
|
self.master_cxx.outputs = [cxxnode]
|
|
self.cnt_cxx += 1
|
|
self.create_compiled_task('cxx', cxxnode)
|
|
x.inputs.append(node)
|
|
|