added support for coords

DanielJDufour · Jun 11, 2023 · 7a806c7 · 7a806c7
1 parent 517e02c
commit 7a806c7
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 4 deletions.
diff --git a/mpoly.js b/mpoly.js
@@ -1,3 +1,4 @@
+const getDepth = require("get-depth");
 const meta = require("@turf/meta");
 const booleanClockwise = require("@turf/boolean-clockwise").default;
 
@@ -33,13 +34,15 @@ function each(geom, callback) {
       }
     });
   } else if (Array.isArray(geom)) {
-    const depth = getDepth(geojson);
+    const depth = getDepth(geom);
     if (depth === 4) {
-      it.forEach(polygon => {
+      geom.forEach(polygon => {
         callback(polygon);
       });
     } else if (depth === 3) {
-      callback(it);
+      callback(geom);
+    } else if (depth === 2) {
+      callback([geom]);
     }
   }
 }

diff --git a/test.js b/test.js
@@ -23,10 +23,29 @@ const { Example, GeometryCollection, LineString, MultiPoint, MultiLineString, Mu
 
 test("mpoly.get (arcgis)", ({ eq }) => {
   Object.entries(arcgisjsons).forEach(([name, data]) => {
-    eq(mpoly.get(arcgisjsons[name]), mpoly.get(geojsons[name]));
+    const results = new Set();
+    results.add(JSON.stringify(mpoly.get(geojsons[name])));
+    results.add(JSON.stringify(mpoly.get(JSON.stringify(geojsons[name]))));
+    results.add(JSON.stringify(mpoly.get(arcgisjsons[name])));
+    results.add(JSON.stringify(mpoly.get(JSON.stringify(arcgisjsons[name]))));
+    eq(results.size, 1);
   });
 });
 
+test("coords", ({ eq }) => {
+  const coords = [
+    [100, 0],
+    [101, 0],
+    [101, 1],
+    [100, 1],
+    [100, 0]
+  ];
+  const expected = [[coords]];
+  eq(mpoly.get(coords), expected);
+  eq(mpoly.get([coords]), expected);
+  eq(mpoly.get([[coords]]), expected);
+});
+
 test("mpoly.get (geojson)", ({ eq }) => {
   eq(mpoly.get(Example), [
     [