def createViewshed(self):
try:
tempEnvironment0 = arcpy.env.extent
arcpy.env.extent = self.buffer
tempEnvironment1 = arcpy.env.cellSize
arcpy.env.cellSize = self.cellsize
tempEnvironment2 = arcpy.env.mask
arcpy.env.mask = self.buffer
outraster = sa.Viewshed(self.islyr, self.obsproc, 1,
"CURVED_EARTH", 0.13)
#outrastertemp = os.path.join(self.scratch, 'outvis')
#outraster.save(outrastertemp)
vshedtmp = os.path.join("in_memory", 'vshedtmp')
vsheddis = os.path.join("in_memory", 'vsheddis')
#vshed_proj = os.path.join(self.scratchgdb, 'vshedproj')
arcpy.AddMessage("temp vshed fc:" + vshedtmp)
arcpy.AddMessage("dissolved vshed fc: " + vsheddis)
arcpy.env.extent = tempEnvironment0
arcpy.env.cellSize = tempEnvironment1
arcpy.env.mask = tempEnvironment2
arcpy.RasterToPolygon_conversion(outraster, vshedtmp, "SIMPLIFY",
"VALUE")
arcpy.Dissolve_management(vshedtmp, vsheddis, "gridcode", "",
"MULTI_PART", "DISSOLVE_LINES")
## if(self.wkidproc != self.wkidout):
## arcpy.AddMessage("Projecting output vshed...")
## arcpy.AddMessage("projected vshed fc: " + vshed_proj)
## arcpy.Project_management(vsheddis, vshed_proj, self.srOut)
## vshed=vshed_proj
## else:
## vshed=vsheddis
#vistmp = os.path.join('in_memory', 'visibility')
vis = os.path.join(self.scratchgdb, 'visibility')
arcpy.AddMessage('creating output viewshed: ' + vis)
arcpy.Clip_analysis(vsheddis, self.mask, vis, "")
arcpy.AddMessage("Coppying to output...")
#arcpy.CopyFeatures_management(vistmp, vis)
fset = arcpy.FeatureSet()
fset.load(vis)
return fset
except arcpy.ExecuteError:
EH = ErrorHandling.ErrorHandling()
line, filename, err = EH.trace()
m = "Python error on " + line + " of " + __file__ + \
" : with error - " + err
arcpy.AddError(m)
"OBJECTID = " + str(row.getValue("OBJECTID")))
# Project the MBG buffer to AZED
obs_buf = os.path.join(env.scratchWorkspace,
"obs_buf_towerlos_" + scrubbedTowerName)
arcpy.Project_management(mbgBufferPrj, obs_buf, strAZED)
delete_me.append(obs_buf)
# Finally ... run Viewshed
arcpy.AddMessage(r"...Calculating Viewshed ...")
vshed = os.path.join(env.scratchWorkspace, "vshed_towerlos")
delete_me.append(vshed)
#Use the visibility tool instead of the viewshed. This will allow is to set the observation height offset for the tower
#outVshed = sa.Visibility(extract_prj,obs_prjTower,z_factor=z_factor,curvature_correction="CURVED_EARTH",refractivity_coefficient=terrestrial_refractivity_coefficient,observer_offset=maxOffset)
outVshed = sa.Viewshed(extract_prj, obs_prjTower, z_factor,
"CURVED_EARTH",
terrestrial_refractivity_coefficient)
outVshed.save(vshed)
# Raster To Polygon
arcpy.AddMessage(r"...Converting to polygons ...")
ras_poly = os.path.join(env.scratchWorkspace,
"ras_poly_towerlos_" + scrubbedTowerName)
arcpy.RasterToPolygon_conversion(vshed, ras_poly, polygon_simplify)
delete_me.append(ras_poly)
# clip output polys to buffer
if RADIUS2_to_infinity != True:
out_buf = os.path.join(env.scratchWorkspace,
"out_buf_towerlos_" + scrubbedTowerName)
arcpy.Buffer_analysis(obs_prjTower, out_buf,
def main():
'''Main RLOS'''
try:
# get/set initial environment
env.overwriteOutput = True
installInfo = arcpy.GetInstallInfo("desktop")
# get observer's vibility modifier maximums
obsMaximums = maxVizModifiers(observers)
removeSPOT = obsMaximums['REMOVE_SPOT']
if removeSPOT is True:
arcpy.AddMessage("Observer SPOT is <NULL>, deleteing field ...")
arcpy.DeleteField_management(observers, "SPOT")
# Do a Minimum Bounding Geometry (MBG) on the input observers
observers_mbg = os.path.join(env.scratchWorkspace, "observers_mbg")
delete_me.append(observers_mbg)
arcpy.AddMessage("Finding observer's minimum bounding envelope ...")
# ENVELOPE would be better but would make it ArcInfo-only.
arcpy.MinimumBoundingGeometry_management(observers, observers_mbg,
"RECTANGLE_BY_AREA")
# Now find the center of the (MBG)
arcpy.AddMessage("Finding center of observers ...")
mbgCenterPoint = os.path.join(env.scratchWorkspace, "mbgCenterPoint")
mbgExtent = arcpy.Describe(observers_mbg).extent
mbgSR = arcpy.Describe(observers_mbg).spatialReference
mbgCenterX = mbgExtent.XMin + (mbgExtent.XMax - mbgExtent.XMin)
mbgCenterY = mbgExtent.YMin + (mbgExtent.YMax - mbgExtent.YMin)
arcpy.CreateFeatureclass_management(os.path.dirname(mbgCenterPoint),
os.path.basename(mbgCenterPoint),
"POINT", "#", "DISABLED",
"DISABLED", mbgSR)
mbgShapeFieldName = arcpy.Describe(mbgCenterPoint).ShapeFieldName
rows = arcpy.InsertCursor(mbgCenterPoint)
feat = rows.newRow()
feat.setValue(mbgShapeFieldName, arcpy.Point(mbgCenterX, mbgCenterY))
rows.insertRow(feat)
del rows
delete_me.append(mbgCenterPoint)
# Get the maximum radius of the observers
maxRad = obsMaximums['RADIUS2']
maxOffset = obsMaximums['OFFSETA']
horizonDistance = 0.0
z_factor = float(zfactor(observers))
if RADIUS2_to_infinity is True:
''' if going to infinity what we really need is the distance to the horizon
based on height/elevation'''
arcpy.AddMessage("Finding horizon distance ...")
result = arcpy.GetCellValue_management(
input_surface,
str(mbgCenterX) + " " + str(mbgCenterY))
centroid_elev = result.getOutput(0)
R2 = float(centroid_elev) + float(maxOffset)
# length, in meters, of semimajor axis of WGS_1984 spheroid.
R = 6378137.0
horizonDistance = math.sqrt(math.pow((R + R2), 2) - math.pow(R, 2))
arcpy.AddMessage(str(horizonDistance) + " meters.")
horizonExtent = (str(mbgCenterX - horizonDistance) + " " +
str(mbgCenterY - horizonDistance) + " " +
str(mbgCenterX + horizonDistance) + " " +
str(mbgCenterY + horizonDistance))
# since we are doing infinity we can drop the RADIUS2 field
arcpy.AddMessage(
"Analysis to edge of surface, dropping RADIUS2 field ...")
arcpy.DeleteField_management(observers, "RADIUS2")
else:
pass
# reset center of AZED using Lat/Lon of MBG center point
# Project point to WGS 84
arcpy.AddMessage("Recentering Azimuthal Equidistant to centroid ...")
mbgCenterWGS84 = os.path.join(env.scratchWorkspace, "mbgCenterWGS84")
arcpy.Project_management(mbgCenterPoint, mbgCenterWGS84, GCS_WGS_1984)
arcpy.AddXY_management(mbgCenterWGS84)
pointx = 0.0
pointy = 0.0
shapeField = arcpy.Describe(mbgCenterWGS84).ShapeFieldName
rows = arcpy.SearchCursor(mbgCenterWGS84)
for row in rows:
feat = row.getValue(shapeField)
pnt = feat.getPart()
pointx = pnt.X
pointy = pnt.Y
del row
del rows
# write new central meridian and latitude of origin...
strAZED = '''PROJCS["World_Azimuthal_Equidistant",
GEOGCS["GCS_WGS_1984",
DATUM["D_WGS_1984",
SPHEROID["WGS_1984",6378137.0,298.257223563]],
PRIMEM["Greenwich",0.0],
UNIT["Degree",0.0174532925199433]],
PROJECTION["Azimuthal_Equidistant"],
PARAMETER["False_Easting",0.0],
PARAMETER["False_Northing",0.0],
PARAMETER["Central_Meridian",' + str(pointx) + '],
PARAMETER["Latitude_Of_Origin",' + str(pointy) + '],
UNIT["Meter",1.0],
AUTHORITY["ESRI",54032]]'''
delete_me.append(mbgCenterWGS84)
# Clip the input surface to the maximum visibilty range and extract
# it to a 1000 x 1000 raster
# if going to infinity then clip to horizion extent
surf_extract = os.path.join(env.scratchWorkspace, "surf_extract")
if RADIUS2_to_infinity is True:
mbgBuffer = os.path.join(env.scratchWorkspace, "mbgBuffer")
arcpy.Buffer_analysis(observers_mbg, mbgBuffer, horizonDistance)
delete_me.append(mbgBuffer)
surfaceSR = arcpy.Describe(input_surface).spatialReference
mbgBufferPrj = os.path.join(env.scratchWorkspace, "mbgBufferPrj")
arcpy.Project_management(mbgBuffer, mbgBufferPrj, surfaceSR)
delete_me.append(mbgBufferPrj)
mbgBufferPrjExtent = arcpy.Describe(mbgBufferPrj).extent
cellSize = max(
float(mbgBufferPrjExtent.width) / 1000.0,
float(mbgBufferPrjExtent.height) / 1000.0)
env.cellSize = cellSize
arcpy.AddMessage(
"Clipping and resampling surface to analysis area with " +
str(cellSize) + " meter cell size ...")
arcpy.Clip_management(input_surface, "#", surf_extract,
mbgBufferPrj)
else:
# buffer MBG by max RADIUS 2 + 10%
mbgBuffer = os.path.join(env.scratchWorkspace, "mbgBuffer")
arcpy.Buffer_analysis(observers_mbg, mbgBuffer,
obsMaximums['RADIUS2'])
delete_me.append(mbgBuffer)
# project buffer to surface SR
surfaceSR = arcpy.Describe(input_surface).spatialReference
mbgBufferPrj = os.path.join(env.scratchWorkspace, "mbgBufferPrj")
arcpy.Project_management(mbgBuffer, mbgBufferPrj, surfaceSR)
delete_me.append(mbgBufferPrj)
# clip surface to projected buffer
arcpy.Clip_management(input_surface, "#", surf_extract,
mbgBufferPrj)
delete_me.append(surf_extract)
# Project surface to the new AZED
extract_prj = os.path.join(env.scratchWorkspace, "extract_prj")
arcpy.AddMessage("Projecting surface ...")
arcpy.ProjectRaster_management(surf_extract, extract_prj, strAZED)
delete_me.append(extract_prj)
# Project observers to the new AZED
obs_prj = os.path.join(env.scratchWorkspace, "obs_prj")
arcpy.AddMessage("Projecting observers ...")
arcpy.Project_management(observers, obs_prj, strAZED)
delete_me.append(obs_prj)
# Project the MBG buffer to AZED
obs_buf = os.path.join(env.scratchWorkspace, "obs_buf")
# if RADIUS2_to_infinity == True:
# arcpy.Buffer_analysis(obs_prj,obs_buf,horizonDistance)
# else:
# arcpy.Project_management(mbgBufferPrj,obs_buf,strAZED)
arcpy.Project_management(mbgBufferPrj, obs_buf, strAZED)
delete_me.append(obs_buf)
# Finally ... run Viewshed
arcpy.AddMessage("Calculating Viewshed ...")
vshed = os.path.join(env.scratchWorkspace, "vshed")
delete_me.append(vshed)
outVshed = sa.Viewshed(extract_prj, obs_prj, 1.0, "CURVED_EARTH",
terrestrial_refractivity_coefficient)
outVshed.save(vshed)
# Raster To Polygon
arcpy.AddMessage("Converting to polygons ...")
ras_poly = os.path.join(env.scratchWorkspace, "ras_poly")
arcpy.RasterToPolygon_conversion(vshed, ras_poly, polygon_simplify)
delete_me.append(ras_poly)
# clip output polys to buffer
if RADIUS2_to_infinity is not True:
out_buf = os.path.join(env.scratchWorkspace, "out_buf")
arcpy.Buffer_analysis(obs_prj, out_buf, "RADIUS2")
delete_me.append(out_buf)
arcpy.Clip_analysis(ras_poly, out_buf, output_rlos)
else:
arcpy.CopyFeatures_management(ras_poly, output_rlos)
# set output
arcpy.SetParameter(2, output_rlos)
# cleanup
arcpy.AddMessage("Removing scratch datasets:")
for ds in delete_me:
arcpy.AddMessage(str(ds))
arcpy.Delete_management(ds)
except arcpy.ExecuteError:
# Get the tool error messages
msgs = arcpy.GetMessages()
arcpy.AddError(msgs)
# print msgs #UPDATE2to3
print(msgs)
except:
# Get the traceback object
tb = sys.exc_info()[2]
tbinfo = traceback.format_tb(tb)[0]
# Concatenate information together concerning the error into a
# message string
pymsg = ("PYTHON ERRORS:\nTraceback info:\n" + tbinfo +
"\nError Info:\n" + str(sys.exc_info()[1]))
msgs = "ArcPy ERRORS:\n" + arcpy.GetMessages() + "\n"
# Return python error messages for use in script tool or Python Window
arcpy.AddError(pymsg)
arcpy.AddError(msgs)
# Print Python error messages for use in Python / Python Window
# print pymsg + "\n" #UPDATE2to3
print(pymsg + "\n")
# print msgs #UPDATE2to3
print(msgs)
