graphica-plot 2.0.0__py3-none-any.whl

graphica/gui/panels/dataset_tree.py

@@ -0,0 +1,424 @@
+"""
+データセットの一覧(ツリー)の管理: 追加・削除・複製、フォルダ、表示/非表示、並べ替え。
+データセットに対する個々の操作は graphica.gui.datasets の機能ごとのクラスにある。
+
+ツリーの項目の UserRole には、データセットなら Dataset そのもの、フォルダなら None が入る。
+選択を読むときは行番号ではなく _get_current_dataset() / _get_selected_datasets() を使う
+(フォルダが入れ子になっていてもデータセットだけを扱えるように)。
+"""
+import copy
+import logging
+import numpy as np
+import pandas as pd
+from PySide6.QtCore import Qt, QTimer
+from PySide6.QtWidgets import (QDialog, QMenu)
+
+from graphica.gui import notify
+from graphica.core.commands import (SetDatasetPropertiesCommand, ReorderDatasetsCommand)
+from graphica.core.dataset import Dataset
+from graphica.gui.datasets.order import DatasetOrder
+from graphica.gui.workers import BUILTIN_DATA_FILE_EXTENSIONS
+from graphica.core.plugin_api import get_registered_importer_extensions
+from graphica.gui.data_editor import DataEditorDialog
+from graphica.gui.datasets.actions_menu import populate_dataset_actions_menu
+from graphica.gui.dialogs import (NewDatasetDialog)
+from graphica.gui.dataset_style_icon import (
+    make_dataset_style_icon, make_dataset_visibility_icon, apply_dataset_visibility_text_style,
+    DATASET_TREE_VISIBILITY_COLUMN,
+)
+
+logger = logging.getLogger(__name__)
+
+
+class DatasetTreePanel:
+    """状態はタブが持ち、この部品はメソッドと信号の配線だけを持つ。PlotterApp の同じ名前は窓口(gui/panels/__init__.py)。"""
+
+    # PlotterApp から同じ名前で使えるもの(テスト・メニュー・ほかの部品が使う)
+    EXPOSED_NAMES = (
+        '_connect_dataset_signals', '_find_dataset_row', '_on_add_dataset', '_on_create_new_dataset',
+        '_on_dataset_order_applied', '_on_dataset_rows_moved', '_on_dataset_search_changed',
+        '_on_dataset_tree_context_menu', '_on_dataset_tree_item_clicked', '_on_duplicate_dataset',
+        '_on_hide_all_datasets', '_on_hide_all_in_folder', '_on_new_folder', '_on_remove_dataset',
+        '_on_rename_dataset_folder', '_on_show_all_datasets', '_on_show_all_in_folder', '_on_show_data_editor',
+        '_push_dataset_property_command', '_refresh_after_dataset_property_change', '_set_all_datasets_visibility',
+        '_set_folder_datasets_visibility', '_sync_dataset_list_and_replot', '_top_level_selected_items',
+    )
+
+    def __init__(self, app):
+        self._app = app
+
+    def _on_add_dataset(self):
+        # プラグインが登録した拡張子も選べるようにする
+        plugin_extensions = get_registered_importer_extensions()
+        plugin_pattern = ''.join(f' *{ext}' for ext in plugin_extensions)
+        builtin_pattern = ' '.join(f'*{ext}' for ext in BUILTIN_DATA_FILE_EXTENSIONS)
+        file_path, _ = notify.get_open_file_name(
+            self._app, "データファイルを選択", "",
+            f"Data Files ({builtin_pattern}{plugin_pattern});;All Files (*)"
+        )
+        if file_path:
+            self._app.load_data(file_path)
+
+    def _on_create_new_dataset(self):
+        """名前・列・行数を指定した空のデータセットを作り、そのままデータエディタで手入力させる。"""
+        dialog = NewDatasetDialog(self._app)
+        if dialog.exec() != QDialog.DialogCode.Accepted:
+            return
+
+        name = dialog.get_dataset_name()
+        column_names = dialog.get_column_names()
+        row_count = dialog.get_row_count()
+
+        df = pd.DataFrame({col: [np.nan] * row_count for col in column_names})
+        x_col_name = column_names[0]
+        y_col_name = column_names[1] if len(column_names) > 1 else column_names[0]
+
+        new_dataset = Dataset(name=name, df=df, x_col_name=x_col_name, y_col_name=y_col_name)
+        self._app._add_dataset_with_undo(new_dataset, self._app._get_target_folder_for_new_dataset(),
+                                    description="新規データセットの作成")
+
+        self._on_show_data_editor()
+
+    def _on_dataset_search_changed(self, text):
+        """名前に検索文字列を含むデータセットだけを出す。フォルダは中に1つでも当たりがあれば出す。"""
+        self._app.dataset_order.filter_by_name(text)
+
+    def _on_new_folder(self):
+        """選択中がフォルダならその中に、そうでなければ一番上に作る。"""
+        name, ok = notify.get_text(self._app, "新しいフォルダ", "フォルダ名:", text="新しいフォルダ")
+        if not ok or not name:
+            return
+
+        current_item = self._app.ui.dataset_list_widget.currentItem()
+        parent_item = None
+        if current_item is not None and current_item.data(0, Qt.ItemDataRole.UserRole) is None:
+            parent_item = current_item
+
+        self._app._add_dataset_folder_item(name, parent_item)
+
+    def _on_rename_dataset_folder(self):
+        """
+        フォルダの構造は Undo の対象外なので、名前の変更も Undo しない。
+        保存時はツリーの表示文字列を読むので、setText だけで保存にも反映される。
+        """
+        current_item = self._app.ui.dataset_list_widget.currentItem()
+        if current_item is None or current_item.data(0, Qt.ItemDataRole.UserRole) is not None:
+            return
+        old_name = current_item.text(0)
+        new_name, ok = notify.get_text(self._app, "フォルダ名を変更", "新しいフォルダ名:", text=old_name)
+        if not ok or not new_name:
+            return
+        current_item.setText(0, new_name)
+
+    def _set_folder_datasets_visibility(self, folder_item, visible):
+        """フォルダの中(サブフォルダも)のデータセットをまとめて表示/非表示にする(Undo 1回分)。"""
+        dataset_items = self._app._flatten_dataset_tree(folder_item)
+        datasets = [item.data(0, Qt.ItemDataRole.UserRole) for item in dataset_items]
+        if not datasets:
+            return
+        is_batch = len(datasets) > 1
+        if is_batch:
+            self._app.undo_stack.beginMacro(f"フォルダ内の表示/非表示切替 ({len(datasets)}件)")
+        for ds in datasets:
+            self._push_dataset_property_command(
+                ds, {'visible': ds.visible}, {'visible': visible},
+                description="データセットの表示/非表示切替"
+            )
+        if is_batch:
+            self._app.undo_stack.endMacro()
+
+    def _set_all_datasets_visibility(self, visible):
+        """フォルダに関係なく、全データセットをまとめて表示/非表示にする(Undo 1回分)。"""
+        datasets = list(self._app.project.datasets)
+        if not datasets:
+            return
+        is_batch = len(datasets) > 1
+        if is_batch:
+            self._app.undo_stack.beginMacro(f"全データセットの表示/非表示切替 ({len(datasets)}件)")
+        for ds in datasets:
+            self._push_dataset_property_command(
+                ds, {'visible': ds.visible}, {'visible': visible},
+                description="データセットの表示/非表示切替"
+            )
+        if is_batch:
+            self._app.undo_stack.endMacro()
+
+    def _on_show_all_datasets(self):
+        self._set_all_datasets_visibility(True)
+
+    def _on_hide_all_datasets(self):
+        self._set_all_datasets_visibility(False)
+
+    def _on_show_all_in_folder(self):
+        current_item = self._app.ui.dataset_list_widget.currentItem()
+        if current_item is None:
+            return
+        self._set_folder_datasets_visibility(current_item, True)
+
+    def _on_hide_all_in_folder(self):
+        current_item = self._app.ui.dataset_list_widget.currentItem()
+        if current_item is None:
+            return
+        self._set_folder_datasets_visibility(current_item, False)
+
+    def _on_dataset_tree_context_menu(self, pos):
+        menu = QMenu(self._app)
+        populate_dataset_actions_menu(self._app, menu)
+        menu.exec(self._app.ui.dataset_list_widget.viewport().mapToGlobal(pos))
+
+    def _top_level_selected_items(self, items):
+        """ほかの選択項目の子孫でないものだけを返す(フォルダと中身が両方選ばれていても二重に消さないため)。"""
+        return DatasetOrder.top_level_items(items)
+
+    def _on_remove_dataset(self):
+        """選択中のデータセットとフォルダ(中身ごと)を Undo できる形で消す。"""
+        selected_items = self._app.ui.dataset_list_widget.selectedItems()
+        if not selected_items:
+            return
+
+        self._app._remove_dataset_items_with_undo(self._top_level_selected_items(selected_items))
+
+    def _find_dataset_row(self, dataset):
+        return self._app.dataset_order.find_row(dataset)
+
+    def _on_dataset_rows_moved(self, source_parent, source_start, source_end, dest_parent, dest_row):
+        """
+        ツリーのドラッグ&ドロップ後、project.datasets をツリーの並び(=描画の重なり順)に合わせる。
+        同じフォルダ内の並べ替えだけ Undo でき、フォルダをまたぐ移動はフォルダ構造と同じく Undo の対象外。
+        """
+        reordered = self._app.dataset_order.tree_order()
+
+        if len(reordered) != len(self._app.project.datasets):
+            logger.warning(
+                "データセットの並べ替え同期に失敗しました (表示数 %d, 実際 %d)。",
+                len(reordered), len(self._app.project.datasets)
+            )
+            return
+
+        old_order = list(self._app.project.datasets)
+        # Dataset の == は使えないので同一性で比べる
+        if [id(d) for d in reordered] == [id(d) for d in old_order]:
+            return
+
+        if source_parent == dest_parent:
+            command = ReorderDatasetsCommand(
+                self._app.project, old_order, reordered,
+                on_applied=self._on_dataset_order_applied,
+                description="データセットの並べ替え"
+            )
+            self._app.undo_stack.push(command)
+        else:
+            self._app.project.datasets = reordered
+            self._app._update_plot()
+
+    def _on_dataset_order_applied(self):
+        """ドロップ処理の最中にツリーを触ると Qt の処理とぶつかるので、同期と再描画は次のイベントループに回す。"""
+        QTimer.singleShot(0, self._sync_dataset_list_and_replot)
+
+    def _sync_dataset_list_and_replot(self):
+        self._app._sync_dataset_list_widget_order()
+        self._app._update_plot()
+
+    def _refresh_after_dataset_property_change(self, dataset, changed_keys=(), old_values=None, new_values=None):
+        """
+        Undo/Redo でデータセットの属性が変わった後に、ツリーの表示・プロパティ欄・描画を合わせる。
+        描き直すのは関係する軸だけ。on_applied は Undo と Redo のどちらの後かを教えないので、
+        subplot_target の新旧どちらの軸も描き直す。
+        """
+        item = self._app._get_dataset_tree_item(dataset)
+        if item is not None:
+            if item.text(0) != dataset.name:
+                item.setText(0, dataset.name)
+            item.setIcon(0, make_dataset_style_icon(dataset))
+            item.setIcon(DATASET_TREE_VISIBILITY_COLUMN, make_dataset_visibility_icon(dataset))
+            apply_dataset_visibility_text_style(item, dataset)
+            if item is self._app.ui.dataset_list_widget.currentItem():
+                self._app.property_panel.update_ui_state()
+
+        axis_index = dataset.subplot_target
+        if axis_index >= len(self._app.canvas.all_axes) or axis_index >= len(self._app.project.all_plot_settings):
+            self._app._update_plot()
+            return
+
+        axis_indices_to_refresh = {axis_index}
+        if 'subplot_target' in changed_keys:
+            if old_values and 'subplot_target' in old_values:
+                axis_indices_to_refresh.add(old_values['subplot_target'])
+            if new_values and 'subplot_target' in new_values:
+                axis_indices_to_refresh.add(new_values['subplot_target'])
+        axis_indices_to_refresh = {
+            i for i in axis_indices_to_refresh
+            if i < len(self._app.canvas.all_axes) and i < len(self._app.project.all_plot_settings)
+        }
+
+        layout_mode = getattr(self._app.project, 'layout_mode', 'grid')
+        if layout_mode == 'free':
+            rows, cols = 0, 0
+        else:
+            rows = self._app.subplot_rows_spinbox.value()
+            cols = self._app.subplot_cols_spinbox.value()
+
+        for idx in axis_indices_to_refresh:
+            self._app.canvas.update_single_axis(
+                idx, self._app.project.datasets, self._app.project.all_plot_settings[idx],
+                rows=rows, cols=cols,
+                share_x_axis=getattr(self._app.project, 'share_x_axis', False),
+                share_y_axis=getattr(self._app.project, 'share_y_axis', False),
+                panel_labels_enabled=self._app.project.panel_labels_enabled,
+            )
+        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._reapply_editor_row_highlight()
+        self._app._refresh_minimap()
+        if hasattr(self._app, 'export_preview_panel'):
+            self._app.export_preview_panel.refresh_preview()
+        self._app._notify_plugins_datasets_changed()
+
+    def _push_dataset_property_command(self, dataset, old_values: dict, new_values: dict, description: str,
+                                       skip_if_unchanged=True):
+        """
+        Dataset のプロパティ変更を Undo/Redo 可能なコマンドとして発行する共通ヘルパー。
+        old_values と new_values が同じ (実質的に変更なし) 場合は何もしない。
+        値に DataFrame を含むときは == で比べられないので skip_if_unchanged=False で呼ぶ。
+        """
+        if skip_if_unchanged and old_values == new_values:
+            return
+        command = SetDatasetPropertiesCommand(
+            dataset, old_values, new_values,
+            on_applied=lambda: self._refresh_after_dataset_property_change(
+                dataset, changed_keys=new_values.keys(), old_values=old_values, new_values=new_values
+            ),
+            description=description
+        )
+        self._app.undo_stack.push(command)
+
+    def _on_duplicate_dataset(self):
+        """選択中のデータセットを、元と同じフォルダに複製する。"""
+        selected_items = [
+            item for item in self._app.ui.dataset_list_widget.selectedItems()
+            if item.data(0, Qt.ItemDataRole.UserRole) is not None
+        ]
+        if not selected_items:
+            return
+
+        copies = []
+        for item in selected_items:
+            original_dataset = item.data(0, Qt.ItemDataRole.UserRole)
+            # DataFrame まで独立させる
+            new_dataset = copy.deepcopy(original_dataset)
+            new_dataset.name = f"{original_dataset.name} (copy)"
+            copies.append((new_dataset, item.parent()))
+
+        def add():
+            tree = self._app.ui.dataset_list_widget
+            tree.blockSignals(True)
+            new_items = [self._app.dataset_order.append(new_dataset, folder) for new_dataset, folder in copies]
+            tree.clearSelection()
+            for new_item in new_items:
+                new_item.setSelected(True)
+            tree.setCurrentItem(new_items[-1])
+            tree.blockSignals(False)
+            self._app.property_panel.update_ui_state()
+            self._app._update_plot()
+
+        self._app._push_dataset_additions(add, [new_dataset for new_dataset, _ in copies], "データセットの複製")
+
+    def _on_show_data_editor(self):
+        dataset = self._app._get_current_dataset()
+        if dataset is None:
+            return
+
+        # 同じデータセットのエディタが開いていれば、作り直さず(並べ替え・位置・選択が消えるので)前に出すだけ
+        if self._app.data_editor_dialog is not None and self._app.data_editor_dialog.dataset is dataset:
+            self._app.data_editor_dialog.show()
+            self._app.data_editor_dialog.raise_()
+            self._app.data_editor_dialog.activateWindow()
+            return
+
+        if self._app.data_editor_dialog:
+            self._app.data_editor_dialog.close()
+            # close() は隠すだけで、親(このタブ)にぶら下がったまま残り続けるので破棄する
+            self._app.data_editor_dialog.deleteLater()
+            del self._app.data_editor_dialog
+            self._app.data_editor_dialog = None
+
+        self._app.data_editor_dialog = DataEditorDialog(dataset, self._app)
+
+        self._app.data_editor_dialog.dataChanged.connect(self._app.property_panel.on_data_structure_changed)
+
+        # エディタで選んだ行をグラフ上で強調する
+        self._app.data_editor_dialog.rowsHighlighted.connect(self._app._on_editor_rows_highlighted)
+
+        # 開いたままグラフを操作できるよう非モーダル
+        self._app.data_editor_dialog.show()
+
+    def _on_dataset_tree_item_clicked(self, item, column):
+        """
+        目アイコンの列のクリックで表示/非表示を切り替える。
+        クリックした項目が複数選択に含まれていれば、選択中のすべてをまとめて切り替える。
+        """
+        if column != DATASET_TREE_VISIBILITY_COLUMN:
+            return
+        dataset = item.data(0, Qt.ItemDataRole.UserRole)
+        if dataset is None:
+            return
+
+        new_value = not dataset.visible
+
+        # Dataset の == は使えないので、in ではなく同一性で調べる
+        selected_datasets = self._app._get_selected_datasets()
+        is_part_of_multi_selection = len(selected_datasets) > 1 and any(
+            ds is dataset for ds in selected_datasets
+        )
+        targets = selected_datasets if is_part_of_multi_selection else [dataset]
+
+        is_batch = len(targets) > 1
+        if is_batch:
+            self._app.undo_stack.beginMacro(f"表示/非表示の一括切替 ({len(targets)}件)")
+        for ds in targets:
+            self._push_dataset_property_command(
+                ds,
+                {'visible': ds.visible},
+                {'visible': new_value},
+                description="データセットの表示/非表示切替"
+            )
+        if is_batch:
+            self._app.undo_stack.endMacro()
+
+    def _connect_dataset_signals(self):
+        self._app.ui.add_dataset_button.clicked.connect(self._on_add_dataset)
+        self._app.new_dataset_button.clicked.connect(self._on_create_new_dataset)
+        self._app.ui.remove_dataset_button.clicked.connect(self._on_remove_dataset)
+        self._app.new_folder_button.clicked.connect(self._on_new_folder)
+        self._app.dataset_search_edit.textChanged.connect(self._on_dataset_search_changed)
+        self._app.ui.dataset_list_widget.currentItemChanged.connect(self._app.property_panel.on_dataset_selected)
+        self._app.ui.dataset_list_widget.customContextMenuRequested.connect(self._on_dataset_tree_context_menu)
+        self._app.ui.dataset_list_widget.itemClicked.connect(self._on_dataset_tree_item_clicked)
+        # ドラッグでの並べ替え(描画の重なり順)を project.datasets に合わせる
+        self._app.ui.dataset_list_widget.model().rowsMoved.connect(self._on_dataset_rows_moved)
+
+
+        # プロパティの欄は表(gui/dataset_bindings.py)の順につなぐ
+        self._app.property_panel.connect_signals()
+
+        self._app.fit_curve_button.clicked.connect(self._app.fitting.fit_current_dataset)
+        self._app.find_peaks_button.clicked.connect(self._app.peaks.find_peaks)
+        self._app.multi_peak_fit_button.clicked.connect(self._app.fitting.multi_peak_fit_current_dataset)
+
+        self._app.duplicate_dataset_button.clicked.connect(self._on_duplicate_dataset)
+        self._app.auto_color_button.clicked.connect(self._app.colors.auto_assign_colors)
+        self._app.manage_palette_action.triggered.connect(self._app.colors.manage_palettes)
+        self._app.colormap_assign_action.triggered.connect(self._app.colors.auto_assign_colors_from_colormap)
+        self._app.view_edit_data_button.clicked.connect(self._on_show_data_editor)
+
+        self._app.x_col_combo.currentTextChanged.connect(self._app.property_panel.on_plot_column_changed)
+        self._app.y_col_combo.currentTextChanged.connect(self._app.property_panel.on_plot_column_changed)
+        self._app.x_err_col_combo.currentTextChanged.connect(self._app.property_panel.on_error_column_changed)
+        self._app.y_err_col_combo.currentTextChanged.connect(self._app.property_panel.on_error_column_changed)
+
+        self._app.z_col_combo.currentTextChanged.connect(self._app.property_panel.on_z_column_changed)

graphica/gui/plot_type_drawers.py

@@ -0,0 +1,131 @@
+"""
+組み込みの plot_type ごとの描き方。プラグインの register_plot_type と同じく「種類名 → 描画関数」の表で引く。
+
+描画関数は (canvas, ax, ds, x, y, plot_kwargs, axis_index) を受け取り、ds.artist にする artist を返す。
+x, y はウォーターフォールのずらし・欠損値の処理・表示用の間引きを済ませた点列。
+plot_kwargs(ウォーターフォールの zorder)はグラデーションの線には渡さない。
+"""
+import logging
+
+import numpy as np
+from scipy.stats import gaussian_kde
+
+from graphica.core.dataset import COLOR_BY_COLUMN_PLOT_TYPE
+
+logger = logging.getLogger(__name__)
+
+
+def _uses_line_gradient(ds):
+    return ds.gradient_enabled and ds.gradient_target in ('line', 'both')
+
+
+def _gradient_line(canvas, ax, ds, x, y):
+    return canvas._add_gradient_line(
+        ax, x, y, ds.color, ds.gradient_color2, ds.linewidth, ds.alpha, ds.linestyle, label=ds.name)
+
+
+def _plain_line(ax, ds, x, y, plot_kwargs, **extra):
+    (artist,) = ax.plot(x, y, color=ds.color, linestyle=ds.linestyle, linewidth=ds.linewidth,
+                        alpha=ds.alpha, label=ds.name, **extra, **plot_kwargs)
+    return artist
+
+
+def _markers(ax, ds, x, y, plot_kwargs, **extra):
+    return ax.scatter(x, y, color=ds.color, marker=ds.marker, s=ds.markersize**2, alpha=ds.alpha,
+                      **extra, **plot_kwargs)
+
+
+def draw_line(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    if _uses_line_gradient(ds):
+        return _gradient_line(canvas, ax, ds, x, y)
+    return _plain_line(ax, ds, x, y, plot_kwargs)
+
+
+def draw_scatter(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    return _markers(ax, ds, x, y, plot_kwargs, label=ds.name)
+
+
+def draw_line_and_scatter(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    if _uses_line_gradient(ds):
+        # LineCollection はマーカーを描けないので、マーカーは単色で重ねる
+        artist = _gradient_line(canvas, ax, ds, x, y)
+        _markers(ax, ds, x, y, plot_kwargs)
+        return artist
+    (artist,) = ax.plot(x, y, color=ds.color, linestyle=ds.linestyle, linewidth=ds.linewidth,
+                        marker=ds.marker, markersize=ds.markersize, alpha=ds.alpha, label=ds.name, **plot_kwargs)
+    return artist
+
+
+def draw_area(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    """0 との間を塗り、輪郭が分かるよう上端に線も重ねる。"""
+    if not ds.gradient_enabled:
+        artist = ax.fill_between(x, y, 0, color=ds.color, alpha=ds.alpha * 0.4, label=ds.name, **plot_kwargs)
+        ax.plot(x, y, color=ds.color, linestyle=ds.linestyle, linewidth=ds.linewidth, alpha=ds.alpha, **plot_kwargs)
+        return artist
+    if ds.gradient_target in ('fill', 'both'):
+        artist = canvas._add_gradient_fill(ax, x, y, ds.color, ds.gradient_color2, ds.alpha * 0.4)
+    else:
+        artist = ax.fill_between(x, y, 0, color=ds.color, alpha=ds.alpha * 0.4, **plot_kwargs)
+    if ds.gradient_target in ('line', 'both'):
+        _gradient_line(canvas, ax, ds, x, y)
+    else:
+        _plain_line(ax, ds, x, y, plot_kwargs)
+    return artist
+
+
+def draw_bar(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    return ax.bar(x, y, color=ds.color, alpha=ds.alpha, label=ds.name, **plot_kwargs)
+
+
+def draw_step(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    # 区間の左端の値を右端まで保つ(サンプリング・イベントデータの慣習)
+    return _plain_line(ax, ds, x, y, plot_kwargs, drawstyle='steps-post')
+
+
+def draw_density_scatter(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    """各点の位置での2次元カーネル密度を色にする。点が少ない・全点が同じ座標だと KDE が失敗するので単色にする。"""
+    try:
+        if len(x) < 3:
+            raise ValueError("点数不足")
+        xy = np.vstack([x, y])
+        density = gaussian_kde(xy)(xy)
+        return ax.scatter(x, y, c=density, cmap=ds.colormap, marker=ds.marker, s=ds.markersize**2,
+                          alpha=ds.alpha, label=ds.name, **plot_kwargs)
+    except (np.linalg.LinAlgError, ValueError):
+        return _markers(ax, ds, x, y, plot_kwargs, label=ds.name)
+
+
+def draw_color_by_column(canvas, ax, ds, x, y, plot_kwargs, axis_index):
+    """
+    z_col_name の列の値で点を色分けする。z は visible_df 経由(ds.z_data)で取る
+    (df から取ると、行を除外したとたん点と色が1つずつずれる)。
+    """
+    z_values = ds.z_data
+    if z_values is not None and len(z_values) == len(x):
+        artist = ax.scatter(x, y, c=z_values, cmap=ds.colormap, vmin=ds.vmin, vmax=ds.vmax,
+                            marker=ds.marker, s=ds.markersize**2, alpha=ds.alpha, label=ds.name, **plot_kwargs)
+        # カラーバーは1軸に1つ。2Dマップが先に登録していればそちらを優先する
+        if axis_index not in canvas._axis_2d_mappables:
+            canvas._axis_2d_mappables[axis_index] = artist
+        return artist
+    if z_values is not None:
+        # 欠損値の方針 'drop' で点が減ったときなど。黙って単色になると理由が分からないのでログに残す
+        logger.warning(
+            "'%s' のZ列の長さ(%d)がプロット点数(%d)と一致しないため、"
+            "単色のScatterとして描画します(欠損値の方針'drop'等で"
+            "配列が短くなっている可能性があります)。",
+            ds.name, len(z_values), len(x),
+        )
+    return _markers(ax, ds, x, y, plot_kwargs, label=ds.name)
+
+
+BUILTIN_PLOT_TYPE_DRAWERS = {
+    'Line': draw_line,
+    'Scatter': draw_scatter,
+    'Line+Scatter': draw_line_and_scatter,
+    'Area': draw_area,
+    'Bar': draw_bar,
+    'Step': draw_step,
+    'Density Scatter': draw_density_scatter,
+    COLOR_BY_COLUMN_PLOT_TYPE: draw_color_by_column,
+}

graphica/gui/plugin_context.py

@@ -0,0 +1,119 @@
+"""PluginContext の本体側の実装。1つのタブ(PlotterApp)と1つのプラグインの組ごとに作る。"""
+import logging
+import weakref
+from dataclasses import fields
+
+
+from graphica.gui import notify
+from graphica.core.app_paths import get_plugin_data_dir
+from graphica.core.color_palettes import normalize_palettes
+from graphica.core.commands import SetDatasetPropertiesCommand
+from graphica.core.dataset import Dataset
+from graphica.core.named_colors import add_named_color, load_named_colors, save_named_colors
+from graphica.core.plugin_context import PluginContext
+
+logger = logging.getLogger(__name__)
+
+_DATASET_FIELDS = frozenset(f.name for f in fields(Dataset))
+
+
+class TabPluginContext(PluginContext):
+
+    def __init__(self, app, plugin_name):
+        # タブを閉じたあとも窓口をプラグインが握っていることがあるので、タブは弱参照で持つ。
+        self._app_ref = weakref.ref(app)
+        self._plugin_name = plugin_name
+        self._datasets_changed_callbacks = []
+        self._selection_changed_callbacks = []
+
+    @property
+    def _app(self):
+        app = self._app_ref()
+        if app is None:
+            raise RuntimeError("このタブはすでに閉じられています。")
+        return app
+
+    def datasets(self):
+        return list(self._app.project.datasets)
+
+    def current_dataset(self):
+        return self._app.current_dataset()
+
+    def selected_datasets(self):
+        return self._app.selected_datasets()
+
+    def add_dataset(self, dataset, description=None):
+        self._app.add_dataset_with_undo(
+            dataset, description=description or f"[{self._plugin_name}] データセットの追加"
+        )
+
+    def set_dataset_properties(self, dataset, values, description=None):
+        unknown = sorted(set(values) - _DATASET_FIELDS)
+        if unknown:
+            raise AttributeError(f"Dataset に無い属性です: {', '.join(unknown)}")
+        app = self._app
+        new_values = dict(values)
+        old_values = {key: getattr(dataset, key) for key in new_values}
+        app.undo_stack.push(SetDatasetPropertiesCommand(
+            dataset, old_values, new_values,
+            on_applied=lambda: app.refresh_after_dataset_property_change(
+                dataset, changed_keys=new_values.keys(), old_values=old_values, new_values=new_values
+            ),
+            description=description or f"[{self._plugin_name}] プロパティの変更",
+        ))
+
+    def redraw(self):
+        self._app.redraw()
+
+    def on_datasets_changed(self, callback):
+        self._datasets_changed_callbacks.append(callback)
+
+    def on_selection_changed(self, callback):
+        self._selection_changed_callbacks.append(callback)
+
+    def _notify_datasets_changed(self):
+        for callback in list(self._datasets_changed_callbacks):
+            self._call_isolated(callback)
+
+    def _notify_selection_changed(self, dataset):
+        for callback in list(self._selection_changed_callbacks):
+            self._call_isolated(callback, dataset)
+
+    def _call_isolated(self, callback, *args):
+        # 通知先のプラグインが失敗しても、描画や選択の処理を止めない。
+        try:
+            callback(*args)
+        except Exception:
+            logger.exception("[plugin:%s] 通知の処理に失敗しました", self._plugin_name)
+
+    @property
+    def parent_widget(self):
+        return self._app_ref()
+
+    def show_message(self, text, title=None):
+        notify.information(self.parent_widget, title or self._plugin_name, text)
+
+    def show_error(self, text, title=None):
+        notify.warning(self.parent_widget, title or self._plugin_name, text)
+
+    @property
+    def data_dir(self):
+        return get_plugin_data_dir(self._plugin_name)
+
+    def named_colors(self):
+        return load_named_colors(self._app.settings)
+
+    def set_named_colors(self, entries):
+        validated = []
+        for entry in entries:
+            validated = add_named_color(validated, entry.get("name"), entry.get("color"))
+        save_named_colors(self._app.settings, validated)
+
+    def color_palettes(self):
+        return {name: list(colors) for name, colors in self._app.colors.load_palettes().items()}
+
+    def set_color_palettes(self, palettes):
+        self._app.colors.save_palettes(normalize_palettes(palettes))
+
+    def active_color_cycle(self):
+        return list(self._app.colors.active_color_cycle())