From 93cb9fd426eebd4395670c718c388c361bb25469 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20G=C3=A2teau?= <agateau@kde.org>
Date: Tue, 29 Jul 2014 13:37:12 +0200
Subject: [PATCH] New role for PartitionModel: SizeRole

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

diff --git a/src/modules/partition/PartitionModel.cpp b/src/modules/partition/PartitionModel.cpp
index 912fd136b..461f2e2c5 100644
--- a/src/modules/partition/PartitionModel.cpp
+++ b/src/modules/partition/PartitionModel.cpp
@@ -172,6 +172,8 @@ PartitionModel::data( const QModelIndex& index, int role ) const
             return colorForPartition( partition, index.row() );
         else
             return QVariant();
+    case SizeRole:
+        return ( partition->lastSector() - partition->firstSector() + 1 ) * m_device->logicalSectorSize();
     default:
         return QVariant();
     }
diff --git a/src/modules/partition/PartitionModel.h b/src/modules/partition/PartitionModel.h
index 50981ae2c..96bb3db4e 100644
--- a/src/modules/partition/PartitionModel.h
+++ b/src/modules/partition/PartitionModel.h
@@ -46,6 +46,13 @@ public:
         PartitionModel* m_model;
     };
 
+    enum
+    {
+        // The raw size, as a qlonglong. This is different from the DisplayRole of
+        // SizeColumn, which is a human-readable string.
+        SizeRole = Qt::UserRole + 1
+    };
+
     enum Column
     {
         NameColumn,