Fix 3d coords for non-points (#160)

developmentseed · Oct 27, 2023 · 400ed8f · 400ed8f
1 parent 712d98a
commit 400ed8f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/lonboard/geoarrow/extension_types.py b/lonboard/geoarrow/extension_types.py
@@ -308,7 +308,7 @@ def construct_geometry_array(
     elif geom_type == GeometryType.LINESTRING:
         assert len(offsets) == 1, "Expected one offsets array"
         (geom_offsets,) = offsets
-        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), 2)
+        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), len(dims))
         parr = pa.ListArray.from_arrays(pa.array(geom_offsets), _parr)
         field = pa.field(
             "geometry",
@@ -321,7 +321,7 @@ def construct_geometry_array(
     elif geom_type == GeometryType.POLYGON:
         assert len(offsets) == 2, "Expected two offsets arrays"
         ring_offsets, geom_offsets = offsets
-        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), 2)
+        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), len(dims))
         _parr1 = pa.ListArray.from_arrays(pa.array(ring_offsets), _parr)
         parr = pa.ListArray.from_arrays(pa.array(geom_offsets), _parr1)
         field = pa.field(
@@ -335,7 +335,7 @@ def construct_geometry_array(
     elif geom_type == GeometryType.MULTIPOINT:
         assert len(offsets) == 1, "Expected one offsets array"
         (geom_offsets,) = offsets
-        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), 2)
+        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), len(dims))
         parr = pa.ListArray.from_arrays(pa.array(geom_offsets), _parr)
         field = pa.field(
             "geometry",
@@ -348,7 +348,7 @@ def construct_geometry_array(
     elif geom_type == GeometryType.MULTILINESTRING:
         assert len(offsets) == 2, "Expected two offsets arrays"
         ring_offsets, geom_offsets = offsets
-        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), 2)
+        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), len(dims))
         _parr1 = pa.ListArray.from_arrays(pa.array(ring_offsets), _parr)
         parr = pa.ListArray.from_arrays(pa.array(geom_offsets), _parr1)
         field = pa.field(
@@ -362,7 +362,7 @@ def construct_geometry_array(
     elif geom_type == GeometryType.MULTIPOLYGON:
         assert len(offsets) == 3, "Expected three offsets arrays"
         ring_offsets, polygon_offsets, geom_offsets = offsets
-        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), 2)
+        _parr = pa.FixedSizeListArray.from_arrays(coords.flatten(), len(dims))
         _parr1 = pa.ListArray.from_arrays(pa.array(ring_offsets), _parr)
         _parr2 = pa.ListArray.from_arrays(pa.array(polygon_offsets), _parr1)
         parr = pa.ListArray.from_arrays(pa.array(geom_offsets), _parr2)