codywinch
1a0bedd640
Some of the files are not accurate to blender-v4.3-release and not due to any goo engine additions Likely due to merges from beyond 4.3 that made it into 4.2 since it is an LTS and gets more support So merging these changes to keep it in line with the release version of 4.3
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 tests/data/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=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)
|