Add remaining pixel filling code

cheonjaeung · Aug 12, 2023 · 38d37f4 · 38d37f4
1 parent c0a352c
commit 38d37f4
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/grid/src/commonMain/kotlin/io/woong/compose/grid/SimpleGridCells.kt b/grid/src/commonMain/kotlin/io/woong/compose/grid/SimpleGridCells.kt
@@ -61,7 +61,10 @@ interface SimpleGridCells {
             val totalSpacing = spacing * (count - 1)
             val totalCellSize = availableSize - totalSpacing
             val cellSize = totalCellSize / count
-            return List(count) { cellSize }
+            val remainingPixels = cellSize % count
+            return List(count) { index ->
+                cellSize + if (remainingPixels > 0 && index == 0) 1 else 0
+            }
         }
 
         override fun equals(other: Any?): Boolean {
@@ -100,7 +103,10 @@ interface SimpleGridCells {
             val totalSpacing = spacing * (count - 1)
             val totalCellSize = availableSize - totalSpacing
             val cellSize = totalCellSize / count
-            return List(count) { cellSize }
+            val remainingPixels = cellSize % count
+            return List(count) { index ->
+                cellSize + if (remainingPixels > 0 && index == 0) 1 else 0
+            }
         }
 
         override fun equals(other: Any?): Boolean {