oafuncs 0.0.80__tar.gz → 0.0.81__tar.gz

{oafuncs-0.0.80/oafuncs.egg-info → oafuncs-0.0.81}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oafuncs
-Version: 0.0.80
+Version: 0.0.81
 Summary: My short description for my project.
 Home-page: https://github.com/Industry-Pays/OAFuncs
 Author: Kun Liu
@@ -13,7 +13,6 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.9.0

oafuncs-0.0.81/oafuncs/__init__.py

@@ -0,0 +1,44 @@
+#!/usr/bin/env python
+# coding=utf-8
+"""
+Author: Liu Kun && 16031215@qq.com
+Date: 2024-09-17 16:09:20
+LastEditors: Liu Kun && 16031215@qq.com
+LastEditTime: 2024-12-13 12:31:06
+FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_s\\__init__.py
+Description:
+EditPlatform: vscode
+ComputerInfo: XPS 15 9510
+SystemInfo: Windows 11
+Python Version: 3.12
+"""
+
+
+# 会导致OAFuncs直接导入所有函数，不符合模块化设计
+# from oafuncs.oa_s.oa_cmap import *
+# from oafuncs.oa_s.oa_data import *
+# from oafuncs.oa_s.oa_draw import *
+# from oafuncs.oa_s.oa_file import *
+# from oafuncs.oa_s.oa_help import *
+# from oafuncs.oa_s.oa_nc import *
+# from oafuncs.oa_s.oa_python import *
+
+# ------------------- 2024-12-13 12:31:06 -------------------
+# path: My_Funcs/OAFuncs/oafuncs/
+from .oa_cmap import *
+from .oa_data import *
+from .oa_draw import *
+from .oa_file import *
+from .oa_help import *
+from .oa_nc import *
+from .oa_python import *
+# ------------------- 2024-12-13 12:31:06 -------------------
+# path: My_Funcs/OAFuncs/oafuncs/oa_down/
+from .oa_down import *
+# ------------------- 2024-12-13 12:31:06 -------------------
+# path: My_Funcs/OAFuncs/oafuncs/oa_sign/
+from .oa_sign import *
+# ------------------- 2024-12-13 12:31:06 -------------------
+# path: My_Funcs/OAFuncs/oafuncs/oa_tool/
+from .oa_tool import *
+# ------------------- 2024-12-13 12:31:06 -------------------

oafuncs-0.0.80/oafuncs - /345/211/257/346/234/254/oa_nc.py" → oafuncs-0.0.81/oafuncs/oa_nc.py

@@ -82,15 +82,27 @@ def _numpy_to_nc_type(numpy_type):
     return numpy_to_nc.get(str(numpy_type), "f4")  # 默认使用 'float32'
 
 
-def write2nc(file, data, varname=None, coords=None, mode='w'):
+def _calculate_scale_and_offset(data, n=16):
+    data_min, data_max = np.nanmin(data), np.nanmax(data)
+    scale_factor = (data_max - data_min) / (2 ** n - 1)
+    add_offset = data_min + 2 ** (n - 1) * scale_factor
+    # S = Q * scale_factor + add_offset
+    return scale_factor, add_offset
+
+
+def write2nc(file, data, varname=None, coords=None, mode='w', scale_offset_switch=True, compile_switch=True):
     """
     description: 写入数据到nc文件
+    
     参数：
     file: 文件路径
     data: 数据
     varname: 变量名
     coords: 坐标，字典，键为维度名称，值为坐标数据
     mode: 写入模式，'w'为写入，'a'为追加
+    scale_offset_switch: 是否使用scale_factor和add_offset，默认为True
+    compile_switch: 是否使用压缩参数，默认为True
+    
     example: write2nc(r'test.nc', data, 'data', {'time': np.linspace(0, 120, 100), 'lev': np.linspace(0, 120, 50)}, 'a')
     """
     # 判断mode是写入还是追加
@@ -115,8 +127,8 @@ def write2nc(file, data, varname=None, coords=None, mode='w'):
     if complete:
         return
     
-    # kwargs = {'zlib': True, 'complevel': 4} # 压缩参数
-    kwargs = {"compression": 'zlib', "complevel": 9}  # 压缩参数
+    kwargs = {'zlib': True, 'complevel': 4} # 压缩参数
+    # kwargs = {"compression": 'zlib', "complevel": 4}  # 压缩参数
 
     # 打开 NetCDF 文件
     with nc.Dataset(file, mode, format="NETCDF4") as ncfile:
@@ -137,7 +149,10 @@ def write2nc(file, data, varname=None, coords=None, mode='w'):
             if add_coords:
                 # 创建新坐标
                 ncfile.createDimension(dim, len(coord_data))
-                ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,), **kwargs)
+                if compile_switch:
+                    ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,), **kwargs)
+                else:
+                    ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,))
                 ncfile.variables[dim][:] = np.array(coord_data)
             
             if isinstance(coord_data, xr.DataArray):
@@ -161,7 +176,25 @@ def write2nc(file, data, varname=None, coords=None, mode='w'):
         if add_var:
             # 创建变量及其维度
             dim_names = tuple(coords.keys())  # 使用coords传入的维度名称
-            ncfile.createVariable(varname, _numpy_to_nc_type(data.dtype), dim_names, **kwargs)
+            if scale_offset_switch:
+                scale_factor, add_offset = _calculate_scale_and_offset(np.array(data))
+                _FillValue = -32767
+                missing_value = -32767
+                dtype = 'i2' # short类型
+            else:
+                dtype = _numpy_to_nc_type(data.dtype)
+            
+            if compile_switch:
+                ncfile.createVariable(varname, dtype, dim_names, **kwargs)
+            else:
+                ncfile.createVariable(varname, dtype, dim_names)
+            
+            if scale_offset_switch: # 需要在写入数据之前设置scale_factor和add_offset
+                ncfile.variables[varname].setncattr('scale_factor', scale_factor)
+                ncfile.variables[varname].setncattr('add_offset', add_offset)
+                ncfile.variables[varname].setncattr('_FillValue', _FillValue)
+                ncfile.variables[varname].setncattr('missing_value', missing_value)
+                
             # ncfile.createVariable('data', 'f4', ('time','lev'))
 
             # 写入数据
@@ -175,6 +208,8 @@ def write2nc(file, data, varname=None, coords=None, mode='w'):
         current_var = ncfile.variables[varname]
         if data.attrs:
             for key, value in data.attrs.items():
+                if key in ["scale_factor", "add_offset", "_FillValue", "missing_value"] and scale_offset_switch:
+                    continue
                 current_var.setncattr(key, value)
 
 

{oafuncs-0.0.80 → oafuncs-0.0.81/oafuncs.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oafuncs
-Version: 0.0.80
+Version: 0.0.81
 Summary: My short description for my project.
 Home-page: https://github.com/Industry-Pays/OAFuncs
 Author: Kun Liu
@@ -13,7 +13,6 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.9.0

oafuncs-0.0.81/oafuncs.egg-info/SOURCES.txt

@@ -0,0 +1,28 @@
+LICENSE.txt
+MANIFEST.in
+README.md
+setup.py
+oafuncs/__init__.py
+oafuncs/oa_cmap.py
+oafuncs/oa_data.py
+oafuncs/oa_draw.py
+oafuncs/oa_file.py
+oafuncs/oa_help.py
+oafuncs/oa_nc.py
+oafuncs/oa_python.py
+oafuncs.egg-info/PKG-INFO
+oafuncs.egg-info/SOURCES.txt
+oafuncs.egg-info/dependency_links.txt
+oafuncs.egg-info/requires.txt
+oafuncs.egg-info/top_level.txt
+oafuncs/oa_down/User_Agent-list.txt
+oafuncs/oa_down/__init__.py
+oafuncs/oa_down/hycom_3hourly.py
+oafuncs/oa_down/literature.py
+oafuncs/oa_down/test_ua.py
+oafuncs/oa_sign/__init__.py
+oafuncs/oa_sign/meteorological.py
+oafuncs/oa_sign/ocean.py
+oafuncs/oa_sign/scientific.py
+oafuncs/oa_tool/__init__.py
+oafuncs/oa_tool/email.py

oafuncs-0.0.81/oafuncs.egg-info/top_level.txt

@@ -0,0 +1 @@
+oafuncs

{oafuncs-0.0.80 → oafuncs-0.0.81}/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'
-VERSION = '0.0.80'
+VERSION = '0.0.81'
 
 # What packages are required for this module to be executed?
 REQUIRED = [
@@ -122,7 +122,7 @@ setup(
     python_requires=REQUIRES_PYTHON,
     url=URL,
     packages=find_packages(
-        exclude=["oa_s", "oa_down", "oa_sign", "oa_tool", "oa_*"]),
+        exclude=["oa_*", "oa_down", "oa_sign", "oa_tool"]),
     # packages=find_packages(exclude=["nc", "file", "*.tests.*", "tests.*"]),
     # If your package is a single module, use this instead of 'packages':
     # py_modules=['mypackage'],
@@ -144,7 +144,6 @@ setup(
         'Programming Language :: Python :: 3.10',
         'Programming Language :: Python :: 3.11',
         'Programming Language :: Python :: 3.12',
-        'Programming Language :: Python :: 3.13',
         'Programming Language :: Python :: Implementation :: CPython',
         'Programming Language :: Python :: Implementation :: PyPy'
     ],

oafuncs-0.0.80/oafuncs/__init__.py

@@ -1,29 +0,0 @@
-#!/usr/bin/env python
-# coding=utf-8
-"""
-Author: Liu Kun && 16031215@qq.com
-Date: 2024-09-17 16:09:20
-LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-12-13 10:56:43
-FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\__init__.py
-Description:
-EditPlatform: vscode
-ComputerInfo: XPS 15 9510
-SystemInfo: Windows 11
-Python Version: 3.11
-"""
-
-# 会导致OAFuncs直接导入所有函数，不符合模块化设计
-from oafuncs.oa_s import (
-    oa_cmap,
-    oa_data,
-    oa_draw,
-    oa_file,
-    oa_help,
-    oa_nc,
-    oa_python,
-)
-
-from .oa_down import *
-from .oa_sign import *
-from .oa_tool import *

oafuncs-0.0.80/oafuncs/oa_down/__init__.py

@@ -1,20 +0,0 @@
-#!/usr/bin/env python
-# coding=utf-8
-'''
-Author: Liu Kun && 16031215@qq.com
-Date: 2024-09-17 16:09:20
-LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-12-01 19:49:40
-FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_down\\__init__.py
-Description:  
-EditPlatform: vscode
-ComputerInfo: XPS 15 9510
-SystemInfo: Windows 11
-Python Version: 3.11
-'''
-
-# from .love_ocean import sign as love_ocean
-# from .meteorological_home import sign as meteorological_home
-
-from .hycom_3hourly import *
-from .literature import *