metradar 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

metradar/trans_polor2grid_func.py

@@ -1,203 +0,0 @@
-'''
-本程序将极坐标格式的ppi数据转换为格点数据，并保存为nc格式
-
-'''
-
-# %%
-import numpy as np
-import xarray as xr
-from decode_fmt_pyart import read_cnrad_fmt
-import os
-import time
-import shutil
-
-def do_ppi(params):
-
-    # sweep_slice,radar,outname
-    sweep_slice = params['sweep_slice']
-    target_reso = params['out_reso']
-    # print(sweep_slice)
-    radar = params['radar']
-
-    ori_reso = radar.range['meters_between_gates']
-    # target_reso = 500
-    reso_list = list(np.arange(0,11,1)*ori_reso)
-    # 从reso_list中找到最接近target_reso的值
-    out_scale = np.argmin(np.abs(np.array(reso_list)-target_reso))
-    # out_scale = int(target_reso/ori_reso)
-    target_reso = ori_reso * out_scale
-
-    # sweep_slice = slice(0,384,None)
-    sweep_azimuths = radar.azimuth['data'][sweep_slice]
-    elv = np.mean(radar.elevation['data'][sweep_slice])
-
-    # 输出的距离库只保留一半
-    ngates = int(len(radar.gate_x['data'][sweep_slice][0])/2) #//2*2 
-    grid_x = np.arange(-1*ngates,ngates+1,out_scale)
-    grid_y = np.arange(-1*ngates,ngates+1,out_scale)
-    total_gates = len(grid_x)
-
-    aa = np.meshgrid(grid_x,grid_y)
-    azi_grid = np.arctan2(aa[0],aa[1])*180/np.pi
-    azi_grid[azi_grid<0]+=360
-    
-    azi_reso = 360/len(sweep_azimuths)
-
-    # 求方位角索引
-    new_azi = azi_grid.flatten()
-    t = new_azi-sweep_azimuths[0]
-    t[t<0]+=360
-    ray_number = np.round(t/azi_reso,0).astype(int)
-    ray_number[ray_number==len(sweep_azimuths)]=0
-    ray_number = np.reshape(ray_number,(total_gates,total_gates))
-
-
-    # 求距离索引
-    dis_grid = np.sqrt(aa[0]**2 + aa[1]**2)
-    dis_grid = np.round(dis_grid.flatten(),0).astype(int)
-    dis_grid = np.reshape(dis_grid,(total_gates,total_gates))
-
-  
-  
-    sweepdata = radar.fields['reflectivity']['data'][sweep_slice]
-    # 对数据进行截断，在径向方向上
-    sweepdata = sweepdata[:,0:ngates]
-    # if np.max(sweepdata.data.flatten()) < -30:
-    #     # 表示该层无有效数据
-    #     return None
-    # data_grid = np.zeros((total_gates,total_gates)) - 9999
-    data_grid = np.zeros((total_gates,total_gates))
-    new_data = data_grid.flatten()
-    new_spdata = sweepdata.flatten()
-
-    pos_out = [i+j*total_gates for i in range(total_gates) for j in range(total_gates) if dis_grid[i,j] < ngates]
-    d_out = [rn*ngates+dg for rn,dg in zip(ray_number.flatten(),dis_grid.flatten()) if dg < ngates]
-   
-    new_data[pos_out]=new_spdata[d_out]
-    new_data[new_data<-10]=np.nan
-    new_data[new_data>=95]=np.nan
-    new_data = new_data *2 + 64
-    new_data[np.isnan(new_data)]=255
-    new_data = new_data.astype(np.uint8)
-    data_grid = np.reshape(new_data,(total_gates,total_gates))
-    outdic={'data':data_grid.transpose(),'elv':elv,'tol_gates':total_gates,
-            'out_reso':target_reso,'datatype':'uint8','scale_factor':2,'add_offset':64,
-            'missing_value':255}
-    print('ele: ' + str(elv) + ' done!')
-
-    del data_grid,new_data,new_spdata,pos_out,d_out,sweepdata
-    return outdic
-
-def do_radar(param):
-    pass
-    try:
-        filename = param['filename']
-        outname = param['outname']
-        out_reso = param['out_reso']
-        if 'bcopy' in param.keys():
-            bcopy = param['bcopy']
-        else:
-            bcopy = 0
-        if 'outpath2' in param.keys():
-            outpath2 = param['outpath2']
-        else:
-            outpath2 = ''
-            print('outpath2 not set!')
-        print('start trans ' + filename)
-
-        # 读取雷达数据
-        radar = read_cnrad_fmt(filename)
-
-        if 'reflectivity' in radar.fields.keys():
-            print('ref in data!')
-
-
-        start  = time.time()
-
-        all_data = []
-        elvs = []
-        
-        nnf=0
-        pre_ele=-999
-        for sweep_slice in radar.iter_slice():
-            # if nnf ==2:
-            #     break
-            params={'sweep_slice':sweep_slice,'radar':radar,'out_reso':out_reso,}
-            cur_ele = np.mean(radar.elevation['data'][sweep_slice])
-            if abs(cur_ele - pre_ele) < 0.2:
-                # print('该仰角已处理过！')
-                # 实际上是跳过了最低仰角的两层径向速度
-                continue
-
-            outdic = do_ppi(params)
-            if outdic is None:
-                continue
-            all_data.append(outdic['data'])
-            elvs.append(outdic['elv'])
-            pre_ele = cur_ele
-            nnf +=1
-        end = time.time()
-       
-
-        #构建xarray数据
-        total_gates = outdic['tol_gates']
-        if total_gates % 2 == 0:
-            out_grid = np.arange(int(-total_gates/2),int(total_gates/2))
-        else:
-            out_grid = np.arange(int(-(total_gates-1)/2),int((total_gates-1)/2)+1)
-        
-        if len(all_data) == 0:
-            print('no data in ' + filename)
-
-            return
-        all_data = np.array(all_data)
-        all_data = all_data.astype(np.uint8)
-        data = xr.DataArray(np.array(all_data),coords=[elvs,out_grid,out_grid],dims=['z','x','y'],name='ref')
-        data.attrs['units'] = 'dBZ'
-        data.attrs['standard_name'] = 'equivalent_reflectivity_factor'
-        data.attrs['long_name'] = 'equivalent_reflectivity_factor'
-        data.attrs['radar_lat'] = radar.latitude['data'][0]
-        data.attrs['radar_lon'] = radar.longitude['data'][0]
-        data.attrs['radar_altitude'] = radar.altitude['data'][0]
-        data.attrs['grid_num'] = total_gates
-        data.attrs['grid_reso'] = outdic['out_reso']
-        data.attrs['elevation'] = elvs
-        data.attrs['obs_range'] = int(outdic['out_reso'] * (total_gates-1)/2)
-        data.attrs['distance_unit'] = 'meter'
-        data.attrs['missing_value'] = 255
-        data.attrs['scale_factor'] = 2
-        data.attrs['add_offset'] = 64
-        data.attrs['datatype'] = 'uint8'
-        data.attrs['decode_method'] = 'dbz = (data - add_offset) / scale_factor'
-
-        if not os.path.exists(os.path.dirname(outname)):
-            os.makedirs(os.path.dirname(outname))
-
-        data.to_netcdf(outname,encoding={'ref': {'zlib': True,'dtype':'uint8'},})
-        print(outname + ' saved!')
-      
-        # copy to outpath2
-        if bcopy == 1:
-            if not os.path.exists(outpath2):
-                os.makedirs(outpath2)
-            shutil.copy(outname,outpath2 + os.sep + os.path.basename(outname))
-            print(outpath2 + os.sep + os.path.basename(outname) + ' copied!')
-         
-
-    except Exception as e:
-        print(e)
-    
-if __name__ == "__main__":
-    rootpath = 'testdata/FMT/'
-    filename = 'Z_RADR_I_Z9200_20240831130000_O_DOR_SAD_CAP_FMT.bin.bz2'
-
-    outname = '/Users/wenjianzhu/Downloads/polargrid/' + filename + '_new.nc'
-    params={'filename':rootpath + filename,'outname':outname,'out_reso':500,'bcopy':0}
-
-    
-    st = time.time()
-    do_radar(params)
-    et = time.time()
-    print(et-st)
-
-    

metradar-0.1.2.dist-info/METADATA

@@ -1,18 +0,0 @@
-Metadata-Version: 2.2
-Name: metradar
-Version: 0.1.2
-Summary: radar data processing
-Author-email: Wenjian Zhu <kevin2075@163.com>
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-Requires-Dist: numpy>=1.18
-Requires-Dist: requests>=2.24.0
-Requires-Dist: arm_pyart>=1.19.5
-Requires-Dist: cartopy>=0.24.0
-Requires-Dist: openpyxl>=3.1.5
-Requires-Dist: pydda==1.2.0
-Requires-Dist: tqdm>=4.67.0
-Requires-Dist: wradlib>=2.4.0