mirror of https://github.com/AxioDL/metaforce.git
Added blender project patching system
This commit is contained in:
Jack Andersen
parent
7a27efa88b
commit
432924ccd5
|
|
@ -1,6 +1,7 @@
|
|||
list(APPEND PY_SOURCES
|
||||
hecl/__init__.py
|
||||
hecl/Nodegrid.py
|
||||
hecl/Patching.py
|
||||
hecl/hmdl/__init__.py
|
||||
hecl/hmdl/HMDLMesh.py
|
||||
hecl/hmdl/HMDLShader.py
|
||||
|
|
|
|||
|
|
@ -0,0 +1,128 @@
|
|||
import bpy
|
||||
import os
|
||||
import os.path
|
||||
from pathlib import Path
|
||||
|
||||
def _mkdir(path):
|
||||
try:
|
||||
os.mkdir(path)
|
||||
except:
|
||||
pass
|
||||
|
||||
def path_components(path):
|
||||
retval = []
|
||||
base, end = os.path.split(path)
|
||||
while end != '':
|
||||
retval.insert(0, end)
|
||||
base, end = os.path.split(base)
|
||||
return retval
|
||||
|
||||
def find_project_root():
|
||||
if bpy.data.filepath == '':
|
||||
return None
|
||||
path = os.path.split(bpy.data.filepath)
|
||||
test_path = os.path.join(path[0], '.hecl')
|
||||
while not os.path.exists(test_path):
|
||||
path = os.path.split(path[0])
|
||||
test_path = os.path.join(path[0], '.hecl')
|
||||
if os.path.exists(test_path):
|
||||
return path[0]
|
||||
return None
|
||||
|
||||
def get_patching_dir(make_dirs=False):
|
||||
proj_root = find_project_root()
|
||||
if not proj_root:
|
||||
return None
|
||||
rel_to_blend = os.path.relpath(bpy.data.filepath, start=proj_root)
|
||||
rel_to_blend_comps = path_components(rel_to_blend)
|
||||
trace_dir = os.path.join(proj_root, '.hecl', 'patches')
|
||||
if not make_dirs and not os.path.exists(trace_dir):
|
||||
return None
|
||||
_mkdir(trace_dir)
|
||||
for comp in rel_to_blend_comps:
|
||||
ext_pair = os.path.splitext(comp)
|
||||
if ext_pair[1] == '.blend':
|
||||
trace_dir = os.path.join(trace_dir, ext_pair[0])
|
||||
if not make_dirs and not os.path.exists(trace_dir):
|
||||
return None
|
||||
_mkdir(trace_dir)
|
||||
return trace_dir
|
||||
trace_dir = os.path.join(trace_dir, comp)
|
||||
if not make_dirs and not os.path.exists(trace_dir):
|
||||
return None
|
||||
_mkdir(trace_dir)
|
||||
|
||||
|
||||
class FILE_OT_hecl_patching_save(bpy.types.Operator):
|
||||
'''Save text datablocks to hecl patching directory'''
|
||||
bl_idname = "file.hecl_patching_save"
|
||||
bl_label = "Save HECL Patches"
|
||||
bl_options = {'REGISTER'}
|
||||
|
||||
def execute(self, context):
|
||||
patching_dir = get_patching_dir(make_dirs=True)
|
||||
if not patching_dir:
|
||||
self.report({'ERROR'}, 'Unable to save patches for ' + bpy.data.filepath)
|
||||
return {'CANCELLED'}
|
||||
count = 0
|
||||
for text in bpy.data.texts:
|
||||
if not text.name.endswith('.py'):
|
||||
continue
|
||||
text_abspath = os.path.join(patching_dir, text.name)
|
||||
text_file = open(text_abspath, 'w')
|
||||
text_file.write(text.as_string())
|
||||
text_file.close()
|
||||
count += 1
|
||||
if count == 1:
|
||||
self.report({'INFO'}, 'saved 1 patch')
|
||||
else:
|
||||
self.report({'INFO'}, 'saved %d patches' % count)
|
||||
return {'FINISHED'}
|
||||
|
||||
class FILE_OT_hecl_patching_load(bpy.types.Operator):
|
||||
'''Load text datablocks from hecl patching directory'''
|
||||
bl_idname = "file.hecl_patching_load"
|
||||
bl_label = "Load HECL Patches"
|
||||
bl_options = {'REGISTER'}
|
||||
|
||||
def execute(self, context):
|
||||
patching_dir = get_patching_dir()
|
||||
if not patching_dir:
|
||||
self.report({'ERROR'}, 'Unable to load patches for ' + bpy.data.filepath)
|
||||
return {'CANCELLED'}
|
||||
p = Path(patching_dir)
|
||||
count = 0
|
||||
for path in p.glob('*.py'):
|
||||
path = path.name
|
||||
text_abspath = os.path.join(patching_dir, path)
|
||||
text_file = open(text_abspath, 'r')
|
||||
if path in bpy.data.texts:
|
||||
text = bpy.data.texts[path]
|
||||
else:
|
||||
text = bpy.data.texts.new(path)
|
||||
text.from_string(text_file.read())
|
||||
text_file.close()
|
||||
count += 1
|
||||
if count == 1:
|
||||
self.report({'INFO'}, 'loaded 1 patch')
|
||||
else:
|
||||
self.report({'INFO'}, 'loaded %d patches' % count)
|
||||
return {'FINISHED'}
|
||||
|
||||
def save_func(self, context):
|
||||
self.layout.operator("file.hecl_patching_save", text="Save HECL Patches")
|
||||
|
||||
def load_func(self, context):
|
||||
self.layout.operator("file.hecl_patching_load", text="Load HECL Patches")
|
||||
|
||||
def register():
|
||||
bpy.utils.register_class(FILE_OT_hecl_patching_save)
|
||||
bpy.utils.register_class(FILE_OT_hecl_patching_load)
|
||||
bpy.types.INFO_MT_file_external_data.append(load_func)
|
||||
bpy.types.INFO_MT_file_external_data.append(save_func)
|
||||
|
||||
def unregister():
|
||||
bpy.utils.unregister_class(FILE_OT_hecl_patching_save)
|
||||
bpy.utils.unregister_class(FILE_OT_hecl_patching_load)
|
||||
bpy.types.INFO_MT_file_external_data.remove(load_func)
|
||||
bpy.types.INFO_MT_file_external_data.remove(save_func)
|
||||
|
|
@ -13,7 +13,7 @@ bl_info = {
|
|||
"category": "System"}
|
||||
|
||||
# Package import
|
||||
from . import hmdl, sact, srea, Nodegrid
|
||||
from . import hmdl, sact, srea, Nodegrid, Patching
|
||||
Nodegrid = Nodegrid.Nodegrid
|
||||
import bpy, os, sys
|
||||
from bpy.app.handlers import persistent
|
||||
|
|
@ -128,6 +128,7 @@ def register():
|
|||
bpy.utils.register_class(hecl_scene_panel)
|
||||
bpy.types.Scene.hecl_auto_select = bpy.props.BoolProperty(name='HECL Auto Select', default=True)
|
||||
bpy.app.handlers.load_post.append(scene_loaded)
|
||||
Patching.register()
|
||||
|
||||
def unregister():
|
||||
bpy.app.handlers.load_post.remove(scene_loaded)
|
||||
|
|
@ -135,6 +136,7 @@ def unregister():
|
|||
sact.unregister()
|
||||
srea.unregister()
|
||||
bpy.utils.unregister_class(hecl_scene_panel)
|
||||
Patching.unregister()
|
||||
|
||||
if __name__ == "__main__":
|
||||
register()
|
||||
|
|
|
|||
|
|
@ -242,14 +242,20 @@ def initialize_nodetree_cycles(mat, pixel_size):
|
|||
diffuse_image_node = nt.nodes.new('ShaderNodeTexImage')
|
||||
gridder.place_node(diffuse_image_node, 1)
|
||||
diffuse_image_node.image = tex_node.texture.image
|
||||
mixrgb_node = nt.nodes.new('ShaderNodeMixRGB')
|
||||
gridder.place_node(mixrgb_node, 1)
|
||||
mixrgb_node.inputs[1].default_value = (1.0,1.0,1.0,1.0)
|
||||
mapping = nt.nodes.new('ShaderNodeMapping')
|
||||
gridder.place_node(mapping, 1)
|
||||
mapping.vector_type = 'TEXTURE'
|
||||
mapping.translation = (1.0,1.0,0.0)
|
||||
mapping.scale = (2.0,2.0,1.0)
|
||||
nt.links.new(diffuse_image_node.outputs[0], diffuse.inputs[0])
|
||||
nt.links.new(diffuse_image_node.outputs[0], mixrgb_node.inputs[2])
|
||||
if nt.nodes['Output'].inputs[1].is_linked:
|
||||
nt.links.new(nt.nodes['Output'].inputs[1].links[0].from_socket, mix_shader.inputs[0])
|
||||
nt.links.new(nt.nodes['Output'].inputs[1].links[0].from_socket, mixrgb_node.inputs[0])
|
||||
nt.links.new(mixrgb_node.outputs[0], transp.inputs[0])
|
||||
nt.links.new(tex_node.inputs[0].links[0].from_socket, mapping.inputs[0])
|
||||
nt.links.new(mapping.outputs[0], diffuse_image_node.inputs[0])
|
||||
|
||||
|
|
|
|||
|
|
@ -132,6 +132,7 @@ while True:
|
|||
bpy.ops.wm.read_homefile()
|
||||
bpy.context.user_preferences.filepaths.save_version = 0
|
||||
if 'FINISHED' in bpy.ops.wm.save_as_mainfile(filepath=cmdargs[1]):
|
||||
bpy.ops.file.hecl_patching_load()
|
||||
writepipeline(b'FINISHED')
|
||||
else:
|
||||
writepipeline(b'CANCELLED')
|
||||
|
|
|
|||
Loading…
Reference in New Issue