lab/lmapper/bake.py

#!/usr/bin/env python
import subprocess
import os
import glob
import sys

root = os.path.dirname(sys.argv[0])
platform = sys.platform

SHADERS = 'shaders'
TEXTURES = 'textures'
RES = 4096

os.environ['DL_SHADERS_PATH'] = os.path.join(root, SHADERS)
os.environ['DL_TEXTURES_PATH'] = os.path.join(root, TEXTURES)

if platform == 'linux2':
    print 'running linux'
    DELIGHT = '/usr/local/3delight-12.0.19/Linux-x86_64'
    os.environ['DELIGHT'] = DELIGHT
    sys.path.append(os.path.join(DELIGHT, 'bin'))
elif platform == 'darwin':
    os.environ['DELIGHT'] = '/Applications/3Delight'


def cleanup():
    for bakefile in glob.glob(os.path.join(root,'*.bake')):
        os.remove(bakefile)
    for tdl in glob.glob(os.path.join(root,'*.tdl')):
        os.remove(tdl)


def gen_includes():
    includes = []
    for file in glob.glob('caster/*'):
        includes.append(file)
    for file in glob.glob('receiver/*'):
        includes.append(file)
    return includes

def bake_post_render(root, res=RES):
    for bakefile in glob.glob(os.path.join(root, '*.bake')):
         bake_target = os.path.splitext(bakefile)[0]
         cmd = ['tdlmake', '-bakeres', str(res) + 'x' + str(res), bakefile, bake_target + '.tdl']
         subprocess.call(cmd)
         cmd = ['tdlmake', bake_target + '.tdl', bake_target + '.tif',]
         subprocess.call(cmd)


print gen_includes()

cleanup()

#subprocess.call(['renderdl', '-res', str(RES), str(RES), '-id', '-progress', 'options.rib'])
subprocess.call(['renderdl', '-res', str(RES), str(RES), '-id','-progress', 'options.rib'])

bake_post_render(root)