self referencing with reference input

orangecontrib/snom/widgets/owpreprocessimage.py

@@ -12,7 +12,7 @@
 
 import Orange.data
 from Orange import preprocess
-from Orange.widgets.widget import Output
+from Orange.widgets.widget import Output, Input
 
 from orangecontrib.spectroscopy.widgets.owhyper import BasicImagePlot
 
@@ -131,8 +131,17 @@ def progress_interrupt(i: float):
         state.set_partial_result((n, data, None))
         return orig_data, data
 
+class SpectralImagePreprocessReference(SpectralImagePreprocess, openclass=True):
+    class Inputs(SpectralImagePreprocess.Inputs):
+        reference = Input("Reference", Orange.data.Table)
+
+    @Inputs.reference
+    def set_reference(self, reference):
+        self.reference_data = reference
+
+
+class OWPreprocessImage(SpectralImagePreprocessReference):
 
-class OWPreprocessImage(SpectralImagePreprocess):
     name = "Preprocess image"
     id = "orangecontrib.snom.widgets.preprocessimage"
     description = "Process image"

orangecontrib/snom/widgets/preprocessors/__init__.py

@@ -6,3 +6,4 @@
 from orangecontrib.snom.widgets.preprocessors import phase_rotation  # noqa: F401
 from orangecontrib.snom.widgets.preprocessors import simple_normalize  # noqa: F401
 from orangecontrib.snom.widgets.preprocessors import linelevel  # noqa: F401
+from orangecontrib.snom.widgets.preprocessors import self_reference # noqa: F401

orangecontrib/snom/widgets/preprocessors/self_reference.py

@@ -0,0 +1,70 @@
+from AnyQt.QtWidgets import QFormLayout, QLabel
+
+from orangecontrib.spectroscopy.widgets.preprocessors.utils import BaseEditorOrange, REFERENCE_DATA_PARAM
+from orangecontrib.spectroscopy.widgets.gui import lineEditIntRange
+
+from pySNOM.images import SelfReference
+
+from orangecontrib.snom.preprocess.utils import (
+    PreprocessImageOpts2DOnlyWhole,
+)
+from orangecontrib.snom.widgets.preprocessors.registry import preprocess_image_editors
+import numpy as np
+
+class SelfRef(PreprocessImageOpts2DOnlyWhole):
+    def __init__(self, reference):
+        self.reference = reference.X
+
+    def transform_image(self, image):
+        ref = (np.reshape(self.reference,np.shape(image)))
+        d = SelfReference(referencedata=ref).transform(
+            image
+        )
+        return d
+
+
+class SelfRefEditor(BaseEditorOrange):
+    name = "Self-referencing"
+    qualname = "orangecontrib.snom.self_reference"
+
+    def __init__(self, parent=None, **kwargs):
+        super().__init__(parent, **kwargs)
+
+        self.reference = None
+
+        form = QFormLayout()
+        self.reference_info = QLabel("Reference data from input!")
+        form.addRow(self.reference_info)
+        self.controlArea.setLayout(form)
+
+    def activateOptions(self):
+        pass  # actions when user starts changing options
+
+    def setParameters(self, params):
+        self.update_reference_info()
+
+    def set_reference_data(self, reference):
+        self.reference = reference
+        self.update_reference_info()
+
+    def update_reference_info(self):
+        if not self.reference:
+            self.reference_info.setText("Reference: missing!")
+            self.reference_info.setStyleSheet("color: red")
+        else:
+            rinfo = "Reference order: N"
+            self.reference_info.setText(rinfo)
+            self.reference_info.setStyleSheet("color: black")
+
+    @classmethod
+    def createinstance(cls, params):
+        params = dict(params)
+        reference = params.get(REFERENCE_DATA_PARAM, None)
+        return SelfRef(reference=reference)
+
+    def set_preview_data(self, data):
+        if data:
+            pass  # TODO any settings
+
+
+preprocess_image_editors.register(SelfRefEditor, 400)