1
/***************************************************************************
2
* Copyright (c) 2008 Werner Mayer <wmayer[at]users.sourceforge.net> *
4
* This file is part of the FreeCAD CAx development system. *
6
* This library is free software; you can redistribute it and/or *
7
* modify it under the terms of the GNU Library General Public *
8
* License as published by the Free Software Foundation; either *
9
* version 2 of the License, or (at your option) any later version. *
11
* This library is distributed in the hope that it will be useful, *
12
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
13
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
14
* GNU Library General Public License for more details. *
16
* You should have received a copy of the GNU Library General Public *
17
* License along with this library; see the file COPYING.LIB. If not, *
18
* write to the Free Software Foundation, Inc., 59 Temple Place, *
19
* Suite 330, Boston, MA 02111-1307, USA *
21
***************************************************************************/
24
#include "PreCompiled.h"
29
#include "GeometryPyCXX.h"
33
// NOLINTBEGIN(readability-identifier-length)
34
int Py::Vector::Vector_TypeCheck(PyObject* obj)
36
return PyObject_TypeCheck(obj, &(Base::VectorPy::Type));
39
bool Py::Vector::accepts(PyObject* obj) const
41
if (obj && Vector_TypeCheck(obj)) {
44
if (obj && PySequence_Check(obj)) {
45
return (PySequence_Size(obj) == 3);
51
Py::Vector::Vector(const Base::Vector3d& vec)
53
set(new Base::VectorPy(vec), true);
57
Py::Vector::Vector(const Base::Vector3f& vec)
59
set(new Base::VectorPy(vec), true);
63
Py::Vector& Py::Vector::operator=(PyObject* rhsp)
72
Py::Vector& Py::Vector::operator=(const Base::Vector3d& vec)
74
set(new Base::VectorPy(vec), true);
78
Py::Vector& Py::Vector::operator=(const Base::Vector3f& vec)
80
set(new Base::VectorPy(vec), true);
84
Base::Vector3d Py::Vector::toVector() const
86
if (Vector_TypeCheck(ptr())) {
87
return static_cast<Base::VectorPy*>(ptr())->value();
90
return Base::getVectorFromTuple<double>(ptr());
96
Py::PythonType& Vector2dPy::behaviors()
98
return Py::PythonClass<Vector2dPy>::behaviors();
101
PyTypeObject* Vector2dPy::type_object()
103
return Py::PythonClass<Vector2dPy>::type_object();
106
bool Vector2dPy::check(PyObject* py)
108
return Py::PythonClass<Vector2dPy>::check(py);
111
Py::PythonClassObject<Vector2dPy> Vector2dPy::create(const Vector2d& vec)
113
return create(vec.x, vec.y);
116
Py::PythonClassObject<Vector2dPy> Vector2dPy::create(double vx, double vy)
118
Py::Callable class_type(type());
120
arg.setItem(0, Py::Float(vx));
121
arg.setItem(1, Py::Float(vy));
122
Py::PythonClassObject<Vector2dPy> py =
123
Py::PythonClassObject<Vector2dPy>(class_type.apply(arg, Py::Dict()));
127
Vector2dPy::Vector2dPy(Py::PythonClassInstance* self, Py::Tuple& args, Py::Dict& kwds)
128
: Py::PythonClass<Vector2dPy>::PythonClass(self, args, kwds)
132
if (!PyArg_ParseTuple(args.ptr(), "|dd", &vx, &vy)) {
133
throw Py::Exception();
140
Vector2dPy::~Vector2dPy() = default;
142
Py::Object Vector2dPy::repr()
146
std::stringstream str;
148
str << static_cast<std::string>(vx.repr()) << ", " << static_cast<std::string>(vy.repr());
151
return Py::String(str.str()); // NOLINT
154
Py::Object Vector2dPy::getattro(const Py::String& name_)
156
// For Py3 either handle __dict__ or implement __dir__ as shown here:
157
// https://stackoverflow.com/questions/48609111/how-is-dir-implemented-exactly-and-how-should-i-know-it
159
std::string name(name_.as_std_string("utf-8"));
161
if (name == "__dict__") {
163
attr.setItem(Py::String("x"), Py::Float(v.x));
164
attr.setItem(Py::String("y"), Py::Float(v.y));
165
return attr; // NOLINT
168
return Py::Float(v.x); // NOLINT
171
return Py::Float(v.y); // NOLINT
174
return genericGetAttro(name_);
177
int Vector2dPy::setattro(const Py::String& name_, const Py::Object& value)
179
std::string name(name_.as_std_string("utf-8"));
181
if (name == "x" && !value.isNull()) {
182
v.x = static_cast<double>(Py::Float(value));
185
if (name == "y" && !value.isNull()) {
186
v.y = static_cast<double>(Py::Float(value));
190
return genericSetAttro(name_, value);
193
Py::Object Vector2dPy::number_negative()
195
return create(-v.x, -v.y); // NOLINT
198
Py::Object Vector2dPy::number_positive()
200
return create(v.x, v.y); // NOLINT
203
Py::Object Vector2dPy::number_absolute()
205
return create(fabs(v.x), fabs(v.y)); // NOLINT
208
Py::Object Vector2dPy::number_invert()
210
throw Py::TypeError("Not defined");
213
Py::Object Vector2dPy::number_int()
215
throw Py::TypeError("Not defined");
218
Py::Object Vector2dPy::number_float()
220
throw Py::TypeError("Not defined");
223
Py::Object Vector2dPy::number_add(const Py::Object& py)
225
Vector2d vec(Py::toVector2d(py));
227
return create(vec); // NOLINT
230
Py::Object Vector2dPy::number_subtract(const Py::Object& py)
232
Vector2d vec(Py::toVector2d(py));
234
return create(vec); // NOLINT
237
Py::Object Vector2dPy::number_multiply(const Py::Object& py)
239
if (PyObject_TypeCheck(py.ptr(), Vector2dPy::type_object())) {
240
Vector2d vec(Py::toVector2d(py));
241
double scalar = v * vec;
242
return Py::Float(scalar); // NOLINT
244
if (py.isNumeric()) {
245
double scale = static_cast<double>(Py::Float(py));
246
return create(v * scale); // NOLINT
249
throw Py::TypeError("Argument must be Vector2d or Float");
252
Py::Object Vector2dPy::number_remainder(const Py::Object&)
254
throw Py::TypeError("Not defined");
257
Py::Object Vector2dPy::number_divmod(const Py::Object&)
259
throw Py::TypeError("Not defined");
262
Py::Object Vector2dPy::number_lshift(const Py::Object&)
264
throw Py::TypeError("Not defined");
267
Py::Object Vector2dPy::number_rshift(const Py::Object&)
269
throw Py::TypeError("Not defined");
272
Py::Object Vector2dPy::number_and(const Py::Object&)
274
throw Py::TypeError("Not defined");
277
Py::Object Vector2dPy::number_xor(const Py::Object&)
279
throw Py::TypeError("Not defined");
282
Py::Object Vector2dPy::number_or(const Py::Object&)
284
throw Py::TypeError("Not defined");
287
Py::Object Vector2dPy::number_power(const Py::Object&, const Py::Object&)
289
throw Py::TypeError("Not defined");
292
Py::Object Vector2dPy::isNull(const Py::Tuple& args)
295
if (args.size() > 0) {
296
tol = static_cast<double>(Py::Float(args[0]));
298
return Py::Boolean(v.IsNull(tol)); // NOLINT
300
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, isNull)
302
Py::Object Vector2dPy::length(const Py::Tuple&)
304
return Py::Float(v.Length()); // NOLINT
306
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, length)
308
Py::Object Vector2dPy::atan2(const Py::Tuple&)
310
return Py::Float(v.Angle()); // NOLINT
312
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, atan2)
314
Py::Object Vector2dPy::square(const Py::Tuple&)
316
return Py::Float(v.Sqr()); // NOLINT
318
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, square)
320
Py::Object Vector2dPy::scale(const Py::Tuple& args)
322
double value = static_cast<double>(Py::Float(args[0]));
326
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, scale)
328
Py::Object Vector2dPy::rotate(const Py::Tuple& args)
330
double value = static_cast<double>(Py::Float(args[0]));
334
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, rotate)
336
Py::Object Vector2dPy::normalize(const Py::Tuple&)
341
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, normalize)
343
Py::Object Vector2dPy::perpendicular(const Py::Tuple& args)
345
bool value = static_cast<bool>(Py::Boolean(args[0]));
346
Base::Vector2d pnt = v.Perpendicular(value);
347
return create(pnt); // NOLINT
349
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, perpendicular)
351
Py::Object Vector2dPy::distance(const Py::Tuple& args)
353
Base::Vector2d pnt = Py::toVector2d(args[0]);
354
return Py::Float(pnt.Distance(v)); // NOLINT
356
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, distance)
358
Py::Object Vector2dPy::isEqual(const Py::Tuple& args)
360
Base::Vector2d pnt = Py::toVector2d(args[0]);
361
double tol = static_cast<double>(Py::Float(args[1]));
362
return Py::Boolean(v.IsEqual(pnt, tol)); // NOLINT
364
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, isEqual)
366
Py::Object Vector2dPy::getAngle(const Py::Tuple& args)
368
Base::Vector2d vec = Py::toVector2d(args[0]);
369
return Py::Float(v.GetAngle(vec)); // NOLINT
371
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, getAngle)
373
Py::Object Vector2dPy::projectToLine(const Py::Tuple& args)
375
Base::Vector2d pnt1 = Py::toVector2d(args[0]);
376
Base::Vector2d pnt2 = Py::toVector2d(args[1]);
377
v.ProjectToLine(pnt1, pnt2);
380
PYCXX_VARARGS_METHOD_DECL(Vector2dPy, projectToLine)
381
// NOLINTEND(readability-identifier-length)
383
void Vector2dPy::init_type()
385
behaviors().name("Vector2d");
386
behaviors().doc("Vector2d class");
387
behaviors().supportGetattro();
388
behaviors().supportSetattro();
389
behaviors().supportRepr();
390
behaviors().supportNumberType();
392
PYCXX_ADD_VARARGS_METHOD(isNull, isNull, "isNull()");
393
PYCXX_ADD_VARARGS_METHOD(length, length, "length()");
394
PYCXX_ADD_VARARGS_METHOD(atan2, atan2, "atan2()");
395
PYCXX_ADD_VARARGS_METHOD(square, square, "square()");
396
PYCXX_ADD_VARARGS_METHOD(scale, scale, "scale()");
397
PYCXX_ADD_VARARGS_METHOD(rotate, rotate, "rotate()");
398
PYCXX_ADD_VARARGS_METHOD(normalize, normalize, "normalize()");
399
PYCXX_ADD_VARARGS_METHOD(perpendicular, perpendicular, "perpendicular()");
400
PYCXX_ADD_VARARGS_METHOD(distance, distance, "distance()");
401
PYCXX_ADD_VARARGS_METHOD(isEqual, isEqual, "isEqual()");
402
PYCXX_ADD_VARARGS_METHOD(getAngle, getAngle, "getAngle()");
403
PYCXX_ADD_VARARGS_METHOD(projectToLine, projectToLine, "projectToLine()");
405
// Call to make the type ready for use
406
behaviors().readyType();