import sys
sys.path.insert(0, op.join(op.dirname(__file__), '..', '..', '..'))
# teslakit
from teslakit.database import Database
from teslakit.estela import Predictor
# --------------------------------------
# Teslakit database
p_data = r'/Users/nico/Projects/TESLA-kit/TeslaKit/data'
db = Database(p_data)
# set site
db.SetSite('TESTNEW')
# --------------------------------------
# load SLP ESTELA predictor and waves data
pred = Predictor(db.paths.site.ESTELA.pred_slp)
pred.Load()
xds_WAVES = db.Load_ESTELA_waves()
# KMA REGRESSION GUIDED parameters
kma_date_ini = '1979-01-22'
kma_date_end = '2011-01-22'
num_clusters = 36
kmarg_alpha = 0.36
sys.path.insert(0, op.join(op.dirname(__file__), '..', '..', '..'))
# teslakit
from teslakit.database import Database
from teslakit.plotting.awt import Plot_AWT_Validation, Plot_AWTs, \
Plot_AWTs_Dates, Plot_AWT_PCs_3D, Plot_AWT_PCs, Plot_EOFs_SST
# --------------------------------------
# Teslakit database
p_data = r'/Users/nico/Projects/TESLA-kit/TeslaKit/data'
db = Database(p_data)
# set site
db.SetSite('KWAJALEIN')
# Load SST AWT KMA and PCA
xds_PCA = db.Load_SST_PCA()
xds_KMA = db.Load_SST_KMA()
# Plot 6 EOFs
Plot_EOFs_SST(xds_PCA, 6)
# Plot 6 PCs 2D
Plot_AWT_PCs(xds_PCA, xds_KMA, n=6)
# load PCs simulated with copula
d_PCs_fit, d_PCs_rnd = db.Load_SST_PCs_fit_rnd()
# Plot PCs 3D
# DEV: override installed teslakit
import sys
sys.path.insert(0, op.join(op.dirname(__file__), '..', '..', '..'))
# teslakit
from teslakit.database import Database
from teslakit.alr import ALR_WRP
# --------------------------------------
# Teslakit database
p_data = r'/Users/nico/Projects/TESLA-kit/TeslaKit/data'
db = Database(p_data)
# set site
db.SetSite('ROI')
# alr test
p_alrw = db.paths.site.SST.alrw
#p_alrw = db.paths.site.MJO.alrw
#p_alrw = db.paths.site.ESTELA.alrw
# ALR wrap
ALRW = ALR_WRP(p_alrw)
# show model report
#ALRW.Report_Fit(terms_fit=True, summary=True)
# show simulation report
ALRW.Report_Sim()