#!/usr/bin/env python
# coding: utf-8

# # Detailed example
# 
# This overview of the most important functions repeats the previous 30-seconds-example, but in more detail and shows additional functionality and alternative steps.

# ## Authentificate & access project

# In[ ]:


import up42
up42.authenticate(project_id="12345", project_api_key="12345")
#up42.authenticate(cfg_file="config.json")

project = up42.initialize_project()
project


# Get information about the available blocks to later construct your workflow.

# In[ ]:


up42.get_blocks(basic=True)


# ## Create or access the workflow
# You can either create a new workflow, use project.get_workflows() to get all existing workflows within the project, or access an exisiting workflow directly via its workflow_id.

# A new workflow is created and filled with tasks ([Sentinel-2 data](https://marketplace.up42.com/block/018dfb34-fc19-4334-8125-14fd7535f979), 
# [Land-Surface-Temperature](https://marketplace.up42.com/block/34767300-5caf-472b-a684-a351212b5c14)). 
# The area of interest and workflow parameters are defined. After running the job, 
# the results are downloaded and visualized.

# In[ ]:


# Create a new, empty workflow.

workflow = project.create_workflow(name="30-seconds-workflow", use_existing=False)
workflow


# In[ ]:


# Add workflow tasks

input_tasks = ["Sentinel-2 Level 2 (BOA) AOI clipped", "Land Surface Temperature Estimation"]
workflow.add_workflow_tasks(input_tasks=input_tasks)

# Check the added tasks.
workflow.get_workflow_tasks(basic=True)


# In[ ]:


# Alternative: Get all existing workflows within the project.

all_workflows = project.get_workflows()
workflow = all_workflows[0]
workflow


# In[ ]:


# Alternative: Directly access the existing workflow the id (has to exist within the accessed project).

UP42_WORKFLOW_ID="7fb2ec8a-45be-41ad-a50f-98ba6b528b98"
workflow = up42.initialize_workflow(workflow_id=UP42_WORKFLOW_ID)

workflow


# ## Select the aoi
# 
# There are multiple ways to select an aoi, you can:
# - Provide aoi the geometry directly in code as a FeatureCollection, Feature, GeoDataFrame, shapely Polygon or list of bounds coordinates.
# - Use up42.draw_aoi() to draw the aoi and export it as a geojson.
# - Use up42.read_vector_file() to read a geojson, json, shapefile, kml or wkt file.
# - Use up42.get_example_aoi() to read multiple provided sample aois.

# In[ ]:


aoi = [13.375966, 52.515068, 13.378314, 52.516639]


# In[ ]:


aoi = up42.read_vector_file("data/aoi_berlin.geojson", as_dataframe=True)
aoi.head(1)


# In[ ]:


#aoi = up42.get_example_aoi(location="Berlin")
#aoi


# In[ ]:


#up42.draw_aoi()


# ## Select the workflow parameters
# 
# There are also multiple ways to construct the workflow input parameters, you can:
# - Provide the parameters directly in code as a json string.
# - Use .get_parameters_info() to get a an overview of all potential parameters for the selected workflow and information about the parameter defaults and ranges.
# - Use .get_input_parameters(aoi_type="bbox", aoi_geometry=aoi) to construct the parameters with the provided aoi and all default parameters. Selecting the aoi_type is independent from the provided aoi, you can e.g. provide a irregular Polygon and still select aoi_type="bbox", then the bounding box of the polygon will be selected.

# In[ ]:


workflow.get_parameters_info()


# In[ ]:


input_parameters = workflow.construct_parameters(geometry=aoi, geometry_operation="bbox", limit=1)

# Further update the input_parameters manually
input_parameters["sentinelhub-s2-aoiclipped:1"].update({"max_cloud_cover":10})
input_parameters


# # Price estimation & Test Job

# In[ ]:


workflow.estimate_job(input_parameters)


# In[ ]:


# Run a test job to query data availability and check the configuration.
test_job = workflow.test_job(input_parameters=input_parameters, track_status=True)
test_results = test_job.get_results_json()
print(test_results)


# ## Run the workflow & download results

# In[ ]:


# Run the actual job.
job = workflow.run_job(input_parameters=input_parameters, track_status=True)


# <p align="center">
#     <img src="_assets/job_running.png" width="700">
# </p>

# ## Download & Display results

# In[ ]:


# Download job result (default downloads to Desktop). Only works after download is finished.
results_fp = job.download_results()


# In[ ]:


job.plot_results(figsize=(6,6), bands=[1], cmap="YlOrBr")


# In[ ]:


#job.map_results(bands=[1])