From fadd69263a881f05ce3f7972196fb8c06d2f43dd Mon Sep 17 00:00:00 2001
From: Samuel Bowman <sam@sambowman.tech>
Date: Wed, 29 Jun 2022 19:21:21 -0400
Subject: [PATCH] PartitionEditor: Add size column

---
 Userland/Applications/PartitionEditor/PartitionModel.cpp | 5 +++++
 Userland/Applications/PartitionEditor/PartitionModel.h   | 1 +
 2 files changed, 6 insertions(+)

diff --git a/Userland/Applications/PartitionEditor/PartitionModel.cpp b/Userland/Applications/PartitionEditor/PartitionModel.cpp
index 6d5d7169d1..a14bf02e91 100644
--- a/Userland/Applications/PartitionEditor/PartitionModel.cpp
+++ b/Userland/Applications/PartitionEditor/PartitionModel.cpp
@@ -4,6 +4,7 @@
  * SPDX-License-Identifier: BSD-2-Clause
  */
 
+#include <AK/NumberFormat.h>
 #include <Applications/PartitionEditor/PartitionModel.h>
 #include <LibPartition/EBRPartitionTable.h>
 #include <LibPartition/GUIDPartitionTable.h>
@@ -22,6 +23,8 @@ String PartitionModel::column_name(int column) const
         return "End Block";
     case Column::TotalBlocks:
         return "Total Blocks";
+    case Column::Size:
+        return "Size";
     default:
         VERIFY_NOT_REACHED();
     }
@@ -47,6 +50,8 @@ GUI::Variant PartitionModel::data(GUI::ModelIndex const& index, GUI::ModelRole r
             return partition.end_block();
         case Column::TotalBlocks:
             return partition.end_block() - partition.start_block() + 1;
+        case Column::Size:
+            return human_readable_size((partition.end_block() - partition.start_block() + 1) * m_partition_table->block_size());
         default:
             VERIFY_NOT_REACHED();
         }
diff --git a/Userland/Applications/PartitionEditor/PartitionModel.h b/Userland/Applications/PartitionEditor/PartitionModel.h
index 268c49c1b9..252fd5a508 100644
--- a/Userland/Applications/PartitionEditor/PartitionModel.h
+++ b/Userland/Applications/PartitionEditor/PartitionModel.h
@@ -18,6 +18,7 @@ public:
         StartBlock,
         EndBlock,
         TotalBlocks,
+        Size,
         __Count,
     };