Update helpers/functional.py

helpers/functional.py

@@ -6,6 +6,7 @@ import pandas as pd
 from io import BytesIO
 import os
 from PIL import Image
+import datetime
 
 # GLOBAL VARIABLES
 if os.path.isfile('helpers/s2l2a_metadata.parquet'):
@@ -20,9 +21,26 @@ else:
     DATASET_NAME = 'Major-TOM/Core-S2L1C'
     l1c_meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
 
+if os.path.isfile('helpers/s1rtc_metadata.parquet'):
+    rtc_meta_path = 'helpers/s1rtc_metadata.parquet'
+else:
+    DATASET_NAME = 'Major-TOM/Core-S1RTC'
+    rtc_meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
+
 grid = Grid(10, latitude_range=(-90,90), longitude_range=(-180,180))
 l2a_df = pd.read_parquet(l2a_meta_path)
 l1c_df = pd.read_parquet(l1c_meta_path)
+rtc_df = pd.read_parquet(rtc_meta_path)
+
+df_dict = {
+    'S2-L2A' : l2a_df,
+    'S2-L1C' : l1c_df,
+    'S1-RTC' : rtc_df
+}
+
+def pretty_date(input):
+    template = '%Y%m%dT%H%M%S' if 'T' in input else '%Y%m%d%H%M%S'
+    return datetime.datetime.strptime(input, template).strftime('%H:%M:%S - %d %b %Y')
 
 # HELPER FUNCTIONS
 def gridcell2ints(grid_string):
@@ -66,7 +84,7 @@ def cell2row(grid_string, meta_df, return_row = False):
     else:
         return None
 
-def map_to_image(map, return_centre=False, return_gridcell=False, l2a=True):
+def map_to_image(map, return_centre=False, return_gridcell=False, return_timestamp=False, source='S2-L2A'):
 
     try:
         # 1. get bounds
@@ -79,7 +97,7 @@ def map_to_image(map, return_centre=False, return_gridcell=False, l2a=True):
     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
+    df = df_dict[source]
     row = cell2row("{}_{}".format(rows[0],cols[0]), df, return_row = True)
 
     if row is not None:
@@ -94,6 +112,9 @@ def map_to_image(map, return_centre=False, return_gridcell=False, l2a=True):
             ret.append((lat,lon))
         if return_gridcell:
             ret.append(meta_row.grid_cell.item())
+        if return_timestamp:
+            ret.append(pretty_date(meta_row.timestamp.item()))
+            
         return ret
     else:
         return None