Fixing some small bugs

BRATProject.py

@@ -15,9 +15,10 @@
 import os
 import arcpy
 import sys
-from SupportingFunctions import make_folder, make_layer, get_execute_error_code, write_xml_element_with_path, find_available_num_prefix
 import XMLBuilder
 reload(XMLBuilder)
+from SupportingFunctions import make_folder, make_layer, get_execute_error_code, write_xml_element_with_path, find_available_num_prefix
+
 XMLBuilder = XMLBuilder.XMLBuilder
 
 

BRAT_table.py

@@ -90,9 +90,12 @@ def main(
     arcpy.env.outputMFlag = "Disabled"
     arcpy.CheckOutExtension("Spatial")
 
+
     # --check input projections--
     validate_inputs(seg_network, road, railroad, canal, is_verbose)
 
+    arcpy.env.outputCoordinateSystem = seg_network
+
     # name and create output folder
     new_output_folder, intermediate_folder, seg_network_copy = build_output_folder(proj_path, out_name, seg_network, road,
                                                                                    should_segment_network, ownership, segment_by_ownership,
@@ -546,7 +549,10 @@ def zSeg(vertex_type, out_field):
         if is_verbose:
             arcpy.AddMessage("Calculating iGeo_Slope...")
         for row in cursor:
-            row[3] = (abs(row[0] - row[1]))/row[2]
+            if row[2] == 0:
+                row[3] = 0.0001
+            else:
+                row[3] = (abs(row[0] - row[1]))/row[2]
             if row[3] == 0.0:
                 row[3] = 0.0001
             cursor.updateRow(row)
@@ -954,6 +960,8 @@ def find_distance_from_feature(out_network, feature, valley_bottom, temp_dir, bu
     ct = int(count.getOutput(0))
     # if there are features, then set the distance from to high value (10000 m)
     if ct < 1:
+        if is_verbose:
+            arcpy.AddMessage('No Data found for ' + new_field_name)
         with arcpy.da.UpdateCursor(out_network, new_field_name) as cursor:
             for row in cursor:
                 row[0] = 10000.0
@@ -963,6 +971,11 @@ def find_distance_from_feature(out_network, feature, valley_bottom, temp_dir, bu
         # set extent to the stream network
         arcpy.env.extent = out_network
         # calculate euclidean distance from input features
+
+        sr = arcpy.Describe(feature_subset).spatialReference
+        unit = sr.linearUnitName
+        arcpy.AddMessage('Raster units are ' + unit)
+
         ed_feature = EucDistance(feature_subset, cell_size = 5) # cell size of 5 m
         # get min distance from feature in the within 30 m buffer of each network segment
         if new_field_name == 'iPC_RoadX':

Capacity_Validation.py

@@ -31,6 +31,8 @@ def main(in_network, output_name, dams=None, da_threshold=None):
     :param da_threshold: Drainage area at which stream is presumably too large for dam building
     :return:
     """
+
+
     if da_threshold == "None":
         da_threshold = None
 
@@ -41,6 +43,8 @@ def main(in_network, output_name, dams=None, da_threshold=None):
     if dams:
         dams = copy_dams_to_inputs(proj_path, dams, in_network)
 
+
+
     if output_name.endswith('.shp'):
         output_network = os.path.join(os.path.dirname(in_network), output_name)
     else:
@@ -101,7 +105,7 @@ def copy_dams_to_inputs(proj_path, dams, in_network):
     """
     if proj_path in dams:
         # The dams input is already in our project folder, so we don't need to copy it over
-        return
+        return dams
 
     inputs_folder = find_folder(proj_path, "Inputs")
     beaver_dams_folder = find_folder(inputs_folder, "*[0-9]*_BeaverDams")