feat: redesign nodes
This commit is contained in:
@@ -1,97 +1,86 @@
|
||||
import bpy
|
||||
from pathlib import Path
|
||||
import shutil
|
||||
from bioxelnodes.utils import get_all_bioxels_objs, get_bioxels_obj
|
||||
from bioxelnodes.utils import get_all_layers, get_container, get_container_layers
|
||||
|
||||
|
||||
class SaveBioxels(bpy.types.Operator):
|
||||
bl_idname = "bioxelnodes.save_bioxels"
|
||||
bl_label = "Save Bioxels"
|
||||
bl_description = "Save Bioxels to Directory."
|
||||
def save_layer(layer, output_dir):
|
||||
name = layer.name
|
||||
|
||||
# "//"
|
||||
output_dir = bpy.path.abspath(output_dir)
|
||||
source_dir = bpy.path.abspath(layer.data.filepath)
|
||||
|
||||
output_path: Path = Path(output_dir, f"{name}.vdb").resolve()
|
||||
source_path: Path = Path(source_dir).resolve()
|
||||
|
||||
if output_path != source_path:
|
||||
shutil.copy(source_path, output_path)
|
||||
|
||||
blend_path = Path(bpy.path.abspath("//")).resolve()
|
||||
layer.data.filepath = bpy.path.relpath(
|
||||
str(output_path), start=str(blend_path))
|
||||
|
||||
|
||||
class SaveBioxelLayers(bpy.types.Operator):
|
||||
bl_idname = "bioxelnodes.save_bioxel_layers"
|
||||
bl_label = "Save Bioxel Layers"
|
||||
bl_description = "Save Bioxel Layers to Directory."
|
||||
bl_options = {'UNDO'}
|
||||
|
||||
@classmethod
|
||||
def poll(cls, context):
|
||||
bioxels_obj = None
|
||||
for obj in bpy.context.selected_objects:
|
||||
bioxels_obj = get_bioxels_obj(obj)
|
||||
break
|
||||
|
||||
return True if bioxels_obj else False
|
||||
container = get_container(bpy.context.active_object)
|
||||
return True if container else False
|
||||
|
||||
def execute(self, context):
|
||||
bioxels_objs = []
|
||||
container = get_container(bpy.context.active_object)
|
||||
|
||||
for obj in bpy.context.selected_objects:
|
||||
bioxels_obj = get_bioxels_obj(obj)
|
||||
if bioxels_obj:
|
||||
bioxels_objs.append(bioxels_obj)
|
||||
|
||||
if len(bioxels_objs) == 0:
|
||||
self.report({"WARNING"}, "Cannot find any bioxels.")
|
||||
if not container:
|
||||
self.report({"WARNING"}, "Cannot find any bioxel container.")
|
||||
return {'FINISHED'}
|
||||
|
||||
for bioxels_obj in bioxels_objs:
|
||||
name = bioxels_obj.parent.name
|
||||
for layer in get_container_layers(container):
|
||||
try:
|
||||
save_layer(layer, context.scene.bioxel_layer_dir)
|
||||
except:
|
||||
self.report(
|
||||
{"WARNING"}, f"Fail to save {layer.name}, skiped")
|
||||
|
||||
# "//"
|
||||
bioxels_dir = bpy.path.abspath(context.scene.bioxels_dir)
|
||||
source_dir = bpy.path.abspath(bioxels_obj.data.filepath)
|
||||
|
||||
output_path: Path = Path(bioxels_dir, f"{name}.vdb").resolve()
|
||||
source_path: Path = Path(source_dir).resolve()
|
||||
|
||||
if output_path != source_path:
|
||||
shutil.copy(source_path, output_path)
|
||||
|
||||
blend_path = Path(bpy.path.abspath("//")).resolve()
|
||||
bioxels_obj.data.filepath = bpy.path.relpath(
|
||||
str(output_path), start=str(blend_path))
|
||||
|
||||
self.report({"INFO"}, f"Successfully saved bioxels")
|
||||
self.report({"INFO"}, f"Successfully saved bioxel layers.")
|
||||
|
||||
return {'FINISHED'}
|
||||
|
||||
|
||||
class SaveAllBioxels(bpy.types.Operator):
|
||||
bl_idname = "bioxelnodes.save_all_bioxels"
|
||||
bl_label = "Save All Bioxels"
|
||||
bl_description = "Save All Bioxels to Directory."
|
||||
class SaveAllBioxelLayers(bpy.types.Operator):
|
||||
bl_idname = "bioxelnodes.save_all_bioxel_layers"
|
||||
bl_label = "Save All Bioxel Layers"
|
||||
bl_description = "Save All Bioxel Layers to Directory."
|
||||
bl_options = {'UNDO'}
|
||||
|
||||
def execute(self, context):
|
||||
|
||||
bioxels_objs = get_all_bioxels_objs()
|
||||
layers = get_all_layers()
|
||||
|
||||
if len(bioxels_objs) == 0:
|
||||
self.report({"WARNING"}, "Cannot find any bioxels.")
|
||||
if len(layers) == 0:
|
||||
self.report({"WARNING"}, "Cannot find any bioxel layer.")
|
||||
return {'FINISHED'}
|
||||
|
||||
for bioxels_obj in bioxels_objs:
|
||||
name = bioxels_obj.parent.name
|
||||
for layer in layers:
|
||||
try:
|
||||
save_layer(layer, context.scene.bioxel_layer_dir)
|
||||
except:
|
||||
self.report(
|
||||
{"WARNING"}, f"Fail to save {layer.name}, skiped")
|
||||
|
||||
# "//"
|
||||
bioxels_dir = bpy.path.abspath(context.scene.bioxels_dir)
|
||||
source_dir = bpy.path.abspath(bioxels_obj.data.filepath)
|
||||
|
||||
output_path: Path = Path(bioxels_dir, f"{name}.vdb").resolve()
|
||||
source_path: Path = Path(source_dir).resolve()
|
||||
|
||||
if output_path != source_path:
|
||||
shutil.copy(source_path, output_path)
|
||||
|
||||
blend_path = Path(bpy.path.abspath("//")).resolve()
|
||||
bioxels_obj.data.filepath = bpy.path.relpath(
|
||||
str(output_path), start=str(blend_path))
|
||||
|
||||
self.report({"INFO"}, f"Successfully saved bioxels")
|
||||
self.report({"INFO"}, f"Successfully saved bioxel layers.")
|
||||
|
||||
return {'FINISHED'}
|
||||
|
||||
|
||||
class BIOXELNODES_PT_Bioxels(bpy.types.Panel):
|
||||
bl_idname = "BIOXELNODES_PT_Bioxels"
|
||||
bl_label = "Bioxels"
|
||||
bl_label = "Bioxel Nodes"
|
||||
bl_space_type = "PROPERTIES"
|
||||
bl_region_type = "WINDOW"
|
||||
bl_context = "scene"
|
||||
@@ -99,5 +88,5 @@ class BIOXELNODES_PT_Bioxels(bpy.types.Panel):
|
||||
def draw(self, context):
|
||||
layout = self.layout
|
||||
scene = context.scene
|
||||
layout.prop(scene, 'bioxels_dir')
|
||||
layout.operator(SaveAllBioxels.bl_idname)
|
||||
layout.prop(scene, 'bioxel_layer_dir')
|
||||
layout.operator(SaveAllBioxelLayers.bl_idname)
|
||||
|
||||
Reference in New Issue
Block a user