graphica-plot 2.0.0__py3-none-any.whl

graphica/gui/datasets/colors.py

@@ -0,0 +1,218 @@
+"""データセットの色: 色の変更、自動配色、名前付きの色の適用、配色パレットの管理。"""
+import json
+import logging
+import matplotlib as mpl
+from PySide6.QtWidgets import (QDialog)
+
+from graphica.core.i18n import tr
+from graphica.gui import notify
+from graphica.core.color_palettes import BUILTIN_PALETTES, default_color_cycle
+from graphica.core.named_colors import POPUP_LIMIT, load_named_colors
+from graphica.gui import app_settings
+from graphica.gui.dialogs import (ColorPaletteDialog)
+
+logger = logging.getLogger(__name__)
+
+
+# カラーマップからの自動配色で選ばせる候補(順序のある系列に向く、知覚的に均一なもの中心)。
+RECOMMENDED_COLORMAPS = ['viridis', 'plasma', 'cividis', 'coolwarm', 'turbo', 'rainbow']
+
+
+class ColorController:
+    """データセットの色と配色パレット。"""
+
+    def __init__(self, host):
+        self._host = host
+
+    def on_color_changed(self, new_color):
+        """色欄で選んだ色を選択中のデータセットすべてに当てる(Undo 1回分)。"""
+        selected_datasets = self._host.selected_datasets()
+        if not selected_datasets:
+            return
+
+        is_batch = len(selected_datasets) > 1
+        if is_batch:
+            self._host.undo_stack.beginMacro(f"データセットの色を一括変更 ({len(selected_datasets)}件)")
+        for dataset in selected_datasets:
+            self._host.push_property_change(
+                dataset,
+                {'color': dataset.color},
+                {'color': new_color},
+                description="データセットの色変更"
+            )
+        if is_batch:
+            self._host.undo_stack.endMacro()
+
+    def on_gradient_color2_changed(self, new_color):
+        """グラデーションの終端色を選択中のデータセットすべてに当てる(Undo 1回分)。"""
+        selected_datasets = self._host.selected_datasets()
+        if not selected_datasets:
+            return
+
+        is_batch = len(selected_datasets) > 1
+        if is_batch:
+            self._host.undo_stack.beginMacro(f"グラデーション終端色を一括変更 ({len(selected_datasets)}件)")
+        for dataset in selected_datasets:
+            self._host.push_property_change(
+                dataset,
+                {'gradient_color2': dataset.gradient_color2},
+                {'gradient_color2': new_color},
+                description="グラデーション終端色の変更"
+            )
+        if is_batch:
+            self._host.undo_stack.endMacro()
+
+    def auto_assign_colors(self):
+        """選択中のデータセットに、今のパレットの色を順に割り当てる。"""
+        selected_datasets = self._host.selected_datasets()
+        if not selected_datasets:
+            return
+
+        color_cycle = self._host.active_color_cycle()
+
+        is_batch = len(selected_datasets) > 1
+        if is_batch:
+            self._host.undo_stack.beginMacro(f"配色の自動割り当て ({len(selected_datasets)}件)")
+        for i, dataset in enumerate(selected_datasets):
+            new_color = color_cycle[i % len(color_cycle)]
+            self._host.push_property_change(
+                dataset,
+                {'color': dataset.color},
+                {'color': new_color},
+                description="配色の自動割り当て"
+            )
+        if is_batch:
+            self._host.undo_stack.endMacro()
+
+    def populate_named_color_menu(self):
+        """
+        「登録した色を適用」の中身を作り直す(登録はいつでも変わるので開くたびに)。
+        登録が無いときは、空のメニューではなく無効な案内項目を1つ出す。
+        """
+        menu = getattr(self._host.parent_widget, '_named_color_apply_menu', None)
+        if menu is None:
+            return
+        menu.clear()
+        entries = load_named_colors(self._host.settings)
+        if not entries:
+            empty_action = menu.addAction(tr("(登録がありません)"))
+            empty_action.setEnabled(False)
+            return
+        # 色欄のポップアップと同じく、並べるのは先頭 POPUP_LIMIT 件まで。
+        for entry in entries[:POPUP_LIMIT]:
+            action = menu.addAction(
+                _named_color_menu_icon(entry["color"]),
+                f'{entry["name"]}	{entry["color"]}')
+            action.triggered.connect(
+                lambda checked=False, c=entry["color"], n=entry["name"]:
+                    self.apply_named_color_to_selection(c, n))
+        if len(entries) > POPUP_LIMIT:
+            more_action = menu.addAction(f"すべての登録色... ({len(entries)}件)")
+            more_action.triggered.connect(self.apply_named_color_from_list)
+
+    def apply_named_color_from_list(self):
+        """登録が POPUP_LIMIT 件を超えたときに、検索欄付きの一覧から選んで適用する。"""
+        from graphica.gui.dialogs import NamedColorPickerDialog
+        dialog = NamedColorPickerDialog(self._host.settings, self._host.parent_widget, title="登録色を適用")
+        if dialog.exec() != dialog.DialogCode.Accepted:
+            return
+        entry = dialog.selected_entry()
+        if entry:
+            self.apply_named_color_to_selection(entry["color"], entry["name"])
+
+    def apply_named_color_to_selection(self, color, name):
+        """登録した色を選択中のデータセットに当てる(Undo 1回分)。"""
+        selected_datasets = self._host.selected_datasets()
+        if not selected_datasets:
+            self._host.show_status("データセットを選択してください。", 4000)
+            return
+
+        # 変わらないものを先に除く。空のマクロでも Qt は Undo を1回分積んでしまう。
+        targets = [ds for ds in selected_datasets if ds.color != color]
+        if not targets:
+            self._host.show_status(f"選択中のデータセットは既に「{name}」の色です。", 4000)
+            return
+
+        is_batch = len(targets) > 1
+        if is_batch:
+            self._host.undo_stack.beginMacro(f"「{name}」の色を適用 ({len(targets)}件)")
+        for dataset in targets:
+            self._host.push_property_change(
+                dataset,
+                {'color': dataset.color},
+                {'color': color},
+                description=f"「{name}」の色を適用",
+            )
+        if is_batch:
+            self._host.undo_stack.endMacro()
+
+    def auto_assign_colors_from_colormap(self):
+        """選択中のデータセットに、カラーマップから等間隔に取った色を割り当てる(順序のある系列向け)。"""
+        selected_datasets = self._host.selected_datasets()
+        if not selected_datasets:
+            return
+
+        cmap_name, ok = notify.get_item(
+            self._host.parent_widget, "カラーマップから自動配色", "使用するカラーマップを選択してください:",
+            RECOMMENDED_COLORMAPS, 0, False
+        )
+        if not ok:
+            return
+
+        cmap = mpl.colormaps[cmap_name]
+        n = len(selected_datasets)
+        # 1件ならカラーマップの中央の色
+        positions = [0.5] if n == 1 else [i / (n - 1) for i in range(n)]
+        new_colors = [mpl.colors.to_hex(cmap(p)) for p in positions]
+
+        is_batch = n > 1
+        if is_batch:
+            self._host.undo_stack.beginMacro(f"カラーマップからの配色 ({n}件)")
+        for dataset, new_color in zip(selected_datasets, new_colors):
+            self._host.push_property_change(
+                dataset, {'color': dataset.color}, {'color': new_color},
+                description="カラーマップからの配色"
+            )
+        if is_batch:
+            self._host.undo_stack.endMacro()
+
+    def load_palettes(self):
+        raw = app_settings.CUSTOM_COLOR_PALETTES.read(self._host.settings)
+        if not raw:
+            return {}
+        try:
+            return json.loads(raw)
+        except (TypeError, ValueError):
+            logger.warning("カスタム配色パレットの読み込みに失敗しました。空として扱います。")
+            return {}
+
+    def save_palettes(self, palettes: dict):
+        app_settings.CUSTOM_COLOR_PALETTES.write(self._host.settings, json.dumps(palettes))
+
+    def active_color_cycle(self):
+        """今のパレットの色。未設定や空なら matplotlib の既定の色の並び。"""
+        active_name = app_settings.ACTIVE_COLOR_PALETTE.read(
+            self._host.settings, default=ColorPaletteDialog.DEFAULT_PALETTE_NAME)
+        if active_name in BUILTIN_PALETTES:
+            return BUILTIN_PALETTES[active_name]
+        palettes = self.load_palettes()
+        if active_name != ColorPaletteDialog.DEFAULT_PALETTE_NAME and palettes.get(active_name):
+            return palettes[active_name]
+        return default_color_cycle()
+
+    def manage_palettes(self):
+        palettes = self.load_palettes()
+        active_name = app_settings.ACTIVE_COLOR_PALETTE.read(
+            self._host.settings, default=ColorPaletteDialog.DEFAULT_PALETTE_NAME)
+
+        dialog = ColorPaletteDialog(palettes, active_name, self._host.parent_widget)
+        if dialog.exec() == QDialog.DialogCode.Accepted:
+            new_palettes, new_active_name = dialog.get_result()
+            self.save_palettes(new_palettes)
+            app_settings.ACTIVE_COLOR_PALETTE.write(self._host.settings, new_active_name)
+
+
+def _named_color_menu_icon(color_name, size=16):
+    """「登録した色を適用」メニューの色見本(色欄のポップアップと同じ描き方)。"""
+    from graphica.gui.color_picker_widget import _color_icon
+    return _color_icon(color_name, size=size)

graphica/gui/datasets/fitting.py

@@ -0,0 +1,70 @@
+"""曲線フィット(単発・一括・多峰分離)と、フィット結果の再表示・注釈への焼き込みのメニューの入口。
+
+本体は gui/datasets/operations/fitting.py。計算は裏で動き、終わったら結果を足す。
+"""
+from graphica.gui.datasets.operations.fitting import (BATCH_FIT_RUNNER, FIT_RESULT_WINDOW, FIT_RUNNER,
+                                                      FITTING_OPERATIONS, MULTI_PEAK_FIT_RUNNER, RUNNER_NAMES,
+                                                      burn_operation, stop_runner)
+from graphica.gui.datasets.operations.runner import OperationState, run_operation
+
+
+class FittingController:
+
+    def __init__(self, host):
+        self._host = host
+        self._state = OperationState()
+
+    @property
+    def result_dialog(self):
+        """結果は非モーダルで出し、グラフと見比べられるようにする。"""
+        return self._state.result_windows.get(FIT_RESULT_WINDOW)
+
+    @property
+    def fit_runner(self):
+        return self._state.runners.get(FIT_RUNNER)
+
+    @fit_runner.setter
+    def fit_runner(self, runner):
+        self._state.runners[FIT_RUNNER] = runner
+
+    @property
+    def batch_fit_runner(self):
+        return self._state.runners.get(BATCH_FIT_RUNNER)
+
+    @batch_fit_runner.setter
+    def batch_fit_runner(self, runner):
+        self._state.runners[BATCH_FIT_RUNNER] = runner
+
+    @property
+    def multi_peak_fit_runner(self):
+        return self._state.runners.get(MULTI_PEAK_FIT_RUNNER)
+
+    @multi_peak_fit_runner.setter
+    def multi_peak_fit_runner(self, runner):
+        self._state.runners[MULTI_PEAK_FIT_RUNNER] = runner
+
+    def shutdown(self):
+        """タブを閉じる前に、実行中の計算を待って片付ける(curve_fit 自体は中断できない)。"""
+        for name in RUNNER_NAMES:
+            runner = self._state.runners.get(name)
+            if runner is not None:
+                stop_runner(runner)
+                self._state.runners[name] = None
+
+    def _run(self, operation):
+        run_operation(operation, self._host, self._state)
+
+    def fit_current_dataset(self):
+        self._run(FITTING_OPERATIONS["fit_current_dataset"])
+
+    def batch_fit_selected(self):
+        self._run(FITTING_OPERATIONS["batch_fit_selected"])
+
+    def multi_peak_fit_current_dataset(self):
+        self._run(FITTING_OPERATIONS["multi_peak_fit_current_dataset"])
+
+    def show_fit_result(self):
+        self._run(FITTING_OPERATIONS["show_fit_result"])
+
+    def burn_fit_result_annotation(self, dataset, fit_result):
+        self._run(burn_operation(dataset, fit_result))

graphica/gui/datasets/host.py

@@ -0,0 +1,162 @@
+"""機能ごとのクラスが本体(PlotterApp の1タブ)に頼むことの窓口。"""
+from contextlib import contextmanager
+
+from graphica.core.axis_settings import axis_setting
+from graphica.core.commands import SetAnnotationsCommand
+
+
+class DatasetHost:
+    """機能ごとのクラスが PlotterApp の内部に直接触らないよう、使う操作だけをここに集める。"""
+
+    def __init__(self, app):
+        self._app = app
+
+    @property
+    def parent_widget(self):
+        return self._app
+
+    @property
+    def undo_stack(self):
+        return self._app.undo_stack
+
+    @contextmanager
+    def undo_macro(self, text, enabled=True):
+        """中の操作を Undo 1回分にまとめる(enabled=False なら何もしない)。"""
+        if enabled:
+            self._app.undo_stack.beginMacro(text)
+        try:
+            yield
+        finally:
+            if enabled:
+                self._app.undo_stack.endMacro()
+
+    def current_dataset(self):
+        return self._app.current_dataset()
+
+    def add_derived_dataset(self, dataset, source, description):
+        """source と同じフォルダにデータセットを足して描き直す(Undo できる。今の選択は変えない)。"""
+        source_item = self._app.dataset_tree_item(source)
+        folder = source_item.parent() if source_item else None
+
+        def add():
+            self._app.dataset_order.append(dataset, folder)
+            self._app.redraw()
+
+        self._app.push_dataset_additions(add, [dataset], description)
+
+    def add_annotation(self, axis_index, annotation, description):
+        self._app.add_annotation(axis_index, annotation, description)
+
+    def annotations(self, axis_index):
+        """その軸の注釈(リストはコピー)。"""
+        return list(axis_setting(self._app.project.all_plot_settings[axis_index], 'annotations'))
+
+    def add_annotations_to_active_axis(self, annotations, description):
+        """今の軸に注釈をまとめて足す(Undo 1回分)。"""
+        project = self._app.project
+        index = project.active_axis_index
+        if index >= len(project.all_plot_settings):
+            return
+        old_annotations = self.annotations(index)
+        self._app.undo_stack.push(SetAnnotationsCommand(
+            project, index, old_annotations, old_annotations + list(annotations),
+            self._app.redraw_appearance, description=description))
+
+    def axis_y_span(self, axis_index):
+        """その軸の今の Y の表示幅。軸が無ければ None。"""
+        axes = self._app.canvas.all_axes
+        if not 0 <= axis_index < len(axes):
+            return None
+        y_lo, y_hi = axes[axis_index].get_ylim()
+        return y_hi - y_lo
+
+    def selected_datasets(self):
+        return self._app.selected_datasets()
+
+    def target_folder_for_new_dataset(self):
+        """ツリーで選ばれているのがフォルダならそのフォルダ(新しいデータセットの置き場所)、それ以外は None。"""
+        return self._app.target_folder_for_new_dataset()
+
+    def add_datasets_to_folder(self, datasets, folder, description):
+        """まとめて足してから1回だけ描き直す(1件ずつ描き直すと件数分のフル再描画になる)。Undo 1回で消える。"""
+        def add():
+            for dataset in datasets:
+                self._app.dataset_order.append(dataset, folder)
+            self._app.redraw()
+
+        self._app.push_dataset_additions(add, list(datasets), description)
+
+    def axis_count(self):
+        return len(self._app.project.all_plot_settings)
+
+    def show_status(self, text, msecs=3000):
+        self._app.statusBar().showMessage(text, msecs)
+
+    def set_fit_button_enabled(self, enabled):
+        self._app.fit_curve_button.setEnabled(enabled)
+
+    def set_multi_peak_fit_button_enabled(self, enabled):
+        self._app.multi_peak_fit_button.setEnabled(enabled)
+
+    def pending_peak_guesses(self):
+        """グラフ上のクリックで置いた多峰フィットの初期値(のコピー)。"""
+        return list(self._app.pending_peak_guesses())
+
+    def finish_peak_placement(self):
+        """置いた初期値を消し、ピーク配置モードを抜ける。"""
+        self._app.clear_pending_peak_guesses()
+        if getattr(self._app, 'peak_placement_mode_enabled', False):
+            self._app.peak_placement_action.setChecked(False)
+            self._app.toggle_peak_placement_mode(False)
+
+    def datasets(self):
+        """タブの全データセット(リストはコピー)。"""
+        return list(self._app.project.datasets)
+
+    def add_dataset(self, dataset, parent_folder=None, select=True):
+        """ツリーと描画に足す(Undo の対象にはしない)。"""
+        return self._app.add_dataset(dataset, parent_folder, select=select)
+
+    def add_dataset_with_undo(self, dataset, parent_folder=None, description="データセットの追加"):
+        self._app.add_dataset_with_undo(dataset, parent_folder=parent_folder, description=description)
+
+    def redraw(self):
+        self._app.redraw()
+
+    def refresh_ui_state(self):
+        """選択中のデータセットに合わせてパネルとメニューの状態を更新する。"""
+        self._app.property_panel.update_ui_state()
+
+    def active_color_cycle(self):
+        return self._app.colors.active_color_cycle()
+
+    @property
+    def project(self):
+        """文書全体。読むためだけに使う(書き換えは host の操作を通す)。"""
+        return self._app.project
+
+    @property
+    def settings(self):
+        """アプリ全体の設定(QSettings)。"""
+        return self._app.settings
+
+    def push_property_change(self, dataset, old_values, new_values, description, skip_if_unchanged=True):
+        """データセットの属性の変更を Undo できる形で積む(skip_if_unchanged なら変化が無いときは何もしない)。"""
+        self._app.push_dataset_property_command(dataset, old_values, new_values, description,
+                                                skip_if_unchanged=skip_if_unchanged)
+
+    def remove_datasets(self, datasets, description=None):
+        """確認なしで Undo できる形で消す。"""
+        items = [item for dataset in datasets
+                 if (item := self._app.dataset_tree_item(dataset)) is not None]
+        self._app.remove_dataset_items_with_undo(items, description=description)
+
+    def sibling_tabs(self):
+        """ほかのタブの (タブの見出し, そのタブの DatasetHost)。タブの窓に入っていなければ空。"""
+        from graphica.gui.main_app_window import MainAppWindow  # main_app_window がこのパッケージを読み込むので遅らせる
+        top = self._app.window()
+        if not isinstance(top, MainAppWindow):
+            return []
+        tabs = top.tab_widget
+        return [(tabs.tabText(i), tabs.widget(i).dataset_host)
+                for i in range(tabs.count()) if tabs.widget(i) is not self._app]

graphica/gui/datasets/operations/__init__.py

@@ -0,0 +1 @@
+"""データセットの処理の定義(操作の本体)と、その流れを担う実行役。"""