get_tif analysis action

datalab/datalab_session/analysis/get_tif.py

@@ -0,0 +1,24 @@
+from datalab.datalab_session.util import create_tif, get_fits, add_file_to_bucket, key_exists, get_s3_url
+
+def get_tif(input: dict):
+  """
+    Checks bucket for tif file and returns the url
+    if the file doesn't exist, generates a new tif file
+    input: dict
+      basename: str
+      source: str
+  """
+
+  basename = input["basename"]
+  file_key = f'{basename}/{basename}.tif'
+
+  # Check in bucket for tif file
+  if(key_exists(file_key)):
+    tif_url = get_s3_url(file_key)
+  else:
+    # If tif file doesn't exist, generate a new tif file
+    fits_path = get_fits(basename)
+    tif_path = create_tif(basename, fits_path)
+    tif_url = add_file_to_bucket(file_key, tif_path)
+
+  return {"tif_url": tif_url}

datalab/datalab_session/util.py

@@ -10,7 +10,7 @@
 from botocore.exceptions import ClientError
 
 from django.conf import settings
-from fits2image.conversions import fits_to_jpg
+from fits2image.conversions import fits_to_jpg, fits_to_tif
 
 log = logging.getLogger()
 log.setLevel(logging.INFO)
@@ -176,6 +176,16 @@ def create_fits(key: str, image_arr: np.ndarray) -> str:
 
   return fits_path
 
+def create_tif(key: str, fits_path: np.ndarray) -> str:
+  """
+    Creates a full sized TIFF file from a FITs
+  """
+  height, width = get_fits_dimensions(fits_path)
+  tif_path = tempfile.NamedTemporaryFile(suffix=f'{key}.tif').name
+  fits_to_tif(fits_path, tif_path, width=width, height=height)
+
+  return tif_path
+
 def create_jpgs(cache_key, fits_paths: str, color=False) -> list:
     """
     Create jpgs from fits files and save them to S3

datalab/datalab_session/views.py

@@ -5,7 +5,7 @@
 from datalab.datalab_session.data_operations.utils import available_operations
 from datalab.datalab_session.analysis.line_profile import line_profile
 from datalab.datalab_session.analysis.source_catalog import source_catalog
-
+from datalab.datalab_session.analysis.get_tif import get_tif
 
 class OperationOptionsApiView(RetrieveAPIView):
     """ View to retrieve the set of operations available, for the UI to use """
@@ -31,6 +31,8 @@ def post(self, request, action):
                 output = line_profile(input)
             case 'source-catalog':
                 output = source_catalog(input)
+            case 'get-tif':
+                output = get_tif(input)
             case _:
                 raise Exception(f'Analysis action {action} not found')