wrfrun 0.1.5__tar.gz → 0.1.6__tar.gz

wrfrun-0.1.6/.gitattributes

@@ -0,0 +1,5 @@
+.gitea/ export-ignore
+.github/ export-ignore
+images/ export-ignore
+.readthedocs.yaml export-ignore
+docs export-ignore

{wrfrun-0.1.5 → wrfrun-0.1.6}/.gitignore

@@ -9,6 +9,7 @@ dist/
 wrfrun.egg-info/
 .idea/
 data/
+docs/build
 
 **/*.1
 **/*.input_step1

{wrfrun-0.1.5 → wrfrun-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wrfrun
-Version: 0.1.5
+Version: 0.1.6
 Summary: wrfrun is a comprehensive toolkit for managing and using WRF
 Keywords: WRF
 Author-Email: Syize <syizeliu@gmail.com>
@@ -18,7 +18,6 @@ Requires-Dist: PyYAML
 Requires-Dist: pandas
 Requires-Dist: f90nml
 Requires-Dist: cdsapi
-Requires-Dist: seafog
 Requires-Dist: matplotlib
 Requires-Dist: Cartopy
 Requires-Dist: wrf-python
@@ -43,10 +42,14 @@ The following are the main features that wrfrun wants to achieve. These features
 - Real-time parsing of WRF logs, feedback on simulation progress.
 - Support for adding more functionality through extensions.
 
-## Install
+## Installation
 
 Install using pip:
 
 ```bash
 pip install wrfrun
-```
+```
+
+## Documentation
+
+Please check [Wiki](https://github.com/Syize/wrfrun/wiki).

{wrfrun-0.1.5 → wrfrun-0.1.6}/README.md

@@ -16,10 +16,14 @@ The following are the main features that wrfrun wants to achieve. These features
 - Real-time parsing of WRF logs, feedback on simulation progress.
 - Support for adding more functionality through extensions.
 
-## Install
+## Installation
 
 Install using pip:
 
 ```bash
 pip install wrfrun
-```
+```
+
+## Documentation
+
+Please check [Wiki](https://github.com/Syize/wrfrun/wiki).

{wrfrun-0.1.5 → wrfrun-0.1.6}/meson.build

@@ -1,7 +1,7 @@
 project(
     'wrfrun',
     'c',
-    version: '0.1.5',
+    version: '0.1.6',
     license: 'BSD-3',
     meson_version: '>=0.64.0',
     default_options: ['warning_level=2'],

{wrfrun-0.1.5 → wrfrun-0.1.6}/pyproject.toml

@@ -1,10 +1,10 @@
 [build-system]
 build-backend = "mesonpy"
-requires = ["meson-python", "numpy<2.0.0"]
+requires = ["meson-python"]
 
 [project]
 name = "wrfrun"
-version = "0.1.5"
+version = "0.1.6"
 readme = "README.md"
 license = { text = "GPL-3.0-or-later" }
 requires-python = '>=3.10'
@@ -14,7 +14,7 @@ maintainers = [
 ]
 keywords = [ "WRF", ]
 description = "wrfrun is a comprehensive toolkit for managing and using WRF"
-dependencies = ["xarray", "netCDF4", "rich", "PyYAML", "pandas", "f90nml", "cdsapi", "seafog", "matplotlib", "Cartopy", "wrf-python", "cfgrib"]
+dependencies = ["xarray", "netCDF4", "rich", "PyYAML", "pandas", "f90nml", "cdsapi", "matplotlib", "Cartopy", "wrf-python", "cfgrib"]
 
 [project.urls]
 homepage = "https://github.com/Syize/wrfrun"

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/data.py

@@ -4,7 +4,7 @@ from os.path import exists, dirname
 from typing import Union, List, Tuple
 
 from pandas import date_range
-from seafog import goos_sst_find_data
+# from seafog import goos_sst_find_data
 
 import cdsapi
 
@@ -315,9 +315,9 @@ def prepare_wps_input_data(area: Tuple[int, int, int, int]):
     download_data(start_date, end_date, hour_step, area, f"{bg_save_path}/pressure.grib",
                   data_format="grib", data_type="pressure", overwrite=True)
 
-    logger.info(f"Download NearGOOS data...")
-    download_data(start_date, end_date, hour_step, area,
-                  save_path=sst_save_path, data_type="goos", overwrite=True)
+    # logger.info(f"Download NearGOOS data...")
+    # download_data(start_date, end_date, hour_step, area,
+    #               save_path=sst_save_path, data_type="goos", overwrite=True)
 
 
 def download_data(
@@ -394,9 +394,10 @@ def download_data(
         )
         pressure_level = None
     elif data_type == "goos":
+        logger.warning(f"NEAR-GOOS SST data hasn't been supported yet")
         # download sst data
-        for _date in date_list:
-            _ = goos_sst_find_data(_date, save_path=save_path)
+        # for _date in date_list:
+        #     _ = goos_sst_find_data(_date, save_path=save_path)
 
         return ""
     else:

wrfrun-0.1.6/wrfrun/extension/goos_sst/README.md

@@ -0,0 +1,10 @@
+# GOOS SST extension
+
+This extension can replace SKT (SKin Temperature) values in ERA5 reanalysis data with the SST (Sea Surface Temperature) from GOOS (Global Ocean Observing System). A new GRIB file will be generated so you can input it to WRF through `ungrib.exe` and `metgrid.exe`.
+
+A [Vtable file](res/Vtable.ERA_GOOS_SST) is also provided for `ungrib.exe`.
+
+## Features
+
+- Replace SKT in ERA5 data with GOOS SST and write new data to a GRIB file.
+- While only ERA5 data is tested, data from other sources is supported theoretically.

wrfrun-0.1.6/wrfrun/extension/littler/README.md

@@ -0,0 +1,9 @@
+# Little R extension
+
+This extension can write data to `LITTLE_R` format for WRF data assimilation.
+
+## Features
+
+- Provide `LittleR` class to write any data to `LITTLE_R` format.
+- Write data to a zip file (end with `.zlr`, that is `Zipped Little R file`).
+- Read data from the `.zlr` file, so you can create `LITTLE_R` file with the data.

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/extension/littler/utils.py

@@ -14,7 +14,7 @@ def to_fstring(var: Union[int, float, bool, str], length: Union[int, Tuple[int,
     :param var: Basic variable, can be one of the ``[int, float, bool, str]``.
     :type var: Union[int, float, bool, str]
     :param length: The length of the output string. 
-        If the type of ``var`` is ``float``, the length must contains two parameters ``(total length, decimal length)``.
+        If the type of ``var`` is ``float``, the length must contain two parameters ``(total length, decimal length)``.
     :type length: Union[int, Tuple[int, int]]
     :return: Converted string.
     :rtype: str

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/extension/meson.build

@@ -6,5 +6,4 @@ py3.install_sources(
     subdir: 'wrfrun/extension'
 )
 
-subdir('goos_sst')
 subdir('littler')

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/model/geodata.py

@@ -44,7 +44,7 @@ def _get_clip_area(index_area: tuple[int, int, int, int], row_num: int, col_num:
     Returns:
         tuple[int, int, int, int]: Clip area.
     """
-    # caluclate tile area
+    # calculate tile area
     tile_area = (
         col_num * tile_x + 1,
         col_num * tile_x + tile_x,

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/model/utils/core.py

@@ -57,7 +57,7 @@ def model_preprocess(exec_name: str, work_path: str):
             copyfile(_file, f"{work_path}/{filename}")
 
 
-def model_postprocess(output_dir: str, save_path: str, startswith: Union[None, str, Tuple[str]] = None, endswith: Union[None, str, Tuple[str]] = None,
+def model_postprocess(output_dir: str, save_path: str, startswith: Union[None, str, Tuple[str, ...]] = None, endswith: Union[None, str, Tuple[str, ...]] = None,
                       outputs: Union[None, str, List[str]] = None, copy_only=True):
     """
     Save the results and logs from the model.

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/res/__init__.py

@@ -4,6 +4,7 @@ from .extension import *
 from .geogrid import *
 from .namelist import *
 from .ungrib import *
+from .job_scheduler import *
 
 
 _RES_PATH = abspath(dirname(__file__))

wrfrun-0.1.6/wrfrun/res/job_scheduler/__init__.py

@@ -0,0 +1,8 @@
+from os.path import abspath, dirname
+
+
+_RES_PATH = abspath(dirname(__file__))
+PBS_TEMPLATE = f"{_RES_PATH}/pbs.template"
+SLURM_TEMPLATE = f"{_RES_PATH}/slurm.template"
+
+__all__ = ["PBS_TEMPLATE", "SLURM_TEMPLATE"]

wrfrun-0.1.6/wrfrun/res/job_scheduler/meson.build

@@ -0,0 +1,8 @@
+py3.install_sources(
+    [
+        '__init__.py',
+        'pbs.template',
+        'slurm.template',
+    ],
+    subdir: 'wrfrun/res/job_scheduler'
+)

wrfrun-0.1.6/wrfrun/res/job_scheduler/pbs.template

@@ -0,0 +1,6 @@
+#PBS -q batch
+#PBS -N wrfrun
+#PBS -o {STDOUT_LOG_PATH}
+#PBS -e {STDERR_LOG_PATH}
+#PBS -l nodes={NODE_NUM}:ppn={CORE_NUM}
+#PBS -l walltime=9999:00:00

wrfrun-0.1.6/wrfrun/res/job_scheduler/slurm.template

@@ -0,0 +1,6 @@
+#SBATCH -J wrfrun                   %Job's name
+#SBATCH -N{NODE_NUM}                %node num
+#SBATCH -n{CORE_NUM}                %core num
+#SBATCH --time=9999:00:00           %maximum run time
+#SBATCH --output={STDOUT_LOG_PATH}  %stdout output log path
+#SBATCH --error={STDERR_LOG_PATH}   %error output log path

{wrfrun-0.1.5 → wrfrun-0.1.6}/wrfrun/res/meson.build

@@ -12,3 +12,4 @@ subdir('geogrid')
 subdir('metgrid')
 subdir('namelist')
 subdir('ungrib')
+subdir('job_scheduler')

wrfrun-0.1.5/.gitattributes

@@ -1,3 +0,0 @@
-.gitea/ export-ignore
-.github/ export-ignore
-images/ export-ignore