Skip to content

Commit

Permalink
Merge pull request #551 from VincentVerelst/issue550-vector-to-raster
Browse files Browse the repository at this point in the history
added vector_to_raster convenience method #550
  • Loading branch information
soxofaan authored Mar 18, 2024
2 parents eddfa77 + 87baca0 commit 66fff76
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 0 deletions.
14 changes: 14 additions & 0 deletions openeo/rest/vectorcube.py
Original file line number Diff line number Diff line change
Expand Up @@ -558,3 +558,17 @@ def apply_dimension(
}
)
return self.process(process_id="apply_dimension", arguments=arguments)

def vector_to_raster(self, target_data_cube):
"""
Converts this vector cube into a :py:class:`~openeo.rest.datacube.DataCube`.
The bounding polygon of homogenous areas of pixels is constructed.
.. warning:: experimental process: not generally supported, API subject to change.
:return: a :py:class:`~openeo.rest.datacube.DataCube`
"""
from openeo.rest.datacube import DataCube

pg_node = PGNode(process_id="vector_to_raster", arguments={"data": self, "target_data_cube": target_data_cube})
return DataCube(pg_node, connection=self._connection, metadata=self.metadata)
19 changes: 19 additions & 0 deletions tests/rest/datacube/test_vectorcube.py
Original file line number Diff line number Diff line change
Expand Up @@ -644,3 +644,22 @@ def test_vectorcube_execute_batch_validation(
else:
assert dummy_backend.validation_requests == []
assert caplog.messages == []


def test_vector_to_raster(s2cube, vector_cube):
raster_cube = vector_cube.vector_to_raster(s2cube)
assert raster_cube.flat_graph() == {
"loadgeojson1": {
"process_id": "load_geojson",
"arguments": {"data": {"type": "Point", "coordinates": [1, 2]}, "properties": []},
},
"loadcollection1": {
"process_id": "load_collection",
"arguments": {"id": "S2", "spatial_extent": None, "temporal_extent": None},
},
"vectortoraster1": {
"process_id": "vector_to_raster",
"arguments": {"data": {"from_node": "loadgeojson1"}, "target_data_cube": {"from_node": "loadcollection1"}},
"result": True,
},
}

0 comments on commit 66fff76

Please sign in to comment.