mirror of
https://github.com/NGSolve/netgen.git
synced 2024-11-12 00:59:16 +05:00
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
from netgen.libngpy._geom2d import *
|
|
from netgen.libngpy._meshing import *
|
|
|
|
tmp_generate_mesh = SplineGeometry.GenerateMesh
|
|
|
|
def geom2d_meshing_func (geom, **args):
|
|
if "mp" in args:
|
|
return tmp_generate_mesh (geom, args["mp"])
|
|
else:
|
|
return tmp_generate_mesh (geom, MeshingParameters (**args))
|
|
|
|
|
|
SplineGeometry.GenerateMesh = geom2d_meshing_func
|
|
|
|
|
|
unit_square = SplineGeometry()
|
|
pnts = [ (0,0), (1,0), (1,1), (0,1) ]
|
|
lines = [ (0,1,1,"bottom"), (1,2,2,"right"), (2,3,3,"top"), (3,0,4,"left") ]
|
|
pnums = [unit_square.AppendPoint(*p) for p in pnts]
|
|
for l1,l2,bc,bcname in lines:
|
|
unit_square.Append( ["line", pnums[l1], pnums[l2]], bc=bcname)
|
|
|
|
|
|
def MakeRectangle (geo, p1, p2, bc=None, bcs=None, **args):
|
|
p1x, p1y = p1
|
|
p2x, p2y = p2
|
|
p1x,p2x = min(p1x,p2x), max(p1x, p2x)
|
|
p1y,p2y = min(p1y,p2y), max(p1y, p2y)
|
|
if not bcs: bcs=4*[bc]
|
|
pts = [geo.AppendPoint(*p) for p in [(p1x,p1y), (p2x, p1y), (p2x, p2y), (p1x, p2y)]]
|
|
for p1,p2,bc in [(0,1,bcs[0]), (1, 2, bcs[1]), (2, 3, bcs[2]), (3, 0, bcs[3])]:
|
|
geo.Append( ["line", pts[p1], pts[p2]], bc=bc, **args)
|
|
|
|
def MakeCircle (geo, c, r, **args):
|
|
cx,cy = c
|
|
pts = [geo.AppendPoint(*p) for p in [(cx,cy-r), (cx+r,cy-r), (cx+r,cy), (cx+r,cy+r), \
|
|
(cx,cy+r), (cx-r,cy+r), (cx-r,cy), (cx-r,cy-r)]]
|
|
for p1,p2,p3 in [(0,1,2), (2,3,4), (4, 5, 6), (6, 7, 0)]:
|
|
geo.Append( ["spline3", pts[p1], pts[p2], pts[p3]], **args)
|
|
|
|
|
|
SplineGeometry.AddCircle = lambda geo, c, r, **args : MakeCircle(geo, c, r, **args)
|
|
SplineGeometry.AddRectangle = lambda geo, p1, p2, **args : MakeRectangle(geo, p1, p2, **args)
|
|
SplineGeometry.AddSegment = lambda *args, **kwargs : SplineGeometry.Append(*args, **kwargs)
|
|
SplineGeometry.AddPoint = lambda *args, **kwargs : SplineGeometry.AppendPoint(*args, **kwargs)
|
|
|
|
|
|
__all__ = ['SplineGeometry', 'unit_square']
|
|
|
|
|
|
|
|
|