nan
7dec3b7ca4
- Implemented `cutout_depth_surface.py` for face separation and connected thickness calculations for depth surface assets. - Created `cutout_depth_surface_layout.py` to arrange nodes for staggered calculation branches in depth surface processing. - Authored `geometry_layouts.py` to define layouts for various geometry assets, including image cutouts and depth planes. - Introduced `node_layout.py` for shared presentation rules, facilitating node arrangement and validation in geometry-node layouts.
210 lines
7.0 KiB
Python
210 lines
7.0 KiB
Python
"""Arrange staggered calculation branches on both sides of the geometry path."""
|
|
|
|
from node_layout import arrange_nodes
|
|
|
|
# Main geometry advances left to right. Local fields grow to either side of it;
|
|
# a long calculation may start before the geometry input or span several stages.
|
|
POSITIONS = {
|
|
"Position": (0, 260),
|
|
"Store Named Attribute": (0, 0),
|
|
"Store Named Attribute.001": (260, 0),
|
|
"Store Named Attribute.002": (520, 0),
|
|
"Store Named Attribute.003": (780, 0),
|
|
"Split Edges": (1060, 0),
|
|
"Switch": (1360, 0),
|
|
# Face sampling stays near its stores; selection forms a dense field cluster.
|
|
"Named Attribute": (-400, -320),
|
|
"Evaluate on Domain": (-180, -320),
|
|
"Image Texture": (20, -320),
|
|
"Named Attribute.003": (260, 260),
|
|
"Named Attribute.001": (-1090, 1060),
|
|
"Separate XYZ": (-870, 1060),
|
|
"Named Attribute.002": (-390, 590),
|
|
"Corners of Edge": (-1090, 910),
|
|
"Face of Corner": (-910, 910),
|
|
"Corners of Edge.001": (-1090, 770),
|
|
"Face of Corner.001": (-910, 770),
|
|
"Evaluate at Index": (-170, 750),
|
|
"Evaluate at Index.002": (-170, 610),
|
|
"Evaluate at Index.001": (-710, 1060),
|
|
"Evaluate at Index.003": (-710, 920),
|
|
"Vector Math": (10, 710),
|
|
"Math": (-530, 1040),
|
|
"Math.001": (-350, 1040),
|
|
"Math.002": (-170, 1040),
|
|
"Math.003": (-350, 830),
|
|
"Math.004": (10, 1040),
|
|
"Math.005": (190, 1040),
|
|
"Math.006": (190, 790),
|
|
"Math.007": (370, 980),
|
|
"Math.008": (190, 620),
|
|
"Math.009": (550, 910),
|
|
"Edge Neighbors": (370, 590),
|
|
"Math.010": (550, 590),
|
|
"Math.011": (730, 810),
|
|
"Evaluate on Domain.001": (730, 670),
|
|
"Evaluate on Domain.002": (910, 670),
|
|
"Math.012": (1090, 670),
|
|
"Math.013": (1360, 280),
|
|
# Cleanup is a separate compact cluster before projection.
|
|
"Face Area": (1460, 780),
|
|
"Mesh Island": (1460, 580),
|
|
"Accumulate Field": (1680, 780),
|
|
"Math.014": (1900, 580),
|
|
"Delete Geometry": (2120, 0),
|
|
# Camera-ray reconstruction runs below; scene projection runs above.
|
|
"Named Attribute.004": (1290, -440),
|
|
"Image Texture.001": (1510, -440),
|
|
"Separate XYZ.001": (1740, -440),
|
|
"Named Attribute.005": (1510, -710),
|
|
"Separate XYZ.002": (1740, -710),
|
|
"Math.017": (1960, -440),
|
|
"Vector Math.001": (2180, -440),
|
|
"Named Attribute.006": (1740, -940),
|
|
"Math.015": (1510, -940),
|
|
"Math.016": (1960, -940),
|
|
"Switch.001": (2600, -240),
|
|
"Store Named Attribute.004": (2920, 0),
|
|
"Named Attribute.007": (2340, 1080),
|
|
"Vector Math.002": (2560, 1080),
|
|
"Separate XYZ.003": (2740, 1080),
|
|
"Math.018": (3100, 730),
|
|
"Math.019": (3280, 730),
|
|
"Math.020": (2740, 910),
|
|
"Combine XYZ": (2920, 1040),
|
|
"Position.001": (2920, 790),
|
|
"Mix": (3100, 950),
|
|
"Separate XYZ.004": (3280, 950),
|
|
"Combine XYZ.001": (3460, 850),
|
|
"Set Position": (3870, 0),
|
|
# Short preparation fields sit above the axis; the long wall-count branch
|
|
# sits below, reading geometry back from the source-vertex store.
|
|
"Position.002": (4560, 340),
|
|
"Store Named Attribute.005": (4560, 0),
|
|
"Vector Math.003": (4780, 340),
|
|
"Set Position.001": (4780, 0),
|
|
"Index": (5010, 340),
|
|
"Store Named Attribute.006": (5010, 0),
|
|
"Edge Vertices": (3470, -610),
|
|
"Vector Math.004": (3700, -610),
|
|
"Edge Neighbors.001": (3470, -850),
|
|
"Math.021": (3700, -850),
|
|
"Attribute Statistic": (3920, -610),
|
|
"Math.022": (4150, -610),
|
|
"Math.023": (4370, -610),
|
|
"Math.024": (4590, -610),
|
|
"Math.025": (4820, -610),
|
|
"Math.026": (5040, -610),
|
|
"Store Named Attribute.007": (5310, 0),
|
|
"Repeat Input": (5620, 0),
|
|
"Extrude Mesh": (5850, 0),
|
|
"Combine XYZ.003": (5620, -340),
|
|
"Separate Geometry": (6070, 0),
|
|
"Join Geometry": (6300, -180),
|
|
"Repeat Output": (6520, 0),
|
|
"Flip Faces.001": (6740, 0),
|
|
"Join Geometry.001": (6970, 0),
|
|
"Named Attribute.010": (5620, -530),
|
|
"Math.027": (5850, -530),
|
|
"Combine XYZ.004": (6070, -530),
|
|
"Set Position.002": (6300, -320),
|
|
"Flip Faces": (6740, -320),
|
|
# Identity decoding above and position restoration below have independent,
|
|
# horizontally spread readers rather than equally tall vertical stacks.
|
|
"Named Attribute.008": (6240, 880),
|
|
"Integer Math": (6470, 880),
|
|
"Named Attribute.009": (6240, 650),
|
|
"Integer Math.001": (6470, 650),
|
|
"Position.004": (6240, 410),
|
|
"Separate XYZ.006": (6470, 410),
|
|
"Combine XYZ.002": (6670, 650),
|
|
"Set Position.003": (7410, 0),
|
|
"Merge by Distance": (7640, 0),
|
|
"Position.003": (6910, -950),
|
|
"Separate XYZ.005": (7160, -950),
|
|
"Named Attribute.011": (6900, -710),
|
|
"Math.028": (7140, -580),
|
|
"Math.029": (7380, -750),
|
|
"Combine XYZ.005": (7600, -680),
|
|
"Named Attribute.012": (7600, -370),
|
|
"Vector Math.005": (7830, -340),
|
|
"Set Position.004": (8080, 0),
|
|
"Math.030": (8310, 260),
|
|
"Switch.002": (8310, 0),
|
|
"Remove Named Attribute": (9000, 0),
|
|
"Set Shade Smooth": (9220, 0),
|
|
"Group Output": (9440, 0),
|
|
}
|
|
|
|
INPUT_REGIONS = (
|
|
(("Store Named Attribute",), (-240, 0)),
|
|
(("Switch",), (1100, -300)),
|
|
(("Image Texture",), (-180, -190)),
|
|
(("Math.002",), (-350, 1140)),
|
|
(("Math.003",), (-530, 830)),
|
|
(("Math.005",), (10, 1140)),
|
|
(("Math.013",), (1100, 320)),
|
|
(("Math.014",), (1680, 400)),
|
|
(("Image Texture.001",), (1290, -270)),
|
|
(("Math.015",), (1290, -940)),
|
|
(("Vector Math.002",), (2340, 1200)),
|
|
(("Math.018", "Math.019"), (2920, 680)),
|
|
(("Mix",), (2920, 900)),
|
|
(("Math.023",), (4150, -820)),
|
|
(("Math.028",), (6900, -540)),
|
|
(("Math.030",), (8080, 300)),
|
|
)
|
|
|
|
|
|
def _route(group, source_name, output, target_name, input_name, points):
|
|
"""Place geometry references around the local branches they serve."""
|
|
source = group.nodes[source_name].outputs[output]
|
|
target = group.nodes[target_name].inputs[input_name]
|
|
for link in list(group.links):
|
|
if link.from_socket == source and link.to_socket == target:
|
|
group.links.remove(link)
|
|
for point in points:
|
|
node = group.nodes.new("NodeReroute")
|
|
node.location = point
|
|
node.select = False
|
|
group.links.new(source, node.inputs[0])
|
|
source = node.outputs[0]
|
|
group.links.new(source, target)
|
|
|
|
|
|
def layout_depth_surface(group):
|
|
"""Keep the geometry path readable while staggering long and short branches."""
|
|
arrange_nodes(group, POSITIONS, INPUT_REGIONS)
|
|
_route(
|
|
group,
|
|
"Repeat Input",
|
|
"Sides",
|
|
"Join Geometry",
|
|
"Geometry",
|
|
((5810, -170), (6280, -170)),
|
|
)
|
|
_route(
|
|
group,
|
|
"Store Named Attribute.007",
|
|
"Geometry",
|
|
"Join Geometry.001",
|
|
"Geometry",
|
|
((5530, 150), (6950, 150)),
|
|
)
|
|
_route(
|
|
group,
|
|
"Store Named Attribute.007",
|
|
"Geometry",
|
|
"Set Position.002",
|
|
"Geometry",
|
|
((5530, -270), (6280, -270)),
|
|
)
|
|
_route(
|
|
group,
|
|
"Set Position",
|
|
"Geometry",
|
|
"Switch.002",
|
|
"True",
|
|
((4110, 240), (8290, 240)),
|
|
)
|