You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

454 lines
23 KiB

#
# This file is part of Magnum.
#
# Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019,
# 2020, 2021, 2022 Vladimír Vondruš <mosra@centrum.cz>
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
# DEALINGS IN THE SOFTWARE.
#
import os
import sys
import unittest
from corrade import containers
from magnum import *
from magnum import scenetools, trade
import magnum
class Filter(unittest.TestCase):
def test_fields(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(0)
scene_refcount = sys.getrefcount(scene)
self.assertEqual(scene.field_count, 3)
self.assertTrue(scene.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(scene.has_field(trade.SceneField.MESH))
fields_to_keep = containers.BitArray.value_init(scene.field_count)
fields_to_keep[scene.field_id(trade.SceneField.TRANSLATION)] = True
fields_to_keep[scene.field_id(trade.SceneField.MESH)] = True
filtered = scenetools.filter_fields(scene, fields_to_keep)
filtered_refcount = sys.getrefcount(filtered)
self.assertEqual(filtered.field_count, 2)
self.assertTrue(filtered.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(filtered.has_field(trade.SceneField.MESH))
self.assertEqual(sys.getrefcount(scene), scene_refcount + 1)
self.assertIs(filtered.owner, scene)
# Subsequent filtering will still reference the original scene, not the
# intermediates
filtered2 = scenetools.filter_fields(filtered, containers.BitArray.direct_init(filtered.field_count, True))
self.assertEqual(filtered2.field_count, 2)
self.assertTrue(filtered2.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(filtered2.has_field(trade.SceneField.MESH))
self.assertEqual(sys.getrefcount(filtered), filtered_refcount)
self.assertEqual(sys.getrefcount(scene), scene_refcount + 2)
self.assertIs(filtered2.owner, scene)
del filtered
self.assertEqual(sys.getrefcount(scene), scene_refcount + 1)
del filtered2
self.assertEqual(sys.getrefcount(scene), scene_refcount)
def test_fields_invalid_size(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(0)
with self.assertRaisesRegex(AssertionError, "expected 3 bits but got 4"):
scenetools.filter_fields(scene, containers.BitArray.value_init(4))
def test_only_fields(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(0)
scene_refcount = sys.getrefcount(scene)
self.assertEqual(scene.field_count, 3)
# Fields that are not present in the mesh are deliberately ignored
filtered = scenetools.filter_only_fields(scene, [trade.SceneField.LIGHT, trade.SceneField.MESH, trade.SceneField.TRANSLATION])
filtered_refcount = sys.getrefcount(filtered)
self.assertEqual(filtered.field_count, 2)
self.assertTrue(filtered.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(filtered.has_field(trade.SceneField.MESH))
self.assertEqual(sys.getrefcount(scene), scene_refcount + 1)
self.assertIs(filtered.owner, scene)
# Subsequent filtering will still reference the original scene, not the
# intermediates
filtered2 = scenetools.filter_only_fields(filtered, [trade.SceneField.MESH, trade.SceneField.TRANSLATION])
self.assertEqual(filtered2.field_count, 2)
self.assertTrue(filtered2.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(filtered2.has_field(trade.SceneField.MESH))
self.assertEqual(sys.getrefcount(filtered), filtered_refcount)
self.assertEqual(sys.getrefcount(scene), scene_refcount + 2)
self.assertIs(filtered2.owner, scene)
del filtered
self.assertEqual(sys.getrefcount(scene), scene_refcount + 1)
del filtered2
self.assertEqual(sys.getrefcount(scene), scene_refcount)
def test_except_fields(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(0)
scene_refcount = sys.getrefcount(scene)
self.assertEqual(scene.field_count, 3)
# Fields that are not present in the mesh are deliberately ignored
filtered = scenetools.filter_except_fields(scene, [trade.SceneField.SKIN, trade.SceneField.PARENT])
filtered_refcount = sys.getrefcount(filtered)
self.assertEqual(filtered.field_count, 2)
self.assertTrue(filtered.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(filtered.has_field(trade.SceneField.MESH))
self.assertEqual(sys.getrefcount(scene), scene_refcount + 1)
self.assertIs(filtered.owner, scene)
# Subsequent filtering will still reference the original scene, not the
# intermediates
filtered2 = scenetools.filter_except_fields(filtered, [trade.SceneField.PARENT])
self.assertEqual(filtered2.field_count, 2)
self.assertTrue(filtered2.has_field(trade.SceneField.TRANSLATION))
self.assertTrue(filtered2.has_field(trade.SceneField.MESH))
self.assertEqual(sys.getrefcount(filtered), filtered_refcount)
self.assertEqual(sys.getrefcount(scene), scene_refcount + 2)
self.assertIs(filtered2.owner, scene)
del filtered
self.assertEqual(sys.getrefcount(scene), scene_refcount + 1)
del filtered2
self.assertEqual(sys.getrefcount(scene), scene_refcount)
def test_field_entries(self):
importer = trade.ImporterManager().load_and_instantiate('GltfImporter')
importer.open_file(os.path.join(os.path.dirname(__file__), "scene.gltf"))
scene = importer.scene(0)
scene_refcount = sys.getrefcount(scene)
self.assertEqual(scene.field_count, 8)
self.assertEqual(scene.field_size(trade.SceneField.PARENT), 4)
self.assertEqual(scene.field_size(trade.SceneField.IMPORTER_STATE), 4)
self.assertEqual(scene.field_size(trade.SceneField.TRANSFORMATION), 4)
self.assertEqual(scene.field_size(trade.SceneField.CAMERA), 2)
# Remove two parents (and importer state, which is linked), one camera
# and all but one transformation
parents_to_keep = containers.BitArray.direct_init(scene.field_size(trade.SceneField.PARENT), True)
parents_to_keep[0] = False
parents_to_keep[2] = False
transformations_to_keep = containers.BitArray.direct_init(scene.field_size(trade.SceneField.TRANSFORMATION), False)
transformations_to_keep[3] = True
cameras_to_keep = containers.BitArray.direct_init(scene.field_size(trade.SceneField.CAMERA), True)
cameras_to_keep[1] = False
filtered1 = scenetools.filter_field_entries(scene, [
(trade.SceneField.PARENT, parents_to_keep),
(trade.SceneField.IMPORTER_STATE, parents_to_keep),
(trade.SceneField.TRANSFORMATION, transformations_to_keep),
(trade.SceneField.CAMERA, cameras_to_keep)
])
filtered2 = scenetools.filter_field_entries(scene, [
(scene.field_id(trade.SceneField.PARENT), parents_to_keep),
(scene.field_id(trade.SceneField.IMPORTER_STATE), parents_to_keep),
(scene.field_id(trade.SceneField.TRANSFORMATION), transformations_to_keep),
(scene.field_id(trade.SceneField.CAMERA), cameras_to_keep)
])
self.assertEqual(filtered1.field_count, 8)
self.assertEqual(filtered2.field_count, 8)
self.assertEqual(filtered1.field_size(trade.SceneField.PARENT), 2)
self.assertEqual(filtered2.field_size(trade.SceneField.PARENT), 2)
self.assertEqual(filtered1.field_size(trade.SceneField.TRANSFORMATION), 1)
self.assertEqual(filtered2.field_size(trade.SceneField.TRANSFORMATION), 1)
self.assertEqual(filtered1.field_size(trade.SceneField.CAMERA), 1)
self.assertEqual(filtered2.field_size(trade.SceneField.CAMERA), 1)
# The original scene isn't referenced by these, it's a full copy
self.assertEqual(sys.getrefcount(scene), scene_refcount)
def test_field_entries_invalid(self):
importer = trade.ImporterManager().load_and_instantiate('GltfImporter')
importer.open_file(os.path.join(os.path.dirname(__file__), "scene.gltf"))
scene = importer.scene(0)
scene_refcount = sys.getrefcount(scene)
self.assertEqual(scene.field_count, 8)
with self.assertRaisesRegex(AssertionError, "index 8 out of range for 8 fields"):
scenetools.filter_field_entries(scene, [
(8, containers.BitArrayView())
])
with self.assertRaisesRegex(AssertionError, "field at index 1 not found"):
scenetools.filter_field_entries(scene, [
(trade.SceneField.CAMERA, containers.BitArray.value_init(2)),
(trade.SceneField.LIGHT, containers.BitArrayView())
])
with self.assertRaisesRegex(AssertionError, "field at index 2 listed more than once"):
scenetools.filter_field_entries(scene, [
(trade.SceneField.CAMERA, containers.BitArray.value_init(2)),
(trade.SceneField.TRANSLATION, containers.BitArray.value_init(3)),
(trade.SceneField.CAMERA, containers.BitArray.value_init(2))
])
with self.assertRaisesRegex(AssertionError, "field 4 listed more than once"):
scenetools.filter_field_entries(scene, [
(scene.field_id(trade.SceneField.CAMERA), containers.BitArray.value_init(2)),
(scene.field_id(trade.SceneField.TRANSLATION), containers.BitArray.value_init(3)),
(scene.field_id(trade.SceneField.CAMERA), containers.BitArray.value_init(2))
])
with self.assertRaisesRegex(AssertionError, "expected 3 bits for field 3 but got 4"):
scenetools.filter_field_entries(scene, [
(scene.field_id(trade.SceneField.TRANSLATION), containers.BitArray.value_init(4))
])
with self.assertRaisesRegex(AssertionError, "expected 3 bits for field at index 1 but got 4"):
scenetools.filter_field_entries(scene, [
(trade.SceneField.CAMERA, containers.BitArray.value_init(2)),
(trade.SceneField.TRANSLATION, containers.BitArray.value_init(4))
])
with self.assertRaisesRegex(NotImplementedError, "filtering string fields is not implemented yet, sorry"):
scenetools.filter_field_entries(scene, [
(scene.field_id(importer.scene_field_for_name('aString')), containers.BitArray.value_init(1))
])
with self.assertRaisesRegex(NotImplementedError, "filtering string fields is not implemented yet, sorry"):
scenetools.filter_field_entries(scene, [
(importer.scene_field_for_name('aString'), containers.BitArray.value_init(1))
])
with self.assertRaisesRegex(NotImplementedError, "filtering bit fields is not implemented yet, sorry"):
scenetools.filter_field_entries(scene, [
(scene.field_id(importer.scene_field_for_name('yes')), containers.BitArray.value_init(2))
])
with self.assertRaisesRegex(NotImplementedError, "filtering bit fields is not implemented yet, sorry"):
scenetools.filter_field_entries(scene, [
(importer.scene_field_for_name('yes'), containers.BitArray.value_init(2))
])
def test_objects(self):
importer = trade.ImporterManager().load_and_instantiate('GltfImporter')
importer.open_file(os.path.join(os.path.dirname(__file__), "scene.gltf"))
scene = importer.scene(0)
scene_refcount = sys.getrefcount(scene)
self.assertEqual(scene.mapping_bound, 4)
self.assertEqual(scene.field_count, 8)
self.assertEqual(scene.field_size(trade.SceneField.PARENT), 4)
self.assertEqual(scene.field_size(trade.SceneField.TRANSFORMATION), 4)
self.assertEqual(scene.field_size(trade.SceneField.TRANSLATION), 3)
self.assertEqual(scene.field_size(trade.SceneField.CAMERA), 2)
# Two parents, transformations and translations gone
objects_to_keep = containers.BitArray.direct_init(scene.mapping_bound, True)
objects_to_keep[0] = False
objects_to_keep[1] = False
filtered = scenetools.filter_objects(scene, objects_to_keep)
self.assertEqual(filtered.mapping_bound, 4)
self.assertEqual(filtered.field_count, 8)
self.assertEqual(filtered.field_size(trade.SceneField.PARENT), 2)
self.assertEqual(filtered.field_size(trade.SceneField.TRANSFORMATION), 2)
self.assertEqual(filtered.field_size(trade.SceneField.TRANSLATION), 1)
self.assertEqual(filtered.field_size(trade.SceneField.CAMERA), 2)
# The original scene isn't referenced by this one, it's a full copy
self.assertEqual(sys.getrefcount(scene), scene_refcount)
def test_objects_invalid_size(self):
importer = trade.ImporterManager().load_and_instantiate('GltfImporter')
importer.open_file(os.path.join(os.path.dirname(__file__), "scene.gltf"))
scene = importer.scene(0)
self.assertEqual(scene.mapping_bound, 4)
with self.assertRaisesRegex(AssertionError, "expected 4 bits but got 3"):
scenetools.filter_objects(scene, containers.BitArray.value_init(3))
class Hierarchy(unittest.TestCase):
def test_parents_breadth_first_children_depth_first(self):
importer = trade.ImporterManager().load_and_instantiate('GltfImporter')
importer.open_file(os.path.join(os.path.dirname(__file__), "scene.gltf"))
scene = importer.scene(0)
parents_breadth_first = scenetools.parents_breadth_first(scene)
children_depth_first = scenetools.children_depth_first(scene)
self.assertEqual(parents_breadth_first, [
# Root objects first, parent always before all its children
(1, -1),
(2, -1),
(3, 2),
(0, 3)
])
self.assertEqual(children_depth_first, [
# Object 1 has no children
(1, 0),
# Object 2 has one direct child and one grandchild, etc
(2, 2),
(3, 1),
(0, 0)
])
def test_parents_breadth_first_children_depth_first_no_hierarchy(self):
importer = trade.ImporterManager().load_and_instantiate('GltfImporter')
importer.open_file(os.path.join(os.path.dirname(__file__), "scene.gltf"))
scene = scenetools.filter_except_fields(importer.scene(0), [trade.SceneField.PARENT])
with self.assertRaisesRegex(AssertionError, "the scene has no hierarchy"):
scenetools.parents_breadth_first(scene)
with self.assertRaisesRegex(AssertionError, "the scene has no hierarchy"):
scenetools.children_depth_first(scene)
def test_absolute_field_transformations2d(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
# The only way to get a 2D scene is via PrimitiveImporter
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(0)
self.assertTrue(scene.is_2d)
transformations1 = scenetools.absolute_field_transformations2d(scene, trade.SceneField.MESH)
transformations2 = scenetools.absolute_field_transformations2d(scene, scene.field_id(trade.SceneField.MESH))
self.assertEqual(len(transformations1), scene.field_size(trade.SceneField.MESH))
self.assertEqual(len(transformations2), scene.field_size(trade.SceneField.MESH))
self.assertEqual(transformations1[0], Matrix3.translation((-4.5, -3.0)))
self.assertEqual(transformations2[0], Matrix3.translation((-4.5, -3.0)))
def test_absolute_field_transformations3d(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(1)
self.assertTrue(scene.is_3d)
transformations1 = scenetools.absolute_field_transformations3d(scene, trade.SceneField.MESH)
transformations2 = scenetools.absolute_field_transformations3d(scene, scene.field_id(trade.SceneField.MESH))
self.assertEqual(len(transformations1), scene.field_size(trade.SceneField.MESH))
self.assertEqual(len(transformations2), scene.field_size(trade.SceneField.MESH))
self.assertEqual(transformations1[0], Matrix4.translation((-4.5, -3.0, 0.0)))
self.assertEqual(transformations2[0], Matrix4.translation((-4.5, -3.0, 0.0)))
def test_absolute_field_transformations_not_found(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene = importer.scene(0)
with self.assertRaises(KeyError):
scenetools.absolute_field_transformations2d(scene, trade.SceneField.LIGHT)
with self.assertRaises(KeyError):
scenetools.absolute_field_transformations3d(scene, trade.SceneField.LIGHT)
with self.assertRaises(IndexError):
scenetools.absolute_field_transformations2d(scene, scene.field_count)
with self.assertRaises(IndexError):
scenetools.absolute_field_transformations3d(scene, scene.field_count)
def test_absolute_field_transformations_not_2d_not_3d(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene2d = importer.scene(0)
scene3d = importer.scene(1)
self.assertFalse(scene2d.is_3d)
self.assertFalse(scene3d.is_2d)
with self.assertRaisesRegex(AssertionError, "the scene is not 2D"):
scenetools.absolute_field_transformations2d(scene3d, trade.SceneField.MESH)
with self.assertRaisesRegex(AssertionError, "the scene is not 2D"):
scenetools.absolute_field_transformations2d(scene3d, scene2d.field_id(trade.SceneField.MESH))
with self.assertRaisesRegex(AssertionError, "the scene is not 3D"):
scenetools.absolute_field_transformations3d(scene2d, trade.SceneField.MESH)
with self.assertRaisesRegex(AssertionError, "the scene is not 3D"):
scenetools.absolute_field_transformations3d(scene2d, scene3d.field_id(trade.SceneField.MESH))
def test_absolute_field_transformations_no_hierarchy(self):
# Static builds with non-static plugins cause assertions with non-owned
# array deleters used by PrimitiveImporter, skip in that case
if magnum.BUILD_STATIC:
self.skipTest("dynamic PrimitiveImporter doesn't work with a static build")
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter')
importer.open_data(containers.ArrayView())
scene2d = scenetools.filter_except_fields(importer.scene(0), [trade.SceneField.PARENT])
scene3d = scenetools.filter_except_fields(importer.scene(1), [trade.SceneField.PARENT])
self.assertFalse(scene2d.is_3d)
self.assertFalse(scene3d.is_2d)
with self.assertRaisesRegex(AssertionError, "the scene has no hierarchy"):
scenetools.absolute_field_transformations2d(scene2d, trade.SceneField.MESH)
with self.assertRaisesRegex(AssertionError, "the scene has no hierarchy"):
scenetools.absolute_field_transformations2d(scene2d, scene2d.field_id(trade.SceneField.MESH))
with self.assertRaisesRegex(AssertionError, "the scene has no hierarchy"):
scenetools.absolute_field_transformations3d(scene3d, trade.SceneField.MESH)
with self.assertRaisesRegex(AssertionError, "the scene has no hierarchy"):
scenetools.absolute_field_transformations3d(scene3d, scene2d.field_id(trade.SceneField.MESH))