feat: add geotiff support valhalla isochrones (#150)

Author: nilsnolde

CHANGELOG.md

@@ -12,8 +12,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 - Added a `km` and a `mi` property to `Direction` class to get the distance in kilometers and miles ([#139](https://github.com/gis-ops/routingpy/issues/139)).
   Distance stored in `Direction.distance` are always in meters.
-
 - Added a `pred_edge_id` property to `Edge` class that is used in `Expansions`.
+- `geotiff` support for Valhalla's isochrones ([#150](https://github.com/mthh/routingpy/pull/150))
 
 ### Fixed
 
@@ -23,6 +23,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 - Added missing support for `kwargs` to Valhalla's `matrix` creation function ([#135](https://github.com/mthh/routingpy/pull/135))
 - Fixed Valhalla's `expansion` method to follow the latest Valhalla API changes ([#142](https://github.com/mthh/routingpy/pull/142))
 - Updating params with `kwargs` [#145](https://github.com/mthh/routingpy/pull/145)
+- Relative import so submoduling works ([#150](https://github.com/mthh/routingpy/pull/150))
 
 ### Removed
 

routingpy/routers/valhalla.py

@@ -16,7 +16,7 @@
 #
 
 from operator import itemgetter
-from typing import List, Optional, Sequence, Union  # noqa: F401
+from typing import List, Optional, Sequence, Union
 
 from .. import utils
 from ..client_base import DEFAULT
@@ -25,6 +25,7 @@
 from ..expansion import Edge, Expansions
 from ..isochrone import Isochrone, Isochrones
 from ..matrix import Matrix
+from ..raster import Raster
 from ..valhalla_attributes import MatchedResults
 
 
@@ -488,6 +489,62 @@ def parse_isochrone_json(response, intervals, locations, interval_type):
 
         return Isochrones(isochrones, response)
 
+    def raster(  # noqa: C901
+        self,
+        locations: List[float],
+        profile: str,
+        intervals: List[int],
+        interval_type: Optional[str] = "time",
+        colors: Optional[List[str]] = None,
+        polygons: Optional[bool] = None,
+        denoise: Optional[float] = None,
+        generalize: Optional[float] = None,
+        preference: Optional[str] = None,
+        options: Optional[dict] = None,
+        avoid_locations: Optional[List[List[float]]] = None,
+        avoid_polygons: Optional[List[List[List[float]]]] = None,
+        date_time: Optional[dict] = None,
+        show_locations: Optional[List[List[float]]] = None,
+        id: Optional[str] = None,
+        dry_run: Optional[bool] = None,
+        **kwargs
+    ):
+        """
+        For parameters see docs for isochrones.
+
+        Returns isochrones/isodistances as GeoTIFF
+        """
+        params = self.get_isochrone_params(
+            locations,
+            profile,
+            intervals,
+            interval_type,
+            colors,
+            polygons,
+            denoise,
+            generalize,
+            preference,
+            options,
+            avoid_locations,
+            avoid_polygons,
+            date_time,
+            show_locations,
+            id,
+            **kwargs
+        )
+        params["format"] = "geotiff"
+
+        return self.parse_raster_response(
+            self.client._request("/isochrone", post_params=params, dry_run=dry_run), max(intervals)
+        )
+
+    @staticmethod
+    def parse_raster_response(response, max_travel_time: int) -> Raster:
+        if response is None:  # pragma: no cover
+            return Raster()
+
+        return Raster(image=response, max_travel_time=max_travel_time)
+
     def matrix(
         self,
         locations: List[List[float]],

tests/raster_example.tiff renamed to tests/raster_otp.tiff

@@ -124,7 +124,7 @@ def test_raster(self):
             ("arriveBy", "false"),
             ("cutoff", convert.seconds_to_iso8601(query["cutoff"])),
         ]
-        with open("tests/raster_example.tiff", "rb") as raster_file:
+        with open("tests/raster_otp.tiff", "rb") as raster_file:
             image = raster_file.read()
             responses.add(
                 responses.GET,

tests/raster_valhalla.tif

@@ -27,6 +27,7 @@
 from routingpy.expansion import Expansions
 from routingpy.isochrone import Isochrone, Isochrones
 from routingpy.matrix import Matrix
+from routingpy.raster import Raster
 from routingpy.valhalla_attributes import (
     MatchedEdge,
     MatchedPoint,
@@ -151,6 +152,24 @@ def test_isodistances(self):
             self.assertIsInstance(i, Isochrone)
             self.assertEqual(i.interval_type, "distance")
 
+    @responses.activate
+    def test_raster(self):
+        query = deepcopy(ENDPOINTS_QUERIES[self.name]["isochrones"])
+        query["format"] = "geotiff"
+        with open("tests/raster_valhalla.tif", "rb") as raster_file:
+            image = raster_file.read()
+            responses.add(
+                responses.POST,
+                "https://api.mapbox.com/valhalla/v1/isochrone",
+                status=200,
+                body=image,
+                content_type="image/tiff",
+            )
+            raster = self.client.raster(**query)
+            self.assertIsInstance(raster, Raster)
+            self.assertEqual(raster.image, image)
+            self.assertEqual(raster.max_travel_time, max(query["intervals"]))
+
     # TODO: test colors having less items than range
     @responses.activate
     def test_full_matrix(self):