Merge pull request #32 from javerbukh/add_ellipse_roi

Add support for elliptical ROI
glue-viz · Jul 2, 2021 · 40aae09 · 40aae09
2 parents 930fd84 + 5376944
commit 40aae09
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/glue_astronomy/translators/regions.py b/glue_astronomy/translators/regions.py
@@ -2,11 +2,11 @@
 from glue.core.subset import RoiSubsetState, RangeSubsetState, OrState, AndState,\
                              XorState, MultiOrState, Subset, MultiRangeSubsetState
 from glue.core.roi import RectangularROI, PolygonalROI, CircularROI, PointROI,\
-                          RangeROI, AbstractMplRoi
+                          RangeROI, AbstractMplRoi, EllipticalROI
 from glue.viewers.image.pixel_selection_subset_state import PixelSubsetState
 
 from regions import RectanglePixelRegion, PolygonPixelRegion, CirclePixelRegion,\
-                    PointPixelRegion, PixCoord
+                    PointPixelRegion, PixCoord, EllipsePixelRegion
 
 
 def range_to_rect(data, ori, low, high):
@@ -57,7 +57,6 @@ def to_object(self, subset):
         subset : `glue.core.subset.Subset`
             The subset to convert to a Region object
         """
-
         data = subset.data
 
         if data.pixel_component_ids[0].axis == 0:
@@ -82,6 +81,8 @@ def to_object(self, subset):
                 return PolygonPixelRegion(PixCoord(roi.vx, roi.vy))
             elif isinstance(roi, CircularROI):
                 return CirclePixelRegion(PixCoord(*roi.get_center()), roi.get_radius())
+            elif isinstance(roi, EllipticalROI):
+                return EllipsePixelRegion(PixCoord(roi.xc, roi.yc), roi.radius_x, roi.radius_y)
             elif isinstance(roi, PointROI):
                 return PointPixelRegion(PixCoord(*roi.center()))
             elif isinstance(roi, RangeROI):