Browse Source

python: expose all Vector variants.

pull/1/head
Vladimír Vondruš 7 years ago
parent
commit
081d32938a
  1. 4
      src/python/magnum/CMakeLists.txt
  2. 4
      src/python/magnum/__init__.py
  3. 2
      src/python/magnum/bootstrap.h
  4. 2
      src/python/magnum/magnum.cpp
  5. 7
      src/python/magnum/math.cpp
  6. 44
      src/python/magnum/math.h
  7. 247
      src/python/magnum/math.vector.h
  8. 90
      src/python/magnum/math.vectorfloat.cpp
  9. 83
      src/python/magnum/math.vectorintegral.cpp
  10. 133
      src/python/magnum/test/test_math.py

4
src/python/magnum/CMakeLists.txt

@ -25,7 +25,9 @@
set(magnum_SRCS
magnum.cpp
math.cpp)
math.cpp
math.vectorfloat.cpp
math.vectorintegral.cpp)
pybind11_add_module(magnum ${magnum_SRCS})
target_include_directories(magnum PRIVATE ${PROJECT_SOURCE_DIR}/src/python)

4
src/python/magnum/__init__.py

@ -31,4 +31,8 @@ __all__ = [
'Deg', 'Rad',
'BoolVector2', 'BoolVector3', 'BoolVector4',
'Vector2', 'Vector3', 'Vector4',
'Vector2d', 'Vector3d', 'Vector4d',
'Vector2i', 'Vector3i', 'Vector4i',
'Vector2ui', 'Vector3ui', 'Vector4ui',
]

2
src/python/magnum/bootstrap.h

@ -34,6 +34,8 @@ using namespace Magnum;
namespace py = pybind11;
void math(py::module& root, py::module& m);
void mathVectorFloat(py::module& root, py::module& m);
void mathVectorIntegral(py::module& root, py::module& m);
}

2
src/python/magnum/magnum.cpp

@ -34,4 +34,6 @@ PYBIND11_MODULE(_magnum, m) {
py::module math = m.def_submodule("math");
magnum::math(m, math);
magnum::mathVectorFloat(m, math);
magnum::mathVectorIntegral(m, math);
}

7
src/python/magnum/math.cpp

@ -31,17 +31,12 @@
#include <Magnum/Math/BoolVector.h>
#include "magnum/bootstrap.h"
#include "magnum/math.h"
namespace magnum {
namespace {
template<class T> std::string repr(const T& value) {
std::ostringstream out;
Debug{&out, Debug::Flag::NoNewlineAtTheEnd} << value;
return out.str();
}
template<class T> void angle(py::class_<T>& c) {
/*
Missing APIs:

44
src/python/magnum/math.h

@ -0,0 +1,44 @@
#ifndef magnum_math_h
#define magnum_math_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 <sstream>
#include <Corrade/Utility/Debug.h>
#include <Magnum/Magnum.h>
#include "magnum/bootstrap.h"
namespace magnum {
template<class T> std::string repr(const T& value) {
std::ostringstream out;
Debug{&out, Debug::Flag::NoNewlineAtTheEnd} << value;
return out.str();
}
}
#endif

247
src/python/magnum/math.vector.h

@ -0,0 +1,247 @@
#ifndef magnum_math_vector_h
#define magnum_math_vector_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/operators.h>
#include <Magnum/Math/Vector4.h>
#include "magnum/math.h"
namespace magnum {
/* Things common for vectors of all sizes and types */
template<class T> void vector(py::module& m, py::class_<T>& c) {
/*
Missing APIs:
from(T*)
Type
construction from different types
VectorNi * VectorN and variants (5)
*/
m
.def("dot", [](const T& a, const T& b) { return Math::dot(a, b); },
"Dot product of two vectors");
c
/* Constructors */
.def_static("zero_init", []() {
return T{Math::ZeroInit};
}, "Construct a zero vector")
.def(py::init(), "Default constructor")
.def(py::init<typename T::Type>(), "Construct a vector with one value for all components")
/* Comparison */
.def(py::self == py::self, "Equality comparison")
.def(py::self != py::self, "Non-equality comparison")
.def(py::self < py::self, "Component-wise less than comparison")
.def(py::self > py::self, "Component-wise greater than comparison")
.def(py::self <= py::self, "Component-wise less than or equal comparison")
.def(py::self >= py::self, "Component-wise greater than or equal comparison")
/* Set / get */
.def("__setitem__", [](T& self, std::size_t i, typename T::Type value) {
self[i] = value;
}, "Set a value at given position")
.def("__getitem__", [](const T& self, std::size_t i) {
return self[i];
}, "Value at given position")
/* Operators */
.def(-py::self, "Negated vector")
.def(py::self += py::self, "Add and assign a vector")
.def(py::self + py::self, "Add a vector")
.def(py::self -= py::self, "Subtract and assign a vector")
.def(py::self - py::self, "Subtract a vector")
.def(py::self *= typename T::Type{}, "Multiply with a scalar and assign")
.def(py::self * typename T::Type{}, "Multiply with a scalar")
.def(py::self /= typename T::Type{}, "Divide with a scalar and assign")
.def(py::self / typename T::Type{}, "Divide with a scalar")
.def(py::self *= py::self, "Multiply a vector component-wise and assign")
.def(py::self * py::self, "Multiply a vector component-wise")
.def(py::self /= py::self, "Divide a vector component-wise and assign")
.def(py::self / py::self, "Divide a vector component-wise")
.def(typename T::Type{} * py::self, "Multiply a scalar with a vector")
.def(typename T::Type{} / py::self, "Divide a vector with a scalar and invert")
/* Member functions common for floating-point and integer types */
.def("is_zero", &T::isZero, "Whether the vector is zero")
.def("dot", static_cast<typename T::Type(T::*)() const>(&T::dot), "Dot product of the vector")
.def("flipped", &T::flipped, "Flipped vector")
.def("sum", &T::sum, "Sum of values in the vector")
.def("product", &T::product, "Product of values in the vector")
.def("min", &T::min, "Minimal value in the vector")
.def("max", &T::max, "Maximal value in the vector")
.def("minmax", &T::minmax, "Minimal and maximal value in the vector")
.def("__repr__", repr<T>, "Object representation");
/* Vector length */
char lenDocstring[] = "Vector size. Returns _.";
lenDocstring[sizeof(lenDocstring) - 3] = '0' + T::Size;
c.def_static("__len__", []() { return int(T::Size); }, lenDocstring);
}
template<class T> void vector2(py::class_<Math::Vector2<T>>& c) {
c
/* Constructors */
.def(py::init<T, T>(), "Constructor")
/* Static constructors */
.def_static("x_axis", &Math::Vector2<T>::xAxis,
"Vector in a direction of X axis (right)", py::arg("length") = T(1))
.def_static("y_axis", &Math::Vector2<T>::yAxis,
"Vector in a direction of Y axis (up)", py::arg("length") = T(1))
.def_static("x_scale", &Math::Vector2<T>::xScale,
"Scaling vector in a direction of X axis (width)", py::arg("scale"))
.def_static("y_scale", &Math::Vector2<T>::yScale,
"Scaling vector in a direction of Y axis (height)", py::arg("scale"))
/* Methods */
.def("perpendicular", &Math::Vector2<T>::perpendicular,
"Perpendicular vector")
/* Properties */
.def_property("x",
static_cast<T(Math::Vector2<T>::*)() const>(&Math::Vector2<T>::x),
[](Math::Vector2<T>& self, T value) { self.x() = value; },
"X component")
.def_property("y",
static_cast<T(Math::Vector2<T>::*)() const>(&Math::Vector2<T>::y),
[](Math::Vector2<T>& self, T value) { self.y() = value; },
"Y component");
}
template<class T> void vector3(py::class_<Math::Vector3<T>>& c) {
c
/* Constructors */
.def(py::init<T, T, T>(), "Constructor")
/* Static constructors */
.def_static("x_axis", &Math::Vector3<T>::xAxis,
"Vector in a direction of X axis (right)", py::arg("length") = T(1))
.def_static("y_axis", &Math::Vector3<T>::yAxis,
"Vector in a direction of Y axis (up)", py::arg("length") = T(1))
.def_static("z_axis", &Math::Vector3<T>::zAxis,
"Vector in a direction of Z axis (backward)", py::arg("length") = T(1))
.def_static("x_scale", &Math::Vector3<T>::xScale,
"Scaling vector in a direction of X axis (width)", py::arg("scale"))
.def_static("y_scale", &Math::Vector3<T>::yScale,
"Scaling vector in a direction of Y axis (height)", py::arg("scale"))
.def_static("z_scale", &Math::Vector3<T>::zScale,
"Scaling vector in a direction of Z axis (depth)", py::arg("scale"))
/* Properties */
.def_property("x",
static_cast<T(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::x),
[](Math::Vector3<T>& self, T value) { self.x() = value; },
"X component")
.def_property("y",
static_cast<T(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::y),
[](Math::Vector3<T>& self, T value) { self.y() = value; },
"Y component")
.def_property("z",
static_cast<T(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::z),
[](Math::Vector3<T>& self, T value) { self.z() = value; },
"Z component")
.def_property("r",
static_cast<T(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::r),
[](Math::Vector3<T>& self, T value) { self.r() = value; },
"R component")
.def_property("g",
static_cast<T(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::g),
[](Math::Vector3<T>& self, T value) { self.g() = value; },
"G component")
.def_property("b",
static_cast<T(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::b),
[](Math::Vector3<T>& self, T value) { self.b() = value; },
"B component")
.def_property("xy",
static_cast<const Math::Vector2<T>(Math::Vector3<T>::*)() const>(&Math::Vector3<T>::xy),
[](Math::Vector3<T>& self, const Math::Vector2<T>& value) { self.xy() = value; },
"XY part of the vector");
}
template<class T> void vector4(py::class_<Math::Vector4<T>>& c) {
c
/* Constructors */
.def(py::init<T, T, T, T>(), "Constructor")
/* Properties */
.def_property("x",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::x),
[](Math::Vector4<T>& self, T value) { self.x() = value; },
"X component")
.def_property("y",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::y),
[](Math::Vector4<T>& self, T value) { self.y() = value; },
"Y component")
.def_property("z",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::z),
[](Math::Vector4<T>& self, T value) { self.z() = value; },
"Z component")
.def_property("w",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::w),
[](Math::Vector4<T>& self, T value) { self.w() = value; },
"W component")
.def_property("r",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::r),
[](Math::Vector4<T>& self, T value) { self.r() = value; },
"R component")
.def_property("g",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::g),
[](Math::Vector4<T>& self, T value) { self.g() = value; },
"G component")
.def_property("b",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::b),
[](Math::Vector4<T>& self, T value) { self.b() = value; },
"B component")
.def_property("a",
static_cast<T(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::a),
[](Math::Vector4<T>& self, T value) { self.a() = value; },
"A component")
.def_property("xyz",
static_cast<const Math::Vector3<T>(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::xyz),
[](Math::Vector4<T>& self, const Math::Vector3<T>& value) { self.xyz() = value; },
"XYZ part of the vector")
.def_property("rgb",
static_cast<const Math::Vector3<T>(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::rgb),
[](Math::Vector4<T>& self, const Math::Vector3<T>& value) { self.rgb() = value; },
"RGB part of the vector")
.def_property("xy",
static_cast<const Math::Vector2<T>(Math::Vector4<T>::*)() const>(&Math::Vector4<T>::xy),
[](Math::Vector4<T>& self, const Math::Vector2<T>& value) { self.xy() = value; },
"XY part of the vector");
}
}
#endif

90
src/python/magnum/math.vectorfloat.cpp

@ -0,0 +1,90 @@
/*
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/math.vector.h"
namespace magnum {
namespace {
template<class T> void vectorFloat(py::module& m, py::class_<T>& c) {
m
.def("angle", [](const T& a, const T& b) { return Radd(Math::angle(a, b)); },
"Angle between normalized vectors", py::arg("normalized_a"), py::arg("normalized_b"));
c
.def("is_normalized", &T::isNormalized, "Whether the vector is normalized")
.def("length", &T::length, "Vector length")
/* Cast needed because these are enabled only for floats */
.def("length_inverted", static_cast<typename T::Type(T::*)() const>(&T::lengthInverted), "Inverse vector length")
.def("normalized", static_cast<T(T::*)() const>(&T::normalized),
"Normalized vector (of unit length)")
.def("resized", static_cast<T(T::*)(typename T::Type) const>(&T::resized),
"Resized vector")
.def("projected", [](const T& self, const T& line) {
return self.projected(line);
}, "Vector projected onto a line")
.def("projected_onto_normalized", [](const T& self, const T& line) {
return self.projectedOntoNormalized(line);
}, "Vector projected onto a normalized line");
}
template<class T> void vectorsFloat(py::module& m, py::class_<Math::Vector2<T>>& vector2_, py::class_<Math::Vector3<T>>& vector3_, py::class_<Math::Vector4<T>>& vector4_) {
vector2_
.def("aspect_ratio", static_cast<T(Math::Vector2<T>::*)() const>(&Math::Vector2<T>::aspectRatio),
"Aspect ratio")
.def("cross", static_cast<T(*)(const Math::Vector2<T>&, const Math::Vector2<T>&)>(Math::cross),
"2D cross product");
vector<Math::Vector2<T>>(m, vector2_);
vectorFloat<Math::Vector2<T>>(m, vector2_);
vector2<T>(vector2_);
vector3_
.def("cross", static_cast<Math::Vector3<T>(*)(const Math::Vector3<T>&, const Math::Vector3<T>&)>(Math::cross),
"Cross product");
vector<Math::Vector3<T>>(m, vector3_);
vectorFloat<Math::Vector3<T>>(m, vector3_);
vector3<T>(vector3_);
vector<Math::Vector4<T>>(m, vector4_);
vectorFloat<Math::Vector4<T>>(m, vector4_);
vector4<T>(vector4_);
}
}
void mathVectorFloat(py::module& root, py::module& m) {
py::class_<Vector2> vector2{root, "Vector2", "Two-component float vector"};
py::class_<Vector3> vector3{root, "Vector3", "Threee-component float vector"};
py::class_<Vector4> vector4{root, "Vector4", "Four-component float vector"};
py::class_<Vector2d> vector2d{root, "Vector2d", "Two-component double vector"};
py::class_<Vector3d> vector3d{root, "Vector3d", "Threee-component double vector"};
py::class_<Vector4d> vector4d{root, "Vector4d", "Four-component double vector"};
vectorsFloat<Float>(m, vector2, vector3, vector4);
vectorsFloat<Double>(m, vector2d, vector3d, vector4d);
}
}

83
src/python/magnum/math.vectorintegral.cpp

@ -0,0 +1,83 @@
/*
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/math.vector.h"
namespace magnum {
namespace {
template<class T> void vectorIntegral(py::class_<T>& c) {
c
.def(py::self %= typename T::Type{}, "Do modulo of an integral vector and assign")
.def(py::self % typename T::Type{}, "Modulo of an integral vector")
.def(py::self %= py::self, "Do module of two integral vectors and assign")
.def(py::self % py::self, "Modulo of two integral vectors")
.def(~py::self, "Bitwise NOT of an integral vector")
.def(py::self &= py::self, "Do bitwise AND of two integral vectors and assign")
.def(py::self & py::self, "Bitwise AND of two integral vectors")
.def(py::self |= py::self, "Do bitwise OR of two integral vectors and assign")
.def(py::self | py::self, "Bitwise OR of two integral vectors")
.def(py::self ^= py::self, "Do bitwise XOR of two integral vectors and assign")
.def(py::self ^ py::self, "Bitwise XOR of two integral vectors")
.def(py::self <<= typename T::Type{}, "Do bitwise left shift of an integral vector and assign")
.def(py::self << typename T::Type{}, "Bitwise left shift of an integral vector")
.def(py::self >>= typename T::Type{}, "Do bitwise right shift of an integral vector and assign")
.def(py::self >> typename T::Type{}, "Bitwise right shift of an integral vector")
.def(py::self *= Float{}, "Multiply an integral vector with a floating-point number and assign")
.def(py::self * Float{}, "Multiply an integral vector with a floating-point number")
.def(Float{} * py::self, "Multiply a floating-point number with an integral vector")
.def(py::self /= Float{}, "Divide an integral vector with a floating-point number and assign")
.def(py::self / Float{}, "Divide an integral vector with a floating-point number");
}
template<class T> void vectorsIntegral(py::module& m, py::class_<Math::Vector2<T>>& vector2_, py::class_<Math::Vector3<T>>& vector3_, py::class_<Math::Vector4<T>>& vector4_) {
vector<Math::Vector2<T>>(m, vector2_);
vectorIntegral<Math::Vector2<T>>(vector2_);
vector2<T>(vector2_);
vector<Math::Vector3<T>>(m, vector3_);
vectorIntegral<Math::Vector3<T>>(vector3_);
vector3<T>(vector3_);
vector<Math::Vector4<T>>(m, vector4_);
vectorIntegral<Math::Vector4<T>>(vector4_);
vector4<T>(vector4_);
}
}
void mathVectorIntegral(py::module& root, py::module& m) {
py::class_<Vector2i> vector2i{root, "Vector2i", "Two-component signed integer vector"};
py::class_<Vector3i> vector3i{root, "Vector3i", "Threee-component signed integral vector"};
py::class_<Vector4i> vector4i{root, "Vector4i", "Four-component signed integral vector"};
py::class_<Vector2ui> vector2ui{root, "Vector2ui", "Two-component unsigned integral vector"};
py::class_<Vector3ui> vector3ui{root, "Vector3ui", "Threee-component unsigned integral vector"};
py::class_<Vector4ui> vector4ui{root, "Vector4ui", "Four-component unsigned integral vector"};
vectorsIntegral<Int>(m, vector2i, vector3i, vector4i);
vectorsIntegral<UnsignedInt>(m, vector2ui, vector3ui, vector4ui);
}
}

133
src/python/magnum/test/test_math.py

@ -102,3 +102,136 @@ class Constants(unittest.TestCase):
def test(self):
self.assertAlmostEqual(math.sqrt2**2, 2.0, 6)
self.assertAlmostEqual(math.sqrt3**2, 3.0)
class Vector(unittest.TestCase):
def test_init(self):
a = Vector4i()
b = Vector3d.zero_init()
c = Vector2i(44, -3)
self.assertEqual(a, Vector4i(0, 0, 0, 0))
self.assertEqual(b, Vector3d(0.0, 0.0, 0.0))
self.assertEqual(c, Vector2i(44, -3))
def test_static_methods(self):
self.assertEqual(Vector2.y_scale(5), Vector2(1, 5))
self.assertEqual(Vector3d.z_axis(-3), Vector3d(0, 0, -3))
self.assertEqual(Vector3i.x_axis(), Vector3i(1, 0, 0))
def test_length(self):
self.assertEqual(Vector3.__len__(), 3)
#self.assertEqual(len(Vector3), 3) TODO: Y not?
self.assertEqual(len(Vector4i()), 4)
def test_properties(self):
a = Vector2i()
a.x = 1
a.y = 2
self.assertEqual(a.x, 1)
self.assertEqual(a.y, 2)
self.assertEqual(a, Vector2i(1, 2))
a = Vector3()
a.x = 1.0
a.y = 2.0
a.z = 3.0
self.assertEqual(a.x, 1.0)
self.assertEqual(a.y, 2.0)
self.assertEqual(a.z, 3.0)
self.assertEqual(a, Vector3(1.0, 2.0, 3.0))
a.xy = (-1.0, -2.0)
self.assertEqual(a.xy, Vector2(-1.0, -2.0))
self.assertEqual(a, Vector3(-1.0, -2.0, 3.0))
a = Vector3()
a.r = 1.0
a.g = 2.0
a.b = 3.0
self.assertEqual(a.r, 1.0)
self.assertEqual(a.g, 2.0)
self.assertEqual(a.b, 3.0)
self.assertEqual(a, Vector3(1.0, 2.0, 3.0))
a = Vector4d()
a.x = 1.0
a.y = 2.0
a.z = 3.0
a.w = 4.0
self.assertEqual(a.x, 1.0)
self.assertEqual(a.y, 2.0)
self.assertEqual(a.z, 3.0)
self.assertEqual(a.w, 4.0)
self.assertEqual(a, Vector4d(1.0, 2.0, 3.0, 4.0))
a = Vector4d()
a.r = 1.0
a.g = 2.0
a.b = 3.0
a.a = 4.0
self.assertEqual(a.r, 1.0)
self.assertEqual(a.g, 2.0)
self.assertEqual(a.b, 3.0)
self.assertEqual(a.a, 4.0)
self.assertEqual(a, Vector4d(1.0, 2.0, 3.0, 4.0))
a.xy = (-1.0, -2.0)
self.assertEqual(a.xy, Vector2d(-1.0, -2.0))
self.assertEqual(a, Vector4d(-1.0, -2.0, 3.0, 4.0))
a.xyz = (0.5, 0.25, 0.125)
self.assertEqual(a.xyz, Vector3d(0.5, 0.25, 0.125))
self.assertEqual(a, Vector4d(0.5, 0.25, 0.125, 4.0))
def test_properties_rgb(self):
a = Vector3()
a.r = 1.0
a.g = 2.0
a.b = 3.0
self.assertEqual(a.r, 1.0)
self.assertEqual(a.g, 2.0)
self.assertEqual(a.b, 3.0)
self.assertEqual(a, Vector3(1.0, 2.0, 3.0))
a = Vector4d()
a.r = 1.0
a.g = 2.0
a.b = 3.0
a.a = 4.0
self.assertEqual(a.r, 1.0)
self.assertEqual(a.g, 2.0)
self.assertEqual(a.b, 3.0)
self.assertEqual(a.a, 4.0)
self.assertEqual(a, Vector4d(1.0, 2.0, 3.0, 4.0))
a.rgb = (0.5, 0.25, 0.125)
self.assertEqual(a.rgb, Vector3d(0.5, 0.25, 0.125))
self.assertEqual(a, Vector4d(0.5, 0.25, 0.125, 4.0))
def test_set_get(self):
a = Vector3(1.0, 3.14, -13.37)
self.assertAlmostEqual(a[1], 3.14, 6)
a[2] = 0.13
self.assertEqual(a, Vector3(1.0, 3.14, 0.13))
b = Vector4i(3, 4, 5, 6)
b.b *= 3
b.xy += Vector2i(1, -1)
self.assertEqual(b, Vector4i(4, 3, 15, 6))
def test_ops(self):
self.assertEqual(math.dot(Vector2(0.5, 3.0), Vector2(2.0, 0.5)), 2.5)
self.assertEqual(Deg(math.angle(
Vector2(0.5, 3.0).normalized(),
Vector2(2.0, 0.5).normalized())), Deg(66.5014333443446))
self.assertEqual(Vector3(1.0, 2.0, 0.3).projected(Vector3.y_axis()),
Vector3.y_axis(2.0))
self.assertEqual(Vector3(1.0, 2.0, 0.3).projected_onto_normalized(Vector3.y_axis()),
Vector3.y_axis(2.0))
def test_ops_number_on_the_left(self):
self.assertEqual(2.0*Vector2(1.0, -3.0), Vector2(2.0, -6.0))
self.assertEqual(6.0/Vector2(2.0, -3.0), Vector2(3.0, -2.0))
def test_repr(self):
self.assertEqual(repr(Vector3(1.0, 3.14, -13.37)), 'Vector(1, 3.14, -13.37)')

Loading…
Cancel
Save