def test_roundtrip_prelit(self):
hierarchy_name = 'testhiera_skl'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True, prelit=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK', prelit=True)
]
hlod = get_hlod('TestModelName', hierarchy_name)
self.filepath = self.outpath() + 'output'
create_data(self, meshes, hlod, hierarchy, [], None, None, [])
# export
self.filepath = self.outpath() + 'output'
export_settings = {'mode': 'HM', 'use_existing_skeleton': False}
save_data(self, export_settings)
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output.w3d'
load(self)
# check created objects
self.assertTrue('output' in bpy.data.objects)
self.assertTrue('output' in bpy.data.armatures)
amt = bpy.data.armatures['output']
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
def test_roundtrip_splitted(self):
hierarchy_name = 'testname_skl'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod(hierarchy_name, hierarchy_name)
boxes = [get_collision_box()]
animation = get_animation(hierarchy_name, xml=True)
self.set_format('W3X')
self.filepath = self.outpath() + 'output_skn'
create_data(self, meshes, hlod, hierarchy, boxes, animation, None)
# export
self.filepath = self.outpath() + 'output_skn'
export_settings = {
'mode': 'HM',
'compression': 'U',
'individual_files': True,
'create_texture_xmls': True,
'use_existing_skeleton': False
}
save_data(self, export_settings)
# check created files
self.assertTrue(os.path.exists(self.outpath() + 'output_skn.w3x'))
self.assertTrue(os.path.exists(self.outpath() +
'output_skn.sword.w3x'))
self.assertTrue(
os.path.exists(self.outpath() + 'output_skn.soldier.w3x'))
self.assertTrue(os.path.exists(self.outpath() +
'output_skn.TRUNK.w3x'))
self.assertTrue(
os.path.exists(self.outpath() + 'output_skn.BOUNDINGBOX.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'testname_skl.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'texture.xml'))
# check created include entries
root = find_root(self, self.outpath() + 'output_skn.w3x')
self.assertEqual(6, len(root.find('Includes').findall('Include')))
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output_skn.w3x'
load(self)
# check created objects
self.assertTrue('testname_skl' in bpy.data.objects)
self.assertTrue('testname_skl' in bpy.data.armatures)
amt = bpy.data.armatures['testname_skl']
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
def test_hierarchy_name_is_container_name_on_HM_and_not_use_existing_skeleton(
self):
hierarchy_name = 'TestName'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod(hierarchy_name, hierarchy_name)
self.filepath = self.outpath() + 'output'
create_data(self, meshes, hlod, hierarchy)
# export
self.filepath = self.outpath() + 'output'
export_settings = {
'mode': 'HM',
'compression': 'U',
'use_existing_skeleton': False
}
save_data(self, export_settings)
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output.w3d'
load(self)
# check created objects
self.assertTrue('output' in bpy.data.armatures)
amt = bpy.data.armatures['output']
self.assertEqual(7, len(amt.bones))
def test_roundtrip_hlod_only_import(self):
hierarchy_name = 'testname_skl'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod(hierarchy_name, hierarchy_name)
create_data(self, meshes, hlod, hierarchy)
# export
self.set_format('W3X')
self.filepath = self.outpath() + 'output'
export_settings = {
'mode': 'HM',
'compression': 'U',
'individual_files': True,
'create_texture_xmls': True,
'use_existing_skeleton': False
}
save_data(self, export_settings)
# remove includes
root = find_root(self, self.outpath() + 'output.w3x')
for child in root:
if child.tag == 'Includes':
root.remove(child)
write(root, self.outpath() + 'output.w3x')
# check created files
self.assertTrue(os.path.exists(self.outpath() + 'output.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'output.sword.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'output.soldier.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'output.TRUNK.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'testname_skl.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'texture.xml'))
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output.w3x'
load(self)
# check created objects
self.assertEqual(2, len(bpy.data.collections))
self.assertTrue('testname_skl' in bpy.data.objects)
self.assertTrue('testname_skl' in bpy.data.armatures)
amt = bpy.data.armatures['testname_skl']
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
def test_roundtrip_no_armature(self):
hierarchy_name = 'TestModelName'
hierarchy = get_hierarchy(hierarchy_name)
hierarchy.pivots = [
get_roottransform(),
get_hierarchy_pivot(name='sword', parent=0),
get_hierarchy_pivot(name='soldier', parent=0),
get_hierarchy_pivot(name='TRUNK', parent=0)
]
meshes = [
get_mesh(name='sword'),
get_mesh(name='soldier'),
get_mesh(name='TRUNK')
]
hlod = get_hlod('TestModelName', hierarchy_name)
hlod.lod_arrays[0].sub_objects = [
get_hlod_sub_object(bone=1, name='containerName.sword'),
get_hlod_sub_object(bone=2, name='containerName.soldier'),
get_hlod_sub_object(bone=3, name='containerName.TRUNK')
]
hlod.lod_arrays[0].header.model_count = len(
hlod.lod_arrays[0].sub_objects)
self.set_format('W3X')
self.filepath = self.outpath() + 'output'
create_data(self, meshes, hlod, hierarchy)
# export
self.filepath = self.outpath() + 'output'
export_settings = {
'mode': 'HM',
'compression': 'U',
'individual_files': False,
'create_texture_xmls': True,
'use_existing_skeleton': False
}
save_data(self, export_settings)
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output.w3x'
load(self)
# check created objects
self.assertTrue('TestModelName' in bpy.data.objects)
self.assertTrue('TestModelName' in bpy.data.armatures)
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
def test_roundtrip_mesh_is_camera_aligned(self):
mesh_struct = get_mesh('camera_aligned')
mesh_struct.header.attrs |= GEOMETRY_TYPE_CAMERA_ALIGNED
create_mesh(self, mesh_struct, bpy.context.scene.collection)
self.set_format('W3X')
self.filepath = self.outpath() + 'output'
export_settings = {
'mode': 'HM',
'compression': 'U',
'individual_files': True,
'create_texture_xmls': True,
'use_existing_skeleton': False
}
save_data(self, export_settings)
def test_roundtrip_HAM(self):
hierarchy_name = 'testname'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod(hierarchy_name, hierarchy_name)
boxes = [get_collision_box()]
animation = get_animation(hierarchy_name, xml=True)
self.set_format('W3X')
self.filepath = self.outpath() + 'output'
create_data(self, meshes, hlod, hierarchy, boxes, animation, None)
# export
self.filepath = self.outpath() + 'output'
export_settings = {
'mode': 'HAM',
'compression': 'U',
'individual_files': False,
'create_texture_xmls': True
}
save_data(self, export_settings)
# check created files
self.assertTrue(os.path.exists(self.outpath() + 'output.w3x'))
self.assertTrue(os.path.exists(self.outpath() + 'texture.xml'))
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output.w3x'
load(self)
# check created objects
self.assertTrue('testname' in bpy.data.objects)
self.assertTrue('testname' in bpy.data.armatures)
amt = bpy.data.armatures['testname']
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
def test_roundtrip_texture_name_with_dots(self):
hierarchy_name = 'testname_skl'
hierarchy = get_hierarchy(hierarchy_name)
mesh = get_mesh(name='sword', skin=True)
mesh.textures = [
get_texture(name='tex.with.dots.in.name.but.not.used.dds'),
get_texture(name='another.tex.with.dots.in.name.dds')
]
meshes = [mesh]
hlod = get_hlod(hierarchy_name, hierarchy_name)
self.set_format('W3X')
self.filepath = self.outpath() + 'output_skn'
create_data(self, meshes, hlod, hierarchy)
# export
self.filepath = self.outpath() + 'output_skn'
export_settings = {
'mode': 'HM',
'compression': 'U',
'individual_files': True,
'create_texture_xmls': True,
'use_existing_skeleton': True
}
save_data(self, export_settings)
# check created files
self.assertTrue(os.path.exists(self.outpath() + 'output_skn.w3x'))
self.assertTrue(os.path.exists(self.outpath() +
'output_skn.sword.w3x'))
self.assertTrue(
os.path.exists(self.outpath() +
'another.tex.with.dots.in.name.xml'))
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output_skn.w3x'
load(self)
self.assertTrue('sword' in bpy.data.objects)
def test_roundtrip_HAM_tc_animation(self):
hierarchy_name = 'TestName'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod(hierarchy_name, hierarchy_name)
boxes = [get_collision_box()]
dazzles = [get_dazzle()]
comp_animation = get_compressed_animation(hierarchy_name)
self.filepath = self.outpath() + 'output'
create_data(self, meshes, hlod, hierarchy, boxes, None, comp_animation,
dazzles)
# export
self.filepath = self.outpath() + 'output'
export_settings = {'mode': 'HAM', 'compression': 'TC'}
save_data(self, export_settings)
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output.w3d'
load(self)
# check created objects
self.assertTrue('output' in bpy.data.armatures)
amt = bpy.data.armatures['output']
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
self.assertTrue('Brakelight' in bpy.data.objects)
def execute(self, context):
print_version(self.info)
if self.will_save_settings:
self.save_settings(context)
export_settings = {
'mode': self.export_mode,
'compression': self.animation_compression,
'use_existing_skeleton': self.use_existing_skeleton,
'individual_files': self.individual_files,
'create_texture_xmls': self.create_texture_xmls
}
return save_data(self, export_settings)
def test_roundtrip(self):
hierarchy_name = 'testhiera_skl'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod('TestModelName', hierarchy_name)
boxes = [get_collision_box()]
animation = get_animation(hierarchy_name, xml=True)
self.filepath = self.outpath() + 'output_skn'
create_data(self, meshes, hlod, hierarchy, boxes, animation, None)
self.set_format('W3X')
# export
self.filepath = self.outpath() + 'output_skn'
export_settings = {
'mode': 'HM',
'individual_files': False,
'use_existing_skeleton': True,
'create_texture_xmls': True
}
save_data(self, export_settings)
self.filepath = self.outpath() + 'testhiera_skl'
export_settings['mode'] = 'H'
save_data(self, export_settings)
self.filepath = self.outpath() + 'output_ani'
export_settings['mode'] = 'A'
export_settings['compression'] = 'U'
save_data(self, export_settings)
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output_skn.w3x'
load(self)
self.filepath = self.outpath() + 'output_ani.w3x'
load(self)
# check created objects
self.assertTrue(hierarchy_name in bpy.data.objects)
self.assertTrue(hierarchy_name in bpy.data.armatures)
amt = bpy.data.armatures[hierarchy_name]
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
def test_roundtrip_compressed_animation(self):
hierarchy_name = 'testhiera_skl'
hierarchy = get_hierarchy(hierarchy_name)
meshes = [
get_mesh(name='sword', skin=True),
get_mesh(name='soldier', skin=True),
get_mesh(name='TRUNK')
]
hlod = get_hlod('testmodelname', hierarchy_name)
boxes = [get_collision_box()]
dazzles = [get_dazzle()]
comp_animation = get_compressed_animation(hierarchy_name)
self.filepath = self.outpath() + 'output_skn'
create_data(self, meshes, hlod, hierarchy, boxes, None, comp_animation,
dazzles)
# export
self.filepath = self.outpath() + 'output_skn'
export_settings = {'mode': 'HM', 'use_existing_skeleton': True}
save_data(self, export_settings)
self.filepath = self.outpath() + 'testhiera_skl'
export_settings['mode'] = 'H'
save_data(self, export_settings)
self.filepath = self.outpath() + 'output_comp_ani'
export_settings['mode'] = 'A'
export_settings['compression'] = 'TC'
save_data(self, export_settings)
# reset scene
bpy.ops.wm.read_homefile(app_template='')
# import
self.filepath = self.outpath() + 'output_skn.w3d'
load(self)
self.filepath = self.outpath() + 'output_comp_ani.w3d'
load(self)
# check created objects
self.assertTrue(hierarchy_name.upper() in bpy.data.objects)
self.assertTrue(hierarchy_name.upper() in bpy.data.armatures)
amt = bpy.data.armatures[hierarchy_name.upper()]
self.assertEqual(7, len(amt.bones))
self.assertTrue('sword' in bpy.data.objects)
self.assertTrue('soldier' in bpy.data.objects)
self.assertTrue('TRUNK' in bpy.data.objects)
self.assertTrue('Brakelight' in bpy.data.objects)
