Implement polygonToCellsCompact

uber · Oct 3, 2023 · e041cbe · e041cbe
1 parent e3102ce
commit e041cbe
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/h3lib/include/polyfill.h b/src/h3lib/include/polyfill.h
@@ -80,7 +80,7 @@ H3Error H3_EXPORT(polygonToCells2)(const GeoPolygon *polygon,
                                            int res, uint32_t flags,
                                            H3Index *out);
 
-H3Error H3_EXPORT(polygonToCells3)(const GeoPolygon *polygon,
+H3Error H3_EXPORT(polygonToCellsCompact)(const GeoPolygon *polygon,
                                            int res, uint32_t flags,
                                            H3Index *out);
 

diff --git a/src/h3lib/lib/polyfill.c b/src/h3lib/lib/polyfill.c
@@ -406,4 +406,17 @@ H3Error H3_EXPORT(polygonToCells2)(const GeoPolygon *polygon, int res,
         out[i++] = iter.cell;
     }
     return iter.error;
+}
+
+/**
+ * Compact implementation for polygonToCells
+ */
+H3Error H3_EXPORT(polygonToCellsCompact)(const GeoPolygon *polygon, int res,
+                                         uint32_t flags, H3Index *out) {
+    IterCellsPolygonCompact iter = iterInitPolygonCompact(polygon, res, flags);
+    int64_t i = 0;
+    for (; iter.cell; iterStepPolygonCompact(&iter)) {
+        out[i++] = iter.cell;
+    }
+    return iter.error;
 }