Add slicer to layer when container is new (#57)
This commit is contained in:
@@ -164,4 +164,23 @@ def container_to_obj(container: Container,
|
|||||||
container_obj=container_obj,
|
container_obj=container_obj,
|
||||||
cache_dir=cache_dir)
|
cache_dir=cache_dir)
|
||||||
|
|
||||||
|
try:
|
||||||
|
layer_node = get_nodes_by_type(
|
||||||
|
node_group, "BioxelNodes_FetchLayer")[0]
|
||||||
|
center_node = add_node_to_graph("ReCenter",
|
||||||
|
node_group,
|
||||||
|
use_link=get_use_link())
|
||||||
|
|
||||||
|
output_node = get_output_node(node_group)
|
||||||
|
|
||||||
|
node_group.links.new(layer_node.outputs[0],
|
||||||
|
center_node.inputs[0])
|
||||||
|
node_group.links.new(center_node.outputs[0],
|
||||||
|
output_node.inputs[0])
|
||||||
|
|
||||||
|
move_node_to_node(center_node, layer_node, (300, 0))
|
||||||
|
bpy.ops.bioxelnodes.add_slicer('EXEC_DEFAULT')
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
return container_obj
|
return container_obj
|
||||||
|
|||||||
@@ -8,9 +8,8 @@ from pathlib import Path
|
|||||||
|
|
||||||
from ..exceptions import CancelledByUser
|
from ..exceptions import CancelledByUser
|
||||||
from ..props import BIOXELNODES_Series
|
from ..props import BIOXELNODES_Series
|
||||||
from ..bioxelutils.node import add_node_to_graph
|
|
||||||
from ..bioxelutils.common import (get_all_layer_objs, get_container_obj,
|
from ..bioxelutils.common import (get_all_layer_objs, get_container_obj,
|
||||||
get_layer_obj, get_nodes_by_type, get_output_node, is_incompatible, move_node_to_node)
|
get_layer_obj, is_incompatible)
|
||||||
from ..bioxelutils.container import (Container,
|
from ..bioxelutils.container import (Container,
|
||||||
add_layers,
|
add_layers,
|
||||||
container_to_obj)
|
container_to_obj)
|
||||||
@@ -18,8 +17,7 @@ from ..bioxel.layer import Layer
|
|||||||
from ..bioxel.parse import (DICOM_EXTS, SUPPORT_EXTS,
|
from ..bioxel.parse import (DICOM_EXTS, SUPPORT_EXTS,
|
||||||
get_ext, parse_volumetric_data)
|
get_ext, parse_volumetric_data)
|
||||||
|
|
||||||
from ..utils import (get_cache_dir, get_use_link,
|
from ..utils import (get_cache_dir, progress_update, progress_bar,
|
||||||
progress_update, progress_bar,
|
|
||||||
select_object)
|
select_object)
|
||||||
|
|
||||||
# 3rd-party
|
# 3rd-party
|
||||||
@@ -856,27 +854,6 @@ class ImportVolumetricDataDialog(bpy.types.Operator):
|
|||||||
|
|
||||||
# Change render setting for better result
|
# Change render setting for better result
|
||||||
if is_first_import:
|
if is_first_import:
|
||||||
try:
|
|
||||||
|
|
||||||
node_group = container_obj.modifiers[0].node_group
|
|
||||||
layer_node = get_nodes_by_type(
|
|
||||||
node_group, "BioxelNodes_FetchLayer")[0]
|
|
||||||
center_node = add_node_to_graph("ReCenter",
|
|
||||||
node_group,
|
|
||||||
use_link=get_use_link())
|
|
||||||
|
|
||||||
output_node = get_output_node(node_group)
|
|
||||||
|
|
||||||
node_group.links.new(layer_node.outputs[0],
|
|
||||||
center_node.inputs[0])
|
|
||||||
node_group.links.new(center_node.outputs[0],
|
|
||||||
output_node.inputs[0])
|
|
||||||
|
|
||||||
move_node_to_node(center_node, layer_node, (300, 0))
|
|
||||||
bpy.ops.bioxelnodes.add_slicer('EXEC_DEFAULT')
|
|
||||||
except:
|
|
||||||
self.report({"WARNING"}, "Fail to create slicer.")
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
bpy.ops.bioxelnodes.render_setting_preset('EXEC_DEFAULT',
|
bpy.ops.bioxelnodes.render_setting_preset('EXEC_DEFAULT',
|
||||||
preset="balance")
|
preset="balance")
|
||||||
|
|||||||
Reference in New Issue
Block a user