metradar 0.1.6__py3-none-any.whl → 0.1.8.2__py3-none-any.whl

metradar/{main_pydda.py → project/wind_retrieval/main_pydda.py}

@@ -4,7 +4,7 @@
 '''
 多普勒雷达三维风场反演脚本
 朱文剑
-2021.4
+
 '''
 # %%
 
@@ -13,7 +13,7 @@ import pydda
 import os
 import matplotlib.pyplot as plt
 import matplotlib
-from get_tlogp_from_sharppy import get_profile
+from metradar.util.get_tlogp_from_sharppy import get_profile
 import numpy as np
 import configparser
 from datetime import datetime
@@ -21,7 +21,8 @@ import math
 import argparse
 import shutil
 import sys
-from decode_fmt_pyart import read_cnrad_fmt
+from metradar.io.decode_fmt_pyart import read_cnrad_fmt
+import xarray as xr
 # import warnings
 # warnings.filterwarnings("ignore")
 
@@ -165,8 +166,6 @@ class GET_WIND_3D:
         self.g_numy = int(len(range(self.ylims[0],self.ylims[1]))*1000/self.yreso+1) # y方向格点数
         self.g_numz = int(len(range(self.zlims[0],self.zlims[1]))*1000/self.zreso+1) # z方向格点数
 
-    
-
     def draw_all_pic(self,bdrawhori=True,):
         resultnames = []
         files = os.listdir(self.outpath)
@@ -187,8 +186,7 @@ class GET_WIND_3D:
             resultnames.append(self.outpath + os.sep + ff)
             self.myprint(ff)
         
-        # ana_lat = 32.787
-        # ana_lon = 119.376    
+ 
         s_x = self.getDistance(self.ana_lat,self.ana_lon,self.ana_lat,self.standard_lon)
         s_y = self.getDistance(self.ana_lat,self.ana_lon,self.standard_lat,self.ana_lon)
         self.myprint('拼图中心经度为：%.3f'%self.standard_lon)
@@ -198,26 +196,22 @@ class GET_WIND_3D:
         self.myprint('X方向偏移量为：%d'%s_x)
         self.myprint('Y方向偏移量为：%d'%s_y)
 
-        # lev_xz = 55
-        # lev_yz = 51
         lev_yz = int((self.xlims[0] - s_x)/(self.xreso/1e3))
         lev_xz = int((s_y- self.ylims[0])/(self.yreso/1e3))
         self.myprint('X方向层次为：%d'%lev_xz)
         self.myprint('Y方向层次为：%d'%lev_yz)
 
 
-        Grids = [pyart.io.read_grid(resultname) for resultname in resultnames]
+        Grids = [xr.open_dataset(resultname) for resultname in resultnames]
 
-        x=Grids[0].x['data']
-        y=Grids[0].y['data']
-        z=Grids[0].z['data']
+        x=Grids[0].x.values
+        y=Grids[0].y.values
+        z=Grids[0].z.values
 
-        uwind=Grids[0].fields['u']['data']
-        vwind=Grids[0].fields['v']['data']
-        wwind=Grids[0].fields['w']['data']
+        uwind=Grids[0]['u'].values
+        vwind=Grids[0]['v'].values
+        wwind=Grids[0]['w'].values
 
-        # vor_stretch = Grids[0].fields['vor_stretch']['data']
-        # vor_tilt = Grids[0].fields['vor_tilt']['data']
 
         picpath_horiz_bar = self.picpath + os.sep + '水平风场'
         picpath_horiz_stream = self.picpath + os.sep + '水平流场'
@@ -254,7 +248,7 @@ class GET_WIND_3D:
                 fig, ax = plt.subplots(figsize=(8, 6))
                 pydda.vis.plot_horiz_xsection_barbs(Grids, ax, 'reflectivity', level=ll,
                                                     # w_vel_contours=[3, 5,],
-                                                    cmap='pyart_NWSRef', vmin=0, vmax=70,
+                                                    cmap='NWSRef', vmin=0, vmax=70,
                                                     barb_spacing_x_km=5,
                                                     barb_spacing_y_km=5,
                                                     show_lobes=False)
@@ -273,7 +267,7 @@ class GET_WIND_3D:
    
                 fig, ax = plt.subplots(figsize=(8, 6))
                 pydda.vis.plot_horiz_xsection_streamlines(Grids, ax, 'reflectivity', level=ll,
-                                                        cmap='pyart_NWSRef', vmin=0, vmax=70,
+                                                        cmap='NWSRef', vmin=0, vmax=70,
                                                         #   w_vel_contours=[3, 5,],
                                                         show_lobes=False,
                                                         )
@@ -287,71 +281,13 @@ class GET_WIND_3D:
                     self.myprint(picpath_horiz_stream + os.sep + picfilename + ' saved!')
 
                 plt.close(fig)
-
-            # # draw stretch vortex pic
-            # for ll in range(self.g_numz):
-            
-            #     fig, axes = plt.subplots(figsize=(8, 6))
-              
-            #     # DPI = fig.get_dpi()
-            #     # fig.set_size_inches(1200.0/float(DPI),900.0/float(DPI))
-            #     plt.tight_layout(pad=0, h_pad=0, w_pad=0, rect=(0.035,0.01,0.91,0.9))
-            #     ax = plt.axes()
-            #     #ax.set_extent([-5000, 5000, -2500, 7500])
-            #     levels=np.arange(-10,12, 2)
-            #     colors = [(1, 0, 1),(0.23,0.58,0.96),(1, 1, 1),(1,0.75,0.24),(1,0,0)]
-            #     cm = LinearSegmentedColormap.from_list("wbyr", colors, N=100)
-            #     cset1=ax.contourf(x, y,vor_stretch[ll,:,:],cmap=cm,levels=levels, extend='both', origin='image')
-            #     ax1 = fig.add_axes([0.912, 0.3, 0.03, 0.55])
-            #     cbar=fig.colorbar(cset1,cax=ax1,orientation='vertical',ticks=levels)
-                
-            #     ax.barbs(x[::3],y[::3],uwind[ll,::3,::3], vwind[ll,::3,::3],flagcolor=(0.1,0.2,0.1,0),length=4.5,linewidth=0.6,
-            #                     barb_increments=dict(half=2, full=4, flag=20),sizes=dict(emptybarb=0.0), barbcolor=['k'],rounding=True,flip_barb=False)
-          
-            #     picfilename = 'vor_stretch_%02d_%s.png'%(ll+1,self.time_files)
-            #     if self.bsavepic:
-            #         fig.savefig(picpath_vor_stretch + os.sep + picfilename, 
-            #                     format='png', transparent=False, dpi=300, pad_inches = 0)
-            #         self.myprint(picpath_vor_stretch + os.sep + picfilename + ' saved!')
-                
-
-            #     plt.close(fig)
-                
-            # # draw tilt vortex pic
-            # for ll in range(self.g_numz):
-            
-            #     fig, axes = plt.subplots(figsize=(8, 6))
-              
-            #     # DPI = fig.get_dpi()
-            #     # fig.set_size_inches(1200.0/float(DPI),900.0/float(DPI))
-            #     plt.tight_layout(pad=0, h_pad=0, w_pad=0, rect=(0.035,0.01,0.91,0.9))
-            #     ax = plt.axes()
-            #     #ax.set_extent([-5000, 5000, -2500, 7500])
-            #     levels=np.arange(-10,12, 2)
-            #     colors = [(1, 0, 1),(0.23,0.58,0.96),(1, 1, 1),(1,0.75,0.24),(1,0,0)]
-            #     cm = LinearSegmentedColormap.from_list("wbyr", colors, N=100)
-            #     cset1=ax.contourf(x, y,vor_tilt[ll,:,:],cmap=cm,levels=levels, extend='both', origin='image')
-            #     ax1 = fig.add_axes([0.912, 0.3, 0.03, 0.55])
-            #     cbar=fig.colorbar(cset1,cax=ax1,orientation='vertical',ticks=levels)
-                
-            #     ax.barbs(x[::3],y[::3],uwind[ll,::3,::3], vwind[ll,::3,::3],flagcolor=(0.1,0.2,0.1,0),length=4.5,linewidth=0.6,
-            #                     barb_increments=dict(half=2, full=4, flag=20),sizes=dict(emptybarb=0.0), barbcolor=['k'],rounding=True,flip_barb=False)
-          
-            #     picfilename = 'vor_tilt_%02d_%s.png'%(ll+1,self.time_files)
-            #     if self.bsavepic:
-            #         fig.savefig(picpath_vor_tilt + os.sep + picfilename, 
-            #                     format='png', transparent=False, dpi=300, pad_inches = 0)
-            #         self.myprint(picpath_vor_tilt + os.sep + picfilename + ' saved!')
-                
-
-            #     plt.close(fig)
                 
         try:
             #=================
             fig, ax = plt.subplots(figsize=(8, 4))
             pydda.vis.plot_xz_xsection_barbs(Grids, ax, 'reflectivity', level=lev_xz,
                                             #  w_vel_contours=[3, 5,],
-                                            cmap='pyart_NWSRef', vmin=0, vmax=70,
+                                            cmap='NWSRef', vmin=0, vmax=70,
                                             barb_spacing_x_km=2.0,
                                             barb_spacing_z_km=1.0)
             ax.set_xlim(self.xlims.tolist())
@@ -370,7 +306,7 @@ class GET_WIND_3D:
         try:
             fig, ax = plt.subplots(figsize=(8, 4))
             pydda.vis.plot_xz_xsection_streamlines(Grids, None, 'reflectivity', level=lev_xz,
-                                                cmap='pyart_NWSRef', vmin=0, vmax=70,
+                                                cmap='NWSRef', vmin=0, vmax=70,
                                                 #  w_vel_contours=[3, 5,],
                                                 )
             ax.set_xlim(self.xlims.tolist())
@@ -389,7 +325,7 @@ class GET_WIND_3D:
             fig, ax = plt.subplots(figsize=(8, 4))
             pydda.vis.plot_yz_xsection_barbs(Grids, None, 'reflectivity', level=lev_yz,
                                             #  w_vel_contours=[ 3, 5, ],
-                                            cmap='pyart_NWSRef', vmin=0, vmax=70,
+                                            cmap='NWSRef', vmin=0, vmax=70,
                                             barb_spacing_y_km=2,
                                             barb_spacing_z_km=1)
             ax.set_xlim(self.ylims.tolist())
@@ -407,7 +343,7 @@ class GET_WIND_3D:
         try:
             fig, ax = plt.subplots(figsize=(8, 4))
             pydda.vis.plot_yz_xsection_streamlines(Grids, None, 'reflectivity', level=lev_yz,
-                                                cmap='pyart_NWSRef', vmin=0, vmax=70,
+                                                cmap='NWSRef', vmin=0, vmax=70,
                                                 #  w_vel_contours=[3, 5, ],
                                                 )
             ax.set_xlim(self.ylims.tolist())
@@ -429,8 +365,7 @@ class GET_WIND_3D:
         
         filepaths = []
         filenames = []
-        # print('kevin')
-        # import pdb; pdb.set_trace()
+
         for rd in self.radars:
             curpath = self.radarfilepath + os.sep + rd
             filepaths.append(curpath)
@@ -440,8 +375,6 @@ class GET_WIND_3D:
             files = [ff for ff in files if ff.find('Z_RADR')>=0]
             times = [datetime.strptime(ff[15:29], '%Y%m%d%H%M%S').timestamp()  for ff in files]
 
-            # files = [ff for ff in files if ff.find('ar2v')>=0]
-            # times = [datetime.strptime(ff[5:13]+ff[14:20], '%Y%m%d%H%M%S').timestamp()  for ff in files]
 
             if len(times) == 0:
                 self.myprint('未找到 %s 雷达站的有效数据文件，请检查！'%rd)
@@ -458,22 +391,22 @@ class GET_WIND_3D:
         self.write_to_progress_main(5)    
 
 
-        # filename1 = 'HAJS.20200612.053920.ar2v'
-        # filename2 = 'NJJS.20200612.054229.ar2v'
-        # filename3 = 'TZJS.20200612.054032.ar2v'
         radar_objs=[]
         grid_objs=[]
         self.outname_grid=[]
         self.valid_radars=len(filenames)
         for nn in range(self.valid_radars):
             outname = filenames[nn].split('.')
-            self.outname_grid.append(filenames[nn].replace(outname[-1],'grid.nc'))
+            self.outname_grid.append(filenames[nn].replace(outname[-1],'wind.nc'))
 
         for nn in range(self.valid_radars):
             pass
 
-            # radar_objs.append(pyart.io.read_nexrad_archive(filepaths[nn] + os.sep + filenames[nn]))
-            radar_objs.append(read_cnrad_fmt(filepaths[nn] + os.sep + filenames[nn]))
+            radar = read_cnrad_fmt(filepaths[nn] + os.sep + filenames[nn])
+            # 退速速模糊
+            newvel = pyart.correct.dealias_region_based(radar, vel_field='velocity')
+            radar.add_field('corrected_velocity', newvel, replace_existing=True)
+            radar_objs.append(radar)
 
             self.myprint('read %s over!'%filenames[nn])
 
@@ -483,15 +416,42 @@ class GET_WIND_3D:
             weighting_function = 'BARNES2',
             grid_shape=(self.g_numz, self.g_numy, self.g_numx),
             grid_limits=(self.g_zlim, self.g_ylim, self.g_xlim),
-            fields=['reflectivity','velocity']))
+            fields=['reflectivity','velocity','corrected_velocity']).to_xarray())
             self.myprint('grid over!')
 
-            # radar1 = pyart.io.read_nexrad_archive(filepath1 + os.sep + filename1)
-            # print('read %s over!'%filename1)
         self.write_to_progress_main(20)
 
         # step 2=============风场反演
-
+        
+        # 给grid对象添加部分属性point_x, point_y, point_z, AZ, EL, point_altitude
+        # 按z，y，x的顺序，进行meshgrid
+        point_z,point_y,point_x = np.mgrid[
+            self.g_zlim[0]:self.g_zlim[1]+self.zreso:self.zreso,
+            self.g_ylim[0]:self.g_ylim[1]+self.yreso:self.yreso,
+            self.g_xlim[0]:self.g_xlim[1]+self.xreso:self.xreso
+        ]
+
+        # 将point_x, point_y, point_z添加到grid对象中
+        for nn in range(len(grid_objs)):
+            grid_objs[nn]['point_x'] = (('z','y','x'), point_x)
+            grid_objs[nn]['point_y'] = (('z','y','x'), point_y)
+            grid_objs[nn]['point_z'] = (('z','y','x'), point_z)
+
+        # 根据z，y，x坐标，计算对应的方位角和仰角，并存储在AZ变量和EL变量中
+        for nn in range(len(grid_objs)):
+            grid_objs[nn]['AZ'] = (('z','y','x'), pyart.core.cartesian_to_antenna(
+                grid_objs[nn]['point_x'].data,
+                grid_objs[nn]['point_y'].data,
+                grid_objs[nn]['point_z'].data)[1])
+            grid_objs[nn]['EL'] = (('z','y','x'), pyart.core.cartesian_to_antenna(
+                grid_objs[nn]['point_x'].data,
+                grid_objs[nn]['point_y'].data,
+                grid_objs[nn]['point_z'].data)[2])
+        # 将高度信息存储在point_altitude变量中,直接采用z坐标值
+        for nn in range(len(grid_objs)):
+            grid_objs[nn]['point_altitude'] = (('z','y','x'), grid_objs[nn]['point_z'].data)
+            
+            
         # 添加探空数据
         if not os.path.exists(self.tlogp_filepath + os.sep + self.tlogp_filename):
             self.myprint(self.tlogp_filepath + os.sep + self.tlogp_filename + ' not exists! please check!')
@@ -499,8 +459,9 @@ class GET_WIND_3D:
             return False
 
         profile = get_profile(self.tlogp_filepath,self.tlogp_filename)
-        u_init, v_init, w_init = pydda.initialization.make_wind_field_from_profile(
-                grid_objs[0], profile, vel_field='velocity')
+        grid_objs[0]= pydda.initialization.make_wind_field_from_profile(
+                grid_objs[0], profile, vel_field='corrected_velocity')
+        
         # u_init, v_init, w_init = None,None,None
         self.myprint('探空数据 %s 加载完毕！'%self.tlogp_filename)
         # 速度退模糊暂时空缺，后期补上
@@ -508,19 +469,38 @@ class GET_WIND_3D:
 
         #进行多雷达三维风场反演
         # try:
-        Grids = pydda.retrieval.get_dd_wind_field(grid_objs, u_init,
-                                            v_init, w_init, Co=100.0, Cm=1500.0,
-                                            Cz=0, vel_name='velocity', refl_field='reflectivity',
-                                            frz=5000.0,max_iterations=120,
-                                            mask_outside_opt=True,upper_bc=1,)# print=self.myprint
+
+        print('=================================')
+        # return new_grid_list, parameters
+        Grids, _ = pydda.retrieval.get_dd_wind_field(
+                                                    grid_objs,
+                                                    Co=100.0,
+                                                    Cm=1500.0,
+                                                    Cx=1e-2,
+                                                    Cy=1e-2,
+                                                    Cz=1e-2,
+                                                    vel_name="corrected_velocity",
+                                                    refl_field="reflectivity",
+                                                    frz=5000.0,
+                                                    engine="tensorflow", # 有GPU就用tensorflow，没有就用scipy或jax
+                                                    mask_outside_opt=True,
+                                                    upper_bc=1,
+                                                )
         # except:
         #     self.myprint('get_dd_wind_field error!')
         #     self.write_to_progress_main(100)
         #     return False
-
-        for nn in range(self.valid_radars):
-            Grids[nn].fields['reflectivity']['data'][Grids[nn].fields['reflectivity']['data']<10] = np.ma.masked
-            pyart.io.write_grid(self.outpath + os.sep + self.outname_grid[nn], Grids[nn])
+        
+        # 计算拉伸涡度和倾斜涡度
+        Grids = self.calc_vortex(Grids)
+        
+        for nn in range(len(Grids)):
+            # 将reflectivity小于10的格点设置为缺省值
+            Grids[nn]['reflectivity'] = Grids[nn]['reflectivity'].where(Grids[nn]['reflectivity']>=10)
+            outname = filenames[nn].split('.')
+            
+            Grids[nn].to_netcdf(self.outpath + os.sep + filenames[nn].replace(outname[-1],'wind.nc'))
+            
 
         # exclude masked gates from the gridding
         # gatefilter = pyart.filters.GateFilter(Grids)
@@ -530,9 +510,6 @@ class GET_WIND_3D:
         self.write_to_progress_main(80)
         # print('All levels = %d'%g_numz)
         
-        # 计算拉伸涡度和倾斜涡度
-        # self.calc_vortex(Grids)
-
         # self.draw_all_pic()
 
         self.write_to_progress_main(100)
@@ -545,55 +522,76 @@ class GET_WIND_3D:
     #计算拉伸涡度和倾斜涡度
     def calc_vortex(self,Grids):
         self.myprint('starting calc_vortex......')
-        files = os.listdir(self.outpath)
-        if len(files)==0:
-            self.myprint('no files in output path!')
-            return False
-        filename = files[0]
-
-        # Grids = pyart.io.read_grid(self.outpath + os.sep + filename)
-        # dataset = nc.Dataset(self.outpath + os.sep + filename)
         
-        uwind=Grids[0].fields['u']['data']
-        vwind=Grids[0].fields['v']['data']
-        wwind=Grids[0].fields['w']['data']
+
         
-        dudz,dudy,dudx = np.gradient(uwind)
-        dvdz,dvdy,dvdx = np.gradient(vwind)
-        dwdz,dwdy,dwdx = np.gradient(wwind)
+        if isinstance(Grids, list):
+            uwind=Grids[0]['u'].values
+            vwind=Grids[0]['v'].values
+            wwind=Grids[0]['w'].values
+        else:
+            uwind=Grids['u'].values
+            vwind=Grids['v'].values
+            wwind=Grids['w'].values
+        # 检查uwind的维度是否是四维
+        if uwind.ndim != 4:
+            raise ValueError("Expected uwind to be a 4D array with dimensions (time, z, y, x)")
         
         vor_stretch_data = np.zeros(uwind.shape,dtype=float)
         vor_tilt_data = np.zeros(uwind.shape,dtype=float)
-        for lev in range(uwind.shape[0]):
-            # picname="stretch_%03d.png"%lev
-            # print(lev)
+        for tt in range(uwind.shape[0]):
+            dudz,dudy,dudx = np.gradient(uwind[tt,:,:,:])
+            dvdz,dvdy,dvdx = np.gradient(vwind[tt,:,:,:])
+            dwdz,dwdy,dwdx = np.gradient(wwind[tt,:,:,:])
             
-            div=dudx[lev,:,:] + dvdy[lev,:,:]
-            vor=dvdx[lev,:,:] - dudy[lev,:,:]
-            
-            u=uwind[lev,:,:]
-            v=vwind[lev,:,:]
-            #dw
-            
-            #拉伸项
-            vor_stretch_data[lev,:,:]=vor*div
+            for lev in range(uwind.shape[1]):
+                # picname="stretch_%03d.png"%lev
+                # print(lev)
+                
+                div=dudx[lev,:,:] + dvdy[lev,:,:]
+                vor=dvdx[lev,:,:] - dudy[lev,:,:]
+                
+                u=uwind[tt,lev,:,:]
+                v=vwind[tt,lev,:,:]
+                #dw
+                
+                #拉伸项
+                vor_stretch_data[tt,lev,:,:]=vor*div
 
-            #倾斜项
-            vor_tilt_data[lev,:,:]=dwdy[lev,:,:]*dudz[lev,:,:]-dwdx[lev,:,:]*dvdz[lev,:,:]
+                #倾斜项
+                vor_tilt_data[tt,lev,:,:]=dwdy[lev,:,:]*dudz[lev,:,:]-dwdx[lev,:,:]*dvdz[lev,:,:]
 
-        
-        vor_stretch = {'data':vor_stretch_data,'_FillValue':-9999.0,'long_name':'stretch vortex','units':'m-2/s-2','coordinates':'z,y,x'}
-        vor_tilt  = {'data':vor_tilt_data,'_FillValue':-9999.0,'long_name':'tilting vortex','units':'m-2/s-2','coordinates':'z,y,x'}
 
-        for nn in range(len(Grids)):
-            Grids[nn].add_field('vor_stretch',vor_stretch,replace_existing=True)
-           
-            Grids[nn].add_field('vor_tilt',vor_tilt,replace_existing=True)
+        if isinstance(Grids, list):
+            for nn in range(len(Grids)):
+                Grids[nn]['vor_stretch'] = (('time','z','y','x'), vor_stretch_data)
+                Grids[nn]['vor_tilt'] = (('time','z','y','x'), vor_tilt_data)
+                
+                # add attrs
+                
+                Grids[nn]['vor_stretch'].attrs['long_name'] = 'stretch vortex'
+                Grids[nn]['vor_stretch'].attrs['units'] = 'm-2/s-2'
+                Grids[nn]['vor_stretch'].attrs['coordinates'] = 't,z,y,x'
+                
+                Grids[nn]['vor_tilt'].attrs['long_name'] = 'tilting vortex'
+                Grids[nn]['vor_tilt'].attrs['units'] = 'm-2/s-2'
+                Grids[nn]['vor_tilt'].attrs['coordinates'] = 't,z,y,x'
             
+        else:
+            Grids['vor_stretch'] = (('time','z','y','x'), vor_stretch_data)
+            Grids['vor_tilt'] = (('time','z','y','x'), vor_tilt_data)
+            Grids['vor_stretch'].attrs['long_name'] = 'stretch vortex'
+            Grids['vor_stretch'].attrs['units'] = 'm-2/s-2'
+            Grids['vor_stretch'].attrs['coordinates'] = 't,z,y,x'
+            
+            Grids['vor_tilt'].attrs['long_name'] = 'tilting vortex'
+            Grids['vor_tilt'].attrs['units'] = 'm-2/s-2'
+            Grids['vor_tilt'].attrs['coordinates'] = 't,z,y,x'
+        
 
-            pyart.io.write_grid(self.outpath + os.sep + self.outname_grid[nn], Grids[nn])
-            self.myprint(self.outname_grid[nn] + ' updated with vor_stretch and vor_tilt!')
-        kkk=0
+        return Grids
+
+            
     
 
 if __name__ == "__main__":
@@ -635,11 +633,16 @@ if __name__ == "__main__":
     parser.add_argument('--drawcross', dest='drawpic_cross', action='store_const', 
                         const=drawpic_cross,
                         help='draw XZ and YZ pic')
+    #是否为debug模式
     BDEBUG=True
 
     if BDEBUG:
         print('当前为debug模式，所以直接读取主目录下的config.ini, 请注意！')
-        mainprog('metradar/config_3dwind.ini')
+        config_path = '/home/wjzhu/OneDrive/PythonCode/MyWork/metradar/metradar/project/wind_retrieval/config_3dwind.ini'
+        if not os.path.exists(config_path):
+            print('%s not exists! please check!'%config_path)
+            sys.exit(-1)
+        mainprog(config_path)
         
     else:
         args = parser.parse_args()

metradar/util/__init__.py

@@ -0,0 +1,9 @@
+"""
+metradar is a python package to read or retrieve radar data
+
+"""
+
+__author__ = "The R & D Center for Weather Forecasting Technology in NMC, CMA"
+
+
+__all__ = [s for s in dir() if not s.startswith('_')]

metradar/{comm_func.py → util/comm_func.py}

@@ -1,9 +1,6 @@
 
 
 import numpy as np
-# from metpy.calc import wind_direction as metpy_wind_direction,geopotential_to_height
-# from metpy.units import units
-
 def calc_td(temperature,relative_humidity):
     # temperature C degree
     # relative_humidity 0~1
@@ -86,41 +83,6 @@ def geopotential_to_height(geopotential):
     `pint.Quantity`
         Corresponding value(s) of height above sea level
 
-    Examples
-    --------
-    >>> import metpy.calc
-    >>> from metpy.units import units
-    >>> height = np.linspace(0, 10000, num=11) * units.m
-    >>> geopot = metpy.calc.height_to_geopotential(height)
-    >>> geopot
-    <Quantity([     0.           9805.11097983 19607.1448853  29406.10316465
-    39201.98726524 48994.79863351 58784.53871501 68571.20895435
-    78354.81079527 88135.34568058 97912.81505219], 'meter ** 2 / second ** 2')>
-    >>> height = metpy.calc.geopotential_to_height(geopot)
-    >>> height
-    <Quantity([     0.   1000.   2000.   3000.   4000.   5000.   6000.   7000.   8000.
-    9000.  10000.], 'meter')>
-
-    See Also
-    --------
-    height_to_geopotential
-
-    Notes
-    -----
-    This calculation approximates :math:`g(z)` as
-
-    .. math:: g(z) = g_0 \left( \frac{R_e}{R_e + z} \right)^2
-
-    where :math:`g_0` is standard gravity. It thereby accounts for the average effects of
-    centrifugal force on apparent gravity, but neglects latitudinal variations due to
-    centrifugal force and Earth's eccentricity.
-
-    (Prior to MetPy v0.11, this formula instead calculated :math:`g(z)` from Newton's Law of
-    Gravitation assuming a spherical Earth and no centrifugal force effects.)
-
-    .. versionchanged:: 1.0
-       Renamed ``geopot`` parameter to ``geopotential``
-
     """
     Re = 6371008.7714
     g = 9.80665
@@ -128,8 +90,6 @@ def geopotential_to_height(geopotential):
 
 if __name__ == "__main__":
     pass
-    # print(wind_speed(3,3))
-    # print(wind_direction(-3,5))
-    # print(metpy_wind_direction(-3*units('m/s'),5*units('m/s')))
+
     g = 9.80665
     print(geopotential_to_height(5884*g))

metradar/util/exceptions.py

@@ -0,0 +1,50 @@
+"""
+Custom Py-ART exceptions.
+
+"""
+
+import warnings
+
+
+class MissingOptionalDependency(Exception):
+    """ Exception raised when a optional dependency is needed by not found. """
+    pass
+
+
+class DeprecatedAttribute(DeprecationWarning):
+    """ Warning category for an attribute which has been renamed/moved.  """
+    pass
+
+
+class DeprecatedFunctionName(DeprecationWarning):
+    """ Warning category for a function which has been renamed/moved. """
+    pass
+
+
+def _deprecated_alias(func, old_name, new_name):
+    """
+
+    A function for creating an alias to a renamed or moved function.
+
+    Parameters
+    ----------
+    func : func
+        The function which has been renamed or moved.
+    old_name, new_name : str
+        Name of the function before and after it was moved or renamed
+        (with namespace if changed).
+
+    Returns
+    -------
+    wrapper : func
+        A wrapper version of func, which issues a DeprecatedFunctionName
+        warning when the called.
+
+    """
+    def wrapper(*args, **kwargs):
+        warnings.warn(
+            ("{0} has been deprecated and will be removed in future " +
+             "versions of Py-ART, pleases use {1}. ").format(
+                 old_name, new_name), category=DeprecatedFunctionName)
+        return func(*args, **kwargs)
+    return wrapper