Support roll angle functionality in future versions of pywwt.

glue-viz · Mar 8, 2023 · 4107043 · 4107043
1 parent 2548e39
commit 4107043
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/glue_wwt/viewer/data_viewer.py b/glue_wwt/viewer/data_viewer.py
@@ -84,6 +84,8 @@ def __gluestate__(self, context):
                 "dec": center.dec.deg,
                 "fov": self._wwt.get_fov().value
             }
+            if hasattr(self._wwt, 'get_roll'):
+                camera["roll"] = self._wwt.get_roll().value
             state["camera"] = camera
         except ViewerNotAvailableError:
             logger.error("Unable to export camera parameters as WWT viewer is not responding.")
@@ -97,5 +99,9 @@ def __setgluestate__(cls, rec, context):
             ra = camera.get("ra", 0)
             dec = camera.get("dec", 0)
             fov = camera.get("fov", 60)
-            viewer._wwt.center_on_coordinates(SkyCoord(ra, dec, unit=u.deg), fov=fov * u.deg, instant=True)
+            roll = camera.get("roll", None)
+            camera_kwargs = dict(fov=fov * u.deg, instant=True)
+            if hasattr(viewer._wwt, 'get_roll') and roll is not None:
+                camera_kwargs["roll"] = roll * u.deg
+            viewer._wwt.center_on_coordinates(SkyCoord(ra, dec, unit=u.deg), **camera_kwargs)
         return viewer