#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Wed Oct 25 20:49:41 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
import geoxarray
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
ifiles = glob.glob('ts.abrupt-4xCO2.*.nc')
ifiles.sort()
func = lambda da: da

if __name__ == '__main__':
    from wyconfig import * #my plot settings
    for ifile in ifiles:
        fig, ax = plt.subplots()
        da = xr.open_dataarray(ifile).load()
        units = da.attrs['units']
        #units = 'K'
        da.pipe(func).plot(ax=ax, label='abrupt-4xCO2')

        model = ifile.split('.')[2]
        ifile_ctl = glob.glob(f'ts.piControl.{model}.*.nc')[0]
        da = xr.open_dataarray(ifile_ctl).load()
        da.pipe(func).plot(ax=ax, label='piControl', color='k')
        
        
        title = os.path.basename(ifile)[:-3] #remove extension ".nc"
        ax.set_title(title)
        ax.set_ylabel(units)

        #savefig
        if 'savefig' in sys.argv or 's' in sys.argv:
            #figname = __file__.replace('.py', f'.png')
            figname = os.path.basename(ifile).replace('.nc', '.png')
            if 'overwritefig' in sys.argv or 'o' in sys.argv:
                wysavefig(figname, overwritefig=True)
            else:
                wysavefig(figname)
        print()
        if 'notshowfig' in sys.argv or 'n' in sys.argv:
            pass
        else:
            if 'plt' in globals(): plt.show()
    
    tt.check(f'**Done**')