feat(Tree): add multiselect (#436)

Author: ceccopierangiolieugenio

libs/pyTermTk/TermTk/TTkWidgets/TTkModelView/treewidget.py

@@ -166,12 +166,14 @@ def itemCollapsed(self) -> pyTTkSignal:
 
     __slots__ = ( '_rootItem', '_cache',
                   '_header', '_columnsPos',
+                  '_selectionMode',
                   '_selectedId', '_selected', '_separatorSelected',
                   '_sortColumn', '_sortOrder', '_sortingEnabled',
                   '_dndMode',
                   # Signals
                   '_itemChanged', '_itemClicked', '_itemDoubleClicked', '_itemExpanded', '_itemCollapsed', '_itemActivated'
                   )
+
     @dataclass(frozen=True)
     class _Cache:
         item: TTkTreeWidgetItem
@@ -180,6 +182,9 @@ class _Cache:
         widgets: list
         firstLine: bool
 
+    _cache:List[_Cache]
+    _selected:List[TTkTreeWidgetItem]
+
     @dataclass(frozen=True)
     class _DropTreeData:
         widget: TTkAbstractScrollView
@@ -188,13 +193,16 @@ class _DropTreeData:
     def __init__(self, *,
                  header=None,
                  sortingEnabled=True,
+                 selectionMode:TTkK.SelectionMode=TTkK.SelectionMode.SingleSelection,
                  dragDropMode:TTkK.DragDropMode=TTkK.DragDropMode.NoDragDrop,
                  **kwargs) -> None:
         '''
         :param header: define the header labels of each column, defaults to []
         :type header: list[TTkString], optional
         :param sortingEnabled: enable the column sorting, defaults to False
         :type sortingEnabled: bool, optional
+        :param selectionMode: This property controls whether the user can select one or many items, defaults to :py:class:`TTkK.SelectionMode.SingleSelection`.
+        :type selectionMode: :py:class:`TTkK.SelectionMode`, optional
         :param dragDropMode: This property holds the drag and drop event the view will act upon, defaults to :py:class:`TTkK.DragDropMode.NoDragDrop`.
         :type dragDropMode: :py:class:`TTkK.DragDropMode`, optional
         '''
@@ -206,8 +214,9 @@ def __init__(self, *,
         self._itemExpanded      = pyTTkSignal(TTkTreeWidgetItem)
         self._itemCollapsed     = pyTTkSignal(TTkTreeWidgetItem)
 
+        self._selectionMode = selectionMode
         self._dndMode = dragDropMode
-        self._selected = None
+        self._selected = []
         self._selectedId = None
         self._separatorSelected = None
         self._header = header if header else []
@@ -283,10 +292,18 @@ def indexOfTopLevelItem(self, item:TTkTreeWidgetItem) -> int:
         '''indexOfTopLevelItem'''
         return self._rootItem.indexOfChild(item)
 
+    def selectionMode(self) -> TTkK.SelectionMode:
+        '''selectionMode'''
+        return self._selectionMode
+
+    def setSelectionMode(self, mode:TTkK.SelectionMode) -> None:
+        '''setSelectionMode'''
+        self._selectionMode = mode
+
     def selectedItems(self) -> list[TTkTreeWidgetItem]:
         '''selectedItems'''
         if self._selected:
-            return [self._selected]
+            return self._selected
         return None
 
     def setHeaderLabels(self, labels:TTkString) -> None:
@@ -395,11 +412,11 @@ def mouseDoubleClickEvent(self, evt:TTkMouseEvent) -> bool:
                     self.itemExpanded.emit(item)
                 else:
                     self.itemCollapsed.emit(item)
-            if self._selected:
-                self._selected.setSelected(False)
+            for _s in self._selected:
+                _s.setSelected(False)
             self._selectedId = y
-            self._selected = item
-            self._selected.setSelected(True)
+            self._selected = [item]
+            item.setSelected(True)
             col = -1
             for i, c in enumerate(self._columnsPos):
                 if x < c:
@@ -449,11 +466,16 @@ def mousePressEvent(self, evt:TTkMouseEvent) -> bool:
                 else:
                     self.itemCollapsed.emit(item)
             else:
-                if self._selected:
-                    self._selected.setSelected(False)
-                self._selectedId = y
-                self._selected = item
-                self._selected.setSelected(True)
+                if self._selectionMode in (TTkK.SelectionMode.SingleSelection,TTkK.SelectionMode.MultiSelection):
+                    if not (
+                         bool(evt.mod & TTkK.ControlModifier) and
+                         self._selectionMode == TTkK.SelectionMode.MultiSelection ):
+                        for _s in self._selected:
+                            _s.setSelected(False)
+                        self._selected.clear()
+                    self._selectedId = y
+                    self._selected.append(item)
+                    item.setSelected(True)
             col = -1
             for i, c in enumerate(self._columnsPos):
                 if x < c:
@@ -492,14 +514,16 @@ def mouseDragEvent(self, evt:TTkMouseEvent) -> bool:
         elif ( self._dndMode & TTkK.DragDropMode.AllowDrag and
                evt.key == TTkMouseEvent.LeftButton and self._selected ):
             drag = TTkDrag()
-            data = TTkTreeWidget._DropTreeData(widget=self,items=[self._selected])
-            text = self._selected.data(0)
-            if text.termWidth() > 30:
-                text = '['+text.substring(to=27)+'...]'
-            else:
-                text = '['+text+']'
-            pm = TTkCanvas(text.termWidth()+2,1)
-            pm.drawTTkString(pos=(0,0),text=text)
+            data = TTkTreeWidget._DropTreeData(widget=self,items=self._selected)
+            text = [(_n.substring(to=27)+'...') if (_n:=_s.data(0)).termWidth()>30 else _n for _s in self._selected[:4]]
+            dh = len(text) + 2
+            dw = max(_t.termWidth() for _t in text[:3])+2
+            pm = TTkCanvas(width=dw,height=dh)
+            for _y,_t in enumerate(text[:3],1):
+                pm.drawTTkString(pos=(1,_y),text=_t)
+            if len(self._selected) > 3:
+                pm.drawText(pos=(1,4),text='...')
+            pm.drawBox(pos=(0,0),size=(dw,dh))
             drag.setPixmap(pm)
             drag.setData(data)
             drag.exec()

libs/pyTermTk/TermTk/TTkWidgets/texedit.py

@@ -1191,7 +1191,7 @@ def currentColorChanged(self) -> pyTTkSignal:
 
         This signal is emitted if the current character color has changed,
         for example caused by a change of the cursor position.
-
+        
         :param color: the new color
         :type color: :py:class:`TTkColor`
         '''
@@ -1202,7 +1202,7 @@ def cursorPositionChanged(self) -> pyTTkSignal:
         .. seealso:: this method is forwarded to :py:meth:`TTkTextEditView.cursorPositionChanged`
 
         This signal is emitted whenever the position of the cursor changed.
-
+        
         :param cursor: the cursor changed.
         :type cursor: :py:class:`TTkTextCursor`
         '''
@@ -1214,7 +1214,7 @@ def undoAvailable(self) -> pyTTkSignal:
 
         This signal is emitted whenever undo operations become available (available is true)
         or unavailable (available is false).
-
+        
         :param available: the availability of undo
         :type available: bool
         '''
@@ -1226,7 +1226,7 @@ def redoAvailable(self) -> pyTTkSignal:
 
         This signal is emitted whenever redo operations become available (available is true)
         or unavailable (available is false).
-
+        
         :param available: the availability of redo
         :type available: bool
         '''