Collada: Added tests for Animation export
This commit is contained in:
Gaia Clary
parent
05b5958a7a
commit
450be3b226
|
|
@ -53,13 +53,14 @@ endif()
|
|||
|
||||
# ------------------------------------------------------------------------------
|
||||
# GENERAL PYTHON CORRECTNESS TESTS
|
||||
macro (COLLADA_TEST module test_name)
|
||||
macro (COLLADA_TEST module test_name blend_file)
|
||||
add_test(
|
||||
NAME collada_${test_name}
|
||||
COMMAND "$<TARGET_FILE:blender>" ${TEST_BLENDER_EXE_PARAMS} ${TEST_SRC_DIR}/collada/${module}/${test_name}.blend
|
||||
--python ${CMAKE_CURRENT_LIST_DIR}/${module}/test_${test_name}.py --
|
||||
NAME collada_${module}_${test_name}
|
||||
COMMAND "$<TARGET_FILE:blender>" ${TEST_BLENDER_EXE_PARAMS} ${TEST_SRC_DIR}/collada/${module}/${blend_file}
|
||||
--python ${CMAKE_CURRENT_LIST_DIR}/${module}/test_${module}_${test_name}.py --
|
||||
--testdir ${TEST_SRC_DIR}/collada/${module}
|
||||
)
|
||||
endmacro()
|
||||
|
||||
COLLADA_TEST(mesh mesh_simple)
|
||||
COLLADA_TEST(mesh simple mesh_simple.blend)
|
||||
COLLADA_TEST(animation simple suzannes_parent_inverse.blend)
|
||||
|
|
|
|||
|
|
@ -0,0 +1,280 @@
|
|||
#!/usr/bin/env python3
|
||||
# ##### BEGIN GPL LICENSE BLOCK #####
|
||||
#
|
||||
# This program is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU General Public License
|
||||
# as published by the Free Software Foundation; either version 2
|
||||
# of the License, or (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program; if not, write to the Free Software Foundation,
|
||||
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
#
|
||||
# ##### END GPL LICENSE BLOCK #####
|
||||
#
|
||||
# Call as follows:
|
||||
# python collada_mesh_simple.py --blender PATH_TO_BLENDER_EXE --testdir PATH_TO_SVN/lib/tests/collada/mesh
|
||||
#
|
||||
import sys
|
||||
import bpy
|
||||
import argparse
|
||||
import functools
|
||||
import shutil
|
||||
import tempfile
|
||||
import unittest
|
||||
import difflib
|
||||
import pathlib
|
||||
from pathlib import Path
|
||||
|
||||
def with_tempdir(wrapped):
|
||||
"""Creates a temporary directory for the function, cleaning up after it returns normally.
|
||||
|
||||
When the wrapped function raises an exception, the contents of the temporary directory
|
||||
remain available for manual inspection.
|
||||
|
||||
The wrapped function is called with an extra positional argument containing
|
||||
the pathlib.Path() of the temporary directory.
|
||||
"""
|
||||
|
||||
@functools.wraps(wrapped)
|
||||
def decorator(*args, **kwargs):
|
||||
dirname = tempfile.mkdtemp(prefix='blender-collada-test')
|
||||
#print("Using tempdir %s" % dirname)
|
||||
try:
|
||||
retval = wrapped(*args, pathlib.Path(dirname), **kwargs)
|
||||
except:
|
||||
print('Exception in %s, not cleaning up temporary directory %s' % (wrapped, dirname))
|
||||
raise
|
||||
else:
|
||||
shutil.rmtree(dirname)
|
||||
return retval
|
||||
|
||||
return decorator
|
||||
|
||||
LINE = "+----------------------------------------------------------------"
|
||||
|
||||
class AbstractColladaTest(unittest.TestCase):
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
cls.testdir = pathlib.Path(args.testdir)
|
||||
|
||||
def checkdae(self, reference, export):
|
||||
"""
|
||||
collada verifier checks if exported dae file is the same as reference dae
|
||||
"""
|
||||
|
||||
ref = open(reference)
|
||||
exp = open(export)
|
||||
diff=difflib.unified_diff(ref.readlines(), exp.readlines(), lineterm='', n=0)
|
||||
ref.close()
|
||||
exp.close()
|
||||
|
||||
diff_count = 0;
|
||||
for line in diff:
|
||||
error = True
|
||||
for prefix in ('---', '+++', '@@'):
|
||||
# Ignore diff metadata
|
||||
if line.startswith(prefix):
|
||||
error=False
|
||||
break
|
||||
else:
|
||||
# Ignore time stamps
|
||||
for ignore in ('<created>', '<modified>', '<authoring_tool>'):
|
||||
if line[1:].strip().startswith(ignore):
|
||||
error=False
|
||||
break
|
||||
if error:
|
||||
diff_count +=1
|
||||
pline = line.strip()
|
||||
if diff_count == 1:
|
||||
print("\n%s" % LINE)
|
||||
print("|Test has errors:")
|
||||
print(LINE)
|
||||
pre = "reference" if pline[0] == "-" else "generated"
|
||||
print ("| %s:%s"% (pre, pline[1:]))
|
||||
|
||||
if diff_count > 0:
|
||||
print(LINE)
|
||||
print("ref :%s" % reference)
|
||||
print("test:%s" % export)
|
||||
print("%s\n" % LINE)
|
||||
|
||||
return diff_count == 0
|
||||
|
||||
class MeshExportTest4(AbstractColladaTest):
|
||||
@with_tempdir
|
||||
def test_export_animation_suzannes_sample_matrix(self, tempdir: pathlib.Path):
|
||||
test = "suzannes_parent_inverse_sample_10_matrix"
|
||||
reference_dae = self.testdir / Path("%s.dae" % test)
|
||||
outfile = tempdir / Path("%s_out.dae" % test)
|
||||
|
||||
bpy.ops.wm.collada_export(filepath="%s" % str(outfile),
|
||||
check_existing=True,
|
||||
filemode=8,
|
||||
display_type='DEFAULT',
|
||||
sort_method='FILE_SORT_ALPHA',
|
||||
apply_modifiers=True,
|
||||
export_mesh_type=0,
|
||||
export_mesh_type_selection='view',
|
||||
selected=True,
|
||||
include_children=True,
|
||||
include_armatures=True,
|
||||
include_shapekeys=False,
|
||||
deform_bones_only=False,
|
||||
include_animations=True,
|
||||
sample_animations=True,
|
||||
sampling_rate=10,
|
||||
active_uv_only=False,
|
||||
use_texture_copies=True,
|
||||
triangulate=False,
|
||||
use_object_instantiation=True,
|
||||
use_blender_profile=True,
|
||||
sort_by_name=False,
|
||||
export_transformation_type=0,
|
||||
export_transformation_type_selection='matrix',
|
||||
export_texture_type=0,
|
||||
export_texture_type_selection='mat',
|
||||
open_sim=False,
|
||||
limit_precision=True,
|
||||
keep_bind_info=False)
|
||||
|
||||
# Now check the resulting Collada file.
|
||||
if not self.checkdae(reference_dae, outfile):
|
||||
self.fail()
|
||||
|
||||
class MeshExportTest3(AbstractColladaTest):
|
||||
@with_tempdir
|
||||
def test_export_animation_suzannes_sample_locrotscale(self, tempdir: pathlib.Path):
|
||||
test = "suzannes_parent_inverse_sample_10_channels"
|
||||
reference_dae = self.testdir / Path("%s.dae" % test)
|
||||
outfile = tempdir / Path("%s_out.dae" % test)
|
||||
|
||||
bpy.ops.wm.collada_export(filepath="%s" % str(outfile),
|
||||
check_existing=True,
|
||||
filemode=8,
|
||||
display_type='DEFAULT',
|
||||
sort_method='FILE_SORT_ALPHA',
|
||||
apply_modifiers=True,
|
||||
export_mesh_type=0,
|
||||
export_mesh_type_selection='view',
|
||||
selected=True,
|
||||
include_children=True,
|
||||
include_armatures=True,
|
||||
include_shapekeys=False,
|
||||
deform_bones_only=False,
|
||||
include_animations=True,
|
||||
sample_animations=True,
|
||||
sampling_rate=10,
|
||||
active_uv_only=False,
|
||||
use_texture_copies=True,
|
||||
triangulate=False,
|
||||
use_object_instantiation=True,
|
||||
use_blender_profile=True,
|
||||
sort_by_name=False,
|
||||
export_transformation_type=0,
|
||||
export_transformation_type_selection='transrotloc',
|
||||
export_texture_type=0,
|
||||
export_texture_type_selection='mat',
|
||||
open_sim=False,
|
||||
limit_precision=True,
|
||||
keep_bind_info=False)
|
||||
|
||||
# Now check the resulting Collada file.
|
||||
if not self.checkdae(reference_dae, outfile):
|
||||
self.fail()
|
||||
|
||||
class MeshExportTest2(AbstractColladaTest):
|
||||
@with_tempdir
|
||||
def test_export_animation_suzannes_keyframe_matrix(self, tempdir: pathlib.Path):
|
||||
test = "suzannes_parent_inverse_keyframes_matrix"
|
||||
reference_dae = self.testdir / Path("%s.dae" % test)
|
||||
outfile = tempdir / Path("%s_out.dae" % test)
|
||||
|
||||
bpy.ops.wm.collada_export(filepath="%s" % str(outfile),
|
||||
check_existing=True,
|
||||
filemode=8,
|
||||
display_type='DEFAULT',
|
||||
sort_method='FILE_SORT_ALPHA',
|
||||
apply_modifiers=True,
|
||||
export_mesh_type=0,
|
||||
export_mesh_type_selection='view',
|
||||
selected=True,
|
||||
include_children=True,
|
||||
include_armatures=True,
|
||||
include_shapekeys=False,
|
||||
deform_bones_only=False,
|
||||
include_animations=True,
|
||||
sample_animations=False,
|
||||
sampling_rate=1,
|
||||
active_uv_only=False,
|
||||
use_texture_copies=True,
|
||||
triangulate=False,
|
||||
use_object_instantiation=True,
|
||||
use_blender_profile=True,
|
||||
sort_by_name=False,
|
||||
export_transformation_type=0,
|
||||
export_transformation_type_selection='matrix',
|
||||
export_texture_type=0,
|
||||
export_texture_type_selection='mat',
|
||||
open_sim=False,
|
||||
limit_precision=True,
|
||||
keep_bind_info=False)
|
||||
|
||||
# Now check the resulting Collada file.
|
||||
if not self.checkdae(reference_dae, outfile):
|
||||
self.fail()
|
||||
|
||||
class MeshExportTest1(AbstractColladaTest):
|
||||
@with_tempdir
|
||||
def test_export_animation_suzannes_keyframe_locrotscale(self, tempdir: pathlib.Path):
|
||||
test = "suzannes_parent_inverse_keyframes_channels"
|
||||
reference_dae = self.testdir / Path("%s.dae" % test)
|
||||
outfile = tempdir / Path("%s_out.dae" % test)
|
||||
|
||||
bpy.ops.wm.collada_export(filepath="%s" % str(outfile),
|
||||
check_existing=True,
|
||||
filemode=8,
|
||||
display_type='DEFAULT',
|
||||
sort_method='FILE_SORT_ALPHA',
|
||||
apply_modifiers=True,
|
||||
export_mesh_type=0,
|
||||
export_mesh_type_selection='view',
|
||||
selected=True,
|
||||
include_children=True,
|
||||
include_armatures=True,
|
||||
include_shapekeys=False,
|
||||
deform_bones_only=False,
|
||||
include_animations=True,
|
||||
sample_animations=False,
|
||||
sampling_rate=1,
|
||||
active_uv_only=False,
|
||||
use_texture_copies=True,
|
||||
triangulate=False,
|
||||
use_object_instantiation=True,
|
||||
use_blender_profile=True,
|
||||
sort_by_name=False,
|
||||
export_transformation_type=0,
|
||||
export_transformation_type_selection='transrotloc',
|
||||
export_texture_type=0,
|
||||
export_texture_type_selection='mat',
|
||||
open_sim=False,
|
||||
limit_precision=True,
|
||||
keep_bind_info=False)
|
||||
|
||||
# Now check the resulting Collada file.
|
||||
if not self.checkdae(reference_dae, outfile):
|
||||
self.fail()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
sys.argv = [__file__] + (sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [])
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('--testdir', required=True)
|
||||
args, remaining = parser.parse_known_args()
|
||||
unittest.main(argv=sys.argv[0:1]+remaining)
|
||||
Loading…
Reference in New Issue