129 lines
3.9 KiB
Python
129 lines
3.9 KiB
Python
import bpy
|
|
from .operators import (AddPlaneCutter, AddCylinderCutter, AddCubeCutter, AddSphereCutter, CombineLabels,
|
|
ConvertToMesh, InvertScalar, FillByLabel, FillByThreshold, FillByRange)
|
|
from .io import ExportVolumeData, ImportAsLabelLayer, ImportAsScalarLayer, ImportVolumeData, AddVolumeData
|
|
from .misc import SaveLayers
|
|
|
|
|
|
class ModifyLayer(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_MODIFY_LAYERS"
|
|
bl_label = "Modify Layer"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(InvertScalar.bl_idname)
|
|
layout.operator(FillByThreshold.bl_idname)
|
|
layout.operator(FillByRange.bl_idname)
|
|
layout.operator(FillByLabel.bl_idname)
|
|
layout.operator(CombineLabels.bl_idname)
|
|
|
|
|
|
class AddCutterMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_CUTTERS"
|
|
bl_label = "Add Cutter"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(AddPlaneCutter.bl_idname)
|
|
layout.operator(AddCylinderCutter.bl_idname)
|
|
layout.operator(AddCubeCutter.bl_idname)
|
|
layout.operator(AddSphereCutter.bl_idname)
|
|
|
|
|
|
class ImportLayerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_LAYERS"
|
|
bl_label = "Import Layer"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator(ImportAsScalarLayer.bl_idname)
|
|
layout.operator(ImportAsLabelLayer.bl_idname)
|
|
|
|
|
|
class BioxelNodesView3DMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_VIEW3D"
|
|
bl_label = "Bioxel Nodes"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.menu(ImportLayerMenu.bl_idname)
|
|
layout.menu(AddCutterMenu.bl_idname)
|
|
layout.operator(ConvertToMesh.bl_idname)
|
|
layout.separator()
|
|
layout.operator(SaveLayers.bl_idname)
|
|
|
|
|
|
class BioxelNodesOutlinerMenu(bpy.types.Menu):
|
|
bl_idname = "BIOXELNODES_MT_OUTLINER"
|
|
bl_label = "Bioxel Nodes"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.menu(ImportLayerMenu.bl_idname)
|
|
layout.menu(AddCutterMenu.bl_idname)
|
|
layout.operator(ConvertToMesh.bl_idname)
|
|
layout.separator()
|
|
layout.menu(ModifyLayer.bl_idname)
|
|
layout.separator()
|
|
layout.operator(SaveLayers.bl_idname)
|
|
|
|
|
|
# def TOPBAR_FILE_IMPORT(self, context):
|
|
# layout = self.layout
|
|
# layout.separator()
|
|
# layout.operator(ImportVolumeData.bl_idname)
|
|
|
|
|
|
# def TOPBAR_FILE_EXPORT(self, context):
|
|
# layout = self.layout
|
|
# layout.separator()
|
|
# layout.operator(ExportVolumeData.bl_idname)
|
|
|
|
|
|
def VIEW3D_OBJECT(self, context):
|
|
layout = self.layout
|
|
layout.menu(BioxelNodesView3DMenu.bl_idname)
|
|
layout.separator()
|
|
|
|
|
|
def OUTLINER_OBJECT(self, context):
|
|
layout = self.layout
|
|
layout.menu(BioxelNodesOutlinerMenu.bl_idname)
|
|
layout.separator()
|
|
|
|
|
|
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)
|
|
layout.operator(ExportVolumeData.bl_idname)
|
|
layout.separator()
|
|
layout.menu(AddCutterMenu.bl_idname)
|
|
layout.operator(ConvertToMesh.bl_idname)
|
|
layout.separator()
|
|
layout.operator(SaveLayers.bl_idname)
|
|
|
|
|
|
def TOPBAR(self, context):
|
|
layout = self.layout
|
|
layout.menu(BioxelNodesTopbarMenu.bl_idname)
|
|
|
|
|
|
def add():
|
|
# bpy.types.TOPBAR_MT_file_import.append(TOPBAR_FILE_IMPORT)
|
|
# bpy.types.TOPBAR_MT_file_export.append(TOPBAR_FILE_EXPORT)
|
|
bpy.types.OUTLINER_MT_object.prepend(OUTLINER_OBJECT)
|
|
bpy.types.VIEW3D_MT_object_context_menu.prepend(VIEW3D_OBJECT)
|
|
bpy.types.TOPBAR_MT_editor_menus.append(TOPBAR)
|
|
|
|
|
|
def remove():
|
|
# bpy.types.TOPBAR_MT_file_import.remove(TOPBAR_FILE_IMPORT)
|
|
# bpy.types.TOPBAR_MT_file_export.remove(TOPBAR_FILE_EXPORT)
|
|
bpy.types.OUTLINER_MT_object.remove(OUTLINER_OBJECT)
|
|
bpy.types.VIEW3D_MT_object_context_menu.remove(VIEW3D_OBJECT)
|
|
bpy.types.TOPBAR_MT_editor_menus.remove(TOPBAR)
|