metradar 0.1.0__py3-none-any.whl

metradar/main_pydda.py

@@ -0,0 +1,653 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'''
+多普勒雷达三维风场反演脚本
+朱文剑
+2021.4
+'''
+# %%
+
+import pyart
+import pydda
+import os
+import matplotlib.pyplot as plt
+import matplotlib
+from get_tlogp_from_sharppy import get_profile
+import numpy as np
+import configparser
+from datetime import datetime
+import math
+import argparse
+import shutil
+import sys
+from decode_fmt_pyart import read_cnrad_fmt
+# import warnings
+# warnings.filterwarnings("ignore")
+
+
+class GET_WIND_3D:
+    # sub function for reading config file
+    def ConfigFetchError(Exception):
+        pass
+
+    def _get_config_from_rcfile(self,rcfile):
+        """
+        Get configure information from config_dk_met_io.ini file.
+        """
+        # print(os.getcwd())
+        rc = rcfile
+        print(rc)
+        try:
+            config = configparser.ConfigParser()
+            config.read(rc,encoding='UTF-8')
+        except IOError as e:
+            raise self.ConfigFetchError(str(e))
+        except Exception as e:
+            raise self.ConfigFetchError(str(e))
+
+        return config
+
+    def myprint(self,message):
+        f = open('process_info.txt','at+',encoding='utf-8')
+        f.write(message)
+        f.write('\n')
+        f.close()
+        print(message)
+
+    def write_to_progress_cross(self,value):
+        f = open(self.picpath + os.sep + 'progress_cross.txt','wt')
+        f.write(str(value))
+        f.close()
+        print('current progress = %d'%value + "%")
+
+    def write_to_progress_main(self,value):
+        f = open('progress_main.txt','wt')
+        f.write(str(value))
+        f.close()
+        print('current progress = %d'%value + "%")
+
+    def rad(self,d):
+        return d * np.pi / 180.0
+
+    def getDistance(self,lat1, lng1, lat2, lng2):
+        # 计算两点间的距离，单位为千米
+        EARTH_REDIUS = 6378.137
+        radLat1 = self.rad(lat1)
+        radLat2 = self.rad(lat2)
+        a = radLat1 - radLat2
+        b = self.rad(lng1) - self.rad(lng2)
+        s = 2 * math.asin(math.sqrt(math.pow(math.sin(a/2), 2) + math.cos(radLat1) * math.cos(radLat2) * math.pow(math.sin(b/2), 2)))
+        s = s * EARTH_REDIUS
+        return s
+
+    def write_to_gisinfo(self,):
+        try:
+            fout = open(self.picpath + os.sep + 'gisinfo.txt','wt',encoding='utf-8')
+            fout.write('minLat=%.3f\n'%self.lat_min[0])
+            fout.write('maxLat=%.3f\n'%self.lat_max[0])
+            fout.write('minLon=%.3f\n'%self.lon_min[0])
+            fout.write('maxLon=%.3f\n'%self.lon_max[0])
+            self.myprint('minLat=%.3f'%self.lat_min[0])
+            self.myprint('maxLat=%.3f'%self.lat_max[0])
+            self.myprint('minLon=%.3f'%self.lon_min[0])
+            self.myprint('maxLon=%.3f'%self.lon_max[0])
+            fout.close()
+            return True
+        except:
+            self.myprint(self.picpath + os.sep + 'gisinfo.txt' + ' open error!')
+            return False
+
+    def __init__(self,inifile='config.ini'):
+        pass
+        self.error=False
+        if os.path.exists('process_info.txt'):
+            os.remove('process_info.txt')
+
+        if not os.path.exists(inifile):
+            self.myprint(inifile + ' 不存在，请检查')
+        config = self._get_config_from_rcfile(inifile)
+        self.xlims = np.array(config['PARAM_SETTINGS']['XLIM'].split(', ')).astype(np.int64)
+        self.ylims = np.array(config['PARAM_SETTINGS']['YLIM'].split(', ')).astype(np.int64)
+        self.zlims = np.array(config['PARAM_SETTINGS']['ZLIM'].split(', ')).astype(int)
+        self.time_files = config['PARAM_SETTINGS']['TIME_FILES']
+        if len(self.time_files) !=14:
+            self.myprint('时间参数设置错误，请重新设置 TIME_FILES 参数')
+            self.error=True
+            return None
+
+        self.standard_lat = float(config['PARAM_SETTINGS']['CENTER_LAT'])
+        self.standard_lon = float(config['PARAM_SETTINGS']['CENTER_LON'])
+
+        self.ana_lat = float(config['PARAM_SETTINGS']['ANA_LAT'])
+        self.ana_lon = float(config['PARAM_SETTINGS']['ANA_LON'])
+        # bshowpic = int(config['PARAM_SETTINGS']['SHOW_PIC'])
+        self.bsavepic = int(config['PARAM_SETTINGS']['SAVE_PIC'])
+        if self.bsavepic == False:
+            self.myprint('参数设置中设定不保存图片！')
+
+        self.bshowpic = False
+        if not self.bshowpic:
+            matplotlib.use('Agg')
+        else:
+            matplotlib.use('MacOSX')
+
+        self.radars = config['PARAM_SETTINGS']['RADAR_SITES'].split(', ')
+        self.radarfilepath = config['PATH_SETTINGS']['RADAR_FILE_PATH']
+        self.outpath = config['PATH_SETTINGS']['OUT_PATH']
+        self.picpath = config['PATH_SETTINGS']['PIC_PATH']
+        self.tlogp_filepath = config['PATH_SETTINGS']['TLOG_PATH']
+        self.tlogp_filename = config['PATH_SETTINGS']['TLOG_FILE']
+
+        if not os.path.exists(self.outpath):
+            os.makedirs(self.outpath)
+
+        if not os.path.exists(self.picpath):
+            os.makedirs(self.picpath)
+
+        self.xreso = int(config['PARAM_SETTINGS']['XRESO'])
+        self.yreso = int(config['PARAM_SETTINGS']['YRESO'])
+        self.zreso = int(config['PARAM_SETTINGS']['ZRESO'])
+
+        self.lon_min,self.lat_min=pyart.core.cartesian_to_geographic_aeqd(self.xlims[0]*1000,self.ylims[0]*1000,self.standard_lon,self.standard_lat)
+        self.lon_max,self.lat_max=pyart.core.cartesian_to_geographic_aeqd(self.xlims[1]*1000,self.ylims[1]*1000,self.standard_lon,self.standard_lat)
+
+        if not self.write_to_gisinfo():
+            self.write_to_progress_main(100)
+            self.error=True
+            self.myprint('write_to_gisinfo error!')
+            return None
+
+        self.g_xlim = (self.xlims[0]*1000,self.xlims[1]*1000)
+        self.g_ylim = (self.ylims[0]*1000,self.ylims[1]*1000)
+        self.g_zlim = (self.zlims[0]*1000,self.zlims[1]*1000)
+        self.g_numx = int(len(range(self.xlims[0],self.xlims[1]))*1000/self.xreso+1) # x方向格点数
+        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)
+        valid_files = []
+
+        self.write_to_progress_cross(0)
+        for ff in files:
+            if ff.find('.nc')<=0:
+                continue
+            else:
+                valid_files.append(ff)
+
+        if len(valid_files) == 0:
+            self.myprint('未找到nc格式的三维风场数据文件，请先调用--getwind方法')
+            return False
+
+        for ff in valid_files:
+            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)
+        self.myprint('拼图中心纬度为：%.3f'%self.standard_lat)
+        self.myprint('垂直剖面经度为：%.3f'%self.ana_lon)
+        self.myprint('垂直剖面纬度为：%.3f'%self.ana_lat)
+        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]
+
+        x=Grids[0].x['data']
+        y=Grids[0].y['data']
+        z=Grids[0].z['data']
+
+        uwind=Grids[0].fields['u']['data']
+        vwind=Grids[0].fields['v']['data']
+        wwind=Grids[0].fields['w']['data']
+
+        # 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 + '水平流场'
+        picpath_xz_bar = self.picpath + os.sep + 'XZ方向垂直风场'
+        picpath_xz_stream = self.picpath + os.sep + 'XZ方向垂直流场'
+        picpath_yz_bar = self.picpath + os.sep + 'YZ方向垂直风场'
+        picpath_yz_stream = self.picpath + os.sep + 'YZ方向垂直流场'
+
+        picpath_vor_stretch = self.picpath + os.sep + '拉伸涡度'
+        picpath_vor_tilt = self.picpath + os.sep + '倾斜涡度'
+ 
+
+
+        if not os.path.exists(picpath_horiz_bar):
+            os.makedirs(picpath_horiz_bar)
+        if not os.path.exists(picpath_horiz_stream):
+            os.makedirs(picpath_horiz_stream)
+        if not os.path.exists(picpath_xz_bar):
+            os.makedirs(picpath_xz_bar)
+        if not os.path.exists(picpath_xz_stream):
+            os.makedirs(picpath_xz_stream)
+        if not os.path.exists(picpath_yz_bar):
+            os.makedirs(picpath_yz_bar)
+        if not os.path.exists(picpath_yz_stream):
+            os.makedirs(picpath_yz_stream)
+        if not os.path.exists(picpath_vor_stretch):
+            os.makedirs(picpath_vor_stretch)
+        if not os.path.exists(picpath_vor_tilt):
+            os.makedirs(picpath_vor_tilt)
+
+        if bdrawhori:
+            # draw barbs
+            for ll in range(10): # self.g_numz
+                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,
+                                                    barb_spacing_x_km=5,
+                                                    barb_spacing_y_km=5,
+                                                    show_lobes=False)
+                ax.set_xlim(self.xlims.tolist())
+                ax.set_ylim(self.ylims.tolist())
+                ax.set_title(ax.get_title().replace('PyDDA','MultiRadar'))
+                picfilename = 'horiz_xsection_barbs_%02d_%s.png'%(ll+1,self.time_files)
+                if self.bsavepic:
+                    fig.savefig(picpath_horiz_bar + os.sep + picfilename, 
+                                format='png', transparent=False, dpi=300, pad_inches = 0)
+                    self.myprint(picpath_horiz_bar + os.sep + picfilename + ' saved!')
+                plt.close(fig)
+
+            # draw streamline
+            for ll in range(10): # self.g_numz
+   
+                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,
+                                                        #   w_vel_contours=[3, 5,],
+                                                        show_lobes=False,
+                                                        )
+                ax.set_xlim(self.xlims.tolist())
+                ax.set_ylim(self.ylims.tolist())
+                ax.set_title(ax.get_title().replace('PyDDA','MultiRadar'))
+                picfilename = 'horiz_xsection_streamlines_%02d_%s.png'%(ll+1,self.time_files)
+                if self.bsavepic:
+                    fig.savefig(picpath_horiz_stream + os.sep + picfilename, 
+                                format='png', transparent=False, dpi=300, pad_inches = 0)
+                    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,
+                                            barb_spacing_x_km=2.0,
+                                            barb_spacing_z_km=1.0)
+            ax.set_xlim(self.xlims.tolist())
+            ax.set_title(ax.get_title().replace('PyDDA','MultiRadar'))
+            picfilename = 'xz_xsection_barbs_%06d_%06d_%s.png'%(self.ana_lat*1000,self.ana_lon*1000,self.time_files)
+            
+            if self.bsavepic:
+                fig.savefig(picpath_xz_bar + os.sep + picfilename, 
+                            format='png', transparent=False, dpi=300, pad_inches = 0)
+                self.myprint(picpath_xz_bar + os.sep + picfilename + ' saved!')
+            plt.close(fig)
+            self.write_to_progress_cross(25)
+        except:
+            self.myprint('可能是垂直剖面的经纬度超出范围了，请重新设置再尝试！')
+        #=================
+        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,
+                                                #  w_vel_contours=[3, 5,],
+                                                )
+            ax.set_xlim(self.xlims.tolist())
+            ax.set_title(ax.get_title().replace('PyDDA','MultiRadar'))
+            picfilename = 'xz_xsection_streamlines_%06d_%06d_%s.png'%(self.ana_lat*1000,self.ana_lon*1000,self.time_files)
+            if self.bsavepic:
+                fig.savefig(picpath_xz_stream + os.sep + picfilename, 
+                            format='png', transparent=False, dpi=300, pad_inches = 0)
+                self.myprint(picpath_xz_stream + os.sep + picfilename + ' saved!')
+            plt.close(fig)
+            self.write_to_progress_cross(50)
+        except:
+            self.myprint('可能是垂直剖面的经纬度超出范围了，请重新设置再尝试！')
+        #=================
+        try:
+            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,
+                                            barb_spacing_y_km=2,
+                                            barb_spacing_z_km=1)
+            ax.set_xlim(self.ylims.tolist())
+            ax.set_title(ax.get_title().replace('PyDDA','MultiRadar'))
+            picfilename = 'yz_xsection_barbs_%06d_%06d_%s.png'%(self.ana_lat*1000,self.ana_lon*1000,self.time_files)
+            if self.bsavepic:
+                fig.savefig(picpath_yz_bar + os.sep + picfilename, 
+                            format='png', transparent=False, dpi=300, pad_inches = 0)
+                self.myprint(picpath_yz_bar + os.sep + picfilename + ' saved!')
+            plt.close(fig)
+            self.write_to_progress_cross(75)
+        except:
+            self.myprint('可能是垂直剖面的经纬度超出范围了，请重新设置再尝试！')
+        #=================
+        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,
+                                                #  w_vel_contours=[3, 5, ],
+                                                )
+            ax.set_xlim(self.ylims.tolist())
+            ax.set_title(ax.get_title().replace('PyDDA','MultiRadar'))
+            picfilename = 'yz_xsection_streamlines_%06d_%06d_%s.png'%(self.ana_lat*1000,self.ana_lon*1000,self.time_files)
+            if self.bsavepic:
+                fig.savefig(picpath_yz_stream + os.sep + picfilename, 
+                            format='png', transparent=False, dpi=300, pad_inches = 0)
+                self.myprint(picpath_yz_stream + os.sep + picfilename + ' saved!')
+            plt.close(fig)
+            self.write_to_progress_cross(100)
+        except:
+            self.myprint('可能是垂直剖面的经纬度超出范围了，请重新设置再尝试！')
+        return True
+
+    def do_wind_retrievel(self):
+
+        self.write_to_progress_main(0)
+        
+        filepaths = []
+        filenames = []
+        # print('kevin')
+        # import pdb; pdb.set_trace()
+        for rd in self.radars:
+            curpath = self.radarfilepath + os.sep + rd
+            filepaths.append(curpath)
+            files = os.listdir(curpath)
+            files = sorted(files)
+
+            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)
+                self.write_to_progress_main(100)
+                return False
+
+            ct = datetime.strptime(self.time_files, '%Y%m%d%H%M%S').timestamp()
+            cidx = np.where(abs(np.array(times) - ct)==min(abs(np.array(times)-ct)))[0][0]
+            if min(abs(np.array(times)-ct)) > 180:
+                self.myprint('查找到 %s 雷达的资料中离规定时间最近的文件的观测时间与规定时间的差超过了180秒，请知悉！'%rd)
+            self.myprint('找到的 %s 雷达的资料中与规定时间最接近的文件名为 %s '%(rd,files[cidx]))
+            filenames.append(files[cidx])
+
+        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'))
+
+        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]))
+
+            self.myprint('read %s over!'%filenames[nn])
+
+            grid_objs.append(pyart.map.grid_from_radars(
+            (radar_objs[nn],),
+            grid_origin=[self.standard_lat,self.standard_lon],
+            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']))
+            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=============风场反演
+
+        # 添加探空数据
+        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!')
+            self.write_to_progress_main(100)
+            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')
+        # u_init, v_init, w_init = None,None,None
+        self.myprint('探空数据 %s 加载完毕！'%self.tlogp_filename)
+        # 速度退模糊暂时空缺，后期补上
+
+
+        #进行多雷达三维风场反演
+        # 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
+        # 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])
+
+        # exclude masked gates from the gridding
+        # gatefilter = pyart.filters.GateFilter(Grids)
+        # gatefilter.exclude_masked('reflectivity')
+
+        self.myprint('write data to nc over!')
+        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)
+        # if bshowpic:
+        #     plt.show()
+
+        if os.path.exists('process_info.txt'):
+            shutil.copyfile('process_info.txt',self.picpath + os.sep + 'process_info.txt')
+
+    #计算拉伸涡度和倾斜涡度
+    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)
+        
+        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)
+            
+            div=dudx[lev,:,:] + dvdy[lev,:,:]
+            vor=dvdx[lev,:,:] - dudy[lev,:,:]
+            
+            u=uwind[lev,:,:]
+            v=vwind[lev,:,:]
+            #dw
+            
+            #拉伸项
+            vor_stretch_data[lev,:,:]=vor*div
+
+            #倾斜项
+            vor_tilt_data[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)
+            
+
+            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
+    
+
+if __name__ == "__main__":
+    '''
+    有两个函数可以调用：
+    python main_pydda.py config_3dwind.ini --getwind
+    python main_pydda.py config_3dwind.ini --drawcross
+    其中第一个用来计算并绘图，第二个用来单独绘制任意点的垂直剖面
+    '''
+    with open('current_pid.txt','wt') as f:
+        f.write('%s'%str(os.getpid()))
+    def mainprog(inifile):
+
+        _getwind = GET_WIND_3D(inifile) 
+        if not _getwind.error:
+            _getwind.do_wind_retrievel()
+            _getwind.draw_all_pic()
+
+        else:
+            print('GET_WIND_3D Init error!')
+            sys.exit(-1)
+    
+    def drawpic_cross(inifile):
+        _getwind = GET_WIND_3D(inifile)
+        if not _getwind.error:
+            _getwind.draw_all_pic(bdrawhori=False)
+        else:
+            print('GET_WIND_3D Init error!')
+            sys.exit(-1)
+
+    parser = argparse.ArgumentParser(description='Do radar 3d wind retrievel.')
+    parser.add_argument('inifile', metavar='inifile', type=str, 
+                    help='ini filename for the mainprog')
+    
+    parser.add_argument('--getwind', dest='mainprog', action='store_const', 
+                        const=mainprog,
+                        help='do 3d wind retrievel')
+    
+    parser.add_argument('--drawcross', dest='drawpic_cross', action='store_const', 
+                        const=drawpic_cross,
+                        help='draw XZ and YZ pic')
+    BDEBUG=True
+
+    if BDEBUG:
+        print('当前为debug模式，所以直接读取主目录下的config.ini, 请注意！')
+        mainprog('metradar/config_3dwind.ini')
+        
+    else:
+        args = parser.parse_args()
+        if not args.mainprog is None:
+            args.mainprog(args.inifile)
+
+        if not args.drawpic_cross is None:
+            args.drawpic_cross(args.inifile)
+
+
+# %%

metradar/make_gif.py

@@ -0,0 +1,24 @@
+
+
+from PIL import Image
+import os
+
+
+filepath = 'pic'
+outpath = 'pic'
+filenames = os.listdir(filepath)
+filenames = sorted(filenames)
+images = []
+
+
+
+for filename in filenames:
+    if filename.startswith('.') or filename.startswith('..'):
+        continue
+    if not filename.endswith('.png'):
+        continue
+
+    images.append(Image.open(filepath + os.sep + filename))
+    print(filename + ' added!')
+images[0].save(outpath + os.sep + '雷达外推示例.gif', format='GIF',
+                    append_images=images[1:], save_all=True, duration=int(500), loop=0)