Sprout/tools/darkroom/darkcult.py

65 lines
2.1 KiB
Python
Executable file

#!/usr/bin/env python
# =============================================================================
# Copyright (c) 2011-2013 Bolero MURAKAMI
# https://github.com/bolero-MURAKAMI/Sprout
#
# Distributed under the Boost Software License, Version 1.0. (See accompanying
# file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
# =============================================================================
import sys
import optparse
import subprocess
import multiprocessing
def compile(command):
sys.stdout.write(".")
sys.stdout.flush()
return subprocess.call(command, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
def main():
parser = optparse.OptionParser(description='darkcult.py')
parser.add_option('--src', type='string')
parser.add_option('--stagedir', type='string')
parser.add_option('--output', type='string')
parser.add_option('--compiler', type='string')
parser.add_option('--width', type='int')
parser.add_option('--height', type='int')
parser.add_option('--tile_width', type='int')
parser.add_option('--tile_height', type='int')
parser.add_option('--compile_options', type='string')
parser.add_option('--darkcult_cpp', type='string')
parser.add_option('--max_procs', type='int')
(opts, args) = parser.parse_args()
pool = multiprocessing.Pool(opts.max_procs if opts.max_procs != 0 else None)
def format_command(x, y):
bin = "%s/%d/%d.out" % (opts.stagedir, y, x)
out = "%s/%d/%d.ppm" % (opts.stagedir, y, x)
return "%s -o %s" \
" %s" \
" -DDARKROOM_SOURCE=\'\"%s\"\'" \
" -DDARKROOM_TOTAL_WIDTH=%d -DDARKROOM_TOTAL_HEIGHT=%d" \
" -DDARKROOM_TILE_WIDTH=%d -DDARKROOM_TILE_HEIGHT=%d" \
" -DDARKROOM_OFFSET_X=%d -DDARKROOM_OFFSET_Y=%d" \
" %s" \
" && %s > %s" \
% (opts.compiler, bin,
opts.compile_options,
opts.src,
opts.width, opts.height,
opts.tile_width, opts.tile_height,
x, y,
opts.darkcult_cpp,
bin, out
)
return any(pool.map(
compile,
[format_command(x, y)
for y in range(0, opts.height, opts.tile_height)
for x in range(0, opts.width, opts.tile_width)
]
))
if __name__=="__main__":
sys.exit(main())