def test_save_multi_influence_per_face_index(self):
converter = DaeConverter(self._get_brres('koopaFigure.brres'),
self._get_tmp('.dae'))
converter.save_model()
original = converter.mdl0
new = DaeConverter(self._get_tmp('.brres'),
converter.mdl_file).load_model()
self.assertTrue(
CheckPositions.positions_equal(original.vertices,
new.vertices,
rtol=0.1,
atol=0.01))
self.assertTrue(CheckPositions.bones_equal(original.bones, new.bones))
def test_kuribo_loads(self):
b = self._get_brres('kuribo.brres')
model = b.models[1]
converter = DaeConverter(b, self._get_test_fname('kuribo.dae'))
converter.load_model()
# converter.brres.save('../../tmp/kuribo.brres', overwrite=True)
self.assertTrue(CheckPositions().positions_equal(
model.vertices, converter.mdl0.vertices, rtol=1e-1, atol=1e-2))
self.assertTrue(CheckPositions().bones_equal(model.bones,
converter.mdl0.bones))
poly = converter.mdl0.objects[0]
for i in range(1, 3):
self.assertTrue(poly.has_uv_matrix(i))
def test_save_and_load_polygons_bound_to_single_material(self):
converter = DaeConverter(self._get_brres('castleflower1.brres'),
self._get_tmp('.dae'))
converter.save_model()
original = converter.mdl0
new = DaeConverter(self._get_tmp('.brres'),
converter.mdl_file).load_model()
self.assertTrue(
CheckPositions.positions_equal(original.vertices,
new.vertices,
rtol=0.1,
atol=0.01))
self.assertTrue(
CheckPositions.bones_equal(
[x.linked_bone for x in original.objects],
[x.linked_bone for x in new.objects]))
def test_map_model_converts_bones_correctly(self):
# Export the map model
dae = self._get_tmp('.dae')
DaeConverter(self.MAP, dae, encode=False).convert()
converter = DaeConverter(self._get_tmp('.brres'), dae)
converter.load_model()
self.assertTrue(CheckPositions.model_equal(self.MAP.models[0], converter.mdl0))
def test_save_multi_bone_as_single_bone(self):
fname = self._get_tmp('.dae')
converter = DaeConverter(
self._get_brres('simple_multi_bone_single_bind.brres'),
fname,
flags=DaeConverter.SINGLE_BONE)
converter.save_model()
# load it to test against original
converter = DaeConverter(self._get_tmp('.brres'), fname)
converter.load_model()
# converter.brres.save(overwrite=True)
original = self._get_brres('simple.brres').models[0]
new = converter.mdl0
self.assertTrue(CheckPositions().bones_equal(original.bones, new.bones,
0.01, 0.001))
self.assertTrue(CheckPositions().positions_equal(
original.vertices, new.vertices, 0.01, 0.001))
def test_save_and_load_multi_bind_equal(self):
fname = self._get_tmp('.dae')
converter = DaeConverter(
self._get_brres('simple_multi_bone_multi_bind.brres'), fname)
converter.save_model()
original = converter.mdl0
converter = DaeConverter(self._get_tmp('.brres'), fname)
converter.load_model()
self.assertTrue(CheckPositions().model_equal(original, converter.mdl0))
def test_save_multi_bone_single_bind(self):
fname = self._get_tmp('.dae')
converter = DaeConverter(
self._get_brres('simple_multi_bone_single_bind.brres'), fname)
converter.save_model()
original = converter.mdl0
converter = DaeConverter(self._get_tmp('.brres'), fname)
converter.load_model()
converter.brres.save(overwrite=True)
self.assertTrue(CheckPositions().model_equal(original, converter.mdl0,
0.01, 0.001))
def test_convert_cow(self):
original = self._get_brres('cow.brres').models[1]
converter = DaeConverter(
Brres(self._get_tmp('.brres'), read_file=False),
self._get_test_fname('cow.dae'))
converter.load_model()
mdl0 = converter.mdl0
# converter.brres.save(overwrite=True)
# Ensure that the bones are correct
self.assertTrue(
CheckPositions.bones_equal(original.bones, mdl0.bones, 0.0001))
def test_convert_single_bone(self):
original = self._get_brres('simple.brres')
converter = DaeConverter(
Brres(self._get_tmp('.brres'), read_file=False),
self._get_test_fname('simple_multi_bone_single_bind.dae'),
flags=DaeConverter.SINGLE_BONE)
converter.load_model()
mdl0 = converter.mdl0
self.assertEqual(len(mdl0.bones), 1)
self.assertTrue(
CheckPositions.positions_equal(original.models[0].vertices,
mdl0.vertices))
def test_convert_multi_bone_single_bind(self):
original = self._get_brres(
'simple_multi_bone_single_bind.brres').models[0]
converter = DaeConverter(
Brres(self._get_tmp('.brres'), read_file=False),
self._get_test_fname('simple_multi_bone_single_bind.dae'))
converter.load_model()
mdl0 = converter.mdl0
# test bones
self.assertEqual(4, len(mdl0.bones))
CheckPositions.bones_equal(mdl0.bones, original.bones, 0.01, 0.001)
# Ensure that the bones are correctly linked
expected_linked_bones = {
'BlackNWhite1': 'simple001_BNW',
'BlackNWhite2': 'simple011',
'GreenCloud': 'simple002_Green',
'Yellow': 'simple'
}
for x in mdl0.objects:
self.assertEqual(expected_linked_bones[x.name], x.linked_bone.name)
self.assertTrue(
CheckPositions.positions_equal(original.vertices, mdl0.vertices))
def test_convert_to_correct_position(self):
# Get reference to beginner_course to test against
beginner_brres = self._get_brres('beginner_course.brres')
original_vertices = beginner_brres.models[0].vertices
# Convert dae
brres = Brres(self._get_tmp('.brres'), read_file=False)
dae = self._get_test_fname('beginner.dae')
converter = DaeConverter(brres, dae)
converter.load_model()
# brres.save(overwrite=True)
new_vertices = brres.models[0].vertices
self.assertTrue(CheckPositions().positions_equal(
original_vertices, new_vertices))
def test_save_multi_bone_scaled_as_single_bone(self):
fname = self._get_tmp('.dae')
converter = DaeConverter(
self._get_brres('simple_multi_bone_scaled.brres'),
fname,
flags=DaeConverter.SINGLE_BONE)
converter.save_model()
# load it to test against original
converter = DaeConverter(self._get_tmp('.brres'), fname)
converter.load_model()
original = self._get_brres('simple.brres').models[0]
self.assertTrue(CheckPositions().positions_equal(
original.vertices, converter.mdl0.vertices))
self.assertNotEqual(original.bones,
converter.mdl0.bones) # should be scaled
def test_save_load_eq(self):
original = self._get_brres('beginner_course.brres')
converter = ObjConverter(original, self._get_tmp('.obj'),
encode=False).convert()
encoder = ObjConverter(self._get_tmp('.brres'), converter.mdl_file).convert()
self.assertTrue(CheckPositions.positions_equal(original.models[0].vertices, encoder.mdl0.vertices))