"""
WRF后处理程序 slp
20230116 byp
"""
import netCDF4
import wrf
import os
import glob
import re


# 读取数据###############################################################开始
dir_name = "./data/WRF/"
dir_output = './output/image/'
list_of_files = glob.glob(dir_name+'*')
latest_file = max(list_of_files, key=os.path.getctime)  # 最新文件
# print(latest_file)
mat = re.search(r"(\d{4}-\d{1,2}-\d{1,2}_\d{1,2})", latest_file)  # 正则匹配时间
dir_time_file = 'wrfout_d01_'+latest_file[mat.span()[0]:]
dir_time = latest_file[mat.span()[0]:mat.span()[1]].replace('-', '').replace('_', '')
data_nc = netCDF4.Dataset(dir_name+dir_time_file)

u = wrf.getvar(data_nc, 'U10', timeidx=1)
# u = u.values[1][10]
# print(u.coords.values())
v = wrf.getvar(data_nc, 'V10', timeidx=1)
# print(v.values)

# Get the lat/lon coordinates
lats, lons = wrf.latlon_coords(u)
print(lons.values[0:1][0][0])
print(lats.values[0:1][0][0])
print(u.values[0:1][0][0])
print(v.values[0:1][0][0])