caiyu/openEMS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

python: implement cylindrical grid support

Browse Source 
Signed-off-by: Thorsten Liebig <Thorsten.Liebig@gmx.de>
This commit is contained in:
Thorsten Liebig 2016-11-13 22:36:57 +01:00
parent 1573f0952e
commit 0f37d1d3a2
2 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/openEMS/openEMS.pxd
View File

@ -32,7 +32,7 @@ cdef extern from "openEMS/openems.h":
void SetCellConstantMaterial(bool val) void SetCellConstantMaterial(bool val)
void SetCylinderCoords(bool val) void SetCylinderCoords(bool val)
#void SetupCylinderMultiGrid(std::vector<double> val) void SetupCylinderMultiGrid(string val)
void SetTimeStepMethod(int val) void SetTimeStepMethod(int val)
void SetTimeStep(double val) void SetTimeStep(double val)

25
python/openEMS/openEMS.pyx
View File

@ -53,7 +53,7 @@ cdef class openEMS:
:param MaxTime: max. real time in seconds to simulate :param MaxTime: max. real time in seconds to simulate
:param OverSampling: nyquist oversampling of time domain dumps :param OverSampling: nyquist oversampling of time domain dumps
:param CoordSystem: choose coordinate system (0 Cartesian, 1 Cylindrical) :param CoordSystem: choose coordinate system (0 Cartesian, 1 Cylindrical)
:param MultiGrid: define a cylindrical sub-grid radius ( not implemented yet ) :param MultiGrid: define a cylindrical sub-grid radius
:param TimeStep: force to use a given timestep (dangerous!) :param TimeStep: force to use a given timestep (dangerous!)
:param TimeStepFactor: reduce the timestep by a given factor (>0 to <=1) :param TimeStepFactor: reduce the timestep by a given factor (>0 to <=1)
:param TimeStepMethod: 1 or 3 chose timestep method (1=CFL, 3=Rennigs (default)) :param TimeStepMethod: 1 or 3 chose timestep method (1=CFL, 3=Rennigs (default))
@ -99,6 +99,9 @@ cdef class openEMS:
if 'CellConstantMaterial' in kw: if 'CellConstantMaterial' in kw:
self.SetCellConstantMaterial(kw['CellConstantMaterial']) self.SetCellConstantMaterial(kw['CellConstantMaterial'])
del kw['CellConstantMaterial'] del kw['CellConstantMaterial']
if 'MultiGrid' in kw:
self.SetMultiGrid(kw['MultiGrid'])
del kw['MultiGrid']
assert len(kw)==0, 'Unknown keyword arguments: "{}"'.format(kw) assert len(kw)==0, 'Unknown keyword arguments: "{}"'.format(kw)
@ -148,10 +151,30 @@ cdef class openEMS:
elif val==1: elif val==1:
self.thisptr.SetCylinderCoords(True) self.thisptr.SetCylinderCoords(True)
def SetMultiGrid(self, radii):
""" SetMultiGrid(radii)
Define radii at which a cylindrical multi grid should be defined.
:param radii: array like, multigrid radii
See Also
--------
openEMS.SetCylinderCoords
"""
assert len(radii)>0, 'SetMultiGrid: invalid multi grid definition'
grid_str = ','.join(['{}'.format(x) for x in radii])
self.thisptr.SetupCylinderMultiGrid(grid_str.encode('UTF-8'))
def SetCylinderCoords(self): def SetCylinderCoords(self):
""" SetCylinderCoords() """ SetCylinderCoords()
Enable use of cylindircal coordinates. Enable use of cylindircal coordinates.
See Also
--------
openEMS.SetMultiGrid
""" """
self.thisptr.SetCylinderCoords(True) self.thisptr.SetCylinderCoords(True)