oafuncs 0.0.98.34__py3-none-any.whl → 0.0.98.36__py3-none-any.whl

oafuncs/oa_cmap.py

@@ -1,7 +1,6 @@
 from typing import List, Optional, Union
 
 import matplotlib as mpl
-import matplotlib.pyplot as plt
 import numpy as np
 import random
 
@@ -24,6 +23,7 @@ def show(
     colormaps : Union[str, mpl.colors.Colormap, List[Union[str, mpl.colors.Colormap]]]
         单个颜色映射或颜色映射列表；可以是名称字符串或颜色映射对象。
     """
+    import matplotlib.pyplot as plt
     if not isinstance(colormaps, list):
         colormaps = [colormaps]
 

oafuncs/oa_data.py

@@ -1,10 +1,9 @@
 from typing import Any, List, Union
 
 import numpy as np
-import salem
 import xarray as xr
 from rich import print
-from scipy.interpolate import interp1d
+
 
 __all__ = ["interp_along_dim", "interp_2d", "ensure_list", "mask_shapefile"]
 
@@ -59,6 +58,7 @@ def interp_along_dim(
         >>> result = interp_along_dim(target_coordinates, source_coordinates, source_data)
         >>> print(result)  # Expected output: [20.0, 30.0]
     """
+    from scipy.interpolate import interp1d
     target_coordinates = np.asarray(target_coordinates)
     if target_coordinates.ndim != 1:
         raise ValueError("[red]target_coordinates must be a 1D array.[/red]")
@@ -177,6 +177,7 @@ def mask_shapefile(
         >>> print(masked_data)  # Expected output: Masked DataArray
 
     """
+    import salem
     try:
         shp_f = salem.read_shapefile(shapefile_path)
         data_da = xr.DataArray(data_array, coords=[("latitude", latitudes), ("longitude", longitudes)])

oafuncs/oa_down/hycom_3hourly.py

@@ -98,6 +98,18 @@ def _get_initial_data():
     data_info["hourly"]["dataset"]["GLBu0.08"]["version"]["93.0"]["classification"] = "var_different"
     data_info["hourly"]["dataset"]["GLBy0.08"]["version"]["93.0"]["classification"] = "var_year_different"
     data_info["hourly"]["dataset"]["ESPC_D"]["version"]["V02"]["classification"] = "single_var_year_different"
+    
+    # lon lat
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["53.X"]["lonlat"] = {'lon_min': -180.00, 'lon_max': 179.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["56.3"]["lonlat"] = {'lon_min': -180.00, 'lon_max': 179.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["57.2"]["lonlat"] = {'lon_min': -180.00, 'lon_max': 179.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["92.8"]["lonlat"] = {'lon_min': 0.00, 'lon_max': 359.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["57.7"]["lonlat"] = {'lon_min': -180.00, 'lon_max': 179.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["92.9"]["lonlat"] = {'lon_min': 0.00, 'lon_max': 359.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBv0.08"]["version"]["93.0"]["lonlat"] = {'lon_min': 0.00, 'lon_max': 359.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["GLBu0.08"]["version"]["93.0"]["lonlat"] = {'lon_min': 0.00, 'lon_max': 359.92, 'lat_min': -80.0, 'lat_max': 80.0}
+    data_info["hourly"]["dataset"]["GLBy0.08"]["version"]["93.0"]["lonlat"] = {'lon_min': 0.00, 'lon_max': 359.92, 'lat_min': -80.0, 'lat_max': 90.0}
+    data_info["hourly"]["dataset"]["ESPC_D"]["version"]["V02"]["lonlat"] = {'lon_min': 0.00, 'lon_max': 359.92, 'lat_min': -80.0, 'lat_max': 90.0}
 
     # download info
     # base url
@@ -851,6 +863,14 @@ def _get_submit_url_var(var, depth, level_num, lon_min, lon_max, lat_min, lat_ma
         which_mode = "level"
     else:
         which_mode = "full"
+    if lon_min == 0.0:
+        lon_min = data_info["hourly"]["dataset"][dataset_name]["version"][version_name]["lonlat"]["lon_min"]
+    if lon_max == 359.92:
+        lon_max = data_info["hourly"]["dataset"][dataset_name]["version"][version_name]["lonlat"]["lon_max"]
+    if lat_min == -80.0:
+        lat_min = data_info["hourly"]["dataset"][dataset_name]["version"][version_name]["lonlat"]["lat_min"]
+    if lat_max == 90.0:
+        lat_max = data_info["hourly"]["dataset"][dataset_name]["version"][version_name]["lonlat"]["lat_max"]
     query_dict = _get_query_dict(var, lon_min, lon_max, lat_min, lat_max, download_time, download_time_end, which_mode, depth, level_num)
     submit_url = _get_submit_url(dataset_name, version_name, var, ymdh_str, query_dict)
     return submit_url
@@ -1194,10 +1214,10 @@ if __name__ == "__main__":
         "start_time": "2018010100",
         "end_time": "2019063000",
         "output_dir": r"G:\Data\HYCOM\china_sea\hourly_24",
-        "lon_min": 105,
-        "lon_max": 135,
-        "lat_min": 10,
-        "lat_max": 45,
+        # "lon_min": 105,
+        # "lon_max": 135,
+        # "lat_min": 10,
+        # "lat_max": 45,
         "workers": 1,
         "overwrite": False,
         "depth": None,

oafuncs/oa_draw.py

@@ -1,12 +1,9 @@
 import warnings
 
 import cartopy.crs as ccrs
-import cartopy.feature as cfeature
-import cv2
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
-from cartopy.mpl.ticker import LatitudeFormatter, LongitudeFormatter
 from rich import print
 
 __all__ = ["fig_minus", "gif", "movie", "setup_map", "MidpointNormalize"]
@@ -74,7 +71,7 @@ def fig_minus(x_axis: plt.Axes = None, y_axis: plt.Axes = None, colorbar: mpl.co
     elif colorbar is not None:
         colorbar.set_ticklabels(out_ticks)
 
-    print("[green]Axis tick labels updated successfully.[/green]")
+    # print("[green]Axis tick labels updated successfully.[/green]")
     return target_object
 
 
@@ -148,6 +145,7 @@ def movie(image_files: list[str], output_video_path: str, fps: int) -> None:
         print("[red]Error:[/red] Image files list is empty.")
         return
 
+    import cv2
     # Read first image to get frame dimensions
     try:
         frame = cv2.imread(image_files[0])
@@ -235,6 +233,7 @@ def setup_map(
     from matplotlib import ticker as mticker
 
     # Add map features
+    import cartopy.feature as cfeature
     if show_land:
         axes.add_feature(cfeature.LAND, facecolor=land_color)
     if show_ocean:
@@ -245,6 +244,7 @@ def setup_map(
         axes.add_feature(cfeature.BORDERS, linewidth=coastline_linewidth, linestyle=":")
 
     # Setup coordinate formatting
+    from cartopy.mpl.ticker import LatitudeFormatter, LongitudeFormatter
     lon_formatter = LongitudeFormatter(zero_direction_label=False, number_format=f".{tick_decimals}f")
     lat_formatter = LatitudeFormatter(number_format=f".{tick_decimals}f")
 
@@ -344,6 +344,23 @@ class MidpointNormalize(mpl.colors.Normalize):
             result = np.clip(result, 0, 1)
 
         return np.ma.masked_array(result)
+    
+    def ticks(self, num_ticks: int = 7) -> np.ndarray:
+        """Generate ticks for the normalization range, centered around vcenter."""
+        if self.vmin is None or self.vmax is None:
+            raise ValueError("vmin and vmax must be set to generate ticks.")
+
+        if num_ticks % 2 == 0:
+            num_ticks += 1
+
+        num_points_side = (num_ticks - 1) // 2 + 1
+
+        negative_ticks = np.linspace(self.vmin, self.vcenter, num_points_side)[:-1]
+        positive_ticks = np.linspace(self.vcenter, self.vmax, num_points_side)[1:]
+
+        ticks = np.concatenate([negative_ticks, [self.vcenter], positive_ticks])
+
+        return ticks
 
     def inverse(self, value: np.ndarray) -> np.ndarray:
         y, x = [self.vmin, self.vcenter, self.vmax], [0, 0.5, 1]

oafuncs/oa_nc.py

@@ -1,7 +1,6 @@
 import os
 from typing import List, Optional, Tuple, Union
 
-import netCDF4 as nc
 import numpy as np
 import xarray as xr
 from rich import print
@@ -112,6 +111,7 @@ def rename(
     Example:
         >>> rename('file.nc', 'old_var', 'new_var')
     """
+    import netCDF4 as nc
     try:
         with nc.Dataset(file_path, "r+") as dataset:
             if old_name not in dataset.variables and old_name not in dataset.dimensions:
@@ -159,6 +159,7 @@ def check(
         return False
 
     try:
+        import netCDF4 as nc
         with nc.Dataset(file_path, "r") as ds_verify:
             if not ds_verify.variables:
                 if print_messages:

{oafuncs-0.0.98.34.dist-info → oafuncs-0.0.98.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oafuncs
-Version: 0.0.98.34
+Version: 0.0.98.36
 Summary: Oceanic and Atmospheric Functions
 Home-page: https://github.com/Industry-Pays/OAFuncs
 Author: Kun Liu

{oafuncs-0.0.98.34.dist-info → oafuncs-0.0.98.36.dist-info}/RECORD

@@ -1,11 +1,11 @@
 oafuncs/__init__.py,sha256=T_-VtnWWllV3Q91twT5Yt2sUapeA051QbPNnBxmg9nw,1456
-oafuncs/oa_cmap.py,sha256=LL2JSIofwVRymgEILjMCB4Iuz7hOYF-QemfNkaZqbk0,13880
-oafuncs/oa_data.py,sha256=Wp7Yn6n-WNddAp1UtwK5CSEdGSrzzgOH5gtaOHBaxtw,8065
+oafuncs/oa_cmap.py,sha256=h9QrOSwbzZJXuS2pjFZkJtck1bq9itPY1natIuIRB3s,13884
+oafuncs/oa_data.py,sha256=u4H1ZazQf2jmGx3IAiaGxLjQHY9cQEMizRIaTQW4UiE,8075
 oafuncs/oa_date.py,sha256=aU2wVIWXyWoRiSQ9dg8sHvShFTxw86RrgbV3Q6tDjD4,6841
-oafuncs/oa_draw.py,sha256=37Zok96Pn7aIpNqZ082cSycT8dSccgwHzScXZXNnfik,13701
+oafuncs/oa_draw.py,sha256=UnUyOcr6KHtEWCZG0TCnHFW0jrjkAW9XWZoUDcSs6oI,14386
 oafuncs/oa_file.py,sha256=fLb0gRhq2AiPl-5ASDHMrx6Z267FmhqNcTV7CdCxTdI,16934
 oafuncs/oa_help.py,sha256=0J5VaZX-cB0c090KxgmktQJBc0o00FsY-4wB8l5y00k,4178
-oafuncs/oa_nc.py,sha256=TMrrPBdPz1IJ-7dMOD_gzzTbG6FHSgm-ZfdlvoJDtcY,15245
+oafuncs/oa_nc.py,sha256=mKNxQ9jPxfRH7xINyrX7tBhitG5gmOKm6Dn7stk5mdw,15279
 oafuncs/oa_python.py,sha256=xYMQnM0cGq9xUCtcoMpnN0LG5Rc_s94tai5nC6CNJ3E,4831
 oafuncs/oa_tool.py,sha256=Zuaoa92wll0YqXGRf0oF_c7wlATtl7bvjCuLt9VLXp0,8046
 oafuncs/_data/hycom.png,sha256=MadKs6Gyj5n9-TOu7L4atQfTXtF9dvN9w-tdU9IfygI,10945710
@@ -22,8 +22,7 @@ oafuncs/_script/plot_dataset.py,sha256=QrA4vOCzWbAJp3hf5YYzgIRUZdJB5_ugepgyT_Yfn
 oafuncs/_script/replace_file_content.py,sha256=wIwvaISFNYWG58BLZHZP9ZgbC5OhoZ-cpR3y25U1EUM,5601
 oafuncs/oa_down/User_Agent-list.txt,sha256=pHaMlElMvZ8TG4vf4BqkZYKqe0JIGkr4kCN0lM1Y9FQ,514295
 oafuncs/oa_down/__init__.py,sha256=IT6oTqaxuV_mC6AwBut0HtkmnVtEu1MyX0x0oS7TKoA,218
-oafuncs/oa_down/hycom_3hourly.py,sha256=y2ZoT1uCpWRJaOMRXzvW1e9f9lXzTTqw5kYVumzms5Q,55266
-oafuncs/oa_down/hycom_3hourly_proxy.py,sha256=xNAPSuTP0fvBAGMv16eeB2TCvNhS2cJq3M5F4GUmPN8,54180
+oafuncs/oa_down/hycom_3hourly.py,sha256=dFXSC_5o-Dic616KrLXir4tEHvCiZt8vGKPEYpXFMmA,57356
 oafuncs/oa_down/idm.py,sha256=vAhRjt_Sb-rKhzFShmSf29QcFTqsHpHXCvTSD1uSXyQ,1455
 oafuncs/oa_down/literature.py,sha256=7Qy5OphcjdRwY2uZ5hmmgK36U_QtVmEUSW0vQaxihC8,10960
 oafuncs/oa_down/read_proxy.py,sha256=HQpr-Mwn0Z8ICAuf63NUUY9p05E_uNWyWmOK46-73Ec,2866
@@ -38,8 +37,8 @@ oafuncs/oa_sign/__init__.py,sha256=JSx1fcWpmNhQBvX_Bmq3xysfSkkFMrjbJASxV_V6aqE,1
 oafuncs/oa_sign/meteorological.py,sha256=3MSjy7HTcvz2zsITkjUMr_0Y027Gas1LFE9pk99990k,6110
 oafuncs/oa_sign/ocean.py,sha256=3uYEzaq-27yVy23IQoqy-clhWu1I_fhPFBAQyT-OF4M,5562
 oafuncs/oa_sign/scientific.py,sha256=moIl2MEY4uitbXoD596JmXookXGQtQsS-8_1NBBTx84,4689
-oafuncs-0.0.98.34.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
-oafuncs-0.0.98.34.dist-info/METADATA,sha256=fUtjS8AeoOMXC6t8r1_JbgOrKytjNumx_gD89e3Sb30,4326
-oafuncs-0.0.98.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-oafuncs-0.0.98.34.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
-oafuncs-0.0.98.34.dist-info/RECORD,,
+oafuncs-0.0.98.36.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
+oafuncs-0.0.98.36.dist-info/METADATA,sha256=N3OWTsj1i-CBXIdbbgy7WZJvEb5mNsHUl2SkoilJ3Oo,4326
+oafuncs-0.0.98.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+oafuncs-0.0.98.36.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
+oafuncs-0.0.98.36.dist-info/RECORD,,