#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Thu Oct 26 15:30:22 EDT 2023
if __name__ == '__main__':
    import sys,os
    from misc.timer import Timer
    tt = Timer(f'[{os.getcwd()}] start ' + ' '.join(sys.argv))
import sys, os.path, os, glob, datetime
#import xarray as xr, numpy as np, pandas as pd, matplotlib.pyplot as plt
#more imports
wython = '/tigress/wenchang/wython'
if wython not in sys.path: sys.path.append(wython); print('added to python path:', wython)
#from misc import get_kws_from_argv
from tigerCMIP.cmip6_model_members_in_common import cmip6_model_members_in_common
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
danames = ['ts', 'pr']
expnames = ['amip-p4K', 'amip', 'abrupt-4xCO2', 'piControl', 'amip-4xCO2']
pairs = [
    ('ts', 'amip-p4K'),
    ('ts', 'amip'),
    ('ts', 'abrupt-4xCO2'),
    ('ts', 'piControl'),
    ('pr', 'amip-p4K'),
    ('pr', 'amip'),
    ('pr', 'abrupt-4xCO2'),
    ('pr', 'piControl'),
    ('pr', 'amip-4xCO2'),
    ]
model_members = cmip6_model_members_in_common(*pairs)
for model_member in model_members:
    print(model_member)
    model, member = model_member.split('_')
    for daname in danames:
        if daname in ('ts',):
            expnames_ = expnames[:-1] #do not need data of ts/amip-4xCO2
        else:
            expnames_ = expnames
        for expname in expnames_:
            ifile = glob.glob(f'/tigress/wenchang/data/cmip6/variables/{daname}/{expname}_v202310/wy_regrid/{daname}.{expname}.{model}.{member}.*.nc')[0]
            ofile = os.path.basename(ifile)
            if os.path.exists(ofile):
                print('[exists]:', ofile)
                continue
            else:
                os.symlink(ifile, ofile)
                print('[linked]:', ofile, '->', ifile)
 
 
if __name__ == '__main__':
    #from wyconfig import * #my plot settings
    
    #savefig
    if 'savefig' in sys.argv or 's' in sys.argv:
        figname = __file__.replace('.py', f'.png')
        if 'overwritefig' in sys.argv or 'o' in sys.argv:
            wysavefig(figname, overwritefig=True)
        else:
            wysavefig(figname)
    tt.check(f'**Done**')
    print()
    if 'notshowfig' in sys.argv or 'n' in sys.argv:
        pass
    else:
        if 'plt' in globals(): plt.show()