#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Wed Sep 18 21:10:30 EDT 2019
from datetime import datetime
import os.path, sys, os, glob
#import matplotlib.pyplot as plt
import xarray as xr, numpy as np, pandas as pd
#import geoxarray
from xlib.io import get_data
print()

expnames = [
    'Pinatubo_PI_ens_noleap',
    'Agung_PI_ens_noleap',
    'StMaria_PI_ens_noleap',
    'Chichon_PI_ens_noleap',
    ]
data_names = ['vcomp']
plev = 850
odata_names = [f'v{plev}'] # data name of output
#data_tags = ['atmos_month', 'atmos_month', 'ice_month'][0:1]
data_tag = 'atmos_month'

if __name__ == '__main__':
    # record the start time
    tformat = '%Y-%m-%d %H:%M:%S'
    t0 = datetime.now()
    print('[time start]:', t0.strftime(tformat))
    
    for data_name, odata_name in zip(data_names, odata_names):
        for expname in expnames: 
            print(expname, data_name, '-->', odata_name)
            get_data(expname, data_name=data_name, odata_name=odata_name, plev=plev, data_tag=data_tag)

    # the end
    t1 = datetime.now()
    dt = t1 - t0
    print('[time end]:', t1.strftime(tformat))
    print('[time used]:', f'{dt.seconds:,} seconds')
    print()