def test12_xml_duplicate_files(variants_scalar_all):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict12/dict.xml')
fr.append(os.path.dirname(filepath))
spectrum_path = os.path.join(mts_root, 'resources/data/ior/Al.eta.spd')
#Export the same file twice, this should only copy it once
scene_dict = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"filename": spectrum_path
}
},
'light2': {
"type": "point",
"intensity": {
"type": "spectrum",
"filename": spectrum_path
}
}
}
dict_to_xml(scene_dict, filepath)
spectra_files = os.listdir(
os.path.join(os.path.split(filepath)[0], 'spectra'))
assert len(spectra_files) == 1 and spectra_files[0] == "Al.eta.spd"
spectrum_path2 = os.path.join(mts_root,
'resources/data/scenes/dict12/Al.eta.spd')
copy(spectrum_path, spectrum_path2)
#Export two files having the same name
scene_dict = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"filename": spectrum_path
}
},
'light2': {
"type": "point",
"intensity": {
"type": "spectrum",
"filename": spectrum_path2
}
}
}
dict_to_xml(scene_dict, filepath)
spectra_files = os.listdir(
os.path.join(os.path.split(filepath)[0], 'spectra'))
assert len(spectra_files) == 2 and spectra_files[
0] == "Al.eta.spd" and spectra_files[1] == "Al.eta(1).spd"
rmtree(os.path.split(filepath)[0])
def test09_xml_decompose_transform(variant_scalar_rgb):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core.xml import load_dict, load_file
from mitsuba.core import Transform4f, Vector3f, Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict09/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'cam': {
'type':
'perspective',
'fov_axis':
'x',
'fov':
35,
'to_world':
Transform4f.look_at(Vector3f(15, 42.3,
25), Vector3f(1.0, 0.0, 0.5),
Vector3f(1.0, 0.0, 0.0))
}
}
dict_to_xml(scene_dict, filepath)
s1 = load_file(filepath)
s2 = load_dict(scene_dict)
vects = [Vector3f(0, 0, 1), Vector3f(0, 1, 0), Vector3f(1, 0, 0)]
tr1 = s1.sensors()[0].world_transform().eval(0)
tr2 = s2.sensors()[0].world_transform().eval(0)
for vec in vects:
assert tr1.transform_point(vec) == tr2.transform_point(vec)
rmtree(os.path.split(filepath)[0])
def test05_xml_split(variant_scalar_rgb):
from mitsuba.core import xml, Point3f
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
import numpy as np
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict05/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'bsdf1': {
'type': 'diffuse'
},
'envmap': {
'type': 'envmap',
'filename': 'resources/data/common/textures/museum.exr'
},
'shape': {
'type': 'sphere',
'bsdf': {
'type': 'ref',
'id': 'bsdf1'
}
}
}
dict_to_xml(scene_dict, filepath)
s1 = xml.load_file(filepath)
dict_to_xml(scene_dict, filepath, split_files=True)
s2 = xml.load_file(filepath)
assert str(s1) == str(s2)
rmtree(os.path.split(filepath)[0])
def test01_xml_save_plugin(variant_scalar_rgb):
from mitsuba.core import xml
from mitsuba.core import Thread
from mitsuba.python.xml import dict_to_xml
fr = Thread.thread().file_resolver()
# Add the path to the mitsuba root folder, so that files are always saved in mitsuba/resources/...
# This way we know where to look for the file in case the unit test fails
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict01/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
"type": "sphere",
"center": [0, 0, -10],
"radius": 10.0,
}
dict_to_xml(scene_dict, filepath)
s1 = xml.load_dict({
'type': 'scene',
'sphere': {
"type": "sphere",
"center": [0, 0, -10],
"radius": 10.0,
}
})
s2 = xml.load_file(filepath)
assert str(s1) == str(s2)
rmtree(os.path.split(filepath)[0])
def test13_xml_multiple_defaults(variant_scalar_rgb):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core.xml import load_file
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict13/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'cam1': {
'type': 'perspective',
'sampler': {
'type': 'independent',
'sample_count': 150
}
},
'cam2': {
'type': 'perspective',
'sampler': {
'type': 'independent',
'sample_count': 250
}
}
}
dict_to_xml(scene_dict, filepath)
scene = load_file(filepath, spp=45)
assert scene.sensors()[0].sampler().sample_count() == scene.sensors(
)[1].sampler().sample_count()
assert scene.sensors()[1].sampler().sample_count() == 45
rmtree(os.path.split(filepath)[0])
def test02_xml_missing_type(variant_scalar_rgb):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict02/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {'my_bsdf': {'type': 'diffuse'}}
with pytest.raises(ValueError) as e:
dict_to_xml(scene_dict, filepath)
e.match("Missing key: 'type'!")
rmtree(os.path.split(filepath)[0])
def test03_xml_references(variant_scalar_rgb):
from mitsuba.core import xml
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict03/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'bsdf1': {
'type': 'diffuse',
'reflectance': {
'type': 'rgb',
'value': [1.0, 0.0, 0.0]
}
},
'texture1': {
'type': 'checkerboard'
},
'bsdf4': {
'type': 'dielectric',
'specular_reflectance': {
'type': 'ref',
'id': 'texture1'
},
'id': 'bsdf2'
},
'shape1': {
'type': 'sphere',
'bsdf': {
'type': 'ref',
'id': 'bsdf1'
}
},
'shape2': {
'type': 'cylinder',
'bsdf': {
'type': 'ref',
'id': 'bsdf2'
}
}
}
s1 = xml.load_dict(scene_dict)
dict_to_xml(scene_dict, filepath)
s2 = xml.load_file(filepath)
assert str(s1.shapes()[0].bsdf()) == str(s2.shapes()[0].bsdf())
rmtree(os.path.split(filepath)[0])
def test08_xml_defaults(variant_scalar_rgb):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
from mitsuba.core.xml import load_dict, load_file
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict08/dict.xml')
fr.append(os.path.dirname(filepath))
spp = 250
resx = 1920
resy = 1080
scene_dict = {
'type': 'scene',
'cam': {
'type': 'perspective',
'fov_axis': 'x',
'fov': 35,
'sampler': {
'type': 'independent',
'sample_count': spp # --> default
},
'film': {
'type': 'hdrfilm',
'width': resx, # --> default
'height': resy # --> default
}
}
}
dict_to_xml(scene_dict, filepath)
# Load a file using default values
s1 = load_file(filepath)
s2 = load_dict(scene_dict)
assert str(s1.sensors()[0].film()) == str(s2.sensors()[0].film())
assert str(s1.sensors()[0].sampler()) == str(s2.sensors()[0].sampler())
# Set new parameters
spp = 45
resx = 2048
resy = 485
# Load a file with options for the rendering parameters
s3 = load_file(filepath, spp=spp, resx=resx, resy=resy)
scene_dict['cam']['sampler']['sample_count'] = spp
scene_dict['cam']['film']['width'] = resx
scene_dict['cam']['film']['height'] = resy
s4 = load_dict(scene_dict)
assert str(s3.sensors()[0].film()) == str(s4.sensors()[0].film())
assert str(s3.sensors()[0].sampler()) == str(s4.sensors()[0].sampler())
rmtree(os.path.split(filepath)[0])
def test06_xml_duplicate_id(variant_scalar_rgb):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict06/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'my-bsdf': {
'type': 'diffuse'
},
'bsdf1': {
'type': 'roughdielectric',
'id': 'my-bsdf'
}
}
with pytest.raises(ValueError) as e:
dict_to_xml(scene_dict, filepath)
e.match("Id: my-bsdf is already used!")
rmtree(os.path.split(filepath)[0])
def test07_xml_invalid_ref(variant_scalar_rgb):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict07/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'bsdf': {
'type': 'diffuse'
},
'sphere': {
'type': 'sphere',
'bsdf': {
'type': 'ref',
'id': 'my-bsdf'
}
}
}
with pytest.raises(ValueError) as e:
dict_to_xml(scene_dict, filepath)
e.match("Id: my-bsdf referenced before export.")
rmtree(os.path.split(filepath)[0])
def test10_xml_rgb(variants_scalar_all):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core.xml import load_dict, load_file
from mitsuba.core import Thread, ScalarColor3f
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict10/dict.xml')
fr.append(os.path.dirname(filepath))
d1 = {
'type': 'scene',
'point': {
"type": "point",
"intensity": {
"type": "rgb",
"value": ScalarColor3f(0.5, 0.2, 0.5)
}
}
}
d2 = {
'type': 'scene',
'point': {
"type": "point",
"intensity": {
"type": "rgb",
"value": [0.5, 0.2, 0.5] # list -> ScalarColor3f
}
}
}
dict_to_xml(d1, filepath)
s1 = load_file(filepath)
dict_to_xml(d2, filepath)
s2 = load_file(filepath)
s3 = load_dict(d1)
assert str(s1) == str(s2)
assert str(s1) == str(s3)
d1 = {
'type': 'scene',
'point': {
"type": "point",
"intensity": {
"type": "rgb",
"value": 0.5 # float -> ScalarColor3f
}
}
}
dict_to_xml(d1, filepath)
s1 = load_file(filepath)
s2 = load_dict(d1)
assert str(s1) == str(s2)
rmtree(os.path.split(filepath)[0])
def test04_xml_point(variant_scalar_rgb):
from mitsuba.core import xml, Point3f
from mitsuba.python.xml import dict_to_xml
from mitsuba.core import Thread
import numpy as np
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict04/dict.xml')
fr.append(os.path.dirname(filepath))
scene_dict = {
'type': 'scene',
'light': {
'type': 'point',
'position': [0.0, 1.0, 2.0]
}
}
dict_to_xml(scene_dict, filepath)
s1 = xml.load_file(filepath)
scene_dict = {
'type': 'scene',
'light': {
'type': 'point',
'position': Point3f(0, 1, 2)
}
}
dict_to_xml(scene_dict, filepath)
s2 = xml.load_file(filepath)
scene_dict = {
'type': 'scene',
'light': {
'type': 'point',
'position': np.array([0, 1, 2])
}
}
dict_to_xml(scene_dict, filepath)
s3 = xml.load_file(filepath)
assert str(s1) == str(s2)
assert str(s1) == str(s3)
rmtree(os.path.split(filepath)[0])
def test11_xml_spectrum(variants_scalar_all):
from mitsuba.python.xml import dict_to_xml
from mitsuba.core.xml import load_dict, load_file
from mitsuba.core import Thread
fr = Thread.thread().file_resolver()
mts_root = str(fr[len(fr) - 1])
filepath = os.path.join(mts_root, 'resources/data/scenes/dict11/dict.xml')
fr.append(os.path.dirname(filepath))
d1 = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"value": [(400, 0.1), (500, 0.2), (600, 0.4), (700, 0.1)]
}
}
}
dict_to_xml(d1, filepath)
s1 = load_file(filepath)
s2 = load_dict(d1)
assert str(s1.emitters()[0]) == str(s2.emitters()[0])
d2 = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"value": 0.44
}
}
}
dict_to_xml(d2, filepath)
s1 = load_file(filepath)
s2 = load_dict(d2)
assert str(s1.emitters()[0]) == str(s2.emitters()[0])
d3 = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"value": [(400, 0.1), (500, 0.2), (300, 0.4)]
}
}
}
with pytest.raises(ValueError) as e:
dict_to_xml(d3, filepath)
e.match("Wavelengths must be sorted in strictly increasing order!")
#wavelengths file
d4 = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"filename": os.path.join(mts_root,
'resources/data/ior/Al.eta.spd')
}
}
}
dict_to_xml(d4, filepath)
s1 = load_file(filepath)
s2 = load_dict(d4)
assert str(s1.emitters()[0]) == str(s2.emitters()[0])
d5 = {
'type': 'scene',
'light': {
"type": "point",
"intensity": {
"type": "spectrum",
"filename": os.path.join(mts_root,
'resources/data/blabla/Pr.spd')
}
}
}
with pytest.raises(ValueError) as e:
dict_to_xml(d5, filepath)
e.match("File '%s' not found!" %
os.path.abspath(d5['light']['intensity']['filename']))
rmtree(os.path.split(filepath)[0])
