import bpy import os import assets import utils def add_armory_library(sdk_path, name): return ('project.addLibrary("../' + bpy.path.relpath(sdk_path + '/' + name)[2:] + '");\n').replace('\\', '/') # Write khafile.js def write_khafilejs(shader_references, asset_references): user_preferences = bpy.context.user_preferences addon_prefs = user_preferences.addons['armory'].preferences sdk_path = addon_prefs.sdk_path # Merge duplicates and sort shader_references = sorted(list(set(shader_references))) asset_references = sorted(list(set(asset_references))) with open('khafile.js', 'w') as f: f.write( """// Auto-generated let project = new Project('""" + bpy.data.worlds['Arm'].ArmProjectName + """'); project.addSources('Sources'); project.addShaders('Sources/Shaders/**'); """) # project.addAssets('build/compiled/Assets/**'); for file in assets.assets: f.write("project.addAssets('" + file + "');\n") f.write(add_armory_library(sdk_path, 'armory')) f.write(add_armory_library(sdk_path, 'iron')) if bpy.data.worlds['Arm'].ArmPhysics != 'Disabled': f.write("project.addDefine('WITH_PHYSICS');\n") f.write(add_armory_library(sdk_path + '/lib/', 'haxebullet')) if bpy.data.worlds['Arm'].ArmPlayLivePatch == True: f.write("project.addDefine('WITH_LIVEPATCH');\n") # Native scripting # f.write(add_armory_library(sdk_path + '/lib/', 'haxeduktape')) for i in range(0, len(shader_references)): # Shaders ref = shader_references[i] # defs = shader_references_defs[i] f.write("project.addShaders('" + ref + ".frag.glsl');\n") f.write("project.addShaders('" + ref + ".vert.glsl');\n") for ref in asset_references: # Assets ref = ref.replace('\\', '/') f.write("project.addAssets('" + ref + "');\n") if bpy.data.worlds['Arm'].ArmPlayConsole: f.write("project.addDefine('WITH_PROFILE');\n") f.write(add_armory_library(sdk_path, 'lib/zui')) font_path = sdk_path + '/armory/Assets/droid_sans.ttf' font_path = font_path.replace('\\', '/') f.write('project.addAssets("' + font_path + '");\n') # f.write(add_armory_library(sdk_path, 'lib/haxeui/haxeui-core')) # f.write(add_armory_library(sdk_path, 'lib/haxeui/haxeui-kha')) # f.write(add_armory_library(sdk_path, 'lib/haxeui/hscript')) if bpy.data.worlds['Arm'].ArmMinimize == False: f.write("project.addDefine('WITH_JSON');\n") if bpy.data.worlds['Arm'].ArmDeinterleavedBuffers == True: f.write("project.addDefine('WITH_DEINTERLEAVED');\n") if bpy.data.worlds['Arm'].generate_gpu_skin == False: f.write("project.addDefine('WITH_CPU_SKIN');\n") for d in assets.khafile_defs: f.write("project.addDefine('" + d + "');\n") config_text = bpy.data.worlds['Arm'].ArmKhafile if config_text != '': f.write(bpy.data.texts[config_text].as_string()) f.write("\n\nresolve(project);\n") def get_project_scene(): wrd = bpy.data.worlds['Arm'] if wrd.ArmPlayActiveScene: return utils.safe_filename(bpy.context.screen.scene.name) else: return utils.safe_filename(wrd.ArmProjectScene) # Write Main.hx def write_main(): wrd = bpy.data.worlds['Arm'] resx, resy = utils.get_render_resolution() #if not os.path.isfile('Sources/Main.hx'): with open('Sources/Main.hx', 'w') as f: f.write( """// Auto-generated package ; class Main { public static inline var projectName = '""" + wrd.ArmProjectName + """'; public static inline var projectPackage = '""" + wrd.ArmProjectPackage + """'; static inline var projectWidth = """ + str(resx) + """; static inline var projectHeight = """ + str(resy) + """; static inline var projectSamplesPerPixel = """ + str(wrd.ArmProjectSamplesPerPixel) + """; static inline var projectScene = '""" + get_project_scene() + """'; public static function main() { iron.sys.CompileTime.importPackage('armory.trait'); iron.sys.CompileTime.importPackage('armory.renderpipeline'); iron.sys.CompileTime.importPackage('""" + wrd.ArmProjectPackage + """'); #if (js && WITH_PHYSICS) untyped __js__(" function loadScript(url, callback) { var head = document.getElementsByTagName('head')[0]; var script = document.createElement('script'); script.type = 'text/javascript'; script.src = url; script.onreadystatechange = callback; script.onload = callback; head.appendChild(script); } "); """) f.write(""" untyped loadScript('ammo.js', start); #else start(); #end } static function start() {""") f.write(""" kha.System.init({title: projectName, width: projectWidth, height: projectHeight, samplesPerPixel: projectSamplesPerPixel}, function() { iron.App.init(function() { var scene = iron.Scene.setActive(projectScene); scene.addTrait(new armory.trait.internal.PhysicsWorld()); iron.App.notifyOnRender(function(g:kha.graphics4.Graphics) { iron.Scene.active.renderFrame(g); }); }); }); } } """) # Write electron.js def write_electronjs(x, y, w, h, winoff, in_viewport): wrd = bpy.data.worlds['Arm'] dev_tools = wrd.ArmPlayDeveloperTools with open('build/electron.js', 'w') as f: f.write( """// Auto-generated 'use strict'; const electron = require('electron'); const app = electron.app; const BrowserWindow = electron.BrowserWindow; let mainWindow; function createWindow () { """) if in_viewport: f.write( """ let point = electron.screen.getCursorScreenPoint(); let targetDisplay = electron.screen.getDisplayNearestPoint(point); """) if utils.get_os() == 'mac': # Perform scale in python f.write("""let scale = 1;""") else: f.write("""let scale = targetDisplay.scaleFactor;""") f.write( """ let _x = Math.floor(""" + str(int(x)) + """ / scale); let _y = Math.floor(""" + str(int(y)) + """ / scale); let _w = Math.floor(""" + str(int(w)) + """ / scale); let _h = Math.floor(""" + str(int(h)) + """ / scale); let _winoff = Math.floor(""" + str(int(winoff)) + """ / scale); let offY = targetDisplay.workAreaSize.height - _winoff; _x = targetDisplay.bounds.x + _x; _y = targetDisplay.bounds.y + _y + offY; mainWindow = new BrowserWindow({x: _x, y: _y, width: _w, height: _h, frame: false, autoHideMenuBar: true, useContentSize: true, movable: false, resizable: false, transparent: true, enableLargerThanScreen: true}); mainWindow.setSkipTaskbar(true); mainWindow.setAlwaysOnTop(true); """) else: f.write( """ mainWindow = new BrowserWindow({width: """ + str(int(w)) + """, height: """ + str(int(h)) + """, autoHideMenuBar: true, useContentSize: true}); """) f.write( """ mainWindow.loadURL('file://' + __dirname + '/html5/index.html'); //mainWindow.loadURL('http://localhost:8040/build/html5/index.html'); mainWindow.on('closed', function() { mainWindow = null; });""") if dev_tools: f.write(""" mainWindow.toggleDevTools();""") f.write(""" } app.on('ready', createWindow); app.on('window-all-closed', function () { app.quit(); }); app.on('activate', function () { if (mainWindow === null) { createWindow(); } }); """) # Write index.html def write_indexhtml(w, h, in_viewport): with open('build/html5/index.html', 'w') as f: f.write( """