diff --git a/libsrc/geom2d/python_geom2d.cpp b/libsrc/geom2d/python_geom2d.cpp index f2b20127..fbfa22d9 100644 --- a/libsrc/geom2d/python_geom2d.cpp +++ b/libsrc/geom2d/python_geom2d.cpp @@ -138,6 +138,47 @@ DLL_HEADER void ExportGeom2d(py::module &m) self.AppendSegment(seg); }), py::arg("point_indices"), py::arg("leftdomain") = 1, py::arg("rightdomain") = py::int_(0)) + + .def("AddCurve", + [] (SplineGeometry2d & self, py::object func, + int leftdomain, int rightdomain, py::object bc, double maxh) + { + int n = 1000; + Array> points; + for (int i = 0; i <= n; i++) + { + double t = double(i)/n; + py::tuple xy = func(t); + double x = py::cast(xy[0]); + double y = py::cast(xy[1]); + points.Append (Point<2>(x,y)); + } + auto spline = new DiscretePointsSeg<2> (points); + SplineSegExt * spex = new SplineSegExt (*spline); + + spex -> leftdom = leftdomain; + spex -> rightdom = rightdomain; + spex->hmax = maxh; + spex->reffak = 1; + spex->copyfrom = -1; + + if (py::extract(bc).check()) + spex->bc = py::extract(bc)(); + else if (py::extract(bc).check()) + { + string bcname = py::extract(bc)(); + spex->bc = self.GetNSplines()+1; + self.SetBCName(spex->bc, bcname); + } + else + spex->bc = self.GetNSplines()+1; + + + self.AppendSegment (spex); + }, py::arg("func"), py::arg("leftdomain") = 1, py::arg("rightdomain") = py::int_(0), + py::arg("bc")=NGDummyArgument(), py::arg("maxh")=1e99, + "Curve is given as parametrization on the interval [0,1]") + .def("SetMaterial", &SplineGeometry2d::SetMaterial) .def("SetDomainMaxH", &SplineGeometry2d::SetDomainMaxh)