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

Tidied up file opening method.

浏览代码
这个提交包含在：
Craig Warren
2022-01-20 09:00:48 +00:00
父节点 b77ba84c5b
当前提交 5d49f5311f
共有 1 个文件被更改，包括 48 次插入 和 48 次删除
下载 Patch 文件 下载 Diff 文件 展开所有文件 折叠所有文件

96
gprMax/geometry_outputs.py
查看文件

@@ -410,55 +410,55 @@ class GeometryObjects:
""" """
# Write the geometry objects to a HDF5 file # Write the geometry objects to a HDF5 file
fdata = h5py.File(self.filename_hdf5, 'w') with h5py.File(self.filename_hdf5, 'w') as fdata:
fdata.attrs['gprMax'] = __version__ fdata.attrs['gprMax'] = __version__
fdata.attrs['Title'] = G.title fdata.attrs['Title'] = G.title
fdata.attrs['dx_dy_dz'] = (G.dx, G.dy, G.dz) fdata.attrs['dx_dy_dz'] = (G.dx, G.dy, G.dz)
# Get minimum and maximum integers of materials in geometry objects volume # Get minimum and maximum integers of materials in geometry objects volume
minmat = np.amin(G.ID[:, self.xs:self.xf + 1, minmat = np.amin(G.ID[:, self.xs:self.xf + 1,
self.ys:self.yf + 1, self.zs:self.zf + 1]) self.ys:self.yf + 1, self.zs:self.zf + 1])
maxmat = np.amax(G.ID[:, self.xs:self.xf + 1, maxmat = np.amax(G.ID[:, self.xs:self.xf + 1,
self.ys:self.yf + 1, self.zs:self.zf + 1]) self.ys:self.yf + 1, self.zs:self.zf + 1])
fdata['/data'] = G.solid[self.xs:self.xf + 1, self.ys:self.yf + fdata['/data'] = G.solid[self.xs:self.xf + 1, self.ys:self.yf +
1, self.zs:self.zf + 1].astype('int16') - minmat 1, self.zs:self.zf + 1].astype('int16') - minmat
pbar.update(self.solidsize) pbar.update(self.solidsize)
fdata['/rigidE'] = G.rigidE[:, self.xs:self.xf + fdata['/rigidE'] = G.rigidE[:, self.xs:self.xf +
1, self.ys:self.yf + 1, self.zs:self.zf + 1] 1, self.ys:self.yf + 1, self.zs:self.zf + 1]
fdata['/rigidH'] = G.rigidH[:, self.xs:self.xf + fdata['/rigidH'] = G.rigidH[:, self.xs:self.xf +
1, self.ys:self.yf + 1, self.zs:self.zf + 1] 1, self.ys:self.yf + 1, self.zs:self.zf + 1]
pbar.update(self.rigidsize) pbar.update(self.rigidsize)
fdata['/ID'] = G.ID[:, self.xs:self.xf + 1, fdata['/ID'] = G.ID[:, self.xs:self.xf + 1,
self.ys:self.yf + 1, self.zs:self.zf + 1] - minmat self.ys:self.yf + 1, self.zs:self.zf + 1] - minmat
pbar.update(self.IDsize) pbar.update(self.IDsize)
# Write materials list to a text file # Write materials list to a text file
# This includes all materials in range whether used in volume or not # This includes all materials in range whether used in volume or not
fmaterials = open(self.filename_materials, 'w') with open(self.filename_materials, 'w') as fmaterials:
for numID in range(minmat, maxmat + 1): for numID in range(minmat, maxmat + 1):
for material in G.materials: for material in G.materials:
if material.numID == numID: if material.numID == numID:
fmaterials.write(f'#material: {material.er:g} {material.se:g} ' fmaterials.write(f'#material: {material.er:g} {material.se:g} '
f'{material.mr:g} {material.sm:g} {material.ID}\n') f'{material.mr:g} {material.sm:g} {material.ID}\n')
if hasattr(material, 'poles'): if hasattr(material, 'poles'):
if 'debye' in material.type: if 'debye' in material.type:
dispersionstr = ('#add_dispersion_debye: ' dispersionstr = ('#add_dispersion_debye: '
f'{material.poles:g} ') f'{material.poles:g} ')
for pole in range(material.poles): for pole in range(material.poles):
dispersionstr += (f'{material.deltaer[pole]:g} ' dispersionstr += (f'{material.deltaer[pole]:g} '
f'{material.tau[pole]:g} ') f'{material.tau[pole]:g} ')
elif 'lorenz' in material.type: elif 'lorenz' in material.type:
dispersionstr = (f'#add_dispersion_lorenz: ' dispersionstr = (f'#add_dispersion_lorenz: '
f'{material.poles:g} ') f'{material.poles:g} ')
for pole in range(material.poles): for pole in range(material.poles):
dispersionstr += (f'{material.deltaer[pole]:g} ' dispersionstr += (f'{material.deltaer[pole]:g} '
f'{material.tau[pole]:g} ' f'{material.tau[pole]:g} '
f'{material.alpha[pole]:g} ') f'{material.alpha[pole]:g} ')
elif 'drude' in material.type: elif 'drude' in material.type:
dispersionstr = (f'#add_dispersion_drude: ' dispersionstr = (f'#add_dispersion_drude: '
f'{material.poles:g} ') f'{material.poles:g} ')
for pole in range(material.poles): for pole in range(material.poles):
dispersionstr += (f'{material.tau[pole]:g} ' dispersionstr += (f'{material.tau[pole]:g} '
f'{material.alpha[pole]:g} ') f'{material.alpha[pole]:g} ')
dispersionstr += material.ID dispersionstr += material.ID
fmaterials.write(dispersionstr + '\n') fmaterials.write(dispersionstr + '\n')