Correct logic for adding blanks sections when numPerPage > 1

mn-pollinators · Jan 3, 2021 · 6e0739b · 6e0739b
1 parent 15d5b43
commit 6e0739b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/app/pages/marker-generator/marker-generator.component.ts b/src/app/pages/marker-generator/marker-generator.component.ts
@@ -31,9 +31,9 @@ interface Page {
    * The marker number.
    */
   value: string;
-  
+
   type: 'Flower' | 'Nest';
-  
+
   /**
    * The marker image, as SVG source code.
    */
@@ -153,9 +153,10 @@ export class MarkerGeneratorComponent implements OnInit {
 
     const backgroundSVGPages = chunk(pages.map(p => p.backgroundSVG), numPerPage);
 
-    if (numPerPage > 1 && pages.length % 2) {
-      columnPages.push([...columnPages.pop(), []]);
-      backgroundSVGPages.push([...backgroundSVGPages.pop(), blankSVG]);
+    if (numPerPage > 1 && pages.length % numPerPage) {
+      const extraColumns = numPerPage - (pages.length % numPerPage);
+      columnPages.push([...columnPages.pop(), ...new Array(extraColumns).fill([])]);
+      backgroundSVGPages.push([...backgroundSVGPages.pop(), ...new Array(extraColumns).fill(blankSVG)]);
     }
 
     const content: Content[] = columnPages.map((c, i, arr) => ({