382 lines
13 KiB
Python
382 lines
13 KiB
Python
import bpy
|
|
|
|
from .constants import MENU_ITEMS, NODE_LIB_FILENAME
|
|
from .node_menu import NodeMenu
|
|
|
|
from .bioxelutils.common import (get_container_obj,
|
|
get_container_layer_objs, get_layer_label,
|
|
get_layer_prop_value)
|
|
from .operators.layer import (FetchLayer, RelocateLayer, RetimeLayer, RenameLayer,
|
|
RemoveSelectedLayers, SaveSelectedLayersCache,
|
|
ResampleLayer, SignScalar, CombineLabels,
|
|
FillByLabel, FillByThreshold, FillByRange)
|
|
from .operators.container import (AddLocator, AddSlicer, ExtractShapeWire,
|
|
SaveContainerLayersCache, RemoveContainerMissingLayers,
|
|
SaveContainer, LoadContainer,
|
|
AddPieCutter, AddPlaneCutter,
|
|
AddCylinderCutter, AddCubeCutter, AddSphereCutter,
|
|
ExtractBboxWire, ExtractSurface, ExtractVolume, ContainerProps)
|
|
|
|
from .operators.io import (ImportAsLabel, ImportAsScalar, ImportAsColor)
|
|
from .operators.misc import (CleanTemp,
|
|
ReLinkNodeLib, RemoveAllMissingLayers, RenderSettingPreset, SaveAllLayersCache, SaveNodeLib, SliceViewer)
|
|
|
|
|
|
class FetchLayerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_ADD_LAYER"
|
|
bl_label = "Fetch Layer"
|
|
|
|
def draw(self, context):
|
|
container_obj = get_container_obj(bpy.context.active_object)
|
|
layer_objs = get_container_layer_objs(container_obj)
|
|
layout = self.layout
|
|
|
|
for layer_obj in layer_objs:
|
|
op = layout.operator(FetchLayer.bl_idname,
|
|
text=get_layer_label(layer_obj))
|
|
op.layer_obj_name = layer_obj.name
|
|
|
|
|
|
class ExtractFromContainerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_PICK"
|
|
bl_label = "Extract from Container"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(ExtractSurface.bl_idname)
|
|
layout.operator(ExtractVolume.bl_idname)
|
|
layout.operator(ExtractShapeWire.bl_idname)
|
|
layout.operator(ExtractBboxWire.bl_idname)
|
|
|
|
|
|
class AddCutterMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_CUTTERS"
|
|
bl_label = "Add a Cutter"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(AddPlaneCutter.bl_idname,
|
|
icon=AddPlaneCutter.bl_icon, text="Plane Cutter")
|
|
layout.operator(AddCylinderCutter.bl_idname,
|
|
icon=AddCylinderCutter.bl_icon, text="Cylinder Cutter")
|
|
layout.operator(AddCubeCutter.bl_idname,
|
|
icon=AddCubeCutter.bl_icon, text="Cube Cutter")
|
|
layout.operator(AddSphereCutter.bl_idname,
|
|
icon=AddSphereCutter.bl_icon, text="Sphere Cutter")
|
|
layout.operator(AddPieCutter.bl_idname,
|
|
icon=AddPieCutter.bl_icon, text="Pie Cutter")
|
|
|
|
|
|
class ImportLayerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_IMPORTLAYER"
|
|
bl_label = "Import Volumetric Data (Init)"
|
|
bl_icon = "FILE_NEW"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(ImportAsScalar.bl_idname,
|
|
text="as Scalar")
|
|
layout.operator(ImportAsLabel.bl_idname,
|
|
text="as Label")
|
|
layout.operator(ImportAsColor.bl_idname,
|
|
text="as Color")
|
|
|
|
|
|
class AddLayerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_ADDLAYER"
|
|
bl_label = "Import Volumetric Data (Add)"
|
|
bl_icon = "FILE_NEW"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(ImportAsScalar.bl_idname,
|
|
text="as Scalar")
|
|
layout.operator(ImportAsLabel.bl_idname,
|
|
text="as Label")
|
|
layout.operator(ImportAsColor.bl_idname,
|
|
text="as Color")
|
|
|
|
|
|
class ModifyLayerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_MODIFYLAYER"
|
|
bl_label = "Modify Layer"
|
|
bl_icon = "FILE_NEW"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(SignScalar.bl_idname,
|
|
icon=SignScalar.bl_icon)
|
|
layout.operator(FillByThreshold.bl_idname,
|
|
icon=FillByThreshold.bl_icon)
|
|
layout.operator(FillByRange.bl_idname,
|
|
icon=FillByRange.bl_icon)
|
|
layout.operator(FillByLabel.bl_idname,
|
|
icon=FillByLabel.bl_icon)
|
|
layout.operator(CombineLabels.bl_idname,
|
|
icon=CombineLabels.bl_icon)
|
|
|
|
|
|
class ReLinkNodeLibMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_RELINK"
|
|
bl_label = "Relink Node Library"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
versions = [{"label": "v0.3.x", "filename": "BioxelNodes_v0.3.x"},
|
|
{"label": "v0.2.x", "filename": "BioxelNodes_v0.2.x"},
|
|
{"label": "v0.1.x", "filename": "BioxelNodes_v0.1.x"}]
|
|
|
|
op = layout.operator(ReLinkNodeLib.bl_idname,
|
|
text="Current Version")
|
|
op.node_lib_filename = NODE_LIB_FILENAME
|
|
layout.separator()
|
|
for version in versions:
|
|
op = layout.operator(ReLinkNodeLib.bl_idname,
|
|
text=version["label"])
|
|
op.node_lib_filename = version["filename"]
|
|
|
|
|
|
class RenderSettingMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_RENDER"
|
|
bl_label = "Render Setting Presets"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
for k, v in RenderSettingPreset.PRESETS.items():
|
|
op = layout.operator(RenderSettingPreset.bl_idname,
|
|
text=v)
|
|
op.preset = k
|
|
|
|
|
|
class DangerZoneMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_DANGER"
|
|
bl_label = "Danger Zone"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(CleanTemp.bl_idname)
|
|
|
|
|
|
class BioxelNodesTopbarMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_TOPBAR"
|
|
bl_label = "Bioxel Nodes"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
layout.menu(ImportLayerMenu.bl_idname,
|
|
icon=ImportLayerMenu.bl_icon)
|
|
layout.operator(LoadContainer.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(SaveAllLayersCache.bl_idname)
|
|
layout.operator(RemoveAllMissingLayers.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(SaveNodeLib.bl_idname)
|
|
layout.menu(ReLinkNodeLibMenu.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.menu(DangerZoneMenu.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.menu(RenderSettingMenu.bl_idname)
|
|
|
|
|
|
def TOPBAR(self, context):
|
|
layout = self.layout
|
|
layout.menu(BioxelNodesTopbarMenu.bl_idname)
|
|
|
|
|
|
class NodeHeadMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_NODE_HEAD"
|
|
bl_label = "Bioxel Nodes"
|
|
bl_icon = "FILE_VOLUME"
|
|
|
|
def draw(self, context):
|
|
|
|
layout = self.layout
|
|
layout.menu(AddLayerMenu.bl_idname,
|
|
icon=AddLayerMenu.bl_icon)
|
|
layout.operator(SaveContainer.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(ContainerProps.bl_idname)
|
|
layout.menu(ExtractFromContainerMenu.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(SaveContainerLayersCache.bl_idname,
|
|
icon=SaveContainerLayersCache.bl_icon)
|
|
layout.operator(RemoveContainerMissingLayers.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(AddLocator.bl_idname,
|
|
icon=AddLocator.bl_icon)
|
|
layout.operator(AddSlicer.bl_idname,
|
|
icon=AddSlicer.bl_icon)
|
|
layout.menu(AddCutterMenu.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.menu(FetchLayerMenu.bl_idname)
|
|
|
|
|
|
class NodeContextMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_NODE_CONTEXT"
|
|
bl_label = "Bioxel Nodes"
|
|
bl_icon = "FILE_VOLUME"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(SaveSelectedLayersCache.bl_idname,
|
|
icon=SaveSelectedLayersCache.bl_icon)
|
|
layout.operator(RemoveSelectedLayers.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(RenameLayer.bl_idname,
|
|
icon=RenameLayer.bl_icon)
|
|
layout.operator(RetimeLayer.bl_idname)
|
|
layout.operator(RelocateLayer.bl_idname)
|
|
|
|
layout.separator()
|
|
layout.operator(ResampleLayer.bl_idname,
|
|
icon=ResampleLayer.bl_icon)
|
|
layout.operator(SignScalar.bl_idname)
|
|
layout.operator(FillByThreshold.bl_idname)
|
|
layout.operator(FillByRange.bl_idname)
|
|
layout.operator(FillByLabel.bl_idname)
|
|
layout.operator(CombineLabels.bl_idname)
|
|
|
|
|
|
def NODE_CONTEXT(self, context):
|
|
container_obj = context.object
|
|
is_geo_nodes = context.area.ui_type == "GeometryNodeTree"
|
|
is_container = get_container_obj(container_obj)
|
|
|
|
if not is_geo_nodes or not is_container:
|
|
return
|
|
|
|
layout = self.layout
|
|
layout.separator()
|
|
layout.menu(NodeContextMenu.bl_idname)
|
|
|
|
|
|
def NODE_HEAD(self, context):
|
|
container_obj = context.object
|
|
is_geo_nodes = context.area.ui_type == "GeometryNodeTree"
|
|
is_container = get_container_obj(container_obj)
|
|
|
|
if not is_geo_nodes or not is_container:
|
|
return
|
|
|
|
layout = self.layout
|
|
layout.separator()
|
|
layout.menu(NodeHeadMenu.bl_idname)
|
|
|
|
|
|
def NODE_PROP(self, context):
|
|
container_obj = context.object
|
|
is_geo_nodes = context.area.ui_type == "GeometryNodeTree"
|
|
is_container = get_container_obj(container_obj)
|
|
self.bl_label = "Group"
|
|
|
|
if not is_geo_nodes or not is_container:
|
|
return
|
|
|
|
if container_obj.modifiers[0].node_group != context.space_data.edit_tree:
|
|
return
|
|
|
|
self.bl_label = "Bioxel Nodes"
|
|
|
|
layer_list_UL = bpy.context.window_manager.bioxelnodes_layer_list_UL
|
|
layer_list = layer_list_UL.layer_list
|
|
layer_list.clear()
|
|
|
|
for layer_obj in get_container_layer_objs(container_obj):
|
|
layer_item = layer_list.add()
|
|
layer_item.label = get_layer_label(layer_obj)
|
|
layer_item.obj_name = layer_obj.name
|
|
layer_item.info_text = "\n".join([f"{prop}: {get_layer_prop_value(layer_obj, prop)}"
|
|
for prop in ["kind",
|
|
"bioxel_size",
|
|
"shape",
|
|
"frame_count",
|
|
"channel_count",
|
|
"min", "max"]])
|
|
|
|
layout = self.layout
|
|
layout.label(text="Layer List")
|
|
split = layout.row()
|
|
split.template_list(listtype_name="BIOXELNODES_UL_layer_list",
|
|
list_id="layer_list",
|
|
dataptr=layer_list_UL,
|
|
propname="layer_list",
|
|
active_dataptr=layer_list_UL,
|
|
active_propname="layer_list_active",
|
|
item_dyntip_propname="info_text",
|
|
rows=20)
|
|
|
|
# sidebar = split.column(align=True)
|
|
# sidebar.menu(AddLayerMenu.bl_idname,
|
|
# icon=AddLayerMenu.bl_icon, text="")
|
|
|
|
# sidebar.operator(SaveContainerLayersCache.bl_idname,
|
|
# icon=SaveContainerLayersCache.bl_icon, text="")
|
|
# sidebar.operator(RemoveContainerMissingLayers.bl_idname,
|
|
# icon=RemoveContainerMissingLayers.bl_icon, text="")
|
|
|
|
# sidebar.separator()
|
|
# sidebar.operator(SaveSelectedLayersCache.bl_idname,
|
|
# icon=SaveSelectedLayersCache.bl_icon, text="")
|
|
# sidebar.operator(RemoveSelectedLayers.bl_idname,
|
|
# icon=RemoveSelectedLayers.bl_icon, text="")
|
|
|
|
# sidebar.separator()
|
|
# sidebar.operator(RenameLayer.bl_idname,
|
|
# icon=RenameLayer.bl_icon, text="")
|
|
# sidebar.operator(ResampleLayer.bl_idname,
|
|
# icon=ResampleLayer.bl_icon, text="")
|
|
# sidebar.operator(RetimeLayer.bl_idname,
|
|
# icon=RetimeLayer.bl_icon, text="")
|
|
|
|
# sidebar.separator()
|
|
# sidebar.operator(SignScalar.bl_idname,
|
|
# icon=SignScalar.bl_icon, text="")
|
|
# sidebar.operator(FillByThreshold.bl_idname,
|
|
# icon=FillByThreshold.bl_icon, text="")
|
|
# sidebar.operator(FillByRange.bl_idname,
|
|
# icon=FillByRange.bl_icon, text="")
|
|
# sidebar.operator(FillByLabel.bl_idname,
|
|
# icon=FillByLabel.bl_icon, text="")
|
|
|
|
# sidebar.separator()
|
|
# layout.separator()
|
|
|
|
|
|
def VIEW3D_TOPBAR(self, context):
|
|
layout = self.layout
|
|
layout.operator(SliceViewer.bl_idname,
|
|
icon=SliceViewer.bl_icon, text="")
|
|
|
|
|
|
node_menu = NodeMenu(
|
|
menu_items=MENU_ITEMS
|
|
)
|
|
|
|
|
|
def add():
|
|
node_menu.register()
|
|
bpy.types.VIEW3D_HT_header.append(VIEW3D_TOPBAR)
|
|
bpy.types.TOPBAR_MT_editor_menus.append(TOPBAR)
|
|
bpy.types.NODE_PT_node_tree_properties.prepend(NODE_PROP)
|
|
bpy.types.NODE_MT_editor_menus.append(NODE_HEAD)
|
|
bpy.types.NODE_MT_context_menu.append(NODE_CONTEXT)
|
|
|
|
|
|
def remove():
|
|
bpy.types.VIEW3D_HT_header.remove(VIEW3D_TOPBAR)
|
|
bpy.types.TOPBAR_MT_editor_menus.remove(TOPBAR)
|
|
bpy.types.NODE_PT_node_tree_properties.remove(NODE_PROP)
|
|
bpy.types.NODE_MT_editor_menus.remove(NODE_HEAD)
|
|
bpy.types.NODE_MT_context_menu.remove(NODE_CONTEXT)
|
|
node_menu.unregister()
|