chore: bump version to 1.0.1 and update related files
This commit is contained in:
@@ -4,11 +4,8 @@ import bpy
|
||||
import numpy as np
|
||||
import pytest
|
||||
|
||||
from anyimage.common.object import modifier_input_identifier, set_modifier_input
|
||||
from tests.support.depth_surface import surface, evaluated
|
||||
from tests.tools.nodes.test_boundary_smoothing import diagonal_surface
|
||||
from tools.nodes.common.cutout_boundary import remove_boundary_triangles
|
||||
from tools.nodes.groups.image_depth_cutout import build_normal_direction, build_normal_weight
|
||||
|
||||
|
||||
def profile_values(obj):
|
||||
@@ -32,23 +29,6 @@ def vertex_uv(obj):
|
||||
result.to_mesh_clear()
|
||||
|
||||
|
||||
def test_boundary_triangle_cleanup_is_one_pass():
|
||||
obj, _ = surface(resolution=2)
|
||||
group = obj.modifiers[0].node_group
|
||||
mesh = bpy.data.meshes.new("Boundary ears")
|
||||
mesh.from_pydata([(0,0,0),(1,0,0),(1,1,0),(0,1,0),(.5,.5,0),(2,0,0)], [],
|
||||
[(0,1,4),(1,2,4),(2,3,4),(3,0,4),(1,5,2)])
|
||||
obj.data = mesh
|
||||
source = group.nodes.new("NodeGroupInput")
|
||||
cleaned = remove_boundary_triangles(group, source.outputs["Geometry"])
|
||||
group.links.new(cleaned, next(n for n in group.nodes if n.type == "GROUP_OUTPUT").inputs["Geometry"])
|
||||
obj.update_tag(refresh={"DATA"})
|
||||
bpy.context.view_layer.update()
|
||||
_, faces = evaluated(obj)
|
||||
assert len(faces) == 4
|
||||
assert all(len(f) == 3 for f in faces)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("iterations", [0, 16])
|
||||
def test_cleaned_sawtooth_smoothing_preserves_face_orientation(iterations):
|
||||
obj, set_value = diagonal_surface(triangles=True)
|
||||
@@ -90,63 +70,3 @@ def test_split_profile_tapers_within_boundary_band():
|
||||
assert values.max() == pytest.approx(.3)
|
||||
set_value("Depth Split", 0)
|
||||
np.testing.assert_allclose(profile_values(obj), .3)
|
||||
|
||||
|
||||
def sampled(obj, group, value, name):
|
||||
"""Store one field on the points and read it back from the evaluated mesh."""
|
||||
data_type, reader = {"VECTOR": ("FLOAT_VECTOR", "vector"), "VALUE": ("FLOAT", "value")}[value.type]
|
||||
store = group.nodes.new("GeometryNodeStoreNamedAttribute")
|
||||
store.data_type, store.domain = data_type, "POINT"
|
||||
store.inputs["Name"].default_value = name
|
||||
source = group.nodes.new("NodeGroupInput")
|
||||
group.links.new(source.outputs["Geometry"], store.inputs["Geometry"])
|
||||
group.links.new(value, store.inputs["Value"])
|
||||
group.links.new(store.outputs[0], next(n for n in group.nodes if n.type == "GROUP_OUTPUT").inputs["Geometry"])
|
||||
obj.update_tag(refresh={"DATA"})
|
||||
bpy.context.view_layer.update()
|
||||
result = obj.evaluated_get(bpy.context.evaluated_depsgraph_get())
|
||||
mesh = result.to_mesh()
|
||||
try:
|
||||
return np.array([getattr(item, reader) for item in mesh.attributes[name].data])
|
||||
finally:
|
||||
result.to_mesh_clear()
|
||||
|
||||
|
||||
def test_normal_weight_follows_profile_only_in_balloon_mode():
|
||||
obj, _ = surface(resolution=2)
|
||||
group = obj.modifiers[0].node_group
|
||||
is_shell = group.nodes.new("FunctionNodeInputBool")
|
||||
profile = group.nodes.new("ShaderNodeValue")
|
||||
profile.outputs[0].default_value = .25
|
||||
weight = build_normal_weight(group, is_shell.outputs[0], profile.outputs[0])
|
||||
np.testing.assert_allclose(sampled(obj, group, weight, "normal_weight"), .25)
|
||||
is_shell.boolean = True
|
||||
np.testing.assert_allclose(sampled(obj, group, weight, "normal_weight"), 0.0)
|
||||
|
||||
|
||||
def test_normal_direction_turns_the_edge_with_edge_turn():
|
||||
obj, _ = surface(resolution=2)
|
||||
group = obj.modifiers[0].node_group
|
||||
smoothed = group.nodes.new("ShaderNodeCombineXYZ")
|
||||
smoothed.inputs["Z"].default_value = 1
|
||||
average = group.nodes.new("ShaderNodeCombineXYZ")
|
||||
average.inputs["X"].default_value = 1
|
||||
weight = group.nodes.new("ShaderNodeValue")
|
||||
controls = group.nodes.new("NodeGroupInput")
|
||||
direction = build_normal_direction(
|
||||
group, controls, smoothed.outputs[0], average.outputs[0], weight.outputs[0],
|
||||
)
|
||||
modifier = obj.modifiers[0]
|
||||
turn = modifier_input_identifier(group, "Edge Turn")
|
||||
|
||||
def result(weight_value, turn_value):
|
||||
weight.outputs[0].default_value = weight_value
|
||||
set_modifier_input(modifier, turn, turn_value)
|
||||
obj.update_tag(refresh={"DATA"})
|
||||
bpy.context.view_layer.update()
|
||||
return sampled(obj, group, direction, "normal_direction")[0]
|
||||
|
||||
np.testing.assert_allclose(result(0.0, 0.0), (0, 0, 1), atol=1e-6)
|
||||
np.testing.assert_allclose(result(0.0, 1.0), (1, 0, 0), atol=1e-6)
|
||||
np.testing.assert_allclose(result(0.0, 2.0), (2, 0, -1) / np.sqrt(5), atol=1e-6)
|
||||
np.testing.assert_allclose(result(1.0, 2.0), (1, 0, 0), atol=1e-6)
|
||||
|
||||
Reference in New Issue
Block a user