def __init__(self, flex):
"""Initialize box."""
super(FittingFunctionBox, self).__init__(
style=Pack(direction=COLUMN, flex=flex)
)
fit_function_box = LineBox()
fit_function_box.add(toga.Label(text="Fitting function:"))
self.fitting_function_selection = toga.Selection(
on_select=self.load_select_fit_function_name,
)
fit_function_box.add(self.fitting_function_selection)
self.fitting_function_syntax = toga.TextInput(
readonly=True,
style=Pack(flex=1, padding_left=BIG_PADDING, padding_right=BIG_PADDING),
)
fit_function_box.add(self.fitting_function_syntax)
self.load_module_button = toga.Button(
label="Load module", on_press=self.load_module
)
fit_function_box.add(self.load_module_button)
self.add(fit_function_box)
self.parameters_box = LineBox()
self.parameters_box.add(toga.Label("Parameters:"))
self.parameter_inputs = toga.TextInput(
readonly=True,
style=Pack(width=PARAMETER_WIDTH),
on_change=lambda widget: self.reset_fit_function(),
)
self.parameters_box.add(self.parameter_inputs)
self.add(self.parameters_box)
self.update_fitting_function_options()
class FittingFunctionBox(toga.Box): # pylint: disable=too-many-instance-attributes
"""Visual box instance for choosing a fitting function."""
fitting_function_box: LineBox
fitting_function_selection: toga.Selection
fitting_function_syntax: toga.TextInput
polynomial_degree_title: toga.Label
polynomial_degree_input: toga.NumberInput
load_module_button: toga.Button
__fitting_function: Optional[FittingFunction] = None
__on_fitting_function_load: Optional[Callable[[FittingFunction],
None]] = None
__polynomial_is_set: bool = False
def __init__(self, flex):
"""Initialize box."""
super().__init__(style=Pack(direction=COLUMN, flex=flex))
self.fitting_function_box = LineBox()
self.fitting_function_box.add(toga.Label(text="Fitting function:"))
self.fitting_function_selection = toga.Selection(
on_select=self.load_select_fitting_function_name, )
self.fitting_function_box.add(self.fitting_function_selection)
self.fitting_function_syntax = toga.TextInput(
readonly=True,
style=Pack(flex=1,
padding_left=BIG_PADDING,
padding_right=BIG_PADDING),
)
self.fitting_function_box.add(self.fitting_function_syntax)
self.load_module_button = toga.Button(label="Load module",
on_press=self.load_module)
self.fitting_function_box.add(self.load_module_button)
self.add(self.fitting_function_box)
self.polynomial_degree_title = toga.Label("Degree:")
self.polynomial_degree_input = toga.NumberInput(
min_value=1,
max_value=5,
default=1,
on_change=lambda widget: self.set_polynomial_degree(),
)
self.update_fitting_function_options()
@property
def fitting_function(self):
"""Getter of the fitting function."""
return self.__fitting_function
@fitting_function.setter
def fitting_function(self, fitting_function):
"""
Setter of the fitting function.
After setting the fit function, run handlers in order to notify other
components of the change.
"""
self.__fitting_function = fitting_function
if self.on_fitting_function_load is not None:
self.on_fitting_function_load(self.fitting_function)
@property
def fitting_function_state(self):
"""Set fit function state."""
return self.fitting_function_selection.value
@property
def on_fitting_function_load(self):
"""on_fitting_function_load getter."""
return self.__on_fitting_function_load
@on_fitting_function_load.setter
def on_fitting_function_load(self, on_fitting_function_load):
"""on_fitting_function_load setter."""
self.__on_fitting_function_load = on_fitting_function_load
def update_fitting_function_options(self):
"""Update the fitting functions options."""
self.fitting_function_selection.items = [NO_VALUE, POLYNOMIAL] + list(
FittingFunctionsRegistry.names())
def load_select_fitting_function_name(self, widget): # pylint: disable=unused-argument
"""Load the selection fitting function from the FittingFunctionRegistry."""
if self.fitting_function_state == POLYNOMIAL and not self.__polynomial_is_set:
self.fitting_function_box.insert(2, self.polynomial_degree_title)
self.fitting_function_box.insert(3, self.polynomial_degree_input)
self.set_polynomial_degree()
self.__polynomial_is_set = True
return
if self.fitting_function_state != POLYNOMIAL and self.__polynomial_is_set:
self.fitting_function_box.remove(self.polynomial_degree_title)
self.fitting_function_box.remove(self.polynomial_degree_input)
self.__polynomial_is_set = False
if self.fitting_function_state == NO_VALUE:
self.fitting_function = None
self.fitting_function_syntax.value = ""
return
self.fitting_function = FittingFunctionsRegistry.load(
self.fitting_function_state)
self.fitting_function_syntax.value = self.fitting_function.syntax
def load_module(self, widget): # pylint: disable=unused-argument
"""
Open a file dialog in order to load user module.
This is done in order to add costume fitting functions.
"""
try:
file_path = self.window.open_file_dialog(
title="Choose module file", multiselect=False)
except ValueError:
return
spec = importlib.util.spec_from_file_location("eddington.dummy",
file_path)
dummy_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(dummy_module)
self.update_fitting_function_options()
def set_polynomial_degree(self):
"""Set fitting function to be polynomial based on given degree."""
degree = int(self.polynomial_degree_input.value)
self.fitting_function = polynomial(degree)
if degree == 1:
self.fitting_function_syntax.value = "a[0] + a[1] * x"
else:
self.fitting_function_syntax.value = "a[0] + a[1] * x + " + " + ".join(
[f"a[{i}] * x ^ {i}" for i in range(2, degree + 1)])
class FittingFunctionBox(toga.Box):
"""Visual box instance for choosing a fitting function."""
fitting_function_selection: toga.Selection
fitting_function_syntax: toga.TextInput
parameters_box: LineBox
parameter_inputs: toga.TextInput
load_module_button: toga.Button
__fit_function: FitFunction = None
__handlers: List[Callable] = []
def __init__(self, flex):
"""Initialize box."""
super(FittingFunctionBox, self).__init__(
style=Pack(direction=COLUMN, flex=flex)
)
fit_function_box = LineBox()
fit_function_box.add(toga.Label(text="Fitting function:"))
self.fitting_function_selection = toga.Selection(
on_select=self.load_select_fit_function_name,
)
fit_function_box.add(self.fitting_function_selection)
self.fitting_function_syntax = toga.TextInput(
readonly=True,
style=Pack(flex=1, padding_left=BIG_PADDING, padding_right=BIG_PADDING),
)
fit_function_box.add(self.fitting_function_syntax)
self.load_module_button = toga.Button(
label="Load module", on_press=self.load_module
)
fit_function_box.add(self.load_module_button)
self.add(fit_function_box)
self.parameters_box = LineBox()
self.parameters_box.add(toga.Label("Parameters:"))
self.parameter_inputs = toga.TextInput(
readonly=True,
style=Pack(width=PARAMETER_WIDTH),
on_change=lambda widget: self.reset_fit_function(),
)
self.parameters_box.add(self.parameter_inputs)
self.add(self.parameters_box)
self.update_fitting_function_options()
def add_handler(self, handler):
"""
Add handler to run whenever the fitting function is updated.
:param handler: Callable
"""
self.__handlers.append(handler)
def update_fitting_function_options(self):
"""Update the fitting functions options."""
self.fitting_function_selection.items = [NO_VALUE] + list(
FitFunctionsRegistry.names()
)
def load_select_fit_function_name(self, widget): # pylint: disable=unused-argument
"""Load the selection fitting function from the FitFunctionRegistry."""
if self.fit_function_state == NO_VALUE:
self.reset_fit_function()
return
self.fit_function = FitFunctionsRegistry.load(self.fit_function_state)
self.fitting_function_syntax.value = self.fit_function.syntax
def load_module(self, widget): # pylint: disable=unused-argument
"""
Open a file dialog in order to load user module.
This is done in order to add costume fitting functions.
"""
try:
file_path = self.window.open_file_dialog(
title="Choose module file", multiselect=False
)
except ValueError:
return
spec = importlib.util.spec_from_file_location("eddington.dummy", file_path)
dummy_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(dummy_module)
self.update_fitting_function_options()
def reset_fit_function(self):
"""Set fit function to be None."""
self.fit_function = None
@property
def fit_function(self):
"""Getter of the fitting function."""
return self.__fit_function
@fit_function.setter
def fit_function(self, fit_function):
"""
Setter of the fitting function.
After setting the fit function, run handlers in order to notify other
components of the change.
"""
self.__fit_function = fit_function
for handler in self.__handlers:
handler(self.fit_function)
@property
def fit_function_state(self):
"""Set fit function state."""
return self.fitting_function_selection.value