Skip to content

Commit

Permalink
support all output formats for delayedvector https://github.com/Open-…
Browse files Browse the repository at this point in the history
  • Loading branch information
JeroenVerstraelen committed Jun 6, 2024
1 parent 6245bb3 commit 4dc1b10
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions openeo_driver/save_result.py
Original file line number Diff line number Diff line change
Expand Up @@ -902,18 +902,19 @@ def to_save_result(data: Any, format: Optional[str] = None, options: Optional[di
options = options or {}
if isinstance(data, SaveResult):
return data
elif isinstance(data, DriverDataCube):
return ImageCollectionResult(data, format=format, options=options)
elif isinstance(data, DriverVectorCube):
return VectorCubeResult(cube=data, format=format, options=options)
elif isinstance(data, DelayedVector):
if format is None or format.lower() == "json":
# TODO #114 EP-3981 add vector cube support: keep features from feature collection
geojsons = [mapping(geometry) for geometry in data.geometries_wgs84]
return JSONResult(geojsons, format=format, options=options)
if format.lower() == "geojson":
return JSONResult(data.geojson, format="geojson", options=options)
raise ValueError(f"Unsupported format {format} for DelayedVector")
else:
data = data.to_driver_vector_cube()
elif isinstance(data, DriverDataCube):
return ImageCollectionResult(data, format=format, options=options)
elif isinstance(data, DriverVectorCube):
return VectorCubeResult(cube=data, format=format, options=options)
elif isinstance(data, DriverMlModel):
return MlModelResult(ml_model = data)
elif isinstance(data, np.ndarray):
Expand Down

0 comments on commit 4dc1b10

Please sign in to comment.