Update helpers/functional.py

helpers/functional.py

@@ -8,14 +8,21 @@ import os
 from PIL import Image
 
 # GLOBAL VARIABLES
-if os.path.isfile('metadata.parquet'):
-    meta_path = 'metadata.parquet'
+if os.path.isfile('helpers/s2l2a_metadata.parquet'):
+    l2a_meta_path = 'helpers/s2l2a_metadata.parquet'
 else:
     DATASET_NAME = 'Major-TOM/Core-S2L2A'
-    meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
+    l2a_meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
+
+if os.path.isfile('helpers/s2l1c_metadata.parquet'):
+    l1c_meta_path = 'helpers/s2l1c_metadata.parquet'
+else:
+    DATASET_NAME = 'Major-TOM/Core-S2L1C'
+    l1c_meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
 
 grid = Grid(10, latitude_range=(-90,90), longitude_range=(-180,180))
-df = pd.read_parquet(meta_path)
+l2a_df = pd.read_parquet(l2a_meta_path)
+l1c_df = pd.read_parquet(l1c_meta_path)
 
 # HELPER FUNCTIONS
 def gridcell2ints(grid_string):
@@ -59,7 +66,7 @@ def cell2row(grid_string, meta_df, return_row = False):
     else:
         return None
 
-def map_to_image(map, return_centre=False):
+def map_to_image(map, return_centre=False, l2a=True):
 
     # 1. get bounds
     bbox = map.get_bbox()
@@ -69,6 +76,7 @@ def map_to_image(map, return_centre=False):
     rows, cols = grid.latlon2rowcol([center[0]], [center[1]])
 
     # 3. translate major-tom cell to row in parquet
+    df = l2a_df if l2a else l1c_df
     row = cell2row("{}_{}".format(rows[0],cols[0]), df, return_row = True)
 
     if row is not None: