ToolCalibration.py 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # File Author: Marius Adrian Stanciu (c) #
  4. # Date: 3/10/2019 #
  5. # MIT Licence #
  6. # ##########################################################
  7. from PyQt5 import QtWidgets, QtCore, QtGui
  8. from FlatCAMTool import FlatCAMTool
  9. from flatcamGUI.GUIElements import FCDoubleSpinner, EvalEntry, FCCheckBox, OptionalInputSection
  10. from flatcamGUI.GUIElements import FCTable, FCComboBox, RadioSet
  11. from flatcamEditors.FlatCAMTextEditor import TextEditor
  12. from shapely.geometry import Point
  13. from shapely.geometry.base import *
  14. from shapely.affinity import scale, skew
  15. import math
  16. from datetime import datetime
  17. import logging
  18. from copy import deepcopy
  19. import gettext
  20. import FlatCAMTranslation as fcTranslate
  21. import builtins
  22. fcTranslate.apply_language('strings')
  23. if '_' not in builtins.__dict__:
  24. _ = gettext.gettext
  25. log = logging.getLogger('base')
  26. class ToolCalibration(FlatCAMTool):
  27. toolName = _("Calibration Tool")
  28. def __init__(self, app):
  29. FlatCAMTool.__init__(self, app)
  30. self.app = app
  31. self.canvas = self.app.plotcanvas
  32. self.decimals = self.app.decimals
  33. # ## Title
  34. title_label = QtWidgets.QLabel("%s" % self.toolName)
  35. title_label.setStyleSheet("""
  36. QLabel
  37. {
  38. font-size: 16px;
  39. font-weight: bold;
  40. }
  41. """)
  42. self.layout.addWidget(title_label)
  43. self.layout.addWidget(QtWidgets.QLabel(''))
  44. # ## Grid Layout
  45. grid_lay = QtWidgets.QGridLayout()
  46. self.layout.addLayout(grid_lay)
  47. grid_lay.setColumnStretch(0, 0)
  48. grid_lay.setColumnStretch(1, 1)
  49. grid_lay.setColumnStretch(2, 0)
  50. step_1 = QtWidgets.QLabel('<b>%s</b>' % _("STEP 1: Acquire Calibration Points"))
  51. step_1.setToolTip(
  52. _("Pick four points by clicking inside the drill holes.\n"
  53. "Those four points should be in the four\n"
  54. "(as much as possible) corners of the Excellon object.")
  55. )
  56. grid_lay.addWidget(step_1, 0, 0, 1, 3)
  57. self.cal_source_lbl = QtWidgets.QLabel("<b>%s:</b>" % _("Source Type"))
  58. self.cal_source_lbl.setToolTip(_("The source of calibration points.\n"
  59. "It can be:\n"
  60. "- Object -> click a hole geo for Excellon or a pad for Gerber\n"
  61. "- Free -> click freely on canvas to acquire the calibration points"))
  62. self.cal_source_radio = RadioSet([{'label': _('Object'), 'value': 'object'},
  63. {'label': _('Free'), 'value': 'free'}],
  64. stretch=False)
  65. grid_lay.addWidget(self.cal_source_lbl, 1, 0)
  66. grid_lay.addWidget(self.cal_source_radio, 1, 1, 1, 2)
  67. self.obj_type_label = QtWidgets.QLabel("%s:" % _("Object Type"))
  68. self.obj_type_combo = FCComboBox()
  69. self.obj_type_combo.addItem(_("Gerber"))
  70. self.obj_type_combo.addItem(_("Excellon"))
  71. self.obj_type_combo.setCurrentIndex(1)
  72. grid_lay.addWidget(self.obj_type_label, 2, 0)
  73. grid_lay.addWidget(self.obj_type_combo, 2, 1, 1, 2)
  74. self.object_combo = FCComboBox()
  75. self.object_combo.setModel(self.app.collection)
  76. self.object_combo.setRootModelIndex(self.app.collection.index(1, 0, QtCore.QModelIndex()))
  77. self.object_combo.setCurrentIndex(1)
  78. self.object_label = QtWidgets.QLabel("%s:" % _("Source object selection"))
  79. self.object_label.setToolTip(
  80. _("FlatCAM Object to be used as a source for reference points.")
  81. )
  82. grid_lay.addWidget(self.object_label, 3, 0, 1, 3)
  83. grid_lay.addWidget(self.object_combo, 4, 0, 1, 3)
  84. self.points_table_label = QtWidgets.QLabel('<b>%s</b>' % _('Calibration Points'))
  85. self.points_table_label.setToolTip(
  86. _("Contain the expected calibration points and the\n"
  87. "ones measured.")
  88. )
  89. grid_lay.addWidget(self.points_table_label, 5, 0, 1, 3)
  90. self.points_table = FCTable()
  91. self.points_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  92. # self.points_table.setSizeAdjustPolicy(QtWidgets.QAbstractScrollArea.AdjustToContents)
  93. grid_lay.addWidget(self.points_table, 6, 0, 1, 3)
  94. self.points_table.setColumnCount(4)
  95. self.points_table.setHorizontalHeaderLabels(
  96. [
  97. '#',
  98. _("Name"),
  99. _("Target"),
  100. _("Found Delta")
  101. ]
  102. )
  103. self.points_table.setRowCount(8)
  104. row = 0
  105. # BOTTOM LEFT
  106. id_item_1 = QtWidgets.QTableWidgetItem('%d' % 1)
  107. flags = QtCore.Qt.ItemIsEnabled
  108. id_item_1.setFlags(flags)
  109. self.points_table.setItem(row, 0, id_item_1) # Tool name/id
  110. self.bottom_left_coordx_lbl = QtWidgets.QLabel('%s' % _('Bot Left X'))
  111. self.points_table.setCellWidget(row, 1, self.bottom_left_coordx_lbl)
  112. self.bottom_left_coordx_tgt = EvalEntry()
  113. self.points_table.setCellWidget(row, 2, self.bottom_left_coordx_tgt)
  114. self.bottom_left_coordx_tgt.setReadOnly(True)
  115. self.bottom_left_coordx_found = EvalEntry()
  116. self.points_table.setCellWidget(row, 3, self.bottom_left_coordx_found)
  117. row += 1
  118. self.bottom_left_coordy_lbl = QtWidgets.QLabel('%s' % _('Bot Left Y'))
  119. self.points_table.setCellWidget(row, 1, self.bottom_left_coordy_lbl)
  120. self.bottom_left_coordy_tgt = EvalEntry()
  121. self.points_table.setCellWidget(row, 2, self.bottom_left_coordy_tgt)
  122. self.bottom_left_coordy_tgt.setReadOnly(True)
  123. self.bottom_left_coordy_found = EvalEntry()
  124. self.points_table.setCellWidget(row, 3, self.bottom_left_coordy_found)
  125. self.bottom_left_coordx_found.set_value(_("Origin"))
  126. self.bottom_left_coordy_found.set_value(_("Origin"))
  127. self.bottom_left_coordx_found.setDisabled(True)
  128. self.bottom_left_coordy_found.setDisabled(True)
  129. row += 1
  130. # BOTTOM RIGHT
  131. id_item_2 = QtWidgets.QTableWidgetItem('%d' % 2)
  132. flags = QtCore.Qt.ItemIsEnabled
  133. id_item_2.setFlags(flags)
  134. self.points_table.setItem(row, 0, id_item_2) # Tool name/id
  135. self.bottom_right_coordx_lbl = QtWidgets.QLabel('%s' % _('Bot Right X'))
  136. self.points_table.setCellWidget(row, 1, self.bottom_right_coordx_lbl)
  137. self.bottom_right_coordx_tgt = EvalEntry()
  138. self.points_table.setCellWidget(row, 2, self.bottom_right_coordx_tgt)
  139. self.bottom_right_coordx_tgt.setReadOnly(True)
  140. self.bottom_right_coordx_found = EvalEntry()
  141. self.points_table.setCellWidget(row, 3, self.bottom_right_coordx_found)
  142. row += 1
  143. self.bottom_right_coordy_lbl = QtWidgets.QLabel('%s' % _('Bot Right Y'))
  144. self.points_table.setCellWidget(row, 1, self.bottom_right_coordy_lbl)
  145. self.bottom_right_coordy_tgt = EvalEntry()
  146. self.points_table.setCellWidget(row, 2, self.bottom_right_coordy_tgt)
  147. self.bottom_right_coordy_tgt.setReadOnly(True)
  148. self.bottom_right_coordy_found = EvalEntry()
  149. self.points_table.setCellWidget(row, 3, self.bottom_right_coordy_found)
  150. row += 1
  151. # TOP LEFT
  152. id_item_3 = QtWidgets.QTableWidgetItem('%d' % 3)
  153. flags = QtCore.Qt.ItemIsEnabled
  154. id_item_3.setFlags(flags)
  155. self.points_table.setItem(row, 0, id_item_3) # Tool name/id
  156. self.top_left_coordx_lbl = QtWidgets.QLabel('%s' % _('Top Left X'))
  157. self.points_table.setCellWidget(row, 1, self.top_left_coordx_lbl)
  158. self.top_left_coordx_tgt = EvalEntry()
  159. self.points_table.setCellWidget(row, 2, self.top_left_coordx_tgt)
  160. self.top_left_coordx_tgt.setReadOnly(True)
  161. self.top_left_coordx_found = EvalEntry()
  162. self.points_table.setCellWidget(row, 3, self.top_left_coordx_found)
  163. row += 1
  164. self.top_left_coordy_lbl = QtWidgets.QLabel('%s' % _('Top Left Y'))
  165. self.points_table.setCellWidget(row, 1, self.top_left_coordy_lbl)
  166. self.top_left_coordy_tgt = EvalEntry()
  167. self.points_table.setCellWidget(row, 2, self.top_left_coordy_tgt)
  168. self.top_left_coordy_tgt.setReadOnly(True)
  169. self.top_left_coordy_found = EvalEntry()
  170. self.points_table.setCellWidget(row, 3, self.top_left_coordy_found)
  171. row += 1
  172. # TOP RIGHT
  173. id_item_4 = QtWidgets.QTableWidgetItem('%d' % 4)
  174. flags = QtCore.Qt.ItemIsEnabled
  175. id_item_4.setFlags(flags)
  176. self.points_table.setItem(row, 0, id_item_4) # Tool name/id
  177. self.top_right_coordx_lbl = QtWidgets.QLabel('%s' % _('Top Right X'))
  178. self.points_table.setCellWidget(row, 1, self.top_right_coordx_lbl)
  179. self.top_right_coordx_tgt = EvalEntry()
  180. self.points_table.setCellWidget(row, 2, self.top_right_coordx_tgt)
  181. self.top_right_coordx_tgt.setReadOnly(True)
  182. self.top_right_coordx_found = EvalEntry()
  183. self.points_table.setCellWidget(row, 3, self.top_right_coordx_found)
  184. row += 1
  185. self.top_right_coordy_lbl = QtWidgets.QLabel('%s' % _('Top Right Y'))
  186. self.points_table.setCellWidget(row, 1, self.top_right_coordy_lbl)
  187. self.top_right_coordy_tgt = EvalEntry()
  188. self.points_table.setCellWidget(row, 2, self.top_right_coordy_tgt)
  189. self.top_right_coordy_tgt.setReadOnly(True)
  190. self.top_right_coordy_found = EvalEntry()
  191. self.points_table.setCellWidget(row, 3, self.top_right_coordy_found)
  192. vertical_header = self.points_table.verticalHeader()
  193. vertical_header.hide()
  194. self.points_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  195. horizontal_header = self.points_table.horizontalHeader()
  196. horizontal_header.setMinimumSectionSize(10)
  197. horizontal_header.setDefaultSectionSize(70)
  198. self.points_table.setSizeAdjustPolicy(QtWidgets.QAbstractScrollArea.AdjustToContents)
  199. # for x in range(4):
  200. # self.points_table.resizeColumnToContents(x)
  201. self.points_table.resizeColumnsToContents()
  202. self.points_table.resizeRowsToContents()
  203. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  204. horizontal_header.resizeSection(0, 20)
  205. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Fixed)
  206. horizontal_header.setSectionResizeMode(2, QtWidgets.QHeaderView.Stretch)
  207. horizontal_header.setSectionResizeMode(3, QtWidgets.QHeaderView.Stretch)
  208. self.points_table.setMinimumHeight(self.points_table.getHeight() + 2)
  209. self.points_table.setMaximumHeight(self.points_table.getHeight() + 3)
  210. # ## Get Points Button
  211. self.start_button = QtWidgets.QPushButton(_("Get Points"))
  212. self.start_button.setToolTip(
  213. _("Pick four points by clicking on canvas if the source choice\n"
  214. "is 'free' or inside the object geometry if the source is 'object'.\n"
  215. "Those four points should be in the four squares of\n"
  216. "the object.")
  217. )
  218. self.start_button.setStyleSheet("""
  219. QPushButton
  220. {
  221. font-weight: bold;
  222. }
  223. """)
  224. grid_lay.addWidget(self.start_button, 7, 0, 1, 3)
  225. separator_line = QtWidgets.QFrame()
  226. separator_line.setFrameShape(QtWidgets.QFrame.HLine)
  227. separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
  228. grid_lay.addWidget(separator_line, 8, 0, 1, 3)
  229. grid_lay.addWidget(QtWidgets.QLabel(''), 9, 0)
  230. # STEP 2 #
  231. step_2 = QtWidgets.QLabel('<b>%s</b>' % _("STEP 2: Verification GCode"))
  232. step_2.setToolTip(
  233. _("Generate GCode file to locate and align the PCB by using\n"
  234. "the four points acquired above.")
  235. )
  236. grid_lay.addWidget(step_2, 10, 0, 1, 3)
  237. self.gcode_title_label = QtWidgets.QLabel('<b>%s</b>' % _('GCode Parameters'))
  238. self.gcode_title_label.setToolTip(
  239. _("Parameters used when creating the GCode in this tool.")
  240. )
  241. grid_lay.addWidget(self.gcode_title_label, 11, 0, 1, 3)
  242. # Travel Z entry
  243. travelz_lbl = QtWidgets.QLabel('%s:' % _("Travel Z"))
  244. travelz_lbl.setToolTip(
  245. _("Height (Z) for travelling between the points.")
  246. )
  247. self.travelz_entry = FCDoubleSpinner()
  248. self.travelz_entry.set_range(-9999.9999, 9999.9999)
  249. self.travelz_entry.set_precision(self.decimals)
  250. self.travelz_entry.setSingleStep(0.1)
  251. grid_lay.addWidget(travelz_lbl, 12, 0)
  252. grid_lay.addWidget(self.travelz_entry, 12, 1, 1, 2)
  253. # Verification Z entry
  254. verz_lbl = QtWidgets.QLabel('%s:' % _("Verification Z"))
  255. verz_lbl.setToolTip(
  256. _("Height (Z) for checking the point.")
  257. )
  258. self.verz_entry = FCDoubleSpinner()
  259. self.verz_entry.set_range(-9999.9999, 9999.9999)
  260. self.verz_entry.set_precision(self.decimals)
  261. self.verz_entry.setSingleStep(0.1)
  262. grid_lay.addWidget(verz_lbl, 13, 0)
  263. grid_lay.addWidget(self.verz_entry, 13, 1, 1, 2)
  264. # Zero the Z of the verification tool
  265. self.zeroz_cb = FCCheckBox('%s' % _("Zero Z tool"))
  266. self.zeroz_cb.setToolTip(
  267. _("Include a sequence to zero the height (Z)\n"
  268. "of the verification tool.")
  269. )
  270. grid_lay.addWidget(self.zeroz_cb, 14, 0, 1, 3)
  271. # Toochange Z entry
  272. toolchangez_lbl = QtWidgets.QLabel('%s:' % _("Toolchange Z"))
  273. toolchangez_lbl.setToolTip(
  274. _("Height (Z) for mounting the verification probe.")
  275. )
  276. self.toolchangez_entry = FCDoubleSpinner()
  277. self.toolchangez_entry.set_range(0.0000, 9999.9999)
  278. self.toolchangez_entry.set_precision(self.decimals)
  279. self.toolchangez_entry.setSingleStep(0.1)
  280. grid_lay.addWidget(toolchangez_lbl, 15, 0)
  281. grid_lay.addWidget(self.toolchangez_entry, 15, 1, 1, 2)
  282. self.z_ois = OptionalInputSection(self.zeroz_cb, [toolchangez_lbl, self.toolchangez_entry])
  283. # ## GCode Button
  284. self.gcode_button = QtWidgets.QPushButton(_("Generate GCode"))
  285. self.gcode_button.setToolTip(
  286. _("Generate GCode file to locate and align the PCB by using\n"
  287. "the four points acquired above.")
  288. )
  289. self.gcode_button.setStyleSheet("""
  290. QPushButton
  291. {
  292. font-weight: bold;
  293. }
  294. """)
  295. grid_lay.addWidget(self.gcode_button, 16, 0, 1, 3)
  296. separator_line1 = QtWidgets.QFrame()
  297. separator_line1.setFrameShape(QtWidgets.QFrame.HLine)
  298. separator_line1.setFrameShadow(QtWidgets.QFrame.Sunken)
  299. grid_lay.addWidget(separator_line1, 17, 0, 1, 3)
  300. grid_lay.addWidget(QtWidgets.QLabel(''), 18, 0, 1, 3)
  301. # STEP 3 #
  302. step_3 = QtWidgets.QLabel('<b>%s</b>' % _("STEP 3: Adjustments"))
  303. step_3.setToolTip(
  304. _("Calculate Scale and Skew factors based on the differences (delta)\n"
  305. "found when checking the PCB pattern. The differences must be filled\n"
  306. "in the fields Found (Delta).")
  307. )
  308. grid_lay.addWidget(step_3, 19, 0, 1, 3)
  309. # ## Factors Button
  310. self.generate_factors_button = QtWidgets.QPushButton(_("Calculate Factors"))
  311. self.generate_factors_button.setToolTip(
  312. _("Calculate Scale and Skew factors based on the differences (delta)\n"
  313. "found when checking the PCB pattern. The differences must be filled\n"
  314. "in the fields Found (Delta).")
  315. )
  316. self.generate_factors_button.setStyleSheet("""
  317. QPushButton
  318. {
  319. font-weight: bold;
  320. }
  321. """)
  322. grid_lay.addWidget(self.generate_factors_button, 20, 0, 1, 3)
  323. separator_line1 = QtWidgets.QFrame()
  324. separator_line1.setFrameShape(QtWidgets.QFrame.HLine)
  325. separator_line1.setFrameShadow(QtWidgets.QFrame.Sunken)
  326. grid_lay.addWidget(separator_line1, 21, 0, 1, 3)
  327. grid_lay.addWidget(QtWidgets.QLabel(''), 22, 0, 1, 3)
  328. # STEP 4 #
  329. step_4 = QtWidgets.QLabel('<b>%s</b>' % _("STEP 4: Adjusted GCode"))
  330. step_4.setToolTip(
  331. _("Generate verification GCode file adjusted with\n"
  332. "the factors above.")
  333. )
  334. grid_lay.addWidget(step_4, 23, 0, 1, 3)
  335. self.scalex_label = QtWidgets.QLabel(_("Scale Factor X:"))
  336. self.scalex_label.setToolTip(
  337. _("Factor for Scale action over X axis.")
  338. )
  339. self.scalex_entry = FCDoubleSpinner()
  340. self.scalex_entry.set_range(0, 9999.9999)
  341. self.scalex_entry.set_precision(self.decimals)
  342. self.scalex_entry.setSingleStep(0.1)
  343. grid_lay.addWidget(self.scalex_label, 24, 0)
  344. grid_lay.addWidget(self.scalex_entry, 24, 1, 1, 2)
  345. self.scaley_label = QtWidgets.QLabel(_("Scale Factor Y:"))
  346. self.scaley_label.setToolTip(
  347. _("Factor for Scale action over Y axis.")
  348. )
  349. self.scaley_entry = FCDoubleSpinner()
  350. self.scaley_entry.set_range(0, 9999.9999)
  351. self.scaley_entry.set_precision(self.decimals)
  352. self.scaley_entry.setSingleStep(0.1)
  353. grid_lay.addWidget(self.scaley_label, 25, 0)
  354. grid_lay.addWidget(self.scaley_entry, 25, 1, 1, 2)
  355. self.scale_button = QtWidgets.QPushButton(_("Apply Scale Factors"))
  356. self.scale_button.setToolTip(
  357. _("Apply Scale factors on the calibration points.")
  358. )
  359. self.scale_button.setStyleSheet("""
  360. QPushButton
  361. {
  362. font-weight: bold;
  363. }
  364. """)
  365. grid_lay.addWidget(self.scale_button, 26, 0, 1, 3)
  366. self.skewx_label = QtWidgets.QLabel(_("Skew Angle X:"))
  367. self.skewx_label.setToolTip(
  368. _("Angle for Skew action, in degrees.\n"
  369. "Float number between -360 and 359.")
  370. )
  371. self.skewx_entry = FCDoubleSpinner()
  372. self.skewx_entry.set_range(-360, 360)
  373. self.skewx_entry.set_precision(self.decimals)
  374. self.skewx_entry.setSingleStep(0.1)
  375. grid_lay.addWidget(self.skewx_label, 27, 0)
  376. grid_lay.addWidget(self.skewx_entry, 27, 1, 1, 2)
  377. self.skewy_label = QtWidgets.QLabel(_("Skew Angle Y:"))
  378. self.skewy_label.setToolTip(
  379. _("Angle for Skew action, in degrees.\n"
  380. "Float number between -360 and 359.")
  381. )
  382. self.skewy_entry = FCDoubleSpinner()
  383. self.skewy_entry.set_range(-360, 360)
  384. self.skewy_entry.set_precision(self.decimals)
  385. self.skewy_entry.setSingleStep(0.1)
  386. grid_lay.addWidget(self.skewy_label, 28, 0)
  387. grid_lay.addWidget(self.skewy_entry, 28, 1, 1, 2)
  388. self.skew_button = QtWidgets.QPushButton(_("Apply Skew Factors"))
  389. self.skew_button.setToolTip(
  390. _("Apply Skew factors on the calibration points.")
  391. )
  392. self.skew_button.setStyleSheet("""
  393. QPushButton
  394. {
  395. font-weight: bold;
  396. }
  397. """)
  398. grid_lay.addWidget(self.skew_button, 29, 0, 1, 3)
  399. # final_factors_lbl = QtWidgets.QLabel('<b>%s</b>' % _("Final Factors"))
  400. # final_factors_lbl.setToolTip(
  401. # _("Generate verification GCode file adjusted with\n"
  402. # "the factors above.")
  403. # )
  404. # grid_lay.addWidget(final_factors_lbl, 27, 0, 1, 3)
  405. #
  406. # self.fin_scalex_label = QtWidgets.QLabel(_("Scale Factor X:"))
  407. # self.fin_scalex_label.setToolTip(
  408. # _("Final factor for Scale action over X axis.")
  409. # )
  410. # self.fin_scalex_entry = FCDoubleSpinner()
  411. # self.fin_scalex_entry.set_range(0, 9999.9999)
  412. # self.fin_scalex_entry.set_precision(self.decimals)
  413. # self.fin_scalex_entry.setSingleStep(0.1)
  414. #
  415. # grid_lay.addWidget(self.fin_scalex_label, 28, 0)
  416. # grid_lay.addWidget(self.fin_scalex_entry, 28, 1, 1, 2)
  417. #
  418. # self.fin_scaley_label = QtWidgets.QLabel(_("Scale Factor Y:"))
  419. # self.fin_scaley_label.setToolTip(
  420. # _("Final factor for Scale action over Y axis.")
  421. # )
  422. # self.fin_scaley_entry = FCDoubleSpinner()
  423. # self.fin_scaley_entry.set_range(0, 9999.9999)
  424. # self.fin_scaley_entry.set_precision(self.decimals)
  425. # self.fin_scaley_entry.setSingleStep(0.1)
  426. #
  427. # grid_lay.addWidget(self.fin_scaley_label, 29, 0)
  428. # grid_lay.addWidget(self.fin_scaley_entry, 29, 1, 1, 2)
  429. #
  430. # self.fin_skewx_label = QtWidgets.QLabel(_("Skew Angle X:"))
  431. # self.fin_skewx_label.setToolTip(
  432. # _("Final value for angle for Skew action, in degrees.\n"
  433. # "Float number between -360 and 359.")
  434. # )
  435. # self.fin_skewx_entry = FCDoubleSpinner()
  436. # self.fin_skewx_entry.set_range(-360, 360)
  437. # self.fin_skewx_entry.set_precision(self.decimals)
  438. # self.fin_skewx_entry.setSingleStep(0.1)
  439. #
  440. # grid_lay.addWidget(self.fin_skewx_label, 30, 0)
  441. # grid_lay.addWidget(self.fin_skewx_entry, 30, 1, 1, 2)
  442. #
  443. # self.fin_skewy_label = QtWidgets.QLabel(_("Skew Angle Y:"))
  444. # self.fin_skewy_label.setToolTip(
  445. # _("Final value for angle for Skew action, in degrees.\n"
  446. # "Float number between -360 and 359.")
  447. # )
  448. # self.fin_skewy_entry = FCDoubleSpinner()
  449. # self.fin_skewy_entry.set_range(-360, 360)
  450. # self.fin_skewy_entry.set_precision(self.decimals)
  451. # self.fin_skewy_entry.setSingleStep(0.1)
  452. #
  453. # grid_lay.addWidget(self.fin_skewy_label, 31, 0)
  454. # grid_lay.addWidget(self.fin_skewy_entry, 31, 1, 1, 2)
  455. # ## Adjusted GCode Button
  456. self.adj_gcode_button = QtWidgets.QPushButton(_("Generate Adjusted GCode"))
  457. self.adj_gcode_button.setToolTip(
  458. _("Generate verification GCode file adjusted with\n"
  459. "the factors above.")
  460. )
  461. self.adj_gcode_button.setStyleSheet("""
  462. QPushButton
  463. {
  464. font-weight: bold;
  465. }
  466. """)
  467. grid_lay.addWidget(self.adj_gcode_button, 35, 0, 1, 3)
  468. separator_line1 = QtWidgets.QFrame()
  469. separator_line1.setFrameShape(QtWidgets.QFrame.HLine)
  470. separator_line1.setFrameShadow(QtWidgets.QFrame.Sunken)
  471. grid_lay.addWidget(separator_line1, 36, 0, 1, 3)
  472. grid_lay.addWidget(QtWidgets.QLabel(''), 37, 0, 1, 3)
  473. # STEP 5 #
  474. step_5 = QtWidgets.QLabel('<b>%s</b>' % _("STEP 5: Calibrate FlatCAM Objects"))
  475. step_5.setToolTip(
  476. _("Adjust the FlatCAM objects\n"
  477. "with the factors determined and verified above.")
  478. )
  479. grid_lay.addWidget(step_5, 38, 0, 1, 3)
  480. self.adj_object_type_combo = QtWidgets.QComboBox()
  481. self.adj_object_type_combo.addItems([_("Gerber"), _("Excellon"), _("Geometry")])
  482. self.adj_object_type_combo.setCurrentIndex(0)
  483. self.adj_object_type_label = QtWidgets.QLabel("%s:" % _("Adjusted object type"))
  484. self.adj_object_type_label.setToolTip(
  485. _("Type of the FlatCAM Object to be adjusted.")
  486. )
  487. grid_lay.addWidget(self.adj_object_type_label, 39, 0, 1, 3)
  488. grid_lay.addWidget(self.adj_object_type_combo, 40, 0, 1, 3)
  489. self.adj_object_combo = FCComboBox()
  490. self.adj_object_combo.setModel(self.app.collection)
  491. self.adj_object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  492. self.adj_object_combo.setCurrentIndex(0)
  493. self.adj_object_label = QtWidgets.QLabel("%s:" % _("Adjusted object selection"))
  494. self.adj_object_label.setToolTip(
  495. _("The FlatCAM Object to be adjusted.")
  496. )
  497. grid_lay.addWidget(self.adj_object_label, 41, 0, 1, 3)
  498. grid_lay.addWidget(self.adj_object_combo, 42, 0, 1, 3)
  499. # ## Adjust Objects Button
  500. self.cal_button = QtWidgets.QPushButton(_("Calibrate"))
  501. self.cal_button.setToolTip(
  502. _("Adjust (scale and/or skew) the objects\n"
  503. "with the factors determined above.")
  504. )
  505. self.cal_button.setStyleSheet("""
  506. QPushButton
  507. {
  508. font-weight: bold;
  509. }
  510. """)
  511. grid_lay.addWidget(self.cal_button, 43, 0, 1, 3)
  512. separator_line2 = QtWidgets.QFrame()
  513. separator_line2.setFrameShape(QtWidgets.QFrame.HLine)
  514. separator_line2.setFrameShadow(QtWidgets.QFrame.Sunken)
  515. grid_lay.addWidget(separator_line2, 44, 0, 1, 3)
  516. grid_lay.addWidget(QtWidgets.QLabel(''), 45, 0, 1, 3)
  517. self.layout.addStretch()
  518. # ## Reset Tool
  519. self.reset_button = QtWidgets.QPushButton(_("Reset Tool"))
  520. self.reset_button.setToolTip(
  521. _("Will reset the tool parameters.")
  522. )
  523. self.reset_button.setStyleSheet("""
  524. QPushButton
  525. {
  526. font-weight: bold;
  527. }
  528. """)
  529. self.layout.addWidget(self.reset_button)
  530. self.mr = None
  531. self.units = ''
  532. # here store 4 points to be used for calibration
  533. self.click_points = list()
  534. # store the status of the grid
  535. self.grid_status_memory = None
  536. self.target_obj = None
  537. # if the mouse events are connected to a local method set this True
  538. self.local_connected = False
  539. # reference for the tab where to open and view the verification GCode
  540. self.gcode_editor_tab = None
  541. # calibrated object
  542. self.cal_object = None
  543. # target values
  544. self.bl_x = 0.0
  545. self.bl_y = 0.0
  546. self.br_x = 0.0
  547. self.br_y = 0.0
  548. self.tl_x = 0.0
  549. self.tl_y = 0.0
  550. self.tr_x = 0.0
  551. self.tr_y = 0.0
  552. # ## Signals
  553. self.cal_source_radio.activated_custom.connect(self.on_cal_source_radio)
  554. self.obj_type_combo.currentIndexChanged.connect(self.on_obj_type_combo)
  555. self.adj_object_type_combo.currentIndexChanged.connect(self.on_adj_obj_type_combo)
  556. self.start_button.clicked.connect(self.on_start_collect_points)
  557. self.gcode_button.clicked.connect(self.generate_verification_gcode)
  558. self.adj_gcode_button.clicked.connect(self.generate_verification_gcode)
  559. self.generate_factors_button.clicked.connect(self.calculate_factors)
  560. self.scale_button.clicked.connect(self.on_scale_button)
  561. self.skew_button.clicked.connect(self.on_skew_button)
  562. self.cal_button.clicked.connect(self.on_cal_button_click)
  563. self.reset_button.clicked.connect(self.set_tool_ui)
  564. def run(self, toggle=True):
  565. self.app.report_usage("ToolCalibration()")
  566. if toggle:
  567. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  568. if self.app.ui.splitter.sizes()[0] == 0:
  569. self.app.ui.splitter.setSizes([1, 1])
  570. else:
  571. try:
  572. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  573. # if tab is populated with the tool but it does not have the focus, focus on it
  574. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  575. # focus on Tool Tab
  576. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  577. else:
  578. self.app.ui.splitter.setSizes([0, 1])
  579. except AttributeError:
  580. pass
  581. else:
  582. if self.app.ui.splitter.sizes()[0] == 0:
  583. self.app.ui.splitter.setSizes([1, 1])
  584. FlatCAMTool.run(self)
  585. self.set_tool_ui()
  586. self.app.ui.notebook.setTabText(2, _("Calibrate Tool"))
  587. def install(self, icon=None, separator=None, **kwargs):
  588. FlatCAMTool.install(self, icon, separator, shortcut='ALT+E', **kwargs)
  589. def set_tool_ui(self):
  590. self.units = self.app.defaults['units'].upper()
  591. if self.local_connected is True:
  592. self.disconnect_cal_events()
  593. self.reset_calibration_points()
  594. self.cal_source_radio.set_value(self.app.defaults['tools_cal_calsource'])
  595. self.travelz_entry.set_value(self.app.defaults['tools_cal_travelz'])
  596. self.verz_entry.set_value(self.app.defaults['tools_cal_verz'])
  597. self.zeroz_cb.set_value(self.app.defaults['tools_cal_zeroz'])
  598. self.toolchangez_entry.set_value(self.app.defaults['tools_cal_toolchangez'])
  599. self.scalex_entry.set_value(1.0)
  600. self.scaley_entry.set_value(1.0)
  601. self.skewx_entry.set_value(0.0)
  602. self.skewy_entry.set_value(0.0)
  603. # calibrated object
  604. self.cal_object = None
  605. self.app.inform.emit('%s...' % _("Tool initialized"))
  606. def on_obj_type_combo(self):
  607. obj_type = self.obj_type_combo.currentIndex()
  608. self.object_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  609. self.object_combo.setCurrentIndex(0)
  610. def on_adj_obj_type_combo(self):
  611. obj_type = self.adj_object_type_combo.currentIndex()
  612. self.adj_object_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  613. self.adj_object_combo.setCurrentIndex(0)
  614. def on_cal_source_radio(self, val):
  615. if val == 'object':
  616. self.obj_type_label.setDisabled(False)
  617. self.obj_type_combo.setDisabled(False)
  618. self.object_label.setDisabled(False)
  619. self.object_combo.setDisabled(False)
  620. else:
  621. self.obj_type_label.setDisabled(True)
  622. self.obj_type_combo.setDisabled(True)
  623. self.object_label.setDisabled(True)
  624. self.object_combo.setDisabled(True)
  625. def on_start_collect_points(self):
  626. if self.cal_source_radio.get_value() == 'object':
  627. selection_index = self.object_combo.currentIndex()
  628. model_index = self.app.collection.index(selection_index, 0, self.object_combo.rootModelIndex())
  629. try:
  630. self.target_obj = model_index.internalPointer().obj
  631. except Exception:
  632. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no source FlatCAM object selected..."))
  633. return
  634. # disengage the grid snapping since it will be hard to find the drills on grid
  635. if self.app.ui.grid_snap_btn.isChecked():
  636. self.grid_status_memory = True
  637. self.app.ui.grid_snap_btn.trigger()
  638. else:
  639. self.grid_status_memory = False
  640. self.mr = self.canvas.graph_event_connect('mouse_release', self.on_mouse_click_release)
  641. if self.app.is_legacy is False:
  642. self.canvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  643. else:
  644. self.canvas.graph_event_disconnect(self.app.mr)
  645. self.local_connected = True
  646. self.reset_calibration_points()
  647. self.app.inform.emit(_("Get First calibration point. Bottom Left..."))
  648. def on_mouse_click_release(self, event):
  649. if event.button == 1:
  650. if self.app.is_legacy is False:
  651. event_pos = event.pos
  652. else:
  653. event_pos = (event.xdata, event.ydata)
  654. pos_canvas = self.canvas.translate_coords(event_pos)
  655. click_pt = Point([pos_canvas[0], pos_canvas[1]])
  656. if self.cal_source_radio.get_value() == 'object':
  657. if self.target_obj.kind.lower() == 'excellon':
  658. for tool, tool_dict in self.target_obj.tools.items():
  659. for geo in tool_dict['solid_geometry']:
  660. if click_pt.within(geo):
  661. center_pt = geo.centroid
  662. self.click_points.append(
  663. (
  664. float('%.*f' % (self.decimals, center_pt.x)),
  665. float('%.*f' % (self.decimals, center_pt.y))
  666. )
  667. )
  668. self.check_points()
  669. else:
  670. for apid, apid_val in self.target_obj.apertures.items():
  671. for geo_el in apid_val['geometry']:
  672. if 'solid' in geo_el:
  673. if click_pt.within(geo_el['solid']):
  674. if isinstance(geo_el['follow'], Point):
  675. center_pt = geo_el['solid'].centroid
  676. self.click_points.append(
  677. (
  678. float('%.*f' % (self.decimals, center_pt.x)),
  679. float('%.*f' % (self.decimals, center_pt.y))
  680. )
  681. )
  682. self.check_points()
  683. else:
  684. self.click_points.append(
  685. (
  686. float('%.*f' % (self.decimals, click_pt.x)),
  687. float('%.*f' % (self.decimals, click_pt.y))
  688. )
  689. )
  690. self.check_points()
  691. def check_points(self):
  692. if len(self.click_points) == 1:
  693. self.bottom_left_coordx_tgt.set_value(self.click_points[0][0])
  694. self.bottom_left_coordy_tgt.set_value(self.click_points[0][1])
  695. self.app.inform.emit(_("Get Second calibration point. Bottom Right..."))
  696. elif len(self.click_points) == 2:
  697. self.bottom_right_coordx_tgt.set_value(self.click_points[1][0])
  698. self.bottom_right_coordy_tgt.set_value(self.click_points[1][1])
  699. self.app.inform.emit(_("Get Third calibration point. Top Left..."))
  700. elif len(self.click_points) == 3:
  701. self.top_left_coordx_tgt.set_value(self.click_points[2][0])
  702. self.top_left_coordy_tgt.set_value(self.click_points[2][1])
  703. self.app.inform.emit(_("Get Forth calibration point. Top Right..."))
  704. elif len(self.click_points) == 4:
  705. self.top_right_coordx_tgt.set_value(self.click_points[3][0])
  706. self.top_right_coordy_tgt.set_value(self.click_points[3][1])
  707. self.app.inform.emit('[success] %s' % _("Done. All four points have been acquired."))
  708. self.disconnect_cal_events()
  709. def reset_calibration_points(self):
  710. self.click_points = list()
  711. self.bottom_left_coordx_tgt.set_value('')
  712. self.bottom_left_coordy_tgt.set_value('')
  713. self.bottom_right_coordx_tgt.set_value('')
  714. self.bottom_right_coordy_tgt.set_value('')
  715. self.top_left_coordx_tgt.set_value('')
  716. self.top_left_coordy_tgt.set_value('')
  717. self.top_right_coordx_tgt.set_value('')
  718. self.top_right_coordy_tgt.set_value('')
  719. def gcode_header(self):
  720. log.debug("ToolCalibration.gcode_header()")
  721. time_str = "{:%A, %d %B %Y at %H:%M}".format(datetime.now())
  722. gcode = '(G-CODE GENERATED BY FLATCAM v%s - www.flatcam.org - Version Date: %s)\n' % \
  723. (str(self.app.version), str(self.app.version_date)) + '\n'
  724. gcode += '(Name: ' + _('Verification GCode for FlatCAM Calibrate Tool') + ')\n'
  725. gcode += '(Units: ' + self.units.upper() + ')\n' + "\n"
  726. gcode += '(Created on ' + time_str + ')\n' + '\n'
  727. gcode += 'G20\n' if self.units.upper() == 'IN' else 'G21\n'
  728. gcode += 'G90\n'
  729. gcode += 'G17\n'
  730. gcode += 'G94\n\n'
  731. return gcode
  732. def close_tab(self):
  733. for idx in range(self.app.ui.plot_tab_area.count()):
  734. if self.app.ui.plot_tab_area.tabText(idx) == _("Gcode Viewer"):
  735. wdg = self.app.ui.plot_tab_area.widget(idx)
  736. wdg.deleteLater()
  737. self.app.ui.plot_tab_area.removeTab(idx)
  738. def generate_verification_gcode(self):
  739. travel_z = '%.*f' % (self.decimals, self.travelz_entry.get_value())
  740. toolchange_z = '%.*f' % (self.decimals, self.toolchangez_entry.get_value())
  741. verification_z = '%.*f' % (self.decimals, self.verz_entry.get_value())
  742. if len(self.click_points) != 4:
  743. self.app.inform.emit('[WARNING_NOTCL] %s' % _("Cancelled. Four points are needed for GCode generation."))
  744. return 'fail'
  745. gcode = self.gcode_header()
  746. if self.zeroz_cb.get_value():
  747. gcode += 'M5\n'
  748. gcode += f'G00 Z{toolchange_z}\n'
  749. gcode += 'M0\n'
  750. gcode += 'G01 Z0\n'
  751. gcode += 'M0\n'
  752. gcode += f'G00 Z{toolchange_z}\n'
  753. gcode += 'M0\n'
  754. gcode += f'G00 Z{travel_z}\n'
  755. gcode += f'G00 X{self.click_points[0][0]} Y{self.click_points[0][1]}\n'
  756. gcode += f'G01 Z{verification_z}\n'
  757. gcode += 'M0\n'
  758. gcode += f'G00 Z{travel_z}\n'
  759. gcode += f'G00 X{self.click_points[2][0]} Y{self.click_points[2][1]}\n'
  760. gcode += f'G01 Z{verification_z}\n'
  761. gcode += 'M0\n'
  762. gcode += f'G00 Z{travel_z}\n'
  763. gcode += f'G00 X{self.click_points[3][0]} Y{self.click_points[3][1]}\n'
  764. gcode += f'G01 Z{verification_z}\n'
  765. gcode += 'M0\n'
  766. gcode += f'G00 Z{travel_z}\n'
  767. gcode += f'G00 X{self.click_points[1][0]} Y{self.click_points[1][1]}\n'
  768. gcode += f'G01 Z{verification_z}\n'
  769. gcode += 'M0\n'
  770. gcode += f'G00 Z{travel_z}\n'
  771. gcode += f'G00 X0 Y0\n'
  772. gcode += f'G00 Z{toolchange_z}\n'
  773. gcode += 'M2'
  774. self.gcode_editor_tab = TextEditor(app=self.app, plain_text=True)
  775. # add the tab if it was closed
  776. self.app.ui.plot_tab_area.addTab(self.gcode_editor_tab, '%s' % _("Gcode Viewer"))
  777. self.gcode_editor_tab.setObjectName('gcode_viewer_tab')
  778. # delete the absolute and relative position and messages in the infobar
  779. self.app.ui.position_label.setText("")
  780. self.app.ui.rel_position_label.setText("")
  781. # first clear previous text in text editor (if any)
  782. self.gcode_editor_tab.code_editor.clear()
  783. self.gcode_editor_tab.code_editor.setReadOnly(False)
  784. self.gcode_editor_tab.code_editor.completer_enable = False
  785. self.gcode_editor_tab.buttonRun.hide()
  786. # Switch plot_area to CNCJob tab
  787. self.app.ui.plot_tab_area.setCurrentWidget(self.gcode_editor_tab)
  788. self.gcode_editor_tab.t_frame.hide()
  789. # then append the text from GCode to the text editor
  790. try:
  791. self.gcode_editor_tab.code_editor.setPlainText(gcode)
  792. except Exception as e:
  793. self.app.inform.emit('[ERROR] %s %s' % ('ERROR -->', str(e)))
  794. return
  795. self.gcode_editor_tab.code_editor.moveCursor(QtGui.QTextCursor.Start)
  796. self.gcode_editor_tab.t_frame.show()
  797. self.app.proc_container.view.set_idle()
  798. self.app.inform.emit('[success] %s...' % _('Loaded Machine Code into Code Editor'))
  799. _filter_ = "G-Code Files (*.nc);;All Files (*.*)"
  800. self.gcode_editor_tab.buttonSave.clicked.disconnect()
  801. self.gcode_editor_tab.buttonSave.clicked.connect(
  802. lambda: self.gcode_editor_tab.handleSaveGCode(name='fc_ver_gcode', filt=_filter_, callback=self.close_tab))
  803. def calculate_factors(self):
  804. origin_x = self.click_points[0][0]
  805. origin_y = self.click_points[0][1]
  806. top_left_x = float('%.*f' % (self.decimals, self.click_points[2][0]))
  807. top_left_y = float('%.*f' % (self.decimals, self.click_points[2][1]))
  808. try:
  809. top_left_dx = float('%.*f' % (self.decimals, self.top_left_coordx_found.get_value()))
  810. except TypeError:
  811. top_left_dx = top_left_x
  812. try:
  813. top_left_dy = float('%.*f' % (self.decimals, self.top_left_coordy_found.get_value()))
  814. except TypeError:
  815. top_left_dy = top_left_y
  816. # top_right_x = float('%.*f' % (self.decimals, self.click_points[3][0]))
  817. # top_right_y = float('%.*f' % (self.decimals, self.click_points[3][1]))
  818. # try:
  819. # top_right_dx = float('%.*f' % (self.decimals, self.top_right_coordx_found.get_value()))
  820. # except TypeError:
  821. # top_right_dx = top_right_x
  822. #
  823. # try:
  824. # top_right_dy = float('%.*f' % (self.decimals, self.top_right_coordy_found.get_value()))
  825. # except TypeError:
  826. # top_right_dy = top_right_y
  827. bot_right_x = float('%.*f' % (self.decimals, self.click_points[1][0]))
  828. bot_right_y = float('%.*f' % (self.decimals, self.click_points[1][1]))
  829. try:
  830. bot_right_dx = float('%.*f' % (self.decimals, self.bottom_right_coordx_found.get_value()))
  831. except TypeError:
  832. bot_right_dx = bot_right_x
  833. try:
  834. bot_right_dy = float('%.*f' % (self.decimals, self.bottom_right_coordy_found.get_value()))
  835. except TypeError:
  836. bot_right_dy = bot_right_y
  837. # ------------------------------------------------------------------------------- #
  838. # --------------------------- FACTORS CALCULUS ---------------------------------- #
  839. # ------------------------------------------------------------------------------- #
  840. if top_left_dy != float('%.*f' % (self.decimals, 0.0)):
  841. # we have scale on Y
  842. scale_y = (top_left_dy + top_left_y - origin_y) / (top_left_y - origin_y)
  843. self.scaley_entry.set_value(scale_y)
  844. if top_left_dx != float('%.*f' % (self.decimals, 0.0)):
  845. # we have skew on X
  846. dx = top_left_dx
  847. dy = top_left_y - origin_y
  848. skew_angle_x = math.degrees(math.atan(dx / dy))
  849. self.skewx_entry.set_value(skew_angle_x)
  850. if bot_right_dx != float('%.*f' % (self.decimals, 0.0)):
  851. # we have scale on X
  852. scale_x = (bot_right_dx + bot_right_x - origin_x) / (bot_right_x - origin_x)
  853. self.scalex_entry.set_value(scale_x)
  854. if bot_right_dy != float('%.*f' % (self.decimals, 0.0)):
  855. # we have skew on Y
  856. dx = bot_right_x - origin_x
  857. dy = bot_right_dy + origin_y
  858. skew_angle_y = math.degrees(math.atan(dy / dx))
  859. self.skewy_entry.set_value(skew_angle_y)
  860. @property
  861. def target_values_from_table(self):
  862. self.bl_x = self.bottom_left_coordx_tgt.get_value()
  863. self.bl_y = self.bottom_left_coordy_tgt.get_value()
  864. self.br_x = self.bottom_right_coordx_tgt.get_value()
  865. self.br_y = self.bottom_right_coordy_tgt.get_value()
  866. self.tl_x = self.top_left_coordx_tgt.get_value()
  867. self.tl_y = self.top_left_coordy_tgt.get_value()
  868. self.tr_x = self.top_right_coordx_tgt.get_value()
  869. self.tr_y = self.top_right_coordy_tgt.get_value()
  870. return (self.bl_x, self.bl_y), (self.br_x, self.br_y), (self.tl_x, self.tl_y), (self.tr_x, self.tr_x)
  871. @target_values_from_table.setter
  872. def target_values_from_table(self, param):
  873. bl_pt, br_pt, tl_pt, tr_pt = param
  874. self.bl_x = bl_pt[0]
  875. self.bl_y = bl_pt[1]
  876. self.br_x = br_pt[0]
  877. self.br_y = br_pt[1]
  878. self.tl_x = tl_pt[0]
  879. self.tl_y = tl_pt[1]
  880. self.tr_x = tr_pt[0]
  881. self.tr_y = tr_pt[1]
  882. self.bottom_left_coordx_tgt.set_value(bl_pt[0])
  883. self.bottom_left_coordy_tgt.set_value(bl_pt[1])
  884. self.bottom_right_coordx_tgt.set_value(br_pt[0])
  885. self.bottom_right_coordy_tgt.set_value(br_pt[1])
  886. self.top_left_coordx_tgt.set_value(tl_pt[0])
  887. self.top_left_coordy_tgt.set_value(tl_pt[1])
  888. self.top_right_coordx_tgt.set_value(tr_pt[0])
  889. self.top_right_coordy_tgt.set_value(tr_pt[1])
  890. def on_scale_button(self):
  891. scalex_fact = self.scalex_entry.get_value()
  892. scaley_fact = self.scaley_entry.get_value()
  893. bl, br, tl, tr = self.target_values_from_table
  894. bl_scaled = scale(Point(bl), xfact=scalex_fact, yfact=scaley_fact, origin=bl)
  895. br_scaled = scale(Point(br), xfact=scalex_fact, yfact=scaley_fact, origin=bl)
  896. tl_scaled = scale(Point(tl), xfact=scalex_fact, yfact=scaley_fact, origin=bl)
  897. tr_scaled = scale(Point(tr), xfact=scalex_fact, yfact=scaley_fact, origin=bl)
  898. scaled_values = (
  899. (bl_scaled.x, bl_scaled.y),
  900. (br_scaled.x, br_scaled.y),
  901. (tl_scaled.x, tl_scaled.y),
  902. (tr_scaled.x, tr_scaled.y)
  903. )
  904. self.target_values_from_table = scaled_values
  905. def on_skew_button(self):
  906. skewx_angle = self.skewx_entry.get_value()
  907. skewy_angle = self.skewy_entry.get_value()
  908. bl, br, tl, tr = self.target_values_from_table
  909. bl_skewed = skew(Point(bl), xs=skewx_angle, ys=skewy_angle, origin=bl)
  910. br_skewed = skew(Point(br), xs=skewx_angle, ys=skewy_angle, origin=bl)
  911. tl_skewed = skew(Point(tl), xs=skewx_angle, ys=skewy_angle, origin=bl)
  912. tr_skewed = skew(Point(tr), xs=skewx_angle, ys=skewy_angle, origin=bl)
  913. skewed_values = (
  914. (bl_skewed.x, bl_skewed.y),
  915. (br_skewed.x, br_skewed.y),
  916. (tl_skewed.x, tl_skewed.y),
  917. (tr_skewed.x, tr_skewed.y)
  918. )
  919. self.target_values_from_table = skewed_values
  920. def on_cal_button_click(self):
  921. # get the FlatCAM object to calibrate
  922. selection_index = self.adj_object_combo.currentIndex()
  923. model_index = self.app.collection.index(selection_index, 0, self.adj_object_combo.rootModelIndex())
  924. try:
  925. self.cal_object = model_index.internalPointer().obj
  926. except Exception as e:
  927. log.debug("ToolCalibration.on_cal_button_click() --> %s" % str(e))
  928. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no FlatCAM object selected..."))
  929. return 'fail'
  930. obj_name = self.cal_object.options["name"] + "_calibrated"
  931. self.app.worker_task.emit({'fcn': self.new_calibrated_object, 'params': [obj_name]})
  932. def new_calibrated_object(self, obj_name):
  933. try:
  934. origin_x = self.click_points[0][0]
  935. origin_y = self.click_points[0][1]
  936. except IndexError as e:
  937. log.debug("ToolCalibration.new_calibrated_object() --> %s" % str(e))
  938. return 'fail'
  939. scalex = self.scalex_entry.get_value()
  940. scaley = self.scaley_entry.get_value()
  941. skewx = self.skewx_entry.get_value()
  942. skewy = self.skewy_entry.get_value()
  943. # create a new object adjusted (calibrated)
  944. def initialize_geometry(obj_init, app):
  945. obj_init.solid_geometry = deepcopy(obj.solid_geometry)
  946. try:
  947. obj_init.follow_geometry = deepcopy(obj.follow_geometry)
  948. except AttributeError:
  949. pass
  950. try:
  951. obj_init.apertures = deepcopy(obj.apertures)
  952. except AttributeError:
  953. pass
  954. try:
  955. if obj.tools:
  956. obj_init.tools = deepcopy(obj.tools)
  957. except Exception as e:
  958. log.debug("App.on_copy_object() --> %s" % str(e))
  959. obj_init.scale(xfactor=scalex, yfactor=scaley, point=(origin_x, origin_y))
  960. obj_init.skew(angle_x=skewx, angle_y=skewy, point=(origin_x, origin_y))
  961. try:
  962. obj_init.source_file = deepcopy(obj.source_file)
  963. except (AttributeError, TypeError):
  964. pass
  965. def initialize_gerber(obj_init, app):
  966. obj_init.solid_geometry = deepcopy(obj.solid_geometry)
  967. try:
  968. obj_init.follow_geometry = deepcopy(obj.follow_geometry)
  969. except AttributeError:
  970. pass
  971. try:
  972. obj_init.apertures = deepcopy(obj.apertures)
  973. except AttributeError:
  974. pass
  975. try:
  976. if obj.tools:
  977. obj_init.tools = deepcopy(obj.tools)
  978. except Exception as e:
  979. log.debug("App.on_copy_object() --> %s" % str(e))
  980. obj_init.scale(xfactor=scalex, yfactor=scaley, point=(origin_x, origin_y))
  981. obj_init.skew(angle_x=skewx, angle_y=skewy, point=(origin_x, origin_y))
  982. try:
  983. obj_init.source_file = self.export_gerber(obj_name=obj_name, filename=None, local_use=obj_init,
  984. use_thread=False)
  985. except (AttributeError, TypeError):
  986. pass
  987. def initialize_excellon(obj_init, app):
  988. obj_init.tools = deepcopy(obj.tools)
  989. # drills are offset, so they need to be deep copied
  990. obj_init.drills = deepcopy(obj.drills)
  991. # slots are offset, so they need to be deep copied
  992. obj_init.slots = deepcopy(obj.slots)
  993. obj_init.scale(xfactor=scalex, yfactor=scaley, point=(origin_x, origin_y))
  994. obj_init.skew(angle_x=skewx, angle_y=skewy, point=(origin_x, origin_y))
  995. obj_init.create_geometry()
  996. obj_init.source_file = self.app.export_excellon(obj_name=obj_name, local_use=obj, filename=None,
  997. use_thread=False)
  998. obj = self.cal_object
  999. obj_name = obj_name
  1000. if obj is None:
  1001. self.app.inform.emit('[WARNING_NOTCL] %s' % _("There is no FlatCAM object selected..."))
  1002. log.debug("ToolCalibration.new_calibrated_object() --> No object to calibrate")
  1003. return 'fail'
  1004. try:
  1005. if obj.kind.lower() == 'excellon':
  1006. self.app.new_object("excellon", str(obj_name), initialize_excellon)
  1007. elif obj.kind.lower() == 'gerber':
  1008. self.app.new_object("gerber", str(obj_name), initialize_gerber)
  1009. elif obj.kind.lower() == 'geometry':
  1010. self.app.new_object("geometry", str(obj_name), initialize_geometry)
  1011. except Exception as e:
  1012. log.debug("ToolCalibration.new_calibrated_object() --> %s" % str(e))
  1013. return "Operation failed: %s" % str(e)
  1014. def disconnect_cal_events(self):
  1015. # restore the Grid snapping if it was active before
  1016. if self.grid_status_memory is True:
  1017. self.app.ui.grid_snap_btn.trigger()
  1018. self.app.mr = self.canvas.graph_event_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
  1019. if self.app.is_legacy is False:
  1020. self.canvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
  1021. else:
  1022. self.canvas.graph_event_disconnect(self.mr)
  1023. self.local_connected = False
  1024. def reset_fields(self):
  1025. self.object_combo.setRootModelIndex(self.app.collection.index(1, 0, QtCore.QModelIndex()))
  1026. self.adj_exc_object_combo.setRootModelIndex(self.app.collection.index(1, 0, QtCore.QModelIndex()))
  1027. self.adj_geo_object_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
  1028. # end of file