Nested partitions mode support in PartitionBarsView.

src/modules/partition/gui/PartitionBarsView.cpp

@@ -52,6 +52,7 @@ static const int SELECTION_MARGIN = qMin( ( EXTENDED_PARTITION_MARGIN - 2 ) / 2,
 
 PartitionBarsView::PartitionBarsView( QWidget* parent )
     : QAbstractItemView( parent )
+    , m_nestedPartitionsMode( NoNestedPartitions )
     , m_hoveredIndex( QModelIndex() )
     , canBeSelected( []( const QModelIndex& ) { return true; } )
 {
@@ -75,6 +76,14 @@ PartitionBarsView::~PartitionBarsView()
 }
 
 
+void
+PartitionBarsView::setNestedPartitionsMode( PartitionBarsView::NestedPartitionsMode mode )
+{
+    m_nestedPartitionsMode = mode;
+    viewport()->repaint();
+}
+
+
 QSize
 PartitionBarsView::minimumSizeHint() const
 {
@@ -216,6 +225,8 @@ PartitionBarsView::drawPartitions( QPainter* painter, const QRect& rect, const Q
             // Make sure we fill the last pixel column
             width = rect.right() - x + 1;
 
+        if ( m_nestedPartitionsMode == DrawNestedPartitions )
+        {
             drawSection( painter, rect, x, width, item.index );
             if ( modl->hasChildren( item.index ) )
             {
@@ -227,6 +238,22 @@ PartitionBarsView::drawPartitions( QPainter* painter, const QRect& rect, const Q
                 );
                 drawPartitions( painter, subRect, item.index );
             }
+        }
+        else
+        {
+            if ( modl->hasChildren( item.index ) )
+            {
+                QRect subRect(
+                    x,
+                    rect.y(),
+                    width,
+                    rect.height()
+                );
+                drawPartitions( painter, subRect, item.index );
+            }
+            else
+                drawSection( painter, rect, x, width, item.index );
+        }
         x += width;
     }
 
@@ -276,12 +303,22 @@ PartitionBarsView::indexAt( const QPoint &point,
         {
             if ( modl->hasChildren( item.index ) )
             {
-                QRect subRect(
+                QRect subRect;
+                if ( m_nestedPartitionsMode == DrawNestedPartitions )
+                    subRect = QRect(
                         x + EXTENDED_PARTITION_MARGIN,
                         rect.y() + EXTENDED_PARTITION_MARGIN,
                         width - 2 * EXTENDED_PARTITION_MARGIN,
                         rect.height() - 2 * EXTENDED_PARTITION_MARGIN
                     );
+                else
+                    subRect = QRect(
+                        x,
+                        rect.y(),
+                        width,
+                        rect.height()
+                    );
+
                 if ( subRect.contains( point ) )
                 {
                     return indexAt( point, subRect, item.index );
@@ -339,12 +376,22 @@ PartitionBarsView::visualRect( const QModelIndex& index,
         if ( modl->hasChildren( item.index ) &&
              index.parent() == item.index )
         {
-            QRect subRect(
+            QRect subRect;
+            if ( m_nestedPartitionsMode == DrawNestedPartitions )
+                subRect = QRect(
                     x + EXTENDED_PARTITION_MARGIN,
                     rect.y() + EXTENDED_PARTITION_MARGIN,
                     width - 2 * EXTENDED_PARTITION_MARGIN,
                     rect.height() - 2 * EXTENDED_PARTITION_MARGIN
                 );
+            else
+                subRect = QRect(
+                    x,
+                    rect.y(),
+                    width,
+                    rect.height()
+                );
+
             QRect candidateVisualRect = visualRect( index, subRect, item.index );
             if ( !candidateVisualRect.isNull() )
                 return candidateVisualRect;

src/modules/partition/gui/PartitionBarsView.h

@@ -35,9 +35,17 @@ class PartitionBarsView : public QAbstractItemView
 {
     Q_OBJECT
 public:
+    enum NestedPartitionsMode
+    {
+        NoNestedPartitions = 0,
+        DrawNestedPartitions
+    };
+
     explicit PartitionBarsView( QWidget* parent = nullptr );
     virtual ~PartitionBarsView();
 
+    void setNestedPartitionsMode( NestedPartitionsMode mode );
+
     QSize minimumSizeHint() const override;
 
     QSize sizeHint() const override;
@@ -75,6 +83,8 @@ private:
     QModelIndex indexAt( const QPoint& point, const QRect& rect, const QModelIndex& parent ) const;
     QRect visualRect( const QModelIndex& index, const QRect& rect, const QModelIndex& parent ) const;
 
+    NestedPartitionsMode m_nestedPartitionsMode;
+
     SelectionFilter canBeSelected;
 
     struct Item