150 lines
4.6 KiB
Python
150 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
# SPDX-FileCopyrightText: 2018-2023 Blender Authors
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
"""
|
|
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 MeshExportTest(AbstractColladaTest):
|
|
@with_tempdir
|
|
def test_export_single_mesh(self, tempdir: pathlib.Path):
|
|
test = "mesh_simple_001"
|
|
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=False,
|
|
export_mesh_type=0,
|
|
export_mesh_type_selection="view",
|
|
selected=False,
|
|
include_children=False,
|
|
include_armatures=False,
|
|
include_shapekeys=True,
|
|
deform_bones_only=False,
|
|
sampling_rate=0,
|
|
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=False,
|
|
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)
|