Enable changing of a whole column at the same time (#56)

Author: PaulJonasJost

src/petab_gui/C.py

@@ -56,4 +56,5 @@
 
 # String constants
 ROW = 'row'
-COLUMN = 'column'
+COLUMN = 'column'
+INDEX = 'index'

src/petab_gui/controllers/table_controllers.py

@@ -42,6 +42,7 @@ def __init__(
         super().__init__()
         self.view = view
         self.model = model
+        self.model.view = self.view.table_view
         self.proxy_model = PandasTableFilterProxy(model)
         self.logger = logger
         self.mother_controller = mother_controller

src/petab_gui/models/pandas_table_model.py

@@ -3,7 +3,7 @@
 from PySide6.QtGui import QColor
 
 from ..C import COLUMNS
-from ..utils import validate_value, create_empty_dataframe, is_invalid
+from ..utils import validate_value, create_empty_dataframe, is_invalid, get_selected
 
 
 class PandasTableModel(QAbstractTableModel):
@@ -24,6 +24,8 @@ def __init__(self, data_frame, allowed_columns, table_type, parent=None):
         if data_frame is None:
             data_frame = create_empty_dataframe(allowed_columns, table_type)
         self._data_frame = data_frame
+        # add a view here, access is needed for selectionModels
+        self.view = None
 
     def rowCount(self, parent=QModelIndex()):
         return self._data_frame.shape[0] + 1  # empty row at the end
@@ -126,11 +128,23 @@ def insertColumn(self, column_name: str):
         return True
 
     def setData(self, index, value, role=Qt.EditRole):
+        if not (index.isValid() and role == Qt.EditRole):
+            return False
+        # check whether multiple rows but only one column is selected
+        multi_row_change, selected = self.check_selection()
+        if not multi_row_change:
+            return self._set_data_single(index, value)
+        # multiple rows but only one column is selected
+        all_set = list()
+        for index in selected:
+            all_set.append(self._set_data_single(index, value))
+        return all(all_set)
+
+    def _set_data_single(self, index, value):
+        """Set the data of a single cell."""
         col_setoff = 0
         if self._has_named_index:
             col_setoff = 1
-        if not (index.isValid() and role == Qt.EditRole):
-            return False
         if index.row() == self._data_frame.shape[0]:
             # empty row at the end
             self.insertRows(index.row(), 1)
@@ -260,6 +274,16 @@ def clear_table(self):
         self._data_frame.drop(self._data_frame.index, inplace=True)
         self.endResetModel()
 
+    def check_selection(self):
+        """Check if multiple rows but only one column is selected."""
+        if self.view is None:
+            return False
+        selected = get_selected(self.view, mode="index")
+        cols = set([index.column() for index in selected])
+        rows = set([index.row() for index in selected])
+        return len(rows) > 1 and len(cols) == 1, selected
+
+
 
 class IndexedPandasTableModel(PandasTableModel):
     """Table model for tables with named index."""

src/petab_gui/utils.py

@@ -7,7 +7,7 @@
 from matplotlib.figure import Figure
 import pandas as pd
 import logging
-from .C import ROW, COLUMN
+from .C import ROW, COLUMN, INDEX
 import antimony
 import os
 import numpy as np
@@ -567,18 +567,22 @@ def get_selected(table_view: QTableView, mode: str = ROW) -> list[int]:
     """
     if not table_view or not isinstance(table_view, QTableView):
         return []
-    if mode not in [ROW, COLUMN]:
+    if mode not in [ROW, COLUMN, INDEX]:
         return []
 
     selection_model = table_view.selectionModel()
     if not selection_model:
         return []
     selected_indexes = selection_model.selectedIndexes()
+    if mode == INDEX:
+        return selected_indexes
     if mode == COLUMN:
         selected_columns = set([index.column() for index in selected_indexes])
         return selected_columns
-    selected_rows = set([index.row() for index in selected_indexes])
-    return selected_rows
+    if mode == ROW:
+        selected_rows = set([index.row() for index in selected_indexes])
+        return selected_rows
+    return None
 
 
 def process_file(filepath, logger):