你已经派生过 gprMax
镜像自地址
https://gitee.com/sunhf/gprMax.git
已同步 2025-08-07 04:56:51 +08:00
transfer existing subgridding modules
这个提交包含在:
@@ -0,0 +1,118 @@
|
||||
"""Class for surface water command."""
|
||||
from .cmds_geometry import UserObjectGeometry
|
||||
from ..exceptions import CmdInputError
|
||||
from ..utilities import round_value
|
||||
from ..materials import Material
|
||||
|
||||
from tqdm import tqdm
|
||||
|
||||
|
||||
class AddSurfaceWater(UserObjectGeometry):
|
||||
"""User class for edge command."""
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
"""Constructor."""
|
||||
super().__init__(**kwargs)
|
||||
self.order = 11
|
||||
self.hash = '#add_surface_water'
|
||||
|
||||
def create(self, grid, uip):
|
||||
""""Create surface water on fractal box."""
|
||||
try:
|
||||
p1 = self.kwargs['p1']
|
||||
p2 = self.kwargs['p2']
|
||||
fractal_box_id = self.kwargs['fractal_box_id']
|
||||
depth = self.kwargs['depth']
|
||||
except KeyError:
|
||||
raise CmdInputError(self.__str__() + ' requires exactly eight parameters')
|
||||
|
||||
# grab the correct fractal volume
|
||||
volumes = [volume for volume in grid.fractalvolumes if volume.ID == fractal_box_id]
|
||||
if volumes:
|
||||
volume = volumes[0]
|
||||
else:
|
||||
raise CmdInputError(self.__str__() + ' Cant find FractalBox {}'.format(fractal_box_id))
|
||||
|
||||
p1, p2 = uip.check_box_points(p1, p2, self.__str__())
|
||||
xs, ys, zs = p1
|
||||
xf, yf, zf = p2
|
||||
|
||||
if depth <= 0:
|
||||
raise CmdInputError(self.__str__() + ' requires a positive value for the depth of water')
|
||||
|
||||
# Check for valid orientations
|
||||
if xs == xf:
|
||||
if ys == yf or zs == zf:
|
||||
raise CmdInputError(self.__str__() + ' dimensions are not specified correctly')
|
||||
if xs != volume.xs and xs != volume.xf:
|
||||
raise CmdInputError(self.__str__() + ' can only be used on the external surfaces of a fractal box')
|
||||
# xminus surface
|
||||
if xs == volume.xs:
|
||||
requestedsurface = 'xminus'
|
||||
# xplus surface
|
||||
elif xf == volume.xf:
|
||||
requestedsurface = 'xplus'
|
||||
filldepthcells = round_value(depth / grid.dx)
|
||||
filldepth = filldepthcells * grid.dx
|
||||
|
||||
elif ys == yf:
|
||||
if xs == xf or zs == zf:
|
||||
raise CmdInputError(self.__str__() + ' dimensions are not specified correctly')
|
||||
if ys != volume.ys and ys != volume.yf:
|
||||
raise CmdInputError(self.__str__() + ' can only be used on the external surfaces of a fractal box')
|
||||
# yminus surface
|
||||
if ys == volume.ys:
|
||||
requestedsurface = 'yminus'
|
||||
# yplus surface
|
||||
elif yf == volume.yf:
|
||||
requestedsurface = 'yplus'
|
||||
filldepthcells = round_value(depth / grid.dy)
|
||||
filldepth = filldepthcells * grid.dy
|
||||
|
||||
elif zs == zf:
|
||||
if xs == xf or ys == yf:
|
||||
raise CmdInputError(self.__str__() + ' dimensions are not specified correctly')
|
||||
if zs != volume.zs and zs != volume.zf:
|
||||
raise CmdInputError(self.__str__() + ' can only be used on the external surfaces of a fractal box')
|
||||
# zminus surface
|
||||
if zs == volume.zs:
|
||||
requestedsurface = 'zminus'
|
||||
# zplus surface
|
||||
elif zf == volume.zf:
|
||||
requestedsurface = 'zplus'
|
||||
filldepthcells = round_value(depth / grid.dz)
|
||||
filldepth = filldepthcells * grid.dz
|
||||
|
||||
else:
|
||||
raise CmdInputError(self.__str__() + ' dimensions are not specified correctly')
|
||||
|
||||
surface = next((x for x in volume.fractalsurfaces if x.surfaceID == requestedsurface), None)
|
||||
if not surface:
|
||||
raise CmdInputError(self.__str__() + ' specified surface {} does not have a rough surface applied'.format(requestedsurface))
|
||||
|
||||
surface.filldepth = filldepthcells
|
||||
|
||||
# Check that requested fill depth falls within range of surface roughness
|
||||
if surface.filldepth < surface.fractalrange[0] or surface.filldepth > surface.fractalrange[1]:
|
||||
raise CmdInputError(self.__str__() + ' requires a value for the depth of water that lies with the range of the requested surface roughness')
|
||||
|
||||
# Check to see if water has been already defined as a material
|
||||
if not any(x.ID == 'water' for x in grid.materials):
|
||||
m = Material(len(grid.materials), 'water')
|
||||
m.averagable = False
|
||||
m.type = 'builtin, debye'
|
||||
m.er = Material.watereri
|
||||
m.deltaer.append(Material.waterdeltaer)
|
||||
m.tau.append(Material.watertau)
|
||||
grid.materials.append(m)
|
||||
if Material.maxpoles == 0:
|
||||
Material.maxpoles = 1
|
||||
|
||||
# Check if time step for model is suitable for using water
|
||||
water = next((x for x in grid.materials if x.ID == 'water'))
|
||||
testwater = next((x for x in water.tau if x < grid.dt), None)
|
||||
if testwater:
|
||||
raise CmdInputError(self.__str__() + ' requires the time step for the model to be less than the relaxation time required to model water.')
|
||||
|
||||
if grid.messages:
|
||||
tqdm.write('Water on surface from {:g}m, {:g}m, {:g}m, to {:g}m, {:g}m, {:g}m with depth {:g}m, added to {}.'.format(xs * grid.dx, ys * grid.dy, zs * grid.dz, xf * grid.dx, yf * grid.dy, zf * grid.dz, filldepth, surface.operatingonID))
|
在新工单中引用
屏蔽一个用户