shancx 1.8.92__py3-none-any.whl → 1.9.33.218__py3-none-any.whl

shancx/Plot/draw_day_CR_SVG.py

@@ -1,275 +0,0 @@
-import matplotlib.pyplot as plt
-import matplotlib 
-matplotlib.use("svg")
-import datetime
-from hjnwtx.colormap import cmp_hjnwtx
-import time
-import os
-import time
-import shutil
- 
-from dateutil.relativedelta import relativedelta
-import glob
-from config import logger
-import argparse
-import pandas as pd 
-import re
-import netCDF4 as nc 
-from multiprocessing import Pool
-from itertools import product
-import numpy as np
-import copy
-import traceback
-
- 
-from multiprocessing import Process, Queue
- 
-import io
-import cartopy.crs as ccrs
-import cartopy.feature as cfeature
-import cartopy.io.shapereader as shpreader
-from shancx import crDir
-
-import matplotlib.pyplot as plt
-import matplotlib.cbook as cbook
-import matplotlib.image as image
-import matplotlib.cm as cm
-import matplotlib.colors as colors
-import io
-import datetime
-import numpy as np
-from multiprocessing import Pool
-import cartopy.crs as ccrs
-import cartopy.feature as cfeature
-import cartopy.io.shapereader as shpreader
-import matplotlib.pyplot as plt
-import matplotlib.cm as cm
-import datetime
-import numpy as np
-from multiprocessing import Pool
-import io
-import os
-from shancx import crDir
-
-# Assuming these are defined somewhere else in your code
-# cmp_hjnwtx = {"radar_nmc": your_cmap_definition}
-# shpreader = your_shpreader_definition
-# cfeature = your_cfeature_definition
-# ccrs = your_ccrs_definition
-
-
-import matplotlib.pyplot as plt
-import datetime
-import os
-import io
-from multiprocessing import Pool
-
-# Assuming these are defined somewhere else in your code
-# cmp_hjnwtx = {"radar_nmc": your_cmap_definition}
-# shpreader = your_shpreader_definition
-# cfeature = your_cfeature_definition
-# ccrs = your_ccrs_definition
-import matplotlib.pyplot as plt
-import datetime
-import os
-import io
-from multiprocessing import Pool
-
-
-import matplotlib.pyplot as plt
-import datetime
-import os
-import io
-from multiprocessing import Pool
-import cairosvg  # Install using pip if not already installed
-
-
-# Assuming these are defined somewhere else in your code
-# cmp_hjnwtx = {"radar_nmc": your_cmap_definition}
-# shpreader = your_shpreader_definition
-# cfeature = your_cfeature_definition
-# ccrs = your_ccrs_definition
-def add_china_map(ax):
-    # Add terrain features on the map
-    ax.add_feature(cfeature.COASTLINE, edgecolor='gray')
-    ax.add_feature(cfeature.BORDERS, linestyle=':', edgecolor='gray')
-    ax.add_feature(cfeature.LAKES, alpha=0.8)
-    
-    # Add province outlines
-    provinces = shpreader.natural_earth(resolution='10m', category='cultural', name='admin_1_states_provinces')
-    provinces_features = shpreader.Reader(provinces).geometries()
-    ax.add_geometries(provinces_features, ccrs.PlateCarree(), facecolor='none', edgecolor='gray', linestyle=':', linewidth=0.5, alpha=0.8)
-
-def draw_subplot(args):
-    index, tp, vmax, vmin, cmap, time_index, name = args
-
-    fig, ax = plt.subplots(figsize=(10, 10), subplot_kw={'projection': ccrs.PlateCarree()})
-
-    ax.set_extent([73, 135, 18, 54], ccrs.PlateCarree())  # Adjust as needed
-
-    add_china_map(ax)
-
-    # Plot the data
-    img = ax.imshow(tp, vmin=vmin, vmax=vmax, cmap=cmap, extent=[73, 134.99, 12.21, 54.2], transform=ccrs.PlateCarree(), alpha=1)
-    
-    # Turn off axis
-    ax.axis('off')
-
-    # Add text annotations
-    ax.text(0.95, 0.95, f'{time_index}', transform=ax.transAxes, color='white', fontsize=20, ha='right', va='bottom')
-    ax.text(0.925, 0.925, f'{str(index)}', transform=ax.transAxes, color='white', fontsize=20, ha='right', va='top')
-
-    fig.tight_layout()
-
-    # Save the figure to a buffer as SVG
-    buf = io.BytesIO()
-    plt.savefig(buf, format='svg', bbox_inches='tight')
-    buf.seek(0)
-    plt.close(fig)
-
-    return (index, buf)
-
-def drawpic(tp, Count, timeList, name="temp"):
-    vmax = 70
-    vmin = 0
-    cmap = cmp_hjnwtx["radar_nmc"]
-
-    # Create arguments list for multiprocessing
-    args_list = [(index, tp[index, :, :], vmax, vmin, cmap, timeList[index], name) for index in range(2 * Count)]
-
-    with Pool(31) as pool:
-        results = pool.map(draw_subplot, args_list)
-
-    # Convert SVG data to PNG using cairosvg
-    png_buffers = []
-    for result in results:
-        svg_data = result[1].getvalue().decode('utf-8')
-        png_bytes = cairosvg.svg2png(bytestring=svg_data.encode('utf-8'))
-        png_buffers.append(io.BytesIO(png_bytes))
-
-    # Create a grid of subplots
-    combined_fig, axs = plt.subplots(2, Count, figsize=(10 * Count, 20))
-
-    for i in range(2):
-        for j in range(Count):
-            index = i * Count + j
-            ax = axs[i, j]
-            # Use the PNG data directly for display
-            png_buf = png_buffers[index]
-            png_buf.seek(0)
-            png_data = plt.imread(png_buf)
-            ax.imshow(png_data, cmap=cmap)  # Display PNG data
-            ax.axis('off')
-
-    # Save the combined figure as SVG
-    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M")
-    outdir = f"./{timeList[0].strftime('%Y%m%d%H%M')}_CR"
-    crDir(outdir)
-    outpath = os.path.join(outdir, f"{name}_{now_str}.svg")
-    
-    plt.tight_layout()
-    plt.savefig(outpath, format='svg', bbox_inches='tight')
-    plt.close(combined_fig)
-
-
- 
-BASE_TARGET_PATH = "/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF"
-
-
-def get_mqpf_paths(UCTstr):
-    year = UCTstr[:4]
-    date = UCTstr[:8]
-    mqpfPath_pattern = os.path.join(BASE_TARGET_PATH,year, date,f"MSP2_WTX_AIW_REF_L88_CHN_{UCTstr}_00000-00300-00006.nc")
-    # mqpfPath_pattern = os.path.join(BASE_TARGET_PATH,year, date,f"MSP2_WTX_AIW_MQPF_L88_CHN_{UCTstr}_00000-00300-00006.nc")
- 
-    return mqpfPath_pattern  
-
-def map_data(conf):
-    CST = conf[0]
-    UCT = CST + relativedelta(hours=-8)
-    UCTstr = UCT.strftime("%Y%m%d%H%M")
-    mqpfPath_pattern = get_mqpf_paths(UCTstr)
-    mqpfPath_list = glob.glob(mqpfPath_pattern)
-    if len(mqpfPath_list) == 0:
-        data_loss = re.findall(r"(2024\d{8}?)",mqpfPath_pattern)
-        print("data_loss",data_loss)
-        print("data_loss_path",mqpfPath_pattern)
-        tj_list.append(data_loss[0]) 
-        return np.full((1, 4200, 6200), np.nan)
-    else:
-        try:
-            with nc.Dataset(mqpfPath_list[0]) as dataNC:
-                mqpf = dataNC.variables["CR"][:]
-                mqpf = mqpf[:1]
-                if mqpf.shape != (1, 4200, 6200):
-                    logger.info(mqpf.shape )
-                    logger.info(mqpfPath_list[0])
-                    print("mqpf shape error",mqpf.shape)
-                    mqpf = mqpf[:,:-1, :-1]
-                    if mqpf.shape != (1, 4200, 6200):
-                       return np.full((1, 4200, 6200), np.nan)      
-                    else:
-                        return mqpf          
-                print("mqpf",UCTstr,mqpf.shape)
-            tj_list1.append(mqpfPath_pattern)
-            return mqpf
-        except Exception as e:
-            print(traceback.format_exc())
-            return np.full((1, 4200, 6200), np.nan)
-
-def options():
-    parser = argparse.ArgumentParser(description='draw CR')
-    # parser.add_argument('--times', type=str, default='202406290000,202406300000')
-    parser.add_argument('--times', type=str, default='202407220000,202407230000')
-    parser.add_argument('--pac', type=str, default='100000')
-    # parser.add_argument('--combine', action='store_true', default=False)
-    parser.add_argument('--combine',action='store_true',default=False)
-    parser.add_argument('--isDebug',action='store_true',default=False)
-    parser.add_argument('--isDraw',action='store_true',default=False)
-    parser.add_argument('--freq', type=str, default="1h")
-    parser.add_argument('--tag',type=str, default=datetime.datetime.now().strftime("%Y%m%d%H%M"))
-    config= parser.parse_args()
-    print(config)
-    config.times = config.times.split(",")
-    config.pac = config.pac.split(",")
-    if len(config.times) == 1:
-        config.times = [config.times[0], config.times[0]]
-    config.times = [datetime.datetime.strptime(config.times[0], "%Y%m%d%H%M"),
-                    datetime.datetime.strptime(config.times[1], "%Y%m%d%H%M")]
-    return config
-if __name__ == '__main__':
-    cfg = options()
-    sCST = cfg.times[0]
-    eCST = cfg.times[-1]
-    sCSTstr = sCST.strftime("%Y%m%d")
-    tj_list = []
-    tj_list1= []
-    start = datetime.datetime.now()
-    timeList = pd.date_range(sCST,eCST,freq="360s",inclusive="left")
-    # for CST in timeList:  
-    productList = product(timeList)
-    with Pool(31) as p:
-         Data = p.map(map_data,productList)
-    end = datetime.datetime.now()
-    print(start-end)
-    Data_con = np.concatenate(Data,axis=0) 
-    loss_len = 240 - Data_con.shape[0]
-    sCSTstr = sCST.strftime("%Y%m%d")
-    eCSTstr = eCST.strftime("%Y%m%d")
-    # Data_con1 = Data_con.filled()  
-    # np.save(f"data_{sCSTstr}_{eCSTstr}.npy",Data_con1) 
-    Data_con_120 = copy.copy(Data_con)
-    drawpic(Data_con_120[:int(len(Data_con_120)/2)], int(len(Data_con_120)/4),timeList[:int(len(Data_con_120)/2)],name=f"temp120_{sCSTstr}_loss_{loss_len}_")
-    print("done 120")
-    end1 = datetime.datetime.now()
-    print(end1-end)
-    Data_con_240 = copy.copy(Data_con)
-    print(Data_con_240[120:].shape)
-    drawpic(Data_con_120[int(len(Data_con_120)/2):], int(len(Data_con_120)/4),timeList[int(len(Data_con_120)/2):],name=f"temp120_240_{sCSTstr}_{loss_len}_")
-    print(datetime.datetime.now()-end1)
-    print("done 120-240") 
-    logger.info("success")
- 
-
-#  "/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF/2024/20240704/MSP2_WTX_AIW_REF_L88_CHN_202407040324_00000-00300-00006.nc"
-

shancx/Plot/draw_day_pre_PNGUS.py

@@ -1,205 +0,0 @@
-import matplotlib.pyplot as plt
-import datetime
-from hjnwtx.colormap import cmp_hjnwtx
-import time
-import os
-import time
-import shutil
- 
-from dateutil.relativedelta import relativedelta
-import glob
-from config import logger
-import argparse
-import pandas as pd 
-import re
-import netCDF4 as nc 
-from multiprocessing import Pool
-from itertools import product
-import numpy as np
-import copy
-import traceback
-
- 
-from multiprocessing import Process, Queue
- 
-import io
-import cartopy.crs as ccrs
-import cartopy.feature as cfeature
-import cartopy.io.shapereader as shpreader
-from hjnwtx.mkNCHJN import mkDir
-
-shape_ = (1, 5701, 8501)
-def add_china_map(ax):
-    # 在地图上添加地形特征
-    ax.add_feature(cfeature.COASTLINE, edgecolor='gray')
-    ax.add_feature(cfeature.BORDERS, linestyle=':', edgecolor='gray')
-    ax.add_feature(cfeature.LAKES, alpha=0.8)
-    # 添加省份轮廓
-    provinces = shpreader.natural_earth(resolution='10m', category='cultural', name='admin_1_states_provinces')
-    provinces_features = shpreader.Reader(provinces).geometries()
-    ax.add_geometries(provinces_features, ccrs.PlateCarree(), facecolor='none', edgecolor='gray', linestyle=':', linewidth=0.5, alpha=0.8)
- 
-def draw_subplot(args):
-    env = [-132.0, -47.0, 0, 57.0]
-    index, tp, vmax, vmin, cmap,time_index, name = args
-    # print(time_index)
-    # logger.info(f"Processing index: {index}")
-    # logger.info(time_index)
-    fig, ax = plt.subplots(figsize=(10, 10))
-
-    # 创建绘图和设置坐标系
-    # fig = plt.figure(figsize=(10, 8))
-    ax = fig.add_subplot(1, 1, 1, projection=ccrs.PlateCarree())
-
-    # 设置图像显示的范围
-    ax.set_extent(env, ccrs.PlateCarree())  # 根据需要调整
-
-    # 添加中国地图的边界和特征，包括省份轮廓
-    add_china_map(ax)
-    tp = np.where(tp >= 0.1, tp, np.nan)
-    ax.imshow(tp, vmin=0, vmax=10, cmap=cmp_hjnwtx["pre_tqw"], transform=ccrs.PlateCarree(), extent=env, alpha=1) 
-
-    # ax.imshow(tp, vmax=vmax, vmin=vmin, cmap=cmap)
-    ax.axis('off')
-    # ax.text(0.5, 0.5, str(index), transform=ax.transAxes, color='white', fontsize=20, ha='center', va='center')
-    ax.text(0.95, 0.95, f'{time_index}', transform=ax.transAxes, color='white', fontsize=20, ha='right', va='bottom')
-    ax.text(0.925, 0.925, f'{str(index)}', transform=ax.transAxes, color='white', fontsize=20, ha='right', va='top')
-    fig.tight_layout()
-
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)   
-    plt.close(fig)
-    return (index, buf)   #返回两个数为元组
- 
-# 修改读取部分
-def drawpic(tp, Count,timeList, name="temp"):
-    vmax = 70
-    vmin = 0
-    cmap = cmp_hjnwtx["radar_nmc"]
-    args_list = [ (index, tp[index, :, :], vmax, vmin, cmap,timeList[index], name)  for index in range(2 * Count)]
-    
-    with Pool(31) as pool:
-        results = pool.map(draw_subplot, args_list)
-    
-    buffers = [result[1] for result in results]
-    combined_fig, axs = plt.subplots(2, Count, figsize=(10 * Count, 20))
-    for i in range(2):
-        for j in range(Count):
-            index = i * Count + j
-            ax = axs[i, j]
-            ax.imshow(plt.imread(buffers[index]), cmap=cmap)
-            ax.axis('off')
-    
-    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M")
-    plt.tight_layout()
-    outpath = f"./{timeList[0].strftime('%Y%m%d%H%M')}_MQPF/{name}_{now_str}.png"
-    mkDir(outpath)
-    plt.savefig(outpath)
-    plt.close(combined_fig)
-
-
- 
-BASE_TARGET_PATH = "/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF_GLB"   #/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF_PMSC
-# BASE_TARGET_PATH = "/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF_PMSC"   #/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF_PMSC
-
-def get_mqpf_paths(UCTstr):
-    year = UCTstr[:4]
-    date = UCTstr[:8]
-    mqpfPath_pattern = os.path.join(BASE_TARGET_PATH,year, date,f"MSP2_WTX_AIW_QPF_L88_NA_{UCTstr}_00000-00300-00006.nc")  #MSP2_WTX_AIW_QPF_L88_NA_202410222254_00000-00300-00006.nc
-    # mqpfPath_pattern = os.path.join(BASE_TARGET_PATH,year, date,f"MSP2_WTX_AIW_MQPF_L88_CHN_{UCTstr}_00000-00300-00006.nc")
- 
-    return mqpfPath_pattern  
-
-def map_data(conf):
-    CST = conf[0]
-    UCT = CST + relativedelta(hours=-8)
-    UCTstr = UCT.strftime("%Y%m%d%H%M")
-    mqpfPath_pattern = get_mqpf_paths(UCTstr)
-    mqpfPath_list = glob.glob(mqpfPath_pattern)
-    if len(mqpfPath_list) == 0:
-        data_loss = re.findall(r"(2024\d{8}?)",mqpfPath_pattern)
-        print("data_loss",data_loss)
-        print("data_loss_path",mqpfPath_pattern)
-        tj_list.append(data_loss[0]) 
-        return np.full(shape_, np.nan)
-    else:
-        try:
-            with nc.Dataset(mqpfPath_list[0]) as dataNC:
-                mqpf = dataNC.variables["MQPF"][:]
-                mqpf = mqpf[:1]
-                if mqpf.shape != shape_:
-                    logger.info(mqpf.shape )
-                    logger.info(mqpfPath_list[0])
-                    print("mqpf shape error",mqpf.shape)
-                    mqpf = mqpf[:,:-1, :-1]
-                    if mqpf.shape != shape_:
-                       return np.full(shape_, np.nan)      
-                    else:
-                        return mqpf          
-                print("mqpf",UCTstr,mqpf.shape)
-            tj_list1.append(mqpfPath_pattern)
-            return mqpf
-        except Exception as e:
-            print(traceback.format_exc())
-            return np.full(shape_, np.nan)
- 
-def options():
-    parser = argparse.ArgumentParser(description='plot radar')
-    # parser.add_argument('--times', type=str, default='202406290000,202406300000')
-    parser.add_argument('--times', type=str, default='202407310000,202408010000')
-    parser.add_argument('--pac', type=str, default='100000')
-    # parser.add_argument('--combine', action='store_true', default=False)
-    parser.add_argument('--combine',action='store_true',default=False)
-    parser.add_argument('--isDebug',action='store_true',default=False)
-    parser.add_argument('--isDraw',action='store_true',default=False)
-    parser.add_argument('--freq', type=str, default="1h")
-    parser.add_argument('--tag',type=str, default=datetime.datetime.now().strftime("%Y%m%d%H%M"))
-    config= parser.parse_args()
-    print(config)
-    config.times = config.times.split(",")
-    config.pac = config.pac.split(",")
-    if len(config.times) == 1:
-        config.times = [config.times[0], config.times[0]]
-    config.times = [datetime.datetime.strptime(config.times[0], "%Y%m%d%H%M"),
-                    datetime.datetime.strptime(config.times[1], "%Y%m%d%H%M")]
-    return config
-if __name__ == '__main__':
-    cfg = options()
-    sCST = cfg.times[0]
-    eCST = cfg.times[-1]
-    sCSTstr = sCST.strftime("%Y%m%d")
-    tj_list = []
-    tj_list1= []
-    start = datetime.datetime.now()
-    logger.info(f"{sCST}")    
-    logger.info(f"{eCST}")    
-    logger.info(f"PNG")    
-    timeList = pd.date_range(sCST,eCST,freq="360s",inclusive="left")
-    # for CST in timeList:  
-    productList = product(timeList)
-    with Pool(31) as p:
-         Data = p.map(map_data,productList)
-    end = datetime.datetime.now()
-    print(start-end)
-    Data_con = np.concatenate(Data,axis=0) 
-    loss_len = 240 - Data_con.shape[0]
-    sCSTstr = sCST.strftime("%Y%m%d")
-    eCSTstr = eCST.strftime("%Y%m%d")
-    # Data_con1 = Data_con.filled()  
-    # np.save(f"data_{sCSTstr}_{eCSTstr}.npy",Data_con1) 
-    Data_con_120 = copy.copy(Data_con)
-    drawpic(Data_con_120[:int(len(Data_con_120)/2)], int(len(Data_con_120)/4),timeList[:int(len(Data_con_120)/2)],name=f"temp120_{sCSTstr}_loss_{loss_len}_")
-    print("done 120")
-    end1 = datetime.datetime.now()
-    print(end1-end)
-    Data_con_240 = copy.copy(Data_con)
-    print(Data_con_240[120:].shape)
-    drawpic(Data_con_120[int(len(Data_con_120)/2):], int(len(Data_con_120)/4),timeList[int(len(Data_con_120)/2):],name=f"temp120_240_{sCSTstr}_{loss_len}_")
-    print(datetime.datetime.now()-end1)
-    print("done 120-240") 
-    logger.info("success")
- 
-
-#  "/mnt/wtx_weather_forecast/WTX_DATA/RADA/MQPF/2024/20240704/MSP2_WTX_AIW_REF_L88_CHN_202407040324_00000-00300-00006.nc"
-

shancx/Plot/glob_nation_map.py

@@ -1,116 +0,0 @@
-import matplotlib.pyplot as plt
-import numpy as np
-import datetime
-from hjnwtx.colormap import cmp_hjnwtx  # Assuming this is your custom colormap library
-import cartopy.crs as ccrs
-import cartopy.feature as cfeature
-import cartopy.io.shapereader as shpreader
-import os
-from shancx import crDir
-
-def Glob(array_dt,cd="CHN",ty="CR"):
-    if cd == "g":
-        env = [-179.617020, 179.632979,-85.098871,85.051128] 
-    elif cd == "US":
-        env = [-132.0, -47.0, 0, 57.0]
-    elif cd == "CHN":
-        env = [73,134.99,12.21,54.2]   
-    else:
-         env = cd
-    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    outpath = f"./radar_nmc/{str(cd)}_{now_str}.png"
-    crDir(outpath)
-    # Create figure and set the coordinate system
-    fig = plt.figure(figsize=(10, 8))
-    ax = fig.add_subplot(1, 1, 1, projection=ccrs.PlateCarree())
-    # Set the extent for the United States
-    ax.set_extent(env, ccrs.PlateCarree())  # Adjust as needed
-    # Add the US map boundaries and features
-    add_glob_map(ax)    
-    # Add data layers
-    if len(array_dt.shape) == 3:
-        for i, img_ch_nel in enumerate(array_dt):
-            ax.imshow(img_ch_nel, vmin=50, vmax=500, cmap=cmp_hjnwtx["radar_nmc"], transform=ccrs.PlateCarree(), extent=env)
-            plt.colorbar(ax.images[0], ax=ax, orientation='vertical')
-            plt.savefig(f"{outpath}_layer_{i}.png")
-            plt.clf()  # Clear the figure to draw the next channel image
-    elif len(array_dt.shape) == 2 and ty =="pre":
-        ax.imshow(array_dt, vmin=0, vmax=10, cmap=cmp_hjnwtx["pre_tqw"], transform=ccrs.PlateCarree(), extent=env)   
-        plt.colorbar(ax.images[0], ax=ax, orientation='vertical')
-        plt.savefig(outpath)
-    else :
-        ax.imshow(array_dt, vmin=0, vmax=72, cmap=cmp_hjnwtx["radar_nmc"], transform=ccrs.PlateCarree(), extent=env)
-        plt.colorbar(ax.images[0], ax=ax, orientation='vertical')
-        plt.savefig(outpath)
-    plt.close(fig)
-
-def add_glob_map(ax): 
-    ax.add_feature(cfeature.COASTLINE, edgecolor='gray')
-    ax.add_feature(cfeature.BORDERS, linestyle=':', edgecolor='gray')
-    ax.add_feature(cfeature.LAKES, alpha=0.8)    
-    # Adding state boundaries
-    if os.path.exists('/home/scx/ne_10m_admin_1_states_provinces.shp'):
-        states = '/home/scx/ne_10m_admin_1_states_provinces.shp'
-    else:
-        states = shpreader.natural_earth(resolution='10m', category='cultural', name='admin_1_states_provinces')  # Automatically download 
-    states_features = shpreader.Reader(states).geometries()    
-    ax.add_geometries(states_features, ccrs.PlateCarree(), facecolor='none', edgecolor='gray', linestyle=':', linewidth=0.5, alpha=0.8)
- 
-def GlobLonLat(array_dt,Lon,Lat,cd="CHN",ty="CR"):   ###  x_coords2 维度 
-    if cd == "g":
-        env = [-179.617020, 179.632979,-85.098871,85.051128] 
-    elif cd == "US":
-        env = [-132.0, -47.0, 0, 57.0]
-    elif cd == "CHN":
-        env = [73,134.99,12.21,54.2]   #[73, 134.99, 12.21, 54.2] 
-    else:
-         env = cd    
-    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    outpath = f"./radar_nmc/{str(cd)}_{now_str}.png"
-    crDir(outpath)
-
-    # Create figure and set the coordinate system
-    fig = plt.figure(figsize=(10, 8))
-    ax = fig.add_subplot(1, 1, 1, projection=ccrs.PlateCarree())
-
-    # Set the extent for the United States
-    ax.set_extent(env, ccrs.PlateCarree())  # Adjust as needed
-
-    # Add the US map boundaries and features
-    add_glob_map(ax)
-    
-    # Add data layers
-    if len(array_dt.shape) == 3:
-        for i, img_ch_nel in enumerate(array_dt):
-            ax.imshow(img_ch_nel, vmin=50, vmax=500, cmap=cmp_hjnwtx["radar_nmc"], transform=ccrs.PlateCarree(), extent=env)
-            plt.colorbar(ax.images[0], ax=ax, orientation='vertical')
-            plt.savefig(f"{outpath}_layer_{i}.png")
-            plt.close()  
-    elif len(array_dt.shape) == 2 and ty =="pre":
-        ax.imshow(array_dt, vmin=0, vmax=10, cmap=cmp_hjnwtx["pre_tqw"], transform=ccrs.PlateCarree(), extent=env)   
-        plt.colorbar(ax.images[0], ax=ax, orientation='vertical')
-        ax.scatter(list(Lon), list(Lat), s=0.5, c='red', marker='o', transform=ccrs.PlateCarree())
-        plt.savefig(outpath)
-    else :
-        ax.imshow(array_dt, vmin=0, vmax=72, cmap=cmp_hjnwtx["radar_nmc"], transform=ccrs.PlateCarree(), extent=env)
-        ax.scatter(list(Lon), list(Lat), s=0.5, c='red', marker='o', transform=ccrs.PlateCarree())
-        plt.colorbar(ax.images[0], ax=ax, orientation='vertical')
-        plt.savefig(outpath)
-    plt.close(fig)
-
-def add_us_map(ax):
-    ax.add_feature(cfeature.COASTLINE, edgecolor='gray')
-    ax.add_feature(cfeature.BORDERS, linestyle=':', edgecolor='gray')
-    ax.add_feature(cfeature.LAKES, alpha=0.8)    
-    # Adding state boundaries
-    if os.path.exists('/home/scx/ne_10m_admin_1_states_provinces.shp'):
-        states = '/home/scx/ne_10m_admin_1_states_provinces.shp'
-    else:
-        states = shpreader.natural_earth(resolution='10m', category='cultural', name='admin_1_states_provinces')  # Automatically download 
-    states_features = shpreader.Reader(states).geometries()    
-    ax.add_geometries(states_features, ccrs.PlateCarree(), facecolor='none', edgecolor='gray', linestyle=':', linewidth=0.5, alpha=0.8)
-
-# Example usage
-# Assuming array_dt is your data array, pass it to drawUS
-# array_dt = np.random.rand(3, 100, 100)  # Example random data; replace with your actual data
-# drawUS(array_dt)

shancx/Plot/radar_nmc.py

@@ -1,61 +0,0 @@
-
-import matplotlib.pyplot as plt
-import datetime
-from hjnwtx.colormap import cmp_hjnwtx
-import os
-import numpy as np
- 
- 
-
-from pathlib import Path
-def MDir(path):
-    path_obj = Path(path)
-    directory = path_obj.parent if path_obj.suffix else path_obj
-    directory.mkdir(parents=True, exist_ok=True)
-
-
-def drawimg(array_dt,ty ="CR",temp = "temp"):
-    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    if len(array_dt.shape)==3:
-        for i , img_ch_nel in enumerate(array_dt): 
-            plt.imshow(img_ch_nel,vmin=0,vmax=10,cmap=cmp_hjnwtx["radar_nmc"])
-            plt.colorbar()
-            outpath = f"./radar_nmc/{temp}_{now_str}.png"
-            MDir(outpath)
-            plt.savefig(outpath)
-            plt.close()
-    if len(array_dt.shape)==2 and ty =="pre":
-            plt.imshow(array_dt,vmin=0,vmax=10,cmap=cmp_hjnwtx["pre_tqw"])
-            plt.colorbar()
-            outpath = f"./radar_nmc/{temp}_{now_str}.png"
-            MDir(outpath)
-            plt.savefig(outpath)
-            plt.close() 
-    else:
-            plt.imshow(array_dt,vmin=0,vmax=72,cmap=cmp_hjnwtx["radar_nmc"])
-            plt.colorbar()
-            outpath = f"./radar_nmc/{temp}_{now_str}.png"
-            MDir(outpath)
-            plt.savefig(outpath)
-            plt.close()  
-
- 
-def drawimg_coor(array_dt, temp="temp"):
-    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    y_coords2, x_coords2 = np.where(array_dt > 0)
-    
-    def plot_and_save(image, path):
-        plt.imshow(image, vmin=0, vmax=10, cmap=cmp_hjnwtx["radar_nmc"])
-        for (x, y) in zip(x_coords2, y_coords2):
-            plt.plot(x, y, 'ro', markersize=25)  # Increase point size
-            plt.text(x, y, f'{(image[y, x] * 6):.1f}', color='white', fontsize=12, ha='center', va='center')  # Label the corresponding value
-        plt.colorbar()
-        MDir(path)
-        plt.savefig(path)
-        plt.close()    
-    if len(array_dt.shape) == 3:
-        for i, img_ch_nel in enumerate(array_dt): 
-            plot_and_save(img_ch_nel, f"./radar_nmc/{temp}_{now_str}.png")
-    elif len(array_dt.shape) == 2:
-        plt.imshow(array_dt, vmin=0, vmax=100, cmap=cmp_hjnwtx["pre_tqw"])
-        plot_and_save(array_dt, f"./radar_nmc/{temp}_{now_str}.png")