{ "cells": [ { "cell_type": "code", "execution_count": 19, "id": "0af4469a-318b-473d-9fde-e2dd7029bb89", "metadata": {}, "outputs": [], "source": [ "# NOTE: use branch feat_restart on the jmccreight fork\n", "\n", "import pathlib as pl\n", "from pprint import pprint\n", "import shutil\n", "\n", "import jupyter_black\n", "import numpy as np\n", "import pywatershed as pws\n", "import xarray as xr\n", "\n", "jupyter_black.load()" ] }, { "cell_type": "code", "execution_count": 28, "id": "1a600752-3850-477c-a7d1-c2af04b8276a", "metadata": {}, "outputs": [], "source": [ "input_dir = pl.Path(\"./output\")\n", "domain_dir = pl.Path(\"../DOMAIN\")\n", "restart_dir = pl.Path(\"./restart/\")\n", "\n", "if restart_dir.exists():\n", " shutil.rmtree(restart_dir)\n", "\n", "t0 = np.datetime64(\"2009-12-31T00:00:00\")\n", "# t0 = np.datetime64(\"2020-01-01T00:00:00\")\n", "timestep = np.timedelta64(24, \"h\")\n", "n_steps = 4000" ] }, { "cell_type": "code", "execution_count": 29, "id": "ae93ec27-53da-4fb2-ba5c-82338685de3a", "metadata": {}, "outputs": [], "source": [ "def get_control(init_time, end_time, restart_read: pl.Path = False):\n", " control = pws.Control.load_prms(\n", " domain_dir / \"nhm.control\", warn_unused_options=False\n", " )\n", " control.options[\"input_dir\"] = input_dir\n", " control.options[\"restart_write\"] = restart_dir\n", " control.options[\"restart_write_freq\"] = \"d\" # 'd'\n", " if restart_read:\n", " control.options[\"restart_read\"] = restart_read\n", " del control.options[\"netcdf_output_dir\"]\n", " del control.options[\"netcdf_output_var_names\"]\n", " control.edit_end_time(end_time)\n", " control.edit_init_start_times(init_time)\n", " return control" ] }, { "cell_type": "code", "execution_count": 30, "id": "556e1eb4-e392-4680-aeb4-d8b1e2a63356", "metadata": {}, "outputs": [], "source": [ "def get_params():\n", " return pws.parameters.PrmsParameters.load(domain_dir / \"myparam.param\")" ] }, { "cell_type": "code", "execution_count": 31, "id": "bbeb3a75-7d55-4150-bf48-c6918733ab22", "metadata": {}, "outputs": [], "source": [ "nhm_processes = [\n", " pws.PRMSGroundwater,\n", " pws.PRMSChannel,\n", "]" ] }, { "cell_type": "code", "execution_count": 32, "id": "54a8d568-d455-41a9-ba70-3d4d10cafa1a", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "PRMSGroundwater jit compiling with numba \n", "PRMSChannel jit compiling with numba \n" ] } ], "source": [ "# no restart, only write restarts\n", "nhm = pws.Model(\n", " nhm_processes,\n", " control=get_control(t0, t0 + timestep * n_steps),\n", " parameters=get_params(),\n", ")" ] }, { "cell_type": "code", "execution_count": 33, "id": "f999083c-26c2-4d0a-a03b-90458c77ef65", "metadata": {}, "outputs": [ { "data": { "text/plain": [ "np.datetime64('2009-12-31T00:00:00')" ] }, "execution_count": 33, "metadata": {}, "output_type": "execute_result" } ], "source": [ "t0" ] }, { "cell_type": "code", "execution_count": 34, "id": "ab017a58-6d4b-479f-b395-a553736792f1", "metadata": {}, "outputs": [ { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "38c1eed3d32f4e0bba86044891bb1234", "version_major": 2, "version_minor": 0 }, "text/plain": [ " 0%| | 0/4000 [00:00