solidworks自动标注-python实现
正文
标注前：
标注后：
标注逻辑：
1、打开solidworks工程图并选中一个视图：
2、执行代码
（执行时需要把swconst.py放在这段代码的文件同一目录下）
import time
import math
import pyautogui
import win32com.client as win32
import threading
import pythoncom
from swconst import constants as swconst
swerrors = win32.VARIANT(pythoncom.VT_BYREF | pythoncom.VT_I4, 3)
swwarnings = win32.VARIANT(pythoncom.VT_BYREF | pythoncom.VT_I4, 3)
def getcirclesort(xpoint1,xcircle2,xcirclepoint2):
#根据第一个参数的指数，从小到大排列三个参数,传入的后两个参数会被清空
xcircleinfo = []
xcircle3 = []
xcirclepoint3 = []
xpoint2 = []
t = len(xpoint1)
for i in range(t):
xpoint1min = xpoint1[0]
xpoint1minindex = 0
for j in range(len(xpoint1)):
if xpoint1[j] < xpoint1min:
xpoint1min = xpoint1[j]
xpoint1minindex = j
xcircle3.append(xcircle2[xpoint1minindex])
xcirclepoint3.append(xcirclepoint2[xpoint1minindex])
xpoint2.append(xpoint1min)
xcircle2.remove(xcircle2[xpoint1minindex])
xcirclepoint2.remove(xcirclepoint2[xpoint1minindex])
xpoint1.remove(xpoint1min)
xcircleinfo.append(xcircle3)
xcircleinfo.append(xcirclepoint3)
xcircleinfo.append(xpoint2)
return(xcircleinfo)
def getonlyinfo(xcircle1,xcirclepoint1,index):
#根据第二个参数去重
onlyinfo = []
xcircle2 = []
xcirclepoint2 = []
xpoint1 = []
for i in range(len(xcircle1)):
if xcirclepoint1[i][index] not in xpoint1:
xpoint1.append(xcirclepoint1[i][index])
xcircle2.append(xcircle1[i])
xcirclepoint2.append(xcirclepoint1[i])
#水平方向用y坐标小的点，竖直方向用x坐标大的点
##########
else:
location = xpoint1.index(xcirclepoint1[i][index])
if index == 0:
if xcirclepoint1[i][1] < xcirclepoint1[location][1]:
xpoint1[location] = xcirclepoint1[i][index]
xcircle2[location] = xcircle1[i]
xcirclepoint2[location] = xcirclepoint1[i]
elif index == 1:
if xcirclepoint1[i][0] > xcirclepoint1[location][0]:
xpoint1[location] = xcirclepoint1[i][index]
xcircle2[location] = xcircle1[i]
xcirclepoint2[location] = xcirclepoint1[i]
#####################
onlyinfo.append(xcircle2)
onlyinfo.append(xcirclepoint2)
onlyinfo.append(xpoint1)
return(onlyinfo)
def getcirclesortinfo(closecircle,closecirclepoint):
#孔排序
circlesortinfo = []
#x方向把孔坐标排序
xcircle1 = []
xcirclepoint1 = []
for i in range(len(closecircle)):
xcircle1.append(closecircle[i])
xcirclepoint1.append(closecirclepoint[i])
#去掉x坐标重复的点
xonlyinfo = getonlyinfo(xcircle1,xcirclepoint1,0)
xcircle2 = xonlyinfo[0]
xcirclepoint2 = xonlyinfo[1]
xpoint1 = xonlyinfo[2]
#x坐标从小到大排序
xcircleinfo = getcirclesort(xpoint1,xcircle2,xcirclepoint2)
xcircle3 = xcircleinfo[0]
xcirclepoint3 = xcircleinfo[1]
xpoint2 = xcircleinfo[2]
#y方向把孔坐标排序
ycircle1 = []
ycirclepoint1 = []
for i in range(len(closecircle)):
ycircle1.append(closecircle[i])
ycirclepoint1.append(closecirclepoint[i])
#去掉y坐标重复的点
xonlyinfo = getonlyinfo(ycircle1,ycirclepoint1,1)
ycircle2 = xonlyinfo[0]
ycirclepoint2 = xonlyinfo[1]
ypoint1 = xonlyinfo[2]
#y坐标从小到大排序
ycircleinfo = getcirclesort(ypoint1,ycircle2,ycirclepoint2)
ycircle3 = ycircleinfo[0]
ycirclepoint3 = ycircleinfo[1]
ypoint2 = ycircleinfo[2]
circlesortinfo.append(xcircle3)
circlesortinfo.append(xcirclepoint3)
circlesortinfo.append(xpoint2)
circlesortinfo.append(ycircle3)
circlesortinfo.append(ycirclepoint3)
circlesortinfo.append(ypoint2)
return(circlesortinfo)
def getboundaryline(view,alllineinfo):
boundarylineinfo = []
xedge = alllineinfo[0]
yedge = alllineinfo[1]
xedgey = alllineinfo[2]
yedgex = alllineinfo[3]
#找出水平线最上和最下
up = xedgey[0]
lineup = xedge[0]
for i in range(len(xedge)):
if xedgey[i] > up:
up = xedgey[i]
lineup = xedge[i]
down = xedgey[0]
linedown = xedge[0]
for i in range(len(xedge)):
if xedgey[i] < down:
down = xedgey[i]
linedown = xedge[i]
#找出竖直线最左和最右
right = yedgex[0]
lineright = yedge[0]
for i in range(len(yedge)):
if yedgex[i] > right:
right = yedgex[i]
lineright = yedge[i]
left = yedgex[0]
lineleft = yedge[0]
for i in range(len(yedge)):
if yedgex[i] < left:
left = yedgex[i]
lineleft = yedge[i]
boundarylineinfo.append(lineup)
boundarylineinfo.append(linedown)
boundarylineinfo.append(lineleft)
boundarylineinfo.append(lineright)
boundarylineinfo.append(right)
boundarylineinfo.append(left)
boundarylineinfo.append(up)
boundarylineinfo.append(down)
return(boundarylineinfo)
def getalledgeinfo(view,vedges):
#输入：视图中的线段
#返回：直线和圆弧信息
#与solidworks通信
alledgeinfo = []
allline = []
alllineparam = []
allcircle = []
allcircleparam1 = []
allcircleparam2 = []
for i in range(len(vedges)):
swcurve = vedges[i].GetCurve
if swcurve.IsLine:
print('这是一条直线')
allline.append(vedges[i])
alllineparam.append(vedges[i].GetCurveParams2)
elif swcurve.IsCircle:
print('这是一个圆')
allcircle.append(vedges[i])
allcircleparam1.append(vedges[i].GetCurveParams2)
allcircleparam2.append(swcurve.CircleParams)
alledgeinfo.append(allline)
alledgeinfo.append(alllineparam)
alledgeinfo.append(allcircle)
alledgeinfo.append(allcircleparam1)
alledgeinfo.append(allcircleparam2)
return(alledgeinfo)
def getalllineinfo(view,alledgeinfo):
alllineinfo = []
allline = alledgeinfo[0]
alllineparam = alledgeinfo[1]
xedge = []
yedge = []
xedgey = []
yedgex = []
for i in range(len(alllineparam)):
start = transform4(view,alllineparam[i][0],alllineparam[i][1],alllineparam[i][2])
end = transform4(view,alllineparam[i][3],alllineparam[i][4],alllineparam[i][5])
#线段两个端点x坐标相等，该线竖直
if start[0] == end[0]:
yedge.append(allline[i])
yedgex.append(start[0])
#线段两个端点y坐标相等，该线水平
if start[1] == end[1]:
xedge.append(allline[i])
xedgey.append(start[1])
alllineinfo.append(xedge)
alllineinfo.append(yedge)
alllineinfo.append(xedgey)
alllineinfo.append(yedgex)
return(alllineinfo)
def getallcircleinfo(view,alledgeinfo):
allcircleinfo = []
allcircle = alledgeinfo[2]
allcircleparam1 = alledgeinfo[3]
allcircleparam2 = alledgeinfo[4]
closecircle = []
closecircleR = []
closecirclepoint = []
for i in range(len(allcircle)):
start = transform4(view,round(allcircleparam1[i][0],4),round(allcircleparam1[i][1],4),round(allcircleparam1[i][2],4))
end = transform4(view,round(allcircleparam1[i][3],4),round(allcircleparam1[i][4],4),round(allcircleparam1[i][5],4))
if start == end:
startangle = math.degrees(allcircleparam1[i][6])
endangle = math.degrees(allcircleparam1[i][7])
if startangle - endangle == 360 or startangle - endangle == -360:
circleR = allcircleparam2[i][6]
center = transform4(view,round(allcircleparam2[i][0],4),round(allcircleparam2[i][1],4),round(allcircleparam2[i][2],4))
center.remove(center[2])
closecircle.append(allcircle[i])
closecircleR.append(circleR)
closecirclepoint.append(center)
allcircleinfo.append(closecircle)
allcircleinfo.append(closecircleR)
allcircleinfo.append(closecirclepoint)
return(allcircleinfo)
def dimensionxcircle(part,scale,vbounds,xcirclepoint3,xcircle3,xmin,xmax,ymin,ymax,lineleft,lineright,yoffset2):
#标注水平方向孔位置尺寸
xpos = (xcirclepoint3[0][0] + xmin)/2 * scale + vbounds[0] + xoffset
ypos = ymin * scale + vbounds[1] + yoffset - yoffset2
part.ClearSelection2(True)
lineleft.Select4(True,arg_Nothing)
xcircle3[0].Select4(True,arg_Nothing)
part.AddVerticalDimension2(xpos,ypos,0)
for i in range(len(xcircle3)-1):
xpos = (xcirclepoint3[i][0] + xcirclepoint3[i+1][0])/2 * scale + vbounds[0] + xoffset
ypos = ymin * scale + vbounds[1] + yoffset - yoffset2
part.ClearSelection2(True)
xcircle3[i].Select4(True,arg_Nothing)
xcircle3[i+1].Select4(True,arg_Nothing)
part.AddHorizontalDimension2(xpos,ypos,0)
def dimensionxmax(part,scale,vbounds,xmin,xmax,ymin,ymax,lineleft,lineright,yoffset2):
xpos = (xmax + xmin)/2 * scale + vbounds[0] + xoffset
ypos = ymin * scale + vbounds[1] + yoffset - yoffset2
part.ClearSelection2(True)
lineleft.Select4(True,arg_Nothing)
lineright.Select4(True,arg_Nothing)
part.AddVerticalDimension2(xpos,ypos,0)
def dimensionycircle(part,scale,vbounds,ycirclepoint3,ycircle3,xmin,xmax,ymin,ymax,lineup,linedown,xoffset2):
#标注竖直方向孔位置尺寸
xpos = xmax * scale + vbounds[0] + xoffset + xoffset2
ypos = (ycirclepoint3[0][1] + ymin)/2 * scale + vbounds[1] + yoffset
part.ClearSelection2(True)
linedown.Select4(True,arg_Nothing)
ycircle3[0].Select4(True,arg_Nothing)
part.AddHorizontalDimension2(xpos,ypos,0)
for i in range(len(ycircle3)-1):
xpos = xmax * scale + vbounds[0] + xoffset + xoffset2
ypos = (ycirclepoint3[i][1] + ycirclepoint3[i+1][1])/2 * scale + vbounds[1] + yoffset
part.ClearSelection2(True)
ycircle3[i].Select4(True,arg_Nothing)
ycircle3[i+1].Select4(True,arg_Nothing)
part.AddVerticalDimension2(xpos,ypos,0)
def dimensionymax(part,scale,vbounds,xmin,xmax,ymin,ymax,lineup,linedown,xoffset2):
#标注竖直最大尺寸
xpos = xmax * scale + vbounds[0] + xoffset + xoffset2
ypos = (ymax + ymin)/2 * scale + vbounds[1] + yoffset
part.ClearSelection2(True)
lineup.Select4(True,arg_Nothing)
linedown.Select4(True,arg_Nothing)
part.AddHorizontalDimension2(xpos,ypos,0)
def getpointsort(allclosecirclepointx1xy3,t):
allclosecirclepointx1xy = []
if t == 0:
q =1
elif t == 1:
q = 0
for i in range(len(allclosecirclepointx1xy3)):
allclosecirclepointx1xy3[i].sort(key=lambda x: (x[t],x[q]))
allclosecirclepointx1xy.append(allclosecirclepointx1xy3[i])
return(allclosecirclepointx1xy)
def transform4(view,x,y,z):
point = swMathUtil.CreatePoint
point.ArrayData=win32.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_R8, (x,y,z))
swviewxform = view.ModelToViewTransform
viewpoint = point.MultiplyTransform(swviewxform)
viewpoint1 = []
viewpoint1.append(round(viewpoint.ArrayData[0],4))
viewpoint1.append(round(viewpoint.ArrayData[1],4))
viewpoint1.append(round(viewpoint.ArrayData[2],4))
return(viewpoint1)
def autodimension(): #自动标注
global swapp
global xoffset
global yoffset
global arg_Nothing
global swMathUtil
#连接solidworks
print('正在连接solidworks')
#swapp = win32.Dispatch("Sldworks.application") #引入sldworks接口
sw_edition = 2016
swapp = win32.Dispatch('SldWorks.Application.{}'.format(sw_edition - 1992))
swapp.Visible = True
swMathUtil = swapp.GetMathUtility
arg_Nothing = win32.VARIANT(pythoncom.VT_DISPATCH, None) #转义VBA中不同变量nothing
print('成功连接solidworks')
part = swapp.ActiveDoc
###
part.SketchManager.AddToDB = True
#设置不弹出输入尺寸值对话框
swapp.SetUserPreferenceToggle(swconst.swInputDimValOnCreate, False)
###
swselectmgr = part.SelectionManager
view = swselectmgr.GetSelectedObject5(1)
view=win32.Dispatch(view)
#vbounds = view.GetOutline
vbounds = [0,0,0]
#视图比例
#viewscale = view.ScaleRatio
#scale = viewscale[0]/viewscale[1]
scale = 1
#偏移
#xoffset = 0.005
#yoffset = 0.005
xoffset = 0
yoffset = 0
vcomps = view.GetVisibleComponents
vedges = view.GetVisibleEntities(vcomps[0],1)
#遍历视图中的线段，返回：直线列表，圆列表
alledgeinfo = getalledgeinfo(view,vedges)
#处理直线信息,返回：水平直线，竖直直线，水平直线y坐标，竖直直线x坐标
alllineinfo = getalllineinfo(view,alledgeinfo)
#处理圆弧信息,返回：整圆，半径，圆心
allcircleinfo = getallcircleinfo(view,alledgeinfo)
closecircle = allcircleinfo[0]
closecircleR = allcircleinfo[1]
closecirclepoint = allcircleinfo[2]
xedge = alllineinfo[0]
yedge = alllineinfo[1]
xedgey = alllineinfo[2]
yedgex = alllineinfo[3]
#找出边界线
boundarylineinfo = getboundaryline(view,alllineinfo)
lineup = boundarylineinfo[0]
linedown = boundarylineinfo[1]
lineleft = boundarylineinfo[2]
lineright = boundarylineinfo[3]
xmax = boundarylineinfo[4]
xmin = boundarylineinfo[5]
ymax = boundarylineinfo[6]
ymin = boundarylineinfo[7]
onlyR = []
for i in range(len(closecircleR)):
if round(closecircleR[i],6) not in onlyR:
onlyR.append(round(closecircleR[i],6))
print('半径')
print(round(closecircleR[i],6))
allclosecircle = []
allclosecirclepoint = []
closecirclemaxx = []
closecirclemaxy = []
for i in range(len(onlyR)):
closecirclei = []
closecirclepointi = []
for j in range(len(closecircleR)):
if round(closecircleR[j],6) == onlyR[i]:
closecirclei.append(closecircle[j])
closecirclepointi.append(closecirclepoint[j])
allclosecircle.append(closecirclei)
allclosecirclepoint.append(closecirclepointi)
maxxi = closecirclepointi[0][0]
maxyi = closecirclepointi[0][1]
for j in range(len(closecirclepointi)):
if closecirclepointi[j][0] > maxxi:
maxxi = closecirclepointi[j][0]
if closecirclepointi[j][1] > maxyi:
maxyi = closecirclepointi[j][1]
closecirclemaxx.append(maxxi)
closecirclemaxy.append(maxyi)
allclosecircle1 = []
allclosecirclepoint1 = []
for i in range(len(allclosecircle)):
allclosecircle1.append(allclosecircle[i])
allclosecirclepoint1.append(allclosecirclepoint[i])
allclosecircleinfox = getcirclesort(closecirclemaxx,allclosecircle,allclosecirclepoint)
allclosecircleinfoy = getcirclesort(closecirclemaxy,allclosecircle1,allclosecirclepoint1)
#x方向
allclosecirclemaxx1 = allclosecircleinfox[2]
allclosecirclex1 = allclosecircleinfox[0]
allclosecirclepointx1 = allclosecircleinfox[1]
allclosecirclepointx1xy3 = [] #存放圆心坐标的x值和y值
for i in range(len(allclosecirclepointx1)):
circlepointj = []
for j in range(len(allclosecirclepointx1[i])):
circlepointi = []
circlepointi.append(allclosecirclepointx1[i][j][0])
circlepointi.append(allclosecirclepointx1[i][j][1])
circlepointj.append(circlepointi)
allclosecirclepointx1xy3.append(circlepointj)
#y方向
allclosecirclemaxy1 = allclosecircleinfoy[2]
allclosecircley1 = allclosecircleinfoy[0]
allclosecirclepointy1 = allclosecircleinfoy[1]
allclosecirclepointy1xy3 = [] #存放圆心坐标的x值和y值
for i in range(len(allclosecirclepointy1)):
circlepointj = []
for j in range(len(allclosecirclepointy1[i])):
circlepointi = []
circlepointi.append(allclosecirclepointy1[i][j][0])
circlepointi.append(allclosecirclepointy1[i][j][1])
circlepointj.append(circlepointi)
allclosecirclepointy1xy3.append(circlepointj)
#xy方向去重
#这里插入一个函数，将allclosecirclepointx1xy按照x方向排序，allclosecirclepointy1xy安装y方向排序
#处理allclosecirclepointx1xy3和allclosecirclepointy1xy3
allclosecirclepointx1xy = getpointsort(allclosecirclepointx1xy3,0)
allclosecirclepointy1xy = getpointsort(allclosecirclepointy1xy3,1)
#####################################
allclosecirclepointxy1 = []
allclosecirclemaxx = []
allclosecirclex = []
allclosecirclepointx = []
for i in range(len(allclosecirclepointx1)):
if allclosecirclepointx1xy[i] not in allclosecirclepointxy1:
allclosecirclepointxy1.append(allclosecirclepointx1xy[i])
allclosecirclemaxx.append(allclosecirclemaxx1[i])
allclosecirclex.append(allclosecirclex1[i])
allclosecirclepointx.append(allclosecirclepointx1[i])
allclosecirclepointxy2 = []
allclosecirclemaxy = []
allclosecircley = []
allclosecirclepointy = []
for i in range(len(allclosecirclepointy1)):
if allclosecirclepointy1xy[i] not in allclosecirclepointxy2:
allclosecirclepointxy2.append(allclosecirclepointy1xy[i])
allclosecirclemaxy.append(allclosecirclemaxy1[i])
allclosecircley.append(allclosecircley1[i])
allclosecirclepointy.append(allclosecirclepointy1[i])
####################################
#yoffset2 = float(e13.get())/1000
#yoffset2step = float(e13.get())/1000
yoffset2 = 10/1000
yoffset2step = 10/1000
for i in range(len(allclosecirclemaxx)):
circlesortinfo = getcirclesortinfo(allclosecirclex[i],allclosecirclepointx[i])
xcircle3 = circlesortinfo[0]
xcirclepoint3 = circlesortinfo[1]
xpoint2 = circlesortinfo[2]
ycircle3 = circlesortinfo[3]
ycirclepoint3 = circlesortinfo[4]
ypoint2 = circlesortinfo[5]
#标注水平方向孔位置尺寸
dimensionxcircle(part,scale,vbounds,xcirclepoint3,xcircle3,xmin,xmax,ymin,ymax,lineleft,lineright,yoffset2)
yoffset2 = yoffset2+yoffset2step
#标注水平最大尺寸
dimensionxmax(part,scale,vbounds,xmin,xmax,ymin,ymax,lineleft,lineright,yoffset2)
#xoffset2 = float(e13.get())/1000
#xoffset2step = float(e13.get())/1000
xoffset2 = 10/1000
xoffset2step = 10/1000
for i in range(len(allclosecirclemaxy)):
circlesortinfo = getcirclesortinfo(allclosecircley[i],allclosecirclepointy[i])
xcircle3 = circlesortinfo[0]
xcirclepoint3 = circlesortinfo[1]
xpoint2 = circlesortinfo[2]
ycircle3 = circlesortinfo[3]
ycirclepoint3 = circlesortinfo[4]
ypoint2 = circlesortinfo[5]
#标注竖直方向孔位置尺寸
dimensionycircle(part,scale,vbounds,ycirclepoint3,ycircle3,xmin,xmax,ymin,ymax,lineup,linedown,xoffset2)
xoffset2 = xoffset2+xoffset2step
#标注竖直最大尺寸
dimensionymax(part,scale,vbounds,xmin,xmax,ymin,ymax,lineup,linedown,xoffset2)
#设置弹出输入尺寸值对话框
swapp.SetUserPreferenceToggle(swconst.swInputDimValOnCreate, True)
#运行中开启这个选项更快
part.SketchManager.AddToDB = False
autodimension()
即可完成以下逻辑：
①：遍历选中视图中的线条，存入列表备用
②：遍历线条，按照直线和圆弧，分别存入列表备用
③：遍历直线，找出最上、最下、最左、最右的直线，存入常量备用
④：遍历圆弧，找出完整的圆，存入列表备用
⑤：圆弧按照半径分类，按照水平和竖直方向排序，分别存入列表备用
⑥：同心圆组成的列表子项去重
⑦：自动标注所有孔的位置尺寸，不同类型的孔分层标注
注：上面代码有一些无意义的操作，例如传递的vbounds和scale两个参数，懒得改了
但是在运行过程中有几个问题没有解决：
一、例如xy平面内有一个圆形，这个圆形在xz平面内会投影成一条直线，这条直线虽然在视图中看起来是一条直线，遍历的时候solidworks api会把它识别成一个圆，会导致多一些没有意义的标注尺寸，如下图：
二、代码连接solidworks2016sp5时运行很快，符合预期，但是在solidworks2018sp5和solidworks2019sp4这两个版本中运行相当慢，相比solidworks2016sp5慢了十倍以上（我暂时只用到这三个版本的solidworks），这就导致我无法正常使用这段代码，因为手动标注比自动标注更快。
这个思路是我在工作中一点点摸索出来的，前前后后搞了一年多才到这个程度，有路过的大佬一起讨论一下，指点一下这两个问题如何解决，感激
1496
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
