AcousticDev/AcousticSo
5
0
派生 0
代码 工单 合并请求 软件包 项目 发布 百科 活动

更改了bellhop.py文件中，nbeams的数目。

浏览代码 
更改了czcore.py文件中，会聚区计算时进行并行运算。
这个提交包含在：
Rose0428
2023-04-02 17:32:53 +08:00
父节点 35c10e68ed
当前提交 6845ee0834
共修改 2 个文件，包含 94 行新增和 79 行删除
下载 Patch 文件 下载 Diff 文件 展开所有文件 折叠所有文件
pyat/bellhop.py
+1 -1
查看文件
@@ -96,7 +96,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
bdy = Bndry(top, bottom) bdy = Bndry(top, bottom)
# Beam params # Beam params
run_type = 'C' # incoherently sum beams, see AT docs for more info run_type = 'C' # incoherently sum beams, see AT docs for more info
nbeams = 0 nbeams = 3000
alpha = np.linspace(-45, 45, 2) # min and max launch angle -20 degrees to 20 degrees alpha = np.linspace(-45, 45, 2) # min and max launch angle -20 degrees to 20 degrees
box = Box(Zmax, max(bathm.r)) # bound the region you let the beams go, depth in meters and range in km box = Box(Zmax, max(bathm.r)) # bound the region you let the beams go, depth in meters and range in km
deltas = 0 # length step of ray trace, 0 means automatically choose deltas = 0 # length step of ray trace, 0 means automatically choose
pyat/czcore.py
+93 -78
查看文件
@@ -1,9 +1,9 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from multiprocessing import Process
from os import system from os import system
import numpy as np import numpy as np
from pyat.AcousticPara import GetEnv, Coord from pyat.AcousticPara import GetEnv, Coord
from scipy.signal import savgol_filter, medfilt from scipy.signal import savgol_filter, medfilt
from pyat.DataBase import DataBase from pyat.DataBase import DataBase
from pyat.readwrite import write_env, write_bathy, read_shd, get_rays from pyat.readwrite import write_env, write_bathy, read_shd, get_rays
from pyat.env import Pos, Source, Dom, cInt, SSPraw, SSP, HS, BotBndry, TopBndry, Bndry, Box, Beam from pyat.env import Pos, Source, Dom, cInt, SSPraw, SSP, HS, BotBndry, TopBndry, Bndry, Box, Beam
@@ -12,18 +12,21 @@ from pyat.config import AtBinPath
file_cz_data = open('cz_data.txt', 'w') file_cz_data = open('cz_data.txt', 'w')
class CzRangeWidth: class CzRangeWidth:
def __init__(self, minRange, maxRange, width): def __init__(self, minRange, maxRange, width):
self.minRange = minRange self.minRange = minRange
self.maxRange = maxRange self.maxRange = maxRange
self.width = width self.width = width
class CzDetail: class CzDetail:
def __init__(self, descript, directCz, reflectCz): def __init__(self, descript, directCz, reflectCz):
self.descript = descript self.descript = descript
self.directCz = directCz self.directCz = directCz
self.reflectCz = reflectCz self.reflectCz = reflectCz
class CzData: class CzData:
def __init__(self, azi, ranges, depths, TL, bathym, Rays, CzDetail): def __init__(self, azi, ranges, depths, TL, bathym, Rays, CzDetail):
self.azi = azi self.azi = azi
@@ -34,23 +37,25 @@ class CzData:
self.Rays = Rays self.Rays = Rays
self.CzDetail = CzDetail self.CzDetail = CzDetail
def calZmax(ssp, sed): def calZmax(ssp, sed):
d = np.array([],dtype=int) d = np.array([], dtype=int)
for it in range(len(ssp)): for it in range(len(ssp)):
d=np.append(d,ssp[it].z[-1]) d = np.append(d, ssp[it].z[-1])
Zmax =max(d) Zmax = max(d)
ssp_idx = np.argmax(d) ssp_idx = np.argmax(d)
return Zmax, ssp_idx return Zmax, ssp_idx
def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base,azi):
# Source and receiving position # Source and receiving position
filename='cz' filename = 'cz'+str(azi)
Rmax = bathm.r[-1] # unit: km Rmax = bathm.r[-1] # unit: km
dr = (1500 / freq) / 1000 # wavelength / 10, unit: km dr = (1500 / freq) / 1000 # wavelength / 10, unit: km
ran = np.linspace(0, Rmax, round(Rmax / dr)) # Receiving range (unit: km) ran = np.linspace(0, Rmax, round(Rmax / dr)) # Receiving range (unit: km)
Zmax, ssp_idx = calZmax(ssp, sed) Zmax, ssp_idx = calZmax(ssp, sed)
"""RD 已修改 8.26""" """RD 已修改 8.26"""
RD = np.linspace(0, Zmax, round(Zmax/10) + 1) RD = np.linspace(0, Zmax, round(Zmax / 10) + 1)
pos = Pos(Source(sd), Dom(ran, RD)) pos = Pos(Source(sd), Dom(ran, RD))
# Range of phase velocity # Range of phase velocity
cInt.Low = 1400 cInt.Low = 1400
@@ -60,31 +65,30 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
NMedia = 1 NMedia = 1
ssp_raw = [] ssp_raw = []
depth = [0] depth = [0]
# Sound speed profile # Sound speed profile
Z = ssp[ssp_idx].z # Depth Z = ssp[ssp_idx].z # Depth
Cp = ssp[ssp_idx].c # Speed of P-wave Cp = ssp[ssp_idx].c # Speed of P-wave
Cs = np.zeros(np.shape(Z)) # Speed of S-wave Cs = np.zeros(np.shape(Z)) # Speed of S-wave
Rho = np.ones(np.shape(Z)) # Density of the media Rho = np.ones(np.shape(Z)) # Density of the media
Ap = np.zeros(np.shape(Z)) # Attenuation of P-wave Ap = np.zeros(np.shape(Z)) # Attenuation of P-wave
As = np.zeros(np.shape(Z)) # Attenuation of S-wave As = np.zeros(np.shape(Z)) # Attenuation of S-wave
ssp_raw.append(SSPraw(Z, Cp, Cs, Rho, Ap, As)) ssp_raw.append(SSPraw(Z, Cp, Cs, Rho, Ap, As))
depth.append(Z[-1]) depth.append(Z[-1])
# Sediment # Sediment
if sed != None: if sed != None:
Z = sed[0].z + ssp[0].z[-1] # Depth
Z = sed[0].z + ssp[0].z[-1] # Depth Cp = sed[0].cp # Speed of P-wave
Cp = sed[0].cp # Speed of P-wave Cs = sed[0].cs # Speed of S-wave
Cs = sed[0].cs # Speed of S-wave Rho = sed[0].rho # Density of the media
Rho = sed[0].rho # Density of the media Ap = sed[0].a_p # Attenuation of P-wave
Ap = sed[0].a_p # Attenuation of P-wave As = sed[0].a_s # Attenuation of S-wave
As = sed[0].a_s # Attenuation of S-wave
ssp_raw.append(SSPraw(Z, Cp, Cs, Rho, Ap, As)) ssp_raw.append(SSPraw(Z, Cp, Cs, Rho, Ap, As))
depth.append(Z[-1]) depth.append(Z[-1])
"""Opt_top 已修改 8.26""" """Opt_top 已修改 8.26"""
Opt_top = 'SVW' Opt_top = 'SVW'
N = np.zeros(NMedia, np.int8) N = np.zeros(NMedia, np.int8)
Sigma = np.zeros(NMedia+1) Sigma = np.zeros(NMedia + 1)
sspK = SSP(ssp_raw, depth, NMedia, Opt_top, N, Sigma) sspK = SSP(ssp_raw, depth, NMedia, Opt_top, N, Sigma)
# Bottom option # Bottom option
@@ -95,7 +99,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
bdy = Bndry(top, bottom) bdy = Bndry(top, bottom)
# Beam params # Beam params
run_type = 'I' # incoherently sum beams, see AT docs for more info run_type = 'I' # incoherently sum beams, see AT docs for more info
nbeams = 0 nbeams = 3000
alpha = np.linspace(-90, 90, 2) # min and max launch angle -20 degrees to 20 degrees alpha = np.linspace(-90, 90, 2) # min and max launch angle -20 degrees to 20 degrees
box = Box(Zmax, 100) # bound the region you let the beams go, depth in meters and range in km box = Box(Zmax, 100) # bound the region you let the beams go, depth in meters and range in km
deltas = 0 # length step of ray trace, 0 means automatically choose deltas = 0 # length step of ray trace, 0 means automatically choose
@@ -103,8 +107,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
"""删除ssp 已修改 8.26""" """删除ssp 已修改 8.26"""
# Write *.env file # Write *.env file
write_env(filename + '.env', 'BELLHOP', 'Pekeris profile', freq, sspK, bdy, pos, beam, cInt, Rmax) write_env(filename + '.env', 'BELLHOP', 'Pekeris profile', freq, sspK, bdy, pos, beam, cInt, Rmax)
write_bathy(filename,bathm) write_bathy(filename, bathm)
system(AtBinPath + "/bellhop " + filename) system(AtBinPath + "/bellhop " + filename)
# Read sound field # Read sound field
@@ -112,10 +115,11 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
TL = -10 * np.log10(np.square(np.abs(pressure))) # Transmission Loss TL = -10 * np.log10(np.square(np.abs(pressure))) # Transmission Loss
return Pos1, TL return Pos1, TL
def call_Bellhop_Rays(freq, sd, rd, bathm, ssp, sed, base):
def call_Bellhop_Rays(freq, sd, rd, bathm, ssp, sed, base,azi):
# Source and receiving position # Source and receiving position
filename = 'cz_ray' filename = 'cz_ray'+str(azi)
Rmax = bathm.r[-1] # unit: km Rmax = bathm.r[-1] # unit: km
dr = (1500 / freq) / 1000 # wavelength / 10, unit: km dr = (1500 / freq) / 1000 # wavelength / 10, unit: km
ran = np.linspace(0, Rmax, round(Rmax / dr)) # Receiving range (unit: km) ran = np.linspace(0, Rmax, round(Rmax / dr)) # Receiving range (unit: km)
@@ -179,78 +183,80 @@ def call_Bellhop_Rays(freq, sd, rd, bathm, ssp, sed, base):
# Read sound field # Read sound field
return get_rays(filename + ".ray") return get_rays(filename + ".ray")
def find_cvgcRays(rays_total): def find_cvgcRays(rays_total):
Rays = [] Rays = []
num_bnc_min = np.Inf num_bnc_min = np.Inf
for ray in rays_total[0]: for ray in rays_total[0]:
if ray.num_top_bnc + ray.num_bot_bnc < num_bnc_min and max(ray.xy[1,:]) > 100: if ray.num_top_bnc + ray.num_bot_bnc < num_bnc_min and max(ray.xy[1, :]) > 100:
num_bnc_min = ray.num_top_bnc + ray.num_bot_bnc num_bnc_min = ray.num_top_bnc + ray.num_bot_bnc
for ray in rays_total[0]: for ray in rays_total[0]:
if ray.num_top_bnc + ray.num_bot_bnc > num_bnc_min + 1: if ray.num_top_bnc + ray.num_bot_bnc > num_bnc_min + 1:
continue continue
elif max(ray.xy[1,:]) < 100: elif max(ray.xy[1, :]) < 100:
continue continue
else: else:
Rays.append(ray) Rays.append(ray)
return Rays return Rays
def call_hjq(r, TL): def call_hjq(r, TL):
TL_R = 20 * np.log10(r) TL_R = 20 * np.log10(r)
out=[] out = []
TL[np.isnan(TL)] = 0 TL[np.isnan(TL)] = 0
TL_R[np.isnan(TL_R)] = 0 TL_R[np.isnan(TL_R)] = 0
dTL = (TL_R - TL) * (TL <= TL_R) dTL = (TL_R - TL) * (TL <= TL_R)
dTL[np.isnan(dTL)] = 0 dTL[np.isnan(dTL)] = 0
test_fliter_s = savgol_filter(dTL, 103, 3) test_fliter_s = savgol_filter(dTL, 103, 3)
test_fliter = medfilt(test_fliter_s,13) test_fliter = medfilt(test_fliter_s, 13)
signal=1 signal = 1
for r_3 in range(len(r)): for r_3 in range(len(r)):
if r[r_3]>30000: if r[r_3] > 30000:
break break
czrangewidth_list = [] czrangewidth_list = []
for i in range(r_3,len(test_fliter)): for i in range(r_3, len(test_fliter)):
czrangewidth = CzRangeWidth(0,0,0) czrangewidth = CzRangeWidth(0, 0, 0)
if(test_fliter[i]>=3): if (test_fliter[i] >= 3):
out.append(i) out.append(i)
if i+1<len(test_fliter): if i + 1 < len(test_fliter):
if test_fliter[i+1]<3: if test_fliter[i + 1] < 3:
m = r[out[0]] m = r[out[0]]
n = r[out[len(out) - 1]] n = r[out[len(out) - 1]]
l=n-m l = n - m
# file_cz_data.write('\n'"会聚区距离：%.1fm"%m +"-%.1fm" %n) # file_cz_data.write('\n'"会聚区距离：%.1fm"%m +"-%.1fm" %n)
# file_cz_data.write('\n'"会聚区宽度：%.1fm\n"%l ) # file_cz_data.write('\n'"会聚区宽度：%.1fm\n"%l )
czrangewidth_list.append(CzRangeWidth(m,n,l)) czrangewidth_list.append(CzRangeWidth(m, n, l))
out.clear() out.clear()
signal=0 signal = 0
else: else:
m = r[out[0]] m = r[out[0]]
n = r[out[len(out) - 1]] n = r[out[len(out) - 1]]
l = n - m l = n - m
# file_cz_data.write('\n'"会聚区距离：%.1fm" % m + "-%.1fm" % n) # file_cz_data.write('\n'"会聚区距离：%.1fm" % m + "-%.1fm" % n)
# file_cz_data.write('\n'"会聚区宽度：%.1fm\n" % l ) # file_cz_data.write('\n'"会聚区宽度：%.1fm\n" % l )
czrangewidth_list.append(CzRangeWidth(m,n,l)) czrangewidth_list.append(CzRangeWidth(m, n, l))
out.clear() out.clear()
signal = 0 signal = 0
# if signal==1: # if signal==1:
# file_cz_data.write('\n'" 没有出现会聚区\n") # file_cz_data.write('\n'" 没有出现会聚区\n")
#滤波器效果图： # 滤波器效果图：
#plt.plot(r, TL) # plt.plot(r, TL)
#plt.plot(r, TL_R) # plt.plot(r, TL_R)
# plt.plot(r,dTL) # plt.plot(r,dTL)
# plt.plot(r, test_fliter) # plt.plot(r, test_fliter)
# plt.gca().invert_yaxis() # plt.gca().invert_yaxis()
# plt.show() # plt.show()
czlen=len(czrangewidth_list)-1 czlen = len(czrangewidth_list) - 1
if czlen>=1: if czlen >= 1:
for i in range(1,czlen,1): for i in range(1, czlen, 1):
if czrangewidth_list[i].minRange-czrangewidth_list[i-1].maxRange<=2000: if czrangewidth_list[i].minRange - czrangewidth_list[i - 1].maxRange <= 2000:
czrangewidth_list[i].minRange = czrangewidth_list[i-1].minRange czrangewidth_list[i].minRange = czrangewidth_list[i - 1].minRange
czrangewidth_list[i].width=czrangewidth_list[i].maxRange-czrangewidth_list[i].minRange czrangewidth_list[i].width = czrangewidth_list[i].maxRange - czrangewidth_list[i].minRange
czrangewidth_list[i - 1]=0 czrangewidth_list[i - 1] = 0
while czrangewidth_list.count(0) > 0: while czrangewidth_list.count(0) > 0:
for j in czrangewidth_list: for j in czrangewidth_list:
@@ -259,19 +265,19 @@ def call_hjq(r, TL):
return czrangewidth_list return czrangewidth_list
def call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=False): def call_fangwei(freq, sd, rd, bathm, ssp, sed, base, azi, isPlot=False):
czdetail = CzDetail("", [], []) czdetail = CzDetail("", [], [])
last=len(ssp[0].c)-1 last = len(ssp[0].c) - 1
# if ssp[0].c[0] > ssp[0].c[last]: # if ssp[0].c[0] > ssp[0].c[last]:
# file_cz_data.write('\n'"直达会聚区：无") # file_cz_data.write('\n'"直达会聚区：无")
# file_cz_data.write('\n'"反射会聚区：") # file_cz_data.write('\n'"反射会聚区：")
# else: # else:
# file_cz_data.write('\n'"直达会聚区：") # file_cz_data.write('\n'"直达会聚区：")
sed = None sed = None
rays_total = call_Bellhop_Rays(freq, sd, rd, bathm, ssp, sed, base) rays_total = call_Bellhop_Rays(freq, sd, rd, bathm, ssp, sed, base, azi)
Rays = find_cvgcRays(rays_total) Rays = find_cvgcRays(rays_total)
[Pos, TL] = call_Bellhop(freq, sd, rd, bathm, ssp, sed, base) [Pos, TL] = call_Bellhop(freq, sd, rd, bathm, ssp, sed, base, azi)
[x, x, x, x, Pos, pressure] = read_shd('cz.shd') [x, x, x, x, Pos, pressure] = read_shd('cz'+str(azi)+'.shd')
TL = -10 * np.log10(np.square(np.abs(pressure))) # Transmission Loss TL = -10 * np.log10(np.square(np.abs(pressure))) # Transmission Loss
# Plot # Plot
if isPlot: if isPlot:
@@ -281,7 +287,7 @@ def call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=False):
plt.gca().invert_yaxis() plt.gca().invert_yaxis()
plt.colorbar() plt.colorbar()
plt.show() plt.show()
czrangewidth_list = call_hjq(Pos.r.range,TL[0, 0, 1,:]) czrangewidth_list = call_hjq(Pos.r.range, TL[0, 0, 1, :])
if ssp[0].c[0] > ssp[0].c[last]: if ssp[0].c[0] > ssp[0].c[last]:
czdetail.directCz = [] czdetail.directCz = []
czdetail.reflectCz = czrangewidth_list czdetail.reflectCz = czrangewidth_list
@@ -290,20 +296,37 @@ def call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=False):
czdetail.reflectCz = [] czdetail.reflectCz = []
return Pos.r.range, Pos.r.depth, TL[0, 0, :, :], bathm, Rays, czdetail return Pos.r.range, Pos.r.depth, TL[0, 0, :, :], bathm, Rays, czdetail
def call_cz(database, freq, lat, lon, azi, sourceDepth=10, receiveDepth=10, isPlot=False): def call_cz(database, freq, lat, lon, azi, sourceDepth=10, receiveDepth=10, isPlot=False):
sd = np.array([sourceDepth]) # Source depth sd = np.array([sourceDepth]) # Source depth
rd = np.linspace(0, receiveDepth, 351) # Receiving depth rd = np.linspace(0, receiveDepth, 351) # Receiving depth
results = [] results = []
ranges, depths, TL, bathym, Rays = [], [], [], [], [] ranges, depths, TL, bathym, Rays = [], [], [], [], []
def child(i):
descripts = ["正北方向", "东北方向", "正东方向", "东南方向", "正南方向", "西南方向", "正西方向", "西北方向"]
theta = np.arange(0, 359, 45)
[bathm, ssp, sed, base] = GetEnv(database, 25, Coord(lon, lat), R=100, azi=theta[i])
if np.mean(bathm.d) < 2000:
# 不会出现会聚区
ddd=[]
else:
ranges, depths, TL, bathym, Rays, czdetail = call_fangwei(freq, sd, rd, bathm, ssp, sed, base, azi=theta[i],isPlot=True)
czdetail.descript = descripts[i]
results.append(CzData(theta[i], ranges, depths, TL, bathym, Rays, czdetail))
return results
def child_0(interval):
results=child(interval)
return results
''' 计算单个方向的会聚区 ''' ''' 计算单个方向的会聚区 '''
if azi != None: if azi != None:
[bathm, ssp, sed, base] = GetEnv(database, 25, Coord(lon, lat), R=100, azi=azi) [bathm, ssp, sed, base] = GetEnv(database, 25, Coord(lon, lat), R=100, azi=azi)
if np.mean(bathm.d) < 2000: if np.mean(bathm.d) < 2000:
return results return results
else: else:
ranges, depths, TL, bathym, Rays, czdetail = call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=isPlot) ranges, depths, TL, bathym, Rays, czdetail = call_fangwei(freq, sd, rd, bathm, ssp, sed, base,
isPlot=isPlot)
czdetail.descript = "{:.1f}°方向".format(azi) czdetail.descript = "{:.1f}°方向".format(azi)
results.append(CzData(0, ranges, depths, TL, bathym, Rays, czdetail)) results.append(CzData(0, ranges, depths, TL, bathym, Rays, czdetail))
return results return results
@@ -314,15 +337,7 @@ def call_cz(database, freq, lat, lon, azi, sourceDepth=10, receiveDepth=10, isPl
# file_cz_data.write("各个方位都不会出现会聚区") # file_cz_data.write("各个方位都不会出现会聚区")
return results return results
else: else:
descripts = ["正北方向", "东北方向", "正东方向", "东南方向", "正南方向", "西南方向", "正西方向", "西北方向"] for pi in range(8):
theta = np.arange(0, 359, 45) pi=Process(target=child_0,args=(pi,))
for i in range(8): pi.start()
[bathm, ssp, sed, base] = GetEnv(database, 25, Coord(lon, lat), R=100, azi=theta[i])
if np.mean(bathm.d) < 2000:
# 不会出现会聚区
continue
ranges, depths, TL, bathym, Rays, czdetail = call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=isPlot)
czdetail.descript = descripts[i]
results.append(CzData(theta[i], ranges, depths, TL, bathym, Rays, czdetail))
return results return results