Add files via upload

Author: asgeirlysdahl

ArcGISProWrapperBegrensSkade_Excavation.py

@@ -42,9 +42,9 @@
 feature_name = arcpy.GetParameterAsText(3)
 coord_syst = arcpy.GetParameterAsText(4)
 
-sr = arcpy.SpatialReference()
-sr.loadFromString(coord_syst)
-output_proj = sr.PCSCode
+output_spatial_ref = arcpy.SpatialReference()
+output_spatial_ref.loadFromString(coord_syst)
+output_proj = output_spatial_ref.PCSCode
 
 corner_name = feature_name + "_CORNER"
 wall_name = feature_name + "_WALL"
@@ -124,20 +124,20 @@
     status_field = None
 
 ##############  GET INPUT PROJECTIONS ####################
-building_proj = Utils_arcpy.getProjCodeFromFC(building_polys_fl)
-excavation_proj = Utils_arcpy.getProjCodeFromFC(excavation_polys_fl)
+building_spatial_ref = arcpy.Describe(building_polys_fl).spatialReference
+excavation_spatial_ref = arcpy.Describe(excavation_polys_fl).spatialReference
 
 ###  GET EXCAVATION ANS BUILDINGS ON SAME PROJECTION #####
 excavation_polys_matched = False
-if  excavation_proj != building_proj:
+if  excavation_spatial_ref != building_spatial_ref:
     arcpy.AddMessage("Matching input projections before clip..")
     excavation_polys_matched = output_folder + os.sep + "exc_match.shp"
-    arcpy.Project_management(excavation_polys_fl, excavation_polys_matched, building_proj)
+    arcpy.Project_management(excavation_polys_fl, excavation_polys_matched, building_spatial_ref)
     excavation_polys_fl = excavation_polys_matched
 
 ################ GET EXCAVATION INFO #####################
 excavation_outline_as_json = Utils_arcpy.getConstructionAsJson(excavation_polys_fl)
-buildingsClipExtent = Utils_arcpy.getBuildingsClipExtentFromConstruction(excavation_outline_as_json, CALCULATION_RANGE, building_proj, logger)
+buildingsClipExtent = Utils_arcpy.getBuildingsClipExtentFromConstruction(excavation_outline_as_json, CALCULATION_RANGE, building_spatial_ref, logger)
 
 ################ EXTRACTING BUILDINGS ##################
 buildings_clip = output_folder + os.sep + "buildings_clip.shp"
@@ -152,7 +152,7 @@
 excavation_polys_projected = False
 buildings_clip_projected = False
 
-if building_proj != output_proj:
+if building_spatial_ref != output_spatial_ref:
 
     arcpy.AddMessage("Projecting bulidings polygon..")
     buildings_clip_projected = output_folder + os.sep + "buil_proj.shp"

ArcGISProWrapperBegrensSkade_Tunnel.py

@@ -42,9 +42,9 @@
 feature_name = arcpy.GetParameterAsText(3)
 coord_syst = arcpy.GetParameterAsText(4)
 
-sr = arcpy.SpatialReference()
-sr.loadFromString(coord_syst)
-output_proj = sr.PCSCode
+output_spatial_ref = arcpy.SpatialReference()
+output_spatial_ref.loadFromString(coord_syst)
+output_proj = output_spatial_ref.PCSCode
 
 corner_name = feature_name + "_CORNER"
 wall_name = feature_name + "_WALL"
@@ -130,20 +130,20 @@
     status_field = None
 
 ##############  GET INPUT PROJECTIONS ###########################
-building_proj = Utils_arcpy.getProjCodeFromFC(building_polys_fl)
-tunnel_proj = Utils_arcpy.getProjCodeFromFC(tunnel_poly_fl)
+building_spatial_ref = arcpy.Describe(building_polys_fl).spatialReference
+tunnel_spatial_ref = arcpy.Describe(tunnel_poly_fl).spatialReference
 
 ########  GET EXCAVATION AND BUILDINGS ON SAME PROJECTION ###########
 tunnel_poly_matched = False
-if tunnel_proj != building_proj:
+if tunnel_spatial_ref != building_spatial_ref:
     arcpy.AddMessage("Matching input projections before clip..")
     tunnel_poly_matched = output_folder + os.sep + "tun_match.shp"
-    arcpy.Project_management(tunnel_poly_fl, tunnel_poly_matched, building_proj)
+    arcpy.Project_management(tunnel_poly_fl, tunnel_poly_matched, building_spatial_ref)
     tunnel_poly_fl = tunnel_poly_matched
 
 ################ GET TUNNEL INFO #####################
 tunnel_outline_as_json = Utils_arcpy.getConstructionAsJson(tunnel_poly_fl)
-buildingsClipExtent = Utils_arcpy.getBuildingsClipExtentFromConstruction(tunnel_outline_as_json, CALCULATION_RANGE, building_proj, logger)
+buildingsClipExtent = Utils_arcpy.getBuildingsClipExtentFromConstruction(tunnel_outline_as_json, CALCULATION_RANGE, building_spatial_ref, logger)
 
 ################ EXTRACTING BUILDINGS ##################
 buildings_clip = output_folder + os.sep + "buildings_clip.shp"
@@ -158,7 +158,7 @@
 tunnel_poly_projected = False
 buildings_clip_projected = False
 
-if building_proj != output_proj:
+if building_spatial_ref != output_spatial_ref:
 
     arcpy.AddMessage("Projecting bulidings polygon..")
     buildings_clip_projected = output_folder + os.sep + "buil_proj.shp"

Utils_arcpy.py

@@ -72,10 +72,9 @@ def getConstructionAsJson(uploaded_construction_outline):
                 bounding_as_json = geom.JSON
                 return json.loads(bounding_as_json)
 
-def getBuildingsClipExtentFromConstruction(constructionJson, margin, working_proj, logger):
+def getBuildingsClipExtentFromConstruction(constructionJson, margin, spatial_ref, logger):
     logger.debug("constructionJson: {}".format(constructionJson))
     construction_area_corners = BegrensSkadeLib.get_construction_corners_from_ArcGIS_json(constructionJson, 2, logger)
-    sr = arcpy.SpatialReference(working_proj)
     minx = 99999999
     miny = 99999999
     maxx = -99999999
@@ -96,7 +95,7 @@ def getBuildingsClipExtentFromConstruction(constructionJson, margin, working_pro
     ur = arcpy.Point(maxx + margin, maxy + margin)
     lr = arcpy.Point(maxx + margin, miny - margin)
 
-    rectangle = arcpy.Polygon(arcpy.Array([ll, ul, ur, lr, ll]), sr)
+    rectangle = arcpy.Polygon(arcpy.Array([ll, ul, ur, lr, ll]), spatial_ref)
 
     logger.debug("Rectangle: {}".format(rectangle.extent.JSON))
     return rectangle