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