Base

create_years_per_job_nml(input_file_path, years_per_job, exist_ok=None)

Splits up list of all years into separate lists of no more than years_per_job in each. A nml file is then created for each of these with name same as input_file_path but with first year in job as a suffix e.g. input_nml becomes input1985.nml with input_info['script_info']['years'] set to years to run for that job.

Parameters:

Name	Type	Description	Default
input_file_path	str	Path to nml file for experiment.	required
years_per_job	int	Numbr of years to run for each job.	required
exist_ok	Optional[bool]	If True, do not raise exception if any file to be created already exists. If False, will overwrite it. If None leaves the existing file unchanged.	None

Returns:

Type	Description
List	List of paths to nml files created e.g. ['/Users/.../input1985.nml', '/Users/.../input1985.nml']

Source code in isca_tools/cesm/base.py

def create_years_per_job_nml(input_file_path: str, years_per_job: int, exist_ok: Optional[bool] = None) -> List:
    """
    Splits up list of all years into separate lists of no more than `years_per_job` in each.
    A `nml` file is then created for each of these with name same as `input_file_path` but with first year
    in job as a suffix e.g. `input_nml` becomes `input1985.nml` with `input_info['script_info']['years']` set to
    years to run for that job.

    Args:
        input_file_path: Path to `nml` file for experiment.
        years_per_job: Numbr of years to run for each job.
        exist_ok: If `True`, do not raise exception if any file to be created already exists.
            If `False`, will overwrite it. If `None` leaves the existing file unchanged.

    Returns:
        List of paths to nml files created e.g. `['/Users/.../input1985.nml', '/Users/.../input1985.nml']`
    """
    input_info = f90nml.read(input_file_path)

    # Determine which years to get data for
    file_dates = get_exp_file_dates(input_info['script_info']['exp_name'], 'atm',
                                    input_info['script_info']['archive_dir'], input_info['script_info']['hist_file'])
    year_files_all = np.unique(file_dates.dt.year).tolist()
    if input_info['script_info']['year_files'] is None:
        years_consider = year_files_all
    else:
        years_consider = parse_int_list(input_info['script_info']['year_files'], lambda x: int(x),
                                        all_values=year_files_all)
    years_jobs = split_list_max_n(years_consider, years_per_job)
    out_file_names = []
    for years in years_jobs:
        input_info['script_info']['year_files'] = years
        out_file_names.append(input_file_path.replace('.nml', f'{years[0]}.nml'))
        if os.path.exists(out_file_names[-1]):
            if exist_ok is None:
                print(f'{years}: Output nml file already exists. Leaving unchanged')
                continue
        input_info.write(out_file_names[-1], force=exist_ok)
        print(f'{years}: Output nml file created')
    return out_file_names

get_pressure(ps, p0, hya, hyb)

Calculates pressure at the hybrid levels, similat to NCAR function.

Parameters:

Name	Type	Description	Default
ps	Union[ndarray, DataArray]	float [n_time x n_lat x n_lon] Array of surface pressures in units of Pa. PS in CESM atmospheric output.	required
p0	float	Surface reference pressure in Pa. P0 in CESM atmospheric output.	required
hya	Union[ndarray, DataArray]	float [n_levels] Hybrid A coefficients. hyam in CESM atmospheric output.	required
hyb	Union[ndarray, DataArray]	float [n_levels] Hybrid B coefficients. hybm in CESM atmospheric output.	required

Returns:

Type	Description
Union[ndarray, DataArray]	float [n_time x n_levels x n_lat x n_lon] Pressure at the hybrid levels in Pa.

Source code in isca_tools/cesm/base.py

def get_pressure(ps: Union[np.ndarray, xr.DataArray], p0: float, hya: Union[np.ndarray, xr.DataArray],
                 hyb: Union[np.ndarray, xr.DataArray]) -> Union[np.ndarray, xr.DataArray]:
    """
    Calculates pressure at the hybrid levels, similat to
    [NCAR function](https://www.ncl.ucar.edu/Document/Functions/Built-in/pres_hybrid_ccm.shtml).

    Args:
        ps: `float [n_time x n_lat x n_lon]`</br>
            Array of surface pressures in units of *Pa*.</br>
            `PS` in CESM atmospheric output.
        p0: Surface reference pressure in *Pa*.</br>
            `P0` in CESM atmospheric output.
        hya: `float [n_levels]`</br>
            Hybrid A coefficients.</br>
            `hyam` in CESM atmospheric output.
        hyb: `float [n_levels]`</br>
            Hybrid B coefficients.</br>
            `hybm` in CESM atmospheric output.

    Returns:
        `float [n_time x n_levels x n_lat x n_lon]`
            Pressure at the hybrid levels in *Pa*.
    """
    return hya * p0 + hyb * ps