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

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.8.dist-info/RECORD

@@ -1,51 +0,0 @@
-wrfrun-0.1.8.dist-info/METADATA,sha256=_Y0huf7Q8hAsLFsgIlPZYF8FDmZ_DKqsGMxcBaewcpQ,2913
-wrfrun-0.1.8.dist-info/WHEEL,sha256=5J4neoE7k6LMgx4Fz1FHgBiO3YevhJGtNQ3muDrdLQM,75
-wrfrun/__init__.py,sha256=l2zGk2qDWa3EXIlwvzqBYZHOU9273TN75Rmvx_7PAYM,58
-wrfrun/data.py,sha256=W6IhopHpp0lhvVo-beuO6NsbfsGDmN5MNf8Bqkt7xsA,14845
-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=zsitfcmbO_Qf_Uh0FzZ4rceZAo3WEYev32jSjgRjtkw,1325
-wrfrun/core/base.py,sha256=zix413Nc0YrgIR6oakVhTN405KUmZ7OKlJTgL0fNUVc,30307
-wrfrun/core/config.py,sha256=g021WEkRZpvsLyzVHlahL9WtAP7zKG9polW3B84eT3k,29252
-wrfrun/core/error.py,sha256=etnImryyqDUggfZOKK1ynOm71GRSFJVGGOwhuiZahTI,2383
-wrfrun/core/replay.py,sha256=nqVmR3Hb4zO6PnROssv3PKw7gvvdi-UqIsjX-NV1WLI,4782
-wrfrun/core/server.py,sha256=RjzFVlRZg60rLMy4Gy5UhZWPUOQP307rT4ttpvbcecQ,8640
-wrfrun/extension/__init__.py,sha256=YDYCS0KD3e-uLvf1K08ktNfP2sUSKpbgkHrvXj21EPQ,913
-wrfrun/extension/utils.py,sha256=OBIAlAXakbh1QhrF2BFlYAFV062ZfJjX0mYJkmcU_RI,2751
-wrfrun/extension/goos_sst/__init__.py,sha256=M8LWGLXqHyPUWijl5925jFmwJ1ih3kHfT2jTxZzwMBA,2126
-wrfrun/extension/goos_sst/core.py,sha256=D0po6PjBqcEareVQO1VjdIUrFz9lJrnAZ5q2sax78Tw,3794
-wrfrun/extension/goos_sst/utils.py,sha256=h0OBCwxS67kG6YNclZLvyPYnT7uNZd3Uq9f6Vg2EaMc,3052
-wrfrun/extension/goos_sst/res/__init__.py,sha256=M-qN-WI0EIvhYo2hwnnIK_fXL3TbR12KzSC9fvjgDaY,164
-wrfrun/extension/goos_sst/res/Vtable.ERA_GOOS_SST,sha256=aX8RSTWL8GVdxf2TgvIoQx5xDVuLi0F5_Eg2e8Gaa-E,529
-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=Hh-GxQPDzpwAEAq-rFh0VcIWzgMDxZVT0c40SIRa6_o,1582
-wrfrun/model/utils.py,sha256=Pyj9D9NMKw3wQcZcM7pJaLG09Vov447AxFKoVeZZU2s,1155
-wrfrun/model/wrf/__init__.py,sha256=T3S1BD9QDoW4uq871QW4_8-2BvCRWYEecz71EcdbMaI,136
-wrfrun/model/wrf/core.py,sha256=srOD7KFF5TgPAcWTw0PLHFjkGjDzb8PYbxuPx5B4bQE,32140
-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=T9b5H6pCljhKUhoR9qwTbdMF04Hrdp3mHwCP-VuhFWE,14791
-wrfrun/model/wrf/_ndown.py,sha256=ZcWttGXiT5FzGCAohmbeNU6bi8GUOUleiTK0ivn2BaI,1548
-wrfrun/model/wrf/scheme.py,sha256=8y85Dbu-GajwjHr3Spw8_tN21xAloD-SnmxJd11nXKE,11254
-wrfrun/model/wrf/vtable.py,sha256=0V4fo0Limnx4oJ2NByq_eHPzj8L4mnHNlrKL34Buw1A,2432
-wrfrun/plot/__init__.py,sha256=9Kn0IgkX10sHEqHJwk7mZV-dP14XMNcvXN8znO89FIw,19
-wrfrun/plot/wps.py,sha256=pvkxbh5760AAM4KaPteMzhFniZZFtkYF31xhzSBa7lo,5552
-wrfrun/res/__init__.py,sha256=YBcVaM2EZZJkDDvyKdk-WZmcFQoTFENZKCwcIQzEf3I,1133
-wrfrun/res/config.toml.template,sha256=LTCEwXSFF2QXnwGRoB0Jw5Qs_vXwdb6Zd7fQtNAz9Uw,4374
-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.8.dist-info/RECORD,,