publicImage/gprMax
1
0
派生 0
镜像自地址 https://gitee.com/sunhf/gprMax.git 已同步 2025-08-07 04:56:51 +08:00
代码 工单 软件包 项目 版本发布 百科 动态

Simplified some logic in init of PMLs.

浏览代码
这个提交包含在：
Craig Warren
2017-01-26 15:16:30 +00:00
父节点 e042b14178
当前提交 46d239e313
共有 1 个文件被更改，包括 12 次插入 和 9 次删除
下载 Patch 文件 下载 Diff 文件 展开所有文件 折叠所有文件

21
gprMax/pml.py
查看文件

@@ -157,12 +157,6 @@ class PML(object):
""" """
self.direction = direction self.direction = direction
if self.direction[0] == 'x':
self.d = G.dx
elif self.direction[0] == 'y':
self.d = G.dy
elif self.direction[0] == 'z':
self.d = G.dz
self.xs = xs self.xs = xs
self.xf = xf self.xf = xf
self.ys = ys self.ys = ys
@@ -172,28 +166,37 @@ class PML(object):
self.nx = xf - xs self.nx = xf - xs
self.ny = yf - ys self.ny = yf - ys
self.nz = zf - zs self.nz = zf - zs
if self.direction[0] == 'x':
self.d = G.dx
self.thickness = self.nx
elif self.direction[0] == 'y':
self.d = G.dy
self.thickness = self.ny
elif self.direction[0] == 'z':
self.d = G.dz
self.thickness = self.nz
self.CFS = G.cfs self.CFS = G.cfs
if not self.CFS: if not self.CFS:
self.CFS = [CFS()] self.CFS = [CFS()]
self.initialise_field_arrays() self.initialise_field_arrays()
def initialise_field_arrays(self): def initialise_field_arrays(self):
"""Initialise arrays to store fields in PML.""" """Initialise arrays to store fields in PML."""
if self.direction[0] == 'x': if self.direction[0] == 'x':
self.thickness = self.nx
self.EPhi1 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz + 1), dtype=floattype) self.EPhi1 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz + 1), dtype=floattype)
self.EPhi2 = np.zeros((len(self.CFS), self.nx + 1, self.ny + 1, self.nz), dtype=floattype) self.EPhi2 = np.zeros((len(self.CFS), self.nx + 1, self.ny + 1, self.nz), dtype=floattype)
self.HPhi1 = np.zeros((len(self.CFS), self.nx, self.ny + 1, self.nz), dtype=floattype) self.HPhi1 = np.zeros((len(self.CFS), self.nx, self.ny + 1, self.nz), dtype=floattype)
self.HPhi2 = np.zeros((len(self.CFS), self.nx, self.ny, self.nz + 1), dtype=floattype) self.HPhi2 = np.zeros((len(self.CFS), self.nx, self.ny, self.nz + 1), dtype=floattype)
elif self.direction[0] == 'y': elif self.direction[0] == 'y':
self.thickness = self.ny
self.EPhi1 = np.zeros((len(self.CFS), self.nx, self.ny + 1, self.nz + 1), dtype=floattype) self.EPhi1 = np.zeros((len(self.CFS), self.nx, self.ny + 1, self.nz + 1), dtype=floattype)
self.EPhi2 = np.zeros((len(self.CFS), self.nx + 1, self.ny + 1, self.nz), dtype=floattype) self.EPhi2 = np.zeros((len(self.CFS), self.nx + 1, self.ny + 1, self.nz), dtype=floattype)
self.HPhi1 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz), dtype=floattype) self.HPhi1 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz), dtype=floattype)
self.HPhi2 = np.zeros((len(self.CFS), self.nx, self.ny, self.nz + 1), dtype=floattype) self.HPhi2 = np.zeros((len(self.CFS), self.nx, self.ny, self.nz + 1), dtype=floattype)
elif self.direction[0] == 'z': elif self.direction[0] == 'z':
self.thickness = self.nz
self.EPhi1 = np.zeros((len(self.CFS), self.nx, self.ny + 1, self.nz + 1), dtype=floattype) self.EPhi1 = np.zeros((len(self.CFS), self.nx, self.ny + 1, self.nz + 1), dtype=floattype)
self.EPhi2 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz + 1), dtype=floattype) self.EPhi2 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz + 1), dtype=floattype)
self.HPhi1 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz), dtype=floattype) self.HPhi1 = np.zeros((len(self.CFS), self.nx + 1, self.ny, self.nz), dtype=floattype)