Generalize different force init signatures

vasturiano · Mar 16, 2021 · 3e26717 · 3e26717
1 parent 033ea14
commit 3e26717
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/limit.js b/src/limit.js
@@ -59,11 +59,9 @@ export default function() {
 
   function initialize() {}
 
-  force.initialize = function(initNodes, nDims) {
-    const numDimensions = [1,2,3].includes(nDims) ? nDims : 2;
-
+  force.initialize = function(initNodes, ...args) {
     nodes = initNodes;
-    nDim = numDimensions;
+    nDim = args.find(arg => [1, 2, 3].includes(arg)) || 2;
     initialize();
   };