Skip to content

Fit selector

FitSelector

Source code in itfit/utils/fit_selector.py 
 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
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
class FitSelector:
    def __init__(self, app: Fitter):
        """Allows to select fits inside a figure, via click.

        Args:
            app (Fitter): Fitter aplication source of fits and data.
        """
        self.app = app

        self.fig, self.ax = plt.subplots()
        x_data, y_data = self.app.data.get_data().T
        line_data, = self.ax.plot(x_data, y_data, '.', c='black', label="data")

        self._key : int|list[int]
        self._mode_: str
        self.fit_lines: list[Line2D] = [line_data]
        self.legend_to_lines_map: dict[Line2D, Line2D] = {}
        self.lines_to_hash: dict[Line2D, int] = {}

        for key, fit in self.app.fits.items():
            line, = self.ax.plot(fit.get_fit_xdata(), fit.get_fit_ydata(), label=f"{fit.fit_manager.name}:{key}")
            self.fit_lines.append(line)
            self.lines_to_hash.update({line : key})


    def set_multiple_selection_mode(self):
        """Starts multiple selection mode picker and sets title.
        """
        self.ax.set_title('Click on legend line to toggle line on/off')
        self._mode_ = 'm'
        legend = self.ax.legend()

        for legend_line, original_line in zip(legend.get_lines(), self.fit_lines):
            legend_line.set_picker(True)
            self.legend_to_lines_map[legend_line] = original_line

        self.fig.canvas.mpl_connect('pick_event', self.on_pick_multiple)
        self.fig.show()

    def set_single_selection_mode(self):
        """Starts single selection mode picker and sets title.
        """
        self.ax.set_title('Click line to select it')
        self._mode_ = 's'
        for line in self.fit_lines:
            line.set_picker(True)

        self.fig.canvas.mpl_connect('pick_event', self.on_pick_single)
        self.fig.show()


    def on_pick_multiple(self, event):
        """Pick event for multiple selection mode.
        """
        legline = event.artist
        origline = self.legend_to_lines_map[legline]
        visible = not origline.get_visible()
        origline.set_visible(visible)

        legline.set_alpha(1.0 if visible else 0.2)
        self.fig.canvas.draw()

    def on_pick_single(self, event):
        """Pick event for single selection mode.
        """
        k = self.lines_to_hash.get(event.artist)
        if isinstance(k, int):
            self._key = k
        self.fig.canvas.stop_event_loop()
        return 

    def get_selected_from_figure(self):
        """Gets the lines selected on multiple selection mode based on visibility.
        """
        result: list[int] = []
        for line in self.fit_lines:
            if line.get_visible():
                if val := self.lines_to_hash.get(line):
                    result.append(val) 
        self.fig.canvas.stop_event_loop()
        self._key = result

    def get_selected(self):
        """Returns the selected fit/s.

        Returns:
            (int, list[int]): Keys of fits selected.
        """
        if self._mode_ == 'm':
            self.get_selected_from_figure()
        plt.close(self.fig)
        return self._key

    def connect_select_one(self):
        """Starts fit selection figure and waits for it to finish.

        Returns:
            (itFit.FitSelector): returns itself.
        """
        self._key = False
        self.set_single_selection_mode()

        if len(self.app.fits.keys()) == 1:
            self._key = self.app._last_fit
            return self

        self.fig.show()
        self.fig.canvas.start_event_loop()
        return self


    def connect_select_multiple(self):
        """Starts fit selection figure and waits for it to finish.

        Returns:
            (itFit.FitSelector): returns itself.
        """
        self._key = False
        self.set_multiple_selection_mode()
        self.fig.show()
        self.fig.canvas.start_event_loop()
        return self

__init__(app)

Allows to select fits inside a figure, via click.

Parameters:

Name Type Description Default
app Fitter

Fitter aplication source of fits and data.

 required
Source code in itfit/utils/fit_selector.py 
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
def __init__(self, app: Fitter):
    """Allows to select fits inside a figure, via click.

    Args:
        app (Fitter): Fitter aplication source of fits and data.
    """
    self.app = app

    self.fig, self.ax = plt.subplots()
    x_data, y_data = self.app.data.get_data().T
    line_data, = self.ax.plot(x_data, y_data, '.', c='black', label="data")

    self._key : int|list[int]
    self._mode_: str
    self.fit_lines: list[Line2D] = [line_data]
    self.legend_to_lines_map: dict[Line2D, Line2D] = {}
    self.lines_to_hash: dict[Line2D, int] = {}

    for key, fit in self.app.fits.items():
        line, = self.ax.plot(fit.get_fit_xdata(), fit.get_fit_ydata(), label=f"{fit.fit_manager.name}:{key}")
        self.fit_lines.append(line)
        self.lines_to_hash.update({line : key})

connect_select_multiple()

Starts fit selection figure and waits for it to finish.

Returns:

Type Description
itFit.FitSelector

returns itself.
Source code in itfit/utils/fit_selector.py 
136
137
138
139
140
141
142
143
144
145
146
def connect_select_multiple(self):
    """Starts fit selection figure and waits for it to finish.

    Returns:
        (itFit.FitSelector): returns itself.
    """
    self._key = False
    self.set_multiple_selection_mode()
    self.fig.show()
    self.fig.canvas.start_event_loop()
    return self

connect_select_one()

Starts fit selection figure and waits for it to finish.

Returns:

Type Description
itFit.FitSelector

returns itself.
Source code in itfit/utils/fit_selector.py 
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
def connect_select_one(self):
    """Starts fit selection figure and waits for it to finish.

    Returns:
        (itFit.FitSelector): returns itself.
    """
    self._key = False
    self.set_single_selection_mode()

    if len(self.app.fits.keys()) == 1:
        self._key = self.app._last_fit
        return self

    self.fig.show()
    self.fig.canvas.start_event_loop()
    return self

get_selected()

Returns the selected fit/s.

Returns:

Type Description
int, list[int]

Keys of fits selected.
Source code in itfit/utils/fit_selector.py 
107
108
109
110
111
112
113
114
115
116
def get_selected(self):
    """Returns the selected fit/s.

    Returns:
        (int, list[int]): Keys of fits selected.
    """
    if self._mode_ == 'm':
        self.get_selected_from_figure()
    plt.close(self.fig)
    return self._key

get_selected_from_figure()

Gets the lines selected on multiple selection mode based on visibility.

Source code in itfit/utils/fit_selector.py 
 96
 97
 98
 99
100
101
102
103
104
105
def get_selected_from_figure(self):
    """Gets the lines selected on multiple selection mode based on visibility.
    """
    result: list[int] = []
    for line in self.fit_lines:
        if line.get_visible():
            if val := self.lines_to_hash.get(line):
                result.append(val) 
    self.fig.canvas.stop_event_loop()
    self._key = result

on_pick_multiple(event)

Pick event for multiple selection mode.

Source code in itfit/utils/fit_selector.py 
76
77
78
79
80
81
82
83
84
85
def on_pick_multiple(self, event):
    """Pick event for multiple selection mode.
    """
    legline = event.artist
    origline = self.legend_to_lines_map[legline]
    visible = not origline.get_visible()
    origline.set_visible(visible)

    legline.set_alpha(1.0 if visible else 0.2)
    self.fig.canvas.draw()

on_pick_single(event)

Pick event for single selection mode.

Source code in itfit/utils/fit_selector.py 
87
88
89
90
91
92
93
94
def on_pick_single(self, event):
    """Pick event for single selection mode.
    """
    k = self.lines_to_hash.get(event.artist)
    if isinstance(k, int):
        self._key = k
    self.fig.canvas.stop_event_loop()
    return

set_multiple_selection_mode()

Starts multiple selection mode picker and sets title.

Source code in itfit/utils/fit_selector.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
def set_multiple_selection_mode(self):
    """Starts multiple selection mode picker and sets title.
    """
    self.ax.set_title('Click on legend line to toggle line on/off')
    self._mode_ = 'm'
    legend = self.ax.legend()

    for legend_line, original_line in zip(legend.get_lines(), self.fit_lines):
        legend_line.set_picker(True)
        self.legend_to_lines_map[legend_line] = original_line

    self.fig.canvas.mpl_connect('pick_event', self.on_pick_multiple)
    self.fig.show()

set_single_selection_mode()

Starts single selection mode picker and sets title.

Source code in itfit/utils/fit_selector.py 
64
65
66
67
68
69
70
71
72
73
def set_single_selection_mode(self):
    """Starts single selection mode picker and sets title.
    """
    self.ax.set_title('Click line to select it')
    self._mode_ = 's'
    for line in self.fit_lines:
        line.set_picker(True)

    self.fig.canvas.mpl_connect('pick_event', self.on_pick_single)
    self.fig.show()