metradar 0.1.0__py3-none-any.whl

metradar/oa_couhua.py

@@ -0,0 +1,166 @@
+'''
+封装凑华的客观分析程序
+2023年4月6日
+朱文剑
+
+'''
+
+# %%
+import numpy as np
+import math
+import time
+from grid import *
+from grid_data import *
+
+# %%迭代函数
+def frprmn2(x,targe,grads,tol = 0.001,caculate_time = 60,show = False):
+    x0 = x
+    n = len(x0)
+    x_min = math.sqrt(1.0/n)
+    df = grads(x0)
+    g_0 = -df
+    h_0 = g_0
+    f_min_line0 = 0
+    f_min_line1 = targe(x0)
+    start_time = time.time()
+    for i in range(10000):
+        d0 = np.dot(df,h_0)
+        x1 = x0 + h_0 * x_min
+        df = grads(x1)
+        d1 = np.dot(df,h_0)
+        if d0 == d1:
+            f_min_line1 = targe(x0)
+            return x0
+        else:
+            x_min = x_min * d0/(d0 - d1)
+            x0 += h_0 * x_min
+        df = grads(x0)
+        g_1 = -df
+        v1 = np.dot(g_1 - g_0, g_1)
+        v2 = np.dot(g_0,g_0)
+        h_0 = g_1 + (v1/v2) * h_0
+        g_0 = g_1
+        if i%10 == 0:
+            f_min_line0 = f_min_line1
+            f_min_line1 = targe(x0)
+            if show:print(str(i) + "  " + str(f_min_line1))
+            delta = 2 * abs(f_min_line1 - f_min_line0) / (abs(f_min_line1) + abs(f_min_line0)) - tol
+            if delta < 0:
+                return x0
+        if(time.time() - start_time > caculate_time):
+            return x0
+    print('迭代分析超过10000次而退出')
+    return x0
+
+# 客观分析算法的代码
+# def interp_sg_oa(sta0,background,sm = 1,effectR = 1000,rate_of_model = 0,show = False):
+def interp_sg_oa(x,y,z,reso,boundary=None,sm = 1,effectR = 1000,rate_of_model = 0,show = False):
+
+    if boundary is None:
+        wlon = int(np.min(x))
+        elon = int(np.max(x))+1
+        slat = int(np.min(y))
+        nlat = int(np.max(y))+1
+    else:
+        wlon = boundary[0]
+        elon = boundary[1]
+        slat = boundary[2]
+        nlat = boundary[3]
+
+    grid0 = grid([wlon,elon,reso],[slat,nlat,reso])
+    background = grid_data(grid0)
+    grd = background.copy()
+    ig = ((x - grid0.slon) // grid0.dlon).astype(dtype = 'int16')
+    jg = ((y - grid0.slat) // grid0.dlat).astype(dtype = 'int16')
+    dx = (x - grid0.slon) / grid0.dlon - ig
+    dy = (y - grid0.slat) / grid0.dlat - jg
+
+    # zhuwj add
+    flag1 = ig+1 <= grid0.nlon-1
+    flag2 = jg+1 <= grid0.nlat-1
+    flag = flag1 & flag2
+    ig = ig[flag]
+    jg = jg[flag]
+    dx = dx[flag]
+    dy = dy[flag]
+    z = z[flag]
+
+    c00 = (1 - dx) * (1 - dy)
+    c01 = dx * (1 - dy)
+    c10 = (1-dx) * dy
+    c11 = dx * dy
+    lat = grd["lat"].values
+    sr = 1/np.power(np.cos(lat*math.pi/180),4)
+    
+    
+
+    def targe(x):
+        grdv =  x.reshape(grid0.nlat,grid0.nlon)
+        dx = grdv[:,:-2] + grdv[:,2:] - 2 * grdv[:,1:-1]
+        dx2 = dx**2
+        sum_dx =np.sum(dx2,axis=1)
+        cost1 = np.sum(sum_dx * sr)
+
+        dy = grdv[:-2,:] + grdv[2:,:] - 2 * grdv[1:-1,:]
+        dy2 = dy**2
+        sum_dy = np.sum(dy2)
+        cost1 = cost1 + sum_dy
+       
+        sta_g = c00 * grdv[jg, ig] + c01 * grdv[jg, ig + 1] + c10 * grdv[jg + 1, ig] + c11 * grdv[
+            jg + 1, ig + 1]
+        error = z - sta_g
+        cost2 = np.sum(error**2)
+        cost3 = 0
+        cost = sm * cost1 + cost2 + cost3
+        return cost
+
+    def grads(x):
+        grdv = x.reshape(grid0.nlat,grid0.nlon)
+        g1 = np.zeros(grdv.shape)
+
+        sr_expend = np.tile(sr,[grid0.nlon-2,1]).T
+
+        dx = 2 * (grdv[:,:-2] + grdv[:,2:] - 2 * grdv[:,1:-1])
+        dx_sr = dx * sr_expend
+        g1[:,:-2] = dx_sr
+        g1[:,2:] += dx_sr
+        g1[:,1:-1] -= 2*dx_sr
+
+        dy = 2 *(grdv[:-2,:] + grdv[2:,:] - 2 * grdv[1:-1,:])
+        g1[:-2,:] += dy
+        g1[2:,:] += dy
+        g1[1:-1,:] -= 2 * dy
+
+        g2 = np.zeros(grdv.shape)
+        sta_g = c00 * grdv[jg, ig] + c01 * grdv[jg, ig + 1] + c10 * grdv[jg + 1, ig] + c11 * grdv[jg + 1, ig + 1]
+        d = 2 * (sta_g - z)
+        g2[jg,ig] += d * c00
+        g2[jg,ig + 1]  += d * c01
+        g2[jg+1,ig] += d * c10
+        g2[jg+1,ig+1] += d * c11
+
+        g3 = 0
+        
+        g = sm * g1 + g2 + g3
+        return g.reshape(-1)
+
+    x = grd.values.reshape(-1)
+    x_oa = frprmn2(x, targe, grads,show  = show)
+    result = grid_data(grid0,x_oa)
+
+    grid_x = np.linspace(wlon, elon, grid0.nlon)
+    grid_y = np.linspace(slat, nlat, grid0.nlat)
+
+    gx, gy = np.meshgrid(grid_x, grid_y)
+
+    return gx, gy,result.data[0,0,0,0,:,:]
+    
+    # return result
+
+
+# %%
+
+if __name__ == "__main__":
+    #读取观测数据
+    pass
+