671 lines
30 KiB
Python
Executable file
671 lines
30 KiB
Python
Executable file
import bpy
|
|
import os
|
|
import shutil
|
|
import glob
|
|
import json
|
|
import stat
|
|
import arm.utils
|
|
import arm.assets as assets
|
|
import arm.make_state as state
|
|
|
|
def add_armory_library(sdk_path, name, rel_path=False):
|
|
if rel_path:
|
|
sdk_path = '../' + os.path.relpath(sdk_path, arm.utils.get_fp()).replace('\\', '/')
|
|
return ('project.addLibrary("' + sdk_path + '/' + name + '");\n').replace('\\', '/').replace('//', '/')
|
|
|
|
def add_assets(path, quality=1.0, use_data_dir=False, rel_path=False):
|
|
if not bpy.data.worlds['Arm'].arm_minimize and path.endswith('.arm'):
|
|
path = path[:-4] + '.json'
|
|
|
|
if rel_path:
|
|
path = os.path.relpath(path, arm.utils.get_fp()).replace('\\', '/')
|
|
|
|
notinlist = not path.endswith('.ttf') # TODO
|
|
s = 'project.addAssets("' + path + '", { notinlist: ' + str(notinlist).lower() + ' '
|
|
if quality < 1.0:
|
|
s += ', quality: ' + str(quality)
|
|
if use_data_dir:
|
|
s += ', destination: "data/{name}"'
|
|
s += '});\n'
|
|
return s
|
|
|
|
def add_shaders(path, rel_path=False):
|
|
if rel_path:
|
|
path = os.path.relpath(path, arm.utils.get_fp())
|
|
return 'project.addShaders("' + path.replace('\\', '/').replace('//', '/') + '");\n'
|
|
|
|
def remove_readonly(func, path, excinfo):
|
|
os.chmod(path, stat.S_IWRITE)
|
|
func(path)
|
|
|
|
def write_khafilejs(is_play, export_physics, export_navigation, export_ui, is_publish, enable_dce, import_traits, import_logicnodes):
|
|
sdk_path = arm.utils.get_sdk_path()
|
|
rel_path = arm.utils.get_relative_paths() # Convert absolute paths to relative
|
|
wrd = bpy.data.worlds['Arm']
|
|
|
|
with open('khafile.js', 'w') as khafile:
|
|
khafile.write(
|
|
"""// Auto-generated
|
|
let project = new Project('""" + arm.utils.safestr(wrd.arm_project_name) + """');
|
|
|
|
project.addSources('Sources');
|
|
""")
|
|
|
|
# Let libraries differentiate between Armory and pure Kha
|
|
assets.add_khafile_def('armory')
|
|
|
|
# Auto-add assets located in Bundled directory
|
|
if os.path.exists('Bundled'):
|
|
for file in glob.glob("Bundled/**", recursive=True):
|
|
if os.path.isfile(file):
|
|
assets.add(file)
|
|
|
|
if os.path.exists('Shaders'):
|
|
# Copy to enable includes
|
|
if os.path.exists(arm.utils.build_dir() + '/compiled/Shaders/Project'):
|
|
shutil.rmtree(arm.utils.build_dir() + '/compiled/Shaders/Project', onerror=remove_readonly)
|
|
shutil.copytree('Shaders', arm.utils.build_dir() + '/compiled/Shaders/Project')
|
|
khafile.write("project.addShaders('" + arm.utils.build_dir() + "/compiled/Shaders/Project/**');\n")
|
|
# for file in glob.glob("Shaders/**", recursive=True):
|
|
# if os.path.isfile(file):
|
|
# assets.add_shader(file)
|
|
|
|
if not os.path.exists('Libraries/armory'):
|
|
khafile.write(add_armory_library(sdk_path, 'armory', rel_path=rel_path))
|
|
|
|
if not os.path.exists('Libraries/iron'):
|
|
khafile.write(add_armory_library(sdk_path, 'iron', rel_path=rel_path))
|
|
|
|
# Project libraries
|
|
if os.path.exists('Libraries'):
|
|
libs = os.listdir('Libraries')
|
|
for lib in libs:
|
|
if os.path.isdir('Libraries/' + lib):
|
|
khafile.write('project.addLibrary("{0}");\n'.format(lib.replace('//', '/')))
|
|
|
|
# Subprojects, merge this with libraries
|
|
if os.path.exists('Subprojects'):
|
|
libs = os.listdir('Subprojects')
|
|
for lib in libs:
|
|
if os.path.isdir('Subprojects/' + lib):
|
|
khafile.write('await project.addProject("Subprojects/{0}");\n'.format(lib))
|
|
|
|
if wrd.arm_audio == 'Disabled':
|
|
assets.add_khafile_def('kha_no_ogg')
|
|
else:
|
|
assets.add_khafile_def('arm_audio')
|
|
|
|
if export_physics:
|
|
assets.add_khafile_def('arm_physics')
|
|
if wrd.arm_physics_engine == 'Bullet':
|
|
assets.add_khafile_def('arm_bullet')
|
|
if not os.path.exists('Libraries/haxebullet'):
|
|
khafile.write(add_armory_library(sdk_path + '/lib/', 'haxebullet', rel_path=rel_path))
|
|
if state.target.startswith('krom'):
|
|
ammojs_path = sdk_path + '/lib/haxebullet/ammo/ammo.wasm.js'
|
|
ammojs_path = ammojs_path.replace('\\', '/').replace('//', '/')
|
|
khafile.write(add_assets(ammojs_path, rel_path=rel_path))
|
|
ammojs_wasm_path = sdk_path + '/lib/haxebullet/ammo/ammo.wasm.wasm'
|
|
ammojs_wasm_path = ammojs_wasm_path.replace('\\', '/').replace('//', '/')
|
|
khafile.write(add_assets(ammojs_wasm_path, rel_path=rel_path))
|
|
elif state.target == 'html5' or state.target == 'node':
|
|
ammojs_path = sdk_path + '/lib/haxebullet/ammo/ammo.js'
|
|
ammojs_path = ammojs_path.replace('\\', '/').replace('//', '/')
|
|
khafile.write(add_assets(ammojs_path, rel_path=rel_path))
|
|
elif wrd.arm_physics_engine == 'Oimo':
|
|
assets.add_khafile_def('arm_oimo')
|
|
if not os.path.exists('Libraries/oimo'):
|
|
khafile.write(add_armory_library(sdk_path + '/lib/', 'oimo', rel_path=rel_path))
|
|
|
|
if export_navigation:
|
|
assets.add_khafile_def('arm_navigation')
|
|
if not os.path.exists('Libraries/haxerecast'):
|
|
khafile.write(add_armory_library(sdk_path + '/lib/', 'haxerecast', rel_path=rel_path))
|
|
if state.target.startswith('krom') or state.target == 'html5':
|
|
recastjs_path = sdk_path + '/lib/haxerecast/js/recast/recast.js'
|
|
recastjs_path = recastjs_path.replace('\\', '/').replace('//', '/')
|
|
khafile.write(add_assets(recastjs_path, rel_path=rel_path))
|
|
|
|
if is_publish:
|
|
assets.add_khafile_def('arm_published')
|
|
if wrd.arm_asset_compression:
|
|
assets.add_khafile_def('arm_compress')
|
|
else:
|
|
pass
|
|
# khafile.write("""project.addParameter("--macro include('armory.trait')");\n""")
|
|
# khafile.write("""project.addParameter("--macro include('armory.trait.internal')");\n""")
|
|
# if export_physics:
|
|
# khafile.write("""project.addParameter("--macro include('armory.trait.physics')");\n""")
|
|
# if wrd.arm_physics_engine == 'Bullet':
|
|
# khafile.write("""project.addParameter("--macro include('armory.trait.physics.bullet')");\n""")
|
|
# else:
|
|
# khafile.write("""project.addParameter("--macro include('armory.trait.physics.oimo')");\n""")
|
|
# if export_navigation:
|
|
# khafile.write("""project.addParameter("--macro include('armory.trait.navigation')");\n""")
|
|
|
|
# if import_logicnodes: # Live patching for logic nodes
|
|
# khafile.write("""project.addParameter("--macro include('armory.logicnode')");\n""")
|
|
|
|
if not wrd.arm_compiler_inline:
|
|
khafile.write("project.addParameter('--no-inline');\n")
|
|
|
|
if enable_dce:
|
|
khafile.write("project.addParameter('-dce full');\n")
|
|
|
|
live_patch = wrd.arm_live_patch and state.target == 'krom'
|
|
if wrd.arm_debug_console or live_patch:
|
|
import_traits.append('armory.trait.internal.Bridge')
|
|
if live_patch:
|
|
assets.add_khafile_def('arm_patch')
|
|
|
|
import_traits = list(set(import_traits))
|
|
for i in range(0, len(import_traits)):
|
|
khafile.write("project.addParameter('" + import_traits[i] + "');\n")
|
|
khafile.write("""project.addParameter("--macro keep('""" + import_traits[i] + """')");\n""")
|
|
|
|
noembed = wrd.arm_cache_build and not is_publish and state.target == 'krom'
|
|
if noembed:
|
|
# Load shaders manually
|
|
assets.add_khafile_def('arm_noembed')
|
|
|
|
noembed = False # TODO: always embed shaders for now, check compatibility with haxe compile server
|
|
|
|
shaders_path = arm.utils.build_dir() + '/compiled/Shaders/*.glsl'
|
|
if rel_path:
|
|
shaders_path = os.path.relpath(shaders_path, arm.utils.get_fp()).replace('\\', '/')
|
|
khafile.write('project.addShaders("' + shaders_path + '", { noembed: ' + str(noembed).lower() + '});\n')
|
|
|
|
if arm.utils.get_gapi() == 'direct3d11':
|
|
# noprocessing flag - gets renamed to .d3d11
|
|
shaders_path = arm.utils.build_dir() + '/compiled/Hlsl/*.glsl'
|
|
if rel_path:
|
|
shaders_path = os.path.relpath(shaders_path, arm.utils.get_fp()).replace('\\', '/')
|
|
khafile.write('project.addShaders("' + shaders_path + '", { noprocessing: true, noembed: ' + str(noembed).lower() + ' });\n')
|
|
|
|
# Move assets for published game to /data folder
|
|
use_data_dir = is_publish and (state.target == 'krom-windows' or state.target == 'krom-linux' or state.target == 'windows-hl' or state.target == 'linux-hl')
|
|
if use_data_dir:
|
|
assets.add_khafile_def('arm_data_dir')
|
|
|
|
ext = 'arm' if wrd.arm_minimize else 'json'
|
|
assets_path = arm.utils.build_dir() + '/compiled/Assets/**'
|
|
assets_path_sh = arm.utils.build_dir() + '/compiled/Shaders/*.' + ext
|
|
if rel_path:
|
|
assets_path = os.path.relpath(assets_path, arm.utils.get_fp()).replace('\\', '/')
|
|
assets_path_sh = os.path.relpath(assets_path_sh, arm.utils.get_fp()).replace('\\', '/')
|
|
dest = ''
|
|
if use_data_dir:
|
|
dest += ', destination: "data/{name}"'
|
|
khafile.write('project.addAssets("' + assets_path + '", { notinlist: true ' + dest + '});\n')
|
|
khafile.write('project.addAssets("' + assets_path_sh + '", { notinlist: true ' + dest + '});\n')
|
|
|
|
shader_data_references = sorted(list(set(assets.shader_datas)))
|
|
for ref in shader_data_references:
|
|
ref = ref.replace('\\', '/').replace('//', '/')
|
|
if '/compiled/' in ref: # Asset already included
|
|
continue
|
|
khafile.write(add_assets(ref, use_data_dir=use_data_dir, rel_path=rel_path))
|
|
|
|
asset_references = sorted(list(set(assets.assets)))
|
|
for ref in asset_references:
|
|
ref = ref.replace('\\', '/').replace('//', '/')
|
|
if '/compiled/' in ref: # Asset already included
|
|
continue
|
|
quality = 1.0
|
|
s = ref.lower()
|
|
if s.endswith('.wav'):
|
|
quality = wrd.arm_sound_quality
|
|
elif s.endswith('.png') or s.endswith('.jpg'):
|
|
quality = wrd.arm_texture_quality
|
|
khafile.write(add_assets(ref, quality=quality, use_data_dir=use_data_dir, rel_path=rel_path))
|
|
|
|
if wrd.arm_sound_quality < 1.0 or state.target == 'html5':
|
|
assets.add_khafile_def('arm_soundcompress')
|
|
|
|
if wrd.arm_texture_quality < 1.0:
|
|
assets.add_khafile_def('arm_texcompress')
|
|
|
|
if wrd.arm_debug_console:
|
|
assets.add_khafile_def('arm_debug')
|
|
khafile.write(add_shaders(sdk_path + "/armory/Shaders/debug_draw/**", rel_path=rel_path))
|
|
|
|
if wrd.arm_verbose_output:
|
|
khafile.write("project.addParameter('--times');\n")
|
|
|
|
if export_ui:
|
|
if not os.path.exists('Libraries/zui'):
|
|
khafile.write(add_armory_library(sdk_path, 'lib/zui', rel_path=rel_path))
|
|
p = sdk_path + '/armory/Assets/font_default.ttf'
|
|
p = p.replace('//', '/')
|
|
khafile.write(add_assets(p.replace('\\', '/'), use_data_dir=use_data_dir, rel_path=rel_path))
|
|
assets.add_khafile_def('arm_ui')
|
|
|
|
if wrd.arm_minimize == False:
|
|
assets.add_khafile_def('arm_json')
|
|
|
|
if wrd.arm_deinterleaved_buffers == True:
|
|
assets.add_khafile_def('arm_deinterleaved')
|
|
|
|
if wrd.arm_batch_meshes == True:
|
|
assets.add_khafile_def('arm_batch')
|
|
|
|
if wrd.arm_stream_scene:
|
|
assets.add_khafile_def('arm_stream')
|
|
|
|
rpdat = arm.utils.get_rp()
|
|
if rpdat.arm_skin != 'Off':
|
|
assets.add_khafile_def('arm_skin')
|
|
|
|
if rpdat.arm_particles != 'Off':
|
|
assets.add_khafile_def('arm_particles')
|
|
|
|
if rpdat.rp_draw_order == 'Shader':
|
|
assets.add_khafile_def('arm_draworder_shader')
|
|
|
|
if arm.utils.get_viewport_controls() == 'azerty':
|
|
assets.add_khafile_def('arm_azerty')
|
|
|
|
if os.path.exists(arm.utils.get_fp() + '/Bundled/config.arm'):
|
|
assets.add_khafile_def('arm_config')
|
|
|
|
if is_publish and wrd.arm_loadscreen:
|
|
assets.add_khafile_def('arm_loadscreen')
|
|
|
|
if wrd.arm_winresize or state.target == 'html5':
|
|
assets.add_khafile_def('arm_resizable')
|
|
|
|
# if bpy.data.scenes[0].unit_settings.system_rotation == 'DEGREES':
|
|
# assets.add_khafile_def('arm_degrees')
|
|
|
|
for d in assets.khafile_defs:
|
|
khafile.write("project.addDefine('" + d + "');\n")
|
|
|
|
for p in assets.khafile_params:
|
|
khafile.write("project.addParameter('" + p + "');\n")
|
|
|
|
if wrd.arm_khafile != None:
|
|
khafile.write(wrd.arm_khafile.as_string())
|
|
|
|
if state.target.startswith('android'):
|
|
bundle = 'org.armory3d.' + wrd.arm_project_package if wrd.arm_project_bundle == '' else wrd.arm_project_bundle
|
|
khafile.write("project.targetOptions.android_native.package = '{0}';\n".format(arm.utils.safestr(bundle)))
|
|
if wrd.arm_winorient != 'Multi':
|
|
khafile.write("project.targetOptions.android_native.screenOrientation = '{0}';\n".format(wrd.arm_winorient.lower()))
|
|
elif state.target.startswith('ios'):
|
|
bundle = 'org.armory3d.' + wrd.arm_project_package if wrd.arm_project_bundle == '' else wrd.arm_project_bundle
|
|
khafile.write("project.targetOptions.ios.bundle = '{0}';\n".format(arm.utils.safestr(bundle)))
|
|
|
|
if wrd.arm_project_icon != '':
|
|
shutil.copy(bpy.path.abspath(wrd.arm_project_icon), arm.utils.get_fp() + '/icon.png')
|
|
|
|
khafile.write("\n\nresolve(project);\n")
|
|
|
|
def get_winmode(arm_winmode):
|
|
if arm_winmode == 'Window':
|
|
return 0
|
|
else: # Fullscreen
|
|
return 1
|
|
|
|
def write_config(resx, resy):
|
|
wrd = bpy.data.worlds['Arm']
|
|
p = arm.utils.get_fp() + '/Bundled'
|
|
if not os.path.exists(p):
|
|
os.makedirs(p)
|
|
output = {}
|
|
output['window_mode'] = get_winmode(wrd.arm_winmode)
|
|
output['window_w'] = int(resx)
|
|
output['window_h'] = int(resy)
|
|
output['window_resizable'] = wrd.arm_winresize
|
|
output['window_maximizable'] = wrd.arm_winresize and wrd.arm_winmaximize
|
|
output['window_minimizable'] = wrd.arm_winminimize
|
|
output['window_vsync'] = wrd.arm_vsync
|
|
rpdat = arm.utils.get_rp()
|
|
output['window_msaa'] = int(rpdat.arm_samples_per_pixel)
|
|
output['window_scale'] = 1.0
|
|
output['rp_supersample'] = float(rpdat.rp_supersampling)
|
|
rp_shadowmap_cube = int(rpdat.rp_shadowmap_cube) if rpdat.rp_shadows else 0
|
|
output['rp_shadowmap_cube'] = rp_shadowmap_cube
|
|
rp_shadowmap_cascade = arm.utils.get_cascade_size(rpdat) if rpdat.rp_shadows else 0
|
|
output['rp_shadowmap_cascade'] = rp_shadowmap_cascade
|
|
output['rp_ssgi'] = rpdat.rp_ssgi != 'Off'
|
|
output['rp_ssr'] = rpdat.rp_ssr != 'Off'
|
|
output['rp_bloom'] = rpdat.rp_bloom != 'Off'
|
|
output['rp_motionblur'] = rpdat.rp_motionblur != 'Off'
|
|
output['rp_gi'] = rpdat.rp_voxelao
|
|
output['rp_dynres'] = rpdat.rp_dynres
|
|
with open(p + '/config.arm', 'w') as f:
|
|
f.write(json.dumps(output, sort_keys=True, indent=4))
|
|
|
|
def write_mainhx(scene_name, resx, resy, is_play, is_publish):
|
|
wrd = bpy.data.worlds['Arm']
|
|
rpdat = arm.utils.get_rp()
|
|
scene_ext = '.lz4' if (wrd.arm_asset_compression and is_publish) else ''
|
|
if scene_ext == '' and not wrd.arm_minimize:
|
|
scene_ext = '.json'
|
|
winmode = get_winmode(wrd.arm_winmode)
|
|
# Detect custom render path
|
|
pathpack = 'armory'
|
|
if os.path.isfile(arm.utils.get_fp() + '/Sources/' + wrd.arm_project_package + '/renderpath/RenderPathCreator.hx'):
|
|
pathpack = wrd.arm_project_package
|
|
elif rpdat.rp_driver != 'Armory':
|
|
pathpack = rpdat.rp_driver.lower()
|
|
|
|
with open('Sources/Main.hx', 'w') as f:
|
|
f.write(
|
|
"""// Auto-generated
|
|
package ;
|
|
class Main {
|
|
public static inline var projectName = '""" + arm.utils.safestr(wrd.arm_project_name) + """';
|
|
public static inline var projectVersion = '""" + arm.utils.safestr(wrd.arm_project_version) + """';
|
|
public static inline var projectPackage = '""" + arm.utils.safestr(wrd.arm_project_package) + """';""")
|
|
|
|
if rpdat.rp_voxelao:
|
|
f.write("""
|
|
public static inline var voxelgiVoxelSize = """ + str(rpdat.arm_voxelgi_dimensions) + " / " + str(rpdat.rp_voxelgi_resolution) + """;
|
|
public static inline var voxelgiHalfExtents = """ + str(round(rpdat.arm_voxelgi_dimensions / 2.0)) + """;""")
|
|
|
|
if rpdat.arm_rp_resolution == 'Custom':
|
|
f.write("""
|
|
public static inline var resolutionSize = """ + str(rpdat.arm_rp_resolution_size) + """;""")
|
|
|
|
f.write("""
|
|
public static function main() {""")
|
|
if rpdat.arm_skin != 'Off':
|
|
f.write("""
|
|
iron.object.BoneAnimation.skinMaxBones = """ + str(rpdat.arm_skin_max_bones) + """;""")
|
|
if rpdat.rp_shadowmap_cascades != '1':
|
|
f.write("""
|
|
iron.object.LightObject.cascadeCount = """ + str(rpdat.rp_shadowmap_cascades) + """;
|
|
iron.object.LightObject.cascadeSplitFactor = """ + str(rpdat.arm_shadowmap_split) + """;""")
|
|
if rpdat.arm_shadowmap_bounds != 1.0:
|
|
f.write("""
|
|
iron.object.LightObject.cascadeBounds = """ + str(rpdat.arm_shadowmap_bounds) + """;""")
|
|
if is_publish and wrd.arm_loadscreen:
|
|
asset_references = list(set(assets.assets))
|
|
loadscreen_class = 'armory.trait.internal.LoadingScreen'
|
|
if os.path.isfile(arm.utils.get_fp() + '/Sources/' + wrd.arm_project_package + '/LoadingScreen.hx'):
|
|
loadscreen_class = wrd.arm_project_package + '.LoadingScreen'
|
|
f.write("""
|
|
armory.system.Starter.numAssets = """ + str(len(asset_references)) + """;
|
|
armory.system.Starter.drawLoading = """ + loadscreen_class + """.render;""")
|
|
f.write("""
|
|
armory.system.Starter.main(
|
|
'""" + arm.utils.safestr(scene_name) + scene_ext + """',
|
|
""" + str(winmode) + """,
|
|
""" + ('true' if wrd.arm_winresize else 'false') + """,
|
|
""" + ('true' if wrd.arm_winminimize else 'false') + """,
|
|
""" + ('true' if (wrd.arm_winresize and wrd.arm_winmaximize) else 'false') + """,
|
|
""" + str(resx) + """,
|
|
""" + str(resy) + """,
|
|
""" + str(int(rpdat.arm_samples_per_pixel)) + """,
|
|
""" + ('true' if wrd.arm_vsync else 'false') + """,
|
|
""" + pathpack + """.renderpath.RenderPathCreator.get
|
|
);
|
|
}
|
|
}
|
|
""")
|
|
|
|
def write_indexhtml(w, h, is_publish):
|
|
wrd = bpy.data.worlds['Arm']
|
|
rpdat = arm.utils.get_rp()
|
|
dest = '/html5' if is_publish else '/debug/html5'
|
|
if not os.path.exists(arm.utils.build_dir() + dest):
|
|
os.makedirs(arm.utils.build_dir() + dest)
|
|
with open(arm.utils.build_dir() + dest + '/index.html', 'w') as f:
|
|
f.write(
|
|
"""<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta charset="utf-8"/>""")
|
|
if rpdat.rp_stereo or wrd.arm_winmode == 'Fullscreen':
|
|
f.write("""
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0, user-scalable=no">
|
|
<style>
|
|
body {
|
|
margin: 0;
|
|
}
|
|
</style>
|
|
""")
|
|
f.write("""
|
|
<title>Armory</title>
|
|
</head>
|
|
<body style="margin: 0; padding: 0;">
|
|
""")
|
|
if rpdat.rp_stereo or wrd.arm_winmode == 'Fullscreen':
|
|
f.write("""
|
|
<canvas style="width: 100vw; height: 100vh; display: block;" id='khanvas' tabindex='-1'></canvas>
|
|
""")
|
|
else:
|
|
f.write("""
|
|
<p align="center"><canvas align="center" style="outline: none;" id='khanvas' width='""" + str(w) + """' height='""" + str(h) + """' tabindex='-1'></canvas></p>
|
|
""")
|
|
f.write("""
|
|
<script src='kha.js'></script>
|
|
</body>
|
|
</html>
|
|
""")
|
|
|
|
add_compiledglsl = ''
|
|
def write_compiledglsl(defs, make_variants):
|
|
rpdat = arm.utils.get_rp()
|
|
shadowmap_size = arm.utils.get_cascade_size(rpdat) if rpdat.rp_shadows else 0
|
|
with open(arm.utils.build_dir() + '/compiled/Shaders/compiled.inc', 'w') as f:
|
|
f.write(
|
|
"""#ifndef _COMPILED_GLSL_
|
|
#define _COMPILED_GLSL_
|
|
""")
|
|
for d in defs:
|
|
if make_variants and d.endswith('var'):
|
|
continue # Write a shader variant instead
|
|
f.write("#define " + d + "\n")
|
|
|
|
f.write("""#if defined(HLSL) || defined(METAL)
|
|
#define _InvY
|
|
#endif
|
|
""")
|
|
|
|
f.write("""const float PI = 3.1415926535;
|
|
const float PI2 = PI * 2.0;
|
|
const vec2 shadowmapSize = vec2(""" + str(shadowmap_size) + """, """ + str(shadowmap_size) + """);
|
|
const float shadowmapCubePcfSize = """ + str((round(rpdat.arm_pcfsize * 100) / 100) / 1000) + """;
|
|
const int shadowmapCascades = """ + str(rpdat.rp_shadowmap_cascades) + """;
|
|
""")
|
|
|
|
if rpdat.rp_water:
|
|
f.write(
|
|
"""const float waterLevel = """ + str(round(rpdat.arm_water_level * 100) / 100) + """;
|
|
const float waterDisplace = """ + str(round(rpdat.arm_water_displace * 100) / 100) + """;
|
|
const float waterSpeed = """ + str(round(rpdat.arm_water_speed * 100) / 100) + """;
|
|
const float waterFreq = """ + str(round(rpdat.arm_water_freq * 100) / 100) + """;
|
|
const vec3 waterColor = vec3(""" + str(round(rpdat.arm_water_color[0] * 100) / 100) + """, """ + str(round(rpdat.arm_water_color[1] * 100) / 100) + """, """ + str(round(rpdat.arm_water_color[2] * 100) / 100) + """);
|
|
const float waterDensity = """ + str(round(rpdat.arm_water_density * 100) / 100) + """;
|
|
const float waterRefract = """ + str(round(rpdat.arm_water_refract * 100) / 100) + """;
|
|
const float waterReflect = """ + str(round(rpdat.arm_water_reflect * 100) / 100) + """;
|
|
""")
|
|
if rpdat.rp_ssgi == 'SSAO' or rpdat.rp_ssgi == 'RTAO' or rpdat.rp_volumetriclight:
|
|
f.write(
|
|
"""const float ssaoRadius = """ + str(round(rpdat.arm_ssgi_radius * 100) / 100) + """;
|
|
const float ssaoStrength = """ + str(round(rpdat.arm_ssgi_strength * 100) / 100) + """;
|
|
const float ssaoScale = """ + ("2.0" if rpdat.arm_ssgi_half_res else "20.0") + """;
|
|
""")
|
|
|
|
if rpdat.rp_ssgi == 'RTGI' or rpdat.rp_ssgi == 'RTAO':
|
|
f.write(
|
|
"""const int ssgiMaxSteps = """ + str(rpdat.arm_ssgi_max_steps) + """;
|
|
const float ssgiRayStep = 0.005 * """ + str(round(rpdat.arm_ssgi_step * 100) / 100) + """;
|
|
const float ssgiStrength = """ + str(round(rpdat.arm_ssgi_strength * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.rp_bloom:
|
|
f.write(
|
|
"""const float bloomThreshold = """ + str(round(rpdat.arm_bloom_threshold * 100) / 100) + """;
|
|
const float bloomStrength = """ + str(round(rpdat.arm_bloom_strength * 100) / 100) + """;
|
|
const float bloomRadius = """ + str(round(rpdat.arm_bloom_radius * 100) / 100) + """;
|
|
""")
|
|
if rpdat.rp_motionblur != 'Off':
|
|
f.write(
|
|
"""const float motionBlurIntensity = """ + str(round(rpdat.arm_motion_blur_intensity * 100) / 100) + """;
|
|
""")
|
|
if rpdat.rp_ssr:
|
|
f.write(
|
|
"""const float ssrRayStep = """ + str(round(rpdat.arm_ssr_ray_step * 100) / 100) + """;
|
|
const float ssrMinRayStep = """ + str(round(rpdat.arm_ssr_min_ray_step * 100) / 100) + """;
|
|
const float ssrSearchDist = """ + str(round(rpdat.arm_ssr_search_dist * 100) / 100) + """;
|
|
const float ssrFalloffExp = """ + str(round(rpdat.arm_ssr_falloff_exp * 100) / 100) + """;
|
|
const float ssrJitter = """ + str(round(rpdat.arm_ssr_jitter * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.arm_ssrs:
|
|
f.write(
|
|
"""const float ssrsRayStep = """ + str(round(rpdat.arm_ssrs_ray_step * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.rp_volumetriclight:
|
|
f.write(
|
|
"""const float volumAirTurbidity = """ + str(round(rpdat.arm_volumetric_light_air_turbidity * 100) / 100) + """;
|
|
const vec3 volumAirColor = vec3(""" + str(round(rpdat.arm_volumetric_light_air_color[0] * 100) / 100) + """, """ + str(round(rpdat.arm_volumetric_light_air_color[1] * 100) / 100) + """, """ + str(round(rpdat.arm_volumetric_light_air_color[2] * 100) / 100) + """);
|
|
const int volumSteps = """ + str(rpdat.arm_volumetric_light_steps) + """;
|
|
""")
|
|
|
|
if rpdat.rp_autoexposure:
|
|
f.write(
|
|
"""const float autoExposureStrength = """ + str(rpdat.arm_autoexposure_strength) + """;
|
|
const float autoExposureSpeed = """ + str(rpdat.arm_autoexposure_speed) + """;
|
|
""")
|
|
|
|
# Compositor
|
|
if rpdat.arm_letterbox:
|
|
f.write(
|
|
"""const float compoLetterboxSize = """ + str(round(rpdat.arm_letterbox_size * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.arm_grain:
|
|
f.write(
|
|
"""const float compoGrainStrength = """ + str(round(rpdat.arm_grain_strength * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.arm_vignette:
|
|
f.write(
|
|
"""const float compoVignetteStrength = """ + str(round(rpdat.arm_vignette_strength * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.arm_sharpen:
|
|
f.write(
|
|
"""const float compoSharpenStrength = """ + str(round(rpdat.arm_sharpen_strength * 100) / 100) + """;
|
|
""")
|
|
|
|
if bpy.data.scenes[0].view_settings.exposure != 0.0:
|
|
f.write(
|
|
"""const float compoExposureStrength = """ + str(round(bpy.data.scenes[0].view_settings.exposure * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.arm_fog:
|
|
f.write(
|
|
"""const float compoFogAmountA = """ + str(round(rpdat.arm_fog_amounta * 100) / 100) + """;
|
|
const float compoFogAmountB = """ + str(round(rpdat.arm_fog_amountb * 100) / 100) + """;
|
|
const vec3 compoFogColor = vec3(""" + str(round(rpdat.arm_fog_color[0] * 100) / 100) + """, """ + str(round(rpdat.arm_fog_color[1] * 100) / 100) + """, """ + str(round(rpdat.arm_fog_color[2] * 100) / 100) + """);
|
|
""")
|
|
|
|
if rpdat.arm_lens_texture_masking:
|
|
f.write(
|
|
"""const float compoCenterMinClip = """ + str(round(rpdat.arm_lens_texture_masking_centerMinClip * 100) / 100) + """;
|
|
const float compoCenterMaxClip = """ + str(round(rpdat.arm_lens_texture_masking_centerMaxClip * 100) / 100) + """;
|
|
const float compoLuminanceMin = """ + str(round(rpdat.arm_lens_texture_masking_luminanceMin * 100) / 100) + """;
|
|
const float compoLuminanceMax = """ + str(round(rpdat.arm_lens_texture_masking_luminanceMax * 100) / 100) + """;
|
|
const float compoBrightnessExponent = """ + str(round(rpdat.arm_lens_texture_masking_brightnessExp * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.rp_chromatic_aberration:
|
|
f.write(
|
|
"""const float compoChromaticStrength = """ + str(round(rpdat.arm_chromatic_aberration_strength * 100) / 100) + """;
|
|
const int compoChromaticSamples = """ + str(rpdat.arm_chromatic_aberration_samples) + """;
|
|
""")
|
|
|
|
if rpdat.arm_chromatic_aberration_type == "Spectral":
|
|
f.write(
|
|
"""const int compoChromaticType = """ + str(1) + """;
|
|
""")
|
|
else:
|
|
f.write(
|
|
"""const int compoChromaticType = """ + str(0) + """;
|
|
""")
|
|
|
|
focus_distance = 0.0
|
|
fstop = 0.0
|
|
if len(bpy.data.cameras) > 0 and bpy.data.cameras[0].dof.use_dof:
|
|
focus_distance = bpy.data.cameras[0].dof.focus_distance
|
|
fstop = bpy.data.cameras[0].dof.aperture_fstop
|
|
|
|
if focus_distance > 0.0:
|
|
f.write(
|
|
"""const float compoDOFDistance = """ + str(round(focus_distance * 100) / 100) + """;
|
|
const float compoDOFFstop = """ + str(round(fstop * 100) / 100) + """;
|
|
const float compoDOFLength = 160.0;
|
|
""") # str(round(bpy.data.cameras[0].lens * 100) / 100)
|
|
|
|
if rpdat.rp_voxelao:
|
|
halfext = round(rpdat.arm_voxelgi_dimensions / 2.0)
|
|
f.write(
|
|
"""const ivec3 voxelgiResolution = ivec3(""" + str(rpdat.rp_voxelgi_resolution) + """, """ + str(rpdat.rp_voxelgi_resolution) + """, """ + str(int(int(rpdat.rp_voxelgi_resolution) * float(rpdat.rp_voxelgi_resolution_z))) + """);
|
|
const vec3 voxelgiHalfExtents = vec3(""" + str(halfext) + """, """ + str(halfext) + """, """ + str(round(halfext * float(rpdat.rp_voxelgi_resolution_z))) + """);
|
|
const float voxelgiOcc = """ + str(round(rpdat.arm_voxelgi_occ * 100) / 100) + """;
|
|
const float voxelgiStep = """ + str(round(rpdat.arm_voxelgi_step * 100) / 100) + """;
|
|
const float voxelgiRange = """ + str(round(rpdat.arm_voxelgi_range * 100) / 100) + """;
|
|
const float voxelgiOffset = """ + str(round(rpdat.arm_voxelgi_offset * 100) / 100) + """;
|
|
const float voxelgiAperture = """ + str(round(rpdat.arm_voxelgi_aperture * 100) / 100) + """;
|
|
""")
|
|
|
|
if rpdat.rp_sss_state == 'On':
|
|
f.write(
|
|
"""const float sssWidth = """ + str(rpdat.arm_sss_width / 10.0) + """;
|
|
""")
|
|
|
|
# Skinning
|
|
if rpdat.arm_skin == 'On':
|
|
f.write(
|
|
"""const int skinMaxBones = """ + str(rpdat.arm_skin_max_bones) + """;
|
|
""")
|
|
|
|
f.write(add_compiledglsl + '\n') # External defined constants
|
|
|
|
f.write("""#endif // _COMPILED_GLSL_
|
|
""")
|
|
|
|
def write_traithx(class_path):
|
|
wrd = bpy.data.worlds['Arm']
|
|
# Split the haxe package syntax in components that will compose the path
|
|
path_components = class_path.split('.')
|
|
# extract the full file name (file + ext) from the components
|
|
class_name = path_components[-1]
|
|
# Create the absolute trait path (os-safe)
|
|
package_path = os.sep.join([arm.utils.get_fp(), 'Sources', arm.utils.safestr(wrd.arm_project_package)] + path_components[:-1])
|
|
if not os.path.exists(package_path):
|
|
os.makedirs(package_path)
|
|
package = '.'.join([arm.utils.safestr(wrd.arm_project_package)] + path_components[:-1]);
|
|
with open(package_path + '/' + class_name + '.hx', 'w') as f:
|
|
f.write(
|
|
"""package """ + package + """;
|
|
|
|
class """ + class_name + """ extends iron.Trait {
|
|
\tpublic function new() {
|
|
\t\tsuper();
|
|
|
|
\t\t// notifyOnInit(function() {
|
|
\t\t// });
|
|
|
|
\t\t// notifyOnUpdate(function() {
|
|
\t\t// });
|
|
|
|
\t\t// notifyOnRemove(function() {
|
|
\t\t// });
|
|
\t}
|
|
}
|
|
""")
|
|
|
|
def write_canvasjson(canvas_name):
|
|
canvas_path = arm.utils.get_fp() + '/Bundled/canvas'
|
|
if not os.path.exists(canvas_path):
|
|
os.makedirs(canvas_path)
|
|
with open(canvas_path + '/' + canvas_name + '.json', 'w') as f:
|
|
f.write(
|
|
"""{ "name": "untitled", "x": 0.0, "y": 0.0, "width": 1280, "height": 720, "theme": "Default Light", "elements": [], "assets": [] }""")
|