Compare commits
43 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
508e6858d7 | ||
| 1c49fbb813 | |||
| 35e405acd8 | |||
|
|
68748f509a | ||
| 436296652b | |||
| c265777313 | |||
| e80a2b6683 | |||
| a6586b7304 | |||
|
|
59bf9bff2d | ||
|
|
d4bc355d94 | ||
| 4e9dc1ce67 | |||
| dc7d9382d6 | |||
|
|
66aafb41bd | ||
| 2c844c01ec | |||
| e5be3ad5db | |||
|
|
29b302531b | ||
| 3ca7c2ac5a | |||
|
|
32138c1da6 | ||
| 86fc18ab69 | |||
|
|
928b15c722 | ||
| 487cea4662 | |||
|
|
92f85ff1f3 | ||
| 63c593f09a | |||
|
|
3c5ff64afd | ||
| 0ebc6dafbd | |||
| c22ab2c85b | |||
| 133702619e | |||
|
|
e384798184 | ||
| 80da400891 | |||
|
|
48700b3001 | ||
| b865d9d78d | |||
| 20c6393344 | |||
|
|
7c71417e5d | ||
| 655b19bc7c | |||
| 051df670a5 | |||
| 9e885865f7 | |||
|
|
21e0b85662 | ||
| 38fb888f00 | |||
|
|
2429eef5e0 | ||
| 2563f398da | |||
| 0cad35e2da | |||
| 65edf40164 | |||
| c3f88b7846 |
@@ -1,31 +1,11 @@
|
|||||||
name: Upload Assets to Release
|
name: Release Version
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
branches:
|
tags:
|
||||||
- "release/*"
|
- 'v*'
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
# draft_release:
|
|
||||||
# name: Draft Release
|
|
||||||
# runs-on: ubuntu-latest
|
|
||||||
# outputs:
|
|
||||||
# upload_url: ${{ steps.draft_release.outputs.upload_url }}
|
|
||||||
# version: ${{ steps.set_env.outputs.version }}
|
|
||||||
# steps:
|
|
||||||
# - name: Set Version Env
|
|
||||||
# id: set_env
|
|
||||||
# run: |
|
|
||||||
# ref_name=${{ github.ref_name }}
|
|
||||||
# echo "version=${ref_name#release/}" >> "$GITHUB_OUTPUT"
|
|
||||||
# - name: Get release
|
|
||||||
# id: draft_release
|
|
||||||
# uses: cardinalby/git-get-release-action@1.2.4
|
|
||||||
# with:
|
|
||||||
# releaseName: Upload
|
|
||||||
# env:
|
|
||||||
# GITHUB_TOKEN: ${{ github.token }}
|
|
||||||
|
|
||||||
draft_release:
|
draft_release:
|
||||||
name: Draft Release
|
name: Draft Release
|
||||||
permissions:
|
permissions:
|
||||||
@@ -50,7 +30,7 @@ jobs:
|
|||||||
env:
|
env:
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|
||||||
build_blender_add-on:
|
build_extension:
|
||||||
name: Build Blender Extension
|
name: Build Blender Extension
|
||||||
needs: draft_release
|
needs: draft_release
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
@@ -66,12 +46,13 @@ jobs:
|
|||||||
uses: actions/setup-python@v4
|
uses: actions/setup-python@v4
|
||||||
with:
|
with:
|
||||||
python-version: 3.11
|
python-version: 3.11
|
||||||
- name: Build Add-on
|
- name: Build Extension
|
||||||
run: |
|
run: | # build.py need tomlkit to edit blender_manifest.toml
|
||||||
pip install tomlkit
|
pip install tomlkit
|
||||||
python build.py ${{ matrix.platform }}
|
python build.py ${{ matrix.platform }}
|
||||||
zip -r package.zip bioxelnodes
|
cd src/bioxelnodes
|
||||||
- name: Upload Release Asset
|
zip -r ../../package.zip .
|
||||||
|
- name: Upload Extension
|
||||||
id: upload-release-asset
|
id: upload-release-asset
|
||||||
uses: actions/upload-release-asset@v1
|
uses: actions/upload-release-asset@v1
|
||||||
env:
|
env:
|
||||||
2
.gitignore
vendored
@@ -133,4 +133,6 @@ blendcache_*
|
|||||||
|
|
||||||
.secrets
|
.secrets
|
||||||
|
|
||||||
|
.vdb
|
||||||
|
|
||||||
!scipy_ndimage/*/**
|
!scipy_ndimage/*/**
|
||||||
11
.vscode/BioxelNodes.code-workspace
vendored
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
{
|
||||||
|
"folders": [
|
||||||
|
{
|
||||||
|
"path": ".."
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"path": "../src/bioxelnodes"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"settings": {}
|
||||||
|
}
|
||||||
36
README.md
@@ -1,4 +1,4 @@
|
|||||||
[中文文档](https://uj6xfhbzp0.feishu.cn/wiki/Qx3VwHuNPimeI8kr6nDcvl1DnHf?from=from_copylink)
|
[中文文档](https://uj6xfhbzp0.feishu.cn/wiki/LPKEwjooSivxjskWHlCcQznjnNf?from=from_copylink)
|
||||||
|
|
||||||
# Bioxel Nodes
|
# Bioxel Nodes
|
||||||
|
|
||||||
@@ -26,18 +26,18 @@ Welcome to our [discord server](https://discord.gg/pYkNyq2TjE), if you have any
|
|||||||
|
|
||||||
## Support Multiple Formats
|
## Support Multiple Formats
|
||||||
|
|
||||||
| Format | EXT | Test |
|
| Format | EXT |
|
||||||
| ------ | ---------------------------------------- | ------- |
|
| ------ | ---------------------------------------- |
|
||||||
| DICOM | .dcm, .DCM, .DICOM, .ima, .IMA | ✅ pass |
|
| DICOM | .dcm, .DCM, .DICOM, .ima, .IMA |
|
||||||
| BMP | .bmp, .BMP | ✅ pass |
|
| BMP | .bmp, .BMP |
|
||||||
| JPEG | .jpg, .JPG, .jpeg, .JPEG | ✅ pass |
|
| JPEG | .jpg, .JPG, .jpeg, .JPEG |
|
||||||
| PNG | .png, .PNG | ✅ pass |
|
| PNG | .png, .PNG |
|
||||||
| TIFF | .tif, .TIF, .tiff, .TIFF | ✅ pass |
|
| TIFF | .tif, .TIF, .tiff, .TIFF |
|
||||||
| Nifti | .nia, .nii, .nii.gz, .hdr, .img, .img.gz | ✅ pass |
|
| Nifti | .nia, .nii, .nii.gz, .hdr, .img, .img.gz |
|
||||||
| Nrrd | .nrrd, .nhdr | ✅ pass |
|
| Nrrd | .nrrd, .nhdr |
|
||||||
| HDF5 | .hdf, .h4, .hdf4, .he2, .h5, .hdf5, .he5 | ✅ pass |
|
| HDF5 | .hdf, .h4, .hdf4, .he2, .h5, .hdf5, .he5 |
|
||||||
| OME | .ome.tiff, .ome.tif | ✅ pass |
|
| OME | .ome.tiff, .ome.tif |
|
||||||
| MRC | .mrc, .mrc.gz, .map, .map.gz | ✅ pass |
|
| MRC | .mrc, .mrc.gz, .map, .map.gz |
|
||||||
|
|
||||||
## Support 4D volumetric data
|
## Support 4D volumetric data
|
||||||
|
|
||||||
@@ -59,16 +59,6 @@ Welcome to our [discord server](https://discord.gg/pYkNyq2TjE), if you have any
|
|||||||
- Only works with Cycles CPU , Cycles GPU (OptiX), EEVEE
|
- Only works with Cycles CPU , Cycles GPU (OptiX), EEVEE
|
||||||
- Section surface cannot be generated when convert to mesh (will be supported soon)
|
- Section surface cannot be generated when convert to mesh (will be supported soon)
|
||||||
|
|
||||||
## Compatible to Newer Version
|
|
||||||
|
|
||||||
**v0.3.x is not compatible to v0.2.x, Updating this addon may break old files. Read the following carefully before upgradation**
|
|
||||||
|
|
||||||
Before upgradation, you need to ask yourself whether this project file will be modified again or not, if it's an archived project file, I would recommend that you run **Bioxel Nodes > Save Staged Data** to make the addon nodes permanent. In this way, there will be no potential problem with the nodes not functioning due to the addon update.
|
|
||||||
|
|
||||||
After the addon update, your old project files may not work either, this may be because you had executed **Save Staged Data**. If so, you need to execute **Bioxel Nodes > Relink Nodes to Addon** to relink them to make sure that the addon's new functionality and the addon nodes are synchronized.
|
|
||||||
|
|
||||||
Also, the older shaders are not based on OSL, so if you find that you can't render volumes, you need to turn on **Open Shading Language (OSL)** in the Render Settings.
|
|
||||||
|
|
||||||
## Roadmap
|
## Roadmap
|
||||||
|
|
||||||
- Better multi-format import experience
|
- Better multi-format import experience
|
||||||
|
|||||||
@@ -1,19 +0,0 @@
|
|||||||
import bpy
|
|
||||||
|
|
||||||
from . import auto_load
|
|
||||||
from . import menus
|
|
||||||
|
|
||||||
|
|
||||||
auto_load.init()
|
|
||||||
|
|
||||||
|
|
||||||
def register():
|
|
||||||
auto_load.register()
|
|
||||||
bpy.types.WindowManager.bioxelnodes_progress_factor = bpy.props.FloatProperty()
|
|
||||||
bpy.types.WindowManager.bioxelnodes_progress_text = bpy.props.StringProperty()
|
|
||||||
menus.add()
|
|
||||||
|
|
||||||
|
|
||||||
def unregister():
|
|
||||||
menus.remove()
|
|
||||||
auto_load.unregister()
|
|
||||||
@@ -1,26 +0,0 @@
|
|||||||
def move_node_to_node(node, target_node, offset=(0, 0)):
|
|
||||||
node.location.x = target_node.location.x + offset[0]
|
|
||||||
node.location.y = target_node.location.y + offset[1]
|
|
||||||
|
|
||||||
|
|
||||||
def move_node_between_nodes(node, target_nodes, offset=(0, 0)):
|
|
||||||
xs = []
|
|
||||||
ys = []
|
|
||||||
for target_node in target_nodes:
|
|
||||||
xs.append(target_node.location.x)
|
|
||||||
ys.append(target_node.location.y)
|
|
||||||
|
|
||||||
node.location.x = sum(xs) / len(xs) + offset[0]
|
|
||||||
node.location.y = sum(ys) / len(ys) + offset[1]
|
|
||||||
|
|
||||||
|
|
||||||
def get_node_type(node):
|
|
||||||
node_type = type(node).__name__
|
|
||||||
if node_type == "GeometryNodeGroup":
|
|
||||||
node_type = node.node_tree.name
|
|
||||||
|
|
||||||
return node_type
|
|
||||||
|
|
||||||
|
|
||||||
def get_nodes_by_type(node_group, type_name: str):
|
|
||||||
return [node for node in node_group.nodes if get_node_type(node) == type_name]
|
|
||||||
@@ -1,38 +0,0 @@
|
|||||||
schema_version = "1.0.0"
|
|
||||||
|
|
||||||
id = "bioxelnodes"
|
|
||||||
version = "0.3.1"
|
|
||||||
name = "Bioxel Nodes"
|
|
||||||
tagline = "For scientific volumetric data visualization in Blender"
|
|
||||||
maintainer = "Ma Nan <icrdr2010@outlook.com>"
|
|
||||||
type = "add-on"
|
|
||||||
website = "https://omoolab.github.io/BioxelNodes/latest"
|
|
||||||
|
|
||||||
tags = ["Geometry Nodes", "Render", "Import-Export"]
|
|
||||||
blender_version_min = "4.2.0"
|
|
||||||
license = ["SPDX:MIT"]
|
|
||||||
copyright = ["2024 OmooLab"]
|
|
||||||
platforms = ["windows-x64", "linux-x64", "macos-arm64", "macos-x64"]
|
|
||||||
|
|
||||||
wheels = [
|
|
||||||
"./wheels/SimpleITK-2.3.1-cp311-cp311-win_amd64.whl",
|
|
||||||
"./wheels/SimpleITK-2.3.1-cp311-cp311-macosx_10_9_x86_64.whl",
|
|
||||||
"./wheels/SimpleITK-2.3.1-cp311-cp311-macosx_11_0_arm64.whl",
|
|
||||||
"./wheels/SimpleITK-2.3.1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl",
|
|
||||||
"./wheels/lxml-5.2.2-cp311-cp311-win_amd64.whl",
|
|
||||||
"./wheels/lxml-5.2.2-cp311-cp311-macosx_10_9_arm64.whl",
|
|
||||||
"./wheels/lxml-5.2.2-cp311-cp311-macosx_10_9_x86_64.whl",
|
|
||||||
"./wheels/lxml-5.2.2-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl",
|
|
||||||
"./wheels/h5py-3.11.0-cp311-cp311-win_amd64.whl",
|
|
||||||
"./wheels/h5py-3.11.0-cp311-cp311-macosx_11_0_arm64.whl",
|
|
||||||
"./wheels/h5py-3.11.0-cp311-cp311-macosx_10_9_x86_64.whl",
|
|
||||||
"./wheels/h5py-3.11.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl",
|
|
||||||
"./wheels/tifffile-2024.7.24-py3-none-any.whl",
|
|
||||||
"./wheels/pyometiff-1.0.0-py3-none-any.whl",
|
|
||||||
"./wheels/mrcfile-1.5.1-py2.py3-none-any.whl",
|
|
||||||
"./wheels/transforms3d-0.4.2-py3-none-any.whl",
|
|
||||||
]
|
|
||||||
|
|
||||||
[permissions]
|
|
||||||
files = "Import/export volume data from/to disk"
|
|
||||||
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
from .menus import CustomNodes
|
|
||||||
@@ -1,147 +0,0 @@
|
|||||||
import bpy
|
|
||||||
from pathlib import Path
|
|
||||||
from .nodes import AddCustomNode
|
|
||||||
|
|
||||||
|
|
||||||
class CustomNodes():
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
menu_items,
|
|
||||||
nodes_file,
|
|
||||||
class_prefix="CUSTOMNODES_MT_NODES",
|
|
||||||
root_label='CustomNodes',
|
|
||||||
root_icon='NONE'
|
|
||||||
) -> None:
|
|
||||||
if not Path(nodes_file).is_file():
|
|
||||||
raise FileNotFoundError(str(Path(nodes_file).resolve()))
|
|
||||||
|
|
||||||
self.menu_items = menu_items
|
|
||||||
self.nodes_file = str(Path(nodes_file).resolve())
|
|
||||||
self.class_prefix = class_prefix
|
|
||||||
self.root_label = root_label
|
|
||||||
self.root_icon = root_icon
|
|
||||||
self.class_prefix = class_prefix
|
|
||||||
|
|
||||||
menu_classes = []
|
|
||||||
self._create_menu_class(
|
|
||||||
items=menu_items,
|
|
||||||
label=root_label,
|
|
||||||
menu_classes=menu_classes
|
|
||||||
)
|
|
||||||
self.menu_classes = menu_classes
|
|
||||||
|
|
||||||
idname = f"{class_prefix}_{root_label.replace(' ', '').upper()}"
|
|
||||||
|
|
||||||
def add_node_menu(self, context):
|
|
||||||
if ('GeometryNodeTree' == bpy.context.area.spaces[0].tree_type):
|
|
||||||
layout = self.layout
|
|
||||||
layout.separator()
|
|
||||||
layout.menu(idname, icon=root_icon)
|
|
||||||
|
|
||||||
self.add_node_menu = add_node_menu
|
|
||||||
|
|
||||||
def _create_menu_class(self, menu_classes, items, label='CustomNodes', icon=0, idname_namespace=None):
|
|
||||||
nodes_file = self.nodes_file
|
|
||||||
idname_namespace = idname_namespace or self.class_prefix
|
|
||||||
idname = f"{idname_namespace}_{label.replace(' ', '').upper()}"
|
|
||||||
|
|
||||||
# create submenu class if item is menu.
|
|
||||||
for item in items:
|
|
||||||
item_items = item.get('items') if item != 'separator' else None
|
|
||||||
if item_items:
|
|
||||||
menu_class = self._create_menu_class(
|
|
||||||
menu_classes=menu_classes,
|
|
||||||
items=item_items,
|
|
||||||
label=item.get('label') or 'CustomNodes',
|
|
||||||
icon=item.get('icon') or 0,
|
|
||||||
idname_namespace=idname
|
|
||||||
)
|
|
||||||
item['menu_class'] = menu_class
|
|
||||||
|
|
||||||
# create menu class
|
|
||||||
class Menu(bpy.types.Menu):
|
|
||||||
bl_idname = idname
|
|
||||||
bl_label = label
|
|
||||||
nodes_file = self.nodes_file
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
|
|
||||||
for item in items:
|
|
||||||
# print(item)
|
|
||||||
if item == "separator":
|
|
||||||
layout.separator()
|
|
||||||
elif item.get('menu_class'):
|
|
||||||
layout.menu(
|
|
||||||
item.get('menu_class').bl_idname,
|
|
||||||
icon=item.get('icon') or 'NONE'
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
op = layout.operator(
|
|
||||||
'customnodes.add_custom_node',
|
|
||||||
text=item.get('label'),
|
|
||||||
icon=item.get('icon') or 'NONE'
|
|
||||||
)
|
|
||||||
op.nodes_file = nodes_file
|
|
||||||
op.node_type = item['node_type']
|
|
||||||
op.node_label = item.get('label') or ""
|
|
||||||
op.node_description = item.get(
|
|
||||||
'node_description') or "Add Custom Node."
|
|
||||||
op.node_link = item.get('link') or True
|
|
||||||
op.node_callback = item.get('node_callback') or ""
|
|
||||||
|
|
||||||
menu_classes.append(Menu)
|
|
||||||
return Menu
|
|
||||||
|
|
||||||
def _find_item(self, found_items, menu_items, node_type: str):
|
|
||||||
|
|
||||||
for item in menu_items:
|
|
||||||
if item == 'separator':
|
|
||||||
continue
|
|
||||||
|
|
||||||
if item.get("node_type") == node_type:
|
|
||||||
found_items.append(item)
|
|
||||||
|
|
||||||
item_items = item.get('items')
|
|
||||||
if item_items:
|
|
||||||
self._find_item(found_items, item_items, node_type)
|
|
||||||
|
|
||||||
def find_item(self, node_type: str):
|
|
||||||
found_items = []
|
|
||||||
self._find_item(found_items, self.menu_items, node_type)
|
|
||||||
return found_items[0] if len(found_items) > 0 else None
|
|
||||||
|
|
||||||
def add_node(self, node_group, node_type: str):
|
|
||||||
item = self.find_item(node_type)
|
|
||||||
op = AddCustomNode()
|
|
||||||
op.nodes_file = self.nodes_file
|
|
||||||
op.node_type = node_type
|
|
||||||
if item:
|
|
||||||
op.node_label = item.get('label') or ""
|
|
||||||
op.node_link = item.get('link') or True
|
|
||||||
op.node_callback = item.get('node_callback') or ""
|
|
||||||
else:
|
|
||||||
op.node_label = ""
|
|
||||||
op.node_link = True
|
|
||||||
op.node_callback = ""
|
|
||||||
return op.add_node(node_group)
|
|
||||||
|
|
||||||
def register(self):
|
|
||||||
for cls in self.menu_classes:
|
|
||||||
bpy.utils.register_class(cls)
|
|
||||||
|
|
||||||
bpy.types.NODE_MT_add.append(self.add_node_menu)
|
|
||||||
bpy.types.Scene.customnodes_node_dir = bpy.props.StringProperty(
|
|
||||||
name="Node Directory",
|
|
||||||
subtype='DIR_PATH',
|
|
||||||
default="//"
|
|
||||||
)
|
|
||||||
|
|
||||||
def unregister(self):
|
|
||||||
bpy.types.NODE_MT_add.remove(self.add_node_menu)
|
|
||||||
try:
|
|
||||||
for cls in reversed(self.menu_classes):
|
|
||||||
bpy.utils.unregister_class(cls)
|
|
||||||
|
|
||||||
except RuntimeError:
|
|
||||||
pass
|
|
||||||
@@ -1,127 +0,0 @@
|
|||||||
import bpy
|
|
||||||
|
|
||||||
|
|
||||||
class AddCustomNode():
|
|
||||||
|
|
||||||
nodes_file: bpy.props.StringProperty(
|
|
||||||
name="nodes_file",
|
|
||||||
subtype='FILE_PATH',
|
|
||||||
default=""
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
node_type: bpy.props.StringProperty(
|
|
||||||
name='node_type',
|
|
||||||
description='',
|
|
||||||
default='',
|
|
||||||
subtype='NONE',
|
|
||||||
maxlen=0
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
node_label: bpy.props.StringProperty(
|
|
||||||
name='node_label',
|
|
||||||
default=''
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
node_description: bpy.props.StringProperty(
|
|
||||||
name="node_description",
|
|
||||||
description="",
|
|
||||||
default="Add custom node group.",
|
|
||||||
subtype="NONE"
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
node_link: bpy.props.BoolProperty(
|
|
||||||
name='node_link',
|
|
||||||
default=True
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
node_callback: bpy.props.StringProperty(
|
|
||||||
name='node_callback',
|
|
||||||
default=''
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def description(cls, context, properties):
|
|
||||||
return properties.node_description
|
|
||||||
|
|
||||||
def assign_node_tree(self, node):
|
|
||||||
node.node_tree = bpy.data.node_groups[self.node_type]
|
|
||||||
|
|
||||||
node.width = 200.0
|
|
||||||
node.label = self.node_label or self.node_type
|
|
||||||
node.name = self.node_type
|
|
||||||
|
|
||||||
if self.node_callback:
|
|
||||||
exec(self.node_callback)
|
|
||||||
|
|
||||||
return node
|
|
||||||
|
|
||||||
def get_node_tree(self, node_type, node_link):
|
|
||||||
# try to get node from current file if exists
|
|
||||||
node_tree = bpy.data.node_groups.get(node_type)
|
|
||||||
# if not exists, get it from asset file.
|
|
||||||
if not node_tree:
|
|
||||||
bpy.ops.wm.append(
|
|
||||||
'EXEC_DEFAULT',
|
|
||||||
directory=f"{self.nodes_file}/NodeTree",
|
|
||||||
filename=node_type,
|
|
||||||
link=node_link,
|
|
||||||
use_recursive=True
|
|
||||||
)
|
|
||||||
|
|
||||||
node_tree = bpy.data.node_groups.get(node_type)
|
|
||||||
if node_tree:
|
|
||||||
# self.recursive_append_material(node_tree)
|
|
||||||
return node_tree
|
|
||||||
else:
|
|
||||||
raise RuntimeError('No custom node found')
|
|
||||||
|
|
||||||
def recursive_append_material(self, node_tree):
|
|
||||||
for child in node_tree.nodes:
|
|
||||||
material_socket = child.inputs.get('Material')
|
|
||||||
if material_socket:
|
|
||||||
print(material_socket.default_value)
|
|
||||||
try:
|
|
||||||
self.recursive_append_material(child.node_tree)
|
|
||||||
except:
|
|
||||||
...
|
|
||||||
|
|
||||||
def add_node(self, node_group):
|
|
||||||
# Deselect all nodes first
|
|
||||||
for node in node_group.nodes:
|
|
||||||
if node.select:
|
|
||||||
node.select = False
|
|
||||||
|
|
||||||
self.get_node_tree(self.node_type, self.node_link)
|
|
||||||
node = node_group.nodes.new("GeometryNodeGroup")
|
|
||||||
self.assign_node_tree(node)
|
|
||||||
node.show_options = False
|
|
||||||
|
|
||||||
return node
|
|
||||||
|
|
||||||
|
|
||||||
class CUSTOMNODES_OT_Add_Custom_Node(bpy.types.Operator, AddCustomNode):
|
|
||||||
bl_idname = "customnodes.add_custom_node"
|
|
||||||
bl_label = "Add Custom Node"
|
|
||||||
bl_options = {"REGISTER", "UNDO"}
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
self.get_node_tree(self.node_type, self.node_link)
|
|
||||||
|
|
||||||
# intended to be called upon button press in the node tree
|
|
||||||
prev_context = bpy.context.area.type
|
|
||||||
bpy.context.area.type = 'NODE_EDITOR'
|
|
||||||
# actually invoke the operator to add a node to the current node tree
|
|
||||||
# use_transform=True ensures it appears where the user's mouse is and is currently
|
|
||||||
# being moved so the user can place it where they wish
|
|
||||||
bpy.ops.node.add_node(
|
|
||||||
'INVOKE_DEFAULT',
|
|
||||||
type='GeometryNodeGroup',
|
|
||||||
use_transform=True
|
|
||||||
)
|
|
||||||
bpy.context.area.type = prev_context
|
|
||||||
node = bpy.context.active_node
|
|
||||||
|
|
||||||
self.assign_node_tree(node)
|
|
||||||
node.show_options = False
|
|
||||||
|
|
||||||
return {"FINISHED"}
|
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
class CancelledByUser(Exception):
|
|
||||||
def __init__(self):
|
|
||||||
message = 'Cancelled by user'
|
|
||||||
super().__init__(message)
|
|
||||||
@@ -1,197 +0,0 @@
|
|||||||
import bpy
|
|
||||||
|
|
||||||
from .bioxelutils.container import get_container_objs_from_selection
|
|
||||||
from .operators.layer import (CombineLabels, SignScalar,
|
|
||||||
FillByLabel, FillByThreshold, FillByRange)
|
|
||||||
from .operators.container import (SaveContainer, LoadContainer,
|
|
||||||
AddPieCutter, AddPlaneCutter,
|
|
||||||
AddCylinderCutter, AddCubeCutter, AddSphereCutter,
|
|
||||||
PickBboxWire, PickMesh, PickVolume)
|
|
||||||
from .operators.io import (ExportVolumetricData,
|
|
||||||
ImportAsLabelLayer, ImportAsScalarLayer)
|
|
||||||
from .operators.misc import (CleanAllCaches,
|
|
||||||
ReLinkNodes, SaveCaches, SaveStagedData)
|
|
||||||
|
|
||||||
|
|
||||||
def container_is_selected():
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
return len(container_objs) > 0
|
|
||||||
|
|
||||||
|
|
||||||
class PickFromContainerMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_PICK"
|
|
||||||
bl_label = "Pick from Container"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
layout.operator(PickMesh.bl_idname)
|
|
||||||
layout.operator(PickVolume.bl_idname)
|
|
||||||
layout.operator(PickBboxWire.bl_idname)
|
|
||||||
|
|
||||||
|
|
||||||
class ModifyLayerMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_MODIFY_LAYERS"
|
|
||||||
bl_label = "Modify Layer"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
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)
|
|
||||||
|
|
||||||
|
|
||||||
class AddCutterMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_CUTTERS"
|
|
||||||
bl_label = "Add a Cutter to Container"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
layout.operator(AddPlaneCutter.bl_idname, text="Plane Cutter")
|
|
||||||
layout.operator(AddCylinderCutter.bl_idname, text="Cylinder Cutter")
|
|
||||||
layout.operator(AddCubeCutter.bl_idname, text="Cube Cutter")
|
|
||||||
layout.operator(AddSphereCutter.bl_idname, text="Sphere Cutter")
|
|
||||||
layout.operator(AddPieCutter.bl_idname, text="Pie Cutter")
|
|
||||||
|
|
||||||
|
|
||||||
class ImportLayerMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_LAYERS"
|
|
||||||
bl_label = "Import Volumetric Data"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
layout.operator(ImportAsScalarLayer.bl_idname, text="as Scalar")
|
|
||||||
layout.operator(ImportAsLabelLayer.bl_idname, text="as Label")
|
|
||||||
|
|
||||||
|
|
||||||
class BioxelNodesView3DMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_VIEW3D"
|
|
||||||
bl_label = "Bioxel Nodes"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
is_selected = container_is_selected()
|
|
||||||
layout.operator(ImportAsScalarLayer.bl_idname,
|
|
||||||
text=ImportAsScalarLayer.bl_label+" (Add to)"
|
|
||||||
if is_selected else ImportAsScalarLayer.bl_label+" (Init)")
|
|
||||||
layout.operator(ImportAsLabelLayer.bl_idname,
|
|
||||||
text=ImportAsLabelLayer.bl_label+" (Add to)"
|
|
||||||
if is_selected else ImportAsLabelLayer.bl_label+" (Init)")
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(AddPlaneCutter.bl_idname)
|
|
||||||
layout.operator(AddCylinderCutter.bl_idname)
|
|
||||||
layout.operator(AddCubeCutter.bl_idname)
|
|
||||||
layout.operator(AddSphereCutter.bl_idname)
|
|
||||||
layout.operator(AddPieCutter.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(PickMesh.bl_idname)
|
|
||||||
layout.operator(PickVolume.bl_idname)
|
|
||||||
layout.operator(PickBboxWire.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(SaveCaches.bl_idname)
|
|
||||||
|
|
||||||
|
|
||||||
class BioxelNodesOutlinerMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_OUTLINER"
|
|
||||||
bl_label = "Bioxel Nodes"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
is_selected = container_is_selected()
|
|
||||||
layout.operator(ImportAsScalarLayer.bl_idname,
|
|
||||||
text=ImportAsScalarLayer.bl_label+" (Add to)"
|
|
||||||
if is_selected else ImportAsScalarLayer.bl_label+" (Init)")
|
|
||||||
layout.operator(ImportAsLabelLayer.bl_idname,
|
|
||||||
text=ImportAsLabelLayer.bl_label+" (Add to)"
|
|
||||||
if is_selected else ImportAsLabelLayer.bl_label+" (Init)")
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(AddPlaneCutter.bl_idname)
|
|
||||||
layout.operator(AddCylinderCutter.bl_idname)
|
|
||||||
layout.operator(AddCubeCutter.bl_idname)
|
|
||||||
layout.operator(AddSphereCutter.bl_idname)
|
|
||||||
layout.operator(AddPieCutter.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(PickMesh.bl_idname)
|
|
||||||
layout.operator(PickVolume.bl_idname)
|
|
||||||
layout.operator(PickBboxWire.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(SaveCaches.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
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 TOPBAR_FILE_IMPORT(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
is_selected = container_is_selected()
|
|
||||||
|
|
||||||
layout.separator()
|
|
||||||
layout.menu(ImportLayerMenu.bl_idname, text="Volumetric Data as Bioxel (Add to)"
|
|
||||||
if is_selected else "Volumetric Data as Bioxel (Init)")
|
|
||||||
|
|
||||||
|
|
||||||
def TOPBAR_FILE_EXPORT(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(ExportVolumetricData.bl_idname,
|
|
||||||
text="Bioxel Layer (.vdb)")
|
|
||||||
|
|
||||||
|
|
||||||
def VIEW3D_OBJECT(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
layout.separator()
|
|
||||||
layout.menu(BioxelNodesView3DMenu.bl_idname, icon="FILE_VOLUME")
|
|
||||||
|
|
||||||
|
|
||||||
def OUTLINER_OBJECT(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
layout.separator()
|
|
||||||
layout.menu(BioxelNodesOutlinerMenu.bl_idname, icon="FILE_VOLUME")
|
|
||||||
|
|
||||||
|
|
||||||
class BioxelNodesTopbarMenu(bpy.types.Menu):
|
|
||||||
bl_idname = "BIOXELNODES_MT_TOPBAR"
|
|
||||||
bl_label = "Bioxel Nodes"
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
is_selected = container_is_selected()
|
|
||||||
|
|
||||||
layout.menu(ImportLayerMenu.bl_idname, text=ImportLayerMenu.bl_label+" (Add to)"
|
|
||||||
if is_selected else ImportLayerMenu.bl_label+" (Init)")
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(LoadContainer.bl_idname)
|
|
||||||
layout.operator(SaveContainer.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.menu(AddCutterMenu.bl_idname)
|
|
||||||
layout.menu(PickFromContainerMenu.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(SaveStagedData.bl_idname)
|
|
||||||
layout.operator(ReLinkNodes.bl_idname)
|
|
||||||
layout.separator()
|
|
||||||
layout.operator(CleanAllCaches.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.append(OUTLINER_OBJECT)
|
|
||||||
bpy.types.VIEW3D_MT_object_context_menu.append(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)
|
|
||||||
@@ -1,228 +0,0 @@
|
|||||||
|
|
||||||
from pathlib import Path
|
|
||||||
from .customnodes import CustomNodes
|
|
||||||
|
|
||||||
NODE_FILE = "BioxelNodes_4.2"
|
|
||||||
|
|
||||||
MENU_ITEMS = [
|
|
||||||
{
|
|
||||||
'label': 'Methods',
|
|
||||||
'icon': 'OUTLINER_DATA_VOLUME',
|
|
||||||
'items': [
|
|
||||||
{
|
|
||||||
'label': 'Mask by Threshold',
|
|
||||||
'icon': 'EMPTY_SINGLE_ARROW',
|
|
||||||
'node_type': 'BioxelNodes_MaskByThreshold',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Mask by Range',
|
|
||||||
'icon': 'IPO_CONSTANT',
|
|
||||||
'node_type': 'BioxelNodes_MaskByRange',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Mask by Label',
|
|
||||||
'icon': 'MESH_CAPSULE',
|
|
||||||
'node_type': 'BioxelNodes_MaskByLabel',
|
|
||||||
'node_description': ''
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Shaders',
|
|
||||||
'icon': 'SHADING_RENDERED',
|
|
||||||
'items': [
|
|
||||||
{
|
|
||||||
'label': 'Membrane Shader',
|
|
||||||
'icon': 'NODE_MATERIAL',
|
|
||||||
'node_type': 'BioxelNodes_AssignMembraneShader',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Solid Shader',
|
|
||||||
'icon': 'SHADING_SOLID',
|
|
||||||
'node_type': 'BioxelNodes_AssignSolidShader',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Slime Shader',
|
|
||||||
'icon': 'OUTLINER_DATA_META',
|
|
||||||
'node_type': 'BioxelNodes_AssignSlimeShader',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Volume Shader',
|
|
||||||
'icon': 'VOLUME_DATA',
|
|
||||||
'node_type': 'BioxelNodes_AssignVolumeShader',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Universal Shader',
|
|
||||||
'icon': 'MATSHADERBALL',
|
|
||||||
'node_type': 'BioxelNodes_AssignUniversalShader',
|
|
||||||
'node_description': ''
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Colors',
|
|
||||||
'icon': 'COLOR',
|
|
||||||
'items': [
|
|
||||||
{
|
|
||||||
'label': 'Color Presets',
|
|
||||||
'icon': 'COLOR',
|
|
||||||
'node_type': 'BioxelNodes_ColorPresets',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Color Presets MRI',
|
|
||||||
'icon': 'COLOR',
|
|
||||||
'node_type': 'BioxelNodes_ColorPresets_MRI',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
"separator",
|
|
||||||
{
|
|
||||||
'label': 'Color Ramp 2',
|
|
||||||
'icon': 'IPO_QUAD',
|
|
||||||
'node_type': 'BioxelNodes_SetColorRamp2',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Color Ramp 3',
|
|
||||||
'icon': 'IPO_CUBIC',
|
|
||||||
'node_type': 'BioxelNodes_SetColorRamp3',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Color Ramp 4',
|
|
||||||
'icon': 'IPO_QUART',
|
|
||||||
'node_type': 'BioxelNodes_SetColorRamp4',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Color Ramp 5',
|
|
||||||
'icon': 'IPO_QUINT',
|
|
||||||
'node_type': 'BioxelNodes_SetColorRamp5',
|
|
||||||
'node_description': ''
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Cutters',
|
|
||||||
'icon': 'MOD_BEVEL',
|
|
||||||
'items': [
|
|
||||||
{
|
|
||||||
'label': 'Cut',
|
|
||||||
'icon': 'MOD_BEVEL',
|
|
||||||
'node_type': 'BioxelNodes_Cut',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
"separator",
|
|
||||||
{
|
|
||||||
'label': 'Primitive Cutter',
|
|
||||||
'icon': 'MOD_LINEART',
|
|
||||||
'node_type': 'BioxelNodes_PrimitiveCutter',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
"separator",
|
|
||||||
{
|
|
||||||
'label': 'Plane Cutter',
|
|
||||||
'icon': 'MESH_PLANE',
|
|
||||||
'node_type': 'BioxelNodes_PlaneObjectCutter',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Cylinder Cutter',
|
|
||||||
'icon': 'MESH_CYLINDER',
|
|
||||||
'node_type': 'BioxelNodes_CylinderObjectCutter',
|
|
||||||
'node_description': '',
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Cube Cutter',
|
|
||||||
'icon': 'MESH_CUBE',
|
|
||||||
'node_type': 'BioxelNodes_CubeObjectCutter',
|
|
||||||
'node_description': '',
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Sphere Cutter',
|
|
||||||
'icon': 'MESH_UVSPHERE',
|
|
||||||
'node_type': 'BioxelNodes_SphereObjectCutter',
|
|
||||||
'node_description': '',
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Pie Cutter',
|
|
||||||
'icon': 'MESH_CONE',
|
|
||||||
'node_type': 'BioxelNodes_PieObjectCutter',
|
|
||||||
'node_description': '',
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Utils',
|
|
||||||
'icon': 'MODIFIER',
|
|
||||||
'items': [
|
|
||||||
{
|
|
||||||
'label': 'Join Component',
|
|
||||||
'icon': 'CONSTRAINT_BONE',
|
|
||||||
'node_type': 'BioxelNodes_JoinComponent',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
"separator",
|
|
||||||
{
|
|
||||||
'label': 'Pick Mesh',
|
|
||||||
'icon': 'OUTLINER_OB_MESH',
|
|
||||||
'node_type': 'BioxelNodes_PickMesh',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Pick Volume',
|
|
||||||
'icon': 'OUTLINER_OB_VOLUME',
|
|
||||||
'node_type': 'BioxelNodes_PickVolume',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Pick Bbox Wire',
|
|
||||||
'icon': 'MESH_CUBE',
|
|
||||||
'node_type': 'BioxelNodes_PickBboxWire',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
"separator",
|
|
||||||
{
|
|
||||||
'label': 'Inflate',
|
|
||||||
'icon': 'OUTLINER_OB_META',
|
|
||||||
'node_type': 'BioxelNodes_M_Inflate',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Smooth',
|
|
||||||
'icon': 'MOD_SMOOTH',
|
|
||||||
'node_type': 'BioxelNodes_M_Smooth',
|
|
||||||
'node_description': ''
|
|
||||||
},
|
|
||||||
{
|
|
||||||
'label': 'Remove Small Island',
|
|
||||||
'icon': 'FORCE_LENNARDJONES',
|
|
||||||
'node_type': 'BioxelNodes_M_RemoveSmallIsland',
|
|
||||||
'node_description': ''
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
custom_nodes = CustomNodes(
|
|
||||||
menu_items=MENU_ITEMS,
|
|
||||||
nodes_file=Path(__file__).parent / f"assets/Nodes/{NODE_FILE}.blend",
|
|
||||||
class_prefix="BIOXELNODES_MT_NODES",
|
|
||||||
root_label='Bioxel Nodes',
|
|
||||||
root_icon="FILE_VOLUME",
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register():
|
|
||||||
custom_nodes.register()
|
|
||||||
|
|
||||||
|
|
||||||
def unregister():
|
|
||||||
custom_nodes.unregister()
|
|
||||||
@@ -1,290 +0,0 @@
|
|||||||
import bpy
|
|
||||||
|
|
||||||
|
|
||||||
import bmesh
|
|
||||||
|
|
||||||
from ..nodes import custom_nodes
|
|
||||||
from ..bioxel.io import load_container, save_container
|
|
||||||
from ..bioxelutils.layer import get_all_layer_objs
|
|
||||||
from ..bioxelutils.container import (container_to_obj, obj_to_container,
|
|
||||||
get_container_objs_from_selection)
|
|
||||||
from ..bioxelutils.node import get_nodes_by_type, move_node_between_nodes, move_node_to_node
|
|
||||||
from .utils import change_render_setting, get_cache_dir, get_preferences, select_object
|
|
||||||
|
|
||||||
|
|
||||||
class SaveContainer(bpy.types.Operator):
|
|
||||||
bl_idname = "bioxelnodes.save_container"
|
|
||||||
bl_label = "Save Container (.bioxel) (BETA)"
|
|
||||||
bl_description = "Clean all caches saved in temp"
|
|
||||||
|
|
||||||
filepath: bpy.props.StringProperty(
|
|
||||||
subtype="FILE_PATH"
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
filename_ext = ".bioxel"
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
return len(container_objs) > 0
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
|
|
||||||
if len(container_objs) == 0:
|
|
||||||
self.report({"WARNING"}, "Cannot find any bioxel container.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
container = obj_to_container(container_objs[0])
|
|
||||||
save_path = f"{self.filepath.split('.')[0]}.bioxel"
|
|
||||||
save_container(container, save_path, overwrite=True)
|
|
||||||
|
|
||||||
self.report({"INFO"}, f"Successfully save to {save_path}")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
context.window_manager.fileselect_add(self)
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
|
|
||||||
class LoadContainer(bpy.types.Operator):
|
|
||||||
bl_idname = "bioxelnodes.load_container"
|
|
||||||
bl_label = "Load Container (.bioxel) (BETA)"
|
|
||||||
bl_description = "Clean all caches saved in temp"
|
|
||||||
|
|
||||||
filepath: bpy.props.StringProperty(
|
|
||||||
subtype="FILE_PATH"
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
filename_ext = ".bioxel"
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
load_path = self.filepath
|
|
||||||
container = load_container(self.filepath)
|
|
||||||
is_first_import = len(get_all_layer_objs()) == 0
|
|
||||||
|
|
||||||
container_obj = container_to_obj(container,
|
|
||||||
scene_scale=0.01,
|
|
||||||
cache_dir=get_cache_dir(context))
|
|
||||||
select_object(container_obj)
|
|
||||||
|
|
||||||
if is_first_import:
|
|
||||||
change_render_setting(context)
|
|
||||||
|
|
||||||
self.report({"INFO"}, f"Successfully load {load_path}")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
context.window_manager.fileselect_add(self)
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
|
|
||||||
class PickObject():
|
|
||||||
bl_options = {'UNDO'}
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
return len(container_objs) > 0
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
|
|
||||||
if len(container_objs) == 0:
|
|
||||||
self.report({"WARNING"}, "Cannot find any bioxel container.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
container_obj = container_objs[0]
|
|
||||||
|
|
||||||
bpy.ops.mesh.primitive_cube_add(
|
|
||||||
size=2, enter_editmode=False, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1))
|
|
||||||
obj = bpy.context.active_object
|
|
||||||
|
|
||||||
obj.name = f"{self.object_type}_{container_obj.name}"
|
|
||||||
|
|
||||||
bpy.ops.node.new_geometry_nodes_modifier()
|
|
||||||
modifier = obj.modifiers[0]
|
|
||||||
node_group = modifier.node_group
|
|
||||||
|
|
||||||
output_node = get_nodes_by_type(node_group, 'NodeGroupOutput')[0]
|
|
||||||
pick_mesh_node = custom_nodes.add_node(node_group,
|
|
||||||
f"BioxelNodes_Pick{self.object_type}")
|
|
||||||
|
|
||||||
pick_mesh_node.inputs[0].default_value = container_obj
|
|
||||||
node_group.links.new(pick_mesh_node.outputs[0], output_node.inputs[0])
|
|
||||||
|
|
||||||
select_object(obj)
|
|
||||||
|
|
||||||
self.report({"INFO"}, f"Successfully picked")
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
class PickMesh(bpy.types.Operator, PickObject):
|
|
||||||
bl_idname = "bioxelnodes.pick_mesh"
|
|
||||||
bl_label = "Pick Mesh"
|
|
||||||
bl_description = "Pick Container Mesh"
|
|
||||||
object_type = "Mesh"
|
|
||||||
|
|
||||||
|
|
||||||
class PickVolume(bpy.types.Operator, PickObject):
|
|
||||||
bl_idname = "bioxelnodes.pick_volume"
|
|
||||||
bl_label = "Pick Volume"
|
|
||||||
bl_description = "Pick Container Volume"
|
|
||||||
object_type = "Volume"
|
|
||||||
|
|
||||||
|
|
||||||
class PickBboxWire(bpy.types.Operator, PickObject):
|
|
||||||
bl_idname = "bioxelnodes.pick_bbox_wire"
|
|
||||||
bl_label = "Pick Bbox Wire"
|
|
||||||
bl_description = "Pick Container Bbox Wire"
|
|
||||||
object_type = "BboxWire"
|
|
||||||
|
|
||||||
|
|
||||||
class AddCutter():
|
|
||||||
bl_options = {'UNDO'}
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
return len(container_objs) > 0
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
|
|
||||||
if len(container_objs) == 0:
|
|
||||||
self.report({"WARNING"}, "Cannot find any bioxel container.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
container_obj = container_objs[0]
|
|
||||||
|
|
||||||
if self.object_type == "plane":
|
|
||||||
node_type = "BioxelNodes_PlaneObjectCutter"
|
|
||||||
bpy.ops.mesh.primitive_plane_add(
|
|
||||||
size=2, enter_editmode=False, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1))
|
|
||||||
elif self.object_type == "cylinder":
|
|
||||||
node_type = "BioxelNodes_CylinderObjectCutter"
|
|
||||||
bpy.ops.mesh.primitive_cylinder_add(
|
|
||||||
radius=1, depth=2, enter_editmode=False, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1))
|
|
||||||
bpy.context.object.rotation_euler[0] = container_obj.rotation_euler[0]
|
|
||||||
elif self.object_type == "cube":
|
|
||||||
node_type = "BioxelNodes_CubeObjectCutter"
|
|
||||||
bpy.ops.mesh.primitive_cube_add(
|
|
||||||
size=2, enter_editmode=False, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1))
|
|
||||||
elif self.object_type == "sphere":
|
|
||||||
node_type = "BioxelNodes_SphereObjectCutter"
|
|
||||||
bpy.ops.mesh.primitive_ico_sphere_add(
|
|
||||||
radius=1, enter_editmode=False, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1))
|
|
||||||
elif self.object_type == "pie":
|
|
||||||
node_type = "BioxelNodes_PieObjectCutter"
|
|
||||||
# Create mesh
|
|
||||||
pie_mesh = bpy.data.meshes.new('Pie')
|
|
||||||
|
|
||||||
# Create object
|
|
||||||
pie = bpy.data.objects.new('Pie', pie_mesh)
|
|
||||||
|
|
||||||
# Link object to scene
|
|
||||||
bpy.context.scene.collection.objects.link(pie)
|
|
||||||
# Get a BMesh representation
|
|
||||||
bm = bmesh.new() # create an empty BMesh
|
|
||||||
bm.from_mesh(pie_mesh) # fill it in from a Mesh
|
|
||||||
|
|
||||||
# Hot to create vertices
|
|
||||||
v_0 = bm.verts.new((0.0, -1.0, 0.0))
|
|
||||||
v_1 = bm.verts.new((-1.0, -1.0, 1.0))
|
|
||||||
v_2 = bm.verts.new((0.0, 1.0, 0.0))
|
|
||||||
v_3 = bm.verts.new((-1.0, 1.0, 1.0))
|
|
||||||
v_4 = bm.verts.new((1.0, -1.0, 1.0))
|
|
||||||
v_5 = bm.verts.new((1.0, 1.0, 1.0))
|
|
||||||
|
|
||||||
# Initialize the index values of this sequence.
|
|
||||||
bm.verts.index_update()
|
|
||||||
|
|
||||||
# How to create a face
|
|
||||||
# it's not necessary to create the edges before, I made it only to show how create
|
|
||||||
# edges too
|
|
||||||
bm.faces.new((v_0, v_1, v_3, v_2))
|
|
||||||
bm.faces.new((v_0, v_2, v_5, v_4))
|
|
||||||
|
|
||||||
# Finish up, write the bmesh back to the mesh
|
|
||||||
bm.to_mesh(pie_mesh)
|
|
||||||
bpy.context.view_layer.objects.active = pie
|
|
||||||
|
|
||||||
cutter_obj = bpy.context.active_object
|
|
||||||
cutter_obj.visible_camera = False
|
|
||||||
cutter_obj.visible_diffuse = False
|
|
||||||
cutter_obj.visible_glossy = False
|
|
||||||
cutter_obj.visible_transmission = False
|
|
||||||
cutter_obj.visible_volume_scatter = False
|
|
||||||
cutter_obj.visible_shadow = False
|
|
||||||
cutter_obj.hide_render = True
|
|
||||||
cutter_obj.display_type = 'WIRE'
|
|
||||||
|
|
||||||
modifier = container_obj.modifiers[0]
|
|
||||||
node_group = modifier.node_group
|
|
||||||
cutter_node = custom_nodes.add_node(node_group, node_type)
|
|
||||||
cutter_node.inputs[0].default_value = cutter_obj
|
|
||||||
|
|
||||||
cut_nodes = get_nodes_by_type(node_group,
|
|
||||||
'BioxelNodes_Cut')
|
|
||||||
output_node = get_nodes_by_type(node_group, 'NodeGroupOutput')[0]
|
|
||||||
if len(cut_nodes) == 0:
|
|
||||||
cut_node = custom_nodes.add_node(node_group, 'BioxelNodes_Cut')
|
|
||||||
if len(output_node.inputs[0].links) == 0:
|
|
||||||
node_group.links.new(cut_node.outputs[0],
|
|
||||||
output_node.inputs[0])
|
|
||||||
move_node_to_node(cut_node, output_node, (-300, 0))
|
|
||||||
else:
|
|
||||||
pre_output_node = output_node.inputs[0].links[0].from_node
|
|
||||||
node_group.links.new(pre_output_node.outputs[0],
|
|
||||||
cut_node.inputs[0])
|
|
||||||
node_group.links.new(cut_node.outputs[0],
|
|
||||||
output_node.inputs[0])
|
|
||||||
move_node_between_nodes(cut_node,
|
|
||||||
[pre_output_node, output_node])
|
|
||||||
|
|
||||||
node_group.links.new(cutter_node.outputs[0],
|
|
||||||
cut_node.inputs[1])
|
|
||||||
|
|
||||||
move_node_to_node(cutter_node, cut_node, (-300, -300))
|
|
||||||
select_object(cutter_obj)
|
|
||||||
else:
|
|
||||||
move_node_to_node(cutter_node, output_node, (0, -100))
|
|
||||||
select_object(container_obj)
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
class AddPlaneCutter(bpy.types.Operator, AddCutter):
|
|
||||||
bl_idname = "bioxelnodes.add_plane_cutter"
|
|
||||||
bl_label = "Add a Plane Cutter"
|
|
||||||
bl_description = "Add a Plane Cutter to Container"
|
|
||||||
object_type = "plane"
|
|
||||||
|
|
||||||
|
|
||||||
class AddCylinderCutter(bpy.types.Operator, AddCutter):
|
|
||||||
bl_idname = "bioxelnodes.add_cylinder_cutter"
|
|
||||||
bl_label = "Add a Cylinder Cutter"
|
|
||||||
bl_description = "Add a Cylinder Cutter to Container"
|
|
||||||
object_type = "cylinder"
|
|
||||||
|
|
||||||
|
|
||||||
class AddCubeCutter(bpy.types.Operator, AddCutter):
|
|
||||||
bl_idname = "bioxelnodes.add_cube_cutter"
|
|
||||||
bl_label = "Add a Cube Cutter"
|
|
||||||
bl_description = "Add a Cube Cutter to Container"
|
|
||||||
object_type = "cube"
|
|
||||||
|
|
||||||
|
|
||||||
class AddSphereCutter(bpy.types.Operator, AddCutter):
|
|
||||||
bl_idname = "bioxelnodes.add_sphere_cutter"
|
|
||||||
bl_label = "Add a Sphere Cutter"
|
|
||||||
bl_description = "Add a Sphere Cutter to Container"
|
|
||||||
object_type = "sphere"
|
|
||||||
|
|
||||||
|
|
||||||
class AddPieCutter(bpy.types.Operator, AddCutter):
|
|
||||||
bl_idname = "bioxelnodes.add_pie_cutter"
|
|
||||||
bl_label = "Add a Pie Cutter"
|
|
||||||
bl_description = "Add a Pie Cutter to Container"
|
|
||||||
object_type = "pie"
|
|
||||||
@@ -1,251 +0,0 @@
|
|||||||
from pathlib import Path
|
|
||||||
import bpy
|
|
||||||
|
|
||||||
import numpy as np
|
|
||||||
|
|
||||||
|
|
||||||
from ..bioxel.layer import Layer
|
|
||||||
from ..bioxelutils.node import get_nodes_by_type
|
|
||||||
from ..bioxelutils.container import add_layers, get_container_obj
|
|
||||||
from ..bioxelutils.layer import obj_to_layer, get_container_layer_objs, get_layer_obj
|
|
||||||
from .utils import get_preferences, select_object
|
|
||||||
|
|
||||||
|
|
||||||
def get_layer_prop_value(layer_obj: bpy.types.Object, prop: str):
|
|
||||||
try:
|
|
||||||
node_group = layer_obj.modifiers[0].node_group
|
|
||||||
layer_node = get_nodes_by_type(node_group, "BioxelNodes__Layer")[0]
|
|
||||||
return layer_node.inputs[prop].default_value
|
|
||||||
except:
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def get_label_layer_selection(self, context):
|
|
||||||
items = [("None", "None", "")]
|
|
||||||
container_obj = get_container_obj(bpy.context.active_object)
|
|
||||||
|
|
||||||
for layer_obj in get_container_layer_objs(container_obj):
|
|
||||||
if get_layer_prop_value(layer_obj, "kind") == "label":
|
|
||||||
items.append((layer_obj.name,
|
|
||||||
layer_obj.name,
|
|
||||||
""))
|
|
||||||
|
|
||||||
return items
|
|
||||||
|
|
||||||
|
|
||||||
class LayerOperator():
|
|
||||||
bl_options = {'UNDO'}
|
|
||||||
base_obj: bpy.types.Object = None
|
|
||||||
|
|
||||||
def add_layer(self, context, layer):
|
|
||||||
preferences = get_preferences(context)
|
|
||||||
cache_dir = Path(preferences.cache_dir, 'VDBs')
|
|
||||||
container_obj = self.base_obj.parent
|
|
||||||
|
|
||||||
add_layers([layer],
|
|
||||||
container_obj,
|
|
||||||
cache_dir)
|
|
||||||
|
|
||||||
select_object(container_obj)
|
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
self.base_obj = get_layer_obj(bpy.context.active_object)
|
|
||||||
return self.execute(context)
|
|
||||||
|
|
||||||
|
|
||||||
class ScalarOperator(LayerOperator):
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
label_objs = [obj for obj in context.selected_ids
|
|
||||||
if get_layer_prop_value(obj, "kind") == "scalar"]
|
|
||||||
return True if len(label_objs) > 0 else False
|
|
||||||
|
|
||||||
|
|
||||||
class LabelOperator(LayerOperator):
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
label_objs = [obj for obj in context.selected_ids
|
|
||||||
if get_layer_prop_value(obj, "kind") == "label"]
|
|
||||||
return True if len(label_objs) > 0 else False
|
|
||||||
|
|
||||||
|
|
||||||
class SignScalar(bpy.types.Operator, ScalarOperator):
|
|
||||||
bl_idname = "bioxelnodes.sign_scalar"
|
|
||||||
bl_label = "Sign Scalar"
|
|
||||||
bl_description = "Sign the scalar value"
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
base_layer = obj_to_layer(self.base_obj)
|
|
||||||
|
|
||||||
modified_layer = base_layer.copy()
|
|
||||||
modified_layer.data = -base_layer.data
|
|
||||||
modified_layer.name = f"{base_layer.name}_Signed"
|
|
||||||
|
|
||||||
self.add_layer(context, modified_layer)
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
class FillOperator(ScalarOperator):
|
|
||||||
def invoke(self, context, event):
|
|
||||||
self.base_obj = get_layer_obj(bpy.context.active_object)
|
|
||||||
scalar_min = get_layer_prop_value(self.base_obj, "min")
|
|
||||||
self.fill_value = min(scalar_min, 0)
|
|
||||||
context.window_manager.invoke_props_dialog(self, width=400)
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
|
|
||||||
class FillByThreshold(bpy.types.Operator, FillOperator):
|
|
||||||
bl_idname = "bioxelnodes.fill_by_threshold"
|
|
||||||
bl_label = "Fill Value by Threshold"
|
|
||||||
bl_description = "Fill Value by Threshold"
|
|
||||||
|
|
||||||
threshold: bpy.props.FloatProperty(
|
|
||||||
name="Threshold",
|
|
||||||
soft_min=0, soft_max=1024,
|
|
||||||
default=128,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
fill_value: bpy.props.FloatProperty(
|
|
||||||
name="Fill Value",
|
|
||||||
soft_min=0, soft_max=1024.0,
|
|
||||||
default=0,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
invert: bpy.props.BoolProperty(
|
|
||||||
name="Invert Area",
|
|
||||||
default=True,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
base_layer = obj_to_layer(self.base_obj)
|
|
||||||
|
|
||||||
data = np.amax(base_layer.data, -1)
|
|
||||||
mask = data <= self.threshold \
|
|
||||||
if self.invert else data > self.threshold
|
|
||||||
|
|
||||||
modified_layer = base_layer.copy()
|
|
||||||
modified_layer.fill(self.fill_value, mask)
|
|
||||||
modified_layer.name = f"{base_layer.name}_{self.threshold}-Filled"
|
|
||||||
|
|
||||||
self.add_layer(context, modified_layer)
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
class FillByRange(bpy.types.Operator, FillOperator):
|
|
||||||
bl_idname = "bioxelnodes.fill_by_range"
|
|
||||||
bl_label = "Fill Value by Range"
|
|
||||||
bl_description = "Fill Value by Range"
|
|
||||||
|
|
||||||
from_min: bpy.props.FloatProperty(
|
|
||||||
name="From Min",
|
|
||||||
soft_min=0, soft_max=1024,
|
|
||||||
default=128,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
from_max: bpy.props.FloatProperty(
|
|
||||||
name="From Max",
|
|
||||||
soft_min=0, soft_max=1024,
|
|
||||||
default=256,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
fill_value: bpy.props.FloatProperty(
|
|
||||||
name="Fill Value",
|
|
||||||
soft_min=0, soft_max=1024.0,
|
|
||||||
default=0,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
invert: bpy.props.BoolProperty(
|
|
||||||
name="Invert Area",
|
|
||||||
default=True,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
base_layer = obj_to_layer(self.base_obj)
|
|
||||||
|
|
||||||
data = np.amax(base_layer.data, -1)
|
|
||||||
mask = (data <= self.from_min) | (data >= self.from_max) if self.invert else \
|
|
||||||
(data > self.from_min) & (data < self.from_max)
|
|
||||||
|
|
||||||
modified_layer = base_layer.copy()
|
|
||||||
modified_layer.fill(self.fill_value, mask)
|
|
||||||
modified_layer.name = f"{base_layer.name}_{self.from_min}-{self.from_max}-Filled"
|
|
||||||
|
|
||||||
self.add_layer(context, modified_layer)
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
class FillByLabel(bpy.types.Operator, FillOperator):
|
|
||||||
bl_idname = "bioxelnodes.fill_by_label"
|
|
||||||
bl_label = "Fill Value by Label"
|
|
||||||
bl_description = "Fill Value by Label Area"
|
|
||||||
|
|
||||||
label_obj_name: bpy.props.EnumProperty(
|
|
||||||
name="Label Layer",
|
|
||||||
items=get_label_layer_selection
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
fill_value: bpy.props.FloatProperty(
|
|
||||||
name="Fill Value",
|
|
||||||
soft_min=0, soft_max=1024.0,
|
|
||||||
default=0,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
invert: bpy.props.BoolProperty(
|
|
||||||
name="Invert Label",
|
|
||||||
default=True,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
label_obj = bpy.data.objects.get(self.label_obj_name)
|
|
||||||
if not label_obj:
|
|
||||||
self.report({"WARNING"}, "Cannot find any label layer.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
base_layer = obj_to_layer(self.base_obj)
|
|
||||||
|
|
||||||
label_layer = obj_to_layer(label_obj)
|
|
||||||
label_layer.resize(base_layer.shape)
|
|
||||||
mask = np.amax(label_layer.data, -1)
|
|
||||||
if self.invert:
|
|
||||||
mask = 1 - mask
|
|
||||||
|
|
||||||
modified_layer = base_layer.copy()
|
|
||||||
modified_layer.fill(self.fill_value, mask)
|
|
||||||
modified_layer.name = f"{base_layer.name}_{label_layer.name}-Filled"
|
|
||||||
|
|
||||||
self.add_layer(context, modified_layer)
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
class CombineLabels(bpy.types.Operator, LabelOperator):
|
|
||||||
bl_idname = "bioxelnodes.combine_labels"
|
|
||||||
bl_label = "Combine Labels"
|
|
||||||
bl_description = "Combine all selected labels"
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
label_objs = [obj for obj in context.selected_ids
|
|
||||||
if get_layer_prop_value(obj, "kind") == "label"]
|
|
||||||
|
|
||||||
if len(label_objs) < 2:
|
|
||||||
self.report({"WARNING"}, "Not enough layers.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
base_obj = label_objs[0]
|
|
||||||
label_objs = label_objs[1:]
|
|
||||||
container_obj = base_obj.parent
|
|
||||||
|
|
||||||
base_layer = obj_to_layer(base_obj)
|
|
||||||
modified_layer = base_layer.copy()
|
|
||||||
label_names = [base_layer.name]
|
|
||||||
|
|
||||||
for label_obj in label_objs:
|
|
||||||
label_layer = obj_to_layer(label_obj)
|
|
||||||
label_layer.resize(base_layer.shape)
|
|
||||||
modified_layer.data = np.maximum(base_layer.data, label_layer.data)
|
|
||||||
label_names.append(label_layer.name)
|
|
||||||
|
|
||||||
modified_layer.name = f"{'-'.join(label_names)}-Combined"
|
|
||||||
|
|
||||||
self.add_layer(context, modified_layer)
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
@@ -1,215 +0,0 @@
|
|||||||
import re
|
|
||||||
import bpy
|
|
||||||
from pathlib import Path
|
|
||||||
import shutil
|
|
||||||
|
|
||||||
|
|
||||||
from ..utils import copy_to_dir
|
|
||||||
from .utils import get_cache_dir
|
|
||||||
from ..nodes import custom_nodes
|
|
||||||
from ..bioxelutils.container import get_container_objs_from_selection
|
|
||||||
from ..bioxelutils.layer import get_all_layer_objs, get_container_layer_objs
|
|
||||||
|
|
||||||
CLASS_PREFIX = "BIOXELNODES_MT_NODES"
|
|
||||||
|
|
||||||
|
|
||||||
class ReLinkNodes(bpy.types.Operator):
|
|
||||||
bl_idname = "bioxelnodes.relink_nodes"
|
|
||||||
bl_label = "Relink Nodes to Addon"
|
|
||||||
bl_description = "Relink all nodes to addon source"
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
file_name = Path(custom_nodes.nodes_file).name
|
|
||||||
for lib in bpy.data.libraries:
|
|
||||||
lib_path = Path(bpy.path.abspath(lib.filepath)).resolve()
|
|
||||||
lib_name = lib_path.name
|
|
||||||
if lib_name == file_name:
|
|
||||||
if str(lib_path) != custom_nodes.nodes_file:
|
|
||||||
lib.filepath = custom_nodes.nodes_file
|
|
||||||
|
|
||||||
self.report({"INFO"}, f"Successfully relinked.")
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
|
|
||||||
def save_layer_cache(layer_obj, output_dir):
|
|
||||||
pattern = r'\.\d{4}\.'
|
|
||||||
|
|
||||||
# "//"
|
|
||||||
output_dir = bpy.path.abspath(output_dir)
|
|
||||||
source_dir = bpy.path.abspath(layer_obj.data.filepath)
|
|
||||||
|
|
||||||
source_path: Path = Path(source_dir).resolve()
|
|
||||||
is_sequence = re.search(pattern, source_path.name) is not None
|
|
||||||
name = layer_obj.name if is_sequence else f"{layer_obj.name}.vdb"
|
|
||||||
output_path: Path = Path(output_dir, name, source_path.name).resolve() \
|
|
||||||
if is_sequence else Path(output_dir, name).resolve()
|
|
||||||
|
|
||||||
if output_path != source_path:
|
|
||||||
copy_to_dir(source_path.parent if is_sequence else source_path,
|
|
||||||
output_path.parent.parent if is_sequence else output_path.parent,
|
|
||||||
new_name=name)
|
|
||||||
|
|
||||||
blend_path = Path(bpy.path.abspath("//")).resolve()
|
|
||||||
|
|
||||||
layer_obj.data.filepath = bpy.path.relpath(
|
|
||||||
str(output_path), start=str(blend_path))
|
|
||||||
|
|
||||||
|
|
||||||
class SaveStagedData(bpy.types.Operator):
|
|
||||||
bl_idname = "bioxelnodes.save_staged_data"
|
|
||||||
bl_label = "Save Staged Data"
|
|
||||||
bl_description = "Save all staged data in this file for sharing"
|
|
||||||
|
|
||||||
save_layer: bpy.props.BoolProperty(
|
|
||||||
name="Save Layer VDB Cache",
|
|
||||||
default=True,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
cache_dir: bpy.props.StringProperty(
|
|
||||||
name="Cache Directory",
|
|
||||||
subtype='DIR_PATH',
|
|
||||||
default="//"
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
save_lib: bpy.props.BoolProperty(
|
|
||||||
name="Save Node Library File",
|
|
||||||
default=True,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
lib_dir: bpy.props.StringProperty(
|
|
||||||
name="Library Directory",
|
|
||||||
subtype='DIR_PATH',
|
|
||||||
default="//"
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
if self.save_lib:
|
|
||||||
files = []
|
|
||||||
for classname in dir(bpy.types):
|
|
||||||
if CLASS_PREFIX in classname:
|
|
||||||
cls = getattr(bpy.types, classname)
|
|
||||||
files.append(cls.nodes_file)
|
|
||||||
files = list(set(files))
|
|
||||||
|
|
||||||
for file in files:
|
|
||||||
file_name = Path(file).name
|
|
||||||
# "//"
|
|
||||||
lib_dir = bpy.path.abspath(self.lib_dir)
|
|
||||||
|
|
||||||
output_path: Path = Path(lib_dir, file_name).resolve()
|
|
||||||
source_path: Path = Path(file).resolve()
|
|
||||||
|
|
||||||
if output_path != source_path:
|
|
||||||
shutil.copy(source_path, output_path)
|
|
||||||
|
|
||||||
for lib in bpy.data.libraries:
|
|
||||||
lib_path = Path(bpy.path.abspath(lib.filepath)).resolve()
|
|
||||||
if lib_path == source_path:
|
|
||||||
blend_path = Path(bpy.path.abspath("//")).resolve()
|
|
||||||
lib.filepath = bpy.path.relpath(
|
|
||||||
str(output_path), start=str(blend_path))
|
|
||||||
|
|
||||||
self.report({"INFO"}, f"Successfully saved to {output_path}")
|
|
||||||
|
|
||||||
if self.save_layer:
|
|
||||||
fails = []
|
|
||||||
for layer in get_all_layer_objs():
|
|
||||||
try:
|
|
||||||
save_layer(layer, self.cache_dir)
|
|
||||||
except:
|
|
||||||
fails.append(layer)
|
|
||||||
|
|
||||||
if len(fails) == 0:
|
|
||||||
self.report({"INFO"}, f"Successfully saved bioxel layers.")
|
|
||||||
else:
|
|
||||||
self.report(
|
|
||||||
{"WARNING"}, f"{','.join([layer.name for layer in fails])} fail to save.")
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
context.window_manager.invoke_props_dialog(self,
|
|
||||||
width=500)
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
return bpy.data.is_saved
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
panel = layout.box()
|
|
||||||
panel.prop(self, "save_layer")
|
|
||||||
panel.prop(self, "cache_dir")
|
|
||||||
panel = layout.box()
|
|
||||||
panel.prop(self, "save_lib")
|
|
||||||
panel.prop(self, "lib_dir")
|
|
||||||
|
|
||||||
|
|
||||||
class SaveCaches(bpy.types.Operator):
|
|
||||||
bl_idname = "bioxelnodes.save_layers"
|
|
||||||
bl_label = "Save Caches"
|
|
||||||
bl_description = "Save Container's caches to directory."
|
|
||||||
|
|
||||||
cache_dir: bpy.props.StringProperty(
|
|
||||||
name="Layer Directory",
|
|
||||||
subtype='DIR_PATH',
|
|
||||||
default="//"
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
return len(container_objs) > 0
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
container_objs = get_container_objs_from_selection()
|
|
||||||
|
|
||||||
if len(container_objs) == 0:
|
|
||||||
self.report({"WARNING"}, "Cannot find any bioxel container.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
fails = []
|
|
||||||
for container_obj in container_objs:
|
|
||||||
for layer_obj in get_container_layer_objs(container_obj):
|
|
||||||
try:
|
|
||||||
save_layer_cache(layer_obj, self.cache_dir)
|
|
||||||
except:
|
|
||||||
fails.append(layer_obj)
|
|
||||||
|
|
||||||
if len(fails) == 0:
|
|
||||||
self.report({"INFO"}, f"Successfully saved bioxel layers.")
|
|
||||||
else:
|
|
||||||
self.report(
|
|
||||||
{"WARNING"}, f"{','.join([layer.name for layer in fails])} fail to save.")
|
|
||||||
|
|
||||||
return {'FINISHED'}
|
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
context.window_manager.invoke_props_dialog(self,
|
|
||||||
width=500)
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
|
|
||||||
class CleanAllCaches(bpy.types.Operator):
|
|
||||||
bl_idname = "bioxelnodes.clear_all_caches"
|
|
||||||
bl_label = "Clean All Caches in Temp"
|
|
||||||
bl_description = "Clean all caches saved in temp"
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
cache_dir = get_cache_dir(context)
|
|
||||||
try:
|
|
||||||
shutil.rmtree(cache_dir)
|
|
||||||
self.report({"INFO"}, f"Successfully cleaned caches.")
|
|
||||||
return {'FINISHED'}
|
|
||||||
except:
|
|
||||||
self.report({"WARNING"},
|
|
||||||
"Fail to clean caches, you may do it manually.")
|
|
||||||
return {'CANCELLED'}
|
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
context.window_manager.invoke_confirm(self,
|
|
||||||
event,
|
|
||||||
message="All caches will be cleaned, include other project files, do you still want to clean?")
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
@@ -1,60 +0,0 @@
|
|||||||
import bpy
|
|
||||||
from pathlib import Path
|
|
||||||
from .. import __package__ as base_package
|
|
||||||
|
|
||||||
|
|
||||||
def change_render_setting(context):
|
|
||||||
preferences = get_preferences(context)
|
|
||||||
if preferences.do_change_render_setting:
|
|
||||||
bpy.context.scene.render.engine = 'CYCLES'
|
|
||||||
try:
|
|
||||||
bpy.context.scene.cycles.shading_system = True
|
|
||||||
bpy.context.scene.cycles.volume_bounces = 12
|
|
||||||
bpy.context.scene.cycles.transparent_max_bounces = 16
|
|
||||||
bpy.context.scene.cycles.volume_preview_step_rate = 10
|
|
||||||
bpy.context.scene.cycles.volume_step_rate = 10
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
try:
|
|
||||||
bpy.context.scene.eevee.use_taa_reprojection = False
|
|
||||||
bpy.context.scene.eevee.volumetric_tile_size = '2'
|
|
||||||
bpy.context.scene.eevee.volumetric_shadow_samples = 128
|
|
||||||
bpy.context.scene.eevee.volumetric_samples = 256
|
|
||||||
bpy.context.scene.eevee.use_volumetric_shadows = True
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
def select_object(target_obj):
|
|
||||||
for obj in bpy.data.objects:
|
|
||||||
obj.select_set(False)
|
|
||||||
|
|
||||||
target_obj.select_set(True)
|
|
||||||
bpy.context.view_layer.objects.active = target_obj
|
|
||||||
|
|
||||||
|
|
||||||
def progress_bar(self, context):
|
|
||||||
row = self.layout.row()
|
|
||||||
row.progress(
|
|
||||||
factor=context.window_manager.bioxelnodes_progress_factor,
|
|
||||||
type="BAR",
|
|
||||||
text=context.window_manager.bioxelnodes_progress_text
|
|
||||||
)
|
|
||||||
row.scale_x = 2
|
|
||||||
|
|
||||||
|
|
||||||
def progress_update(context, factor, text=""):
|
|
||||||
context.window_manager.bioxelnodes_progress_factor = factor
|
|
||||||
context.window_manager.bioxelnodes_progress_text = text
|
|
||||||
|
|
||||||
|
|
||||||
def get_preferences(context):
|
|
||||||
return context.preferences.addons[base_package].preferences
|
|
||||||
|
|
||||||
|
|
||||||
def get_cache_dir(context):
|
|
||||||
preferences = get_preferences(context)
|
|
||||||
cache_path = Path(preferences.cache_dir, 'VDBs')
|
|
||||||
cache_path.mkdir(parents=True, exist_ok=True)
|
|
||||||
return str(cache_path)
|
|
||||||
@@ -1,23 +0,0 @@
|
|||||||
import bpy
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
class BioxelNodesPreferences(bpy.types.AddonPreferences):
|
|
||||||
bl_idname = __package__
|
|
||||||
|
|
||||||
cache_dir: bpy.props.StringProperty(
|
|
||||||
name="Set Cache Directory",
|
|
||||||
subtype='DIR_PATH',
|
|
||||||
default=str(Path(Path.home(), '.bioxelnodes'))
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
do_change_render_setting: bpy.props.BoolProperty(
|
|
||||||
name="Change Render Setting",
|
|
||||||
default=True,
|
|
||||||
) # type: ignore
|
|
||||||
|
|
||||||
def draw(self, context):
|
|
||||||
layout = self.layout
|
|
||||||
|
|
||||||
layout.label(text="Configuration")
|
|
||||||
layout.prop(self, 'cache_dir')
|
|
||||||
layout.prop(self, "do_change_render_setting")
|
|
||||||
@@ -1,5 +0,0 @@
|
|||||||
import bpy
|
|
||||||
|
|
||||||
class BIOXELNODES_Series(bpy.types.PropertyGroup):
|
|
||||||
id: bpy.props.StringProperty() # type: ignore
|
|
||||||
label: bpy.props.StringProperty() # type: ignore
|
|
||||||
@@ -1,32 +0,0 @@
|
|||||||
from pathlib import Path
|
|
||||||
import shutil
|
|
||||||
|
|
||||||
|
|
||||||
def copy_to_dir(source_path, dir_path, new_name=None, exist_ok=True):
|
|
||||||
source = Path(source_path)
|
|
||||||
target = Path(dir_path)
|
|
||||||
|
|
||||||
# Check if the source exists
|
|
||||||
if not source.exists():
|
|
||||||
raise FileNotFoundError
|
|
||||||
|
|
||||||
# Check if the target exists
|
|
||||||
if not target.exists():
|
|
||||||
target.mkdir(parents=True, exist_ok=True)
|
|
||||||
|
|
||||||
target_path = target / new_name if new_name else target / source.name
|
|
||||||
# If source is a file, copy it to the target directory
|
|
||||||
if source.is_file():
|
|
||||||
try:
|
|
||||||
shutil.copy(source, target_path)
|
|
||||||
except shutil.SameFileError:
|
|
||||||
if exist_ok:
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
raise shutil.SameFileError
|
|
||||||
# If source is a directory, copy its contents to the target directory
|
|
||||||
elif source.is_dir():
|
|
||||||
shutil.copytree(source, target_path, dirs_exist_ok=exist_ok)
|
|
||||||
|
|
||||||
if not target_path.exists():
|
|
||||||
raise Exception
|
|
||||||
15
build.py
@@ -16,7 +16,8 @@ required_packages = ["SimpleITK==2.3.1",
|
|||||||
"pyometiff==1.0.0",
|
"pyometiff==1.0.0",
|
||||||
"mrcfile==1.5.1",
|
"mrcfile==1.5.1",
|
||||||
"h5py==3.11.0",
|
"h5py==3.11.0",
|
||||||
"transforms3d==0.4.2"]
|
"transforms3d==0.4.2",
|
||||||
|
"tifffile==2024.7.24"]
|
||||||
|
|
||||||
|
|
||||||
platforms = {"windows-x64": Platform(pypi_suffix="win_amd64",
|
platforms = {"windows-x64": Platform(pypi_suffix="win_amd64",
|
||||||
@@ -40,8 +41,8 @@ def run_python(args: str):
|
|||||||
|
|
||||||
|
|
||||||
def build_extension(platform: Platform, python_version="3.11") -> None:
|
def build_extension(platform: Platform, python_version="3.11") -> None:
|
||||||
wheel_dirpath = Path("./bioxelnodes/wheels")
|
wheel_dirpath = Path("./src/bioxelnodes/wheels")
|
||||||
toml_filepath = Path("bioxelnodes/blender_manifest.toml")
|
toml_filepath = Path("./src/bioxelnodes/blender_manifest.toml")
|
||||||
scipy_ndimage_dirpath = Path("./scipy_ndimage", platform.blender_tag)
|
scipy_ndimage_dirpath = Path("./scipy_ndimage", platform.blender_tag)
|
||||||
|
|
||||||
# download required_packages
|
# download required_packages
|
||||||
@@ -59,7 +60,7 @@ def build_extension(platform: Platform, python_version="3.11") -> None:
|
|||||||
f.name.replace("universal2", "arm64")))
|
f.name.replace("universal2", "arm64")))
|
||||||
|
|
||||||
for ndimage_filepath in scipy_ndimage_dirpath.iterdir():
|
for ndimage_filepath in scipy_ndimage_dirpath.iterdir():
|
||||||
to_filepath = Path("./bioxelnodes/bioxel/scipy", ndimage_filepath.name)
|
to_filepath = Path("./src/bioxelnodes/bioxel/scipy", ndimage_filepath.name)
|
||||||
shutil.copy(ndimage_filepath, to_filepath)
|
shutil.copy(ndimage_filepath, to_filepath)
|
||||||
|
|
||||||
# Load the TOML file
|
# Load the TOML file
|
||||||
@@ -70,12 +71,6 @@ def build_extension(platform: Platform, python_version="3.11") -> None:
|
|||||||
manifest["wheels"] = [f"./wheels/{f.name}"
|
manifest["wheels"] = [f"./wheels/{f.name}"
|
||||||
for f in wheel_dirpath.glob('*.whl')]
|
for f in wheel_dirpath.glob('*.whl')]
|
||||||
|
|
||||||
# build = tomlkit.table(True)
|
|
||||||
# generated = tomlkit.table()
|
|
||||||
# generated["platforms"] = [platform.blender_tag]
|
|
||||||
# generated["wheels"] = [f"./wheels/{f.name}"
|
|
||||||
# for f in wheel_dirpath.glob('*.whl')]
|
|
||||||
|
|
||||||
# build.append('generated', generated)
|
# build.append('generated', generated)
|
||||||
# manifest.append('build', build)
|
# manifest.append('build', build)
|
||||||
|
|
||||||
|
|||||||
99
docs/SARS-Cov-2.md
Normal file
@@ -0,0 +1,99 @@
|
|||||||
|
# SARS-Cov-2
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
In this tutorial, we'll make a structural visualzation of the SARS-Cov-2 Virus. Please see [step by step](step_by_step.md) to make sure you have a basic understanding of how to use the addon, and make sure the addon version is v1.0.2 or higher!
|
||||||
|
|
||||||
|
Research data is from paper published in September 2020 in Cell Press ([https://www.cell.com/cell/fulltext/S0092-8674(20)31159-4](<https://www.cell.com/cell/fulltext/S0092-8674(20)31159-4>)) by Li Sai team at Tsinghua University. This research revealed the molecular assembly of the authentic SARS-CoV-2 virus using cryoelectron tomography (cryo-ET) and subtomogram averaging (STA). Here is an official video of this study
|
||||||
|
|
||||||
|
<iframe width="560" height="315" src="https://www.youtube.com/embed/JSKl5VY4k5w?si=t4LhrCRebxrX_Saf&controls=0" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>
|
||||||
|
|
||||||
|
"It remains enigmatic how coronaviruses pack the ∼30-kb RNA into the ∼80-nm-diameter viral lumen. Are the RNPs ordered relative to each other to avoid RNA entangling, knotting, or even damage or are they involved in virus assembly?" Three assembly classes were proposed in the paper, we will make a visualzation of the "eggs-in-a-nest"-shaped RNPs assembly, based on real cryo-ET data.
|
||||||
|
|
||||||
|
## Download and import data
|
||||||
|
|
||||||
|
The data of this research is hosted on EMDB ([https://www.ebi.ac.uk/emdb](https://www.ebi.ac.uk/emdb)), and the website of SARS-Cov-2 virus is [here](https://www.ebi.ac.uk/emdb/EMD-30430). Open the page, find "Download" button and select the first selection "3D Volume (.map.gz)".
|
||||||
|
|
||||||
|
In case you can't download data, you can download here.
|
||||||
|
|
||||||
|
[SARS-Cov-2.map](https://drive.google.com/file/d/1LMybsmTVbwQ38_eqAx6hbZTc5p2fdcjK/view?usp=sharing)
|
||||||
|
|
||||||
|
After downloading, put it into any directory and import the data. In the import options, it is recommended to adjust the Bioxel Size to 5, in order to reduce the shape of the data to increase the speed of calculation and rendering. If you are confident in the performance of your hardware, you can leave the original 2.72 unchanged. After adjusting the Bioxel Size, the shape of the converted data will be calculated below based on the current bioxel size. The amount of data will increase exponentially with larger shape, and may result in OOM (out of memory).
|
||||||
|
|
||||||
|
## Cutout the Virus
|
||||||
|
|
||||||
|
After importing, you can get the virus by creating and connecting nodes and setting parameters as shown below
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Place the light and the camera in the way as below. The backlight is a area light, color white, intensity 500W, spread 90°, and inside the virus, place a point light, color `FFD08D`, intensity 5W. Camera focal length is 200 mm. the background color is pure black. The Look of the color management is High Contrast.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Continue editing the node, followed by "Set Properies" and "Set Color by Ramp 4" node, the parameters of these two nodes are set as shown in the figure, where the color part, from top to bottom, is set to `E1D0FC` (1.0), `FFE42D` (0.5), `3793FF` (0.5), `FFFF8EC` (0.1) (alpha value in parentheses). Color alpha also affects density.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
If you feel that rendering is too slow, see [here](improve_performance.md) for suggestions
|
||||||
|
|
||||||
|
## Color the RNPs alone
|
||||||
|
|
||||||
|
The current rendering is pretty good, but considering that my goal is to make it clear how the RNPs are arranged within the virus, the RNPs should be colored differently than the others. So how to color the RNPs alone?
|
||||||
|
|
||||||
|
First of all, we have to separate the RNPs. The value of RNPs is between the membrane and S protein, so it is very difficult to separate. The good thing is that the virus is spherical, we can separate the RNPs part of the virus from the rest by using a sphere cutter. To do so, in the Geometry Nodes panel menu of the container, click **Bioxel Nodes > Add a Cutter > Sphere Cutter**, and adjust the scale and position of the newly created sphere object named "Sphere_Cutter" appropriately, so that it can just separate the internal RNPs and membrane of the virus.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
The process of changing the position of the "Sphere_Cutter" object can be very laggy, then you can turn off "With Surface" and do it in Slice Viewer mode. Once you are happy with it, you can restore the settings.
|
||||||
|
|
||||||
|
Then we turn on Invert in the "Sphere Cutter" node and you can see the opposite result. This way we've managed to separate the two, next let's color them separately and finally merge them with Join Component. Create and join the nodes as shown below, where the second Cutter node is copied and the color value in Set Color is `FFDDFE` (0.5). If all is well, the rendering should look like below.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
At this point, the rendering of a SARS-Cov-2 virus is complete!
|
||||||
|
|
||||||
|
## Render RNPs and Membranes
|
||||||
|
|
||||||
|
Realistic rendering is great, but Non-photorealistic rendering (NPR) is better for presenting structural information. So I will render ultrastructure of the RNPs assembly with toon shader.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
The silhouettes of RNPs uses Blender's Line Art feature, but Line Art must be based on a mesh. Select the nodes that are extruding the RNPs (i.e., the "Cut" nodes that are in charge of cutting out the RNPs), and right-click, click **Bioxel Nodes > Extract Mesh**. This gives you a new mesh object of the RNPs.
|
||||||
|
|
||||||
|
Extract the membrane mesh in the same way.
|
||||||
|
|
||||||
|
The membrane has a much higher value than the rest of the virus, so let's create a new "Cutout" node, also after the ReCenter, but with the threshold adjusted to 0.13. After the Cutout, connect to the "To Surface" node and parameterize it as shown in the figure, noting that the Remove Island Threshold is set to 1000 so that any fragments smaller than 1000 pts be eliminated. Then select the "To Surface" node, and perform the operation of extracting the mesh.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
The mesh of the membrane needs to be cut in half, this can be done with the Box Trim brush in sculpt mode or with the Boolean modifier. the mesh of the RNPs may also need some repair work, I cleaned up some of the broken surfaces of the RNPs. all the meshes are now ready to be used.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
For Line Art, please see the video tutorial from Blender Secrets.
|
||||||
|
|
||||||
|
<iframe width="560" height="315" src="https://www.youtube.com/embed/aIWdBq7-ias?si=CrcStx5VVJBwDpzu&controls=0" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>
|
||||||
|
|
||||||
|
Finally, an infographic on the ultrastructure of RNPs is done!
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
If you have trouble following the docs, you can download the project files.
|
||||||
|
|
||||||
|
[SARS-Cov-2.zip](https://drive.google.com/file/d/15GpIoIjVAE-Jr98zWo7oupuk1KfRVPmk/view?usp=sharing)
|
||||||
|
|
||||||
|
## Homework
|
||||||
|
|
||||||
|
Other data relevant to the study are provided in the paper, you could try visualizing them.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Go to the official EMDB website https://www.ebi.ac.uk/emdb/ and enter the EMD number in the search box to get them.
|
||||||
|
|
||||||
|
The Electron Microscopy Data Bank (EMDB) is a public repository for cryogenic-sample Electron Microscopy (cryoEM) volumes and representative tomograms of macromolecular complexes and subcellular structures. If you want to get data on other virus, just enter their name in the search box. For example, hepatitis B virus, enter it into the search box, then select Virus in the Sample Type on the left side, and download the data you need (it is better to find the corresponding paper to clarify the data information).
|
||||||
|
|
||||||
|
**Try to visualize hepatitis B virus.**
|
||||||
|
|
||||||
|

|
||||||
118
docs/advanced_usage.md
Normal file
@@ -0,0 +1,118 @@
|
|||||||
|
# Advanced Usage
|
||||||
|
|
||||||
|
## Concepts
|
||||||
|
|
||||||
|
In the previous section, I mentioned some key concepts, I'll describe them in detail so that you can better understand the following operations.
|
||||||
|
|
||||||
|
### Container
|
||||||
|
|
||||||
|
First is the **Container**, it is the easiest to describe, you can think of it as a scene, or a solver container. it is the stage that carries all stuffs.
|
||||||
|
|
||||||
|
### Layer
|
||||||
|
|
||||||
|
Second is the **Layer**, which represents different fields of data under the same location in container of the same subject. Just as view layers in any map app, one as roads, one as satellite view, while in Bioxel, one as MRI, one as CT, one as electrical signals, etc. Some layers are for material, such as color, density, etc., we call them **Material Layer**. One container contains different layers of one subject each. Any external data imported into the container is converted into the layer.
|
||||||
|
|
||||||
|
### Component
|
||||||
|
|
||||||
|
Finally, the **Component**, which is a fragment of **Material Layer**. Multiple materials (Matters) may exist within the same area with different meanings of presentation, like objects in blender, each representing a part of the scene. Bioxel Nodes provides a series of nodes and tools that allow the user to create material layers (Components) based on the other layers (serve as source data), In the previous section, we used the CT data (layer) to make the material (matter) of skull, this process known as materialization.
|
||||||
|
|
||||||
|
## Attache the Muscles to the Skull
|
||||||
|
|
||||||
|
Multiple components can exist in one container, just as multiple objects can exist in one scene. They can be controlled individually or combined with each other to build complex scenes. Next, let's try cutouting muscle from the VHP anatomical images and attaching the muscles to the skull that created in the previous section. The anatomical images are very large, so I've cropped and scaled them down to 1/4 of their original size for ease of use. Download the file and unzip it into a new directory.
|
||||||
|
|
||||||
|
[VHP_M_AnatomicalImages_Head.zip](https://drive.google.com/file/d/164AjQX0tmgUpWZlleJ5i1IKHco4ytDu2/view?usp=drive_link)
|
||||||
|
|
||||||
|
You can also download the data directly from the official website, [https://data.lhncbc.nlm.nih.gov/public/Visible-Human/Male-Images/PNG_format/head/index.html](https://data.lhncbc.nlm.nih.gov/public/Visible-Human/Male-Images/PNG_format/head/index.html)
|
||||||
|
|
||||||
|
First import the CT data as described in [step by step](step_by_step.md). Then import the anatomical images, but unlike before, this time from the Geometry Nodes menu of the existing container (not the top menu), click **Bioxel Nodes > Import Volumetric Data (Init) > as Color**, locate to the unzipped folder, and select any .png file (do not multi-select, and do not select the folder). This way the imported layer will be in the same container as the previous CT data. You can also import by dragging it into the container geometry node interface instead of the 3D Viewport interface. Also note that **as Color** (not as Scalar) is clicked because the anatomical images are RGB data.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Let's take more time in the import options this time 🫡, first let's talk about what the **Original Spacing** is in the options, you can simply think of it as the original "pixel size" of a image, since the volume is 3D, its pixels (i.e., voxels) are cuboids with a length, width, and height rather than 2D rectangles. And their length, width, and height size is the **Spacing**.
|
||||||
|
|
||||||
|
Normally, Bioxel Nodes can read the spacing from the data header, and you don't need to fill in the value, but there is no record of that in any PNG image, so you need to input it manually. in the official description of the VHP, it is mentioned that each pixel of the anatomical images corresponds to the size of .33mm, and the thickness of each slice is 1mm, so the Original Spacing of the anatomical images should be set to `(0.33, 0.33, 1)`.
|
||||||
|
|
||||||
|
However, the file I provided is scaled by 1/4 for each image, so the X and Y axes need to be multiplied by 4. So the final Original Spacing should be set to `(1.33, 1.33, 1)`, which should be understandable, right? (If you are importing an official file, then it should still be set to `(0.33, 0.33, 1)`)
|
||||||
|
Layer Name should be set to "Anatomical Images", other options remain default, click Ok.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
After importing, connect the "Cutout by Hue" node (Add > Bioxel Nodes > Component > Cutout by Hue) after the newly created "Fetch Layer" node, and turn on the Slice Viewer mode preview, as shown below.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
In addition to spacing, the Dicom file also records the subject's position and orientation, but those information is apparently not recorded in PNG format. This results in a mismatch between the CT data and the anatomical images in position. We need to manually align this two layers: First, set the threshold of the "Cutout by Threshold" node to 0.2 so that the full head can be cutouted and displayed. Then add "ReCenter" node before "Cutout by Hue" node and "Cutout by Threshold" node each, then add Locater to the "anatomical images" workflow (see [step by step](step_by_step.md) for details), and finally add "Join Component" node (Add > Bioxel Nodes > Component > Join Component), and connect them as shown below, with the parameters set as shown.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
The "Join Component" node allows you to merge two components together. Move and rotate the Locater object so that the two components are exactly the same in position. If you're having trouble matching them up, here's the exact transform values to fill in the Locater object properties.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Next let's cutout the muscles individually.
|
||||||
|
|
||||||
|
First let's talk about the "Cutout by Hue" node. Literally, it cutout regions by hue of color. If you select this node and press "M" key to temporarily mute it, you'll see that the data becomes a cube with blue parts in Slice Viewer mode. the default value of Hue in the "Cutout by Hue" node is red, meaning that the node retains areas of the body that are close to red, and strip out the blue, which is on the opposite side of the red in the hue ring, leaving the human body parts.
|
||||||
|
|
||||||
|
If you turn down the value of Sensitivity, you'll see that the visabile part is decreasing, leaving only the part more closer to red, and the muscle happens to be red, so we can easily cutout the muscle. The color of the specimen tissue is not as vibrant as it is in the live state, and we can fine-tune the color to restore the vibrancy by using the "Set Color by Color" node (Add > Bioxel Nodes > Property > Set Color by Color).
|
||||||
|
|
||||||
|
To finally put it all together, connect and set the nodes as shown below, where the two colors in "Set Color by Ramp 2" node are `E7D2C5` and `FFEDEC`. If everything is working correctly, you'll get the rendering below.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
You can also add "Transfrom" and "Cutter" nodes at the end and all components.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
The multimodal data of VHP do not have any deformation or scaling differences, but only positional differences, so all data can be aligned easily. However, in most cases, the differences of different modalities are so great that they need to be registered first, which is not possible with Bioxel Nodes at the moment (but it is in plan). Nevertheless, you can build different components based on the same layer and combine them together. For example, based on the CT data, with the "Cutout by Range" node, you can eliminate the low value regions (fat), eliminate the high value regions (bone), and you can still get the muscle. And then attach it to the skull.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
However, the CT values of brain and muscle are very close to each other, so it is extremely difficult to separate the brain from the muscle by threshold. So how should cutout the brain?
|
||||||
|
|
||||||
|
## Cutout the Brain
|
||||||
|
|
||||||
|
We can use AI technology to realize the division of brain regions, this process is called Segmentation, and the brain regions are as label. There are many models and methods for medical image segmentation, here we recommend [Total Segmentator](https://github.com/wasserth/TotalSegmentator).
|
||||||
|
|
||||||
|
It is an out-of-the-box CT/MRI medical image segmentation Python library. You can deploy Total Segmentator locally or process your data through the official online site [https://totalsegmentator.com](https://totalsegmentator.com).
|
||||||
|
|
||||||
|
Let me demonstrate the operation, we are still using the CT data from VHP. Click on "Drop DICOM.zip or ..." and select the VHP_M_CT_Head.zip file. Enter "brain" below "Selected task", press Enter to confirm. Finally click "Process data" button.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Wait for the uploading and processing, after all, you will be redirected to the following page, click "Download results" button, download and unzip segmentations.zip to any folder directory.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
In case you can't get the brain segment (label), you can download it directly (without unzipping it).
|
||||||
|
|
||||||
|
[VHP_M_CT_Head_Brain.nii.gz](https://drive.google.com/file/d/1KGiZ3G11YLXkGszSrvWQ9kTF4TMpxUPd/view?usp=drive_link)
|
||||||
|
|
||||||
|
Next we import the brain segment (label) into the container in the same way as we imported the anatomical imagess(under the same container), but this time selecting **as Label**, with the Layer Name set to "Brain", and leaving the other options as default. Once the import is complete, a new "Fetch Layer" node will be added. Create and set the node as shown below, and you should be able to see the surface of brain.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
Do you see the "step" like shape on surface? this is due to the low precision of the AI segmentation results. So How to addressing the issue? Select the Fetch Layer node of the brain label, right-click to pop up menu, click **Bioxel Nodes > Resample Value**, in the dialog box, change the "Smooth Size" to 3 or higher (the higher it is, the slower the calculation will be), click OK and wait for the calculation to finish. When it is done, a new layer will be added, replace the original one, is it smoother? The following figure lists different levels of smoothing.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
The smoother it is, the more comfortable it looks, but it also means that more details are lost and the calculation is slower.
|
||||||
|
|
||||||
|
But if you cut the brain, you will find that the cross-section is homogeneous, this is because the label type layer is does not contain any modality data itself. Bioxel Nodes provides a function to fill in any layer values by label, if you set the value below brain tissue in the non-brain region of CT data, then any "Cutout" node will only keep the brain region.
|
||||||
|
|
||||||
|
The operation is like this, select the "Fetch Layer" node of the CT data, right-click to pop up the menu, click **Bioxel Nodes > Fill Value by Label**, in the dialog box, Label is set to the newly imported brain label layer, Fill Value is set to -100 (as long as it is less than the Hu value of the brain tissue), click OK. wait for the data processing, a new data layer will be added to the container when it is finished.
|
||||||
|
|
||||||
|
As shown in the following figure, connect the nodes and set the node parameters, in which the two colors of "Set Color by Ramp 2" node are `B26267` and `EBC5B7`, and the "Cutout by Range" and "Set Color by Ramp 2" node are changed to "Value" mode (the default is "Factor" mode), because the maximum and minimum values of the brain are too close to each other and difficult to adjust in "Factor" mode. To make the brain tissue look more transparent, I add the "Set Properties" (Add > Bioxel Nodes > Property > Set Properties) node at the end and changed the "Density" to 0.5.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
(Still feel not smooth enough? It's really hard to get rid of "step" surface completely at the moment)
|
||||||
|
|
||||||
|
Now the brain is done. Finally we combine all together and you get something like the following.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
🤗 If you've followed the docs up to this point, give yourself a round of applause, you've mastered this addon! If you have trouble following the docs instruction, you can download the project files to understand how the nodes work.
|
||||||
|
|
||||||
|
[VHP.zip](https://drive.google.com/file/d/1EHB7sxNSvJNwmoTfFnpyOSsm2M_2Y0hO/view?usp=sharing)
|
||||||
|
|
||||||
|
AI segmentation is an essential part of volumetric data visualization. Therefore, the future Bioxel Nodes will integrate some commonly used AI segmentation models, so that users can complete the volumetric data visualization in place!
|
||||||
BIN
docs/assets/SARS-Cov-2/image-1.png
Normal file
|
After Width: | Height: | Size: 1.0 MiB |
BIN
docs/assets/SARS-Cov-2/image-10.png
Normal file
|
After Width: | Height: | Size: 29 KiB |
BIN
docs/assets/SARS-Cov-2/image-11.png
Normal file
|
After Width: | Height: | Size: 1.8 MiB |
BIN
docs/assets/SARS-Cov-2/image-2.png
Normal file
|
After Width: | Height: | Size: 246 KiB |
BIN
docs/assets/SARS-Cov-2/image-3.png
Normal file
|
After Width: | Height: | Size: 963 KiB |
BIN
docs/assets/SARS-Cov-2/image-4.png
Normal file
|
After Width: | Height: | Size: 354 KiB |
BIN
docs/assets/SARS-Cov-2/image-5.png
Normal file
|
After Width: | Height: | Size: 278 KiB |
BIN
docs/assets/SARS-Cov-2/image-6.png
Normal file
|
After Width: | Height: | Size: 366 KiB |
BIN
docs/assets/SARS-Cov-2/image-7.png
Normal file
|
After Width: | Height: | Size: 591 KiB |
BIN
docs/assets/SARS-Cov-2/image-8.png
Normal file
|
After Width: | Height: | Size: 184 KiB |
BIN
docs/assets/SARS-Cov-2/image-9.png
Normal file
|
After Width: | Height: | Size: 413 KiB |
BIN
docs/assets/SARS-Cov-2/image.png
Normal file
|
After Width: | Height: | Size: 1.8 MiB |
BIN
docs/assets/advanced_usage/image-1.png
Normal file
|
After Width: | Height: | Size: 41 KiB |
BIN
docs/assets/advanced_usage/image-10.png
Normal file
|
After Width: | Height: | Size: 185 KiB |
BIN
docs/assets/advanced_usage/image-11.png
Normal file
|
After Width: | Height: | Size: 326 KiB |
BIN
docs/assets/advanced_usage/image-12.png
Normal file
|
After Width: | Height: | Size: 288 KiB |
BIN
docs/assets/advanced_usage/image-13.png
Normal file
|
After Width: | Height: | Size: 412 KiB |
BIN
docs/assets/advanced_usage/image-2.png
Normal file
|
After Width: | Height: | Size: 337 KiB |
BIN
docs/assets/advanced_usage/image-3.png
Normal file
|
After Width: | Height: | Size: 328 KiB |
BIN
docs/assets/advanced_usage/image-4.png
Normal file
|
After Width: | Height: | Size: 505 KiB |
BIN
docs/assets/advanced_usage/image-5.png
Normal file
|
After Width: | Height: | Size: 340 KiB |
BIN
docs/assets/advanced_usage/image-6.png
Normal file
|
After Width: | Height: | Size: 369 KiB |
BIN
docs/assets/advanced_usage/image-7.png
Normal file
|
After Width: | Height: | Size: 343 KiB |
BIN
docs/assets/advanced_usage/image-8.png
Normal file
|
After Width: | Height: | Size: 338 KiB |
BIN
docs/assets/advanced_usage/image-9.png
Normal file
|
After Width: | Height: | Size: 351 KiB |
BIN
docs/assets/advanced_usage/image.png
Normal file
|
After Width: | Height: | Size: 76 KiB |
|
Before Width: | Height: | Size: 2.5 MiB After Width: | Height: | Size: 3.5 MiB |
|
Before Width: | Height: | Size: 9.1 KiB |
|
Before Width: | Height: | Size: 148 KiB |
|
Before Width: | Height: | Size: 423 KiB |
|
Before Width: | Height: | Size: 8.6 KiB |
|
Before Width: | Height: | Size: 19 KiB |
|
Before Width: | Height: | Size: 5.1 MiB |
|
Before Width: | Height: | Size: 213 KiB |
|
Before Width: | Height: | Size: 333 KiB |
|
Before Width: | Height: | Size: 96 KiB |
|
Before Width: | Height: | Size: 63 KiB |
|
Before Width: | Height: | Size: 549 KiB |
|
Before Width: | Height: | Size: 770 KiB |
|
Before Width: | Height: | Size: 43 KiB |
|
Before Width: | Height: | Size: 917 KiB |
|
Before Width: | Height: | Size: 30 KiB |
BIN
docs/assets/improve_performance/image-1.png
Normal file
|
After Width: | Height: | Size: 1.3 MiB |
BIN
docs/assets/improve_performance/image-2.png
Normal file
|
After Width: | Height: | Size: 903 KiB |
BIN
docs/assets/improve_performance/image.png
Normal file
|
After Width: | Height: | Size: 126 KiB |
|
Before Width: | Height: | Size: 40 KiB After Width: | Height: | Size: 40 KiB |
|
Before Width: | Height: | Size: 22 KiB |
|
Before Width: | Height: | Size: 174 KiB |
|
Before Width: | Height: | Size: 45 KiB |
|
Before Width: | Height: | Size: 66 KiB |
|
Before Width: | Height: | Size: 71 KiB |
|
Before Width: | Height: | Size: 77 KiB |
|
Before Width: | Height: | Size: 81 KiB |
|
Before Width: | Height: | Size: 75 KiB |
|
Before Width: | Height: | Size: 45 KiB |
|
Before Width: | Height: | Size: 140 KiB |
|
Before Width: | Height: | Size: 62 KiB |
|
Before Width: | Height: | Size: 57 KiB |
|
Before Width: | Height: | Size: 96 KiB |
|
Before Width: | Height: | Size: 86 KiB |
|
Before Width: | Height: | Size: 53 KiB |
|
Before Width: | Height: | Size: 41 KiB |
|
Before Width: | Height: | Size: 40 KiB |
|
Before Width: | Height: | Size: 75 KiB |
|
Before Width: | Height: | Size: 75 KiB |
|
Before Width: | Height: | Size: 75 KiB |
|
Before Width: | Height: | Size: 59 KiB |
BIN
docs/assets/step_by_step/image-1.png
Normal file
|
After Width: | Height: | Size: 177 KiB |
BIN
docs/assets/step_by_step/image-2.png
Normal file
|
After Width: | Height: | Size: 278 KiB |
BIN
docs/assets/step_by_step/image-3.png
Normal file
|
After Width: | Height: | Size: 284 KiB |
BIN
docs/assets/step_by_step/image-4.png
Normal file
|
After Width: | Height: | Size: 311 KiB |
BIN
docs/assets/step_by_step/image-5.png
Normal file
|
After Width: | Height: | Size: 369 KiB |
BIN
docs/assets/step_by_step/image-6.png
Normal file
|
After Width: | Height: | Size: 350 KiB |
BIN
docs/assets/step_by_step/image-7.png
Normal file
|
After Width: | Height: | Size: 320 KiB |
BIN
docs/assets/step_by_step/image-8.png
Normal file
|
After Width: | Height: | Size: 314 KiB |
BIN
docs/assets/step_by_step/image-9.png
Normal file
|
After Width: | Height: | Size: 83 KiB |