复制CAD图形到arcgis

复制CAD图形到arcgis

-- coding: utf-8 --

import arcpy, glob, os
import pythonaddins, pythoncom
import win32com.client as win32#输出dwg文件
layer = ‘’
def vtInt(list):
“”“列表转化为整数”“”
return win32.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_I2, list)
def vtVariant(list):
“”“列表转化为变体”“”
return win32.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_VARIANT, list)
def vtfloat(lst):
return win32.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_R8, lst)
def sel_point(text):
list_xy = []
list_xy1 = []
for i in range(int(len(text)/2)):
list_xy.append(text[2i])
list_xy.append(text[2i+1])
list_xy.append(0)

    x = text[2 * i]
    y = text[2 * i + 1]
    list_xy1.append([x, y])
#list_xy1.append([text[0], text[1]])
return list_xy1

class ButtonClass11(object):
“”“Implementation for fzcadtx_addin.button_1 (Button)”“”
def init(self):
self.enabled = True
self.checked = False
def onClick(self):
global layer
if layer == ‘’:
layer = ‘CAD’
arcpy.DefineProjection_management(layer, 4526)

class ButtonClass12(object):
“”“Implementation for fzcadtx_addin.button_2 (Button)”“”
def init(self):
self.enabled = True
self.checked = False
def onClick(self):
list1 = []
with arcpy.da.SearchCursor(layer, (“SHAPE@”, “SHAPE@AREA”)) as cursor:
for row in cursor:
for part in row[0]:
list = []
for point in part:
if point:
list.append(point.X)
list.append(point.Y)
list.append(0)
else:
list1.append(list)
list = []
list1.append(list)
try:
wincad = win32.Dispatch(“AutoCAD.Application.17.1”)
except:
wincad = win32.Dispatch(“AutoCAD.Application”)
doc = wincad.ActiveDocument
msp = doc.ModelSpace
for list_xy in list1:
points = vtfloat(list_xy)
plineObj = msp.AddPolyline(points)
plineObj.Closed = True

#复制CAD图形到arcgis
class ButtonClass2(object):
“”“Implementation for fzcadtx_addin.button (Button)”“”
def init(self):
self.enabled = True
self.checked = False
def onClick(self):
try:
global layer
try:
wincad = win32.Dispatch(“AutoCAD.Application.17”)
except:
wincad = win32.Dispatch(“AutoCAD.Application”)
doc = wincad.ActiveDocument
msp = doc.ModelSpace
try:
doc.SelectionSets.Item(“SS1”).Delete()
except:
print(“Delete selection failed”)
slt = doc.SelectionSets.Add(“SS1”) # 创建选择集
slt.SelectOnScreen() # 选择
i = 0
if layer == ‘’ or layer == r’D:\CAD.shp’:
list_dj = glob.glob(r"D:\cad.*")
for path in list_dj:
try:
os.remove(path)#polygon
except:
pass
arcpy.CreateFeatureclass_management(“D:”, “CAD.shp”, “POLYGON”, “”, “DISABLED”,
“DISABLED”)
layer = r’D:\CAD.shp’
cur = arcpy.da.InsertCursor(layer, (“SHAPE@”))
for item in slt:
xy = sel_point(item.Coordinates)
array = arcpy.Array()
point = arcpy.Point()
ii = 1
for xys in xy:
point.ID = ii
point.X = xys[0]
point.Y = xys[1]
array.add(point)
ii += 1
polygon = arcpy.Polygon(array)
cur.insertRow([polygon])
del cur
arcpy.RefreshActiveView()
except Exception as e:
pythonaddins.MessageBox(e, ‘INFO’, 0)

class ComboBoxClass1(object):
“”“Implementation for fzcadtx_addin.combobox (ComboBox)”“”
def init(self):
self.items = [‘’,‘’,‘’,‘’,‘’,‘’,‘’,‘’]
self.editable = True
self.enabled = True
self.dropdownWidth = ‘WWWWWWWWWWWWWWWWWWWWWWW’
self.width = ‘WWWWWW’
def onSelChange(self, selection):
global layer
layer = selection
def onEditChange(self, text):
pass
def onFocus(self, focused):
if focused:
self.mxd = arcpy.mapping.MapDocument(‘current’)
layers = arcpy.mapping.ListLayers(self.mxd)
self.items = []
self.items.append(‘’)
for layer in layers:
self.items.append(layer.name)
def onEnter(self):
pass
def refresh(self):
pass