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

wrfrun/data.py

@@ -1,14 +1,15 @@
 from datetime import datetime
 from os import makedirs
-from os.path import exists, dirname
-from typing import Union, List, Tuple
+from os.path import dirname, exists
+from typing import List, Tuple, Union
 
 from pandas import date_range
-# from seafog import goos_sst_find_data
 
-from .core.config import WRFRUNConfig
+from .core import get_wrfrun_config
 from .utils import logger
 
+# from seafog import goos_sst_find_data
+
 # lazy initialize
 CDS_CLIENT = None
 
@@ -264,8 +265,7 @@ def find_era5_data(date: Union[List[str], List[datetime]], area: Tuple[int, int,
     # check if we need to add pressure_level to params dict
     if dataset == ERA5CONFIG.DATASET_ERA5_PRESSURE_LEVEL:
         if pressure_level is None:
-            logger.error(
-                f"You need to provide pressure levels to download data")
+            logger.error("You need to provide pressure levels to download data")
             exit(1)
         # convert value to str
         if not isinstance(pressure_level, list):
@@ -277,12 +277,11 @@ def find_era5_data(date: Union[List[str], List[datetime]], area: Tuple[int, int,
         if _check_pressure_level(pressure_level):   # type: ignore
             params_dict["pressure_level"] = pressure_level
         else:
-            logger.error(
-                f"You have passed wrong pressure level to download data, check it")
+            logger.error("You have passed wrong pressure level to download data, check it")
             exit(1)
 
     # download data
-    logger.info(f"Downloading data to {save_path}, it may take several tens of minutes, please wait...")
+    logger.info(f"Downloading data to '{save_path}', it may take several tens of minutes, please wait...")
 
     if CDS_CLIENT is None:
         import cdsapi
@@ -299,29 +298,28 @@ def prepare_wps_input_data(area: Tuple[int, int, int, int]):
     Args:
         area (Tuple[int, int, int, int]): Range of longitude and latitude, `[lon1, lon2, lat1, lat2]`.
     """
-    wrf_config = WRFRUNConfig.get_model_config("wrf")
+    wrf_config = get_wrfrun_config().get_model_config("wrf")
     # get start and end date from config
     start_date = wrf_config["time"]["start_date"]
     end_date = wrf_config["time"]["end_date"]
 
     # remove second part
-    start_date = start_date[:-3]
-    end_date = end_date[:-3]
+    start_date = start_date.strftime("%Y-%m-%d %H:%M")
+    end_date = end_date.strftime("%Y-%m-%d %H:%M")
 
     # get hour step
     hour_step = wrf_config["time"]["input_data_interval"] // 3600
 
     # get data save path
-    bg_save_path = wrf_config["wps_input_data_folder"]
-    sst_save_path = wrf_config["near_goos_data_folder"]
+    data_save_path = get_wrfrun_config().get_input_data_path()
 
     # download data
     logger.info(f"Download background data of surface level...")
-    download_data(start_date, end_date, hour_step, area, f"{bg_save_path}/surface.grib",
+    download_data(start_date, end_date, hour_step, area, f"{data_save_path}/surface.grib",
                   data_format="grib", data_type="surface", overwrite=True)
 
     logger.info(f"Download background data of pressure level...")
-    download_data(start_date, end_date, hour_step, area, f"{bg_save_path}/pressure.grib",
+    download_data(start_date, end_date, hour_step, area, f"{data_save_path}/pressure.grib",
                   data_format="grib", data_type="pressure", overwrite=True)
 
     # logger.info(f"Download NearGOOS data...")

wrfrun/extension/littler/core.py

@@ -20,7 +20,7 @@ from zipfile import ZipFile
 from pandas import DataFrame, read_csv
 import numpy as np
 
-from wrfrun.core import WRFRUNConfig
+from wrfrun.core import get_wrfrun_config
 from wrfrun.utils import logger
 
 
@@ -849,7 +849,7 @@ class LittleR(LittleRData):
         if not file_path.endswith(".zlr"):
             file_path = f"{file_path}.zlr"
 
-        file_path = WRFRUNConfig.parse_resource_uri(file_path)
+        file_path = get_wrfrun_config().parse_resource_uri(file_path)
 
         with ZipFile(file_path, "w") as zip_file:
             with zip_file.open("header", "w") as header_file:
@@ -868,7 +868,7 @@ class LittleR(LittleRData):
         :return: ``LittleR`` instance.
         :rtype: LittleR
         """
-        file_path = WRFRUNConfig.parse_resource_uri(file_path)
+        file_path = get_wrfrun_config().parse_resource_uri(file_path)
 
         with ZipFile(file_path, "r") as zip_file:
             with zip_file.open("header", "r") as header_file:

wrfrun/extension/utils.py

@@ -15,7 +15,7 @@ from os.path import exists
 from shutil import copyfile
 from typing import List, Optional
 
-from wrfrun.core import WRFRUNConfig
+from wrfrun.core import get_wrfrun_config
 from wrfrun.utils import check_path, logger
 
 
@@ -23,7 +23,7 @@ def extension_postprocess(output_dir: str, extension_id: str, outputs: Optional[
     """
     This function provides a unified postprocessing interface for all extensions.
 
-    This function will save outputss and logs in ``output_dir`` to the ``{output_path}/extension_id`` and ``{output_path}/extension_id/logs``,
+    This function will save outputs and logs in ``output_dir`` to the ``{output_path}/extension_id`` and ``{output_path}/extension_id/logs``,
     in which ``output_path`` is defined in ``config.toml``.
     Files end with ``.log`` are treated as log files, while others are treated as outputs.
     You can save specific outputs by giving their names through the argument ``outputs``.
@@ -37,6 +37,7 @@ def extension_postprocess(output_dir: str, extension_id: str, outputs: Optional[
     :param outputs: A list contains multiple filenames. Files in this will be treated as outputs.
     :type outputs: list
     """
+    WRFRUNConfig = get_wrfrun_config()
     output_path = WRFRUNConfig.WRFRUN_OUTPUT_PATH
     output_save_path = f"{output_path}/{extension_id}"
     log_save_path = f"{output_path}/{extension_id}/logs"

wrfrun/model/plot.py

@@ -9,7 +9,7 @@ from cartopy.mpl.geoaxes import GeoAxes
 from cartopy.mpl.gridliner import LATITUDE_FORMATTER, LONGITUDE_FORMATTER
 from haversine.haversine import Direction, Unit, inverse_haversine
 
-from wrfrun.core import WRFRUNConfig
+from wrfrun.core import get_wrfrun_config
 from wrfrun.utils import check_path, logger
 
 
@@ -183,7 +183,7 @@ def plot_domain_area(
             logger.error("You need to give 'model_name' if `domain_settings == None`")
             raise ValueError("You need to give 'model_name' if `domain_settings == None`")
 
-        user_settings = WRFRUNConfig.get_model_config(model_name)["domain"]
+        user_settings = get_wrfrun_config().get_model_config(model_name)["domain"]
         domain_settings: DomainSetting = {
             "dx": user_settings["dx"],
             "dy": user_settings["dy"],
@@ -241,6 +241,10 @@ def plot_domain_area(
             end_lat, _ = inverse_haversine((ref_lat, ref_lon), false_northing, direction=Direction.NORTH, unit=Unit.METERS)
             ax.set_extent([start_lon, end_lon, start_lat, end_lat])
 
+        case crs.LambertConformal:
+            false_easting, false_northing = _calculate_x_y_offset(domain_settings)
+            ax.set_extent([0, false_easting * 2, 0, false_northing * 2], crs=proj)
+
         case _:
             logger.error(f"Unsupported project type: {type(proj)}")
 
@@ -253,6 +257,7 @@ def generate_domain_area():
     :return:
     :rtype:
     """
+    WRFRUNConfig = get_wrfrun_config()
     save_path = WRFRUNConfig.parse_resource_uri(WRFRUNConfig.WRFRUN_OUTPUT_PATH)
     check_path(save_path)
     save_path = abspath(save_path)

wrfrun/model/utils.py

@@ -13,9 +13,9 @@ Utility functions used by models.
 from os import listdir
 from shutil import move
 
-from ..core import WRFRUNConfig
+from ..core import get_wrfrun_config
 from ..utils import check_path, logger
-from ..workspace.wrf import WORKSPACE_MODEL_WRF
+from ..workspace.wrf import get_wrf_workspace_path
 
 
 def clear_model_logs():
@@ -23,8 +23,9 @@ def clear_model_logs():
     This function can automatically collect unsaved log files,
     and save them to the corresponding output directory of the ``Executable``.
     """
+    WRFRUNConfig = get_wrfrun_config()
     work_status = WRFRUNConfig.WRFRUN_WORK_STATUS
-    work_path = WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WRF)
+    work_path = WRFRUNConfig.parse_resource_uri(get_wrf_workspace_path("wrf"))
 
     log_files = [x for x in listdir(work_path) if x.startswith("rsl.") or x.endswith(".log")]
 

wrfrun/model/wrf/core.py

@@ -24,13 +24,12 @@ from os.path import abspath, basename, exists
 from shutil import copyfile, move, rmtree
 from typing import Optional, Union
 
-from wrfrun.core import ExecutableBase, FileConfigDict, InputFileError, NamelistIDError, WRFRUNConfig, WRFRUNExecDB
-from wrfrun.workspace.wrf import WORKSPACE_MODEL_WPS, WORKSPACE_MODEL_WRF
+from wrfrun.core import ExecutableBase, FileConfigDict, InputFileError, NamelistIDError, WRFRUNConfig, WRFRUNExecDB, get_wrfrun_config
 from wrfrun.utils import logger
-from .utils import reconcile_namelist_metgrid, process_after_ndown
-from .namelist import (prepare_dfi_namelist, prepare_wps_namelist, prepare_wrf_namelist,
-                       prepare_wrfda_namelist, get_ungrib_out_dir_path, get_ungrib_out_prefix,
-                       set_ungrib_out_prefix, set_metgrid_fg_names)
+from wrfrun.workspace.wrf import get_wrf_workspace_path
+from .namelist import (get_ungrib_out_dir_path, get_ungrib_out_prefix, prepare_dfi_namelist, prepare_wps_namelist, prepare_wrf_namelist, prepare_wrfda_namelist,
+                       set_metgrid_fg_names, set_ungrib_out_prefix)
+from .utils import process_after_ndown, reconcile_namelist_metgrid
 from .vtable import VtableFiles
 from ..base import NamelistName
 
@@ -40,13 +39,14 @@ def _check_and_prepare_namelist():
     This function check if namelists needed by WPS/WRF have been loaded,
     and prepare namelist if check fails.
     """
-    if not WRFRUNConfig.check_namelist("wps"):
+    wrfrun_config = get_wrfrun_config()
+    if not wrfrun_config.check_namelist("wps"):
         prepare_wps_namelist()
 
-    if not WRFRUNConfig.check_namelist("wrf"):
+    if not wrfrun_config.check_namelist("wrf"):
         prepare_wrf_namelist()
 
-    if not WRFRUNConfig.check_namelist("wrfda"):
+    if not wrfrun_config.check_namelist("wrfda"):
         prepare_wrfda_namelist()
 
 
@@ -78,7 +78,7 @@ class GeoGrid(ExecutableBase):
             mpi_cmd = "mpirun"
             mpi_core_num = core_num
 
-        super().__init__(name="geogrid", cmd="./geogrid.exe", work_path=WORKSPACE_MODEL_WPS, mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
+        super().__init__(name="geogrid", cmd="./geogrid.exe", work_path=get_wrf_workspace_path('wps'), mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
 
         self.geogrid_tbl_file = geogrid_tbl_file
 
@@ -112,14 +112,14 @@ class GeoGrid(ExecutableBase):
         if not WRFRUNConfig.IS_IN_REPLAY:
             if self.geogrid_tbl_file is not None:
                 tbl_file: FileConfigDict = {
-                    "file_path": self.geogrid_tbl_file, "save_path": f"{WORKSPACE_MODEL_WPS}/geogrid",
+                    "file_path": self.geogrid_tbl_file, "save_path": f"{get_wrf_workspace_path('wps')}/geogrid",
                     "save_name": "GEOGRID.TBL", "is_data": False, "is_output": False
                 }
                 self.add_input_files(tbl_file)
 
         super().before_exec()
 
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WPS}/{NamelistName.WPS}", "wps")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wps')}/{NamelistName.WPS}", "wps")
 
         # print debug logs
         logger.debug("Namelist settings of 'geogrid':")
@@ -149,7 +149,7 @@ class LinkGrib(ExecutableBase):
         """
         self._link_grib_input_path = "./input_grib_data_dir"
 
-        super().__init__(name="link_grib", cmd=["./link_grib.csh", f"{self._link_grib_input_path}/*", "."], work_path=WORKSPACE_MODEL_WPS)
+        super().__init__(name="link_grib", cmd=["./link_grib.csh", f"{self._link_grib_input_path}/*", "."], work_path=get_wrf_workspace_path('wps'))
         self.grib_dir_path = grib_dir_path
 
     def generate_custom_config(self):
@@ -170,7 +170,7 @@ class LinkGrib(ExecutableBase):
                 logger.error(f"GRIB file directory not found: {_grib_dir_path}")
                 raise FileNotFoundError(f"GRIB file directory not found: {_grib_dir_path}")
 
-            save_path = f"{WORKSPACE_MODEL_WPS}/{self._link_grib_input_path}"
+            save_path = f"{get_wrf_workspace_path('wps')}/{self._link_grib_input_path}"
             save_path = WRFRUNConfig.parse_resource_uri(save_path)
             if exists(save_path):
                 rmtree(save_path)
@@ -178,7 +178,7 @@ class LinkGrib(ExecutableBase):
             for _file in listdir(_grib_dir_path):
                 _file_config: FileConfigDict = {
                     "file_path": f"{_grib_dir_path}/{_file}",
-                    "save_path": f"{WORKSPACE_MODEL_WPS}/{self._link_grib_input_path}",
+                    "save_path": f"{get_wrf_workspace_path('wps')}/{self._link_grib_input_path}",
                     "save_name": _file, "is_data": True, "is_output": False,
                 }
                 self.add_input_files(_file_config)
@@ -202,7 +202,7 @@ class UnGrib(ExecutableBase):
                                 Defaults to ``input_data_path`` set in user's config file.
         :type input_data_path: str
         """
-        super().__init__(name="ungrib", cmd="./ungrib.exe", work_path=WORKSPACE_MODEL_WPS)
+        super().__init__(name="ungrib", cmd="./ungrib.exe", work_path=get_wrf_workspace_path('wps'))
 
         self.vtable_file = vtable_file
         self.input_data_path = input_data_path
@@ -263,7 +263,7 @@ class UnGrib(ExecutableBase):
 
             _file_config: FileConfigDict = {
                 "file_path": self.vtable_file,
-                "save_path": WORKSPACE_MODEL_WPS,
+                "save_path": get_wrf_workspace_path('wps'),
                 "save_name": "Vtable",
                 "is_data": False,
                 "is_output": False
@@ -272,7 +272,7 @@ class UnGrib(ExecutableBase):
 
         super().before_exec()
 
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WPS}/{NamelistName.WPS}", "wps")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wps')}/{NamelistName.WPS}", "wps")
 
         # print debug logs
         logger.debug("Namelist settings of 'ungrib':")
@@ -325,7 +325,7 @@ class MetGrid(ExecutableBase):
             mpi_cmd = "mpirun"
             mpi_core_num = core_num
 
-        super().__init__(name="metgrid", cmd="./metgrid.exe", work_path=WORKSPACE_MODEL_WPS, mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
+        super().__init__(name="metgrid", cmd="./metgrid.exe", work_path=get_wrf_workspace_path('wps'), mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
 
         self.geogrid_data_path = geogrid_data_path
         self.ungrib_data_path = ungrib_data_path
@@ -377,7 +377,7 @@ class MetGrid(ExecutableBase):
         if not WRFRUNConfig.IS_IN_REPLAY and not WRFRUNConfig.FAKE_SIMULATION_MODE:
             # check input of metgrid.exe
             # try to search input files in the output path if workspace is clear.
-            file_list = listdir(WRFRUNConfig.parse_resource_uri(WORKSPACE_MODEL_WPS))
+            file_list = listdir(WRFRUNConfig.parse_resource_uri(get_wrf_workspace_path('wps')))
 
             if "geo_em.d01.nc" not in file_list:
 
@@ -394,7 +394,7 @@ class MetGrid(ExecutableBase):
                     for _file in geogrid_file_list:
                         _file_config = {
                             "file_path": f"{self.geogrid_data_path}/{_file}",
-                            "save_path": WORKSPACE_MODEL_WPS,
+                            "save_path": get_wrf_workspace_path('wps'),
                             "save_name": _file,
                             "is_data": True,
                             "is_output": True
@@ -427,7 +427,11 @@ class MetGrid(ExecutableBase):
 
         super().before_exec()
 
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WPS}/{NamelistName.WPS}", "wps")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wps')}/{NamelistName.WPS}", "wps")
+
+        # print debug logs
+        logger.debug("Namelist settings of 'metgrid':")
+        logger.debug(WRFRUNConfig.get_namelist("wps"))
 
     def after_exec(self):
         if not WRFRUNConfig.IS_IN_REPLAY:
@@ -470,7 +474,7 @@ class Real(ExecutableBase):
 
         _check_and_prepare_namelist()
 
-        super().__init__(name="real", cmd="./real.exe", work_path=WORKSPACE_MODEL_WRF, mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
+        super().__init__(name="real", cmd="./real.exe", work_path=get_wrf_workspace_path('wrf'), mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
 
         self.metgrid_data_path = metgrid_data_path
 
@@ -511,7 +515,7 @@ class Real(ExecutableBase):
             for _file in file_list:
                 _file_config: FileConfigDict = {
                     "file_path": f"{self.metgrid_data_path}/{_file}",
-                    "save_path": WORKSPACE_MODEL_WRF,
+                    "save_path": get_wrf_workspace_path('wrf'),
                     "save_name": _file,
                     "is_data": True,
                     "is_output": True
@@ -520,7 +524,11 @@ class Real(ExecutableBase):
 
         super().before_exec()
 
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WRF}/{NamelistName.WRF}", "wrf")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wrf')}/{NamelistName.WRF}", "wrf")
+
+        # print debug logs
+        logger.debug("Namelist settings of 'real':")
+        logger.debug(WRFRUNConfig.get_namelist("wrf"))
 
     def after_exec(self):
         if not WRFRUNConfig.IS_IN_REPLAY:
@@ -566,7 +574,7 @@ class WRF(ExecutableBase):
 
         _check_and_prepare_namelist()
 
-        super().__init__(name="wrf", cmd="./wrf.exe", work_path=WORKSPACE_MODEL_WRF, mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
+        super().__init__(name="wrf", cmd="./wrf.exe", work_path=get_wrf_workspace_path('wrf'), mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
 
         self.input_file_dir_path = input_file_dir_path
         self.restart_file_dir_path = restart_file_dir_path
@@ -625,7 +633,7 @@ class WRF(ExecutableBase):
                 for _file in file_list:
                     _file_config: FileConfigDict = {
                         "file_path": f"{self.input_file_dir_path}/{_file}",
-                        "save_path": WORKSPACE_MODEL_WRF,
+                        "save_path": get_wrf_workspace_path('wrf'),
                         "save_name": _file,
                         "is_data": True,
                         "is_output": is_output
@@ -647,7 +655,7 @@ class WRF(ExecutableBase):
                 for _file in file_list:
                     _file_config: FileConfigDict = {
                         "file_path": f"{self.restart_file_dir_path}/{_file}",
-                        "save_path": WORKSPACE_MODEL_WRF,
+                        "save_path": get_wrf_workspace_path('wrf'),
                         "save_name": _file,
                         "is_data": True,
                         "is_output": False
@@ -656,7 +664,11 @@ class WRF(ExecutableBase):
 
         super().before_exec()
 
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WRF}/{NamelistName.WRF}", "wrf")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wrf')}/{NamelistName.WRF}", "wrf")
+
+        # print debug logs
+        logger.debug("Namelist settings of 'wrf':")
+        logger.debug(WRFRUNConfig.get_namelist("wrf"))
 
     def after_exec(self):
         if not WRFRUNConfig.IS_IN_REPLAY:
@@ -701,7 +713,7 @@ class DFI(ExecutableBase):
             mpi_cmd = "mpirun"
             mpi_core_num = core_num
 
-        super().__init__(name="dfi", cmd="./wrf.exe", work_path=WORKSPACE_MODEL_WRF, mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
+        super().__init__(name="dfi", cmd="./wrf.exe", work_path=get_wrf_workspace_path('wrf'), mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
 
         self.input_file_dir_path = input_file_dir_path
         self.update_real_output = update_real_output
@@ -755,7 +767,7 @@ class DFI(ExecutableBase):
                 for _file in file_list:
                     _file_config: FileConfigDict = {
                         "file_path": f"{self.input_file_dir_path}/{_file}",
-                        "save_path": WORKSPACE_MODEL_WRF,
+                        "save_path": get_wrf_workspace_path('wrf'),
                         "save_name": _file,
                         "is_data": True,
                         "is_output": is_output
@@ -769,7 +781,7 @@ class DFI(ExecutableBase):
             prepare_dfi_namelist()
 
         super().before_exec()
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WRF}/{NamelistName.WRF}", "dfi")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wrf')}/{NamelistName.WRF}", "dfi")
 
     def after_exec(self):
         if not WRFRUNConfig.IS_IN_REPLAY:
@@ -823,7 +835,7 @@ class NDown(ExecutableBase):
 
         _check_and_prepare_namelist()
 
-        super().__init__(name="ndown", cmd="./ndown.exe", work_path=WORKSPACE_MODEL_WRF, mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
+        super().__init__(name="ndown", cmd="./ndown.exe", work_path=get_wrf_workspace_path('wrf'), mpi_use=mpi_use, mpi_cmd=mpi_cmd, mpi_core_num=mpi_core_num)
 
         self.wrfout_file_path = wrfout_file_path
         self.real_output_dir_path = real_output_dir_path
@@ -853,7 +865,7 @@ class NDown(ExecutableBase):
         """
         self.real_output_dir_path = self.custom_config["real_output_dir_path"]
         self.update_namelist = self.custom_config["update_namelist"]
-        WRFRUNConfig.update_namelist(self.custom_config["namelist"])
+        WRFRUNConfig.update_namelist(self.custom_config["namelist"], "wrf")
 
     def before_exec(self):
         WRFRUNConfig.check_wrfrun_context(True)
@@ -872,14 +884,14 @@ class NDown(ExecutableBase):
 
         wrfndi_file_config: FileConfigDict = {
             "file_path": f"{self.real_output_dir_path}/wrfinput_d02",
-            "save_path": WORKSPACE_MODEL_WRF,
+            "save_path": get_wrf_workspace_path('wrf'),
             "save_name": "wrfndi_d02",
             "is_data": True,
             "is_output": is_output
         }
         wrfout_file_config: FileConfigDict = {
             "file_path": self.wrfout_file_path,
-            "save_path": WORKSPACE_MODEL_WRF,
+            "save_path": get_wrf_workspace_path('wrf'),
             "save_name": "wrfout_d01",
             "is_data": True,
             "is_output": False
@@ -889,7 +901,7 @@ class NDown(ExecutableBase):
 
         super().before_exec()
 
-        WRFRUNConfig.write_namelist(f"{WORKSPACE_MODEL_WRF}/{NamelistName.WRF}", "wrf")
+        WRFRUNConfig.write_namelist(f"{get_wrf_workspace_path('wrf')}/{NamelistName.WRF}", "wrf")
 
     def after_exec(self):
         self.add_output_files(save_path=self._log_save_path, startswith="rsl.", outputs="namelist.input")

wrfrun/model/wrf/geodata.py

@@ -5,7 +5,7 @@ from typing import OrderedDict, Union
 import numpy as np
 from xarray import DataArray
 
-from wrfrun.core import WRFRUNConfig
+from wrfrun.core import get_wrfrun_config
 from wrfrun.utils import logger
 
 # for example: 00001-00200.00201-00400
@@ -75,6 +75,7 @@ def parse_geographical_data_index(index_path: str) -> OrderedDict:
     # since the index file is very similar to fortran namelist file,
     # we can manually add "&index" and "/" and parse it as a namelist
     # temp file store path
+    WRFRUNConfig = get_wrfrun_config()
     temp_file = f"{WRFRUNConfig.WRFRUN_TEMP_PATH}/geogrid_data.index"
     temp_file = WRFRUNConfig.parse_resource_uri(temp_file)
 

wrfrun/model/wrf/namelist.py

@@ -1,14 +1,13 @@
 from datetime import datetime, timedelta
-from os.path import exists, dirname, basename
+from os.path import basename, dirname, exists
 from typing import Union
 
 from wrfrun.core import WRFRUNConfig
 from wrfrun.res import NAMELIST_DFI, NAMELIST_WPS, NAMELIST_WRF, NAMELIST_WRFDA
 from wrfrun.utils import logger
-from wrfrun.workspace.wrf import WORKSPACE_MODEL_WPS
+from wrfrun.workspace.wrf import get_wrf_workspace_path
 from .scheme import *
 
-
 UNGRIB_OUTPUT_DIR = "./outputs"
 
 
@@ -20,7 +19,7 @@ def get_ungrib_out_dir_path() -> str:
     :rtype: str
     """
     wif_prefix = WRFRUNConfig.get_namelist("wps")["ungrib"]["prefix"]
-    wif_path = f"{WORKSPACE_MODEL_WPS}/{dirname(wif_prefix)}"
+    wif_path = f"{get_wrf_workspace_path('wps')}/{dirname(wif_prefix)}"
 
     return wif_path
 

wrfrun/model/wrf/utils.py

@@ -4,7 +4,7 @@ from typing import Dict
 
 from xarray import open_dataset
 
-from wrfrun import WRFRUNConfig
+from wrfrun import get_wrfrun_config
 from wrfrun.utils import logger
 
 
@@ -65,7 +65,7 @@ def reconcile_namelist_metgrid(metgrid_path: str):
         }
     }
 
-    WRFRUNConfig.update_namelist(update_values, "wrf")
+    get_wrfrun_config().update_namelist(update_values, "wrf")
 
 
 def process_after_ndown():
@@ -77,6 +77,7 @@ def process_after_ndown():
 
     :return:
     """
+    WRFRUNConfig = get_wrfrun_config()
     namelist_data = WRFRUNConfig.get_namelist("wrf")
 
     for section in namelist_data:

wrfrun/model/wrf/vtable.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass
 
-from wrfrun.core import WRFRUNConfig
-from wrfrun.workspace.wrf import WORKSPACE_MODEL_WPS
-
+from wrfrun.core import WRFRunConfig, set_register_func
+from wrfrun.workspace.wrf import get_wrf_workspace_path
 
 VTABLE_URI = ":WRFRUN_VTABLE:"
 
@@ -58,9 +57,13 @@ class VtableFiles:
     UKMO_NO_HEIGHTS = f"{VTABLE_URI}/Vtable.UKMO_no_heights"
 
 
-# register uri
-if not WRFRUNConfig.check_resource_uri(VTABLE_URI):
-    WRFRUNConfig.register_resource_uri(VTABLE_URI, f"{WORKSPACE_MODEL_WPS}/ungrib/Variable_Tables")
+def _register_vtable_uri(wrfrun_config: WRFRunConfig):
+    # register uri
+    if not wrfrun_config.check_resource_uri(VTABLE_URI):
+        wrfrun_config.register_resource_uri(VTABLE_URI, f"{get_wrf_workspace_path('wps')}/ungrib/Variable_Tables")
+
+
+set_register_func(_register_vtable_uri)
 
 
 __all__ = ["VtableFiles"]

wrfrun/res/__init__.py

@@ -1,10 +1,17 @@
 from os.path import abspath, dirname
 
-from wrfrun.core import WRFRUNConfig
+from wrfrun.core import set_register_func, WRFRunConfig
 
 
 RES_PATH = abspath(dirname(__file__))
-WRFRUNConfig.register_resource_uri(WRFRUNConfig.WRFRUN_RESOURCE_PATH, RES_PATH)
+
+
+def _register_res_uri(wrfrun_config: WRFRunConfig):
+    wrfrun_config.register_resource_uri(wrfrun_config.WRFRUN_RESOURCE_PATH, RES_PATH)
+
+
+set_register_func(_register_res_uri)
+
 
 RUN_SH_TEMPLATE = ":WRFRUN_RESOURCE_PATH:/run.template.sh"
 EXT_NCL_PLOT_SCRIPT = ":WRFRUN_RESOURCE_PATH:/extension/plotgrids.ncl"
@@ -19,7 +26,12 @@ NAMELIST_WPS = ":WRFRUN_RESOURCE_PATH:/namelist/namelist.wps.template"
 CONFIG_MAIN_TOML_TEMPLATE = ":WRFRUN_RESOURCE_PATH:/config/config.template.toml"
 CONFIG_WRF_TOML_TEMPLATE = ":WRFRUN_RESOURCE_PATH:/config/wrf.template.toml"
 
-WRFRUNConfig.set_config_template_path(CONFIG_MAIN_TOML_TEMPLATE)
+
+def _set_config_template_path(wrfrun_config: WRFRunConfig):
+    wrfrun_config.set_config_template_path(CONFIG_MAIN_TOML_TEMPLATE)
+
+
+set_register_func(_set_config_template_path)
 
 
 __all__ = ["RUN_SH_TEMPLATE", "EXT_NCL_PLOT_SCRIPT", "SCHEDULER_LSF_TEMPLATE", "SCHEDULER_PBS_TEMPLATE", "SCHEDULER_SLURM_TEMPLATE", "NAMELIST_WRFDA", "NAMELIST_DFI", "NAMELIST_REAL", "NAMELIST_WRF", "NAMELIST_WPS", "CONFIG_MAIN_TOML_TEMPLATE", "CONFIG_WRF_TOML_TEMPLATE"]

wrfrun/res/config/config.template.toml

@@ -1,4 +1,11 @@
 # Config template of wrfrun.
+
+# Work directory.
+# By setting work directory, you can let wrfrun work and save runtime files in a single directory.
+# You can also set work_dir="" to let wrfrun work in the standard path.
+work_dir = "./.wrfrun"
+
+
 # Path of the directory which contains input data.
 input_data_path = ""