arcgis python脚本

Script Name: Spider_Create
#Script Label: Create Spider Diagrams
#
#Description: Custom python script developed to create spider diagrams from points contained
#in one or two feature classes. If using two feature classes, one feature class must contain
#the origins and the other must contain the destinations.
#
#Source Script File: Spider_Create.py
#Programming Language: Python 2.4
#ArcGIS Version: ArcGIS 9.2
#Author: Anthony Palmer, U.S. Army Corps of Engineers, Tennessee-Tombigbee Waterway
#Email: anthony.g.palmer@sam.usace.army.mil
#Date: 04/04/2007
#
#Parameters:
#
# Display Name Data Type Type Direction MultiValue Dependency
# argv[1] Origin Point Feature Class Feature Class Required Input No
# argv[2] Select Subset of Origins SQL Expression Optional Input No Origin Point Feature Class
# argv[3] Select Origin Key Field Field Required Input No Origin Point Feature Class
# argv[4] Destination Point Feature Class Feature Class Required Input No
# argv[2] Select Subset of Destinations SQL Expression Optional Input No Destination Point Feature Class
# argv[6] Select Destination Key Field Field Required Input No Destination Point Feature Class
# argv[7] Output Polyline Feature Class Feature Class Required Output No
#----------------------------------------------------------------------------------------------------------------------
#
#import modules and create the geoprocessor object
#
import arcgisscripting, os, sys, string, math
gp = arcgisscripting.create()
#
# Get the input and output feature class names, key fields, and optional sql expressions
#
orgFC = sys.argv[1] # input feature class
orgExp = sys.argv[2] # expression identifying origins
orgkey = sys.argv[3] # origin key field
desFC = sys.argv[4] # destination feature class
desExp = sys.argv[5] # expression identifying destinations
deskey = sys.argv[6] # destination key field
outFC = sys.argv[7] # output polyline feature class
#
#Verify origin and destination features classes are points and determine their projection.
#
orgdesc = gp.Describe(orgFC)
desdesc = gp.Describe(desFC)
orgShapeField = orgdesc.ShapeFieldName
desShapeField = desdesc.ShapeFieldName
orgsr = orgdesc.SpatialReference
dessr = desdesc.SpatialReference
orgsrtype = str(orgsr.Type)
dessrtype = str(dessr.Type)
if orgdesc.ShapeType == "Point" and desdesc.ShapeType == "Point":
if orgsrtype == "Unknown" or dessrtype == "Unknown":
gp.AddError("Error: Either the projection for the origins, destinations, or both is undefined. (Exiting Script)")
sys.exit()
elif orgsrtype != dessrtype:
gp.AddError("Error: The origin and destination feature classes must have the same projection. (Exiting Script)")
sys.exit()
else:
pass
else:
gp.AddError("Error: Either the origins, destinations, or both are not point feature classes. (Exiting Script)")
sys.exit()
#
#Verify origin and destination key fields and identify field types
#
orgfields = gp.ListFields(orgFC)
orgfield = orgfields.Next()
desfields = gp.ListFields(desFC)
desfield = desfields.Next()
try:
while orgfield:
if orgfield.Name == orgkey:
orgftype = orgfield.Type
orgflength = orgfield.Length
break
else:
pass
orgfield = orgfields.Next()
if orgftype == "String":
orgntype = "Text"
orgnlength = orgflength
elif orgftype == "Integer":
orgntype = "Long"
elif orgftype == "SmallInteger":
orgntype = "Short"
else:
gp.AddError("Origin Key field type is not a string, integer, or small integer. (Exiting Script).")
sys.exit()
del orgfields
del orgfield
while desfield:
if desfield.Name == deskey:
desftype = desfield.Type
desflength = desfield.Length
break
else:
pass
desfield = desfields.Next()
if desftype == "String":
desntype = "Text"
desnlength = desflength
elif desftype == "Integer":
desntype = "Long"
elif desftype == "SmallInteger":
desntype = "Short"
else:
gp.AddError("Destination Key field type is not a string, integer, or small integer. (Exiting Script).")
sys.exit()
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Problem encountered identifying the origin-destination key fields type and length. (Exiting Script)")
sys.exit()
#
#Build origin coordinate-key list from origin feature class
#
def org_lists(orgFC):
if orgExp != "#":
in_Layer = "in_Layer"
org_Layer = "org_Layer"
gp.MakeFeatureLayer_management(orgFC, in_Layer, "", "", "")
gp.SelectLayerByAttribute_management(in_Layer, "NEW_SELECTION", orgExp)
gp.MakeFeatureLayer_management(in_Layer, org_Layer, "", "", "")
rows = gp.SearchCursor(org_Layer)
row = rows.next()
orgxy = []
orgkeylist = []
try:
while row:
ashape = row.shape
pnt = ashape.GetPart()
pntx = pnt.x
pnty = pnt.y
if str(pntx) == "1.#QNAN" or str(pnty) == "1.#QNAN":
pass
else:
orgkeyid = row.GetValue(orgkey)
orgkeylist.append(orgkeyid)
orgxy.append(float(pntx))
orgxy.append(float(pnty))
row = rows.next()
del rows
del row
if len(orgxy) == 0:
gp.AddError("Valid origin values could not be identified. This may be because records were selected /
with 'Not a Number' coordinate values. Common for 'unmatched records' in a geocoded address database. (Exiting Script)")
sys.exit()
else:
pass
gp.AddMessage("Origin coordinates and key values added to origin list.")

return (orgxy, orgkeylist)
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Problem encountered creating origin coordinate-key list. (Exiting script)")
del rows
del row
sys.exit()
else:
rows = gp.SearchCursor(orgFC)
row = rows.next()
orgxy = []
orgkeylist = []
try:
while row:
ashape = row.shape
pnt = ashape.GetPart()
pntx = pnt.x
pnty = pnt.y
if str(pntx) == "1.#QNAN" or str(pnty) == "1.#QNAN":
pass
else:
orgkeyid = row.GetValue(orgkey)
orgkeylist.append(orgkeyid)
orgxy.append(float(pntx))
orgxy.append(float(pnty))
row = rows.next()
del rows
del row
if len(orgxy) == 0:
gp.AddError("Valid origin values could not be identified. This may be because records were selected /
with 'Not a Number' coordinate values. Common for 'unmatched records' in a geocoded address database. (Exiting Script)")
sys.exit()
else:
pass
gp.AddMessage("Origin coordinates and key values added to origin list.")
return (orgxy, orgkeylist)
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Problem encountered creating origin coordinate-key list. (Exiting script)")
del rows
del row
sys.exit()
#
#Build destination coordinate-key list from destination feature class
#
def des_lists(desFC):
if desExp != "#":
in_Layer2 = "in_Layer2"
des_Layer = "des_Layer"
gp.MakeFeatureLayer_management(desFC, in_Layer2, "", "", "")
gp.SelectLayerByAttribute_management(in_Layer2, "NEW_SELECTION", desExp)
gp.MakeFeatureLayer_management(in_Layer2, des_Layer, "", "", "")
rows = gp.SearchCursor(des_Layer)
row = rows.next()
desxy = []
deskeylist = []
try:
while row:
ashape = row.shape
pnt = ashape.GetPart()
pntx = pnt.x
pnty = pnt.y
if str(pntx) == "1.#QNAN" or str(pnty) == "1.#QNAN":
pass
else:
deskeyid = row.GetValue(deskey)
deskeylist.append(deskeyid)
desxy.append(float(pntx))
desxy.append(float(pnty))
row = rows.next()
del rows
del row
if len(desxy) == 0:
gp.AddError("Valid destination values could not be identified. This may be because records were selected /
with 'Not a Number' coordinate values. Common for 'unmatched records' in a geocoded address database. (Exiting Script)")
sys.exit()
else:
pass
gp.AddMessage("Destination coordinates and key values added to destination list.")
return (desxy, deskeylist)
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Problem encountered creating destination coordinate-key list. (Exiting script)")
del rows
del row
sys.exit()
else:
rows = gp.SearchCursor(desFC)
row = rows.next()
desxy = []
deskeylist = []
try:
while row:
ashape = row.shape
pnt = ashape.GetPart()
pntx = pnt.x
pnty = pnt.y
if str(pntx) == "1.#QNAN" or str(pnty) == "1.#QNAN":
pass
else:
deskeyid = row.GetValue(deskey)
deskeylist.append(deskeyid)
desxy.append(float(pntx))
desxy.append(float(pnty))
row = rows.next()
del rows
del row
if len(desxy) == 0:
gp.AddError("Valid destination values could not be identified. This may be because records were selected /
with 'Not a Number' coordinate values. Common for 'unmatched records' in a geocoded address database. (Exiting Script)")
sys.exit()
else:
pass
gp.AddMessage("Destination coordinates and key values added to destination list.")
return (desxy, deskeylist)
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Problem encountered creating destination coordinate-key list. (Exiting script)")
del rows
del row
sys.exit()
#
#Assign values returned by the org_lists and des_lists functions to list variables.
#
org_lists = org_lists(orgFC)
des_lists = des_lists(desFC)
#
#Create the polyline feature class and add 'ORG_ID', 'DES_ID' & 'DES_LENGTH' fields'.
#
try:
gp.CreateFeatureClass(os.path.dirname(outFC),os.path.basename(outFC),"Polyline","#","#","#",orgsr)
gp.AddMessage("Polyline feature class created successfully.")
if orgntype == "text":
gp.AddField(outFC, "ORG_ID", orgntype, "#", "#", orgnlength)
else:
gp.AddField(outFC, "ORG_ID", orgntype, "#", "#", "#", "#", "#", "#", "#")
if desntype == "text":
gp.AddField(outFC, "DES_ID", desntype, "#", "#", desnlength)
else:
gp.AddField(outFC, "DES_ID", desntype, "#", "#", "#", "#", "#", "#", "#")
gp.AddField(outFC, "DES_LENGTH", "DOUBLE", "18", "6", "#", "#", "#", "#", "#")
gp.AddMessage("Added 'ORG_ID', 'DES_ID' & 'DES_LENGTH' Fields to the table.")
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Problem encountered creating feature class. (Exiting script)")
sys.exit()
#
#Construct polyline features
#
orgxy = org_lists[0]
orgkeylist = org_lists[1]
orglen = len(orgxy)
desxy = des_lists[0]
deskeylist = des_lists[1]
deslen = len(desxy)
gp.AddMessage("Beginning Polyline Construction...")
cur = gp.InsertCursor(outFC)
lineArray = gp.CreateObject("Array")
pnt = gp.CreateObject("Point")
dinc = 0
oinc = 0
odlist = []
okey = 0
dkey = 0
try:
while oinc < orglen:
orgx = oinc
orgy = oinc + 1
dinc = 0
dkey = 0
while dinc < deslen:
desx = dinc
desy = dinc + 1
deltaxy = math.sqrt((desxy[desx]-orgxy[orgx])**2 + (desxy[desy]-orgxy[orgy])**2)
if deltaxy == 0:
dinc = dinc + 2
dkey = dkey + 1
else:
odlist.append(orgkeylist[okey])
odlist.append(deskeylist[dkey])
pnt.x = orgxy[orgx]
pnt.y = orgxy[orgy]
lineArray.add(pnt)
pnt.x = desxy[desx]
pnt.y = desxy[desy]
lineArray.add(pnt)
feat = cur.NewRow()
feat.shape = lineArray
cur.InsertRow(feat)
lineArray.RemoveAll()
dinc = dinc + 2
dkey = dkey + 1
oinc = oinc + 2
okey = okey + 1
del cur
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Unable to construct polyline. (Exiting Script)")
del cur
sys.exit()
#
#Populate 'ORG_ID', 'DES_ID' & 'DES_LENGTH' fields with values.
#
rows = gp.UpdateCursor(outFC)
row = rows.next()
gp.AddMessage("Updating 'ORG_ID', 'DES_ID' & 'DES_LENGTH' Fields...")
ikey = 0
try:
while row:
dfeat = row.shape
dlength = dfeat.Length
oval = odlist[ikey]
dval = odlist[ikey + 1]
row.SetValue("ORG_ID", oval)
row.SetValue("DES_ID", dval)
row.SetValue("DES_LENGTH", dlength)
rows.UpdateRow(row)
ikey = ikey + 2
row = rows.next()
del rows
del row
gp.AddMessage("Completed update of 'ORG_ID', 'DES_ID' & 'DES_LENGTH fields.")
except:
gp.AddWarning(gp.getmessages())
gp.AddError("Encountered a problem updating 'ORG_ID', 'DES_ID' & 'DES_LENGTH' fields. (Exiting Script)")
del rows
del row
sys.exit()
#
del gp