Peter Müller / NOD3 single dish reduction software

title = "Linear Scaling"
tip = "linear calibration to data (scaling)"

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, **args):
        class CalibrateParam(DataSet):
            a = FloatItem('a', default=1.)
            b = FloatItem('b', default=0.)
            unit = StringItem('unit', default="mk/Tcal")
        if args == {}:
           param = CalibrateParam(_("Linear calibration"), "y = a*x + b")
        else:
           param = self.parent.ScriptParameter(title, args)
        name = title.replace(" ", "")
        self.parent.compute_11(name, lambda m, p: self.function(m, p), param)

    def function(self, m, p):
        m.data = p.a*m.data + p.b
        m.header["BUNIT"] = str(p.unit)
        return m, p