distancerasters 0.3.2__tar.gz → 0.3.3__tar.gz

{distancerasters-0.3.2/src/distancerasters.egg-info → distancerasters-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: distancerasters
-Version: 0.3.2
+Version: 0.3.3
 Summary: Generate distance raster using arbitrary sets of spatial features
 Home-page: https://github.com/sgoodm/python-distance-rasters
 Author: Seth Goodman
@@ -11,6 +11,13 @@ Keywords: raster,distance,spatial
 Classifier: Programming Language :: Python :: 3
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
+Requires-Dist: numpy
+Requires-Dist: rasterio
+Requires-Dist: fiona
+Requires-Dist: affine
+Requires-Dist: scipy>=1.6.0
+Requires-Dist: rasterstats
+Requires-Dist: tqdm
 
 # distance-rasters
 
@@ -135,7 +142,7 @@ You can run tests and coverage checks locally, or you can fork the repository an
 
 To run tests and coverage checks locally, you can use the following commands:
 ```sh
-pip install pytest coverage shapely geopandas
+pip install pytest coverage
 coverage run -m pytest ./
 coverage html
 ```

{distancerasters-0.3.2 → distancerasters-0.3.3}/README.md

@@ -121,7 +121,7 @@ You can run tests and coverage checks locally, or you can fork the repository an
 
 To run tests and coverage checks locally, you can use the following commands:
 ```sh
-pip install pytest coverage shapely geopandas
+pip install pytest coverage
 coverage run -m pytest ./
 coverage html
 ```

{distancerasters-0.3.2 → distancerasters-0.3.3}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = distancerasters
-version = 0.3.2
+version = 0.3.3
 description = Generate distance raster using arbitrary sets of spatial features
 long_description = file:README.md
 long_description_content_type = text/markdown
@@ -25,6 +25,7 @@ install_requires =
 	affine
 	scipy>=1.6.0
 	rasterstats
+	tqdm
 
 [options.packages.find]
 where = src

{distancerasters-0.3.2 → distancerasters-0.3.3}/src/distancerasters/main.py

@@ -1,13 +1,18 @@
 import time
+from itertools import product
+
 import numpy as np
 from affine import Affine
 from scipy.spatial import KDTree
-from .utils import export_raster, convert_index_to_coords, calc_haversine_distance
+from tqdm import tqdm
+
+from .utils import (calc_haversine_distance, convert_index_to_coords,
+                    export_raster)
 
 
 class DistanceRaster(object):
 
-    def __init__(self, raster_array, affine=None, conditional=None, output_path=None):
+    def __init__(self, raster_array, affine=None, conditional=None, output_path=None, progress_bar=False):
         """build distance array from raster array
 
         Args
@@ -46,6 +51,7 @@ class DistanceRaster(object):
         self.raster_array = raster_array
         self.pixel_size = pixel_size
         self.affine = affine
+        self.progress_bar = progress_bar
 
         self.tree = None
         self.dist_array = None
@@ -77,34 +83,33 @@ class DistanceRaster(object):
 
         t_start = time.time()
 
-        for r in range(nrows):
-
-            for c in range(ncols):
+        row_col_iterable = product(range(nrows), range(ncols))
 
-                cur_index = (r, c)
-                min_dist, min_index = self.tree.query([cur_index])
-                min_dist = min_dist[0]
-                min_index = self.tree.data[min_index[0]]
+        for r, c in tqdm(row_col_iterable, total=nrows*ncols, disable=(not self.progress_bar)):
+            cur_index = (r, c)
+            min_dist, min_index = self.tree.query([cur_index])
+            min_dist = min_dist[0]
+            min_index = self.tree.data[min_index[0]]
 
-                if self.affine is not None:
-                    if cur_index[1] == min_index[1]:
-                        # columns are same meaning nearest is either vertical or self.
-                        # no correction needed, just convert to km
-                        dd_min_dist = min_dist * self.pixel_size
-                        km_min_dist = dd_min_dist * 111.321
+            if self.affine is not None:
+                if cur_index[1] == min_index[1]:
+                    # columns are same meaning nearest is either vertical or self.
+                    # no correction needed, just convert to km
+                    dd_min_dist = min_dist * self.pixel_size
+                    km_min_dist = dd_min_dist * 111.321
 
-                    else:
-                        km_min_dist = calc_haversine_distance(
-                            convert_index_to_coords(cur_index, self.affine),
-                            convert_index_to_coords(min_index, self.affine),
-                        )
+                else:
+                    km_min_dist = calc_haversine_distance(
+                        convert_index_to_coords(cur_index, self.affine),
+                        convert_index_to_coords(min_index, self.affine),
+                    )
 
-                    val = km_min_dist * 1000
+                val = km_min_dist * 1000
 
-                else:
-                    val = min_dist
+            else:
+                val = min_dist
 
-                self.dist_array[r][c] = val
+            self.dist_array[r][c] = val
 
         print("Distance calc run time: {0} seconds".format(round(time.time() - t_start, 4)))
 

{distancerasters-0.3.2 → distancerasters-0.3.3/src/distancerasters.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: distancerasters
-Version: 0.3.2
+Version: 0.3.3
 Summary: Generate distance raster using arbitrary sets of spatial features
 Home-page: https://github.com/sgoodm/python-distance-rasters
 Author: Seth Goodman
@@ -11,6 +11,13 @@ Keywords: raster,distance,spatial
 Classifier: Programming Language :: Python :: 3
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
+Requires-Dist: numpy
+Requires-Dist: rasterio
+Requires-Dist: fiona
+Requires-Dist: affine
+Requires-Dist: scipy>=1.6.0
+Requires-Dist: rasterstats
+Requires-Dist: tqdm
 
 # distance-rasters
 
@@ -135,7 +142,7 @@ You can run tests and coverage checks locally, or you can fork the repository an
 
 To run tests and coverage checks locally, you can use the following commands:
 ```sh
-pip install pytest coverage shapely geopandas
+pip install pytest coverage
 coverage run -m pytest ./
 coverage html
 ```

{distancerasters-0.3.2 → distancerasters-0.3.3}/src/distancerasters.egg-info/requires.txt

@@ -4,3 +4,4 @@ fiona
 affine
 scipy>=1.6.0
 rasterstats
+tqdm