| import os | |
| import utils | |
| import rioxarray as rx | |
| from senHub import SenHub | |
| from sentinelhub import SHConfig, MimeType | |
| config = SHConfig() | |
| config.instance_id = '352670fb-2edf-4abd-90c8-437485a2403e' | |
| config.sh_client_id = 'ca95f10f-443c-4c60-9a36-98950292bb9b' | |
| config.sh_client_secret = 'rNFGRxGNiNFrXJfGyHIkVRyGOrdWNsfI' | |
| config.sh_timesfm_IP = "34.121.141.161" | |
| def Download_image_in_given_date(clientName, metric, df, field, date, mime_type = MimeType.TIFF): | |
| sen_obj = SenHub(config, mime_type = mime_type) | |
| download_path = f'./data/{clientName}/raw/{metric}/{date}/field_{field}/' | |
| bbox = utils.calculate_bbox(df, field) | |
| evalscript = utils.Scripts[metric] | |
| sen_obj.set_dir(download_path) | |
| sen_obj.make_bbox(bbox) | |
| sen_obj.make_request(evalscript, date) | |
| data = sen_obj.download_data() | |
| return data | |
| def mask_downladed_image(clientName, metric, df, field, date): | |
| download_path = utils.get_downloaded_location_img_path(clientName, metric, date, field) | |
| im = rx.open_rasterio(download_path) | |
| field_vals = df.loc[df['name'] == field] | |
| field_geom = field_vals.geometry | |
| crs = field_vals.crs | |
| clipped = im.rio.clip(field_geom, crs, drop=True) | |
| save_dir_path = f'./data/{clientName}/processed/{metric}/{date}/field_{field}/' | |
| os.makedirs(save_dir_path, exist_ok=True) | |
| save_tiff_path = save_dir_path + 'masked.tiff' | |
| clipped.rio.to_raster(save_tiff_path) | |
| return save_tiff_path | |
| def convert_maske_image_to_geodataframe(clientName, metric, df, field, date, crs): | |
| imagePath = utils.get_masked_location_img_path(clientName, metric, date, field) | |
| im = rx.open_rasterio(imagePath) | |
| gdf = utils.tiff_to_geodataframe(im, metric, date, crs) | |
| save_dir_path = f'./data/{clientName}/curated/{metric}/{date}/field_{field}/' | |
| os.makedirs(save_dir_path, exist_ok=True) | |
| save_geojson_path = save_dir_path + 'masked.geojson' | |
| gdf.to_file(save_geojson_path, driver='GeoJSON') | |
| return save_geojson_path | |