def test03_depth_packet_stairs(variant_packet_rgb):
from mitsuba.core import Ray3f as Ray3fX, Properties
from mitsuba.render import Scene
if mitsuba.core.MTS_ENABLE_EMBREE:
pytest.skip("EMBREE enabled")
props = Properties("scene")
props["_unnamed_0"] = create_stairs_packet(11)
scene = Scene(props)
mitsuba.set_variant("scalar_rgb")
from mitsuba.core import Ray3f, Vector3f
n = 4
inv_n = 1.0 / (n - 1)
rays = Ray3fX.zero(n * n)
d = [0, 0, -1]
wavelengths = []
for x in range(n):
for y in range(n):
o = Vector3f(x * inv_n, y * inv_n, 2)
o = o * 0.999 + 0.0005
rays[x * n + y] = Ray3f(o, d, 0, 100, 0.5, wavelengths)
res_naive = scene.ray_intersect_naive(rays)
res = scene.ray_intersect(rays)
res_shadow = scene.ray_test(rays)
# TODO: spot-check (here, we only check consistency)
assert ek.all(res_shadow == res.is_valid())
compare_results(res_naive, res, atol=1e-6)
def test03_ray_intersect(variant_scalar_rgb):
from mitsuba.core import xml, Ray3f, Transform4f
# Scalar
scene = xml.load_dict({
"type": "scene",
"foo": {
"type": "rectangle",
"to_world": Transform4f.scale((2.0, 0.5, 1.0))
}
})
n = 15
coords = ek.linspace(Float, -1, 1, n)
rays = [
Ray3f(o=[a, a, 5], d=[0, 0, -1], time=0.0, wavelengths=[])
for a in coords
]
si_scalar = []
valid_count = 0
for i in range(n):
its_found = scene.ray_test(rays[i])
si = scene.ray_intersect(rays[i])
assert its_found == (abs(coords[i]) <= 0.5)
assert si.is_valid() == its_found
si_scalar.append(si)
valid_count += its_found
assert valid_count == 7
try:
mitsuba.set_variant('packet_rgb')
from mitsuba.core import xml, Ray3f as Ray3fX
except ImportError:
pytest.skip("packet_rgb mode not enabled")
# Packet
scene_p = xml.load_dict({
"type": "scene",
"foo": {
"type": "rectangle",
"to_world": Transform4f.scale((2.0, 0.5, 1.0))
}
})
packet = Ray3fX.zero(n)
for i in range(n):
packet[i] = rays[i]
si_p = scene_p.ray_intersect(packet)
its_found_p = scene_p.ray_test(packet)
assert ek.all(si_p.is_valid() == its_found_p)
for i in range(n):
assert ek.allclose(si_p.t[i], si_scalar[i].t)
def test03_ray_intersect(variant_scalar_rgb):
if mitsuba.core.MTS_ENABLE_EMBREE:
pytest.skip("EMBREE enabled")
from mitsuba.core.xml import load_string
from mitsuba.core import Ray3f
# Scalar
scene = load_string("""<scene version="2.0.0">
<shape type="rectangle">
<transform name="to_world">
<scale x="2" y="0.5" z="1"/>
</transform>
</shape>
</scene>""")
n = 15
coords = ek.linspace(Float, -1, 1, n)
rays = [
Ray3f(o=[a, a, 5], d=[0, 0, -1], time=0.0, wavelengths=[])
for a in coords
]
si_scalar = []
valid_count = 0
for i in range(n):
# print(rays[i])
its_found = scene.ray_test(rays[i])
si = scene.ray_intersect(rays[i])
assert its_found == (abs(coords[i]) <= 0.5)
assert si.is_valid() == its_found
si_scalar.append(si)
valid_count += its_found
assert valid_count == 7
try:
mitsuba.set_variant('packet_rgb')
from mitsuba.core.xml import load_string as load_string_packet
from mitsuba.core import Ray3f as Ray3fX
except ImportError:
pytest.skip("packet_rgb mode not enabled")
# Packet
scene_p = load_string_packet("""<scene version="2.0.0">
<shape type="rectangle">
<transform name="to_world">
<scale x="2" y="0.5" z="1"/>
</transform>
</shape>
</scene>""")
packet = Ray3fX.zero(n)
for i in range(n):
packet[i] = rays[i]
si_p = scene_p.ray_intersect(packet)
its_found_p = scene_p.ray_test(packet)
assert ek.all(si_p.is_valid() == its_found_p)
for i in range(n):
assert ek.allclose(si_p.t[i], si_scalar[i].t)