refactor: plot.add simplified with array.flat (closes #68)

jakebeamish · Sep 14, 2024 · 0596735 · 0596735
1 parent 483fe0f
commit 0596735
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/source/Plot.js b/source/Plot.js
@@ -105,17 +105,8 @@ plot.generate = () => {
 plot.draw();
    */
   add(shape) {
-    if (Array.isArray(shape)) {
-      shape.forEach((item) => {
-        if (Array.isArray(item)) {
-          this.add(item);
-        } else {
-          this.addSingleShape(item);
-        }
-      });
-    } else {
-      this.addSingleShape(shape);
-    }
+    const shapes = Array.isArray(shape) ? shape.flat(Infinity) : [shape];
+    shapes.forEach((item) => this.addSingleShape(item));
   }
 
   /**