graphica-plot 2.0.0__py3-none-any.whl

graphica/gui/tools/layout_edit.py

@@ -0,0 +1,285 @@
+"""自由配置のレイアウト: サブプロットをドラッグか数値で好きな位置と大きさに置く。
+
+位置は all_plot_settings[i]['free_rect'] に (left, bottom, width, height)(Figure に対する 0〜1)で持つ。
+ドラッグも数値の入力も、ax.set_position() と free_rect への書き込みという同じ経路を通す(食い違わないように)。
+"""
+import logging
+from graphica.core.axis_settings import axis_setting
+
+logger = logging.getLogger(__name__)
+
+# 右下の角をつかんでいるとみなす距離
+RESIZE_HANDLE_TOLERANCE_PX = 12
+MIN_FREE_RECT_SIZE = 0.05
+
+
+class LayoutEditTool:
+    """状態はこのツールが持ち、PlotterApp の同じ名前は窓口(gui/tools/__init__.py)。"""
+
+    # PlotterApp から同じ名前で読み書きできるもの(テスト・メニュー・ほかの mixin が使う)
+    EXPOSED_NAMES = (
+        '_find_axis_at_point', '_layout_drag_state', '_layout_edit_leave_cid', '_layout_edit_motion_cid',
+        '_layout_edit_press_cid', '_layout_edit_release_cid', '_layout_selected_axis_index', '_on_add_free_subplot',
+        '_on_free_layout_position_spinbox_changed', '_on_layout_motion', '_on_layout_press', '_on_layout_release',
+        '_on_remove_free_subplot', '_on_toggle_free_layout', '_sync_canvas_axes_state_and_side_panels',
+        '_sync_free_layout_position_controls', '_toggle_layout_edit_mode', 'layout_edit_mode_enabled',
+    )
+
+    def __init__(self, app):
+        self._app = app
+        self.layout_edit_mode_enabled = False
+        self._layout_edit_press_cid = None
+        self._layout_edit_motion_cid = None
+        self._layout_edit_release_cid = None
+        self._layout_edit_leave_cid = None     # ドラッグ中に図の外へ出たとき用
+        self._layout_drag_state = None
+        # クリックで選んだ軸(ドラッグとは別)。位置と大きさの数値欄の対象
+        self._layout_selected_axis_index = None
+
+    def _on_toggle_free_layout(self, checked):
+        self._app.project.layout_mode = 'free' if checked else 'grid'
+
+        self._app.subplot_rows_spinbox.setEnabled(not checked)
+        self._app.subplot_cols_spinbox.setEnabled(not checked)
+        self._app.add_free_subplot_button.setEnabled(checked)
+        self._app.remove_free_subplot_button.setEnabled(checked)
+        self._app.layout_edit_action.setEnabled(checked)
+        # 自由配置には「同じ行・列」が無いので軸の共有は使えない(値は残し、グリッドに戻せばそのまま効く)
+        self._app.share_x_checkbox.setEnabled(not checked)
+        self._app.share_y_checkbox.setEnabled(not checked)
+
+        if not checked and self._app.layout_edit_action.isChecked():
+            self._app.layout_edit_action.setChecked(False)
+            self._toggle_layout_edit_mode(False)
+
+        if not checked:
+            self._layout_selected_axis_index = None
+            self._app.free_layout_position_group.setVisible(False)
+
+        if checked:
+            # 矩形をまだ持たないサブプロットには既定の矩形を割り当てる
+            for i, settings in enumerate(self._app.project.all_plot_settings):
+                if not axis_setting(settings, 'free_rect'):
+                    settings['free_rect'] = self._app.canvas._default_free_rect(i)
+
+        self._app._update_plot()
+
+    def _on_add_free_subplot(self):
+        """空の Axes を1つ足すだけなので、全体を描き直さず canvas.add_free_axis() で足りる。"""
+        default_settings = self._app._gather_settings_from_ui()
+        new_settings = default_settings.copy()
+        # 注釈などは引き継がない。浅いコピーなので、差し替えないと全部で共有してしまう
+        new_settings['annotations'] = []
+        new_settings['legend_order'] = []
+        index = len(self._app.project.all_plot_settings)
+        new_settings['free_rect'] = self._app.canvas._default_free_rect(index)
+        self._app.project.all_plot_settings.append(new_settings)
+
+        self._app._update_subplot_combos()
+        self._app.canvas.add_free_axis(
+            self._app.project.datasets, new_settings, panel_labels_enabled=self._app.project.panel_labels_enabled,
+        )
+        self._sync_canvas_axes_state_and_side_panels()
+
+    def _on_remove_free_subplot(self):
+        """末尾のサブプロットを消す(最低1つは残す)。ほかの番号は変わらないので、全体を描き直さずに済む。"""
+        if len(self._app.project.all_plot_settings) <= 1:
+            return
+        self._app.project.all_plot_settings.pop()
+        new_total = len(self._app.project.all_plot_settings)
+        if self._app.project.active_axis_index >= new_total:
+            self._app.project.active_axis_index = new_total - 1
+
+        # 消えたサブプロットの系列はどこにも描かれなくなるので、新しい末尾に移す
+        for dataset in self._app.project.datasets:
+            if dataset.subplot_target >= new_total:
+                dataset.subplot_target = new_total - 1
+
+        self._app._update_subplot_combos()
+        self._app._apply_settings_to_ui_controls(self._app.project.all_plot_settings[self._app.project.active_axis_index])
+
+        self._app.canvas.remove_last_free_axis(self._app.project.datasets)
+        new_last_index = new_total - 1
+        self._app.canvas.update_single_axis(
+            new_last_index, self._app.project.datasets, self._app.project.all_plot_settings[new_last_index],
+            rows=0, cols=0, panel_labels_enabled=self._app.project.panel_labels_enabled,
+        )
+        self._sync_canvas_axes_state_and_side_panels()
+
+        if (self._layout_selected_axis_index is not None and
+                self._layout_selected_axis_index >= len(self._app.project.all_plot_settings)):
+            self._layout_selected_axis_index = None
+        self._sync_free_layout_position_controls()
+
+    def _sync_canvas_axes_state_and_side_panels(self):
+        """Axes を足したり消したりした後、_update_plot() がしている UI の同期を、全体を描き直さずに行う。"""
+        is_secondary_visible = any(sa is not None for sa in self._app.canvas.all_secondary_axes)
+        self._app.tick_direction_y2_label.setVisible(is_secondary_visible)
+        self._app.major_tick_direction_y2_combo.setVisible(is_secondary_visible)
+        self._app.minor_tick_direction_y2_combo.setVisible(is_secondary_visible)
+        self._app.y2_label_text_label.setVisible(is_secondary_visible)
+        self._app.y2_label_text_edit.setVisible(is_secondary_visible)
+
+        self._app.all_axes = self._app.canvas.all_axes
+        self._app.all_secondary_axes = self._app.canvas.all_secondary_axes
+
+        if hasattr(self._app, 'export_preview_panel'):
+            self._app.export_preview_panel.refresh_preview()
+        self._app._reapply_editor_row_highlight()
+        self._app._refresh_minimap()
+
+    def _toggle_layout_edit_mode(self, checked):
+        self.layout_edit_mode_enabled = checked
+
+        if checked:
+            self._app._deactivate_other_mouse_modes('layout_edit')
+
+            self._layout_edit_press_cid = self._app.canvas.mpl_connect('button_press_event', self._on_layout_press)
+            self._layout_edit_motion_cid = self._app.canvas.mpl_connect('motion_notify_event', self._on_layout_motion)
+            self._layout_edit_release_cid = self._app.canvas.mpl_connect('button_release_event', self._on_layout_release)
+            # 図の外でボタンを離すと button_release_event が届かず、ボタンを押していないのに動かし続けるので、
+            # 図から出た時点で離したことにする
+            self._layout_edit_leave_cid = self._app.canvas.mpl_connect('figure_leave_event', self._on_layout_release)
+            self._app.statusBar().showMessage(
+                "レイアウト編集モード: プロット内部をドラッグで移動、右下端をドラッグでリサイズします", 5000
+            )
+        else:
+            for cid_attr in ('_layout_edit_press_cid', '_layout_edit_motion_cid', '_layout_edit_release_cid',
+                              '_layout_edit_leave_cid'):
+                cid = getattr(self, cid_attr, None)
+                if cid is not None:
+                    self._app.canvas.mpl_disconnect(cid)
+                    setattr(self, cid_attr, None)
+            self._layout_drag_state = None
+            self._layout_selected_axis_index = None
+            self._app.free_layout_position_group.setVisible(False)
+
+    def _find_axis_at_point(self, x_px, y_px):
+        """Figure 内のピクセル座標(原点は左下)にある軸の番号。無ければ None。"""
+        for index, ax in enumerate(self._app.canvas.all_axes):
+            bbox = ax.bbox
+            if bbox.x0 <= x_px <= bbox.x1 and bbox.y0 <= y_px <= bbox.y1:
+                return index
+        return None
+
+    def _on_layout_press(self, event):
+        if not self.layout_edit_mode_enabled or event.x is None or event.y is None:
+            return
+        axis_index = self._find_axis_at_point(event.x, event.y)
+        if axis_index is None:
+            self._layout_selected_axis_index = None
+            self._sync_free_layout_position_controls()
+            return
+
+        # ドラッグしなくても選択にする(数値の欄に今の矩形を出す)
+        self._layout_selected_axis_index = axis_index
+        self._sync_free_layout_position_controls()
+
+        ax = self._app.canvas.all_axes[axis_index]
+        bbox = ax.bbox
+        near_corner = (
+            abs(event.x - bbox.x1) <= RESIZE_HANDLE_TOLERANCE_PX and
+            abs(event.y - bbox.y0) <= RESIZE_HANDLE_TOLERANCE_PX
+        )
+        pos = ax.get_position()
+        self._layout_drag_state = {
+            'axis_index': axis_index,
+            'mode': 'resize' if near_corner else 'move',
+            'start_mouse': (event.x, event.y),
+            'start_rect': (pos.x0, pos.y0, pos.width, pos.height),
+        }
+
+    def _on_layout_motion(self, event):
+        if not self.layout_edit_mode_enabled or self._layout_drag_state is None:
+            return
+        if event.x is None or event.y is None:
+            return
+
+        state = self._layout_drag_state
+        fig_width_px, fig_height_px = self._app.canvas.fig.get_size_inches() * self._app.canvas.fig.dpi
+        if fig_width_px <= 0 or fig_height_px <= 0:
+            return
+
+        delta_x = (event.x - state['start_mouse'][0]) / fig_width_px
+        delta_y = (event.y - state['start_mouse'][1]) / fig_height_px
+        left, bottom, width, height = state['start_rect']
+
+        if state['mode'] == 'move':
+            new_left = left + delta_x
+            new_bottom = bottom + delta_y
+            new_width, new_height = width, height
+        else:
+            # 右下の角のドラッグは、左上を固定して伸縮する
+            new_width = max(MIN_FREE_RECT_SIZE, width + delta_x)
+            new_height = max(MIN_FREE_RECT_SIZE, height - delta_y)
+            new_left = left
+            new_bottom = bottom + height - new_height
+
+        ax = self._app.canvas.all_axes[state['axis_index']]
+        ax.set_position([new_left, new_bottom, new_width, new_height])
+        self._app.canvas.draw_idle()
+
+        self._sync_free_layout_position_controls()
+
+    def _on_layout_release(self, event):
+        """最終的な位置を設定に保存する。"""
+        if not self.layout_edit_mode_enabled or self._layout_drag_state is None:
+            return
+
+        axis_index = self._layout_drag_state['axis_index']
+        self._layout_drag_state = None
+
+        if axis_index >= len(self._app.canvas.all_axes) or axis_index >= len(self._app.project.all_plot_settings):
+            return
+        ax = self._app.canvas.all_axes[axis_index]
+        pos = ax.get_position()
+        self._app.project.all_plot_settings[axis_index]['free_rect'] = (pos.x0, pos.y0, pos.width, pos.height)
+
+        self._sync_free_layout_position_controls()
+
+    def _sync_free_layout_position_controls(self):
+        """選んだサブプロットの矩形を数値の欄に出す。自由配置でないか、選択が無ければ欄ごと隠す。
+
+        値を入れる間は通知を止める(止めないと _on_free_layout_position_spinbox_changed が呼ばれて循環する)。
+        """
+        is_free_layout = getattr(self._app.project, 'layout_mode', 'grid') == 'free'
+        axis_index = self._layout_selected_axis_index
+
+        if (not is_free_layout or axis_index is None or
+                axis_index >= len(self._app.canvas.all_axes)):
+            self._app.free_layout_position_group.setVisible(False)
+            return
+
+        ax = self._app.canvas.all_axes[axis_index]
+        pos = ax.get_position()
+        for spinbox, value in (
+            (self._app.free_layout_x_spinbox, pos.x0),
+            (self._app.free_layout_y_spinbox, pos.y0),
+            (self._app.free_layout_width_spinbox, pos.width),
+            (self._app.free_layout_height_spinbox, pos.height),
+        ):
+            spinbox.blockSignals(True)
+            spinbox.setValue(value)
+            spinbox.blockSignals(False)
+        self._app.free_layout_position_group.setVisible(True)
+
+    def _on_free_layout_position_spinbox_changed(self, _value=None):
+        """数値の欄から、ドラッグと同じ経路(set_position → draw_idle → free_rect)で更新する。"""
+        axis_index = self._layout_selected_axis_index
+        if (axis_index is None or
+                axis_index >= len(self._app.canvas.all_axes) or
+                axis_index >= len(self._app.project.all_plot_settings)):
+            return
+
+        new_left = self._app.free_layout_x_spinbox.value()
+        new_bottom = self._app.free_layout_y_spinbox.value()
+        new_width = max(MIN_FREE_RECT_SIZE, self._app.free_layout_width_spinbox.value())
+        new_height = max(MIN_FREE_RECT_SIZE, self._app.free_layout_height_spinbox.value())
+
+        ax = self._app.canvas.all_axes[axis_index]
+        ax.set_position([new_left, new_bottom, new_width, new_height])
+        self._app.canvas.draw_idle()
+
+        self._app.project.all_plot_settings[axis_index]['free_rect'] = (
+            new_left, new_bottom, new_width, new_height
+        )

graphica/gui/tools/manager.py

@@ -0,0 +1,78 @@
+"""マウスの7つのモード(データカーソル / 注釈 / 自由配置 / 範囲選択 / ピーク配置 / スライス抽出 / 領域強調)を、
+どれか1つだけが有効になるよう MOUSE_MODES の表でまとめて管理する。
+
+モードを足すときは MOUSE_MODES に1行足す。各モードの _toggle_*_mode は、有効になるときの先頭で
+self._deactivate_other_mouse_modes('<名前>') を呼ぶだけでよい。QAction は toggled ではなく triggered に
+つないであるので、setChecked(False) だけではスロットが呼ばれない。解除ではチェックを外すのと
+_toggle_*_mode(False) の呼び出しの両方をする。
+"""
+import logging
+from collections import namedtuple
+
+logger = logging.getLogger(__name__)
+
+
+# name:          モードの識別子(_deactivate_other_mouse_modes に渡す名前)
+# flag_attr:     PlotterApp 上の「有効かどうか」を持つ属性名
+# action_attr:   ツールバーの QAction を持つ属性名
+# toggle_method: ON/OFF を切り替えるメソッド名
+MouseMode = namedtuple("MouseMode", "name flag_attr action_attr toggle_method")
+
+MOUSE_MODES = (
+    MouseMode("cursor", "cursor_mode_enabled",
+              "cursor_action", "_toggle_cursor_mode"),
+    MouseMode("annotation", "annotation_mode_enabled",
+              "annotation_action", "_toggle_annotation_mode"),
+    MouseMode("layout_edit", "layout_edit_mode_enabled",
+              "layout_edit_action", "_toggle_layout_edit_mode"),
+    MouseMode("range_select", "range_select_mode_enabled",
+              "range_select_action", "_toggle_range_select_mode"),
+    MouseMode("peak_placement", "peak_placement_mode_enabled",
+              "peak_placement_action", "_toggle_peak_placement_mode"),
+    MouseMode("slice_extraction", "slice_extraction_mode_enabled",
+              "slice_extraction_action", "_toggle_slice_extraction_mode"),
+    MouseMode("region_highlight", "region_highlight_mode_enabled",
+              "region_highlight_action", "_toggle_region_highlight_mode"),
+)
+
+MOUSE_MODES_BY_NAME = {mode.name: mode for mode in MOUSE_MODES}
+
+
+class ToolManager:
+    """モードの排他。タブの MOUSE_MODES の表を見て、フラグ・QAction・切り替えはタブの名前(ツールへの窓口)で触る。"""
+
+    def __init__(self, app):
+        self._app = app
+
+    def deactivate_others(self, active_name):
+        """active_name 以外のモードを解除する。有効でないモードには何もしないので、解除から呼び戻されて循環しない。"""
+        if active_name not in MOUSE_MODES_BY_NAME:
+            # 表への登録漏れ。動作は続けるが、開発中に気づけるようログに残す
+            logger.warning(
+                "未登録のマウスモード名です: %s (gui/tools/manager.py の "
+                "MOUSE_MODES に追加してください)", active_name
+            )
+
+        for mode in self._app.MOUSE_MODES:
+            if mode.name == active_name:
+                continue
+            if not getattr(self._app, mode.flag_attr, False):
+                continue
+
+            action = getattr(self._app, mode.action_attr, None)
+            if action is not None:
+                action.setChecked(False)
+
+            toggle = getattr(self._app, mode.toggle_method, None)
+            if callable(toggle):
+                toggle(False)
+            else:
+                # メソッドが無くても、フラグは落として「2つ有効」を残さない
+                setattr(self._app, mode.flag_attr, False)
+
+    def active_name(self):
+        """有効なモードの名前。無ければ None。"""
+        for mode in self._app.MOUSE_MODES:
+            if getattr(self._app, mode.flag_attr, False):
+                return mode.name
+        return None

graphica/gui/tools/peak_placement.py

@@ -0,0 +1,132 @@
+"""ピーク配置モード: 左クリックで多峰分離の初期値を置き、右クリックで近いものを消す。
+
+_pending_peak_guesses は [{'center', 'height', 'width'}, ...] で、MultiPeakFitDialog の initial_guesses にそのまま渡す。
+ダイアログを閉じたら(OK でもキャンセルでも)消す。仮のマーカーは redraw_all() で消えていることがあるので、
+消すときの例外は無視する。
+"""
+import logging
+
+logger = logging.getLogger(__name__)
+
+# 仮の幅は表示中の X の範囲に対する割合(ダイアログで直せるので大まかでよい)
+PEAK_PLACEMENT_DEFAULT_WIDTH_FRACTION = 0.05
+PEAK_PLACEMENT_DELETE_TOLERANCE_PX = 15
+
+
+class PeakPlacementTool:
+    """状態はこのツールが持ち、PlotterApp の同じ名前は窓口(gui/tools/__init__.py)。"""
+
+    # PlotterApp から同じ名前で読み書きできるもの(テスト・メニュー・ほかの mixin が使う)
+    EXPOSED_NAMES = (
+        '_clear_pending_peak_guesses', '_draw_pending_peak_marker', '_on_peak_placement_press',
+        '_peak_placement_press_cid', '_pending_peak_guesses', '_pending_peak_markers',
+        '_remove_nearest_pending_peak_guess', '_toggle_peak_placement_mode', 'peak_placement_mode_enabled',
+    )
+
+    def __init__(self, app):
+        self._app = app
+        self.peak_placement_mode_enabled = False
+        self._peak_placement_press_cid = None
+        self._pending_peak_guesses = []   # [{'center':, 'height':, 'width':}, ...]
+        self._pending_peak_markers = []   # [(guess, axvline, plot point), ...]
+
+    def _toggle_peak_placement_mode(self, checked):
+        self.peak_placement_mode_enabled = checked
+
+        if checked:
+            self._app._deactivate_other_mouse_modes('peak_placement')
+
+            self._peak_placement_press_cid = self._app.canvas.mpl_connect(
+                'button_press_event', self._on_peak_placement_press
+            )
+            self._app.statusBar().showMessage(
+                "ピーク配置モード: クリックで多峰分離フィットの初期値(中心・高さ)を追加、"
+                "右クリックで直近のマーカーを削除します", 5000
+            )
+        else:
+            if getattr(self, '_peak_placement_press_cid', None) is not None:
+                self._app.canvas.mpl_disconnect(self._peak_placement_press_cid)
+                self._peak_placement_press_cid = None
+
+    def _on_peak_placement_press(self, event):
+        if not getattr(self, 'peak_placement_mode_enabled', False):
+            return
+        if event.inaxes is None or event.xdata is None or event.ydata is None:
+            return
+
+        if event.button == 3:  # 右クリックは近いマーカーを消す
+            self._remove_nearest_pending_peak_guess(event)
+            return
+        if event.button != 1:
+            return
+
+        ax = event.inaxes
+        x_min, x_max = ax.get_xlim()
+        # X のずらしは平行移動なので、幅は表示座標でもデータ座標でも同じ
+        width = abs(x_max - x_min) * PEAK_PLACEMENT_DEFAULT_WIDTH_FRACTION or 1.0
+
+        # 初期値はフィットでデータ座標と突き合わされるので、クリック位置(表示座標)をデータ座標に戻す。
+        # マーカーはクリックした場所に出す
+        dataset = self._app._get_current_dataset()
+        data_x, data_y = self._app.canvas.display_to_data(
+            dataset, float(event.xdata), float(event.ydata)
+        ) if dataset is not None else (float(event.xdata), float(event.ydata))
+
+        guess = {'center': float(data_x), 'height': float(data_y), 'width': float(width)}
+        self._pending_peak_guesses.append(guess)
+        self._draw_pending_peak_marker(ax, guess, float(event.xdata), float(event.ydata))
+        self._app.statusBar().showMessage(
+            f"ピーク初期値を追加しました({len(self._pending_peak_guesses)}件、"
+            f"X={guess['center']:.4g}, Y={guess['height']:.4g})", 3000
+        )
+
+    def _draw_pending_peak_marker(self, ax, guess, display_x, display_y):
+        """マーカーは表示座標に描く(guess はデータ座標で持つ)。"""
+        line = ax.axvline(display_x, color='#E4572E', linestyle=':', linewidth=1, zorder=100)
+        point, = ax.plot(
+            [display_x], [display_y],
+            marker='x', color='#E4572E', markersize=8, zorder=101, linestyle='None',
+        )
+        self._pending_peak_markers.append((guess, line, point))
+        self._app.canvas.draw_idle()
+
+    def _remove_nearest_pending_peak_guess(self, event):
+        if not self._pending_peak_guesses:
+            return
+
+        ax = event.inaxes
+        click_px = ax.transData.transform((event.xdata, event.ydata))
+
+        best_i, best_distance = None, None
+        for i, (_guess, _line, point) in enumerate(self._pending_peak_markers):
+            # guess はデータ座標なので、描いたマーカーの位置(表示座標)で比べる
+            marker_x = point.get_xdata()[0]
+            marker_y = point.get_ydata()[0]
+            pos_px = ax.transData.transform((marker_x, marker_y))
+            distance = ((pos_px[0] - click_px[0]) ** 2 + (pos_px[1] - click_px[1]) ** 2) ** 0.5
+            if best_distance is None or distance < best_distance:
+                best_distance, best_i = distance, i
+
+        if best_i is None or best_distance > PEAK_PLACEMENT_DELETE_TOLERANCE_PX:
+            return
+
+        guess, line, point = self._pending_peak_markers.pop(best_i)
+        self._pending_peak_guesses.remove(guess)
+        for artist in (line, point):
+            try:
+                artist.remove()
+            except (ValueError, NotImplementedError):
+                pass
+        self._app.canvas.draw_idle()
+
+    def _clear_pending_peak_guesses(self):
+        """ダイアログを閉じたら(OK でもキャンセルでも)仮のマーカーごと消す。"""
+        for _guess, line, point in self._pending_peak_markers:
+            for artist in (line, point):
+                try:
+                    artist.remove()
+                except (ValueError, NotImplementedError):
+                    pass
+        self._pending_peak_markers = []
+        self._pending_peak_guesses = []
+        self._app.canvas.draw_idle()