287 lines
12 KiB
Python
287 lines
12 KiB
Python
import os
|
|
import shutil
|
|
import arm.assets as assets
|
|
import arm.utils
|
|
import bpy
|
|
import stat
|
|
import subprocess
|
|
import webbrowser
|
|
from bpy.types import Menu, Panel, UIList
|
|
from bpy.props import *
|
|
|
|
def remove_readonly(func, path, excinfo):
|
|
os.chmod(path, stat.S_IWRITE)
|
|
func(path)
|
|
|
|
def update_gapi_custom(self, context):
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
def update_gapi_win(self, context):
|
|
if os.path.isdir(arm.utils.get_fp_build() + '/windows-build'):
|
|
shutil.rmtree(arm.utils.get_fp_build() + '/windows-build', onerror=remove_readonly)
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
def update_gapi_linux(self, context):
|
|
if os.path.isdir(arm.utils.get_fp_build() + '/linux-build'):
|
|
shutil.rmtree(arm.utils.get_fp_build() + '/linux-build', onerror=remove_readonly)
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
def update_gapi_mac(self, context):
|
|
if os.path.isdir(arm.utils.get_fp_build() + '/osx-build'):
|
|
shutil.rmtree(arm.utils.get_fp_build() + '/osx-build', onerror=remove_readonly)
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
def update_gapi_android(self, context):
|
|
if os.path.isdir(arm.utils.get_fp_build() + '/android-build'):
|
|
shutil.rmtree(arm.utils.get_fp_build() + '/android-build', onerror=remove_readonly)
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
def update_gapi_ios(self, context):
|
|
if os.path.isdir(arm.utils.get_fp_build() + '/ios-build'):
|
|
shutil.rmtree(arm.utils.get_fp_build() + '/ios-build', onerror=remove_readonly)
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
def update_gapi_html5(self, context):
|
|
bpy.data.worlds['Arm'].arm_recompile = True
|
|
assets.invalidate_compiled_data(self, context)
|
|
|
|
class ArmExporterListItem(bpy.types.PropertyGroup):
|
|
name = bpy.props.StringProperty(
|
|
name="Name",
|
|
description="A name for this item",
|
|
default="Preset")
|
|
|
|
arm_project_rp = bpy.props.StringProperty(
|
|
name="Render Path",
|
|
description="A name for this item",
|
|
default="Path")
|
|
|
|
arm_project_scene = StringProperty(name="Scene", description="Scene to load when launching")
|
|
|
|
arm_project_target = EnumProperty(
|
|
items = [('html5', 'HTML5 (JS)', 'html5'),
|
|
('windows-hl', 'Windows (C)', 'windows-hl'),
|
|
('krom-windows', 'Windows (Krom)', 'krom-windows'),
|
|
('macos-hl', 'macOS (C)', 'macos-hl'),
|
|
('krom-macos', 'macOS (Krom)', 'krom-macos'),
|
|
('linux-hl', 'Linux (C)', 'linux-hl'),
|
|
('krom-linux', 'Linux (Krom)', 'krom-linux'),
|
|
('ios-hl', 'iOS (C)', 'ios-hl'),
|
|
('android-native-hl', 'Android (C)', 'android-native-hl'),
|
|
('node', 'Node (JS)', 'node'),
|
|
('custom', 'Custom', 'custom'),],
|
|
name="Target", default='html5', description='Build platform')
|
|
|
|
arm_project_khamake = StringProperty(name="Khamake", description="Specify arguments for the 'node Kha/make' call")
|
|
|
|
arm_gapi_custom = EnumProperty(
|
|
items = [('opengl', 'OpenGL', 'opengl'),
|
|
('vulkan', 'Vulkan', 'vulkan'),
|
|
('direct3d11', 'Direct3D11', 'direct3d11'),
|
|
('direct3d12', 'Direct3D12', 'direct3d12'),
|
|
('metal', 'Metal', 'metal')],
|
|
name="Graphics API", default='opengl', description='Based on currently selected target', update=update_gapi_custom)
|
|
arm_gapi_win = EnumProperty(
|
|
items = [('opengl', 'Auto', 'opengl'),
|
|
('opengl', 'OpenGL', 'opengl'),
|
|
('vulkan', 'Vulkan', 'vulkan'),
|
|
('direct3d11', 'Direct3D11', 'direct3d11'),
|
|
('direct3d12', 'Direct3D12', 'direct3d12')],
|
|
name="Graphics API", default='opengl', description='Based on currently selected target', update=update_gapi_win)
|
|
arm_gapi_linux = EnumProperty(
|
|
items = [('opengl', 'Auto', 'opengl'),
|
|
('opengl', 'OpenGL', 'opengl'),
|
|
('vulkan', 'Vulkan', 'vulkan')],
|
|
name="Graphics API", default='opengl', description='Based on currently selected target', update=update_gapi_linux)
|
|
arm_gapi_android = EnumProperty(
|
|
items = [('opengl', 'Auto', 'opengl'),
|
|
('opengl', 'OpenGL', 'opengl'),
|
|
('vulkan', 'Vulkan', 'vulkan')],
|
|
name="Graphics API", default='opengl', description='Based on currently selected target', update=update_gapi_android)
|
|
arm_gapi_mac = EnumProperty(
|
|
items = [('opengl', 'Auto', 'opengl'),
|
|
('opengl', 'OpenGL', 'opengl'),
|
|
('metal', 'Metal', 'metal')],
|
|
name="Graphics API", default='opengl', description='Based on currently selected target', update=update_gapi_mac)
|
|
arm_gapi_ios = EnumProperty(
|
|
items = [('opengl', 'Auto', 'opengl'),
|
|
('opengl', 'OpenGL', 'opengl'),
|
|
('metal', 'Metal', 'metal')],
|
|
name="Graphics API", default='opengl', description='Based on currently selected target', update=update_gapi_ios)
|
|
arm_gapi_html5 = EnumProperty(
|
|
items = [('webgl', 'Auto', 'webgl'),
|
|
('webgl', 'WebGL2', 'webgl')],
|
|
name="Graphics API", default='webgl', description='Based on currently selected target', update=update_gapi_html5)
|
|
|
|
class ArmExporterList(bpy.types.UIList):
|
|
def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
|
|
# We could write some code to decide which icon to use here...
|
|
custom_icon = 'OBJECT_DATAMODE'
|
|
|
|
# Make sure your code supports all 3 layout types
|
|
if self.layout_type in {'DEFAULT', 'COMPACT'}:
|
|
row = layout.row()
|
|
row.prop(item, "name", text="", emboss=False, icon=custom_icon)
|
|
col = row.column()
|
|
col.alignment = 'RIGHT'
|
|
col.label(text=item.arm_project_target)
|
|
|
|
elif self.layout_type in {'GRID'}:
|
|
layout.alignment = 'CENTER'
|
|
layout.label(text="", icon = custom_icon)
|
|
|
|
class ArmExporterListNewItem(bpy.types.Operator):
|
|
# Add a new item to the list
|
|
bl_idname = "arm_exporterlist.new_item"
|
|
bl_label = "Add a new item"
|
|
|
|
def execute(self, context):
|
|
mdata = bpy.data.worlds['Arm']
|
|
mdata.arm_exporterlist.add()
|
|
mdata.arm_exporterlist_index = len(mdata.arm_exporterlist) - 1
|
|
if len(mdata.arm_rplist) > mdata.arm_exporterlist_index:
|
|
mdata.arm_exporterlist[-1].arm_project_rp = mdata.arm_rplist[mdata.arm_rplist_index].name
|
|
return{'FINISHED'}
|
|
|
|
class ArmExporterListDeleteItem(bpy.types.Operator):
|
|
# Delete the selected item from the list
|
|
bl_idname = "arm_exporterlist.delete_item"
|
|
bl_label = "Deletes an item"
|
|
|
|
@classmethod
|
|
def poll(self, context):
|
|
""" Enable if there's something in the list """
|
|
mdata = bpy.data.worlds['Arm']
|
|
return len(mdata.arm_exporterlist) > 0
|
|
|
|
def execute(self, context):
|
|
mdata = bpy.data.worlds['Arm']
|
|
list = mdata.arm_exporterlist
|
|
index = mdata.arm_exporterlist_index
|
|
|
|
list.remove(index)
|
|
|
|
if index > 0:
|
|
index = index - 1
|
|
|
|
mdata.arm_exporterlist_index = index
|
|
return{'FINISHED'}
|
|
|
|
class ArmExporterListMoveItem(bpy.types.Operator):
|
|
# Move an item in the list
|
|
bl_idname = "arm_exporterlist.move_item"
|
|
bl_label = "Move an item in the list"
|
|
direction = bpy.props.EnumProperty(
|
|
items=(
|
|
('UP', 'Up', ""),
|
|
('DOWN', 'Down', ""),))
|
|
|
|
def move_index(self):
|
|
# Move index of an item render queue while clamping it
|
|
mdata = bpy.data.worlds['Arm']
|
|
index = mdata.arm_exporterlist_index
|
|
list_length = len(mdata.arm_exporterlist) - 1
|
|
new_index = 0
|
|
|
|
if self.direction == 'UP':
|
|
new_index = index - 1
|
|
elif self.direction == 'DOWN':
|
|
new_index = index + 1
|
|
|
|
new_index = max(0, min(new_index, list_length))
|
|
mdata.arm_exporterlist.move(index, new_index)
|
|
mdata.arm_exporterlist_index = new_index
|
|
|
|
def execute(self, context):
|
|
mdata = bpy.data.worlds['Arm']
|
|
list = mdata.arm_exporterlist
|
|
index = mdata.arm_exporterlist_index
|
|
|
|
if self.direction == 'DOWN':
|
|
neighbor = index + 1
|
|
self.move_index()
|
|
|
|
elif self.direction == 'UP':
|
|
neighbor = index - 1
|
|
self.move_index()
|
|
else:
|
|
return{'CANCELLED'}
|
|
return{'FINISHED'}
|
|
|
|
class ArmExporterSpecialsMenu(bpy.types.Menu):
|
|
bl_label = "More"
|
|
bl_idname = "arm_exporterlist_specials"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator("arm.exporter_open_folder")
|
|
layout.operator("arm.exporter_gpuprofile")
|
|
|
|
class ArmoryExporterOpenFolderButton(bpy.types.Operator):
|
|
'''Open published folder'''
|
|
bl_idname = 'arm.exporter_open_folder'
|
|
bl_label = 'Open Folder'
|
|
|
|
def execute(self, context):
|
|
wrd = bpy.data.worlds['Arm']
|
|
if len(wrd.arm_exporterlist) == 0:
|
|
return {'CANCELLED'}
|
|
item = wrd.arm_exporterlist[wrd.arm_exporterlist_index]
|
|
p = os.path.join(arm.utils.get_fp_build(), item.arm_project_target)
|
|
if os.path.exists(p):
|
|
webbrowser.open('file://' + p)
|
|
return{'FINISHED'}
|
|
|
|
class ArmExporterGpuProfileButton(bpy.types.Operator):
|
|
'''GPU profile'''
|
|
bl_idname = 'arm.exporter_gpuprofile'
|
|
bl_label = 'Open in RenderDoc'
|
|
|
|
def execute(self, context):
|
|
p = arm.utils.get_renderdoc_path()
|
|
if p == '':
|
|
self.report({'ERROR'}, 'Configure RenderDoc path in Armory add-on preferences')
|
|
return {'CANCELLED'}
|
|
pbin = ''
|
|
base = arm.utils.get_fp_build()
|
|
ext1 = '/krom-windows/' + arm.utils.safestr(bpy.data.worlds['Arm'].arm_project_name) + '.exe'
|
|
ext2 = '/krom-linux/' + arm.utils.safestr(bpy.data.worlds['Arm'].arm_project_name)
|
|
if os.path.exists(base + ext1):
|
|
pbin = base + ext1
|
|
elif os.path.exists(base + ext2):
|
|
pbin = base + ext2
|
|
if pbin == '':
|
|
self.report({'ERROR'}, 'Publish project using Krom target first')
|
|
return {'CANCELLED'}
|
|
subprocess.Popen([p, pbin])
|
|
return{'FINISHED'}
|
|
|
|
def register():
|
|
bpy.utils.register_class(ArmExporterListItem)
|
|
bpy.utils.register_class(ArmExporterList)
|
|
bpy.utils.register_class(ArmExporterListNewItem)
|
|
bpy.utils.register_class(ArmExporterListDeleteItem)
|
|
bpy.utils.register_class(ArmExporterListMoveItem)
|
|
bpy.utils.register_class(ArmExporterSpecialsMenu)
|
|
bpy.utils.register_class(ArmExporterGpuProfileButton)
|
|
bpy.utils.register_class(ArmoryExporterOpenFolderButton)
|
|
|
|
bpy.types.World.arm_exporterlist = bpy.props.CollectionProperty(type=ArmExporterListItem)
|
|
bpy.types.World.arm_exporterlist_index = bpy.props.IntProperty(name="Index for my_list", default=0)
|
|
|
|
def unregister():
|
|
bpy.utils.unregister_class(ArmExporterListItem)
|
|
bpy.utils.unregister_class(ArmExporterList)
|
|
bpy.utils.unregister_class(ArmExporterListNewItem)
|
|
bpy.utils.unregister_class(ArmExporterListDeleteItem)
|
|
bpy.utils.unregister_class(ArmExporterListMoveItem)
|
|
bpy.utils.unregister_class(ArmExporterSpecialsMenu)
|
|
bpy.utils.unregister_class(ArmExporterGpuProfileButton)
|
|
bpy.utils.unregister_class(ArmoryExporterOpenFolderButton)
|