Vladimír Vondruš
3 years ago
16 changed files with 340 additions and 7 deletions
@ -0,0 +1,40 @@
|
||||
.. |
||||
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. |
||||
.. |
||||
|
||||
.. py:function:: magnum.scenetools.flatten_transformation_hierarchy2d |
||||
:raise KeyError: If :p:`field` does not exist in :p:`scene` |
||||
:raise IndexError: If :p:`field_id` negative or not less than |
||||
:ref:`trade.SceneData.field_count` |
||||
:raise AssertionError: If :p:`scene` is not 2D |
||||
:raise AssertionError: If :p:`scene` does not have |
||||
:ref:`trade.SceneField.PARENT` |
||||
|
||||
.. py:function:: magnum.scenetools.flatten_transformation_hierarchy3d |
||||
:raise KeyError: If :p:`field` does not exist in :p:`scene` |
||||
:raise IndexError: If :p:`field_id` negative or not less than |
||||
:ref:`trade.SceneData.field_count` |
||||
:raise AssertionError: If :p:`scene` is not 2D |
||||
:raise AssertionError: If :p:`scene` does not have |
||||
:ref:`trade.SceneField.PARENT` |
||||
@ -0,0 +1,138 @@
|
||||
/*
|
||||
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. |
||||
*/ |
||||
|
||||
#include <pybind11/pybind11.h> |
||||
#include <pybind11/stl.h> /* for std::vector */ |
||||
#include <Corrade/Containers/Optional.h> |
||||
#include <Corrade/Containers/ArrayViewStl.h> |
||||
#include <Magnum/Math/Matrix3.h> |
||||
#include <Magnum/Math/Matrix4.h> |
||||
#include <Magnum/SceneTools/FlattenTransformationHierarchy.h> |
||||
#include <Magnum/Trade/SceneData.h> |
||||
|
||||
#include "magnum/bootstrap.h" |
||||
|
||||
namespace magnum { |
||||
|
||||
void scenetools(py::module_& m) { |
||||
m.doc() = "Scene manipulation and optimization tools"; |
||||
|
||||
#ifndef MAGNUM_BUILD_STATIC |
||||
/* These are a part of the same module in the static build, no need to
|
||||
import (also can't import because there it's _magnum.*) */ |
||||
py::module_::import("magnum.trade"); |
||||
#endif |
||||
|
||||
m |
||||
.def("flatten_transformation_hierarchy2d", [](const Trade::SceneData& scene, Trade::SceneField field, const Matrix3& globalTransformation) { |
||||
const Containers::Optional<UnsignedInt> fieldId = scene.findFieldId(field); |
||||
if(!fieldId) { |
||||
PyErr_SetNone(PyExc_KeyError); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.is2D()) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene is not 2D"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.hasField(Trade::SceneField::Parent)) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene has no hierarchy"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
|
||||
/** @todo maybe do a caster for arrays, finally?! */ |
||||
std::vector<Matrix3> out(scene.fieldSize(*fieldId)); |
||||
SceneTools::flattenTransformationHierarchy2DInto(scene, *fieldId, out, globalTransformation); |
||||
return out; |
||||
}, "Flatten a 2D transformation hierarchy for given field", py::arg("scene"), py::arg("field"), py::arg("global_transformation") = Matrix3{}) |
||||
.def("flatten_transformation_hierarchy2d", [](const Trade::SceneData& scene, UnsignedInt fieldId, const Matrix3& globalTransformation) { |
||||
if(fieldId >= scene.fieldCount()) { |
||||
PyErr_SetNone(PyExc_IndexError); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.is2D()) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene is not 2D"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.hasField(Trade::SceneField::Parent)) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene has no hierarchy"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
|
||||
/** @todo maybe do a caster for arrays, finally?! */ |
||||
std::vector<Matrix3> out(scene.fieldSize(fieldId)); |
||||
SceneTools::flattenTransformationHierarchy2DInto(scene, fieldId, out, globalTransformation); |
||||
return out; |
||||
}, "Flatten a 2D transformation hierarchy for given field ID", py::arg("scene"), py::arg("field_id"), py::arg("global_transformation") = Matrix3{}) |
||||
.def("flatten_transformation_hierarchy3d", [](const Trade::SceneData& scene, Trade::SceneField field, const Matrix4& globalTransformation) { |
||||
const Containers::Optional<UnsignedInt> fieldId = scene.findFieldId(field); |
||||
if(!fieldId) { |
||||
PyErr_SetNone(PyExc_KeyError); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.is3D()) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene is not 3D"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.hasField(Trade::SceneField::Parent)) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene has no hierarchy"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
|
||||
/** @todo maybe do a caster for arrays, finally?! */ |
||||
std::vector<Matrix4> out(scene.fieldSize(*fieldId)); |
||||
SceneTools::flattenTransformationHierarchy3DInto(scene, *fieldId, out, globalTransformation); |
||||
return out; |
||||
}, "Flatten a 3D transformation hierarchy for given field", py::arg("scene"), py::arg("field"), py::arg("global_transformation") = Matrix4{}) |
||||
.def("flatten_transformation_hierarchy3d", [](const Trade::SceneData& scene, UnsignedInt fieldId, const Matrix4& globalTransformation) { |
||||
if(fieldId >= scene.fieldCount()) { |
||||
PyErr_SetNone(PyExc_IndexError); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.is3D()) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene is not 3D"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
if(!scene.hasField(Trade::SceneField::Parent)) { |
||||
PyErr_SetString(PyExc_AssertionError, "the scene has no hierarchy"); |
||||
throw py::error_already_set{}; |
||||
} |
||||
|
||||
/** @todo maybe do a caster for arrays, finally?! */ |
||||
std::vector<Matrix4> out(scene.fieldSize(fieldId)); |
||||
SceneTools::flattenTransformationHierarchy3DInto(scene, fieldId, out, globalTransformation); |
||||
return out; |
||||
}, "Flatten a 3D transformation hierarchy for given field ID", py::arg("scene"), py::arg("field_id"), py::arg("global_transformation") = Matrix4{}); |
||||
} |
||||
|
||||
} |
||||
|
||||
#ifndef MAGNUM_BUILD_STATIC |
||||
/* TODO: remove declaration when https://github.com/pybind/pybind11/pull/1863
|
||||
is released */ |
||||
extern "C" PYBIND11_EXPORT PyObject* PyInit_scenetools(); |
||||
PYBIND11_MODULE(scenetools, m) { |
||||
magnum::scenetools(m); |
||||
} |
||||
#endif |
||||
@ -0,0 +1,119 @@
|
||||
# |
||||
# 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 unittest |
||||
|
||||
from corrade import containers |
||||
from magnum import * |
||||
from magnum import scenetools, trade |
||||
|
||||
class FlattenTransformationHierarchy(unittest.TestCase): |
||||
def test_2d(self): |
||||
# 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) |
||||
|
||||
transformations = scenetools.flatten_transformation_hierarchy2d(scene, trade.SceneField.MESH) |
||||
self.assertEqual(len(transformations), scene.field_size(trade.SceneField.MESH)) |
||||
self.assertEqual(transformations[0], Matrix3.translation((-4.5, -3.0))) |
||||
|
||||
def test_2d_field_id(self): |
||||
# 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) |
||||
|
||||
mesh_field = scene.field_id(trade.SceneField.MESH) |
||||
|
||||
transformations = scenetools.flatten_transformation_hierarchy2d(scene, mesh_field) |
||||
self.assertEqual(len(transformations), scene.field_size(mesh_field)) |
||||
self.assertEqual(transformations[0], Matrix3.translation((-4.5, -3.0))) |
||||
|
||||
def test_3d(self): |
||||
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter') |
||||
importer.open_data(containers.ArrayView()) |
||||
|
||||
scene = importer.scene(1) |
||||
self.assertTrue(scene.is_3d) |
||||
|
||||
transformations = scenetools.flatten_transformation_hierarchy3d(scene, trade.SceneField.MESH) |
||||
self.assertEqual(len(transformations), scene.field_size(trade.SceneField.MESH)) |
||||
self.assertEqual(transformations[0], Matrix4.translation((-4.5, -3.0, 0.0))) |
||||
|
||||
def test_3d_field_id(self): |
||||
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter') |
||||
importer.open_data(containers.ArrayView()) |
||||
|
||||
scene = importer.scene(1) |
||||
self.assertTrue(scene.is_3d) |
||||
|
||||
mesh_field = scene.field_id(trade.SceneField.MESH) |
||||
|
||||
transformations = scenetools.flatten_transformation_hierarchy3d(scene, mesh_field) |
||||
self.assertEqual(len(transformations), scene.field_size(mesh_field)) |
||||
self.assertEqual(transformations[0], Matrix4.translation((-4.5, -3.0, 0.0))) |
||||
|
||||
def test_field_not_found(self): |
||||
importer = trade.ImporterManager().load_and_instantiate('PrimitiveImporter') |
||||
importer.open_data(containers.ArrayView()) |
||||
|
||||
scene = importer.scene(0) |
||||
|
||||
with self.assertRaises(KeyError): |
||||
scenetools.flatten_transformation_hierarchy2d(scene, trade.SceneField.LIGHT) |
||||
with self.assertRaises(KeyError): |
||||
scenetools.flatten_transformation_hierarchy3d(scene, trade.SceneField.LIGHT) |
||||
with self.assertRaises(IndexError): |
||||
scenetools.flatten_transformation_hierarchy2d(scene, scene.field_count) |
||||
with self.assertRaises(IndexError): |
||||
scenetools.flatten_transformation_hierarchy3d(scene, scene.field_count) |
||||
|
||||
def test_not_2d_not_3d(self): |
||||
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.flatten_transformation_hierarchy2d(scene3d, trade.SceneField.MESH) |
||||
with self.assertRaisesRegex(AssertionError, "the scene is not 2D"): |
||||
scenetools.flatten_transformation_hierarchy2d(scene3d, scene2d.field_id(trade.SceneField.MESH)) |
||||
with self.assertRaisesRegex(AssertionError, "the scene is not 3D"): |
||||
scenetools.flatten_transformation_hierarchy3d(scene2d, trade.SceneField.MESH) |
||||
with self.assertRaisesRegex(AssertionError, "the scene is not 3D"): |
||||
scenetools.flatten_transformation_hierarchy3d(scene2d, scene3d.field_id(trade.SceneField.MESH)) |
||||
|
||||
def test_no_hierarchy(self): |
||||
# TODO implement once it's possible to create / import a scene without |
||||
# the parent field |
||||
pass |
||||
Loading…
Reference in new issue