483 lines
15 KiB
C++
483 lines
15 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2008 Werner Mayer <wmayer[at]users.sourceforge.net> *
|
|
* *
|
|
* This file is part of the FreeCAD CAx development system. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Library General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU Library General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Library General Public *
|
|
* License along with this library; see the file COPYING.LIB. If not, *
|
|
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
|
* Suite 330, Boston, MA 02111-1307, USA *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
|
|
#include "PreCompiled.h"
|
|
#ifndef _PreComp_
|
|
# include <gp_Ax1.hxx>
|
|
# include <gp_Dir.hxx>
|
|
# include <gp_Pnt.hxx>
|
|
# include <gp_Vec.hxx>
|
|
# include <gp_Trsf.hxx>
|
|
# include <Geom_Geometry.hxx>
|
|
# include <Geom_Curve.hxx>
|
|
# include <Geom_Surface.hxx>
|
|
# include <Precision.hxx>
|
|
# include <Standard_Failure.hxx>
|
|
|
|
# include <boost/uuid/uuid_io.hpp>
|
|
#endif
|
|
|
|
#include <Base/GeometryPyCXX.h>
|
|
#include <Base/Matrix.h>
|
|
#include <Base/MatrixPy.h>
|
|
#include <Base/Vector3D.h>
|
|
#include <Base/VectorPy.h>
|
|
#include <Base/Rotation.h>
|
|
#include <Base/Placement.h>
|
|
#include <Base/PlacementPy.h>
|
|
|
|
#include "OCCError.h"
|
|
#include "Geometry.h"
|
|
#include "GeometryPy.h"
|
|
#include "GeometryPy.cpp"
|
|
|
|
#include "GeometryExtensionPy.h"
|
|
#include "TopoShape.h"
|
|
#include "TopoShapePy.h"
|
|
|
|
using namespace Part;
|
|
|
|
// returns a string which represents the object e.g. when printed in python
|
|
std::string GeometryPy::representation(void) const
|
|
{
|
|
return "<Geometry object>";
|
|
}
|
|
|
|
PyObject *GeometryPy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper
|
|
{
|
|
// never create such objects with the constructor
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"You cannot create an instance of the abstract class 'Geometry'.");
|
|
return 0;
|
|
}
|
|
|
|
// constructor method
|
|
int GeometryPy::PyInit(PyObject* /*args*/, PyObject* /*kwd*/)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::mirror(PyObject *args)
|
|
{
|
|
PyObject* o;
|
|
if (PyArg_ParseTuple(args, "O!", &(Base::VectorPy::Type),&o)) {
|
|
Base::Vector3d vec = static_cast<Base::VectorPy*>(o)->value();
|
|
gp_Pnt pnt(vec.x, vec.y, vec.z);
|
|
getGeometryPtr()->handle()->Mirror(pnt);
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_Clear();
|
|
PyObject* axis;
|
|
if (PyArg_ParseTuple(args, "O!O!", &(Base::VectorPy::Type),&o,
|
|
&(Base::VectorPy::Type),&axis)) {
|
|
Base::Vector3d pnt = static_cast<Base::VectorPy*>(o)->value();
|
|
Base::Vector3d dir = static_cast<Base::VectorPy*>(axis)->value();
|
|
gp_Ax1 ax1(gp_Pnt(pnt.x,pnt.y,pnt.z), gp_Dir(dir.x,dir.y,dir.z));
|
|
getGeometryPtr()->handle()->Mirror(ax1);
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "either a point (vector) or axis (vector, vector) must be given");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::rotate(PyObject *args)
|
|
{
|
|
PyObject* o;
|
|
if (!PyArg_ParseTuple(args, "O!", &(Base::PlacementPy::Type),&o))
|
|
return 0;
|
|
|
|
Base::Placement* plm = static_cast<Base::PlacementPy*>(o)->getPlacementPtr();
|
|
Base::Rotation rot(plm->getRotation());
|
|
Base::Vector3d pnt, dir;
|
|
double angle;
|
|
|
|
rot.getValue(dir, angle);
|
|
pnt = plm->getPosition();
|
|
|
|
gp_Ax1 ax1(gp_Pnt(pnt.x,pnt.y,pnt.z), gp_Dir(dir.x,dir.y,dir.z));
|
|
getGeometryPtr()->handle()->Rotate(ax1, angle);
|
|
Py_Return;
|
|
}
|
|
|
|
PyObject* GeometryPy::scale(PyObject *args)
|
|
{
|
|
PyObject* o;
|
|
double scale;
|
|
Base::Vector3d vec;
|
|
if (PyArg_ParseTuple(args, "O!d", &(Base::VectorPy::Type),&o, &scale)) {
|
|
vec = static_cast<Base::VectorPy*>(o)->value();
|
|
gp_Pnt pnt(vec.x, vec.y, vec.z);
|
|
getGeometryPtr()->handle()->Scale(pnt, scale);
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_Clear();
|
|
if (PyArg_ParseTuple(args, "O!d", &PyTuple_Type,&o, &scale)) {
|
|
vec = Base::getVectorFromTuple<double>(o);
|
|
gp_Pnt pnt(vec.x, vec.y, vec.z);
|
|
getGeometryPtr()->handle()->Scale(pnt, scale);
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "either vector or tuple and float expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::transform(PyObject *args)
|
|
{
|
|
PyObject* o;
|
|
if (!PyArg_ParseTuple(args, "O!", &(Base::MatrixPy::Type),&o))
|
|
return 0;
|
|
Base::Matrix4D mat = static_cast<Base::MatrixPy*>(o)->value();
|
|
gp_Trsf trf;
|
|
trf.SetValues(mat[0][0],mat[0][1],mat[0][2],mat[0][3],
|
|
mat[1][0],mat[1][1],mat[1][2],mat[1][3],
|
|
mat[2][0],mat[2][1],mat[2][2],mat[2][3]
|
|
#if OCC_VERSION_HEX < 0x060800
|
|
, 0.00001,0.00001
|
|
#endif
|
|
); //precision was removed in OCCT CR0025194
|
|
getGeometryPtr()->handle()->Transform(trf);
|
|
Py_Return;
|
|
}
|
|
|
|
PyObject* GeometryPy::translate(PyObject *args)
|
|
{
|
|
PyObject* o;
|
|
Base::Vector3d vec;
|
|
if (PyArg_ParseTuple(args, "O!", &(Base::VectorPy::Type),&o)) {
|
|
vec = static_cast<Base::VectorPy*>(o)->value();
|
|
gp_Vec trl(vec.x, vec.y, vec.z);
|
|
getGeometryPtr()->handle()->Translate(trl);
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_Clear();
|
|
if (PyArg_ParseTuple(args, "O!", &PyTuple_Type,&o)) {
|
|
vec = Base::getVectorFromTuple<double>(o);
|
|
gp_Vec trl(vec.x, vec.y, vec.z);
|
|
getGeometryPtr()->handle()->Translate(trl);
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "either vector or tuple expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::copy(PyObject *args)
|
|
{
|
|
if (!PyArg_ParseTuple(args, ""))
|
|
return NULL;
|
|
|
|
Part::Geometry* geom = this->getGeometryPtr();
|
|
PyTypeObject* type = this->GetType();
|
|
PyObject* cpy = 0;
|
|
// let the type object decide
|
|
if (type->tp_new)
|
|
cpy = type->tp_new(type, this, 0);
|
|
if (!cpy) {
|
|
PyErr_SetString(PyExc_TypeError, "failed to create copy of geometry");
|
|
return 0;
|
|
}
|
|
|
|
Part::GeometryPy* geompy = static_cast<Part::GeometryPy*>(cpy);
|
|
// the PyMake function must have created the corresponding instance of the 'Geometry' subclass
|
|
// so delete it now to avoid a memory leak
|
|
if (geompy->_pcTwinPointer) {
|
|
Part::Geometry* clone = static_cast<Part::Geometry*>(geompy->_pcTwinPointer);
|
|
delete clone;
|
|
}
|
|
geompy->_pcTwinPointer = geom->copy();
|
|
return cpy;
|
|
}
|
|
|
|
PyObject* GeometryPy::clone(PyObject *args)
|
|
{
|
|
if (!PyArg_ParseTuple(args, ""))
|
|
return NULL;
|
|
|
|
Part::Geometry* geom = this->getGeometryPtr();
|
|
PyTypeObject* type = this->GetType();
|
|
PyObject* cpy = 0;
|
|
// let the type object decide
|
|
if (type->tp_new)
|
|
cpy = type->tp_new(type, this, 0);
|
|
if (!cpy) {
|
|
PyErr_SetString(PyExc_TypeError, "failed to create clone of geometry");
|
|
return 0;
|
|
}
|
|
|
|
Part::GeometryPy* geompy = static_cast<Part::GeometryPy*>(cpy);
|
|
// the PyMake function must have created the corresponding instance of the 'Geometry' subclass
|
|
// so delete it now to avoid a memory leak
|
|
if (geompy->_pcTwinPointer) {
|
|
Part::Geometry* clone = static_cast<Part::Geometry*>(geompy->_pcTwinPointer);
|
|
delete clone;
|
|
}
|
|
geompy->_pcTwinPointer = geom->clone();
|
|
return cpy;
|
|
}
|
|
|
|
PyObject* GeometryPy::setExtension(PyObject *args)
|
|
{
|
|
PyObject* o;
|
|
if (PyArg_ParseTuple(args, "O!", &(GeometryExtensionPy::Type),&o)) {
|
|
Part::GeometryExtension * ext;
|
|
ext = static_cast<GeometryExtensionPy *>(o)->getGeometryExtensionPtr();
|
|
|
|
// make copy of Python managed memory and wrap it in smart pointer
|
|
auto cpy = ext->copy();
|
|
|
|
this->getGeometryPtr()->setExtension(std::move(cpy));
|
|
Py_Return;
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A geometry extension object was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::getExtensionOfType(PyObject *args)
|
|
{
|
|
char* o;
|
|
if (PyArg_ParseTuple(args, "s", &o)) {
|
|
|
|
Base::Type type = Base::Type::fromName(o);
|
|
|
|
if(type != Base::Type::badType()) {
|
|
try {
|
|
std::shared_ptr<GeometryExtension> ext(this->getGeometryPtr()->getExtension(type));
|
|
|
|
// we create a copy and transfer this copy's memory management responsibility to Python
|
|
PyObject* cpy = static_cast<GeometryExtensionPy *>(ext->getPyObject())->copy(Py::new_reference_to(Py::Tuple(size_t(0))));
|
|
|
|
return cpy;
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
catch(const std::bad_weak_ptr&) {
|
|
PyErr_SetString(PartExceptionOCCError, "Geometry extension does not exist anymore.");
|
|
return 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PartExceptionOCCError, "Exception type does not exist");
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A string with the name of the geometry extension type was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::getExtensionOfName(PyObject *args)
|
|
{
|
|
char* o;
|
|
if (PyArg_ParseTuple(args, "s", &o)) {
|
|
|
|
try {
|
|
std::shared_ptr<GeometryExtension> ext(this->getGeometryPtr()->getExtension(std::string(o)));
|
|
|
|
// we create a copy and transfer this copy's memory management responsibility to Python
|
|
PyObject* cpy = static_cast<GeometryExtensionPy *>(ext->getPyObject())->copy(Py::new_reference_to(Py::Tuple(size_t(0))));
|
|
|
|
return cpy;
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
catch(const std::bad_weak_ptr&) {
|
|
PyErr_SetString(PartExceptionOCCError, "Geometry extension does not exist anymore.");
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A string with the name of the geometry extension was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::hasExtensionOfType(PyObject *args)
|
|
{
|
|
char* o;
|
|
if (PyArg_ParseTuple(args, "s", &o)) {
|
|
|
|
Base::Type type = Base::Type::fromName(o);
|
|
|
|
if(type != Base::Type::badType()) {
|
|
try {
|
|
return Py::new_reference_to(Py::Boolean(this->getGeometryPtr()->hasExtension(type)));
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PartExceptionOCCError, "Exception type does not exist");
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A string with the type of the geometry extension was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::hasExtensionOfName(PyObject *args)
|
|
{
|
|
char* o;
|
|
if (PyArg_ParseTuple(args, "s", &o)) {
|
|
|
|
try {
|
|
return Py::new_reference_to(Py::Boolean(this->getGeometryPtr()->hasExtension(std::string(o))));
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A string with the type of the geometry extension was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::deleteExtensionOfType(PyObject *args)
|
|
{
|
|
char* o;
|
|
if (PyArg_ParseTuple(args, "s", &o)) {
|
|
|
|
Base::Type type = Base::Type::fromName(o);
|
|
|
|
if(type != Base::Type::badType()) {
|
|
try {
|
|
this->getGeometryPtr()->deleteExtension(type);
|
|
Py_Return;
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
PyErr_SetString(PartExceptionOCCError, "Type does not exist");
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A string with a type object was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::deleteExtensionOfName(PyObject *args)
|
|
{
|
|
char* o;
|
|
if (PyArg_ParseTuple(args, "s", &o)) {
|
|
|
|
try {
|
|
this->getGeometryPtr()->deleteExtension(std::string(o));
|
|
Py_Return;
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
PyErr_SetString(PartExceptionOCCError, "A string with the name of the extension was expected");
|
|
return 0;
|
|
}
|
|
|
|
PyObject* GeometryPy::getExtensions(PyObject *args)
|
|
{
|
|
if (!PyArg_ParseTuple(args, "")){
|
|
PyErr_SetString(PartExceptionOCCError, "No arguments were expected");
|
|
return NULL;
|
|
}
|
|
|
|
try {
|
|
const std::vector<std::weak_ptr<GeometryExtension>> ext = this->getGeometryPtr()->getExtensions();
|
|
|
|
PyObject* list = PyList_New(ext.size());
|
|
|
|
Py::Tuple tuple(ext.size());
|
|
|
|
for (std::size_t i=0; i<ext.size(); ++i) {
|
|
|
|
std::shared_ptr<GeometryExtension> p = ext[i].lock();
|
|
|
|
if(p) {
|
|
// we create a python copy and add it to the list
|
|
Py::Tuple args(size_t(0));
|
|
PyObject* cpy = static_cast<GeometryExtensionPy *>(p->getPyObject())->copy(Py::new_reference_to(Py::Tuple(size_t(0))));
|
|
|
|
PyList_SetItem( list, i, cpy);
|
|
}
|
|
}
|
|
|
|
return list;
|
|
}
|
|
catch(const Base::ValueError& e) {
|
|
PyErr_SetString(PartExceptionOCCError, e.what());
|
|
return 0;
|
|
}
|
|
|
|
}
|
|
|
|
Py::Boolean GeometryPy::getConstruction(void) const
|
|
{
|
|
return Py::Boolean(getGeometryPtr()->Construction);
|
|
}
|
|
|
|
void GeometryPy::setConstruction(Py::Boolean arg)
|
|
{
|
|
if (getGeometryPtr()->getTypeId() != Part::GeomPoint::getClassTypeId())
|
|
getGeometryPtr()->Construction = arg;
|
|
}
|
|
|
|
Py::String GeometryPy::getTag(void) const
|
|
{
|
|
std::string tmp = boost::uuids::to_string(getGeometryPtr()->getTag());
|
|
return Py::String(tmp);
|
|
}
|
|
|
|
PyObject *GeometryPy::getCustomAttributes(const char* /*attr*/) const
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int GeometryPy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/)
|
|
{
|
|
return 0;
|
|
}
|