1 files changed, 147 insertions, 0 deletions
diff --git a/src/silx/gui/plot/actions/medfilt.py b/src/silx/gui/plot/actions/medfilt.py
new file mode 100644
index 0000000..f86a377
--- /dev/null
+++ b/src/silx/gui/plot/actions/medfilt.py
@@ -0,0 +1,147 @@
+# coding: utf-8
+# /*##########################################################################
+#
+# Copyright (c) 2004-2020 European Synchrotron Radiation Facility
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+# ###########################################################################*/
+"""
+:mod:`silx.gui.plot.actions.medfilt` provides a set of QAction to apply filter
+on data contained in a :class:`.PlotWidget`.
+
+The following QAction are available:
+
+- :class:`MedianFilterAction`
+- :class:`MedianFilter1DAction`
+- :class:`MedianFilter2DAction`
+
+"""
+
+from __future__ import division
+
+__authors__ = ["V.A. Sole", "T. Vincent", "P. Knobel"]
+__license__ = "MIT"
+
+__date__ = "10/10/2018"
+
+from .PlotToolAction import PlotToolAction
+from silx.gui.widgets.MedianFilterDialog import MedianFilterDialog
+from silx.math.medianfilter import medfilt2d
+import logging
+
+_logger = logging.getLogger(__name__)
+
+
+class MedianFilterAction(PlotToolAction):
+    """QAction to plot the pixels intensities diagram
+
+    :param plot: :class:`.PlotWidget` instance on which to operate
+    :param parent: See :class:`QAction`
+    """
+
+    def __init__(self, plot, parent=None):
+        PlotToolAction.__init__(self,
+                                plot,
+                                icon='median-filter',
+                                text='median filter',
+                                tooltip='Apply a median filter on the image',
+                                parent=parent)
+        self._originalImage = None
+        self._legend = None
+        self._filteredImage = None
+
+    def _createToolWindow(self):
+        popup = MedianFilterDialog(parent=self.plot)
+        popup.sigFilterOptChanged.connect(self._updateFilter)
+        return popup
+
+    def _connectPlot(self, window):
+        PlotToolAction._connectPlot(self, window)
+        self.plot.sigActiveImageChanged.connect(self._updateActiveImage)
+        self._updateActiveImage()
+
+    def _disconnectPlot(self, window):
+        PlotToolAction._disconnectPlot(self, window)
+        self.plot.sigActiveImageChanged.disconnect(self._updateActiveImage)
+
+    def _updateActiveImage(self):
+        """Set _activeImageLegend and _originalImage from the active image"""
+        self._activeImageLegend = self.plot.getActiveImage(just_legend=True)
+        if self._activeImageLegend is None:
+            self._originalImage = None
+            self._legend = None
+        else:
+            self._originalImage = self.plot.getImage(self._activeImageLegend).getData(copy=False)
+            self._legend = self.plot.getImage(self._activeImageLegend).getName()
+
+    def _updateFilter(self, kernelWidth, conditional=False):
+        if self._originalImage is None:
+            return
+
+        self.plot.sigActiveImageChanged.disconnect(self._updateActiveImage)
+        filteredImage = self._computeFilteredImage(kernelWidth, conditional)
+        self.plot.addImage(data=filteredImage,
+                           legend=self._legend,
+                           replace=True)
+        self.plot.sigActiveImageChanged.connect(self._updateActiveImage)
+
+    def _computeFilteredImage(self, kernelWidth, conditional):
+        raise NotImplementedError('MedianFilterAction is a an abstract class')
+
+    def getFilteredImage(self):
+        """
+        :return: the image with the median filter apply on"""
+        return self._filteredImage
+
+
+class MedianFilter1DAction(MedianFilterAction):
+    """Define the MedianFilterAction for 1D
+
+    :param plot: :class:`.PlotWidget` instance on which to operate
+    :param parent: See :class:`QAction`
+    """
+    def __init__(self, plot, parent=None):
+        MedianFilterAction.__init__(self,
+                                    plot,
+                                    parent=parent)
+
+    def _computeFilteredImage(self, kernelWidth, conditional):
+        assert(self.plot is not None)
+        return medfilt2d(self._originalImage,
+                         (kernelWidth, 1),
+                         conditional)
+
+
+class MedianFilter2DAction(MedianFilterAction):
+    """Define the MedianFilterAction for 2D
+
+    :param plot: :class:`.PlotWidget` instance on which to operate
+    :param parent: See :class:`QAction`
+    """
+    def __init__(self, plot, parent=None):
+        MedianFilterAction.__init__(self,
+                                    plot,
+                                    parent=parent)
+
+    def _computeFilteredImage(self, kernelWidth, conditional):
+        assert(self.plot is not None)
+        return medfilt2d(self._originalImage,
+                         (kernelWidth, kernelWidth),
+                         conditional)