Vladimír Vondruš
7 years ago
6 changed files with 266 additions and 164 deletions
@ -0,0 +1,156 @@ |
|||||||
|
#ifndef magnum_scenegraph_h |
||||||
|
#define magnum_scenegraph_h |
||||||
|
/*
|
||||||
|
This file is part of Magnum. |
||||||
|
|
||||||
|
Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019 |
||||||
|
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 <Corrade/Utility/FormatStl.h> |
||||||
|
#include <Magnum/SceneGraph/Object.h> |
||||||
|
#include <Magnum/SceneGraph/Scene.h> |
||||||
|
|
||||||
|
#include "Magnum/SceneGraph/Python.h" |
||||||
|
|
||||||
|
#include "magnum/bootstrap.h" |
||||||
|
|
||||||
|
namespace magnum { |
||||||
|
|
||||||
|
template<class Transformation> void scene(py::class_<SceneGraph::Scene<Transformation>>& c) { |
||||||
|
c.def(py::init(), "Constructor"); |
||||||
|
} |
||||||
|
|
||||||
|
template<UnsignedInt dimensions, class T, class Transformation> void object(py::class_<SceneGraph::Object<Transformation>, SceneGraph::PyObject<SceneGraph::Object<Transformation>>, SceneGraph::AbstractObject<dimensions, T>, SceneGraph::PyObjectHolder<SceneGraph::Object<Transformation>>>& c) { |
||||||
|
c |
||||||
|
.def(py::init_alias<SceneGraph::PyObject<SceneGraph::Object<Transformation>>*>(), |
||||||
|
"Constructor", py::arg("parent") = nullptr) |
||||||
|
.def(py::init_alias<SceneGraph::Scene<Transformation>*>(), |
||||||
|
"Constructor", py::arg("parent") = nullptr) |
||||||
|
|
||||||
|
/* Properties */ |
||||||
|
.def_property_readonly("scene", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self) { |
||||||
|
return static_cast<SceneGraph::Scene<Transformation>*>(self.scene()); |
||||||
|
}, "Scene or None if the object is not a part of any scene") |
||||||
|
.def_property("parent", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self) { |
||||||
|
return static_cast<SceneGraph::PyObject<SceneGraph::Object<Transformation>>*>(self.parent()); |
||||||
|
}, [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, py::object parentobj) { |
||||||
|
SceneGraph::Object<Transformation>* parent; |
||||||
|
if(py::isinstance<SceneGraph::PyObject<SceneGraph::Object<Transformation>>>(parentobj)) |
||||||
|
parent = py::cast<SceneGraph::PyObject<SceneGraph::Object<Transformation>>*>(parentobj); |
||||||
|
else if(py::isinstance<SceneGraph::Scene<Transformation>>(parentobj)) |
||||||
|
parent = py::cast<SceneGraph::Scene<Transformation>*>(parentobj); |
||||||
|
else if(py::isinstance<py::none>(parentobj)) |
||||||
|
parent = nullptr; |
||||||
|
else throw py::type_error{Utility::formatString("expected Scene, Object or None, got {}", std::string(py::str{parentobj.get_type()}))}; |
||||||
|
|
||||||
|
/* Decrease refcount if a parent is removed, increase it if a
|
||||||
|
parent gets added */ |
||||||
|
if(self.parent() && !parent) py::cast(&self).dec_ref(); |
||||||
|
else if(!self.parent() && parent) py::cast(&self).inc_ref(); |
||||||
|
|
||||||
|
self.setParent(parent); |
||||||
|
}, "Parent object or None if this is the root object") |
||||||
|
|
||||||
|
/* Transformation APIs common to all implementations */ |
||||||
|
.def_property("transformation", |
||||||
|
&SceneGraph::PyObject<SceneGraph::Object<Transformation>>::transformation, |
||||||
|
&SceneGraph::PyObject<SceneGraph::Object<Transformation>>::setTransformation, |
||||||
|
"Object transformation") |
||||||
|
.def("absolute_transformation", &SceneGraph::PyObject<SceneGraph::Object<Transformation>>::absoluteTransformation, |
||||||
|
"Transformation relative to the root object") |
||||||
|
.def("reset_transformation", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self) { |
||||||
|
self.resetTransformation(); |
||||||
|
}, "Reset the transformation") |
||||||
|
.def("transform", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const typename Transformation::DataType& transformation) { |
||||||
|
self.transform(transformation); |
||||||
|
}, "Transform the object") |
||||||
|
.def("transform_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const typename Transformation::DataType& transformation) { |
||||||
|
self.transformLocal(transformation); |
||||||
|
}, "Transform the object as a local transformation") |
||||||
|
.def("translate", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const VectorTypeFor<dimensions, typename Transformation::DataType::Type>& vector) { |
||||||
|
self.translate(vector); |
||||||
|
}, "Translate the object") |
||||||
|
.def("translate_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const VectorTypeFor<dimensions, typename Transformation::DataType::Type>& vector) { |
||||||
|
self.translateLocal(vector); |
||||||
|
}, "Translate the object as a local transformation"); |
||||||
|
} |
||||||
|
|
||||||
|
template<UnsignedInt dimensions, class T, class Transformation> void object2D(py::class_<SceneGraph::Object<Transformation>, SceneGraph::PyObject<SceneGraph::Object<Transformation>>, SceneGraph::AbstractObject<dimensions, T>, SceneGraph::PyObjectHolder<SceneGraph::Object<Transformation>>>& c) { |
||||||
|
c |
||||||
|
.def("rotate", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotate(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object") |
||||||
|
.def("rotate_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateLocal(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object as a local transformation"); |
||||||
|
} |
||||||
|
|
||||||
|
template<UnsignedInt dimensions, class T, class Transformation> void object3D(py::class_<SceneGraph::Object<Transformation>, SceneGraph::PyObject<SceneGraph::Object<Transformation>>, SceneGraph::AbstractObject<dimensions, T>, SceneGraph::PyObjectHolder<SceneGraph::Object<Transformation>>>& c) { |
||||||
|
c |
||||||
|
.def("rotate", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle, const Math::Vector3<typename Transformation::DataType::Type>& normalizedAxis) { |
||||||
|
self.rotate(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle), normalizedAxis); |
||||||
|
}, "Rotate the object as a local transformation", py::arg("angle"), py::arg("normalized_axis")) |
||||||
|
.def("rotate_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle, const Math::Vector3<typename Transformation::DataType::Type>& normalizedAxis) { |
||||||
|
self.rotateLocal(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle), normalizedAxis); |
||||||
|
}, "Rotate the object as a local transformation", py::arg("angle"), py::arg("normalized_axis")) |
||||||
|
.def("rotate_x", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateX(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object around X axis") |
||||||
|
.def("rotate_x_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateXLocal(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object around X axis as a local transformation") |
||||||
|
.def("rotate_y", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateY(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object around Y axis") |
||||||
|
.def("rotate_y_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateYLocal(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object around Y axis as a local transformation") |
||||||
|
.def("rotate_z", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateZ(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object around Z axis") |
||||||
|
.def("rotate_z_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const Radd angle) { |
||||||
|
self.rotateZLocal(static_cast<Math::Rad<typename Transformation::DataType::Type>>(angle)); |
||||||
|
}, "Rotate the object around Z axis as a local transformation"); |
||||||
|
} |
||||||
|
|
||||||
|
template<UnsignedInt dimensions, class T, class Transformation> void objectScale(py::class_<SceneGraph::Object<Transformation>, SceneGraph::PyObject<SceneGraph::Object<Transformation>>, SceneGraph::AbstractObject<dimensions, T>, SceneGraph::PyObjectHolder<SceneGraph::Object<Transformation>>>& c) { |
||||||
|
c |
||||||
|
.def("scale", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const VectorTypeFor<dimensions, typename Transformation::DataType::Type>& vector) { |
||||||
|
self.scale(vector); |
||||||
|
}, "Scale the object") |
||||||
|
.def("scale_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const VectorTypeFor<dimensions, typename Transformation::DataType::Type>& vector) { |
||||||
|
self.scaleLocal(vector); |
||||||
|
}, "Scale the object as a local transformation") |
||||||
|
.def("reflect", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const VectorTypeFor<dimensions, typename Transformation::DataType::Type>& vector) { |
||||||
|
self.reflect(vector); |
||||||
|
}, "Reflect the object") |
||||||
|
.def("reflect_local", [](SceneGraph::PyObject<SceneGraph::Object<Transformation>>& self, const VectorTypeFor<dimensions, typename Transformation::DataType::Type>& vector) { |
||||||
|
self.reflectLocal(vector); |
||||||
|
}, "Reflect the object as a local transformation"); |
||||||
|
} |
||||||
|
|
||||||
|
void scenegraphMatrix(py::module& m); |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,54 @@ |
|||||||
|
/*
|
||||||
|
This file is part of Magnum. |
||||||
|
|
||||||
|
Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019 |
||||||
|
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 <Magnum/SceneGraph/MatrixTransformation2D.h> |
||||||
|
#include <Magnum/SceneGraph/MatrixTransformation3D.h> |
||||||
|
|
||||||
|
#include "scenegraph.h" |
||||||
|
|
||||||
|
namespace magnum { |
||||||
|
|
||||||
|
void scenegraphMatrix(py::module& m) { |
||||||
|
py::module matrix = m.def_submodule("matrix"); |
||||||
|
matrix.doc() = "General matrix-based scene graph implementation"; |
||||||
|
|
||||||
|
py::class_<SceneGraph::Scene<SceneGraph::MatrixTransformation2D>> scene2D_{matrix, "Scene2D", "Two-dimensional scene with matrix-based transformation implementation"}; |
||||||
|
scene(scene2D_); |
||||||
|
|
||||||
|
py::class_<SceneGraph::Scene<SceneGraph::MatrixTransformation3D>> scene3D_{matrix, "Scene3D", "Three-dimensional scene with matrix-based transformation implementation"}; |
||||||
|
scene(scene3D_); |
||||||
|
|
||||||
|
py::class_<SceneGraph::Object<SceneGraph::MatrixTransformation2D>, SceneGraph::PyObject<SceneGraph::Object<SceneGraph::MatrixTransformation2D>>, SceneGraph::AbstractObject2D, SceneGraph::PyObjectHolder<SceneGraph::Object<SceneGraph::MatrixTransformation2D>>> object2D_{matrix, "Object2D", "Two-dimensional object with matrix-based transformation implementation"}; |
||||||
|
object(object2D_); |
||||||
|
object2D(object2D_); |
||||||
|
objectScale(object2D_); |
||||||
|
|
||||||
|
py::class_<SceneGraph::Object<SceneGraph::MatrixTransformation3D>, SceneGraph::PyObject<SceneGraph::Object<SceneGraph::MatrixTransformation3D>>, SceneGraph::AbstractObject3D, SceneGraph::PyObjectHolder<SceneGraph::Object<SceneGraph::MatrixTransformation3D>>> object3D_{matrix, "Object3D", "Three-dimensional object with matrix-based transformation implementation"}; |
||||||
|
object(object3D_); |
||||||
|
object3D(object3D_); |
||||||
|
objectScale(object3D_); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,51 @@ |
|||||||
|
# |
||||||
|
# This file is part of Magnum. |
||||||
|
# |
||||||
|
# Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019 |
||||||
|
# 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 sys |
||||||
|
import unittest |
||||||
|
|
||||||
|
from magnum import * |
||||||
|
from magnum import scenegraph |
||||||
|
from magnum.scenegraph.matrix import Object3D, Scene3D |
||||||
|
|
||||||
|
class Object(unittest.TestCase): |
||||||
|
def test_transformation(self): |
||||||
|
scene = Scene3D() |
||||||
|
|
||||||
|
a = Object3D(scene) |
||||||
|
a.rotate_local(Deg(35.0), Vector3.x_axis()) |
||||||
|
self.assertEqual(a.transformation, Matrix4.rotation_x(Deg(35.0))) |
||||||
|
self.assertEqual(a.absolute_transformation(), Matrix4.rotation_x(Deg(35.0))) |
||||||
|
|
||||||
|
b = Object3D(a) |
||||||
|
b.translate((3.0, 4.0, 5.0)) |
||||||
|
self.assertEqual(b.transformation, Matrix4.translation((3.0, 4.0, 5.0))) |
||||||
|
self.assertEqual(b.absolute_transformation(), |
||||||
|
Matrix4.rotation_x(Deg(35.0))@ |
||||||
|
Matrix4.translation((3.0, 4.0, 5.0))) |
||||||
|
|
||||||
|
c = Object3D(scene) |
||||||
|
self.assertEqual(c.transformation, Matrix4.identity_init()) |
||||||
|
self.assertEqual(c.absolute_transformation(), Matrix4.identity_init()) |
||||||
Loading…
Reference in new issue