Removed OSM underground floors from nstories count for better results

NHERI-SimCenter · Mar 31, 2024 · 179374f · 179374f
1 parent 3b37512
commit 179374f
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/brails/workflow/FootprintHandler.py b/brails/workflow/FootprintHandler.py
@@ -37,7 +37,7 @@
 # Barbaros Cetiner
 #
 # Last updated:
-# 03-29-2024  
+# 03-30-2024  
 
 import math
 import json
@@ -54,6 +54,11 @@
 import concurrent.futures
 from requests.adapters import HTTPAdapter, Retry
 import unicodedata
+import warnings
+
+# Set a custom warning message format:
+warnings.formatwarning = lambda message, category, filename, lineno, line=None: \
+                         f"{category.__name__}: {message}\n"
 
 class FootprintHandler:
     def __init__(self): 
@@ -287,8 +292,7 @@ def height2float(inpstr,lengthUnit):
                        'height':'buildingheight',           
                        }
 
-            levelkeys = {'building:levels','roof:levels',
-                         'building:levels:underground'}
+            levelkeys = {'building:levels','roof:levels'} # Excluding 'building:levels:underground'
             otherattrkeys = set(attrmap.keys())
             datakeys = levelkeys.union(otherattrkeys)
 
@@ -1021,7 +1025,8 @@ def fp_source_selector(self):
             elif self.fpSource=='usastr':
                 footprints, attributes = get_usastruct_footprints(self.queryarea,lengthUnit)
             else:
-                print('Unimplemented footprint source. Setting footprint source to OSM')
+                warnings.warn('Unimplemented footprint source. Setting footprint source to OSM',
+                              UserWarning)
                 footprints, attributes = get_osm_footprints(self.queryarea,lengthUnit)
             return footprints, attributes