{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e9cedd8043c54aceb1d5e8d381b4bcd0",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Map(center=[23.916287330629192, -102.18542161782723], controls=(WidgetControl(options=['position', 'transparen…"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import ee\n",
    "ee.Initialize()\n",
    "import geemap\n",
    "\n",
    "# Carga de Terra Climate y declaracion de parámetros\n",
    "img_coll = ee.ImageCollection(\"IDAHO_EPSCOR/TERRACLIMATE\")\n",
    "scale_img_coll = 4638.3\n",
    "img_coll_start_year = 1958 # Empezando en Enero\n",
    "img_coll_end_year = 2022 # Terminando en Diciembre\n",
    "\n",
    "# Geometria de México\n",
    "fc_polygons_countries = ee.FeatureCollection(\"USDOS/LSIB/2017\")\n",
    "geom_mexico = fc_polygons_countries.filter(ee.Filter.eq(\"COUNTRY_NA\",\"Mexico\")).first().geometry()\n",
    "\n",
    "# Banda de interés\n",
    "banda_interes = \"pdsi\"\n",
    "\n",
    "# Escalar a unidades \"reales\"\n",
    "def scalling_funcion_pdsi(img):\n",
    "    return img.multiply(0.01).copyProperties(img, img.propertyNames())\n",
    "\n",
    "# Dar como propiedad a la imagen: date_month y date_year\n",
    "def create_month_year(img):\n",
    "    full_date = ee.Date(ee.Number(img.get(\"system:time_start\")))\n",
    "    date_year = ee.Number(full_date.get(\"year\"))\n",
    "    date_month = ee.Number(full_date.get(\"month\"))\n",
    "    return img.set({\"date_month\": date_month, \"date_year\": date_year})\n",
    "\n",
    "\n",
    "# Cargar datos, seleccionar banda, escalarlos y limitarlos a la mexico\n",
    "img_coll_pdsi = img_coll.select(banda_interes).filterBounds(geom_mexico).map(scalling_funcion_pdsi)\n",
    "\n",
    "# Etiquetar date_month y date_year como prop de la imagen\n",
    "img_coll_pdsi_tag_month_year = img_coll_pdsi.map(create_month_year)#.map(lambda img: img.clip(geom_mexico))\n",
    "\n",
    "labels_dates = list()\n",
    "for year in range(1958,2023):\n",
    "    for month in list(map(lambda x: '0'+str(x) if x <= 9 else str(x), range(1,13))):\n",
    "        date_year_month_str = str(year) + \"-\" + month + \"-01\"\n",
    "        labels_dates.append(date_year_month_str)\n",
    "\n",
    "mapVisParamsPDSI = dict(min = -5, max = 5, palette = [\"FB322C\", \"FFFFFF\", \"039DAB\"])\n",
    "\n",
    "Mapa = geemap.Map(center=(22, -100), zoom=4, height='800px', basemap = \"CartoDB.DarkMatter\")\n",
    "Mapa.center_object(geom_mexico, 5)\n",
    "Mapa.add_time_slider(img_coll_pdsi_tag_month_year, mapVisParamsPDSI,\n",
    "                     layer_name = \"Índice de Severidad de Sequía de Palmer\",\n",
    "                     labels = labels_dates,\n",
    "                     position='bottomleft',\n",
    "                     slider_length = \"300px\",\n",
    "                     time_interval=1)\n",
    "Mapa\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.15"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}