ForkMaster/netgen
1
0
Fork 0
mirror of https://github.com/NGSolve/netgen.git synced 2024-11-13 17:48:34 +05:00
Code Issues Packages Projects Releases Wiki Activity

py::init constructors

Browse Source
This commit is contained in:
Joachim Schöberl 2018-03-08 22:19:11 +01:00
parent cf0d3f6682
commit 5de403ffd8
1 changed files with 125 additions and 124 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
libsrc/meshing/python_mesh.cpp
View File

@ -204,39 +204,41 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
;
py::class_<Element>(m, "Element3D")
.def("__init__", [](Element *instance, int index, py::list vertices)
.def(py::init([](int index, py::list vertices)
{
Element * newel = nullptr;
if (py::len(vertices) == 4)
{
new (instance) Element(TET);
newel = new Element(TET);
for (int i = 0; i < 4; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
else if (py::len(vertices) == 5)
{
new (instance) Element(PYRAMID);
newel = new Element(PYRAMID);
for (int i = 0; i < 5; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
else if (py::len(vertices) == 6)
{
new (instance) Element(PRISM);
newel = new Element(PRISM);
for (int i = 0; i < 6; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
else if (py::len(vertices) == 8)
{
new (instance) Element(HEX);
newel = new Element(HEX);
for (int i = 0; i < 8; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
else
throw NgException ("cannot create element");
},
return newel;
}),
py::arg("index")=1,py::arg("vertices"),
"create volume element"
)
@ -261,35 +263,34 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
;
py::class_<Element2d>(m, "Element2D")
.def("__init__",
[](Element2d *instance, int index, py::list vertices)
.def(py::init ([](int index, py::list vertices)
{
Element2d * newel = nullptr;
if (py::len(vertices) == 3)
{
new (instance) Element2d(TRIG);
newel = new Element2d(TRIG);
for (int i = 0; i < 3; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
return;
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
if (py::len(vertices) == 4)
else if (py::len(vertices) == 4)
{
new (instance) Element2d(QUAD);
newel = new Element2d(QUAD);
for (int i = 0; i < 4; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
return;
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
if (py::len(vertices) == 6)
else if (py::len(vertices) == 6)
{
new (instance) Element2d(TRIG6);
newel = new Element2d(TRIG6);
for(int i = 0; i<6; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance->SetIndex(index);
return;
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel->SetIndex(index);
}
else
throw NgException("Inconsistent number of vertices in Element2D");
},
return newel;
}),
py::arg("index")=1,py::arg("vertices"),
"create surface element"
)
@ -313,21 +314,21 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
;
py::class_<Segment>(m, "Element1D")
.def("__init__",
[](Segment *instance, py::list vertices, py::list surfaces, int index)
.def(py::init([](py::list vertices, py::list surfaces, int index)
{
new (instance) Segment();
Segment * newel = new Segment();
for (int i = 0; i < 2; i++)
(*instance)[i] = py::extract<PointIndex>(vertices[i])();
instance -> si = index;
(*newel)[i] = py::extract<PointIndex>(vertices[i])();
newel -> si = index;
// needed for codim2 in 3d
instance -> edgenr = index;
newel -> edgenr = index;
if (len(surfaces))
{
instance->surfnr1 = py::extract<int>(surfaces[0])();
instance->surfnr2 = py::extract<int>(surfaces[1])();
newel->surfnr1 = py::extract<int>(surfaces[0])();
newel->surfnr2 = py::extract<int>(surfaces[1])();
}
},
return newel;
}),
py::arg("vertices"),
py::arg("surfaces")=py::list(),
py::arg("index")=1,
@ -370,13 +371,13 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
py::class_<Element0d>(m, "Element0D")
.def("__init__",
[](Element0d *instance, PointIndex vertex, int index)
.def(py::init([](PointIndex vertex, int index)
{
new (instance) Element0d;
Element0d * instance = new Element0d;
instance->pnum = vertex;
instance->index = index;
},
return instance;
}),
py::arg("vertex"),
py::arg("index")=1,
"create point element"
@ -397,15 +398,15 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
py::class_<FaceDescriptor>(m, "FaceDescriptor")
.def(py::init<const FaceDescriptor&>())
.def("__init__",
[](FaceDescriptor *instance, int surfnr, int domin, int domout, int bc)
.def(py::init([](int surfnr, int domin, int domout, int bc)
{
new (instance) FaceDescriptor();
FaceDescriptor * instance = new FaceDescriptor();
instance->SetSurfNr(surfnr);
instance->SetDomainIn(domin);
instance->SetDomainOut(domout);
instance->SetBCProperty(bc);
},
return instance;
}),
py::arg("surfnr")=1,
py::arg("domin")=1,
py::arg("domout")=py::int_(0),
@ -796,12 +797,11 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
typedef MeshingParameters MP;
py::class_<MP> (m, "MeshingParameters")
.def(py::init<>())
.def("__init__",
[](MP *instance, double maxh, bool quad_dominated, int optsteps2d, int optsteps3d,
.def(py::init([](double maxh, bool quad_dominated, int optsteps2d, int optsteps3d,
MESHING_STEP perfstepsend, int only3D_domain, const string & meshsizefilename,
double grading, double curvaturesafety, double segmentsperedge)
{
new (instance) MeshingParameters;
MP * instance = new MeshingParameters;
instance->maxh = maxh;
instance->quad = int(quad_dominated);
instance->optsteps2d = optsteps2d;
@ -813,7 +813,8 @@ DLL_HEADER void ExportNetgenMeshing(py::module &m)
instance->grading = grading;
instance->curvaturesafety = curvaturesafety;
instance->segmentsperedge = segmentsperedge;
},
return instance;
}),
py::arg("maxh")=1000,
py::arg("quad_dominated")=false,
py::arg("optsteps2d") = 3,