Update helpers/functional.py

helpers/functional.py

@@ -66,7 +66,7 @@ def cell2row(grid_string, meta_df, return_row = False):
     else:
         return None
 
-def map_to_image(map, return_centre=False, l2a=True):
+def map_to_image(map, return_centre=False, return_gridcell=False, l2a=True):
 
     # 1. get bounds
     bbox = map.get_bbox()
@@ -81,13 +81,16 @@ def map_to_image(map, return_centre=False, l2a=True):
 
     if row is not None:
         parquet_url, parquet_row, meta_row = row
+        print(meta_row)
         img = row2image(parquet_url, parquet_row)
         # 4. acquire image # X. update map
         lat, lon = meta_row.centre_lat.item(), meta_row.centre_lon.item()
 
+        ret = [img]
         if return_centre:
-            return img, (lat,lon)
-        else:
-            return img
+            ret.append((lat,lon))
+        if return_gridcell:
+            ret.append(meta_row.grid_cell.item())
+        return ret
     else:
         return None