From 2000553a107d4f925017bb2399c9ae9e6e44b2f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20G=C3=A2teau?= <agateau@kde.org>
Date: Mon, 30 Jun 2014 15:03:45 +0200
Subject: [PATCH] PartitionModel: Add getter for Partition

---
 src/modules/partition/PartitionModel.cpp | 11 +++++++++++
 src/modules/partition/PartitionModel.h   |  2 ++
 2 files changed, 13 insertions(+)

diff --git a/src/modules/partition/PartitionModel.cpp b/src/modules/partition/PartitionModel.cpp
index 90be12439..3d5384643 100644
--- a/src/modules/partition/PartitionModel.cpp
+++ b/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 );
+}
diff --git a/src/modules/partition/PartitionModel.h b/src/modules/partition/PartitionModel.h
index a9bd3e403..384d7cbb1 100644
--- a/src/modules/partition/PartitionModel.h
+++ b/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;