python对 CAD图斑面积的统计及标注

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
import win32com.client
import pythoncom
import tkinter as tk
from tkinter import filedialog, messagebox
import numpy
from tkinter import ttk

'''打开选择文件夹对话框'''
window = tk.Tk()
window.title("CAD标注及面积统计  by: 放放风")
window.geometry('720x300+800+200')  # 290 160为窗口大小，+1000 +10 定义窗口弹出时的默认展示位置

window.resizable( 0, 0 )
window['background'] = 'DimGray'

r_value = tk.IntVar()

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, step=2, deduction=None):
    new_list = []
    for counte in range(0, len(old_list), step):
        new_list.append(list(old_list[counte:counte + step])[:deduction])
    return new_list


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


class Extract:

    def __init__(self):
        self.face_properties = []
        self.r = None

    def r_print(self):  # 获取单选框值
        self.r = r_value.get()

    def Select_element(self):
        acad = win32com.client.Dispatch("AutoCAD.Application")
        doc = acad.ActiveDocument
        doc.Utility.Prompt("n醉后不知天在水n满船清梦压星河n")
        mp = doc.ModelSpace  # 模型空间
        tk.messagebox.showinfo('提示', "请在屏幕拾取图元，以Enter键结束")
        polyline_name = text1_var.get()
        polyline_number = text2_var.get()
        polyline_mm_round = text3_var.get()  # 平方米保留位数
        polyline_mu_round = text4_var.get()  # 亩保留位数
        font_size = text5_var.get()
        font_height = text6_var.get()

        mm_Company = text7_var.get()
        Mu_Company = text8_var.get()
        while True:
            try:
                doc.SelectionSets.Item("SS1").Delete()
            except:
                tk.messagebox.showinfo('警告', "Delete selection failed")
            slt = doc.SelectionSets.Add("SS1")
            slt.SelectOnScreen()
            Text_point = doc.Utility.GetPoint()  # 获取屏幕指定坐标
            Text_point_number = vtpnt(Text_point[0], Text_point[1])  # 转为cad坐标格式,编号坐标
            Text_point_area = vtpnt(Text_point[0], Text_point[1] - font_height)  # 平方米/亩标注坐标

            for x in slt:
                if x.ObjectName == "AcDbPolyline":

                    Polygon_area = x.Area  # 多边形面积
                    Polygon_Mu_area = Polygon_area / 666.6666666667
                    Polygon_area = str(rrd(Polygon_area, polyline_mm_round)) + mm_Company  #
                    Polygon_Mu_area = str(rrd(Polygon_Mu_area, polyline_mu_round)) + Mu_Company  #

                    polyline_text_number = polyline_name + str(polyline_number)  # 前缀+编号
                    tree.insert("", "end", text=polyline_number,
                                values=(polyline_text_number, Polygon_area, Polygon_Mu_area))
                    self.face_properties.append(
                        [polyline_text_number, "t", Polygon_area, "t", Polygon_Mu_area, "n"])  # {编号：多边形平方米面积}

                    mp.AddText(polyline_text_number, Text_point_number, font_size)  # 添加编号注记

                    if self.r == 2:
                        mp.AddText(Polygon_area, Text_point_area, font_size)  # 添加平方米注记
                    elif self.r == 3:
                        mp.AddText(Polygon_Mu_area, Text_point_area, font_size)  # 添加亩注记
                    polyline_number += 1

    def save_text(self):
        Save_path = filedialog.askdirectory()
        new_filer = open(Save_path + "/面积汇总.txt", "w")
        new_lines = ["地块编号", "t", "平方米", "t", "亩", "n"]
        new_filer.writelines(new_lines)
        for items in self.face_properties:
            new_filer.writelines(items)
        new_filer.close()
        messagebox.showinfo("提示", "成功")


objectA = Extract()
tk.Button(window, text="拾取及标注", width=15, height=1, command=objectA.Select_element, bg="Silver").grid(row=8, column=3)
tk.Button(window, text="导出文本", width=15, height=1, command=objectA.save_text, bg="Silver").grid(row=8, column=4)

tree = ttk.Treeview(window, height=12, show="headings")
tree.grid(row=0, column=3, rowspan=8, columnspan=2)
tree["columns"] = ("Num", "Area(m2)", "Area(Mu)")

tree.column("Num", width=150)
tree.column("Area(m2)", width=150)
tree.column("Area(Mu)", width=150)

tree.heading("Num", text="编号")
tree.heading("Area(m2)", text="面积（m²）")
tree.heading("Area(Mu)", text="面积（Mu）")

lable0 = tk.Label(window, text="[  文本属性  ]", width=38).grid(row=0, column=0, columnspan=2)

lable1 = tk.Label(window, text="[  编号前缀  ]", width=15).grid(row=1, column=0)

text1_var = tk.StringVar()  # 获取text_1输入的值
text1_var.set("地块")
text1 = tk.Entry(window, textvariable=text1_var, bd=5).grid(row=1, column=1)

lable2 = tk.Label(window, text="[编号起始位置]", width=15).grid(row=2, column=0)
text2_var = tk.IntVar()  # 获取text_2输入的值
text2_var.set("1")
text2 = tk.Entry(window, textvariable=text2_var, bd=5).grid(row=2, column=1)

tk.Radiobutton(window, text='[平方米]', variable=r_value, value=2, command=objectA.r_print, width=12).grid(row=3, column=0)
lable3 = tk.Label(window, text="[保留位数]", width=15).grid(row=4, column=0)
text3_var = tk.IntVar()  # 获取text_3输入的值
text3_var.set("2")
text3 = tk.Entry(window, textvariable=text3_var, bd=5).grid(row=4, column=1)

text7_var = tk.StringVar()  # 获取text_3输入的值
text7_var.set(r"平方米")
text7 = tk.Entry(window, textvariable=text7_var, bd=5).grid(row=3, column=1)

tk.Radiobutton(window, text='[   亩   ]', variable=r_value, value=3, command=objectA.r_print, width=12).grid(row=5,
                                                                                                            column=0)
lable4 = tk.Label(window, text="[保留位数]", width=15).grid(row=6, column=0)
text4_var = tk.IntVar()  # 获取text_3输入的值
text4_var.set("3")
text4 = tk.Entry(window, textvariable=text4_var, bd=5).grid(row=6, column=1)

text8_var = tk.StringVar()  # 获取text_3输入的值
text8_var.set(r"亩")
text8 = tk.Entry(window, textvariable=text8_var, bd=5).grid(row=5, column=1)

lable5 = tk.Label(window, text="[注记大小]", width=15).grid(row=7, column=0)
text5_var = tk.IntVar()  # 获取text_3输入的值
text5_var.set("1.5")
text5 = tk.Entry(window, textvariable=text5_var, bd=5).grid(row=7, column=1)

lable6 = tk.Label(window, text="[注记间隔]", width=15).grid(row=8, column=0)
text6_var = tk.IntVar()  # 获取text_3输入的值
text6_var.set("2.5")
text6 = tk.Entry(window, textvariable=text6_var, bd=5).grid(row=8, column=1)

window.mainloop()