import act
import numpy as np
import xarray as xr
import matplotlib.pyplot as plt

username = 'rppoland18'
token = '3127641b73e77c24'
datastream = 'gucqcrad1longM1.c2'
startdate = '2021-09-01'
enddate = '2023-06-15'

result = act.discovery.download_arm_data(username, token, datastream, startdate, enddate)
rad = act.io.read_arm_netcdf(result)
rad

wf_rad = rad.sel(time=slice('2022-01-26T06:30:00','2022-01-27T00:00:00'))
bkg_rad = rad.sel(time=slice('2022-01-03T00:00:00','2022-01-04T00:00:00'))
dust_rad = rad.sel(time=slice('2023-04-06T00:00:00','2023-04-07T00:00:00'))
# dust_rad.down_long_hemisp.plot()
# dust_rad.up_long_hemisp.plot()
fig,(ax1,ax2) = plt.subplots(1,2,figsize=[20,8],layout='constrained')
fig.supxlabel('Time')
fig.supylabel('Downwelling IR Rad. / Upwelling IR Rad.')
ax1.plot((wf_rad.down_long_hemisp/wf_rad.up_long_hemisp),color='red') # Ratio of downwelling and upwelling IR radiation
ax1.set_ylim([0.6,0.9])
ax2.plot((bkg_rad.down_long_hemisp/bkg_rad.up_long_hemisp),color='black') # Ratio of downwelling and upwelling IR radiation
ax2.set_ylim([0.6,0.9])