#!/usr/bin/env python
"""
Build hydroelectric inflow time-series for each country.

Relevant Settings
-----------------

.. code:: yaml

    countries:

    renewable:
        hydro:
            cutout:
            clip_min_inflow:

Inputs
------

- ``data/bundle/EIA_hydro_generation_2000_2014.csv``:
- ``resources/country_shapes.geojson``: confer :ref:`shapes`
- ``"cutouts/" + config["renewable"]['hydro']['cutout']``: confer :ref:`cutout`

Outputs
-------

- ``resources/profile_hydro.nc``:

Description
-----------

.. seealso::
    build_renewable_profiles
"""

import os
import atlite
import pandas as pd
import geopandas as gpd
from vresutils import hydro as vhydro
import logging


if __name__ == "__main__":
    logger.basicConfig(level=snakemake.config['logging_level'])

    config = snakemake.config['renewable']['hydro']
    cutout = atlite.Cutout(config['cutout'],
                        cutout_dir=os.path.dirname(snakemake.input.cutout))

    countries = snakemake.config['countries']
    country_shapes = gpd.read_file(snakemake.input.country_shapes).set_index('name')['geometry'].reindex(countries)
    country_shapes.index.name = 'countries'

    eia_stats = vhydro.get_eia_annual_hydro_generation(snakemake.input.eia_hydro_generation).reindex(columns=countries)
    inflow = cutout.runoff(shapes=country_shapes,
                        smooth=True,
                        lower_threshold_quantile=True,
                        normalize_using_yearly=eia_stats)

    if 'clip_min_inflow' in config:
        inflow.values[inflow.values < config['clip_min_inflow']] = 0.

    inflow.to_netcdf(snakemake.output[0])