oafuncs 0.0.98.12__tar.gz → 0.0.98.13__tar.gz

{oafuncs-0.0.98.12/oafuncs.egg-info → oafuncs-0.0.98.13}/PKG-INFO

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

{oafuncs-0.0.98.12 → oafuncs-0.0.98.13}/oafuncs/_script/netcdf_merge.py

@@ -3,10 +3,8 @@ from typing import List, Optional, Union
 
 import numpy as np
 import xarray as xr
-from dask.diagnostics import ProgressBar
 
 from oafuncs import pbar
-from oafuncs._script.netcdf_write import _calculate_scale_and_offset  # 新增导入
 
 
 def merge_nc(file_list: Union[str, List[str]], var_name: Optional[Union[str, List[str]]] = None, dim_name: Optional[str] = None, target_filename: Optional[str] = None) -> None:
@@ -27,6 +25,7 @@ def merge_nc(file_list: Union[str, List[str]], var_name: Optional[Union[str, Lis
         merge(file_list, var_name=['u', 'v'], dim_name='time', target_filename='merged.nc')
         merge(file_list, var_name=None, dim_name='time', target_filename='merged.nc')
     """
+    from oafuncs._script.netcdf_write import save_to_nc
 
     if target_filename is None:
         target_filename = "merged.nc"
@@ -53,7 +52,7 @@ def merge_nc(file_list: Union[str, List[str]], var_name: Optional[Union[str, Lis
     # 初始化合并数据字典
     merged_data = {}
 
-    for i, file in pbar(enumerate(file_list), description="Reading files", color="#f8bbd0", total=len(file_list)):
+    for i, file in pbar(enumerate(file_list), description="Reading files", total=len(file_list)):
         with xr.open_dataset(file) as ds:
             for var in var_names:
                 data_var = ds[var]
@@ -66,48 +65,24 @@ def merge_nc(file_list: Union[str, List[str]], var_name: Optional[Union[str, Lis
                     else:
                         merged_data[var] = data_var.fillna(0)
 
-    for var in pbar(merged_data, description="Merging variables", color="#9b45d1"):
+    for var in pbar(merged_data, description="Merging variables"):
         if isinstance(merged_data[var], list):
             # 判断类型，时间类型用NaT填充
             if np.issubdtype(merged_data[var][0].dtype, np.datetime64):
                 merged_data[var] = xr.concat(merged_data[var], dim=dim_name).fillna(np.datetime64("NaT"))
             else:
                 merged_data[var] = xr.concat(merged_data[var], dim=dim_name).fillna(0)
-        # print(f"Variable '{var}' merged: min={merged_data[var].min().values:.3f}, max={merged_data[var].max().values:.3f}, mean={merged_data[var].mean().values:.3f}")
 
-    # 修改写入数据部分，支持压缩并设置基数和比例因子
-    # print("\nWriting data to file ...")
     if os.path.exists(target_filename):
         print("Warning: The target file already exists. Removing it ...")
         os.remove(target_filename)
-
-    with xr.Dataset(merged_data) as merged_dataset:
-        encoding = {}
-        for var in merged_dataset.data_vars:
-            data = merged_dataset[var].values
-            if data.dtype.kind in {"i", "u", "f"}:  # 仅对数值型数据进行压缩
-                # 统一调用 netcdf_write 中的 scale/offset 计算
-                scale_factor, add_offset = _calculate_scale_and_offset(data, n=16)
-                encoding[var] = {
-                    "zlib": True,
-                    "complevel": 4,
-                    "dtype": "int16",
-                    "scale_factor": scale_factor,
-                    "add_offset": add_offset,
-                    "_FillValue": -32767,
-                }
-            else:
-                encoding[var] = {"zlib": True, "complevel": 4}
-
-        # 确保写入时不会因编码问题导致数据丢失
-        # merged_dataset.to_netcdf(target_filename, encoding=encoding)
-        delayed_write = merged_dataset.to_netcdf(target_filename, encoding=encoding, compute=False)
-        with ProgressBar():
-            delayed_write.compute()
+    
+    save_to_nc(target_filename, merged_data)
 
     print(f'\nFile "{target_filename}" has been successfully created.')
 
 
+
 # Example usage
 if __name__ == "__main__":
     files_to_merge = ["file1.nc", "file2.nc", "file3.nc"]

{oafuncs-0.0.98.12 → oafuncs-0.0.98.13}/oafuncs/oa_tool.py

@@ -2,6 +2,7 @@ import datetime
 import logging
 from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, Union
 
+
 from rich import print
 
 from ._script.parallel import ParallelExecutor
@@ -174,6 +175,8 @@ def pbar(
     """
     from ._script.cprogressbar import ColorProgressBar
     import random
+    
+    # number = random.randint(1, 999)
 
     def _generate_random_color_hex():
         """Generate a random color in hexadecimal format."""
@@ -186,9 +189,11 @@ def pbar(
         color = _generate_random_color_hex()
 
     style = f"bold {color if color != 'None' else 'green'}"
+    # print(f"[{style}]~*^* {description} *^*~ -> {number:03d}[/{style}]")
     print(f"[{style}]~*^* {description} *^*~[/{style}]")
 
-    description=''
+    # description=f'{number:03d}'
+    description = ""
     
     return ColorProgressBar(
         iterable=iterable,

{oafuncs-0.0.98.12 → oafuncs-0.0.98.13/oafuncs.egg-info}/PKG-INFO

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

{oafuncs-0.0.98.12 → oafuncs-0.0.98.13}/setup.py

@@ -18,7 +18,7 @@ URL = "https://github.com/Industry-Pays/OAFuncs"
 EMAIL = "liukun0312@stu.ouc.edu.cn"
 AUTHOR = "Kun Liu"
 REQUIRES_PYTHON = ">=3.9.0"  # 2025/03/13
-VERSION = "0.0.98.12"
+VERSION = "0.0.98.13"
 
 # What packages are required for this module to be executed?
 REQUIRED = [