146 lines
4.8 KiB
C++
146 lines
4.8 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 <Geom_Parabola.hxx>
|
|
#endif
|
|
|
|
#include <Base/VectorPy.h>
|
|
#include <Base/GeometryPyCXX.h>
|
|
|
|
#include "OCCError.h"
|
|
#include "Geometry.h"
|
|
#include <Mod/Part/App/ParabolaPy.h>
|
|
#include <Mod/Part/App/ParabolaPy.cpp>
|
|
|
|
using namespace Part;
|
|
|
|
// returns a string which represents the object e.g. when printed in python
|
|
std::string ParabolaPy::representation(void) const
|
|
{
|
|
return "<Parabola object>";
|
|
}
|
|
|
|
PyObject *ParabolaPy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper
|
|
{
|
|
// create a new instance of ParabolaPy and the Twin object
|
|
return new ParabolaPy(new GeomParabola);
|
|
}
|
|
|
|
// constructor method
|
|
int ParabolaPy::PyInit(PyObject* args, PyObject* /*kwd*/)
|
|
{
|
|
if (PyArg_ParseTuple(args, "")) {
|
|
Handle_Geom_Parabola c = Handle_Geom_Parabola::DownCast
|
|
(getGeometryPtr()->handle());
|
|
c->SetFocal(1.0);
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
PyObject* ParabolaPy::compute(PyObject *args)
|
|
{
|
|
PyObject *p1, *p2, *p3;
|
|
if (!PyArg_ParseTuple(args, "O!O!O!",
|
|
&Base::VectorPy::Type,&p1,
|
|
&Base::VectorPy::Type,&p2,
|
|
&Base::VectorPy::Type,&p3))
|
|
return 0;
|
|
Base::Vector3d v1 = Py::Vector(p1,false).toVector();
|
|
Base::Vector3d v2 = Py::Vector(p2,false).toVector();
|
|
Base::Vector3d v3 = Py::Vector(p3,false).toVector();
|
|
Base::Vector3d c = (v1-v2) % (v3-v2);
|
|
double zValue = v1.z;
|
|
if (fabs(c.Length()) < 0.0001) {
|
|
PyErr_SetString(PartExceptionOCCError, "Points are collinear");
|
|
return 0;
|
|
}
|
|
|
|
Base::Matrix4D m;
|
|
Base::Vector3d v;
|
|
m[0][0] = v1.y * v1.y;
|
|
m[0][1] = v1.y;
|
|
m[0][2] = 1;
|
|
m[1][0] = v2.y * v2.y;
|
|
m[1][1] = v2.y;
|
|
m[1][2] = 1;
|
|
m[2][0] = v3.y * v3.y;
|
|
m[2][1] = v3.y;
|
|
m[2][2] = 1.0;
|
|
v.x = v1.x;
|
|
v.y = v2.x;
|
|
v.z = v3.x;
|
|
m.inverseGauss();
|
|
v = m * v;
|
|
double a22 = v.x;
|
|
double a10 = -0.5;
|
|
double a20 = v.y/2.0;
|
|
double a00 = v.z;
|
|
Handle_Geom_Parabola curve = Handle_Geom_Parabola::DownCast(getGeometryPtr()->handle());
|
|
curve->SetFocal(0.5*fabs(a10/a22));
|
|
curve->SetLocation(gp_Pnt((a20*a20-a22*a00)/(2*a22*a10), -a20/a22, zValue));
|
|
|
|
Py_Return;
|
|
}
|
|
|
|
Py::Float ParabolaPy::getFocal(void) const
|
|
{
|
|
Handle_Geom_Parabola curve = Handle_Geom_Parabola::DownCast(getGeometryPtr()->handle());
|
|
return Py::Float(curve->Focal());
|
|
}
|
|
|
|
void ParabolaPy::setFocal(Py::Float arg)
|
|
{
|
|
Handle_Geom_Parabola curve = Handle_Geom_Parabola::DownCast(getGeometryPtr()->handle());
|
|
curve->SetFocal((double)arg);
|
|
}
|
|
|
|
Py::Object ParabolaPy::getFocus(void) const
|
|
{
|
|
Handle_Geom_Parabola c = Handle_Geom_Parabola::DownCast
|
|
(getGeometryPtr()->handle());
|
|
gp_Pnt loc = c->Focus();
|
|
return Py::Vector(Base::Vector3d(loc.X(), loc.Y(), loc.Z()));
|
|
}
|
|
|
|
Py::Float ParabolaPy::getParameter(void) const
|
|
{
|
|
Handle_Geom_Parabola curve = Handle_Geom_Parabola::DownCast(getGeometryPtr()->handle());
|
|
return Py::Float(curve->Parameter());
|
|
}
|
|
|
|
PyObject *ParabolaPy::getCustomAttributes(const char* /*attr*/) const
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int ParabolaPy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
|