import win32com.client
import pythoncom
import tkinter as tk
from tkinter import messagebox
import numpy
import math
'''打开选择文件夹对话框'''
window = tk.Tk()
window.title("权属面积批量修改0.1")
window.geometry('380x30+800+200') # 290 160为窗口大小，+1000 +10 定义窗口弹出时的默认展示位置
window.resizable(0, 0)
window['background'] = 'DimGray'
def vtpnt(x, y, z=0):
"""坐标点转化为浮点数"""
return win32com.client.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_R8, (x, y, z))
def vtobj(obj):
"""转化为对象数组"""
return win32com.client.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_DISPATCH, obj)
def vtFloat(list):
"""列表转化为浮点数"""
return win32com.client.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_R8, list)
def vtInt(list):
"""列表转化为整数"""
return win32com.client.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_I2, list)
def vtVariant(list):
"""列表转化为变体"""
return win32com.client.VARIANT(pythoncom.VT_ARRAY | pythoncom.VT_VARIANT, list)
def list_format_conversion(old_list, leixing=False, step=2, deduction=None):
if leixing is False:
leixing = list
else:
leixing = tuple
new_list = []
for counte in range(0, len(old_list), step):
new_list.append(leixing(old_list[counte:counte + step])[:deduction])
return new_list
'''
进位，返回str
'''
def rrd(point, f):
if f == 0:
point = str(int(point))
elif f > 0:
SP = str(numpy.around(point, f))
sy = SP.split(".")
n = len(sy[1])
if n < f:
sj = f - n
point = sy[0] + "." + sy[1] + "0" * sj
else:
point = SP
elif f < 0:
tk.messagebox.showinfo('提示', "取位不可为负")
return point
'''
计算顺时针或逆时针
green方法判断面积正负
'''
def polyline_order(P):
n = len(P)
P.append(P[0])
S = 0
for i in range(0, n):
S = S + (P[i][0] + P[i + 1][0]) * (P[i + 1][1] - P[i][1])
if S > 0:
return False # 逆时针
else:
return True # 顺时针
def compute_polygon_area(points):
point_num = len(points)
if (point_num < 3): return 0.0
s = points[0][1] * (points[point_num - 1][0] - points[1][0])
# for i in range(point_num): # (int i = 1 i < point_num ++i):
for i in range(1, point_num): # 有小伙伴发现一个bug，这里做了修改，但是没有测试，需要使用的亲请测试下，以免结果不正确。
s += points[i][1] * (points[i - 1][0] - points[(i + 1) % point_num][0])
return numpy.around(abs(s / 2.0), 4)
'''
计算方位角
'''
def cross_point(line1, line2): # 计算交点函数
x1 = line1[0] # 取四点坐标
y1 = line1[1]
x2 = line1[2]
y2 = line1[3]
x3 = line2[0]
y3 = line2[1]
x4 = line2[2]
y4 = line2[3]
k1 = (y2 - y1) * 1.0 / (x2 - x1) # 计算k1,由于点均为整数，需要进行浮点数转化
b1 = y1 * 1.0 - x1 * k1 * 1.0 # 整型转浮点型是关键
if (x4 - x3) == 0: # L2直线斜率不存在操作
k2 = None
b2 = 0
else:
k2 = (y4 - y3) * 1.0 / (x4 - x3) # 斜率存在操作
b2 = y3 * 1.0 - x3 * k2 * 1.0
if k2 == None:
x = x3
else:
x = (b2 - b1) * 1.0 / (k1 - k2)
y = k1 * x * 1.0 + b1 * 1.0
return [x, y]
def azimuth(x1, y1, x2, y2):
angle = 0
dx = x2 - x1
dy = y2 - y1
if dx == 0 and dy > 0: # 判断象限
angle = 0
if dx == 0 and dy < 0:
angle = 180
if dy == 0 and dx > 0:
angle = 90
if dy == 0 and dx < 0:
angle = 270
if dx > 0 and dy > 0:
angle = math.atan(dx / dy) * 180 / math.pi
elif dx < 0 and dy > 0:
angle = 360 + math.atan(dx / dy) * 180 / math.pi
elif dx < 0 and dy < 0:
angle = 180 + math.atan(dx / dy) * 180 / math.pi
elif dx > 0 and dy < 0:
angle = 180 + math.atan(dx / dy) * 180 / math.pi
angle = 90 - angle
angle = math.radians(angle)
return angle # j
'''
计算西北角
'''
def distance(point, poly):
d = {}
min_d = []
azimuth_2 = {}
for Endpoint in poly:
distance_list = math.sqrt((Endpoint[0] - point[0]) ** 2 + (Endpoint[1] - point[1]) ** 2)
d.update({tuple(Endpoint): distance_list})
for key, val in d.items():
if val == min(d.values()):
min_d.append(key)
if len(min_d) > 1:
for x, y in min_d:
angle = azimuth(x, y, point[0], point[1])
azimuth_2.update({(x, y): angle})
min_point = max(azimuth_2, key=azimuth_2.get)
return min_point
elif len(min_d) == 1:
min_point = min_d[0]
return min_point
'''
重画西北角
'''
def Select_element():
global acad, doc, mp, original_area
acad = win32com.client.Dispatch("AutoCAD.Application")
doc = acad.ActiveDocument
doc.Utility.Prompt("\n醉后不知天在水\n满船清梦压星河\n")
mp = doc.ModelSpace # 模型空间
tk.messagebox.showinfo('提示', "请在屏幕拾取图元，以Enter键结束")
try:
doc.SelectionSets.Item("SS1").Delete()
except:
tk.messagebox.showinfo('警告', "Delete selection failed")
slt = doc.SelectionSets.Add("SS1")
slt.SelectOnScreen()
mmm = float(text1_var.get())
for x in slt:
if x.ObjectName == "AcDbPolyline":
original_area = x.Area # 面积
old_coord = list_format_conversion(x.Coordinates, leixing=True)
new_coord = tuple({}.fromkeys(old_coord).keys()) # 剔除重复点
list4 = []
for ier in old_coord:
if ier not in list4:
list4.append(ier)
else:
center, radius = vtpnt(ier[0], ier[1], 0), 2
cenobj = acad.ActiveDocument.Layers.Add("01重复点")
acad.ActiveDocument.ActiveLayer = cenobj
ClrNum = 2
cenobj.color = ClrNum
mp.AddCircle(center, radius)
Polyline_coordinates = [list(x) for x in new_coord] # 改变坐标格式【【1，2】，【4，3】，【6，7】】
if not polyline_order(Polyline_coordinates): # 若逆时针，反转列表
Polyline_coordinates.reverse()
poly_x = [x[0] for x in Polyline_coordinates]
poly_y = [y[1] for y in Polyline_coordinates]
max_y = max(poly_y)
min_x = min(poly_x)
northwest_point = [min_x, max_y] # 矩形框西北角
northwest_point_end_point_distance = distance(northwest_point, Polyline_coordinates) # 图斑最靠近图框西北角的坐标
count = Polyline_coordinates.index(list(northwest_point_end_point_distance)) # 查找在列表中的位置
Polyline_coordinates = Polyline_coordinates[:-1]
for _ in range(0, count):
delete = Polyline_coordinates.pop(0)
Polyline_coordinates.append(delete)
XY1 = Polyline_coordinates[0]
XY2 = Polyline_coordinates[1]
XY3 = Polyline_coordinates[2]
XY4 = Polyline_coordinates[3]
angle = azimuth(XY4[0], XY4[1], XY1[0], XY1[1]) # 角度
angle2 = azimuth(XY3[0], XY3[1], XY2[0], XY2[1]) # 角度
angle3 = azimuth(XY4[0], XY4[1], XY3[0], XY3[1]) # 角度
angle4 = math.radians(180) + (angle3 - angle) - math.radians(90)
area = mmm - original_area # 面积差值
d1 = math.sqrt((XY3[0] - XY4[0]) ** 2 + (XY3[1] - XY4[1]) ** 2) # 1，2 点距离
d5 = math.sqrt((XY2[0] - XY1[0]) ** 2 + (XY2[1] - XY1[1]) ** 2) # 1，2 点距离
d2 = (area * 2) / (d1 + d5) # 延长距离
d3 = d2 / math.cos(angle4)
xMargin1 = XY4[0] - math.cos(angle) * d3
yMargin1 = XY4[1] - math.sin(angle) * d3
xMargin2 = xMargin1 + math.cos(angle3) * d1
yMargin2 = yMargin1 + math.sin(angle3) * d1
line1 = [xMargin1, yMargin1, xMargin2, yMargin2]
line2 = XY2 + XY3
dSanDian = cross_point(line1, line2)
Polyline_coordinates[3] = [xMargin1, yMargin1]
Polyline_coordinates[2] = [dSanDian[0], dSanDian[1]]
ff = compute_polygon_area(Polyline_coordinates) - mmm
if ff > 0.0:
while True:
d3 = d3 - 0.000001
xMargin3 = XY4[0] - math.cos(angle) * (d3)
yMargin3 = XY4[1] - math.sin(angle) * (d3)
xMargin4 = xMargin3 + math.cos(angle3) * d1
yMargin4 = yMargin3 + math.sin(angle3) * d1
line1 = [xMargin3, yMargin3, xMargin4, yMargin4]
line2 = XY2 + XY3
dSanDian = cross_point(line1, line2)
Polyline_coordinates[3] = [xMargin3, yMargin3]
Polyline_coordinates[2] = [dSanDian[0], dSanDian[1]]
sd = compute_polygon_area(Polyline_coordinates)
if (mmm - 0.00004) <= sd <= (mmm + 0.00004):
break
elif ff < 0.0:
while True:
d3 = d3 + 0.000001
xMargin3 = XY4[0] - math.cos(angle) * d3
yMargin3 = XY4[1] - math.sin(angle) * d3
xMargin4 = xMargin3 + math.cos(angle3) * d1
yMargin4 = yMargin3 + math.sin(angle3) * d1
line1 = [xMargin3, yMargin3, xMargin4, yMargin4]
line2 = XY2 + XY3
dSanDian = cross_point(line1, line2)
Polyline_coordinates[3] = [xMargin3, yMargin3]
Polyline_coordinates[2] = [dSanDian[0], dSanDian[1]]
v = compute_polygon_area(Polyline_coordinates)
if (mmm - 0.00004) <= v <= (mmm + 0.00004):
break
else:
pass
Polyline_coordinates = numpy.array([m for n in Polyline_coordinates for m in n], dtype=numpy.float)
points = vtFloat(Polyline_coordinates)
# 设定图层
layerobj = acad.ActiveDocument.Layers.Add("02调整后面积图层")
acad.ActiveDocument.ActiveLayer = layerobj
# 多段线颜色
ClrNum = 1
layerobj.color = ClrNum
polylineobj = mp.AddLightweightPolyline(points)
polylineobj.Closed = True
messagebox.showinfo("tip","完成。")
lable1 = tk.Label(window, text="[面积替换:平方米]", width=15).grid(row=0, column=0)
text1_var = tk.StringVar() # 获取text_1输入的值
text1_var.set(r'400')
text1 = tk.Entry(window, textvariable=text1_var, bd=5).grid(row=0, column=1)
tk.Button(window, text="开始", width=15, height=1, command=Select_element, bg="Silver").grid(row=0, column=2)
window.mainloop()