Add functionality to reproject .tif file (#139)

arup-group · Aug 25, 2022 · 4fcc3d3 · 4fcc3d3
1 parent 44a4035
commit 4fcc3d3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/genet/utils/elevation.py b/genet/utils/elevation.py
@@ -4,7 +4,8 @@
 
 def get_elevation_image(elevation_tif):
     xarr_file = rioxarray.open_rasterio(elevation_tif)
-
+    if str(xarr_file.rio.crs) != 'EPSG:4326':
+        xarr_file = xarr_file.rio.write_crs(4326, inplace=True)
     return xarr_file[0, :, :]
 
 

diff --git a/tests/test_data/elevation/hk_elevation_example_crs_2326.tif b/tests/test_data/elevation/hk_elevation_example_crs_2326.tif
diff --git a/tests/test_utils_elevation.py b/tests/test_utils_elevation.py
@@ -7,6 +7,7 @@
 
 elevation_test_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "test_data", "elevation"))
 tif_path = os.path.join(elevation_test_folder, 'hk_elevation_example.tif')
+tif_path_crs_not_4326 = os.path.join(elevation_test_folder, 'hk_elevation_example_crs_2326.tif')
 array_path = os.path.join(elevation_test_folder, 'elevation_image.nc')
 
 def test_output_type_get_elevation_image():
@@ -20,6 +21,12 @@ def test_get_elevation_image():
     assert output == image
 
 
+def test_get_elevation_image_with_crs_not_4326():
+    output = elevation.get_elevation_image(tif_path_crs_not_4326)
+    image = xr.open_dataset(array_path)
+    assert output == image
+
+
 def test_get_elevation_data():
     # need to figure out how to load save and load in an xarray.DataArray - only managed xarray.Dataset
     image = elevation.get_elevation_image(tif_path)