oafuncs 0.0.98.33__py3-none-any.whl → 0.0.98.35__py3-none-any.whl

oafuncs/_script/plot_dataset.py

@@ -91,7 +91,7 @@ def plot_2d(data: xr.DataArray, output_path: str, data_range: Optional[Tuple[flo
         lon_lat_ratio = np.abs(np.max(lon_range) - np.min(lon_range)) / (np.max(lat_range) - np.min(lat_range))
         figsize = (10, 10 / lon_lat_ratio)
         fig, ax = plt.subplots(figsize=figsize, subplot_kw={"projection": ccrs.PlateCarree()})
-        oafuncs.oa_draw.add_cartopy(ax, lon_range, lat_range)
+        oafuncs.oa_draw.setup_map(ax, lon_range, lat_range)
     else:
         fig, ax = plt.subplots(figsize=(10, 8))
 

oafuncs/oa_cmap.py

@@ -1,10 +1,10 @@
 from typing import List, Optional, Union
 
 import matplotlib as mpl
-import matplotlib.pyplot as plt
 import numpy as np
+import random
 
-__all__ = ["show", "to_color", "create", "get"]
+__all__ = ["show", "to_color", "create", "get", "random_color"]
 
 
 # ** 将cmap用填色图可视化（官网摘抄函数）
@@ -23,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]
 
@@ -304,6 +305,14 @@ def get(colormap_name: Optional[str] = None, show_available: bool = False) -> Op
             return mpl.colormaps.get_cmap("rainbow")  # 默认返回 'rainbow'
 
 
+# ** 生成随机颜色
+def random_color():
+    """Generate a random color in hexadecimal format."""
+    r = random.randint(0, 255)
+    g = random.randint(0, 255)
+    b = random.randint(0, 255)
+    return f"#{r:02x}{g:02x}{b:02x}"
+
 if __name__ == "__main__":
     # ** 测试自制cmap
     colors = ["#C2B7F3", "#B3BBF2", "#B0CBF1", "#ACDCF0", "#A8EEED"]

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_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])
@@ -231,61 +229,11 @@ def setup_map(
     right_labels: bool = False,
     top_labels: bool = False,
 ) -> plt.Axes:
-    """Setup a complete cartopy map with customizable features.
-
-    Args:
-        axes (plt.Axes): The axes to setup as a map.
-        longitude_data (np.ndarray, optional): Array of longitudes to set map extent.
-        latitude_data (np.ndarray, optional): Array of latitudes to set map extent.
-        map_projection (ccrs.Projection, optional): Coordinate reference system. Defaults to PlateCarree.
-
-        show_land (bool, optional): Whether to show land features. Defaults to True.
-        show_ocean (bool, optional): Whether to show ocean features. Defaults to True.
-        show_coastline (bool, optional): Whether to show coastlines. Defaults to True.
-        show_borders (bool, optional): Whether to show country borders. Defaults to False.
-        land_color (str, optional): Color of land. Defaults to "lightgrey".
-        ocean_color (str, optional): Color of oceans. Defaults to "lightblue".
-        coastline_linewidth (float, optional): Line width for coastlines. Defaults to 0.5.
-
-        show_gridlines (bool, optional): Whether to show gridlines. Defaults to False.
-        longitude_ticks (list[float], optional): Longitude tick positions.
-        latitude_ticks (list[float], optional): Latitude tick positions.
-        tick_decimals (int, optional): Number of decimal places for tick labels. Defaults to 0.
-
-        grid_color (str, optional): Gridline color. Defaults to "k".
-        grid_alpha (float, optional): Gridline transparency. Defaults to 0.5.
-        grid_style (str, optional): Gridline style. Defaults to "--".
-        grid_width (float, optional): Gridline width. Defaults to 0.5.
-
-        show_labels (bool, optional): Whether to show coordinate labels. Defaults to True.
-        left_labels (bool, optional): Show labels on left side. Defaults to True.
-        bottom_labels (bool, optional): Show labels on bottom. Defaults to True.
-        right_labels (bool, optional): Show labels on right side. Defaults to False.
-        top_labels (bool, optional): Show labels on top. Defaults to False.
-
-    Returns:
-        plt.Axes: The configured map axes.
-
-    Examples:
-        >>> # Basic map setup
-        >>> ax = setup_map(ax)
-
-        >>> # Map with gridlines and custom extent
-        >>> ax = setup_map(ax, longitude_data=lon, latitude_data=lat, show_gridlines=True)
-
-        >>> # Customized map
-        >>> ax = setup_map(
-        ...     ax,
-        ...     show_gridlines=True,
-        ...     longitude_ticks=[0, 30, 60],
-        ...     latitude_ticks=[-30, 0, 30],
-        ...     land_color='wheat',
-        ...     ocean_color='lightcyan'
-        ... )
-    """
+    """Setup a complete cartopy map with customizable features."""
     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:
@@ -296,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")
 
@@ -327,14 +276,14 @@ def setup_map(
             current_extent = axes.get_extent(crs=map_projection)
             lon_range = current_extent[1] - current_extent[0]
             # Generate reasonable tick spacing
-            tick_spacing = 5 if lon_range <= 30 else (10 if lon_range <= 90 else 20)
+            tick_spacing = 1 if lon_range <= 10 else (5 if lon_range <= 30 else (10 if lon_range <= 90 else 20))
             longitude_ticks = np.arange(np.ceil(current_extent[0] / tick_spacing) * tick_spacing, current_extent[1] + tick_spacing, tick_spacing)
 
         if latitude_ticks is None:
             current_extent = axes.get_extent(crs=map_projection)
             lat_range = current_extent[3] - current_extent[2]
             # Generate reasonable tick spacing
-            tick_spacing = 5 if lat_range <= 30 else (10 if lat_range <= 90 else 20)
+            tick_spacing = 1 if lat_range <= 10 else (5 if lat_range <= 30 else (10 if lat_range <= 90 else 20))
             latitude_ticks = np.arange(np.ceil(current_extent[2] / tick_spacing) * tick_spacing, current_extent[3] + tick_spacing, tick_spacing)
 
         # Set tick positions and formatters
@@ -343,12 +292,18 @@ def setup_map(
         axes.xaxis.set_major_formatter(lon_formatter)
         axes.yaxis.set_major_formatter(lat_formatter)
 
+        # Control label visibility based on input parameters
+        axes.tick_params(axis="x", labelbottom=bottom_labels, labeltop=top_labels)
+        axes.tick_params(axis="y", labelleft=left_labels, labelright=right_labels)
+
     # 只要传入经纬度数据就自动设置范围
     # 范围必须在cartopy添加地图特征之后设置，因为添加特征可能会改变axes的范围
     if longitude_data is not None and latitude_data is not None:
         # 过滤掉NaN，避免极端值影响
-        lon_valid = np.asarray(longitude_data)[~np.isnan(longitude_data)]
-        lat_valid = np.asarray(latitude_data)[~np.isnan(latitude_data)]
+        lon_data = np.asarray(longitude_data)
+        lat_data = np.asarray(latitude_data)
+        lon_valid = lon_data[~np.isnan(lon_data)]
+        lat_valid = lat_data[~np.isnan(lat_data)]
         if lon_valid.size > 0 and lat_valid.size > 0:
             lon_min, lon_max = np.min(lon_valid), np.max(lon_valid)
             lat_min, lat_max = np.min(lat_valid), np.max(lat_valid)
@@ -389,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.33.dist-info → oafuncs-0.0.98.35.dist-info}/METADATA

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

{oafuncs-0.0.98.33.dist-info → oafuncs-0.0.98.35.dist-info}/RECORD

@@ -1,11 +1,11 @@
 oafuncs/__init__.py,sha256=T_-VtnWWllV3Q91twT5Yt2sUapeA051QbPNnBxmg9nw,1456
-oafuncs/oa_cmap.py,sha256=KKEM3Dx0RYYFdLaj0eX6p1iQ1IcKR0RZmt7a3bMGO3Y,13609
-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=oPW1dlL0dKIfx4u18djBJfokhZAAAV7t-280DU4MblQ,15967
+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
@@ -18,12 +18,11 @@ oafuncs/_script/netcdf_modify.py,sha256=XDlAEToe_lwfAetkBSENqU5df-wnH7MGuxNTjG1g
 oafuncs/_script/netcdf_write.py,sha256=GvyUyUhzMonzSp3y4pT8ZAfbQrsh5J3dLnmINYJKhuE,21422
 oafuncs/_script/parallel.py,sha256=07-BJVHxXJNlrOrhrSGt7qCZiKWq6dBvNDBA1AANYnI,8861
 oafuncs/_script/parallel_test.py,sha256=0GBqZOX7IaCOKF2t1y8N8YYu53GJ33OkfsWgpvZNqM4,372
-oafuncs/_script/plot_dataset.py,sha256=4jhUZNIc2DLHnk5GH0rotzhOq0cAMmB7rhMJKorYObo,16329
+oafuncs/_script/plot_dataset.py,sha256=QrA4vOCzWbAJp3hf5YYzgIRUZdJB5_ugepgyT_YfnaY,16327
 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/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.33.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
-oafuncs-0.0.98.33.dist-info/METADATA,sha256=SSKQ7VF-hVUtg_YT91r5QfPfL_sAMlIF5Cz0jPuTAbI,4326
-oafuncs-0.0.98.33.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-oafuncs-0.0.98.33.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
-oafuncs-0.0.98.33.dist-info/RECORD,,
+oafuncs-0.0.98.35.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
+oafuncs-0.0.98.35.dist-info/METADATA,sha256=s4W9Zkj2_ALKyPBleO3l791AVeekgC508wE7XNRrpcU,4326
+oafuncs-0.0.98.35.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+oafuncs-0.0.98.35.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
+oafuncs-0.0.98.35.dist-info/RECORD,,