Peter Müller / NOD3 single dish reduction software

title = "Threshold"
tip = "clips values lesser than 'min'"

import numpy as np

from guidata.dataset.datatypes import DataSet
from guidata.dataset.dataitems import (IntItem, FloatArrayItem, StringItem,
                                       ChoiceItem, FloatItem, DictItem,
                                       BoolItem)
from guiqwt.config import _

class NOD3_App:

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

    def compute_app(self):
        class Param(DataSet):
            Min = FloatItem('Min:', default=1.)
            Dummy = BoolItem('Dummy:', default=False)
        param = Param(_("Thresholding"), "truncate y < min")
        name = title.replace(" ", "")
        self.parent.compute_11(name, lambda m, p: self.function(m, p), param) 

    def function(self, m, p):
        m.data = np.clip(m.data, p.Min, m.data.max())
        if p.Dummy:
           m.data = np.where(m.data == p.Min, np.nan, m.data)
        return m, p