nan
32bcd45d7b
- Implemented `build_mesh_plane.py` to create a geometry node group for mesh planes with customizable parameters. - Created `build_shadeless.py` for a shader node group that displays color without scene lighting while preserving shadows and reflections. - Added `validate.py` to validate the integrity of the newly created node groups and their inputs. - Introduced `validate_cutout.py` to ensure the functionality of cutout node groups and their dependencies. - Developed `validate_mesh_plane.py` to test the mesh plane creation and its modifier settings.
157 lines
6.9 KiB
Python
157 lines
6.9 KiB
Python
"""Runtime Geometry Nodes for Depth Plane."""
|
|
|
|
import bpy
|
|
|
|
from ..common.node_asset import load_mesh_plane_node_group
|
|
from ..common.node_graph import group_input, interface_socket
|
|
|
|
def depth_plane_node_group():
|
|
group_name = "O Image Depth Plane"
|
|
group = bpy.data.node_groups.get(group_name)
|
|
if group is not None:
|
|
return group
|
|
if group is None:
|
|
group = bpy.data.node_groups.new(group_name, "GeometryNodeTree")
|
|
else:
|
|
group.nodes.clear()
|
|
group.interface.clear()
|
|
group.is_modifier = True
|
|
interface_socket(group, "Geometry", "INPUT", "NodeSocketGeometry")
|
|
subdivide = interface_socket(group, "Subdivide", "INPUT", "NodeSocketInt")
|
|
subdivide.default_value = 7
|
|
subdivide.min_value = 0
|
|
subdivide.max_value = 10
|
|
thickness = interface_socket(group, "Thickness", "INPUT", "NodeSocketFloat")
|
|
thickness.default_value = 0.1
|
|
thickness.min_value = 0.0
|
|
thickness.subtype = "DISTANCE"
|
|
depth_amount = interface_socket(group, "Depth Amount", "INPUT", "NodeSocketFloat")
|
|
depth_amount.default_value = 1.0
|
|
depth_amount.min_value = 0.0
|
|
depth_image = interface_socket(group, "Depth Image", "INPUT", "NodeSocketImage")
|
|
depth_image.hide_in_modifier = True
|
|
material = interface_socket(group, "Material", "INPUT", "NodeSocketMaterial")
|
|
material.hide_in_modifier = True
|
|
interface_socket(group, "Geometry", "OUTPUT", "NodeSocketGeometry")
|
|
|
|
nodes = group.nodes
|
|
links = group.links
|
|
geometry_input = group_input(
|
|
nodes,
|
|
(-1850.0, 450.0),
|
|
{"Geometry", "Subdivide", "Material"},
|
|
)
|
|
shape_input = group_input(
|
|
nodes,
|
|
(-1250.0, 850.0),
|
|
{"Thickness", "Depth Amount"},
|
|
)
|
|
image_input = group_input(nodes, (-1050.0, 50.0), {"Depth Image"})
|
|
cleanup_input = group_input(nodes, (650.0, -200.0), {"Thickness"})
|
|
group_output = nodes.new("NodeGroupOutput")
|
|
group_output.location = (1600.0, 450.0)
|
|
mesh_plane = nodes.new("GeometryNodeGroup")
|
|
mesh_plane.node_tree = load_mesh_plane_node_group()
|
|
mesh_plane.inputs["Thickness"].default_value = 0.0
|
|
mesh_plane.location = (-1550.0, 450.0)
|
|
links.new(geometry_input.outputs["Geometry"], mesh_plane.inputs["Geometry"])
|
|
links.new(geometry_input.outputs["Subdivide"], mesh_plane.inputs["Subdivide"])
|
|
links.new(geometry_input.outputs["Material"], mesh_plane.inputs["Material"])
|
|
|
|
extrude = nodes.new("GeometryNodeExtrudeMesh")
|
|
extrude.mode = "FACES"
|
|
extrude.inputs["Offset"].default_value = (0.0, 0.0, -1.0)
|
|
extrude.inputs["Individual"].default_value = False
|
|
extrude.location = (-1200.0, 450.0)
|
|
links.new(mesh_plane.outputs["Geometry"], extrude.inputs["Mesh"])
|
|
links.new(shape_input.outputs["Thickness"], extrude.inputs["Offset Scale"])
|
|
geometry = extrude.outputs["Mesh"]
|
|
|
|
uv = nodes.new("GeometryNodeInputNamedAttribute")
|
|
uv.data_type = "FLOAT_VECTOR"
|
|
uv.inputs["Name"].default_value = "UVMap"
|
|
uv.location = (-1000.0, 50.0)
|
|
texture = nodes.new("GeometryNodeImageTexture")
|
|
texture.location = (-800.0, 50.0)
|
|
links.new(image_input.outputs["Depth Image"], texture.inputs["Image"])
|
|
links.new(uv.outputs["Attribute"], texture.inputs["Vector"])
|
|
separate = nodes.new("ShaderNodeSeparateXYZ")
|
|
separate.location = (-600.0, 50.0)
|
|
links.new(texture.outputs["Color"], separate.inputs["Vector"])
|
|
raw_depth = nodes.new("ShaderNodeMath")
|
|
raw_depth.operation = "MULTIPLY"
|
|
raw_depth.location = (-350.0, 150.0)
|
|
links.new(separate.outputs["X"], raw_depth.inputs[0])
|
|
links.new(shape_input.outputs["Depth Amount"], raw_depth.inputs[1])
|
|
inward_limit = nodes.new("ShaderNodeMath")
|
|
inward_limit.operation = "MULTIPLY"
|
|
inward_limit.inputs[1].default_value = -0.98
|
|
inward_limit.location = (-350.0, -50.0)
|
|
links.new(shape_input.outputs["Thickness"], inward_limit.inputs[0])
|
|
clamp = nodes.new("ShaderNodeMath")
|
|
clamp.operation = "MAXIMUM"
|
|
clamp.location = (-100.0, 50.0)
|
|
links.new(raw_depth.outputs[0], clamp.inputs[0])
|
|
links.new(inward_limit.outputs[0], clamp.inputs[1])
|
|
has_thickness = nodes.new("ShaderNodeMath")
|
|
has_thickness.operation = "GREATER_THAN"
|
|
has_thickness.inputs[1].default_value = 1e-6
|
|
has_thickness.location = (-350.0, -250.0)
|
|
links.new(shape_input.outputs["Thickness"], has_thickness.inputs[0])
|
|
choose_depth = nodes.new("GeometryNodeSwitch")
|
|
choose_depth.input_type = "FLOAT"
|
|
choose_depth.location = (150.0, 50.0)
|
|
links.new(has_thickness.outputs[0], choose_depth.inputs["Switch"])
|
|
links.new(raw_depth.outputs[0], choose_depth.inputs["False"])
|
|
links.new(clamp.outputs[0], choose_depth.inputs["True"])
|
|
combine = nodes.new("ShaderNodeCombineXYZ")
|
|
combine.location = (350.0, 50.0)
|
|
links.new(choose_depth.outputs["Output"], combine.inputs["Z"])
|
|
front = nodes.new("FunctionNodeBooleanMath")
|
|
front.operation = "NOT"
|
|
front.location = (150.0, 300.0)
|
|
links.new(extrude.outputs["Top"], front.inputs[0])
|
|
set_position = nodes.new("GeometryNodeSetPosition")
|
|
set_position.location = (550.0, 450.0)
|
|
links.new(geometry, set_position.inputs["Geometry"])
|
|
links.new(front.outputs[0], set_position.inputs["Selection"])
|
|
links.new(combine.outputs["Vector"], set_position.inputs["Offset"])
|
|
geometry = set_position.outputs["Geometry"]
|
|
|
|
zero_thickness = nodes.new("ShaderNodeMath")
|
|
zero_thickness.operation = "LESS_THAN"
|
|
zero_thickness.inputs[1].default_value = 1e-6
|
|
zero_thickness.location = (750.0, -150.0)
|
|
links.new(cleanup_input.outputs["Thickness"], zero_thickness.inputs[0])
|
|
back_or_side = nodes.new("FunctionNodeBooleanMath")
|
|
back_or_side.operation = "OR"
|
|
back_or_side.location = (750.0, 150.0)
|
|
links.new(extrude.outputs["Top"], back_or_side.inputs[0])
|
|
links.new(extrude.outputs["Side"], back_or_side.inputs[1])
|
|
delete_selection = nodes.new("FunctionNodeBooleanMath")
|
|
delete_selection.operation = "AND"
|
|
delete_selection.location = (950.0, 50.0)
|
|
links.new(zero_thickness.outputs[0], delete_selection.inputs[0])
|
|
links.new(back_or_side.outputs[0], delete_selection.inputs[1])
|
|
delete_geometry = nodes.new("GeometryNodeDeleteGeometry")
|
|
delete_geometry.domain = "FACE"
|
|
delete_geometry.location = (1100.0, 450.0)
|
|
links.new(geometry, delete_geometry.inputs["Geometry"])
|
|
links.new(delete_selection.outputs[0], delete_geometry.inputs["Selection"])
|
|
shade_smooth = nodes.new("GeometryNodeSetShadeSmooth")
|
|
shade_smooth.domain = "FACE"
|
|
shade_smooth.inputs["Shade Smooth"].default_value = True
|
|
shade_smooth.location = (1350.0, 450.0)
|
|
links.new(delete_geometry.outputs["Geometry"], shade_smooth.inputs["Geometry"])
|
|
links.new(shade_smooth.outputs["Geometry"], group_output.inputs["Geometry"])
|
|
group.use_fake_user = True
|
|
return group
|
|
|
|
|
|
def plane_node_group(shape):
|
|
if shape == "PLANE":
|
|
return load_mesh_plane_node_group()
|
|
if shape == "DEPTH_PLANE":
|
|
return depth_plane_node_group()
|
|
raise ValueError(f"Unsupported Plane shape: {shape}")
|