experiments = ['temev', 'temfa', 'temeq', 'temel', 'temeg', 'temeb', 'temfA', 'temfv', 'temfq', 'temfl', 'temfg', 'temfb', 'temez', 'temeu', 'temep', 'temek', 'temef', 'temea', 'temfz', 'temfu', 'temfp', 'temfk', 'temff', 'temey', 'temet', 'temeo', 'temej', 'temee', 'temfy', 'temft', 'temfo', 'temfj', 'temfe', 'temex', 'temes', 'temen', 'temei', 'temed', 'temfx', 'temfs', 'temfn', 'temfi', 'temfd', 'temew', 'temer', 'temem', 'temeh', 'temec', 'temfw', 'temfr', 'temfm', 'temfh', 'temfc'] ds_hice = xr.open_dataset('/nfs/annie/eeymr/work/data/glac_mw/GLAC1DHiceF26.nc') ds_pointer = xr.open_dataset('/nfs/annie/eeymr/work/data/glac_mw/GLAC1DdrainagePointerF26ka.nc') ds_wfix = xr.open_dataset("/nfs/annie/earpal/database/experiments/temev/inidata/temev.qrparm.waterfix.nc") for experiment in experiments: lsm_path = f"/nfs/annie/earpal/database/experiments/{experiment}/inidata/{experiment}.qrparm.omask.nc" ds_lsm = xr.open_dataset(lsm_path) routed_mw = routing.routing(ds_hice, ds_pointer, ds_lsm, mode_flux="Volume", mode_lon="double", mode_shape="cross", mode_smooth="differential") saving.saving(routed_mw, ds_lsm, experiment, mode="routed", start_year=-26, end_year=0, step=100, mode_smooth="diff") spreaded_mw = spreading.spreading(routed_mw, ds_lsm, ds_wfix) saving.saving(spreaded_mw, ds_lsm, experiment, mode="spreaded", start_year=-26, end_year=0, step=100, mode_smooth="diff") plotting.plot_discharge_ts( "/nfs/see-fs-01_users/eeymr/work/outputs/proj_glac_mw/glac.-26_0_100.diff_s/temev.discharge.glac_mw.nc", "/nfs/annie/earpal/database/experiments/temev/inidata/temev.qrparm.omask.nc", unit="kg/m2/s", out="save")
import glac_mw.plotting as plotting path_discharge = '/nfs/annie/eeymr/work/outputs/proj_glac_mw/glac.-28_2_100.diff_s/temev.discharge.glac_mw.nc' path_lsm = '/nfs/annie/earpal/database/experiments/temev/inidata/temev.qrparm.omask.nc' plotting.plot_discharge_ts(path_discharge, path_lsm, unit="kg/m2/s", out="save")