🚑 Fix partition sizes in the custom partitioning view

Author: mmstick

src/Views/PartitioningView.vala

@@ -192,7 +192,7 @@ public class Installer.PartitioningView : AbstractInstallerView {
                 partitions.add (partition);
             }
 
-            var disk_bar = new DiskBar (model, path, size, (owned) partitions);
+            var disk_bar = new DiskBar (model, path, size, sector_size, (owned) partitions);
             label_sizer.add_widget (disk_bar.label);
             disk_list.pack_start (disk_bar);
         }
@@ -248,7 +248,7 @@ public class Installer.PartitioningView : AbstractInstallerView {
             }
         }
 
-        var disk_bar = new DiskBar (model, path, size, (owned) partitions);
+        var disk_bar = new DiskBar (model, path, size, sector_size, (owned) partitions);
         label_sizer.add_widget (disk_bar.label);
         disk_list.pack_start (disk_bar);
     }

src/Widgets/DiskBar.vala

@@ -22,6 +22,7 @@ public class Installer.DiskBar: Gtk.Grid {
     public string disk_name { get; construct; }
     public string disk_path { get; construct; }
     public uint64 size { get; construct; }
+    public uint64 sector_size { get; construct; }
     public Gee.ArrayList<PartitionBar> partitions { get; construct; }
 
     public Gtk.Box label;
@@ -34,13 +35,15 @@ public class Installer.DiskBar: Gtk.Grid {
         string model,
         string path,
         uint64 size,
+        uint64 sector_size,
         Gee.ArrayList<PartitionBar> partitions
     ) {
         Object (
             disk_name: model,
             disk_path: path,
             partitions: partitions,
-            size: size
+            size: size,
+            sector_size: sector_size
         );
     }
 
@@ -66,7 +69,7 @@ public class Installer.DiskBar: Gtk.Grid {
             used += partition.get_size ();
         }
 
-        return size - (used * 512);
+        return size - (used * sector_size);
     }
 
     private void generate_legend () {
@@ -79,7 +82,7 @@ public class Installer.DiskBar: Gtk.Grid {
         legend.add (legend_container);
 
         foreach (PartitionBar p in partitions) {
-            add_legend (p.path, p.get_size () * 512, Distinst.strfilesys (p.filesystem), p.vg, p.menu);
+            add_legend (p.path, p.get_size () * sector_size, Distinst.strfilesys (p.filesystem), p.vg, p.menu);
         }
 
         if (size / 100 < unused) {
@@ -165,7 +168,7 @@ public class Installer.DiskBar: Gtk.Grid {
 
     private void update_sector_lengths (Gee.ArrayList<PartitionBar> partitions, Gtk.Allocation alloc) {
         var alloc_width = alloc.width;
-        var disk_sectors = this.size / 512;
+        var disk_sectors = this.size / sector_size;
 
         int[] lengths = {};
         for (int x = 0; x < partitions.size; x++) {