#!/usr/bin/env python
# This software is open source software available under the BSD-3 license.
#
# Copyright (c) 2022 Triad National Security, LLC. All rights reserved.
# Copyright (c) 2022 Lawrence Livermore National Security, LLC. All rights
# reserved.
# Copyright (c) 2022 UT-Battelle, LLC. All rights reserved.
#
# Additional copyright and license information can be found in the LICENSE file
# distributed with this code, or at
# https://raw.githubusercontent.com/MPAS-Dev/MPAS-Analysis/main/LICENSE
"""
Entry points for downloading data for MPAS-Analysis
"""
# Authors
# -------
# Xylar Asay-Davis, Phillip J. Wolfram, Milena Veneziani
import argparse
import importlib.resources as resources
import os
from mpas_analysis.shared.io.download import download_files
[docs]
def download_analysis_data():
"""
Entry point for downloading the input data set from public repository for
MPAS-Analysis to work. The input data set includes: pre-processed
observations data, MPAS mapping files and MPAS regional mask files
(which are used for the MOC computation), for a subset of MPAS meshes.
"""
parser = argparse.ArgumentParser(
description=__doc__, formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument("-o", "--outDir", dest="outDir", required=True,
help="Directory where MPAS-Analysis input data will"
"be downloaded")
parser.add_argument("-d", "--dataset", dest="dataset", default='analysis',
help="Directory where MPAS-Analysis input data will"
"be downloaded")
args = parser.parse_args()
try:
os.makedirs(args.outDir)
except OSError:
pass
urlBase = 'https://web.lcrc.anl.gov/public/e3sm/diagnostics'
resource = resources.files('mpas_analysis.obs').joinpath(
f'{args.dataset}_input_files')
with resource.open('r') as f:
analysisFileList = f.read()
# remove any empty strings from the list
analysisFileList = list(filter(None, analysisFileList.split('\n')))
download_files(analysisFileList, urlBase, args.outDir, verify=True)