'''
待完善。
此代碼實現(xiàn)了，根據(jù)標注文本的
屬性，數(shù)值，位置，及 容差，
去判斷 設計 和 實測兩圖中的同一位置的尺寸。
如果是同一位置的尺寸，則進行比較，
并把結果存成表格，到運行此代碼的當前目錄。

此代碼運行時，要讀取的 dwg文件 必須處于打開狀態(tài)。
且 不能在 移動(pan) 模式。

啟動代碼：
python dwg_measurements_comparison4.py [8]

其中，8代表，判定兩圖尺寸為同一尺寸的最大距離，
單位：米（圖上單位）。自己決定具體是多少。

注意：
啟動此代碼后，首先要在cad軟件中打開 對比圖，
當該圖讀完后，提示切換到實測圖時，請在cad軟件中切換。
切換完成后，回車，即可。

包的安裝：
pip install pyautocad
注：
1.該操作會自動安裝 comtypes模塊。
2.如要使用tables 命令，要另外安裝xlrd 和 tablib

'''
from pyautocad import Autocad
import sys
from pyautocad.contrib.tables import Table
import re

acad = Autocad(create_if_not_exists=True)
def getDescription_Measurement_TextPositions():
 '''
 此函數(shù)用于讀取 實測圖 的尺寸標注的 屬性，尺寸，尺寸位置。
 并返回結果。
 目前實測圖帶屬性，對比圖不帶。
 '''
 print('正在讀取 ', acad.doc.Name, ' ...')
 description_measurement_textPositions = []
 for obj in acad.iter_objects('Dimension'):
  description_measurement_textPositions.append(
   (obj.GetXData("MyDimDist")[1][1],
   round(obj.Measurement,2), 
   obj.TextPosition)
  )
 return description_measurement_textPositions

def getMeasurement_TextPositions():
 '''
 此函數(shù)用于讀取 對比圖 的尺寸，尺寸位置。
 并返回結果。
 '''
 print('正在讀取 ', acad.doc.Name, ' ...')
 measurements_textPositions = []
 for a in acad.iter_objects('Dimension'):
  measurements_textPositions.append((round(a.Measurement,2), a.TextPosition))  
 return measurements_textPositions

def isTheSameMeasurement(point1,point2,tolerance):
 '''
 point1, 類似這樣(82.37, (81953.97462829649, 276686.2885731713, 0.0))，
 82.37，代表標注的尺寸，后邊代表，該尺寸在圖上顯示的位置坐標。
 point2, 類似這樣('車間二;長', 82.44, (81951.56923143109, 276679.7827104012, 0.0))
 
 此函數(shù)通過 兩個標注的距離來判斷，
 兩個尺寸，是否是同一位置處的尺寸。
 是，return True
 否，return False

 tolerance，設計/實測圖的同一位置兩個尺寸標注允許的距離差。
 即，在這個距離差之內(nèi)，認為是同一個對象的尺寸，可以進行比對。
 ''' 
 p1x = point1[1][0]
 p1y = point1[1][1]
 p2x = point2[2][0]
 p2y = point2[2][1]
 d = ((p1x - p2x) ** 2 + (p1y - p2y) ** 2) ** 0.5
 if d < tolerance:
  return True
 else:
  return False

def handleData(lst):
 '''
 此函數(shù)用于處理讀取到的原始數(shù)據(jù)，
 把原始數(shù)據(jù)分成三類：
 長，寬，間距 三個列表如下：
 lengthLst，widthLst，distanceLst，
 并返回。
 '''
 lengthLst = []
 widthLst = []
 distanceLst = []
 for i in lst:
  key = i[0].split(';')[1]
  if key == '間距':
   distanceLst.append(i)
  elif key == '長':
   lengthLst.append(i)
  elif key == '寬':
   widthLst.append(i)
 return lengthLst,widthLst,distanceLst

def handleLengthWidth(lengthLst,widthLst):
 '''
 此函數(shù)用于處理長度列表和寬度列表，
 組合成一個列表，即報告中需要的數(shù)據(jù)結構。

 其中，連廊只有寬度，需單獨處理。
 '''
 tableContents = []
 tableName = '竣工建（構）筑物滿外尺寸對比表'
 tableHead = ['\\', '發(fā)證長度', '實測長度', '長度差值(允許誤差值)',
  '發(fā)證寬度', '實測寬度', '寬度差值(允許誤差值)'  
 ]
 tableContents.append(tableHead)
 # 處理連廊尺寸。
 for w in widthLst:
  keyW = w[0].split(';')[0]
  if re.match('連廊', keyW):
   w2 = [keyW]
   w2.extend(['---','---','---'])
   w2.extend(w[1:])
   tableContents.append(w2)
 # 處理同時有長寬的尺寸。
 for l in lengthLst:
  keyL = l[0].split(';')[0]
  for w in widthLst:
   keyW = w[0].split(';')[0]   
   if keyL == keyW:
    w2 = w[1:]
    l[0] = l[0].split(';')[0]
    l.extend(w2)
    tableContents.append(l)
    break
 tableContents.sort()
 return tableName,tableContents

def handleDistance(distanceLst):
 '''
 此函數(shù)用于處理建筑物 間距尺寸。 
 '''
 tableContents = []
 tableName = '竣工建（構）筑物間距對比表'
 tableHead = [
  '\\', '發(fā)證間距', '實測間距', 
  '差值', '允許誤差值'
 ]
 tableContents.append(tableHead)
 for dl in distanceLst:
  dl[0] = dl[0][:-3]
  new = dl[-1][6:-1]
  dl[-1] = dl[-1][:5]
  dl.append(new)
  tableContents.append(dl)
 tableContents.sort()
 return tableName,tableContents

def handleDJ():
 '''
 此函數(shù)用于處理地界特征點。
 '''
 tableContents = []
 tableName = '用地界址坐標表'
 tableHead = [
  '點號', 'X坐標(米)', 'Y坐標(米)'
 ]
 tableContents.append(tableHead)
 area = 0
 for obj in acad.iter_objects("PolyLine"):
  if obj.Layer == 'DJHX':
   area = '%.1f' % obj.Area
   t = obj.Coordinates
   if 0 in t:
    DJHX = [('%.3f' % t[i], '%.3f' % t[i+1], t[i+2]) for i in range(0,len(t),3)]
   else:
    DJHX = [('%.3f' % t[i], '%.3f' % t[i+1], 0) for i in range(0,len(t),2)]
   break
 for i in range(len(DJHX)):
  tableContents.append(['J' + str(i + 1), DJHX[i][1], DJHX[i][0]])
 tableContents.append(['用地面積', area, '平方米'])
 return tableName,tableContents 

def write_to_table(tableName,tableContents):
 table = Table()
 for row in tableContents:
  table.writerow(row)
 table.save(tableName + '.xls', 'xls')

def main(tolerance):
 i = input('請在CAD軟件中打開 對比圖。打開了嗎？[Y]')
 if i == '' or i.upper() == 'Y':
  # dmt1, 第一次讀取的尺寸, 設計尺寸。
  dmt1 = getMeasurement_TextPositions()
  print('此圖有效尺寸數(shù)：', len(dmt1), ' 個')
  # print(dmt1)
  print()
 i = input('請在CAD軟件中切換到 實測圖。切換了嗎？[Y]')
 if i == '' or i.upper() == 'Y':
  # dmt2, 第二次讀取的尺寸，實測尺寸。
  dmt2 = getDescription_Measurement_TextPositions()
  print('此圖有效尺寸數(shù)：', len(dmt2), ' 個')
  # print(dmt2)
  print()
 # print(dmt1,dmt2,sep='\n\n')
 # d_value, 存儲兩個同位置尺寸的 屬性，設計尺寸，實測尺寸，及差值。
 d_value = []
 for m1 in dmt1:
  for m2 in dmt2:
   if isTheSameMeasurement(m1,m2,tolerance):
    d_value.append([
     m2[0],'%.2f' % m1[0],'%.2f' % m2[1], 
     '%.2f' % (m2[1] - m1[0]) + '(±' + '%.2f' % (m1[0] * 0.005) + ')'
    ])
    break
 print('比對兩圖尺寸數(shù):', len(d_value), ' 個')
 print('兩圖中判定為同一尺寸的容差：', tolerance, ' 米（圖上單位)。') 

 lengthLst,widthLst,distanceLst = handleData(d_value)
 # print(lengthLst,widthLst,distanceLst,sep='\n')
 tableName,tableContents = handleLengthWidth(lengthLst,widthLst)
 print()
 print(tableName)
 write_to_table(tableName,tableContents)

 tableName,tableContents = handleDistance(distanceLst)
 print(tableName)
 write_to_table(tableName,tableContents)

 tableName,tableContents = handleDJ()
 print(tableName)
 write_to_table(tableName,tableContents)

if __name__ == '__main__':
 try:
  tolerance = sys.argv[1]
  main(int(tolerance))
 except IndexError:
  # 默認兩圖尺寸相差2米內(nèi)算同一個尺寸。
  main(2)

from pyautocad import Autocad,APoint
 
p1 = APoint(10,20)
p2 = APoint(10,80)
 
acad = Autocad(create_if_not_exists = True)
acad.model.AddLine(p1,p2)