import sys, datetime, copy, os
from dateutil.relativedelta import relativedelta
import numpy as np; np.set_printoptions(threshold=sys.maxsize)
import matplotlib; matplotlib.use('Agg')
import matplotlib.pyplot as plt
import matplotlib.colors as clr
from mpl_toolkits.basemap import Basemap

import seaborn as sns
from matplotlib.colors import ListedColormap
from netCDF4 import Dataset

nc   = Dataset('/work/users/hmo/remoclic/Phuong/PHD/BC_2023/Tan/NC1/All_Cha_landpoints.nc','r') #FD_Char_1961-2020_1yrs_Daily_Reg.nc','r')
Chars  = nc.variables['cha'][:]
Model  = nc.variables['mem'][:]
Npoint = nc.variables['point'][:]
aph    = nc.variables['aph'][:]
org    = nc.variables['org'][:]
qm     = nc.variables['qm'][:]

aph[aph<0] = np.nan
org[org<0] = np.nan
qm[qm<0] = np.nan

aph=aph[~np.isnan(aph)]
org=org[~np.isnan(org)]
qm=qm[~np.isnan(qm)]

xlabels = ["CNRM","CSIRO","ECEA","GFDL","HadG","MPI","ENS"]
title = ["DF","DD","DS","DI"]
#Amax = [ 7, 150, 4000, 85, 35, 35 ]
Amax = [ 20, 80, 100, 10 ]

Xlab=[]
Del1=[]
Del2=[]
Del3=[]
for i in range(0,len(xlabels)):
    del1=1 + i*4
    del2=2 + i*4
    del3=3 + i*4
    xlab=i+1
    Del1.append(del1)
    Del2.append(del2)
    Del3.append(del3)
    Xlab.append(xlab)

nbox = len(Chars)
#fig, ax = plt.subplots(nbox,figsize=(6,10),sharey=False)
fig, ax = plt.subplots(nbox,figsize=(8,10),sharey=False)
plt.subplots_adjust(hspace=0.4)
plt.subplots_adjust(top=1)
#plt.subplots_adjust(left=0.01)
#plt.subplots_adjust(right=0.02)

c1 = "red"
c2 = "blue"
c3 = "green"
for k in range(len(Chars)):
    data1 = aph[:,:,k]
    data2 = org[:,:,k]
    data3 = qm[:,:,k]

#    data1 = data1[data1>=0]
#    data2 = data2[data2>=0]
#    data3 = data3[data3>=0]

#    data1[data1==-99] = np.nan
#    data2[data2==-99] = np.nan
#    data3[data3==-99] = np.nan

#    ymax = round(np.nanmax(data)); ymin = round(np.nanmin(data))
    ymax = Amax[k]; ymin = 0 
    yticks = np.linspace(ymin,ymax,5)
#    ax[k].boxplot(data1, sym='k+')
"""
    ax[k].boxplot(data1, positions = Del1,
                    notch = True,
                    vert = True,
                    patch_artist = True,
                   # labels = list_of_files,
                   # labels = Del1,
                   ## labels = Xlab,
                    boxprops=dict(facecolor=c1, color=c1))
    ax[k].boxplot(data2, positions = Del2,
                    notch = True,
                    vert = True,
                    patch_artist = True,
                   # labels = list_of_files,
                   ## labels = Xlab,
                    boxprops=dict(facecolor=c2, color=c2))
    ax[k].boxplot(data3, positions = Del3,
                    notch = True,
                    vert = True,
                    patch_artist = True,
                   # labels = list_of_files,
                   ## labels = Xlab,
                    boxprops=dict(facecolor=c3, color=c3))

"""
    
    sns.boxplot(y='model',x='data',data=data1)
    sns.boxplot(y='model',x='data',data=data2)
    sns.boxplot(y='model',x='data',data=data3)

#    ax[k].set_title('var'+str(k+1))
    ax[k].set_title(title[k])
#    ax[k].set_xticklabels(xlabels)

    ax[k].set_yticks(yticks)

#plt.show()
fig.savefig('Boxpl_Dr_Chars_test.pdf', bbox_inches="tight", pad_inches=.05, dpi=600)