nan
5b2dc59f79
- Introduced `frames.py` to manage field complex calculations and their arrangement within node groups. - Added functionality to create and remove field frames based on node arrangements. - Developed `passages.py` to find unobstructed paths for connections around calculation frames. - Implemented `references.py` to route shared results on peripheral tracks, ensuring efficient distribution of outputs. - Created `sections.py` to reserve space for field calculations adjacent to geometry consumers. - Added `sources.py` to position shared calculation sources effectively beside their distribution channels. - Enhanced outline smoothing with `outline_smoothing.py` to adapt camera depth along original outlines.
86 lines
3.8 KiB
Python
86 lines
3.8 KiB
Python
"""Apply boundary-preserving smoothing within a narrow cut-edge band."""
|
|
|
|
from .nodes import evaluate_field, group_input
|
|
from .depth_surface import _math
|
|
from .smoothing import edge_boundary_field, expand_smooth
|
|
|
|
|
|
def boundary_influence(group, protected=False):
|
|
"""Fade a boundary field through two rings while excluding protected points."""
|
|
nodes, links = group.nodes, group.links
|
|
boundary = evaluate_field(group, edge_boundary_field(nodes, links), "FLOAT", "POINT")
|
|
unprotected = _math(group, "SUBTRACT", 1, protected)
|
|
cut = _math(group, "MULTIPLY", boundary, unprotected)
|
|
blur = nodes.new("GeometryNodeBlurAttribute")
|
|
blur.data_type = "FLOAT"
|
|
blur.inputs["Iterations"].default_value = 2
|
|
blur.inputs["Weight"].default_value = 1
|
|
links.new(cut, blur.inputs["Value"])
|
|
return _math(group, "MULTIPLY", _math(group, "MAXIMUM", cut, blur.outputs["Value"]), unprotected)
|
|
|
|
|
|
def smooth_boundary_depth(group, depth, protected=False):
|
|
"""Blur point depth on the current mesh and apply it only in the edge band."""
|
|
nodes, links = group.nodes, group.links
|
|
depth = evaluate_field(group, depth, "FLOAT", "POINT")
|
|
iterations = group_input(nodes, {"Boundary Smooth"}).outputs["Boundary Smooth"]
|
|
influence = boundary_influence(group, protected)
|
|
blur = nodes.new("GeometryNodeBlurAttribute")
|
|
blur.data_type = "FLOAT"
|
|
blur.inputs["Weight"].default_value = 0.5
|
|
links.new(depth, blur.inputs["Value"])
|
|
links.new(iterations, blur.inputs["Iterations"])
|
|
mix = nodes.new("ShaderNodeMix")
|
|
mix.data_type = "FLOAT"
|
|
links.new(influence, mix.inputs[0])
|
|
links.new(depth, mix.inputs[2])
|
|
links.new(blur.outputs["Value"], mix.inputs[3])
|
|
choose = nodes.new("GeometryNodeSwitch")
|
|
choose.input_type = "FLOAT"
|
|
links.new(influence, choose.inputs["Switch"])
|
|
links.new(depth, choose.inputs["False"])
|
|
links.new(mix.outputs[0], choose.inputs["True"])
|
|
enabled = nodes.new("GeometryNodeSwitch")
|
|
enabled.input_type = "FLOAT"
|
|
links.new(iterations, enabled.inputs["Switch"])
|
|
links.new(depth, enabled.inputs["False"])
|
|
links.new(choose.outputs[0], enabled.inputs["True"])
|
|
return enabled.outputs[0]
|
|
|
|
|
|
def smooth_boundary_camera(group, camera, protected=False):
|
|
"""Reconstruct blurred camera depth along each unchanged projection ray."""
|
|
nodes, links = group.nodes, group.links
|
|
axes = nodes.new("ShaderNodeSeparateXYZ")
|
|
links.new(camera, axes.inputs[0])
|
|
depth = axes.outputs["Z"]
|
|
smoothed = smooth_boundary_depth(group, depth, protected)
|
|
return reconstruct_camera_depth(group, camera, depth, smoothed)
|
|
|
|
|
|
def reconstruct_camera_depth(group, camera, depth, smoothed):
|
|
"""Move each valid camera point along its original projection ray."""
|
|
nodes, links = group.nodes, group.links
|
|
changed = _math(group, "GREATER_THAN", _math(group, "ABSOLUTE", _math(group, "SUBTRACT", smoothed, depth)), 0)
|
|
enabled = _math(group, "MULTIPLY", changed, _math(group, "GREATER_THAN", depth, 1e-8))
|
|
projected = nodes.new("ShaderNodeVectorMath")
|
|
projected.operation = "SCALE"
|
|
links.new(camera, projected.inputs[0])
|
|
links.new(_math(group, "DIVIDE", smoothed, depth), projected.inputs[3])
|
|
choose = nodes.new("GeometryNodeSwitch")
|
|
choose.input_type = "VECTOR"
|
|
links.new(enabled, choose.inputs["Switch"])
|
|
links.new(camera, choose.inputs["False"])
|
|
links.new(projected.outputs[0], choose.inputs["True"])
|
|
return choose.outputs[0]
|
|
|
|
|
|
def smooth_cut_boundary(group, geometry, original_boundary=False):
|
|
"""Blend O Smooth's pinned-boundary result through a two-ring edge mask."""
|
|
influence = boundary_influence(group, original_boundary)
|
|
# The support band depends on topology, which stays unchanged in the loop.
|
|
return expand_smooth(
|
|
group, geometry, iterations_name="Boundary Smooth", weight_name="Smooth Weight",
|
|
influence=influence, pin_sharp=False,
|
|
)
|