filter_prewitt.py 963 Bytes
title = "Prewitt Filter"
tip = "2-Dim Prewitt filter"
onein = True

import numpy as np
from scipy.ndimage import filters

from guidata.dataset.datatypes import DataSet
from guidata.dataset.dataitems import (IntItem, FloatArrayItem, StringItem,
                                       ChoiceItem, FloatItem, DictItem,
                                       BoolItem)
from guiqwt.config import _
from guiqwt.plot import CurveDialog
from guidata.qt.QtGui import QSplitter, QListWidget

class NOD3_App():

    def __init__(self, parent):
        self.parent = parent
        self.parent.activateWindow()

    def compute_app(self, **args):
        name = title.replace(" ", "")
        param = None
        self.parent.compute_11(name, lambda m, p: self.function(m, p), param, onein)

    def function(self, m, p):
        data = self.parent.nan_check(m.data, np.nanmin(m.data))
        m.data = filters.prewitt(data, mode='constant', cval=np.nan)
        return m, p