Surface: Various fixes

This commit is contained in:
tomate44
2022-08-18 14:58:05 +02:00
committed by wwmayer
parent a783d5bbfc
commit 7e69d7e38b
11 changed files with 1047 additions and 0 deletions

View File

@@ -0,0 +1,194 @@
///***************************************************************************
// * Copyright (c) 2022 Matteo Grellier <matteogrellier@gmail.com> *
// * *
// * 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 <BRepAdaptor_Curve.hxx>
#include <TColStd_Array1OfReal.hxx>
#include <TColgp_Array1OfPnt.hxx>
#include <TopoDS.hxx>
#endif
#include <Base/GeometryPyCXX.h>
#include <Base/VectorPy.h>
#include <Mod/Part/App/TopoShapePy.h>
#include "Blending/BlendPoint.h"
#include "Blending/BlendPointPy.h"
#include "Blending/BlendPointPy.cpp"
using namespace Surface;
std::string BlendPointPy::representation(void) const
{
Base::Vector3d bp = getBlendPointPtr()->vectors[0];
std::stringstream str;
str << "G" << getBlendPointPtr()->getContinuity()
<< " BlendPoint at (" << bp.x << ", " << bp.y << ", " << bp.z << "), ";
return str.str();
}
PyObject *BlendPointPy::PyMake(struct _typeobject *, PyObject *, PyObject *)// Python wrapper
{
// create a new instance of BlendPointPy
return new BlendPointPy(new BlendPoint);
}
int BlendPointPy::PyInit(PyObject* args, PyObject*)
{
PyObject *plist;
std::vector<Base::Vector3d> vecs;
if (PyArg_ParseTuple(args, "O", &plist)) {
Py::Sequence list(plist);
if (list.size() == 0) {
vecs.emplace_back(Base::Vector3d(0, 0, 0));
}
else {
for (Py::Sequence::iterator it = list.begin(); it != list.end(); ++it) {
Py::Vector v(*it);
Base::Vector3d vec = v.toVector();
vecs.emplace_back(vec);
}
}
this->getBlendPointPtr()->vectors = vecs;
return 0;
}
PyErr_Clear();
if (PyArg_ParseTuple(args, "")) {
vecs.emplace_back(Base::Vector3d(0, 0, 0));
this->getBlendPointPtr()->vectors = vecs;
return 0;
}
double param;
int cont;
PyObject *pcObj;
PyErr_Clear();
// Create a curve with an edge, parameter and continiuity.
if (PyArg_ParseTuple(args, "O!di", &(Part::TopoShapePy::Type), &pcObj, &param, &cont)) {
try {
gp_Pnt Pt;
TopoDS_Shape shape = static_cast<Part::TopoShapePy *>(pcObj)->getTopoShapePtr()->getShape();
const TopoDS_Edge &e = TopoDS::Edge(shape);
BRepAdaptor_Curve adapt(e);
adapt.D0(param, Pt);
Base::Vector3d bv(Pt.X(), Pt.Y(), Pt.Z());
vecs.emplace_back(bv);
for (int i = 1; i <= cont; i++) {
gp_Vec v1 = adapt.DN(param, i);
Base::Vector3d bbv1(v1.X(), v1.Y(), v1.Z());
vecs.emplace_back(bbv1);
}
this->getBlendPointPtr()->vectors = vecs;
return 0;
}
catch (const std::exception &e) {
std::cerr << e.what() << '\n';
return -1;
}
}
return -1;
}
PyObject *BlendPointPy::setSize(PyObject *args)
{
double size = 1.0;
if (!PyArg_ParseTuple(args, "d", &size))
return nullptr;
try {
getBlendPointPtr()->setSize(size);
Py_Return;
}
catch (Standard_Failure &e) {
PyErr_SetString(PyExc_Exception, "Failed to set size");
return nullptr;
}
}
PyObject *BlendPointPy::getSize(PyObject *args)
{
if (!PyArg_ParseTuple(args, ""))
return nullptr;
int nb = getBlendPointPtr()->nbVectors();
if (nb >= 2) {
double bpTangentLength = getBlendPointPtr()->vectors[1].Length();
return Py_BuildValue("d", bpTangentLength);
}
return nullptr;
}
Py::List BlendPointPy::getVectors() const
{
try {
BlendPoint *bp = getBlendPointPtr();
std::vector<Base::Vector3d> p = bp->vectors;
Py::List vecs;
for (size_t i = 0; i < p.size(); i++) {
Base::VectorPy *vec = new Base::VectorPy(Base::Vector3d(
p[i].x, p[i].y, p[i].z));
vecs.append(Py::asObject(vec));
}
return vecs;
}
catch (Standard_Failure &e) {
PyErr_SetString(PyExc_RuntimeError,
e.GetMessageString());
return Py::List();
}
}
PyObject *BlendPointPy::setvectors(PyObject *args)
{
BlendPoint *bp = getBlendPointPtr();
PyObject *plist;
if (!PyArg_ParseTuple(args, "O", &plist)) {
PyErr_SetString(PyExc_TypeError, "List of vectors required.");
return nullptr;
}
try {
Py::Sequence list(plist);
std::vector<Base::Vector3d> vecs;
for (Py::Sequence::iterator it = list.begin(); it != list.end(); ++it) {
Py::Vector v(*it);
Base::Vector3d pole = v.toVector();
vecs.emplace_back(pole);
}
bp->vectors = vecs;
Py_Return;
}
catch (Standard_Failure &e) {
PyErr_SetString(PyExc_RuntimeError, e.GetMessageString());
}
return nullptr;
}
PyObject *BlendPointPy::getCustomAttributes(const char * /*attr*/) const
{
return nullptr;
}
int BlendPointPy::setCustomAttributes(const char * /*attr*/, PyObject * /*obj*/)
{
return 0;
}