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

更改了czcore.py文件中，会聚区计算时进行并行运算。

pyat/bellhop.py

@@ -96,7 +96,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
     bdy = Bndry(top, bottom)
     # Beam params
     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
     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

pyat/czcore.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-
+from multiprocessing import Process
 from os import system
 import numpy as np
 from pyat.AcousticPara import GetEnv, Coord
@@ -12,18 +12,21 @@ from pyat.config import AtBinPath
 
 file_cz_data = open('cz_data.txt', 'w')
 
+
 class CzRangeWidth:
     def __init__(self, minRange, maxRange, width):
         self.minRange = minRange
         self.maxRange = maxRange
         self.width = width
 
+
 class CzDetail:
     def __init__(self, descript, directCz, reflectCz):
         self.descript = descript
         self.directCz = directCz
         self.reflectCz = reflectCz
 
+
 class CzData:
     def __init__(self, azi, ranges, depths, TL, bathym, Rays, CzDetail):
         self.azi = azi
@@ -34,23 +37,25 @@ class CzData:
         self.Rays = Rays
         self.CzDetail = CzDetail
 
+
 def calZmax(ssp, sed):
-    d = np.array([],dtype=int)
+    d = np.array([], dtype=int)
     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)
     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
-    filename='cz'
+    filename = 'cz'+str(azi)
     Rmax = bathm.r[-1]  # unit: km
     dr = (1500 / freq) / 1000  # wavelength / 10, unit: km
     ran = np.linspace(0, Rmax, round(Rmax / dr))  # Receiving range (unit: km)
     Zmax, ssp_idx = calZmax(ssp, sed)
     """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))
     # Range of phase velocity
     cInt.Low = 1400
@@ -72,7 +77,6 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
 
     # Sediment
     if sed != None:
-
         Z = sed[0].z + ssp[0].z[-1]  # Depth
         Cp = sed[0].cp  # Speed of P-wave
         Cs = sed[0].cs  # Speed of S-wave
@@ -84,7 +88,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
     """Opt_top 已修改 8.26"""
     Opt_top = 'SVW'
     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)
 
     #   Bottom option
@@ -95,7 +99,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
     bdy = Bndry(top, bottom)
     # Beam params
     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
     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
@@ -103,8 +107,7 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
     """删除ssp 已修改 8.26"""
     # Write *.env file
     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)
     # Read sound field
@@ -113,9 +116,10 @@ def call_Bellhop(freq, sd, rd, bathm, ssp, sed, base):
 
     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
-    filename = 'cz_ray'
+    filename = 'cz_ray'+str(azi)
     Rmax = bathm.r[-1]  # unit: km
     dr = (1500 / freq) / 1000  # wavelength / 10, 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
     return get_rays(filename + ".ray")
 
+
 def find_cvgcRays(rays_total):
     Rays = []
     num_bnc_min = np.Inf
     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
 
     for ray in rays_total[0]:
         if ray.num_top_bnc + ray.num_bot_bnc > num_bnc_min + 1:
             continue
-        elif max(ray.xy[1,:]) < 100:
+        elif max(ray.xy[1, :]) < 100:
             continue
         else:
             Rays.append(ray)
     return Rays
 
+
 def call_hjq(r, TL):
     TL_R = 20 * np.log10(r)
-    out=[]
+    out = []
     TL[np.isnan(TL)] = 0
     TL_R[np.isnan(TL_R)] = 0
     dTL = (TL_R - TL) * (TL <= TL_R)
     dTL[np.isnan(dTL)] = 0
     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)):
-        if r[r_3]>30000:
+        if r[r_3] > 30000:
             break
 
     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)
-            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]]
                     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\n"%l )
-                    czrangewidth_list.append(CzRangeWidth(m,n,l))
+                    czrangewidth_list.append(CzRangeWidth(m, n, l))
                     out.clear()
-                    signal=0
+                    signal = 0
             else:
                 m = r[out[0]]
                 n = r[out[len(out) - 1]]
                 l = n - m
                 # file_cz_data.write('\n'"会聚区距离：%.1fm" % m + "-%.1fm" % n)
                 # file_cz_data.write('\n'"会聚区宽度：%.1fm\n" % l )
-                czrangewidth_list.append(CzRangeWidth(m,n,l))
+                czrangewidth_list.append(CzRangeWidth(m, n, l))
                 out.clear()
                 signal = 0
     # if signal==1:
     # 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, test_fliter)
     # plt.gca().invert_yaxis()
     # 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:
         for j in czrangewidth_list:
@@ -259,19 +265,19 @@ def call_hjq(r, TL):
     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("", [], [])
-    last=len(ssp[0].c)-1
+    last = len(ssp[0].c) - 1
     # if ssp[0].c[0] > ssp[0].c[last]:
     # file_cz_data.write('\n'"直达会聚区：无")
     # file_cz_data.write('\n'"反射会聚区：")
     # else:
     # file_cz_data.write('\n'"直达会聚区：")
     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)
-    [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
     # Plot
     if isPlot:
@@ -281,7 +287,7 @@ def call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=False):
         plt.gca().invert_yaxis()
         plt.colorbar()
         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]:
         czdetail.directCz = []
         czdetail.reflectCz = czrangewidth_list
@@ -290,20 +296,37 @@ def call_fangwei(freq, sd, rd, bathm, ssp, sed, base, isPlot=False):
         czdetail.reflectCz = []
     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):
     sd = np.array([sourceDepth])  # Source depth
     rd = np.linspace(0, receiveDepth, 351)  # Receiving depth
 
     results = []
     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:
         [bathm, ssp, sed, base] = GetEnv(database, 25, Coord(lon, lat), R=100, azi=azi)
         if np.mean(bathm.d) < 2000:
             return results
         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)
             results.append(CzData(0, ranges, depths, TL, bathym, Rays, czdetail))
             return results
@@ -314,15 +337,7 @@ def call_cz(database, freq, lat, lon, azi, sourceDepth=10, receiveDepth=10, isPl
         # file_cz_data.write("各个方位都不会出现会聚区")
         return results
     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