# Apache OCW lib immports
from ocw.dataset import Dataset, Bounds
import ocw.data_source.local as local
import ocw.data_source.rcmed as rcmed
import ocw.dataset_processor as dsp
import ocw.evaluation as evaluation
import ocw.metrics as metrics
import ocw.plotter as plotter
import ocw.utils as utils

import datetime
import numpy as np
import numpy.ma as ma

OUTPUT_PLOT = "subregions"

# Spatial and temporal configurations
LAT_MIN = -45.0
LAT_MAX = 42.24
LON_MIN = -24.0
LON_MAX = 60.0
START_SUB = datetime.datetime(2000, 01, 1)
END_SUB = datetime.datetime(2007, 12, 31)

# regridding parameters
gridLonStep = 0.5
gridLatStep = 0.5

# Regrid
print("... regrid")
new_lats = np.arange(LAT_MIN, LAT_MAX, gridLatStep)
new_lons = np.arange(LON_MIN, LON_MAX, gridLonStep)

list_of_regions = [
    Bounds(-10.0, 0.0, 29.0, 36.5, START_SUB, END_SUB),
    Bounds(0.0, 10.0,  29.0, 37.5, START_SUB, END_SUB),
    Bounds(10.0, 20.0, 25.0, 32.5, START_SUB, END_SUB),
    Bounds(20.0, 33.0, 25.0, 32.5, START_SUB, END_SUB),
    Bounds(-19.3, -10.2, 12.0, 20.0, START_SUB, END_SUB),
    Bounds(15.0, 30.0, 15.0, 25.0, START_SUB, END_SUB),
    Bounds(-10.0, 10.0, 7.3, 15.0, START_SUB, END_SUB),
    Bounds(-10.9, 10.0, 5.0, 7.3,  START_SUB, END_SUB),
    Bounds(33.9, 40.0,  6.9, 15.0, START_SUB, END_SUB),
    Bounds(10.0, 25.0,  0.0, 10.0, START_SUB, END_SUB),
    Bounds(10.0, 25.0, -10.0,  0.0, START_SUB, END_SUB),
    Bounds(30.0, 40.0, -15.0,  0.0, START_SUB, END_SUB),
    Bounds(33.0, 40.0, 25.0, 35.0, START_SUB, END_SUB)]

# for plotting the subregions
plotter.draw_subregions(list_of_regions, new_lats,
                        new_lons, OUTPUT_PLOT, fmt='png')