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

metradar/{mosaic_quickdraw.py → graph/draw_comp_mosaic.py}

@@ -1,6 +1,8 @@
+
+# _*_ coding: utf-8 _*_
+
 '''
-将不同层次的mosacdata stack在一起
-ZhuWJ
+快速绘制三维雷达组网拼图
 
 '''
 
@@ -17,10 +19,20 @@ from matplotlib.font_manager import FontProperties
 import pandas as pd
 from pyart.graph import common
 from datetime import datetime,timedelta
-from parse_pal import parse_pro
+from metradar.util.parse_pal import parse_pro,parse
 import xarray as xr
 
-def add_china_map_2cartopy(ax, name='province', facecolor='none',
+from metradar.config import CONFIG
+
+# 资源文件路径
+RESOURCES_PATH = CONFIG.get('SETTING','RESOURCES_PATH')
+FONT_FILE = RESOURCES_PATH + '/fonts/YaHeiConsolasHybrid_1.12.ttf'
+COLOR_FILE=RESOURCES_PATH + '/gr2_colors/default_BR_PUP2.pal'
+MAP_PATH = RESOURCES_PATH + '/国省市县审图号(GS (2019) 3082号/'
+STATION_FILE = RESOURCES_PATH + '/stations/cma_city_station_info.dat'
+
+
+def add_china_map_2cartopy(ax, name='province', facecolor='none',transform=None,
                            edgecolor='c', lw=2, **kwargs):
     """
     Draw china boundary on cartopy map.
@@ -34,40 +46,56 @@ def add_china_map_2cartopy(ax, name='province', facecolor='none',
     """
 
     # map name
-    names = {'nation': "bou1_4p", 'province': "bou2_4p",
-            'county': "BOUNT_poly", 'river': "hyd1_4l",
-            'river_high': "hyd2_4l"}
-
-    # get shape filename
-    shpfile = "../common/resources/maps/" + names[name] + ".shp"
-
-    # add map
-    ax.add_geometries(
-        Reader(shpfile).geometries(), ccrs.PlateCarree(),
-        path_effects=[path_effects.Stroke(linewidth=lw+1, foreground=[0.8,0.8,0.8]),path_effects.Normal()],
-        facecolor=facecolor, edgecolor=edgecolor, lw=lw, **kwargs)
+    names = {'nation': "BOUL_G", 'province': "BOUL_S",
+            'city': "BOUL_D", 'county': "BOUL_X",}
+
+    # add 省界
+    if transform is None:
+        transform = ccrs.PlateCarree()
+
+    if name == 'county':
+        # 添加县边界
+        shpfile = MAP_PATH + os.sep + names[name] + ".shp"
+        ax.add_geometries(
+            Reader(shpfile).geometries(), transform,
+            path_effects=[path_effects.Stroke(linewidth=lw, foreground=[1,1,1]),path_effects.Normal()],
+            facecolor=facecolor, edgecolor=edgecolor, lw=lw-0.5, **kwargs)
+
+    if name == 'city':
+        # 添加市边界
+        shpfile = MAP_PATH + os.sep + names[name] + ".shp"
+        ax.add_geometries(
+            Reader(shpfile).geometries(), transform,
+            path_effects=[path_effects.Stroke(linewidth=lw+0.2, foreground=[1,1,1]),path_effects.Normal()],
+            facecolor=facecolor, edgecolor=edgecolor, lw=lw-0.2, **kwargs)
+        
+    # 添加省边界
+    if name == 'province':
+        shpfile = MAP_PATH + os.sep + names[name] + ".shp"
+        ax.add_geometries(
+            Reader(shpfile).geometries(), transform,
+            path_effects=[path_effects.Stroke(linewidth=lw+0.5, foreground=[1,1,1]),path_effects.Normal()],
+            facecolor=facecolor, edgecolor=edgecolor, lw=lw+0.2, **kwargs)
         
 def draw_gisinfo(ax,slat,nlat,wlon,elon):
 
     # 添加中文地名
 
-    filename = '/Users/wenjianzhu/Library/CloudStorage/OneDrive-个人/PythonCode/MyWork/nmc_met_radar/resources/中文地理信息原始文件/cma_city_station_info.xlsx'
+    city_file = STATION_FILE
   
-    data_guojia = pd.read_excel(filename, 
-                        header=0,skiprows=0,index_col=None,
-                        )
+    cities = pd.read_csv(city_file,  delimiter=r"\s+") # cma_city_station_info
    
     gis_lats=[]
     gis_lons=[]
     gis_name=[]
-    for ng in range(len(data_guojia['经度'])):
-        gis_lons.append(data_guojia['经度'][ng])
-        gis_lats.append(data_guojia['纬度'][ng])
-        gis_name.append(data_guojia['城市名'][ng])
+    for ng in range(len(cities['lon'])):
+        gis_lons.append(cities['lon'][ng])
+        gis_lats.append(cities['lat'][ng])
+        gis_name.append(cities['city_name'][ng])
     
 
-    fontname='/Users/wenjianzhu/Library/CloudStorage/OneDrive-个人/PythonCode/MyWork/nmc_met_radar/resources/fonts/msyhbd.ttc'
-    font2=FontProperties(fname=fontname, size=8)
+   
+    font2=FontProperties(fname=FONT_FILE, size=8)
     geodetic_transform = ccrs.Geodetic()._as_mpl_transform(ax)
     text_transform = offset_copy(geodetic_transform, units='dots', x=-5)
     aa = (np.array(gis_lats)<nlat) & (np.array(gis_lats) > slat)
@@ -127,14 +155,7 @@ def quick_draw_mosaic_cross(grid:pyart.core.Grid,params):
     vmin = 0
     vmax = 75
 
-    cmapname = 'pyart_NWSRef'
-
-    #
-    #cmapname = 'pyart_NWSVel'
-    #cmapname = 'pyart_NWS_SPW'
-    #cmapname = 'pyart_RefDiff'
-    #cmapname = 'pyart_NWS_SPW'
-    #cmapname = 'pyart_NWS_SPW'
+    cmapname = 'NWSRef'
 
     # Panel 1: PPI plot of the second tilt.
     ax1 = fig.add_axes(map_panel_axes, projection=projection)
@@ -168,17 +189,6 @@ def quick_draw_mosaic_cross(grid:pyart.core.Grid,params):
     plt.close()
 
 
-
-def draw_topo(ax=None):
-    # 读取地形数据
-    filename = '/Users/wenjianzhu/Downloads/srtm_southchina.nc'
-    topo = xr.open_dataset(filename)
-    # ax = plt.axes()
-    if ax is None:
-        ax = plt.gca()
-    ax.contour(topo['longitude'].values,topo['latitude'].values,topo['topo'],levels=[250,500,750,1000],colors='gray')
-
-    pass
 def draw_composite_operational(gridfile,outpath='.',level=4,vmin=0,vmax=75,colorfile=None):
 
     if not os.path.exists(gridfile):
@@ -192,15 +202,14 @@ def draw_composite_operational(gridfile,outpath='.',level=4,vmin=0,vmax=75,color
     if not os.path.exists(outpath):
         os.makedirs(outpath)
 
-    fontname='/Users/wenjianzhu/Library/CloudStorage/OneDrive-个人/PythonCode/MyWork/nmc_met_radar/resources/fonts/msyhbd.ttc'
     (filepath,filename) = os.path.split(gridfile)
     tstr = filename[7:19]
     tt = datetime.strptime(tstr,'%Y%m%d%H%M') + timedelta(hours=8)
 
     tstr_date = '%04d年%02d月%02d日'%(tt.year,tt.month,tt.day)
     tstr_time = '%02d时%02d分（北京时）'%(tt.hour,tt.minute)
-    font_large=FontProperties(fname=fontname, size=14)
-    font_mid=FontProperties(fname=fontname, size=10)
+    font_large=FontProperties(fname=FONT_FILE, size=14)
+    font_mid=FontProperties(fname=FONT_FILE, size=10)
 
     min_ref=10
     am = grid.fields['reflectivity']['data'].mask
@@ -248,29 +257,23 @@ def draw_composite_operational(gridfile,outpath='.',level=4,vmin=0,vmax=75,color
     
     field = 'reflectivity'
     if colorfile is None:
-        cmapname = 'pyart_NWSRef'
+        cmapname = 'NWSRef'
         cmap = common.parse_cmap(cmapname, field)
     else:
-        # colorfile='gr2_colors/default_BR_PUP2.pal'
-        # colorfile = 'gr2_colors/BR_WDTB_Bright.pal'
-
-        outdic= parse_pro(colorfile)
-        cmap=outdic['cmap']
-        norm=outdic['norm']
+        cmap,norm=parse(colorfile)
+        # outdic= parse(colorfile)
+        # cmap=outdic['cmap']
+        # norm=outdic['norm']
 
     display.plot_grid(field, level, vmin=vmin, vmax=vmax,
                     ax=ax_main,colorbar_flag=False,title_flag=False,
                     projection=projection,axislabels_flag=False,title=None,
                     cmap=cmap,norm=norm)
-    
-    # add_china_map_2cartopy(ax_main, name='province', facecolor='none',edgecolor=None, lw=1)
+    add_china_map_2cartopy(ax_main, name='province', facecolor='none',edgecolor=None, lw=1)
     draw_gisinfo(ax_main,slat=ylim_south,nlat=ylim_north,wlon=xlim_west,elon=xlim_east)
-    # draw_topo()
-    # point_lat = 30.72
-    # point_lon = 108.57
-    # ax_main.plot(point_lon,point_lat,marker='o',color='k',markersize=6,transform=ccrs.PlateCarree())
-    infostr = '中央气象台-天气预报技术研发室 制作(试验产品)'
-    # infostr = '贵州省气象台制作（试验产品）'
+    
+    infostr = '基于 metradar 制作'
+
     fig.text(0.05,0.03, infostr, va="center", ha="left", font=font_mid ) 
 
     ax_main.set_ylim([ylim_south, ylim_north])
@@ -319,20 +322,20 @@ def draw_composite_operational(gridfile,outpath='.',level=4,vmin=0,vmax=75,color
     plt.close()
     # plt.show()
     # create axis
-
+    print('save figure to ' + outpath + os.sep + filename.replace('.nc','.png'))
     pass
 
 if __name__ == "__main__":
 
     
-    path = '/Users/wenjianzhu/Downloads/20230727重庆/FMT/mosaic_pyart'
-    outpath = '/Users/wenjianzhu/Downloads/20230727重庆/FMT/mosaic_3d_pic'
+    path = '/mnt/e/metradar_test/vpr/mosaic/20230731_daxing'
+    outpath = '/mnt/e/metradar_test/vpr/mosaic/20230731_daxing/mosac_pic'
 
-    colorfile='/Users/wenjianzhu/Library/CloudStorage/OneDrive-个人/PythonCode/MyWork/nmc_met_radar/resources/gr2_colors/default_BR_PUP2.pal'
+    print(os.listdir(path))
     for file in os.listdir(path):
         if not file.endswith('.nc'):
             continue
-        draw_composite_operational(path + os.sep + file,outpath,colorfile=colorfile)
+        draw_composite_operational(path + os.sep + file,outpath,colorfile=None)
         pass