PartitionModel: Add getter for Partition

src/modules/partition/PartitionModel.cpp

@@ -82,3 +82,14 @@ PartitionModel::fillPartitionList( PartitionNode* parent )
         fillPartitionList( partition );
     }
 }
+
+Partition*
+PartitionModel::partitionForIndex( const QModelIndex& index ) const
+{
+    int row = index.row();
+    if ( row < 0 || row >= m_partitionList.count() )
+    {
+        return nullptr;
+    }
+    return m_partitionList.at( row );
+}

src/modules/partition/PartitionModel.h

@@ -33,6 +33,8 @@ public:
     int rowCount( const QModelIndex& parent = QModelIndex() ) const Q_DECL_OVERRIDE;
     QVariant data( const QModelIndex& index, int role = Qt::DisplayRole ) const Q_DECL_OVERRIDE;
 
+    Partition* partitionForIndex( const QModelIndex& index ) const;
+
 private:
     Device* m_device;
     QList< Partition* > m_partitionList;