feat(vortex-geo): Arrow import/export for the native Point type#8374
feat(vortex-geo): Arrow import/export for the native Point type#8374HarukiMoriarty wants to merge 4 commits into
Conversation
Merging this PR will not alter performance
Warning Please fix the performance issues or acknowledge them on CodSpeed. Performance Changes
Tip Investigate this regression by commenting Comparing 
Footnotes
|
Register ArrowImportVTable/ArrowExportVTable for the Point extension type, mapping geoarrow.point fields (separated coordinates) to and from vortex.geo.point with the CRS carried through. Interleaved coordinates are rejected at field import. The CRS metadata conversion is shared with the WKB vtables, and the coordinate storage dtype now has a canonical constructor inverse to coordinate_dimension. Signed-off-by: Nemo Yu <zyu379@wisc.edu>
HarukiMoriarty
force-pushed
the
nemo/geo-point-arrow
branch
from
638d514 to
2654b3a
Compare
| let cases = [ | ||
| (Dimension::Xy, ["x", "y"].as_slice()), | ||
| (Dimension::Xyz, ["x", "y", "z"].as_slice()), | ||
| (Dimension::Xym, ["x", "y", "m"].as_slice()), | ||
| (Dimension::Xyzm, ["x", "y", "z", "m"].as_slice()), | ||
| ]; |
There was a problem hiding this comment.
nit: we have `rstest::cases` for this. just search for that in the repo and you'll see several examples. you might need to import rstest into this crate to use it
There was a problem hiding this comment.
Done, also applied to other places.
| if point_meta.coord_type() != CoordType::Separated { | ||
| vortex_bail!( | ||
| "geoarrow.point with interleaved coordinates is not supported; \ | ||
| re-encode with separated (struct) coordinates" | ||
| ); | ||
| } |
There was a problem hiding this comment.
there is a vortex_ensure!() macro which captures the if-not-vortex_bail flow
Parameterize the table-driven tests with rstest cases, collapse the three binary-family WKB export tests into one, and replace the if-then-vortex_bail flows with vortex_ensure!. Signed-off-by: Nemo Yu <zyu379@wisc.edu>
2 participants
Summary
This PR adds support for import/export to Arrow for the
Pointextension type, as thegeoarrow.pointArrow extension with separated (struct) coordinates.Stacked on #8372.
Testing
Unit tests are added to exercise both code paths, plus a Vortex → Arrow → Vortex round-trip.