nan
938d95db0f
- Implemented a preview tool for rectifying images with interactive aspect controls in `preview.py`. - Added tests for depth metadata loading to ensure consistent schema usage in `test_depth_metadata.py`. - Developed comprehensive tests for cutout calibration processes in `test_cutout_calibration.py`, including stability checks and depth calibration. - Created tests for frame compositing operations in `test_frame_compositing.py`, ensuring correct handling of depth and transparency. - Introduced material slot inheritance functionality in `material_slots.py` to manage geometry material assignments. - Built shared shader nodes for material processing in `shader.py`, including roughness and bump mapping setups.
47 lines
2.3 KiB
Python
47 lines
2.3 KiB
Python
import bpy
|
|
import numpy as np
|
|
import pytest
|
|
from anyimage.operators.cutout_tool.object import load_cutout_node_group
|
|
from anyimage.operators.cutout_tool.operators import normal_mode_for_shape
|
|
from anyimage.operators.cutout_tool.geometry import refine_balloon_triangles
|
|
from anyimage.operators.cutout_tool.balloon import poisson_balloon_profile
|
|
|
|
|
|
@pytest.mark.parametrize("enabled", [False, True])
|
|
def test_depth_presets_always_generate_the_correct_normal_space(enabled):
|
|
assert normal_mode_for_shape("DEPTH_SOLID", enabled) == "OBJECT"
|
|
assert normal_mode_for_shape("DEPTH_BALLOON", enabled) == "TANGENT"
|
|
for shape in ("FLAT", "SOLID"):
|
|
assert normal_mode_for_shape(shape, enabled) == ("TANGENT" if enabled else "NONE")
|
|
|
|
|
|
def test_new_cutout_does_not_reuse_a_saved_wrapper():
|
|
bpy.ops.wm.read_factory_settings(use_empty=True)
|
|
saved = bpy.data.node_groups.new("O Image Cutout", "GeometryNodeTree")
|
|
saved.interface.new_socket(name="Shape", in_out="INPUT", socket_type="NodeSocketMenu")
|
|
saved.use_fake_user = True
|
|
current = load_cutout_node_group("SOLID")
|
|
assert current != saved and current.get("anyimage_asset_revision") == 13
|
|
assert saved in bpy.data.node_groups.values()
|
|
assert current == load_cutout_node_group("FLAT")
|
|
assert any(item.name == "Shape" for item in saved.interface.items_tree)
|
|
|
|
|
|
def test_depth_cutout_loads_uniform_thickness_revision_and_preserves_existing_group():
|
|
bpy.ops.wm.read_factory_settings(use_empty=True)
|
|
saved = bpy.data.node_groups.new("O Image Depth Cutout", "GeometryNodeTree")
|
|
saved["anyimage_asset_revision"] = 11
|
|
current = load_cutout_node_group("DEPTH_SOLID")
|
|
assert current != saved and current["anyimage_asset_revision"] == 26
|
|
assert saved in bpy.data.node_groups.values()
|
|
assert saved["anyimage_asset_revision"] == 11
|
|
assert load_cutout_node_group("DEPTH_SOLID") == current
|
|
|
|
|
|
def test_refined_boundary_ears_have_interior_balloon_samples():
|
|
points = np.array([(0, 0), (1, 0), (1, 1), (0, 1)], dtype=float)
|
|
points, faces, boundary = refine_balloon_triangles(points, [(0, 1, 2), (0, 2, 3)], np.ones(4, bool))
|
|
profile = poisson_balloon_profile(points, faces, boundary)
|
|
assert np.all(profile[boundary] == 0)
|
|
assert all(any(profile[index] > 0 for index in face) for face in faces)
|