wrfrun 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

wrfrun/workspace/core.py

@@ -0,0 +1,92 @@
+"""
+wrfrun.workspace.core
+#####################
+
+Core functions to prepare ``wrfrun`` workspace.
+
+.. autosummary::
+    :toctree: generated/
+
+    prepare_workspace
+"""
+
+from os.path import exists
+from shutil import rmtree
+
+from wrfrun import WRFRUNConfig
+from wrfrun.utils import check_path, logger
+from .wrf import prepare_wrf_workspace, check_wrf_workspace
+
+
+def prepare_workspace():
+    """
+    Initialize ``wrfrun`` workspace.
+
+    This function will check following paths,
+    and create them if them don't exist:
+
+    1. ``/tmp/wrfrun``
+    2. ``$HOME/.config/wrfrun``
+    3. ``$HOME/.config/wrfrun/replay``
+    4. ``$HOME/.config/wrfrun/model``
+
+    It will call other responding functions to initialize workspace for numerical models:
+
+    1. :doc:`WPS/WRF model </api/workspace.wrf>`
+    """
+    logger.info(f"Initialize main workspace.")
+
+    wrfrun_temp_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_TEMP_PATH)
+    workspace_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_WORKSPACE_ROOT)
+    replay_work_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_WORKSPACE_REPLAY)
+    output_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_OUTPUT_PATH)
+
+    if exists(workspace_path):
+        logger.info(f"Remove old files in workspace.")
+        rmtree(workspace_path)
+
+    # check folder
+    check_path(wrfrun_temp_path)
+    check_path(replay_work_path)
+    check_path(output_path)
+
+    func_map = {
+        "wrf": prepare_wrf_workspace
+    }
+    model_configs = WRFRUNConfig["model"]
+
+    for model_name in model_configs:
+        func_map[model_name](model_configs[model_name])
+
+
+def check_workspace() -> bool:
+    """
+    Check if workspace exists.
+
+    :return: ``True`` if workspace exists, ``False`` otherwise.
+    :rtype: bool
+    """
+
+    wrfrun_temp_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_TEMP_PATH)
+    workspace_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_WORKSPACE_ROOT)
+    replay_work_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_WORKSPACE_REPLAY)
+    output_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_OUTPUT_PATH)
+
+    flag = True
+    flag = flag & exists(wrfrun_temp_path) & exists(replay_work_path) & exists(output_path) & exists(workspace_path)
+
+    func_map = {
+        "wrf": check_wrf_workspace
+    }
+    model_configs = WRFRUNConfig["model"]
+
+    for model_name in model_configs:
+        if model_name == "debug_level":
+            continue
+
+        flag = flag & func_map[model_name](model_configs[model_name])
+
+    return True
+
+
+__all__ = ["prepare_workspace", "check_workspace"]

wrfrun/workspace/wrf.py

@@ -0,0 +1,121 @@
+"""
+wrfrun.workspace.wrf
+####################
+
+Functions to prepare workspace for WPS/WRF model.
+
+.. autosummary::
+    :toctree: generated/
+
+    prepare_wrf_workspace
+    check_wrf_workspace
+"""
+
+from os import listdir, makedirs, symlink
+from os.path import exists
+
+from wrfrun.core import WRFRUNConfig
+from wrfrun.utils import logger, check_path
+
+
+WORKSPACE_MODEL_WPS = f"{WRFRUNConfig.WRFRUN_WORKSPACE_MODEL}/WPS"
+WORKSPACE_MODEL_WRF = f"{WRFRUNConfig.WRFRUN_WORKSPACE_MODEL}/WRF"
+WORKSPACE_MODEL_WRFDA = f"{WRFRUNConfig.WRFRUN_WORKSPACE_MODEL}/WRFDA"
+
+
+def prepare_wrf_workspace(model_config: dict):
+    """
+    Initialize workspace for WPS/WRF model.
+
+    This function will check following paths,
+    and create them or delete old files inside:
+
+    1. ``$HOME/.config/wrfrun/model/WPS``
+    2. ``$HOME/.config/wrfrun/model/WRF``
+    3. ``$HOME/.config/wrfrun/model/WRFDA``
+
+    :param model_config: Model config.
+    :type model_config: dict
+    """
+    logger.info(f"Initialize workspace for WPS/WRF.")
+
+    wps_path = model_config["wps_path"]
+    wrf_path = model_config["wrf_path"]
+    wrfda_path = model_config["wrfda_path"]
+
+    if not (wps_path and wrf_path):
+        logger.warning(f"No WPS/WRF model installation path given, skip initialization.")
+        return
+
+    if wps_path:
+        if not exists(wps_path):
+            logger.error(f"Your WPS path is wrong.")
+            raise FileNotFoundError(f"Your WPS path is wrong.")
+
+        wps_work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WPS)
+        check_path(wps_work_path, f"{wps_work_path}/outputs", force=True)
+
+        file_list = [x for x in listdir(wps_path) if x not in ["geogrid", "namelist.wps"]]
+        _ = [symlink(f"{wps_path}/{file}", f"{wps_work_path}/{file}") for file in file_list]
+        makedirs(f"{wps_work_path}/geogrid")
+        symlink(f"{wps_path}/geogrid/GEOGRID.TBL", f"{wps_work_path}/geogrid/GEOGRID.TBL")
+
+    if wrf_path:
+        if not exists(wrf_path):
+            logger.error(f"Your WRF path is wrong.")
+            raise FileNotFoundError(f"Your WRF path is wrong.")
+
+        wrf_work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WRF)
+        check_path(wrf_work_path, force=True)
+
+        file_list = [x for x in listdir(f"{wrf_path}/run") if not x.startswith("namelist")]
+        _ = [symlink(f"{wrf_path}/run/{file}", f"{wrf_work_path}/{file}") for file in file_list]
+
+    if wrfda_path:
+        if not exists(wrfda_path):
+            logger.error(f"Your WRFDA path is wrong.")
+            raise FileNotFoundError(f"Your WRFDA path is wrong.")
+
+        wrfda_work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WRFDA)
+        check_path(wrfda_work_path, force=True)
+
+        file_list = ["da_wrfvar.exe", "da_update_bc.exe"]
+        _ = [symlink(f"{wrfda_path}/var/build/{file}", f"{wrfda_work_path}/{file}") for file in file_list]
+
+        file_list = listdir(f"{wrfda_path}/var/run")
+        _ = [symlink(f"{wrfda_path}/var/run/{file}", f"{wrfda_work_path}/{file}") for file in file_list]
+
+        symlink(f"{wrfda_path}/run/LANDUSE.TBL", f"{wrfda_work_path}/LANDUSE.TBL")
+
+
+def check_wrf_workspace(model_config: dict) -> bool:
+    """
+    Check if WPS/WRF workspace exists.
+
+    :param model_config: Model config.
+    :type model_config: dict
+    :return: ``True`` if WPS/WRF workspace exists, ``False`` otherwise.
+    :rtype: bool
+    """
+    wps_path = model_config["wps_path"]
+    wrf_path = model_config["wrf_path"]
+    wrfda_path = model_config["wrfda_path"]
+
+    flag = True
+
+    if wps_path:
+        wps_work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WPS)
+        flag = flag & exists(wps_work_path)
+
+    if wrf_path:
+        wrf_work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WRF)
+        flag = flag & exists(wrf_work_path)
+
+    if wrfda_path:
+        wrfda_work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WRFDA)
+        flag = flag & exists(wrfda_work_path)
+
+    return flag
+
+
+__all__ = ["prepare_wrf_workspace", "WORKSPACE_MODEL_WRF", "WORKSPACE_MODEL_WPS", "WORKSPACE_MODEL_WRFDA", "check_wrf_workspace"]

{wrfrun-0.1.7.dist-info → wrfrun-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wrfrun
-Version: 0.1.7
+Version: 0.1.9
 Summary: wrfrun is a comprehensive toolkit for managing and using WRF
 Keywords: WRF
 Author-Email: Syize <syizeliu@gmail.com>
@@ -10,8 +10,8 @@ Project-URL: homepage, https://github.com/Syize/wrfrun
 Project-URL: repository, https://github.com/Syize/wrfrun
 Project-URL: documentation, https://wrfrun.syize.cn
 Project-URL: Bug Tracker, https://github.com/Syize/wrfrun/issues
-Requires-Python: <3.13
-Requires-Dist: numpy<2.0.0
+Requires-Python: >=3.10
+Requires-Dist: numpy
 Requires-Dist: xarray
 Requires-Dist: netCDF4
 Requires-Dist: rich
@@ -24,6 +24,7 @@ Requires-Dist: wrf-python
 Requires-Dist: cfgrib
 Requires-Dist: tomli
 Requires-Dist: tomli-w
+Requires-Dist: haversine
 Description-Content-Type: text/markdown
 
 # wrfrun: A toolkit to control WRF

wrfrun-0.1.9.dist-info/RECORD

@@ -0,0 +1,62 @@
+wrfrun-0.1.9.dist-info/METADATA,sha256=RPtBDS1Ctr0OatYSqezZsgRqGj621sS-dijdt4oN2Vk,2932
+wrfrun-0.1.9.dist-info/WHEEL,sha256=5J4neoE7k6LMgx4Fz1FHgBiO3YevhJGtNQ3muDrdLQM,75
+wrfrun-0.1.9.dist-info/entry_points.txt,sha256=G3358n6wW1qZF3hSH9umxrWHJewN-6N1BUloacN2tks,50
+wrfrun/__init__.py,sha256=l2zGk2qDWa3EXIlwvzqBYZHOU9273TN75Rmvx_7PAYM,58
+wrfrun/cli.py,sha256=xaMxiHQCHvgHWKlPavYXkC9hGIo7JpUZaIsV8ZKr_SM,4300
+wrfrun/data.py,sha256=W6IhopHpp0lhvVo-beuO6NsbfsGDmN5MNf8Bqkt7xsA,14845
+wrfrun/run.py,sha256=aYRQ-YqjIt5g4MQEnphmZIwanT9TBlYCddcyXxtgD_s,8918
+wrfrun/utils.py,sha256=BdGP5u2Y2e3tqU0dawfHsTeqZ0q4Ic4MCGR8RU4FJ90,8388
+wrfrun/core/__init__.py,sha256=zsitfcmbO_Qf_Uh0FzZ4rceZAo3WEYev32jSjgRjtkw,1325
+wrfrun/core/base.py,sha256=LXtol0AQlDiZYMZeQT3qRZVcQ-F0ETE7sMXdI1jriLw,30324
+wrfrun/core/config.py,sha256=gljrE1j8qFNJb_VVJwhQ3E2pmapT9cFUg1pyWizEvjA,27457
+wrfrun/core/error.py,sha256=etnImryyqDUggfZOKK1ynOm71GRSFJVGGOwhuiZahTI,2383
+wrfrun/core/replay.py,sha256=jZWN68IgUroxuCzg7EO0FXxe2zwokxJnquu4t81hp7Y,4782
+wrfrun/core/server.py,sha256=Kns4an_wqnrYsi0PJxJgQJY8ShjNDnvrxY5V7gsbDec,8660
+wrfrun/extension/__init__.py,sha256=YDYCS0KD3e-uLvf1K08ktNfP2sUSKpbgkHrvXj21EPQ,913
+wrfrun/extension/utils.py,sha256=mmcyqxOEVGSBu8t8ukB2SeRp9X_PLR0kWvfptBC_WWo,2458
+wrfrun/extension/goos_sst/__init__.py,sha256=QsVheVVL5K0_k-eg8-xe2MLAJ845iivXzGHbp2qARh4,2101
+wrfrun/extension/goos_sst/core.py,sha256=_s53vq0nAwnwWxExD8-73p5seCWLO1XVzRh5T8JofAw,3870
+wrfrun/extension/goos_sst/utils.py,sha256=bomXbhpOuNOYm2rL1SiOeSxVbSaKWfVxX51KoODgBn0,3469
+wrfrun/extension/goos_sst/res/__init__.py,sha256=-Ey6iFHzQYZNOGhVrSkMjOwiXVgt3P6eaumq7pOmh_E,511
+wrfrun/extension/goos_sst/res/Vtable.ERA_GOOS_SST,sha256=UfbzWKEE2pnQO4Q2mkMmkl7pv_bDSxVGaN2jEYfMVzE,534
+wrfrun/extension/littler/__init__.py,sha256=bLiGIpB_FpTyio44CjIzwwcahyFWxGW-gECfBOIG9Y8,2032
+wrfrun/extension/littler/core.py,sha256=0Si4phAUiCwgjEQnDfWKSOvtnyitWYmrDWkB6_2VjUg,34173
+wrfrun/model/__init__.py,sha256=PYA6ZUwL4u3heRIMX_qkLvAA6jDrY4kugD8rCvnl-2Y,803
+wrfrun/model/base.py,sha256=WnVNu12ICJgThfZR7kK6_XLCQWVYFsHrdyAz8wl_YRc,218
+wrfrun/model/plot.py,sha256=SvD5_2sLaLqYvuvcZVX1x3XTrgMW_bsGP-Npr6RKHVU,10335
+wrfrun/model/utils.py,sha256=5_ffMJCPi_n3-Mgaci9F2eMk2fY7_UrcjmK0JSsZZr0,1206
+wrfrun/model/wrf/__init__.py,sha256=KJij1JMTcj3i5e2LcJJJudA99LrsJIVHeMj9ktT9Y_4,1752
+wrfrun/model/wrf/core.py,sha256=Ykw6ElY3mUelWOThDFT--ChnwY7zgcbyooLuJNhGqV4,35627
+wrfrun/model/wrf/exec_wrap.py,sha256=REOtNLCc6IcM2yaaSXoXGgXFA-xjMHmoTiA9B3ZezEA,4418
+wrfrun/model/wrf/geodata.py,sha256=X9OUOm0__5NI7sl2z1F9_ur5wZ4P0HkpBNcRZQYKof0,9740
+wrfrun/model/wrf/namelist.py,sha256=ksxHlh-LtxDBF73cCe8J0wpyGrSvM1dBgPgwK2XseGo,16918
+wrfrun/model/wrf/scheme.py,sha256=8y85Dbu-GajwjHr3Spw8_tN21xAloD-SnmxJd11nXKE,11254
+wrfrun/model/wrf/utils.py,sha256=DuKghfqBT77p_beDKJ3EyyPMuRmBJCgCElvs3YheWak,3709
+wrfrun/model/wrf/vtable.py,sha256=1el_TctQfPVTYnZn0nJNgfDvVXy6q5ZzVIyUgWRfFUs,2478
+wrfrun/plot/__init__.py,sha256=9Kn0IgkX10sHEqHJwk7mZV-dP14XMNcvXN8znO89FIw,19
+wrfrun/plot/wps.py,sha256=pvkxbh5760AAM4KaPteMzhFniZZFtkYF31xhzSBa7lo,5552
+wrfrun/res/__init__.py,sha256=cIjX9Was6nhms-aBLUY0V1393of-LzoSRcWcD_HGgn0,1437
+wrfrun/res/run.template.sh,sha256=k-r4lOOarscmSdiBXGHPnv3oeiRe-qW-VhOBia27ZGU,101
+wrfrun/res/config/config.template.toml,sha256=DjtFjy2sL_P6w63XfLDkmDWhWKyo6qrlwBy69cnbvXA,1378
+wrfrun/res/config/wrf.template.toml,sha256=qQS3OHAXBMaKWgj2qT7jLeDu6tyngPgin7Q-XcE3ZkQ,3189
+wrfrun/res/extension/plotgrids.ncl,sha256=B0mvH1H1j_w7EEana9HmU1XJZtG0w9IccQ54zXpbQG0,7546
+wrfrun/res/namelist/namelist.input.da_wrfvar.template,sha256=Cwc-XPu_spJeQte4duWrulPBOLRMEBtn0mIn0pgMmKY,4912
+wrfrun/res/namelist/namelist.input.dfi.template,sha256=E7MVbIvMopkAM7xGUC4vniC1WOUVbIbdLfkTKHqzX_o,6591
+wrfrun/res/namelist/namelist.input.real.template,sha256=DDUiArtBFmzBwVjkZW4NOrBR34eIOk1vV0VsyL0stsk,6214
+wrfrun/res/namelist/namelist.input.wrf.template,sha256=myrKi79sQ8ABBsmQJkcgN0WLbWJdtMVPIjuAC1MTMuM,6213
+wrfrun/res/namelist/namelist.wps.template,sha256=HlA7-SHs4C-cKRb3h6D0Kl1Y-5VSJ1Lw9hLiXWGAN0Q,1017
+wrfrun/res/namelist/parame.in.template,sha256=vR8JSix20FAKGqj6jK8QuEAeWkGvg8_iptdVlzjPX6o,259
+wrfrun/res/scheduler/lsf.template,sha256=KorLfqTFTqTTctDE9Edgkixi7JLWrcDNRpZknXuxbEQ,111
+wrfrun/res/scheduler/pbs.template,sha256=kP4pGMAq2uz0OZUAWii7dO4ECLC924c8SnciFeBL5QY,155
+wrfrun/res/scheduler/slurm.template,sha256=sQNjkKzOftipDD4kYmzhZkh8fg0M9uaQsdLjDakCVwA,336
+wrfrun/scheduler/__init__.py,sha256=pOqnCAl2-_yC7zPS2VQL5RMltQN4zc0koYS_MESKtHo,1130
+wrfrun/scheduler/env.py,sha256=N_K5yZFRXMEXvW3yjSideEDPWZzoWpbRhFS9LJoZ3R8,1262
+wrfrun/scheduler/lsf.py,sha256=E_sKUk4_RZMBSkHctzDi2sEAy8UrteZvgpbuU2zeSvo,1158
+wrfrun/scheduler/pbs.py,sha256=fqGqoIWG4lh2bT_NqxPxL3XIdPo8keWG2MFqZmOMKpI,1174
+wrfrun/scheduler/script.py,sha256=si2pzrKCOutICcke4t1c5kmCnVlTS9PZJF5MJTp3YTk,2403
+wrfrun/scheduler/slurm.py,sha256=Y3a4MAt5omDgVUtONaL2ajV609HbOZl2opjIGkHUa7g,1196
+wrfrun/scheduler/utils.py,sha256=5BGPEB-oMn4ifo3XZOkDOwAvjWXi69UHLR5HMabwUAM,206
+wrfrun/workspace/__init__.py,sha256=e7ijFWaYRMmBbso0pXnsAc0oUnkQ3t3UguhY3a0U6Qk,1456
+wrfrun/workspace/core.py,sha256=wj9f8XzafazpxXuO_wtConOEhkHwJLyPTwhubXrn20s,2617
+wrfrun/workspace/wrf.py,sha256=wp5pQgKB9B-7JGc2psRIyj_pI7iG344jphBBB0b6PsY,4098
+wrfrun-0.1.9.dist-info/RECORD,,

wrfrun-0.1.9.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+wrfrun = wrfrun.cli:main_entry
+

wrfrun/model/wrf/_ndown.py

@@ -1,39 +0,0 @@
-from wrfrun.core import WRFRUNConfig
-from wrfrun.utils import logger
-
-
-def process_after_ndown():
-    """
-    After running ndown.exe, namelist settings are supposed to be changed,
-    so WRF can simulate a higher resolution domain according to `WRF User's Guide <https://www2.mmm.ucar.edu/wrf/users/wrf_users_guide/build/html/running_wrf.html#wrf-nesting>`_.
-    `wrfrun` provide this function to help you change these settings which have multiple values for each domain.
-    The first value will be removed to ensure the value of higher resolution domain is the first value.
-
-    :return:
-    """
-    namelist_data = WRFRUNConfig.get_namelist("wrf")
-
-    for section in namelist_data:
-        if section in ["bdy_control", "namelist_quilt"]:
-            continue
-
-        for key in namelist_data[section]:
-            if key in ["grid_id", "parent_id", "i_parent_start", "j_parent_start", "parent_grid_ratio", "parent_time_step_ratio", "eta_levels"]:
-                continue
-
-            if isinstance(namelist_data[section][key], list):
-
-                if len(namelist_data[section][key]) > 1:
-                    namelist_data[section][key] = namelist_data[section][key][1:]
-
-    namelist_data["domains"]["max_dom"] = 1
-
-    time_ratio = namelist_data["domains"]["parent_time_step_ratio"][1]
-    namelist_data["domains"]["time_step"] = namelist_data["domains"]["time_step"] // time_ratio
-
-    WRFRUNConfig.update_namelist(namelist_data, "wrf")
-
-    logger.info(f"Update namelist after running ndown.exe")
-
-
-__all__ = ["process_after_ndown"]

wrfrun/pbs.py

@@ -1,86 +0,0 @@
-from os import environ
-from os.path import exists, dirname, abspath
-
-from .core import WRFRUNConfig
-from .res import TASKSYS_PBS_TEMPLATE
-from .utils import logger
-
-
-def prepare_pbs_script(main_file_path: str):
-    """Prepare the bash script to be submitted to PBS work system.
-
-    Args:
-        main_file_path (str): The path of main Python file.
-    """
-    # check main file
-    if not exists(main_file_path):
-        logger.error(f"Wrong path of main Python file: {main_file_path}")
-        raise FileNotFoundError
-
-    # get absolute path of parent directory
-    dir_path = abspath(dirname(main_file_path))
-
-    # read log path and PBS setting from config
-    log_path = WRFRUNConfig.get_log_path()
-    PBS_setting = WRFRUNConfig.get_job_scheduler_config()
-
-    # set PBS log path
-    stdout_log_path = f"{log_path}/PBS.log"
-    stderr_log_path = f"{log_path}/PBS.log"
-
-    # set environment parameter
-    env_settings = ''
-    if len(PBS_setting["env_settings"]) != 0:
-        for key in PBS_setting["env_settings"]:
-            env_settings += f"{key}={PBS_setting['env_settings'][key]}\n"
-
-    # set command
-    exec_cmd = f"{PBS_setting['python_interpreter']} {main_file_path}"
-
-    # read template and write to file
-    pbs_template_path = WRFRUNConfig.parse_resource_uri(TASKSYS_PBS_TEMPLATE)
-    with open(f"{dir_path}/run.sh", "w") as f:
-
-        with open(pbs_template_path, "r") as f_template:
-            template = f_template.read()
-
-        template = template.format(
-            STDOUT_LOG_PATH=stdout_log_path,
-            STDERR_LOG_PATH=stderr_log_path,
-            NODE_NUM=PBS_setting["node_num"],
-            CORE_NUM=PBS_setting["core_num"],
-            ENV_SETTINGS=env_settings,
-            WORK_PATH=dir_path,
-            WORK_COMMAND=exec_cmd
-        )
-
-        f.write(template)
-
-    logger.info(
-        f"PBS script has been generated and write to file {dir_path}/run.sh. Check it and submit it to PBS system to run wrfrun.")
-
-
-def get_core_num() -> int:
-    """Read core num from config.
-
-    Returns:
-        int: Core number.
-    """
-    return WRFRUNConfig["core_num"]
-
-
-def in_pbs() -> bool:
-    """Check if we're in a PBS work.
-
-    Returns:
-        bool: True if we're in, False is we aren't.
-    """
-    # check if we're a PBS task
-    if "PBS_ENVIRONMENT" in environ:
-        # we're in a PBS task
-        return True
-    else:
-        return False
-
-
-__all__ = ["prepare_pbs_script", "get_core_num", "in_pbs"]

wrfrun/res/run.sh.template

@@ -1,16 +0,0 @@
-#!/bin/bash
-#PBS -q batch
-##PBS -q odasc
-#PBS -N py-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
-
-ulimit -s unlimited
-
-{ENV_SETTINGS}
-
-cd {WORK_PATH}
-
-{WORK_COMMAND}

wrfrun/workspace.py

@@ -1,88 +0,0 @@
-"""
-This file contains functions to interact with WRF workspace
-"""
-
-from os import listdir, makedirs, symlink
-from os.path import exists
-
-from .core import WRFRUNConfig
-from .utils import check_path, logger
-
-
-def prepare_workspace():
-    """Initialize workspace
-
-    """
-    logger.info(f"Initialize workspace...")
-
-    # extract WRF path
-    wrf_config = WRFRUNConfig.get_model_config("wrf")
-    wps_path = wrf_config["wps_path"]
-    wrf_path = wrf_config["wrf_path"]
-    wrfda_path = wrf_config["wrfda_path"]
-
-    WRFRUN_TEMP_PATH = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_TEMP_PATH)
-    WORK_PATH = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_WORKSPACE_PATH)
-    REPLAY_WORK_PATH = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_REPLAY_WORK_PATH)
-    WPS_WORK_PATH = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WPS_WORK_PATH)
-    WRF_WORK_PATH = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRF_WORK_PATH)
-    WRFDA_WORK_PATH = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFDA_WORK_PATH)
-    output_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_OUTPUT_PATH)
-
-    # check folder
-    check_path(WRFRUN_TEMP_PATH)
-    check_path(REPLAY_WORK_PATH, force=True)
-    check_path(output_path)
-
-    if exists(wrfda_path):
-        WRFRUNConfig.USE_WRFDA = True
-
-    # create folder to run WPS, and WRF
-    # check the path
-    if not (exists(wps_path) and exists(wrf_path)):
-        logger.error(f"Your WPS or WRF path is wrong")
-        raise FileNotFoundError(f"Your WPS or WRF path is wrong")
-
-    # remove old file
-    if exists(WORK_PATH):
-        logger.info(f"Remove old files...")
-    check_path(WPS_WORK_PATH, f"{WPS_WORK_PATH}/outputs",
-               WRF_WORK_PATH, WRFDA_WORK_PATH, force=True)
-    logger.info(f"Link essential files...")
-
-    # link {wps_path}/*
-    # collect file except folder geogrid
-    file_list = [
-        x for x in listdir(wps_path) if x not in ["geogrid", "namelist.wps"]
-    ]
-    for file in file_list:
-        symlink(f"{wps_path}/{file}", f"{WPS_WORK_PATH}/{file}")
-
-    # create folder geogrid and link default GEOGRID file
-    makedirs(f"{WPS_WORK_PATH}/geogrid")
-    symlink(
-        f"{wps_path}/geogrid/GEOGRID.TBL", f"{WPS_WORK_PATH}/geogrid/GEOGRID.TBL"
-    )
-
-    # # link {wrf_path}/run/*, except namelist.input
-    file_list = [x for x in listdir(
-        f"{wrf_path}/run") if not x.startswith("namelist")]
-    for file in file_list:
-        symlink(f"{wrf_path}/run/{file}", f"{WRF_WORK_PATH}/{file}")
-
-    if WRFRUNConfig.USE_WRFDA:
-        # # link {wrfda_path}/bin/*.exe
-        file_list = ["da_wrfvar.exe", "da_update_bc.exe"]
-        for file in file_list:
-            symlink(f"{wrfda_path}/var/build/{file}", f"{WRFDA_WORK_PATH}/{file}")
-
-        # # link {wrfda_path}/var/run/*
-        file_list = listdir(f"{wrfda_path}/var/run")
-        for file in file_list:
-            symlink(f"{wrfda_path}/var/run/{file}", f"{WRFDA_WORK_PATH}/{file}")
-
-        # # link {wrfda_path}/run/LANDUSE.TBL
-        symlink(f"{wrfda_path}/run/LANDUSE.TBL", f"{WRFDA_WORK_PATH}/LANDUSE.TBL")
-
-
-__all__ = ["prepare_workspace"]

wrfrun-0.1.7.dist-info/RECORD

@@ -1,46 +0,0 @@
-wrfrun-0.1.7.dist-info/METADATA,sha256=nrPBmS0UxcjIiCN9wLyL798-r3OKpirBJUPlWrzobw8,2912
-wrfrun-0.1.7.dist-info/WHEEL,sha256=5J4neoE7k6LMgx4Fz1FHgBiO3YevhJGtNQ3muDrdLQM,75
-wrfrun/__init__.py,sha256=l2zGk2qDWa3EXIlwvzqBYZHOU9273TN75Rmvx_7PAYM,58
-wrfrun/data.py,sha256=AjkDSZqkLROnWLw9qDOBUctoQmJ3R1bkfjnPzLMjPfs,14751
-wrfrun/pbs.py,sha256=gzRHbYqpU-c8lynu39TUKNvmPntvrPwfQVxK337yvjw,2454
-wrfrun/run.py,sha256=0mvuI98gLjq1gP1eaLUZiU1LJMCjtL668aKZTwu3n2g,9114
-wrfrun/utils.py,sha256=v3ZRu3Rd8-pZjo7I0PrJtgBN1zIBAt_nVPs0BP_DFwY,8118
-wrfrun/workspace.py,sha256=5tWIIFxwAE4hkdg_wJALGDc3P0bfwb_IRTIyPIjN-tE,3105
-wrfrun/core/__init__.py,sha256=bf9Fp6VS5i9xB10lazYBOkJapF2jnQpyiB6qucVjp0I,107
-wrfrun/core/base.py,sha256=S9Mqr3LOpos0CP0TsFwtCQl0tLIObIhjX7VxjS0LRDY,23172
-wrfrun/core/config.py,sha256=v7zdhAthlMmjv_BWINxkmgNHLvKZdzQhgM5B-Nlsl7M,23795
-wrfrun/core/error.py,sha256=fCWVC4-GTyDH2obzu9-D_bzF8_uzqZoUi7Ffy0B5qoc,1324
-wrfrun/core/replay.py,sha256=oyV9QppZG99GbAksQqoRxeBQ3jjXU-7YqkpxHRyfBvc,3229
-wrfrun/core/server.py,sha256=KK_yj4J-5F_s-rKO5FK7e8WojsnuZMbhWtMceWYbaBY,6524
-wrfrun/extension/__init__.py,sha256=alIDGBnxWH4JvP-UW-7N99seBBi0r1GV1h8f1ERFBec,21
-wrfrun/extension/utils.py,sha256=OBIAlAXakbh1QhrF2BFlYAFV062ZfJjX0mYJkmcU_RI,2751
-wrfrun/extension/littler/__init__.py,sha256=pG-YPVG0gJIJ6s4xcAz9S_CnUxpUcz33wl9eNUSgxGk,20
-wrfrun/extension/littler/utils.py,sha256=TsN3GiAFszPWF7RNM_hYdARRY_OLL8mYm87UyZupUtU,21615
-wrfrun/model/__init__.py,sha256=lQNNnh3evXZytLAjdnhlIEFWerP1L0XAgDAUQhqIRyE,119
-wrfrun/model/base.py,sha256=WnVNu12ICJgThfZR7kK6_XLCQWVYFsHrdyAz8wl_YRc,218
-wrfrun/model/plot.py,sha256=ATG5nl5kV8YfoeNFIPLCUnnIQIrfIFNGPb9d8raM6S0,1582
-wrfrun/model/utils.py,sha256=Pyj9D9NMKw3wQcZcM7pJaLG09Vov447AxFKoVeZZU2s,1155
-wrfrun/model/wrf/__init__.py,sha256=T3S1BD9QDoW4uq871QW4_8-2BvCRWYEecz71EcdbMaI,136
-wrfrun/model/wrf/core.py,sha256=FEzRXUKK5dENxev3wR_GON5n1Zxq4Hj-Bs_BNP7P_ok,32134
-wrfrun/model/wrf/exec_wrap.py,sha256=gIsGZDAjzRzxt24eo8t-laIK4UBiyKf_ysfAZLlE1og,3918
-wrfrun/model/wrf/geodata.py,sha256=X9OUOm0__5NI7sl2z1F9_ur5wZ4P0HkpBNcRZQYKof0,9740
-wrfrun/model/wrf/_metgrid.py,sha256=i2qILM_1LkdnBZqXXL3swVhMRvhFvq4bOcRPGI6kSzg,2248
-wrfrun/model/wrf/namelist.py,sha256=t_JuWYj4oNNVwzTMKo6BNlv2LIq4t3JC7ZgVkaKfNbA,14923
-wrfrun/model/wrf/_ndown.py,sha256=ZcWttGXiT5FzGCAohmbeNU6bi8GUOUleiTK0ivn2BaI,1548
-wrfrun/model/wrf/scheme.py,sha256=F7HxoaNkkPpjkse5t9sGS3CJDvn_yBdHoaCG7WLdbqY,8524
-wrfrun/model/wrf/vtable.py,sha256=0V4fo0Limnx4oJ2NByq_eHPzj8L4mnHNlrKL34Buw1A,2432
-wrfrun/plot/__init__.py,sha256=9Kn0IgkX10sHEqHJwk7mZV-dP14XMNcvXN8znO89FIw,19
-wrfrun/plot/wps.py,sha256=AQ1B6llNUrPU3PR8XKNLSzs_H-v60a0Wep-8Mcto6NM,5319
-wrfrun/res/__init__.py,sha256=YBcVaM2EZZJkDDvyKdk-WZmcFQoTFENZKCwcIQzEf3I,1133
-wrfrun/res/config.toml.template,sha256=aVXCSzhbV3P-MQCZOk2g-y0auRmqgr3jW920W3wks9g,4386
-wrfrun/res/run.sh.template,sha256=ZN7k0a9ONOtZUQ67JiTva-0DNVJG5O7BdOXQ8-LCscs,247
-wrfrun/res/extension/plotgrids.ncl,sha256=B0mvH1H1j_w7EEana9HmU1XJZtG0w9IccQ54zXpbQG0,7546
-wrfrun/res/namelist/namelist.input.da_wrfvar.template,sha256=Cwc-XPu_spJeQte4duWrulPBOLRMEBtn0mIn0pgMmKY,4912
-wrfrun/res/namelist/namelist.input.dfi.template,sha256=E7MVbIvMopkAM7xGUC4vniC1WOUVbIbdLfkTKHqzX_o,6591
-wrfrun/res/namelist/namelist.input.real.template,sha256=DDUiArtBFmzBwVjkZW4NOrBR34eIOk1vV0VsyL0stsk,6214
-wrfrun/res/namelist/namelist.input.wrf.template,sha256=myrKi79sQ8ABBsmQJkcgN0WLbWJdtMVPIjuAC1MTMuM,6213
-wrfrun/res/namelist/namelist.wps.template,sha256=HlA7-SHs4C-cKRb3h6D0Kl1Y-5VSJ1Lw9hLiXWGAN0Q,1017
-wrfrun/res/namelist/parame.in.template,sha256=vR8JSix20FAKGqj6jK8QuEAeWkGvg8_iptdVlzjPX6o,259
-wrfrun/res/job_scheduler/pbs.template,sha256=m7eyvZkzQRpjs43CvSeZ1biHYNyEJJkgXt3IU5ZhCmg,148
-wrfrun/res/job_scheduler/slurm.template,sha256=BrKL8DcHfpFfv0y0z7Qe_cqjwfzMyF-3V6vM72ehr9Y,312
-wrfrun-0.1.7.dist-info/RECORD,,