Merge pull request #2 from mudhoney/feature/faster-multibatch

Feature/faster multibatch

Author: mudhoney

app/hgs2hpc.py

@@ -3,6 +3,7 @@
 import astropy.units as u
 from sunpy.coordinates import frames, transform_with_sun_center
 from sunpy.physics.differential_rotation import solar_rotate_coordinate
+from typing import List, Dict
 
 from frames import get_helioviewer_frame, get_earth_frame
 
@@ -26,6 +27,7 @@ def hgs2hpc(lat: float, lon: float, coord_time: Time, target: Time) -> SkyCoord:
         Desired observation time
     """
     hv_frame = get_helioviewer_frame(target)
+
     with transform_with_sun_center():
         coord = SkyCoord(
             lon * u.deg,
@@ -37,3 +39,43 @@ def hgs2hpc(lat: float, lon: float, coord_time: Time, target: Time) -> SkyCoord:
         hpc = coord.transform_to(earth_frame)
         # Then apply the rotation as seen from Helioviewer
         return solar_rotate_coordinate(hpc, hv_frame.observer)
+
+
+def hgs2hpc_batch(coordinates: List[Dict], target: Time) -> List[Dict]:
+    """
+    Batch process multiple HGS to HPC coordinate transformations
+
+    Parameters
+    ----------
+    coordinates : List[Dict]
+        List of coordinate dictionaries with keys: lat, lon, coord_time
+    target : Time
+        Target observation time (same for all coordinates)
+
+    Returns
+    -------
+    List[Dict]
+        List of results with keys: x, y
+    """
+    if not coordinates:
+        return []
+
+    hv_frame = get_helioviewer_frame(target)
+
+    with transform_with_sun_center():
+        lats = [c["lat"] for c in coordinates]
+        lons = [c["lon"] for c in coordinates]
+        coord_time = [c["coord_time"] for c in coordinates]
+        coord = SkyCoord(
+            lons,
+            lats,
+            unit="deg,deg",
+            frame=frames.HeliographicStonyhurst,
+            obstime=coord_time,
+        )
+        # First convert to an hpc coordinate
+        earth_frame = get_earth_frame(coord_time)
+        hpc = coord.transform_to(earth_frame)
+        # Then apply the rotation as seen from Helioviewer
+        result = solar_rotate_coordinate(hpc, hv_frame.observer)
+    return [{"x": c.Tx.value.item(), "y": c.Ty.value.item()} for c in result]

app/main.py

@@ -4,7 +4,7 @@
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import Field
 
-from hgs2hpc import hgs2hpc
+from hgs2hpc import hgs2hpc, hgs2hpc_batch
 from normalizer import normalize_hpc, gse_frame, jsonify_skycoord
 from ephemeris import get_position
 from validation import AstropyTime, HvBaseModel
@@ -64,12 +64,15 @@ class Hgs2HpcBatchInput(HvBaseModel):
 )
 def _hgs2hpc_post(params: Hgs2HpcBatchInput):
     "Convert a latitude/longitude coordinate to the equivalent helioprojective coordinate at the given target time"
-    coords = map(
-        lambda c: hgs2hpc(c.lat, c.lon, c.coord_time, params.target), params.coordinates
-    )
-    return {
-        "coordinates": [{"x": coord.Tx.value, "y": coord.Ty.value} for coord in coords]
-    }
+    # Prepare coordinates for batch processing
+    coords_input = [
+        {"lat": c.lat, "lon": c.lon, "coord_time": c.coord_time}
+        for c in params.coordinates
+    ]
+
+    results = hgs2hpc_batch(coords_input, params.target)
+
+    return {"coordinates": results}
 
 
 class NormalizeHpcQueryParameters(HvBaseModel):