162 lines
4.7 KiB
Python
162 lines
4.7 KiB
Python
|
|
from pathlib import Path
|
|
from .customnodes import CustomNodes
|
|
import bpy
|
|
|
|
|
|
def add_driver_to_node_factory(source_prop, target_prop):
|
|
callback_str = f"""
|
|
import bpy
|
|
from bioxelnodes.utils import add_direct_driver, get_bioxels_obj
|
|
bioxels_obj = get_bioxels_obj(bpy.context.active_object)
|
|
if bioxels_obj:
|
|
container_obj = bioxels_obj.parent
|
|
add_direct_driver(
|
|
target=node,
|
|
target_prop='{target_prop}',
|
|
source=container_obj,
|
|
source_prop='{source_prop}'
|
|
)
|
|
else:
|
|
print('Cannot find any bioxels.')
|
|
"""
|
|
return callback_str
|
|
|
|
|
|
def set_prop_to_node_factory(source_prop, target_prop):
|
|
callback_str = f"""
|
|
import bpy
|
|
from bioxelnodes.utils import get_bioxels_obj
|
|
bioxels_obj = get_bioxels_obj(bpy.context.active_object)
|
|
if bioxels_obj:
|
|
container_obj = bioxels_obj.parent
|
|
node.inputs.get('{target_prop}').default_value = container_obj.get(
|
|
'{source_prop}')
|
|
else:
|
|
print('Cannot find any bioxels.')
|
|
"""
|
|
return callback_str
|
|
|
|
|
|
MENU_ITEMS = [
|
|
{
|
|
'label': 'Presets',
|
|
'icon': 'MESH_MONKEY',
|
|
'items': [
|
|
{
|
|
'label': 'Segment',
|
|
'icon': 'BONE_DATA',
|
|
'node_type': 'BioxelNodes_Segment',
|
|
'node_description': '',
|
|
'node_callback': "\n".join([
|
|
set_prop_to_node_factory("bioxel_size", "Detail Size"),
|
|
set_prop_to_node_factory("bioxels_offset", "Offset"),
|
|
])
|
|
}
|
|
# {
|
|
# 'label': 'Anatomy',
|
|
# 'icon': 'CONSTRAINT_BONE',
|
|
# 'node_type': 'BioxelNodes_Anatomy',
|
|
# 'node_description': '',
|
|
# 'node_callback': "\n".join([
|
|
# set_prop_to_node_factory("bioxel_size", "Detail Size"),
|
|
# set_prop_to_node_factory("bioxels_offset", "Offset"),
|
|
# ])
|
|
# }
|
|
]
|
|
},
|
|
{
|
|
'label': 'Segment',
|
|
'icon': 'OUTLINER_DATA_VOLUME',
|
|
'items': [
|
|
{
|
|
'label': 'Segment by Level',
|
|
'icon': 'EMPTY_SINGLE_ARROW',
|
|
'node_type': 'BioxelNodes_SegmentByLevel',
|
|
'node_description': '',
|
|
'node_callback': "\n".join([
|
|
set_prop_to_node_factory("bioxel_size", "Detail Size"),
|
|
set_prop_to_node_factory("bioxels_offset", "Offset"),
|
|
])
|
|
},
|
|
{
|
|
'label': 'Segment by Range',
|
|
'icon': 'IPO_CONSTANT',
|
|
'node_type': 'BioxelNodes_SegmentByRange',
|
|
'node_description': '',
|
|
'node_callback': "\n".join([
|
|
set_prop_to_node_factory("bioxel_size", "Detail Size"),
|
|
set_prop_to_node_factory("bioxels_offset", "Offset"),
|
|
])
|
|
}
|
|
# {
|
|
# 'label': 'Segment by Layers',
|
|
# 'icon': 'ALIGN_JUSTIFY',
|
|
# 'node_type': 'BioxelNodes_SegmentsByLayers',
|
|
# 'node_description': '',
|
|
# 'node_callback': "\n".join([
|
|
# set_prop_to_node_factory("bioxel_size", "Detail Size"),
|
|
# set_prop_to_node_factory("bioxels_offset", "Offset"),
|
|
# ])
|
|
# }
|
|
]
|
|
},
|
|
{
|
|
'label': 'Shader',
|
|
'icon': 'SHADING_RENDERED',
|
|
'items': [
|
|
{
|
|
'label': 'Segment Shader',
|
|
'icon': 'OUTLINER_OB_VOLUME',
|
|
'node_type': 'BioxelNodes_SegmentShader',
|
|
'node_description': ''
|
|
}
|
|
]
|
|
}
|
|
]
|
|
|
|
NODE_FILE = "BioxelNodes_4.0"
|
|
|
|
if bpy.app.version >= (4, 1, 0):
|
|
MENU_ITEMS.append({
|
|
'label': 'Slicer',
|
|
'icon': 'CLIPUV_DEHLT',
|
|
'items': [
|
|
{
|
|
'label': 'RAS Slicer',
|
|
'icon': 'EMPTY_DATA',
|
|
'node_type': 'BioxelNodes_RAS-Slicer',
|
|
'node_description': ''
|
|
},
|
|
{
|
|
'label': 'Axis Slicer',
|
|
'icon': 'OUTLINER_OB_EMPTY',
|
|
'node_type': 'BioxelNodes_AxisSlicer',
|
|
'node_description': '',
|
|
},
|
|
{
|
|
'label': 'Box Slicer',
|
|
'icon': 'MOD_WIREFRAME',
|
|
'node_type': 'BioxelNodes_BoxSlicer',
|
|
'node_description': '',
|
|
}
|
|
]
|
|
})
|
|
NODE_FILE = "BioxelNodes_4.1"
|
|
|
|
|
|
custom_nodes = CustomNodes(
|
|
menu_items=MENU_ITEMS,
|
|
nodes_file=Path(__file__).parent / f"assets/Nodes/{NODE_FILE}.blend",
|
|
root_label='Bioxel Nodes',
|
|
root_icon="FILE_VOLUME",
|
|
)
|
|
|
|
|
|
def register():
|
|
custom_nodes.register()
|
|
|
|
|
|
def unregister():
|
|
custom_nodes.unregister()
|