metradar 0.1.0__py3-none-any.whl

metradar/get_cross_section_from_pyart.py

@@ -0,0 +1,354 @@
+'''
+从pyart的radar object中获取任意垂直剖面的数据
+
+朱文剑
+2021.10.14
+
+'''
+
+import numpy as np
+from pyart.core import Radar
+import math
+# import time
+
+R = 6371 # km
+#该程序用来求直线方程系数
+#目前没考虑垂直X轴的直线，需要完善
+#kevin2075@163.com
+def linefunc(x,y,x1,y1): #直接输入两个坐标就可以求出一条直线方程。
+# if x~=x1
+    a=(y1-y)/(x1-x)
+    b=y-a*x
+    return [a,b]
+
+
+# 将多个数或列表组合成一个列表
+def myconcat(vars:tuple):
+    result=[]
+    for vartmp in vars:
+        if len(np.array(vartmp).shape)==0:
+            result.append(vartmp)
+        else:
+            for var in vartmp:
+                result.append(var)
+   
+    return result
+
+def get_cross_radar(radar:Radar,params):
+
+    # antenna_to_cartesian(147, 260, 1.5)
+    # startangle=None,startrange=None,endangle=None,endrange=None
+    # startangle = 8.5
+    # startrange = 69.23 #km
+
+    # endangle = 18.6;
+    # endrange =  59.03 #km
+    startangle = params['crs_start_azi']
+    startrange = params['crs_start_range']
+    endangle = params['crs_end_azi']
+    endrange = params['crs_end_range']
+    toph = params['top_height']
+
+    startx = startrange*np.sin(np.radians(startangle))
+    starty = startrange*np.cos(np.radians(startangle))
+    endx   = endrange*np.sin(np.radians(endangle))
+    endy   = endrange*np.cos(np.radians(endangle))
+
+    baseh = 0; # km 垂直剖面的起始高度
+    # toph = 20; # km 垂直剖面的上限高度
+
+    validzlv=[]
+    validvlv=[]
+    validReles=[]
+    validDeles=[]
+    outdic=dict()
+
+    for il in range(radar.nsweeps):
+        # if sweep(il).RGates > 0
+        # if il !=1 and il !=3:
+        #     validzlv.append(il)
+        #     validReles.append(radar.get_elevation(il).mean())
+        # if il !=0 and il !=2:
+        #     validvlv.append(il)
+        #     validDeles.append(radar.get_elevation(il).mean())
+        validzlv.append(il)
+        validReles.append(round(radar.get_elevation(il).mean(),1))
+        validvlv.append(il)
+        validDeles.append(round(radar.get_elevation(il).mean(),1))
+
+    tol_length = int(np.sqrt(np.power((endx-startx),2) + np.power((endy - starty),2)))
+
+    #求方程坐标，这里暂时没考虑X=0的情况，后面要完善这一点
+    [a,b] = linefunc(startx,starty,endx,endy)
+
+    # 获取反射率垂直剖面
+        
+    #构建变量，确定坐标维度
+    xreso = 0.05#km
+    if endx < startx:
+        xreso = -1*xreso
+
+    xnum = int(tol_length  / abs(xreso))
+    zreso = 0.05# km
+    znum = int((toph - baseh)  / zreso)
+
+    
+
+    beamwidth = radar.get_azimuth(validzlv[-1])
+    beamwidth = np.diff(beamwidth)
+    beamwidth = np.mean(beamwidth[beamwidth>0])
+    topel_R = max(validReles) + beamwidth / 2
+    baseel_R = min(validReles) - beamwidth / 2
+
+    topel_D = max(validDeles) + beamwidth / 2
+    baseel_D = min(validDeles) - beamwidth / 2
+
+    tmp=[]
+    for nn in range(len(validReles)-1):
+        tmp.append(np.mean([validReles[nn],validReles[nn+1]]))
+
+    tmp_newRel = sorted(myconcat((tmp,validReles)))
+
+    tmp=[]
+    for nn in range(len(validDeles)-1):
+        tmp.append(np.mean([validDeles[nn],validDeles[nn+1]]))
+
+    tmp_newDel = sorted(tmp+validDeles)
+
+    newRel = [baseel_R+tmp_newRel+topel_R]
+    newDel = [baseel_D+tmp_newDel+topel_D]
+
+    # for varkey in radar.fields.keys():
+     
+    nflag=0
+    outdic=dict()
+    outdic['vertical_km'] = toph - baseh
+    outdic['horizontal_km'] = tol_length
+    outdic['xreso'] = xreso
+    outdic['yreso'] = zreso
+    
+    for varkey in radar.fields.keys():
+        outdic[varkey]=[]
+        outdic[varkey] = np.zeros([znum,xnum],dtype='float')*np.nan
+
+    
+    # 纯 for 循环版本 =====================
+    valideles=[]
+    validlvl=[]
+    for varkey in radar.fields.keys():
+        print(varkey)
+        if varkey == 'reflectivity' :
+            valideles = validReles
+            validlvl = validzlv
+        elif varkey == 'velocity':
+            valideles = validDeles
+            validlvl = validvlv 
+        else:
+            # print('暂时仅支持回波强度和径向速度')
+            continue
+        
+        curfd = []
+        for il in range(radar.nsweeps):
+            curfd.append(radar.get_field(il,varkey))#sweep(targetsp).dbz(az,gates);
+
+        for iz in range(znum):
+            for ix in range(xnum):
+                tmpcurx = startx + (ix * xreso) * math.cos(math.atan(a))
+                
+                tmpcury = a * tmpcurx + b
+                tmpaz = (90 - math.degrees(math.atan2(tmpcury,tmpcurx)) + 360)%360
+                
+                curx = math.sqrt(math.pow(tmpcurx,2) + math.pow(tmpcury,2))#startrange + xreso * ix 
+                curz = baseh + iz * zreso
+                
+                #计算相对雷达的仰角
+                # ele = math.degrees(math.atan2(curz,curx))
+                # 这里必须用math.pow，如果直接用称号会有较大误差
+                ele = math.asin((math.pow((curz+R),2) - math.pow(curx,2) - math.pow(R,2))/2/curx/R)/np.pi*180
+
+                if ele < baseel_R or ele > topel_R:
+                    continue
+
+                tmp = list(abs(np.array(valideles) - ele))
+                idx = tmp.index(min(tmp))
+                idx_el = idx
+                
+                targetsp = validlvl[idx_el]
+                
+                #获取方位角角标
+                tmp = list(abs(radar.get_azimuth(targetsp) - tmpaz))
+                idx = tmp.index(min(tmp))
+                az = idx
+                
+                #经向距离，由该页面的公式进行反算
+                #https://arm-doe.github.io/pyart/API/generated/pyart.core.antenna_to_cartesian.html#pyart.core.antenna_to_cartesian
+                #根据s=R*??的公式反算r
+                
+                cur_range = (R+curz)*math.sin(curx/R)/math.cos(math.radians(valideles[idx_el]))
+                gates = math.floor(cur_range/(radar.range['meters_between_gates'] / 1000)+0.5)-1
+                # gates = math.floor(curx/math.cos(math.radians(valideles[idx_el]))/(radar.range['meters_between_gates'] / 1000)+0.5)-1
+                
+                if gates < radar.ngates:
+                    #disp(sweep(targetsp).dbz(az,gates));
+                    # if curfd[targetsp].data[az,gates] > -30 and not curfd[targetsp].mask[az,gates]:
+                    # if curfd[targetsp].data[az,gates] > -5:
+                    if not curfd[targetsp].mask[az,gates]:
+                        outdic[varkey][iz,ix] = curfd[targetsp].data[az,gates]
+                        nflag+=1
+    
+    return outdic
+
+# main function
+def get_cross_radar_old(radar:Radar,params):
+    # startangle=None,startrange=None,endangle=None,endrange=None
+    # startangle = 8.5
+    # startrange = 69.23 #km
+
+    # endangle = 18.6;
+    # endrange =  59.03 #km
+    startangle = params['crs_start_azi']
+    startrange = params['crs_start_range']
+    endangle = params['crs_end_azi']
+    endrange = params['crs_end_range']
+    toph = params['top_height']
+
+    startx = startrange*np.sin(np.radians(startangle))
+    starty = startrange*np.cos(np.radians(startangle))
+    endx   = endrange*np.sin(np.radians(endangle))
+    endy   = endrange*np.cos(np.radians(endangle))
+
+    baseh = 0; # km 垂直剖面的起始高度
+    # toph = 20; # km 垂直剖面的上限高度
+
+    validzlv=[]
+    validvlv=[]
+    validReles=[]
+    validDeles=[]
+    outdic=dict()
+
+    for il in range(radar.nsweeps):
+        # if sweep(il).RGates > 0
+        # if il !=1 and il !=3:
+        #     validzlv.append(il)
+        #     validReles.append(radar.get_elevation(il).mean())
+        # if il !=0 and il !=2:
+        #     validvlv.append(il)
+        #     validDeles.append(radar.get_elevation(il).mean())
+        validzlv.append(il)
+        validReles.append(round(radar.get_elevation(il).mean(),1))
+        validvlv.append(il)
+        validDeles.append(round(radar.get_elevation(il).mean(),1))
+
+    tol_length = np.sqrt(np.power((endx-startx),2) + np.power((endy - starty),2))
+
+    #求方程坐标，这里暂时没考虑X=0的情况，后面要完善这一点
+    [a,b] = linefunc(startx,starty,endx,endy)
+
+    # 获取反射率垂直剖面
+        
+    #构建变量，确定坐标维度
+    xreso = 0.25#km
+    if endx < startx:
+        xreso = -1*xreso
+
+    xnum = int(tol_length  / abs(xreso))
+    yreso = 0.25# km
+    ynum = int((toph - baseh)  / yreso)
+
+    
+
+    beamwidth = radar.get_azimuth(validzlv[-1])
+    beamwidth = np.diff(beamwidth)
+    beamwidth = np.mean(beamwidth[beamwidth>0])
+    topel_R = max(validReles) + beamwidth / 2
+    baseel_R = min(validReles) - beamwidth / 2
+
+    topel_D = max(validDeles) + beamwidth / 2
+    baseel_D = min(validDeles) - beamwidth / 2
+
+    tmp=[]
+    for nn in range(len(validReles)-1):
+        tmp.append(np.mean([validReles[nn],validReles[nn+1]]))
+
+    tmp_newRel = sorted(myconcat((tmp,validReles)))
+
+    tmp=[]
+    for nn in range(len(validDeles)-1):
+        tmp.append(np.mean([validDeles[nn],validDeles[nn+1]]))
+
+    tmp_newDel = sorted(tmp+validDeles)
+
+    newRel = [baseel_R+tmp_newRel+topel_R]
+    newDel = [baseel_D+tmp_newDel+topel_D]
+
+    # for varkey in radar.fields.keys():
+     
+    nflag=0
+    outdic=dict()
+    outdic['vertical_km'] = toph - baseh
+    outdic['horizontal_km'] = tol_length
+    outdic['xreso'] = xreso
+    outdic['yreso'] = yreso
+    
+    for varkey in radar.fields.keys():
+        outdic[varkey]=[]
+        outdic[varkey] = np.zeros([ynum,xnum],dtype='float')*np.nan
+
+    
+    # 纯 for 循环版本 =====================
+    valideles=[]
+    validlvl=[]
+    for varkey in radar.fields.keys():
+        print(varkey)
+        if varkey == 'reflectivity' :
+            valideles = validReles
+            validlvl = validzlv
+        elif varkey == 'velocity':
+            valideles = validDeles
+            validlvl = validvlv 
+        else:
+            # print('暂时仅支持回波强度和径向速度')
+            continue
+        
+        curfd = []
+        for il in range(radar.nsweeps):
+            curfd.append(radar.get_field(il,varkey))#sweep(targetsp).dbz(az,gates);
+
+        for iy in range(ynum):
+            for ix in range(xnum):
+                tmpcurx = startx + (ix * xreso) * math.cos(math.atan(a))
+                
+                tmpcury = a * tmpcurx + b
+                tmpaz = (90 - math.degrees(math.atan2(tmpcury,tmpcurx)) + 360)%360
+                
+                curx = math.sqrt(math.pow(tmpcurx,2) + math.pow(tmpcury,2))#startrange + xreso * ix 
+                cury = baseh + iy * yreso
+                
+                #计算相对雷达的仰角
+                ele = math.degrees(math.atan2(cury,curx))
+                # if ele < baseel_R: ele = baseel_R
+                # if ele > topel_R: continue
+                if ele < baseel_R or ele > topel_R:
+                    continue
+
+                tmp = list(abs(np.array(valideles) - ele))
+                idx = tmp.index(min(tmp))
+                idx_el = idx
+                
+                targetsp = validlvl[idx_el]
+                
+                #获取方位角角标
+                tmp = list(abs(radar.get_azimuth(targetsp) - tmpaz))
+                idx = tmp.index(min(tmp))
+                az = idx
+                
+                gates = math.ceil(curx/math.cos(math.radians(valideles[idx_el]))/(radar.range['meters_between_gates'] / 1000))
+                
+                if gates < radar.ngates:
+                    #disp(sweep(targetsp).dbz(az,gates));
+                    if curfd[targetsp].data[az,gates] > -30:
+                        outdic[varkey][iy,ix] = curfd[targetsp].data[az,gates]
+                        nflag+=1
+    
+    return outdic
+

metradar/get_tlogp_from_sharppy.py

@@ -0,0 +1,93 @@
+'''
+从sharppy格式的探空数据中获取风廓线，作为背景场
+2021.3.9
+朱文剑
+'''
+
+# %%
+import os
+import re
+import numpy as np
+from pyart.core.wind_profile import HorizontalWindProfile
+import warnings
+warnings.filterwarnings("ignore")
+
+def get_profile(filepath,filename):
+    # filepath = '/Users/wenjianzhu/Downloads/CMADAAS/tlogp'
+    # filename = '20061208.58238'
+
+    fin = open(filepath + os.sep + filename,'rt')
+
+    nflag_title=0
+    nflag_raw=0
+    nflag=0
+    profile=dict()
+    defaultvalue = '99999'
+    for line in fin.readlines():
+        # print(line)
+        nflag = nflag + 1
+        if line.find('%TITLE%') >=0 :
+            nflag_title = nflag
+        if line.find('%RAW%') >=0 :
+            nflag_raw = nflag
+
+            # 遇到文档结束
+        if line.find('%END%') >=0 :
+            break
+        #获取站号，时间信息
+        if nflag == nflag_title + 1:
+            profile['staname'] = line.split(' ')[0]
+        
+        #获取经纬度信息
+        if nflag == nflag_title + 2:
+            tmp = re.findall(r"\d+\.?\d*",line)
+            if len(tmp) == 2:
+                profile['lon'] = tmp[0]
+                profile['lat'] = tmp[1]
+            else:
+                profile['lon'] = defaultvalue
+                profile['lat'] = defaultvalue
+        
+        #获取变量名称
+        if nflag == nflag_title + 3:
+            tmp = line.split(' ')
+            varnames = [i.strip('\n') for i in tmp if i != '']
+            # print(varnames)
+            if len(varnames) ==6:
+                for dd in range(len(varnames)):
+                    profile[varnames[dd]]=[]
+                    # exec('{} = []'.format(varnames[dd]))
+            else:
+                print('变量个数不够，请检查')
+                break
+
+        # 数据
+        if nflag > nflag_raw:
+            tmp = re.findall(r"\d+\.?\d*",line)
+            if len(tmp) ==6:
+                values = np.fromstring( line, sep=',')
+                value_str = [str(value) for value in values]
+                for dd in range(len(value_str)):
+                    # exec('{}.append({})'.format(varnames[dd], value_str[dd]))
+                    profile[varnames[dd]].append(values[dd])
+            else:
+                continue
+    # for dd in range(len(varnames)):
+    #     exec("profile.update({'%s':%s})"%(varnames[dd],varnames[dd]))
+
+    outprofile = HorizontalWindProfile(
+            profile['HGHT'], np.array(profile['WSPD'])/1.94, profile['WDIR'], latitude=np.tile(float(profile['lat']), len(profile['HGHT'])), longitude=np.tile(float(profile['lon']), len(profile['HGHT'])))
+
+    fin.close()
+    print('profile read over!')
+    return outprofile   
+
+if __name__ == '__main__':
+    filepath = 'tlogp'
+    filename = '22060608.59280'
+    profile = get_profile(filepath,filename)
+    pass
+
+
+
+# %%