fit_removeBack.py
2.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
title = "Remove Background"
tip = "2-Dim background remove 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 ListBox(QSplitter):
def __init__(self, parent):
QSplitter.__init__(self, parent)
self.imagelist = QListWidget(self)
self.addWidget(self.imagelist)
self.properties = DataSetEditGroupBox(_("Properties"), self.ImageParam)
self.properties.setEnabled(False)
self.addWidget(self.properties)
class NOD3_App():
def __init__(self, parent):
self.parent = parent
self.parent.activateWindow()
def compute_app(self, **args):
class Param(DataSet):
Box = IntItem("Box:", default=11, min=5)
Filter = ChoiceItem("Filter", (("Median", "Median"), ("Percentile", "Percentile"),
("Rank", "Rank")),
default = "Percentile")
name = title.replace(" ", "")
if args == {}:
param = Param(title.replace(" ", ""), "2-dim background filter<br>")
else:
param = self.parent.ScriptParameter(name, args)
param = self.parent.read_defaults(param)
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))
if p.Filter == "Median":
m.data -= filters.median_filter(data, size=(p.Box, p.Box))
elif p.Filter == "Percentile":
m.data -= filters.percentile_filter(data, p.Box, size=(p.Box, p.Box), mode='mirror')
elif p.Filter == "Rank":
m.data -= filters.rank_filter(data, p.Box, size=(p.Box, p.Box), mode='mirror')
return m, p