from oggm import cfg, utils, workflow, tasks, graphics
from oggm.core import flowline
import salem
import xarray as xr
import pandas as pd
import geopandas as gpd
import matplotlib.pyplot as plt
import os


cfg.initialize(logging_level='WARNING')

# Where to store the data 
cfg.PATHS['working_dir'] = os.environ.get('OGGM_WORKDIR')
cfg.PARAMS['continue_on_error'] = True
odir = os.environ.get('OGGM_OUTDIR')
utils.mkdir(odir)

rgi_reg = os.environ.get('OGGM_RGI_REG')
rgi_reg = '{:02}'.format(int(rgi_reg))

rgidf = gpd.read_file(utils.get_rgi_region_file(rgi_reg, version='62'))

base_url = 'https://cluster.klima.uni-bremen.de/~oggm/gdirs/oggm_v1.4/L1-L2_files/centerlines'
gdirs = workflow.init_glacier_directories(rgidf, from_prepro_level=2, prepro_border=10, prepro_base_url=base_url)

utils.write_centerlines_to_shape(gdirs, to_tar=True,
                                 path=os.path.join(odir, 'RGI{}_centerlines.shp'.format(rgi_reg)))
utils.write_centerlines_to_shape(gdirs, to_tar=True, flowlines_output=True,
                                 path=os.path.join(odir, 'RGI{}_flowlines.shp'.format(rgi_reg)))
utils.write_centerlines_to_shape(gdirs, to_tar=True, geometrical_widths_output=True,
                                 path=os.path.join(odir, 'RGI{}_geom_widths.shp'.format(rgi_reg)))
utils.write_centerlines_to_shape(gdirs, to_tar=True, corrected_widths_output=True,
                                 path=os.path.join(odir, 'RGI{}_widths.shp'.format(rgi_reg)))