nan
988332548e
- Implemented a comprehensive test suite for the clipboard image module, covering various functionalities including operator registration, context handling, and image processing. - Created a new mesh plane asset with customizable parameters and integrated it into the geometry node system. - Developed a validation script to ensure the correct behavior of the mesh plane asset, including input handling and output verification.
104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
from pathlib import Path
|
|
import sys
|
|
|
|
import bpy
|
|
|
|
|
|
PROJECT_ROOT = Path(__file__).parents[1]
|
|
sys.path.insert(0, str(PROJECT_ROOT / "src"))
|
|
|
|
from anyimage.operators.material import (
|
|
get_modifier_input,
|
|
modifier_input_identifier,
|
|
set_modifier_input,
|
|
)
|
|
from anyimage.operators.mesh_plane import (
|
|
add_mesh_plane_processing,
|
|
create_mesh_plane_mesh,
|
|
)
|
|
|
|
|
|
def evaluated_mesh(obj):
|
|
bpy.context.view_layer.update()
|
|
evaluated = obj.evaluated_get(bpy.context.evaluated_depsgraph_get())
|
|
return evaluated, evaluated.to_mesh()
|
|
|
|
|
|
def main():
|
|
bpy.ops.wm.read_factory_settings(use_empty=True)
|
|
mesh = create_mesh_plane_mesh("Mesh Plane", (-1.0, 1.0, -0.5, 0.5))
|
|
assert len(mesh.polygons) == 1
|
|
material = bpy.data.materials.new("Mesh Plane Material")
|
|
obj = bpy.data.objects.new("Mesh Plane", mesh)
|
|
bpy.context.scene.collection.objects.link(obj)
|
|
modifier = add_mesh_plane_processing(obj, material)
|
|
group = modifier.node_group
|
|
assert group.name == "O Mesh Plane"
|
|
assert group.is_modifier
|
|
inputs = {
|
|
item.name: item
|
|
for item in group.interface.items_tree
|
|
if item.item_type == "SOCKET" and item.in_out == "INPUT"
|
|
}
|
|
assert list(inputs) == ["Geometry", "Subdivide", "Thickness", "Material"]
|
|
assert inputs["Subdivide"].default_value == 0
|
|
assert inputs["Subdivide"].max_value == 10
|
|
assert inputs["Thickness"].default_value == 0.0
|
|
assert inputs["Material"].hide_in_modifier
|
|
assert get_modifier_input(modifier, inputs["Material"].identifier) == material
|
|
|
|
evaluated, result = evaluated_mesh(obj)
|
|
try:
|
|
assert len(result.polygons) == 2
|
|
assert len(result.vertices) == 6
|
|
assert "UVMap" in result.attributes
|
|
uv_values = {
|
|
tuple(round(value, 6) for value in loop.uv)
|
|
for loop in result.uv_layers["UVMap"].data
|
|
}
|
|
assert uv_values == {
|
|
(0.0, 0.0),
|
|
(0.5, 0.0),
|
|
(1.0, 0.0),
|
|
(0.0, 1.0),
|
|
(0.5, 1.0),
|
|
(1.0, 1.0),
|
|
}
|
|
assert all(abs(vertex.co.z) < 1e-6 for vertex in result.vertices)
|
|
finally:
|
|
evaluated.to_mesh_clear()
|
|
|
|
set_modifier_input(modifier, modifier_input_identifier(group, "Subdivide"), 3)
|
|
obj.update_tag(refresh={"DATA"})
|
|
evaluated, result = evaluated_mesh(obj)
|
|
try:
|
|
assert len(result.polygons) == 128
|
|
assert len(result.vertices) == 153
|
|
finally:
|
|
evaluated.to_mesh_clear()
|
|
|
|
set_modifier_input(modifier, modifier_input_identifier(group, "Subdivide"), 2)
|
|
set_modifier_input(modifier, modifier_input_identifier(group, "Thickness"), 1.2)
|
|
obj.update_tag(refresh={"DATA"})
|
|
evaluated, result = evaluated_mesh(obj)
|
|
try:
|
|
assert len(result.polygons) == 184
|
|
z_values = [vertex.co.z for vertex in result.vertices]
|
|
assert abs(min(z_values) + 0.6) < 1e-6
|
|
assert abs(max(z_values) - 0.6) < 1e-6
|
|
assert len({round(value, 6) for value in z_values}) == 6
|
|
edge_users = {edge.key: 0 for edge in result.edges}
|
|
for polygon in result.polygons:
|
|
for edge_key in polygon.edge_keys:
|
|
edge_users[edge_key] += 1
|
|
assert all(users == 2 for users in edge_users.values())
|
|
finally:
|
|
evaluated.to_mesh_clear()
|
|
|
|
if bpy.app.version >= (5, 0, 0):
|
|
assert group.library is not None
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|