| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497 |
- # ##########################################################
- # FlatCAM: 2D Post-processing for Manufacturing #
- # http://flatcam.org #
- # File Modified: Marius Adrian Stanciu (c) #
- # Date: 3/10/2019 #
- # MIT Licence #
- # ##########################################################
- from FlatCAMTool import FlatCAMTool
- from copy import copy, deepcopy
- from ObjectCollection import *
- from shapely.geometry import base
- import gettext
- import FlatCAMTranslation as fcTranslate
- import builtins
- fcTranslate.apply_language('strings')
- if '_' not in builtins.__dict__:
- _ = gettext.gettext
- class ToolPaint(FlatCAMTool, Gerber):
- toolName = _("Paint Tool")
- def __init__(self, app):
- self.app = app
- FlatCAMTool.__init__(self, app)
- Geometry.__init__(self, geo_steps_per_circle=self.app.defaults["geometry_circle_steps"])
- # ## Title
- title_label = QtWidgets.QLabel("%s" % self.toolName)
- title_label.setStyleSheet("""
- QLabel
- {
- font-size: 16px;
- font-weight: bold;
- }
- """)
- self.layout.addWidget(title_label)
- self.tools_frame = QtWidgets.QFrame()
- self.tools_frame.setContentsMargins(0, 0, 0, 0)
- self.layout.addWidget(self.tools_frame)
- self.tools_box = QtWidgets.QVBoxLayout()
- self.tools_box.setContentsMargins(0, 0, 0, 0)
- self.tools_frame.setLayout(self.tools_box)
- # ## Form Layout
- form_layout = QtWidgets.QFormLayout()
- self.tools_box.addLayout(form_layout)
- # ################################################
- # ##### Type of object to be painted #############
- # ################################################
- self.type_obj_combo = QtWidgets.QComboBox()
- self.type_obj_combo.addItem("Gerber")
- self.type_obj_combo.addItem("Excellon")
- self.type_obj_combo.addItem("Geometry")
- # we get rid of item1 ("Excellon") as it is not suitable
- self.type_obj_combo.view().setRowHidden(1, True)
- self.type_obj_combo.setItemIcon(0, QtGui.QIcon("share/flatcam_icon16.png"))
- self.type_obj_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
- self.type_obj_combo_label = QtWidgets.QLabel('%s:' % _("Obj Type"))
- self.type_obj_combo_label.setToolTip(
- _("Specify the type of object to be painted.\n"
- "It can be of type: Gerber or Geometry.\n"
- "What is selected here will dictate the kind\n"
- "of objects that will populate the 'Object' combobox.")
- )
- self.type_obj_combo_label.setMinimumWidth(60)
- form_layout.addRow(self.type_obj_combo_label, self.type_obj_combo)
- # ################################################
- # ##### The object to be painted #################
- # ################################################
- self.obj_combo = QtWidgets.QComboBox()
- self.obj_combo.setModel(self.app.collection)
- self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
- self.obj_combo.setCurrentIndex(1)
- self.object_label = QtWidgets.QLabel('%s:' % _("Object"))
- self.object_label.setToolTip(_("Object to be painted."))
- form_layout.addRow(self.object_label, self.obj_combo)
- e_lab_0 = QtWidgets.QLabel('')
- form_layout.addRow(e_lab_0)
- # ### Tools ## ##
- self.tools_table_label = QtWidgets.QLabel('<b>%s</b>' % _('Tools Table'))
- self.tools_table_label.setToolTip(
- _("Tools pool from which the algorithm\n"
- "will pick the ones used for painting.")
- )
- self.tools_box.addWidget(self.tools_table_label)
- self.tools_table = FCTable()
- self.tools_box.addWidget(self.tools_table)
- self.tools_table.setColumnCount(4)
- self.tools_table.setHorizontalHeaderLabels(['#', _('Diameter'), _('TT'), ''])
- self.tools_table.setColumnHidden(3, True)
- # self.tools_table.setSortingEnabled(False)
- # self.tools_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
- self.tools_table.horizontalHeaderItem(0).setToolTip(
- _("This is the Tool Number.\n"
- "Painting will start with the tool with the biggest diameter,\n"
- "continuing until there are no more tools.\n"
- "Only tools that create painting geometry will still be present\n"
- "in the resulting geometry. This is because with some tools\n"
- "this function will not be able to create painting geometry.")
- )
- self.tools_table.horizontalHeaderItem(1).setToolTip(
- _("Tool Diameter. It's value (in current FlatCAM units) \n"
- "is the cut width into the material."))
- self.tools_table.horizontalHeaderItem(2).setToolTip(
- _("The Tool Type (TT) can be:<BR>"
- "- <B>Circular</B> with 1 ... 4 teeth -> it is informative only. Being circular, <BR>"
- "the cut width in material is exactly the tool diameter.<BR>"
- "- <B>Ball</B> -> informative only and make reference to the Ball type endmill.<BR>"
- "- <B>V-Shape</B> -> it will disable de Z-Cut parameter in the resulting geometry UI form "
- "and enable two additional UI form fields in the resulting geometry: V-Tip Dia and "
- "V-Tip Angle. Adjusting those two values will adjust the Z-Cut parameter such "
- "as the cut width into material will be equal with the value in the Tool Diameter "
- "column of this table.<BR>"
- "Choosing the <B>V-Shape</B> Tool Type automatically will select the Operation Type "
- "in the resulting geometry as Isolation."))
- self.order_label = QtWidgets.QLabel('<b>%s:</b>' % _('Tool order'))
- self.order_label.setToolTip(_("This set the way that the tools in the tools table are used.\n"
- "'No' --> means that the used order is the one in the tool table\n"
- "'Forward' --> means that the tools will be ordered from small to big\n"
- "'Reverse' --> menas that the tools will ordered from big to small\n\n"
- "WARNING: using rest machining will automatically set the order\n"
- "in reverse and disable this control."))
- self.order_radio = RadioSet([{'label': _('No'), 'value': 'no'},
- {'label': _('Forward'), 'value': 'fwd'},
- {'label': _('Reverse'), 'value': 'rev'}])
- self.order_radio.setToolTip(_("This set the way that the tools in the tools table are used.\n"
- "'No' --> means that the used order is the one in the tool table\n"
- "'Forward' --> means that the tools will be ordered from small to big\n"
- "'Reverse' --> menas that the tools will ordered from big to small\n\n"
- "WARNING: using rest machining will automatically set the order\n"
- "in reverse and disable this control."))
- form = QtWidgets.QFormLayout()
- self.tools_box.addLayout(form)
- form.addRow(QtWidgets.QLabel(''), QtWidgets.QLabel(''))
- form.addRow(self.order_label, self.order_radio)
- # ### Add a new Tool ## ##
- hlay = QtWidgets.QHBoxLayout()
- self.tools_box.addLayout(hlay)
- self.addtool_entry_lbl = QtWidgets.QLabel('<b>%s:</b>' % _('Tool Dia'))
- self.addtool_entry_lbl.setToolTip(
- _("Diameter for the new tool.")
- )
- self.addtool_entry = FCEntry2()
- # hlay.addWidget(self.addtool_label)
- # hlay.addStretch()
- hlay.addWidget(self.addtool_entry_lbl)
- hlay.addWidget(self.addtool_entry)
- grid2 = QtWidgets.QGridLayout()
- self.tools_box.addLayout(grid2)
- self.addtool_btn = QtWidgets.QPushButton(_('Add'))
- self.addtool_btn.setToolTip(
- _("Add a new tool to the Tool Table\n"
- "with the diameter specified above.")
- )
- # self.copytool_btn = QtWidgets.QPushButton('Copy')
- # self.copytool_btn.setToolTip(
- # "Copy a selection of tools in the Tool Table\n"
- # "by first selecting a row in the Tool Table."
- # )
- self.deltool_btn = QtWidgets.QPushButton(_('Delete'))
- self.deltool_btn.setToolTip(
- _("Delete a selection of tools in the Tool Table\n"
- "by first selecting a row(s) in the Tool Table.")
- )
- grid2.addWidget(self.addtool_btn, 0, 0)
- # grid2.addWidget(self.copytool_btn, 0, 1)
- grid2.addWidget(self.deltool_btn, 0, 2)
- self.empty_label_0 = QtWidgets.QLabel('')
- self.tools_box.addWidget(self.empty_label_0)
- grid3 = QtWidgets.QGridLayout()
- self.tools_box.addLayout(grid3)
- # Overlap
- ovlabel = QtWidgets.QLabel('%s:' % _('Overlap Rate'))
- ovlabel.setToolTip(
- _("How much (fraction) of the tool width to overlap each tool pass.\n"
- "Example:\n"
- "A value here of 0.25 means 25%% from the tool diameter found above.\n\n"
- "Adjust the value starting with lower values\n"
- "and increasing it if areas that should be painted are still \n"
- "not painted.\n"
- "Lower values = faster processing, faster execution on PCB.\n"
- "Higher values = slow processing and slow execution on CNC\n"
- "due of too many paths.")
- )
- self.paintoverlap_entry = FCDoubleSpinner()
- self.paintoverlap_entry.set_precision(3)
- self.paintoverlap_entry.setWrapping(True)
- self.paintoverlap_entry.setRange(0.000, 0.999)
- self.paintoverlap_entry.setSingleStep(0.1)
- grid3.addWidget(ovlabel, 1, 0)
- grid3.addWidget(self.paintoverlap_entry, 1, 1)
- # Margin
- marginlabel = QtWidgets.QLabel('%s:' % _('Margin'))
- marginlabel.setToolTip(
- _("Distance by which to avoid\n"
- "the edges of the polygon to\n"
- "be painted.")
- )
- grid3.addWidget(marginlabel, 2, 0)
- self.paintmargin_entry = FCEntry()
- grid3.addWidget(self.paintmargin_entry, 2, 1)
- # Method
- methodlabel = QtWidgets.QLabel('%s:' % _('Method'))
- methodlabel.setToolTip(
- _("Algorithm for painting:\n"
- "- Standard: Fixed step inwards.\n"
- "- Seed-based: Outwards from seed.\n"
- "- Line-based: Parallel lines.")
- )
- grid3.addWidget(methodlabel, 3, 0)
- self.paintmethod_combo = RadioSet([
- {"label": _("Standard"), "value": "standard"},
- {"label": _("Seed-based"), "value": "seed"},
- {"label": _("Straight lines"), "value": "lines"}
- ], orientation='vertical', stretch=False)
- grid3.addWidget(self.paintmethod_combo, 3, 1)
- # Connect lines
- pathconnectlabel = QtWidgets.QLabel('%s:' % _("Connect"))
- pathconnectlabel.setToolTip(
- _("Draw lines between resulting\n"
- "segments to minimize tool lifts.")
- )
- grid3.addWidget(pathconnectlabel, 4, 0)
- self.pathconnect_cb = FCCheckBox()
- grid3.addWidget(self.pathconnect_cb, 4, 1)
- contourlabel = QtWidgets.QLabel('%s:' % _("Contour"))
- contourlabel.setToolTip(
- _("Cut around the perimeter of the polygon\n"
- "to trim rough edges.")
- )
- grid3.addWidget(contourlabel, 5, 0)
- self.paintcontour_cb = FCCheckBox()
- grid3.addWidget(self.paintcontour_cb, 5, 1)
- restlabel = QtWidgets.QLabel('%s:' % _("Rest M."))
- restlabel.setToolTip(
- _("If checked, use 'rest machining'.\n"
- "Basically it will clear copper outside PCB features,\n"
- "using the biggest tool and continue with the next tools,\n"
- "from bigger to smaller, to clear areas of copper that\n"
- "could not be cleared by previous tool, until there is\n"
- "no more copper to clear or there are no more tools.\n\n"
- "If not checked, use the standard algorithm.")
- )
- grid3.addWidget(restlabel, 6, 0)
- self.rest_cb = FCCheckBox()
- grid3.addWidget(self.rest_cb, 6, 1)
- # Polygon selection
- selectlabel = QtWidgets.QLabel('%s:' % _('Selection'))
- selectlabel.setToolTip(
- _("How to select Polygons to be painted.\n\n"
- "- 'Area Selection' - left mouse click to start selection of the area to be painted.\n"
- "Keeping a modifier key pressed (CTRL or SHIFT) will allow to add multiple areas.\n"
- "- 'All Polygons' - the Paint will start after click.\n"
- "- 'Reference Object' - will do non copper clearing within the area\n"
- "specified by another object.")
- )
- grid3.addWidget(selectlabel, 7, 0)
- # grid3 = QtWidgets.QGridLayout()
- self.selectmethod_combo = RadioSet([
- {"label": _("Single Polygon"), "value": "single"},
- {"label": _("Area Selection"), "value": "area"},
- {"label": _("All Polygons"), "value": "all"},
- {"label": _("Reference Object"), "value": "ref"}
- ], orientation='vertical', stretch=False)
- self.selectmethod_combo.setToolTip(
- _("How to select Polygons to be painted.\n\n"
- "- 'Area Selection' - left mouse click to start selection of the area to be painted.\n"
- "Keeping a modifier key pressed (CTRL or SHIFT) will allow to add multiple areas.\n"
- "- 'All Polygons' - the Paint will start after click.\n"
- "- 'Reference Object' - will do non copper clearing within the area\n"
- "specified by another object.")
- )
- grid3.addWidget(self.selectmethod_combo, 7, 1)
- form1 = QtWidgets.QFormLayout()
- self.tools_box.addLayout(form1)
- self.box_combo_type_label = QtWidgets.QLabel('%s:' % _("Ref. Type"))
- self.box_combo_type_label.setToolTip(
- _("The type of FlatCAM object to be used as paint reference.\n"
- "It can be Gerber, Excellon or Geometry.")
- )
- self.box_combo_type = QtWidgets.QComboBox()
- self.box_combo_type.addItem(_("Gerber Reference Box Object"))
- self.box_combo_type.addItem(_("Excellon Reference Box Object"))
- self.box_combo_type.addItem(_("Geometry Reference Box Object"))
- form1.addRow(self.box_combo_type_label, self.box_combo_type)
- self.box_combo_label = QtWidgets.QLabel('%s:' % _("Ref. Object"))
- self.box_combo_label.setToolTip(
- _("The FlatCAM object to be used as non copper clearing reference.")
- )
- self.box_combo = QtWidgets.QComboBox()
- self.box_combo.setModel(self.app.collection)
- self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
- self.box_combo.setCurrentIndex(1)
- form1.addRow(self.box_combo_label, self.box_combo)
- self.box_combo.hide()
- self.box_combo_label.hide()
- self.box_combo_type.hide()
- self.box_combo_type_label.hide()
- # GO Button
- self.generate_paint_button = QtWidgets.QPushButton(_('Create Paint Geometry'))
- self.generate_paint_button.setToolTip(
- _("- 'Area Selection' - left mouse click to start selection of the area to be painted.\n"
- "Keeping a modifier key pressed (CTRL or SHIFT) will allow to add multiple areas.\n"
- "- 'All Polygons' - the Paint will start after click.\n"
- "- 'Reference Object' - will do non copper clearing within the area\n"
- "specified by another object.")
- )
- self.tools_box.addWidget(self.generate_paint_button)
- self.tools_box.addStretch()
- self.obj_name = ""
- self.paint_obj = None
- self.bound_obj_name = ""
- self.bound_obj = None
- self.units = ''
- self.paint_tools = {}
- self.tooluid = 0
- self.first_click = False
- self.cursor_pos = None
- self.mouse_is_dragging = False
- self.sel_rect = []
- # store here the default data for Geometry Data
- self.default_data = {}
- self.default_data.update({
- "name": '_paint',
- "plot": self.app.defaults["geometry_plot"],
- "cutz": self.app.defaults["geometry_cutz"],
- "vtipdia": 0.1,
- "vtipangle": 30,
- "travelz": self.app.defaults["geometry_travelz"],
- "feedrate": self.app.defaults["geometry_feedrate"],
- "feedrate_z": self.app.defaults["geometry_feedrate_z"],
- "feedrate_rapid": self.app.defaults["geometry_feedrate_rapid"],
- "dwell": self.app.defaults["geometry_dwell"],
- "dwelltime": self.app.defaults["geometry_dwelltime"],
- "multidepth": self.app.defaults["geometry_multidepth"],
- "ppname_g": self.app.defaults["geometry_ppname_g"],
- "depthperpass": self.app.defaults["geometry_depthperpass"],
- "extracut": self.app.defaults["geometry_extracut"],
- "toolchange": self.app.defaults["geometry_toolchange"],
- "toolchangez": self.app.defaults["geometry_toolchangez"],
- "endz": self.app.defaults["geometry_endz"],
- "spindlespeed": self.app.defaults["geometry_spindlespeed"],
- "toolchangexy": self.app.defaults["geometry_toolchangexy"],
- "startz": self.app.defaults["geometry_startz"],
- "tooldia": self.app.defaults["tools_painttooldia"],
- "paintmargin": self.app.defaults["tools_paintmargin"],
- "paintmethod": self.app.defaults["tools_paintmethod"],
- "selectmethod": self.app.defaults["tools_selectmethod"],
- "pathconnect": self.app.defaults["tools_pathconnect"],
- "paintcontour": self.app.defaults["tools_paintcontour"],
- "paintoverlap": self.app.defaults["tools_paintoverlap"]
- })
- self.tool_type_item_options = ["C1", "C2", "C3", "C4", "B", "V"]
- # ## Signals
- self.addtool_btn.clicked.connect(self.on_tool_add)
- self.addtool_entry.returnPressed.connect(self.on_tool_add)
- # self.copytool_btn.clicked.connect(lambda: self.on_tool_copy())
- self.tools_table.itemChanged.connect(self.on_tool_edit)
- self.deltool_btn.clicked.connect(self.on_tool_delete)
- self.generate_paint_button.clicked.connect(self.on_paint_button_click)
- self.selectmethod_combo.activated_custom.connect(self.on_radio_selection)
- self.order_radio.activated_custom[str].connect(self.on_order_changed)
- self.rest_cb.stateChanged.connect(self.on_rest_machining_check)
- self.box_combo_type.currentIndexChanged.connect(self.on_combo_box_type)
- self.type_obj_combo.currentIndexChanged.connect(self.on_type_obj_index_changed)
- def on_type_obj_index_changed(self, index):
- obj_type = self.type_obj_combo.currentIndex()
- self.obj_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
- self.obj_combo.setCurrentIndex(0)
- def install(self, icon=None, separator=None, **kwargs):
- FlatCAMTool.install(self, icon, separator, shortcut='ALT+P', **kwargs)
- def run(self, toggle=True):
- self.app.report_usage("ToolPaint()")
- if toggle:
- # if the splitter is hidden, display it, else hide it but only if the current widget is the same
- if self.app.ui.splitter.sizes()[0] == 0:
- self.app.ui.splitter.setSizes([1, 1])
- else:
- try:
- if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
- # if tab is populated with the tool but it does not have the focus, focus on it
- if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
- # focus on Tool Tab
- self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
- else:
- self.app.ui.splitter.setSizes([0, 1])
- except AttributeError:
- pass
- else:
- if self.app.ui.splitter.sizes()[0] == 0:
- self.app.ui.splitter.setSizes([1, 1])
- FlatCAMTool.run(self)
- self.set_tool_ui()
- self.app.ui.notebook.setTabText(2, _("Paint Tool"))
- def reset_usage(self):
- self.obj_name = ""
- self.paint_obj = None
- self.bound_obj = None
- self.first_click = False
- self.cursor_pos = None
- self.mouse_is_dragging = False
- self.sel_rect = []
- def on_radio_selection(self):
- if self.selectmethod_combo.get_value() == "ref":
- self.box_combo.show()
- self.box_combo_label.show()
- self.box_combo_type.show()
- self.box_combo_type_label.show()
- else:
- self.box_combo.hide()
- self.box_combo_label.hide()
- self.box_combo_type.hide()
- self.box_combo_type_label.hide()
- if self.selectmethod_combo.get_value() == 'single':
- # disable rest-machining for single polygon painting
- self.rest_cb.set_value(False)
- self.rest_cb.setDisabled(True)
- # delete all tools except first row / tool for single polygon painting
- # list_to_del = list(range(1, self.tools_table.rowCount()))
- # if list_to_del:
- # self.on_tool_delete(rows_to_delete=list_to_del)
- # # disable addTool and delTool
- # self.addtool_entry.setDisabled(True)
- # self.addtool_btn.setDisabled(True)
- # self.deltool_btn.setDisabled(True)
- # self.tools_table.setContextMenuPolicy(Qt.NoContextMenu)
- if self.selectmethod_combo.get_value() == 'area':
- # disable rest-machining for single polygon painting
- self.rest_cb.set_value(False)
- self.rest_cb.setDisabled(True)
- else:
- self.rest_cb.setDisabled(False)
- self.addtool_entry.setDisabled(False)
- self.addtool_btn.setDisabled(False)
- self.deltool_btn.setDisabled(False)
- self.tools_table.setContextMenuPolicy(Qt.ActionsContextMenu)
- def on_order_changed(self, order):
- if order != 'no':
- self.build_ui()
- def on_rest_machining_check(self, state):
- if state:
- self.order_radio.set_value('rev')
- self.order_label.setDisabled(True)
- self.order_radio.setDisabled(True)
- else:
- self.order_label.setDisabled(False)
- self.order_radio.setDisabled(False)
- def set_tool_ui(self):
- self.tools_frame.show()
- self.reset_fields()
- # ## Init the GUI interface
- self.order_radio.set_value(self.app.defaults["tools_paintorder"])
- self.paintmargin_entry.set_value(self.default_data["paintmargin"])
- self.paintmethod_combo.set_value(self.default_data["paintmethod"])
- self.selectmethod_combo.set_value(self.default_data["selectmethod"])
- self.pathconnect_cb.set_value(self.default_data["pathconnect"])
- self.paintcontour_cb.set_value(self.default_data["paintcontour"])
- self.paintoverlap_entry.set_value(self.default_data["paintoverlap"])
- # make the default object type, "Geometry"
- self.type_obj_combo.setCurrentIndex(2)
- # updated units
- self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
- if self.units == "IN":
- self.addtool_entry.set_value(0.039)
- else:
- self.addtool_entry.set_value(1)
- self.tools_table.setupContextMenu()
- self.tools_table.addContextMenu(
- "Add", lambda: self.on_tool_add(dia=None, muted=None), icon=QtGui.QIcon("share/plus16.png"))
- self.tools_table.addContextMenu(
- "Delete", lambda:
- self.on_tool_delete(rows_to_delete=None, all=None), icon=QtGui.QIcon("share/delete32.png"))
- # set the working variables to a known state
- self.paint_tools.clear()
- self.tooluid = 0
- self.default_data.clear()
- self.default_data.update({
- "name": '_paint',
- "plot": self.app.defaults["geometry_plot"],
- "cutz": self.app.defaults["geometry_cutz"],
- "vtipdia": 0.1,
- "vtipangle": 30,
- "travelz": self.app.defaults["geometry_travelz"],
- "feedrate": self.app.defaults["geometry_feedrate"],
- "feedrate_z": self.app.defaults["geometry_feedrate_z"],
- "feedrate_rapid": self.app.defaults["geometry_feedrate_rapid"],
- "dwell": self.app.defaults["geometry_dwell"],
- "dwelltime": self.app.defaults["geometry_dwelltime"],
- "multidepth": self.app.defaults["geometry_multidepth"],
- "ppname_g": self.app.defaults["geometry_ppname_g"],
- "depthperpass": self.app.defaults["geometry_depthperpass"],
- "extracut": self.app.defaults["geometry_extracut"],
- "toolchange": self.app.defaults["geometry_toolchange"],
- "toolchangez": self.app.defaults["geometry_toolchangez"],
- "endz": self.app.defaults["geometry_endz"],
- "spindlespeed": self.app.defaults["geometry_spindlespeed"],
- "toolchangexy": self.app.defaults["geometry_toolchangexy"],
- "startz": self.app.defaults["geometry_startz"],
- "tooldia": self.app.defaults["tools_painttooldia"],
- "paintmargin": self.app.defaults["tools_paintmargin"],
- "paintmethod": self.app.defaults["tools_paintmethod"],
- "selectmethod": self.app.defaults["tools_selectmethod"],
- "pathconnect": self.app.defaults["tools_pathconnect"],
- "paintcontour": self.app.defaults["tools_paintcontour"],
- "paintoverlap": self.app.defaults["tools_paintoverlap"]
- })
- # call on self.on_tool_add() counts as an call to self.build_ui()
- # through this, we add a initial row / tool in the tool_table
- self.on_tool_add(self.app.defaults["tools_painttooldia"], muted=True)
- # if the Paint Method is "Single" disable the tool table context menu
- if self.default_data["selectmethod"] == "single":
- self.tools_table.setContextMenuPolicy(Qt.NoContextMenu)
- def build_ui(self):
- try:
- # if connected, disconnect the signal from the slot on item_changed as it creates issues
- self.tools_table.itemChanged.disconnect()
- except (TypeError, AttributeError):
- pass
- # updated units
- self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
- sorted_tools = []
- for k, v in self.paint_tools.items():
- sorted_tools.append(float('%.4f' % float(v['tooldia'])))
- order = self.order_radio.get_value()
- if order == 'fwd':
- sorted_tools.sort(reverse=False)
- elif order == 'rev':
- sorted_tools.sort(reverse=True)
- else:
- pass
- n = len(sorted_tools)
- self.tools_table.setRowCount(n)
- tool_id = 0
- for tool_sorted in sorted_tools:
- for tooluid_key, tooluid_value in self.paint_tools.items():
- if float('%.4f' % tooluid_value['tooldia']) == tool_sorted:
- tool_id += 1
- id = QtWidgets.QTableWidgetItem('%d' % int(tool_id))
- id.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
- row_no = tool_id - 1
- self.tools_table.setItem(row_no, 0, id) # Tool name/id
- # Make sure that the drill diameter when in MM is with no more than 2 decimals
- # There are no drill bits in MM with more than 3 decimals diameter
- # For INCH the decimals should be no more than 3. There are no drills under 10mils
- if self.units == 'MM':
- dia = QtWidgets.QTableWidgetItem('%.2f' % tooluid_value['tooldia'])
- else:
- dia = QtWidgets.QTableWidgetItem('%.4f' % tooluid_value['tooldia'])
- dia.setFlags(QtCore.Qt.ItemIsEnabled)
- tool_type_item = QtWidgets.QComboBox()
- for item in self.tool_type_item_options:
- tool_type_item.addItem(item)
- tool_type_item.setStyleSheet('background-color: rgb(255,255,255)')
- idx = tool_type_item.findText(tooluid_value['tool_type'])
- tool_type_item.setCurrentIndex(idx)
- tool_uid_item = QtWidgets.QTableWidgetItem(str(int(tooluid_key)))
- self.tools_table.setItem(row_no, 1, dia) # Diameter
- self.tools_table.setCellWidget(row_no, 2, tool_type_item)
- # ## REMEMBER: THIS COLUMN IS HIDDEN IN OBJECTUI.PY # ##
- self.tools_table.setItem(row_no, 3, tool_uid_item) # Tool unique ID
- # make the diameter column editable
- for row in range(tool_id):
- self.tools_table.item(row, 1).setFlags(
- QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
- # all the tools are selected by default
- self.tools_table.selectColumn(0)
- #
- self.tools_table.resizeColumnsToContents()
- self.tools_table.resizeRowsToContents()
- vertical_header = self.tools_table.verticalHeader()
- vertical_header.hide()
- self.tools_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
- horizontal_header = self.tools_table.horizontalHeader()
- horizontal_header.setMinimumSectionSize(10)
- horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
- horizontal_header.resizeSection(0, 20)
- horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
- # self.tools_table.setSortingEnabled(True)
- # sort by tool diameter
- # self.tools_table.sortItems(1)
- self.tools_table.setMinimumHeight(self.tools_table.getHeight())
- self.tools_table.setMaximumHeight(self.tools_table.getHeight())
- # we reactivate the signals after the after the tool adding as we don't need to see the tool been populated
- self.tools_table.itemChanged.connect(self.on_tool_edit)
- def on_combo_box_type(self):
- obj_type = self.box_combo_type.currentIndex()
- self.box_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
- self.box_combo.setCurrentIndex(0)
- def on_tool_add(self, dia=None, muted=None):
- try:
- self.tools_table.itemChanged.disconnect()
- except TypeError:
- pass
- if dia:
- tool_dia = dia
- else:
- try:
- tool_dia = float(self.addtool_entry.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- tool_dia = float(self.addtool_entry.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- return
- if tool_dia is None:
- self.build_ui()
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Please enter a tool diameter to add, in Float format."))
- return
- # construct a list of all 'tooluid' in the self.tools
- tool_uid_list = []
- for tooluid_key in self.paint_tools:
- tool_uid_item = int(tooluid_key)
- tool_uid_list.append(tool_uid_item)
- # find maximum from the temp_uid, add 1 and this is the new 'tooluid'
- if not tool_uid_list:
- max_uid = 0
- else:
- max_uid = max(tool_uid_list)
- self.tooluid = int(max_uid + 1)
- tool_dias = []
- for k, v in self.paint_tools.items():
- for tool_v in v.keys():
- if tool_v == 'tooldia':
- tool_dias.append(float('%.4f' % v[tool_v]))
- if float('%.4f' % tool_dia) in tool_dias:
- if muted is None:
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Adding tool cancelled. Tool already in Tool Table."))
- self.tools_table.itemChanged.connect(self.on_tool_edit)
- return
- else:
- if muted is None:
- self.app.inform.emit('[success] %s' %
- _("New tool added to Tool Table."))
- self.paint_tools.update({
- int(self.tooluid): {
- 'tooldia': float('%.4f' % tool_dia),
- 'offset': 'Path',
- 'offset_value': 0.0,
- 'type': 'Iso',
- 'tool_type': 'V',
- 'data': dict(self.default_data),
- 'solid_geometry': []
- }
- })
- self.build_ui()
- def on_tool_edit(self):
- old_tool_dia = ''
- try:
- self.tools_table.itemChanged.disconnect()
- except TypeError:
- pass
- tool_dias = []
- for k, v in self.paint_tools.items():
- for tool_v in v.keys():
- if tool_v == 'tooldia':
- tool_dias.append(float('%.4f' % v[tool_v]))
- for row in range(self.tools_table.rowCount()):
- try:
- new_tool_dia = float(self.tools_table.item(row, 1).text())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- new_tool_dia = float(self.tools_table.item(row, 1).text().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- return
- tooluid = int(self.tools_table.item(row, 3).text())
- # identify the tool that was edited and get it's tooluid
- if new_tool_dia not in tool_dias:
- self.paint_tools[tooluid]['tooldia'] = new_tool_dia
- self.app.inform.emit('[success] %s' %
- _("Tool from Tool Table was edited."))
- self.build_ui()
- return
- else:
- # identify the old tool_dia and restore the text in tool table
- for k, v in self.paint_tools.items():
- if k == tooluid:
- old_tool_dia = v['tooldia']
- break
- restore_dia_item = self.tools_table.item(row, 1)
- restore_dia_item.setText(str(old_tool_dia))
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Edit cancelled. New diameter value is already in the Tool Table."))
- self.build_ui()
- # def on_tool_copy(self, all=None):
- # try:
- # self.tools_table.itemChanged.disconnect()
- # except:
- # pass
- #
- # # find the tool_uid maximum value in the self.tools
- # uid_list = []
- # for key in self.paint_tools:
- # uid_list.append(int(key))
- # try:
- # max_uid = max(uid_list, key=int)
- # except ValueError:
- # max_uid = 0
- #
- # if all is None:
- # if self.tools_table.selectedItems():
- # for current_row in self.tools_table.selectedItems():
- # # sometime the header get selected and it has row number -1
- # # we don't want to do anything with the header :)
- # if current_row.row() < 0:
- # continue
- # try:
- # tooluid_copy = int(self.tools_table.item(current_row.row(), 3).text())
- # max_uid += 1
- # self.paint_tools[int(max_uid)] = dict(self.paint_tools[tooluid_copy])
- # for td in self.paint_tools:
- # print("COPIED", self.paint_tools[td])
- # self.build_ui()
- # except AttributeError:
- # self.app.inform.emit("[WARNING_NOTCL] Failed. Select a tool to copy.")
- # self.build_ui()
- # return
- # except Exception as e:
- # log.debug("on_tool_copy() --> " + str(e))
- # # deselect the table
- # # self.ui.geo_tools_table.clearSelection()
- # else:
- # self.app.inform.emit("[WARNING_NOTCL] Failed. Select a tool to copy.")
- # self.build_ui()
- # return
- # else:
- # # we copy all tools in geo_tools_table
- # try:
- # temp_tools = dict(self.paint_tools)
- # max_uid += 1
- # for tooluid in temp_tools:
- # self.paint_tools[int(max_uid)] = dict(temp_tools[tooluid])
- # temp_tools.clear()
- # self.build_ui()
- # except Exception as e:
- # log.debug("on_tool_copy() --> " + str(e))
- #
- # self.app.inform.emit("[success] Tool was copied in the Tool Table.")
- def on_tool_delete(self, rows_to_delete=None, all=None):
- try:
- self.tools_table.itemChanged.disconnect()
- except TypeError:
- pass
- deleted_tools_list = []
- if all:
- self.paint_tools.clear()
- self.build_ui()
- return
- if rows_to_delete:
- try:
- for row in rows_to_delete:
- tooluid_del = int(self.tools_table.item(row, 3).text())
- deleted_tools_list.append(tooluid_del)
- except TypeError:
- deleted_tools_list.append(rows_to_delete)
- for t in deleted_tools_list:
- self.paint_tools.pop(t, None)
- self.build_ui()
- return
- try:
- if self.tools_table.selectedItems():
- for row_sel in self.tools_table.selectedItems():
- row = row_sel.row()
- if row < 0:
- continue
- tooluid_del = int(self.tools_table.item(row, 3).text())
- deleted_tools_list.append(tooluid_del)
- for t in deleted_tools_list:
- self.paint_tools.pop(t, None)
- except AttributeError:
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Delete failed. Select a tool to delete."))
- return
- except Exception as e:
- log.debug(str(e))
- self.app.inform.emit('[success] %s' %
- _("Tool(s) deleted from Tool Table."))
- self.build_ui()
- def on_paint_button_click(self):
- # init values for the next usage
- self.reset_usage()
- self.app.report_usage(_("on_paint_button_click"))
- # self.app.call_source = 'paint'
- # #####################################################
- # ######### Reading Parameters ########################
- # #####################################################
- self.app.inform.emit(_("Paint Tool. Reading parameters."))
- try:
- overlap = float(self.paintoverlap_entry.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- overlap = float(self.paintoverlap_entry.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- return
- if overlap >= 1 or overlap < 0:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Overlap value must be between 0 (inclusive) and 1 (exclusive)"))
- return
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Click inside the desired polygon."))
- connect = self.pathconnect_cb.get_value()
- contour = self.paintcontour_cb.get_value()
- select_method = self.selectmethod_combo.get_value()
- self.obj_name = self.obj_combo.currentText()
- # Get source object.
- try:
- self.paint_obj = self.app.collection.get_by_name(str(self.obj_name))
- except Exception as e:
- log.debug("ToolPaint.on_paint_button_click() --> %s" % str(e))
- self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
- (_("Could not retrieve object: %s"),
- self.obj_name))
- return
- if self.paint_obj is None:
- self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
- (_("Object not found"),
- self.paint_obj))
- return
- # test if the Geometry Object is multigeo and return Fail if True because
- # for now Paint don't work on MultiGeo
- if self.paint_obj.multigeo is True:
- self.app.inform.emit('[ERROR_NOTCL] %s...' %
- _("Can't do Paint on MultiGeo geometries"))
- return 'Fail'
- o_name = '%s_multitool_paint' % self.obj_name
- # use the selected tools in the tool table; get diameters
- tooldia_list = list()
- if self.tools_table.selectedItems():
- for x in self.tools_table.selectedItems():
- try:
- tooldia = float(self.tools_table.item(x.row(), 1).text())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- tooldia = float(self.tools_table.item(x.row(), 1).text().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- continue
- tooldia_list.append(tooldia)
- else:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("No selected tools in Tool Table."))
- return
- if select_method == "all":
- self.paint_poly_all(self.paint_obj,
- tooldia=tooldia_list,
- outname=o_name,
- overlap=overlap,
- connect=connect,
- contour=contour)
- elif select_method == "single":
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Click inside the desired polygon."))
- # use the first tool in the tool table; get the diameter
- # tooldia = float('%.4f' % float(self.tools_table.item(0, 1).text()))
- # To be called after clicking on the plot.
- def doit(event):
- # do paint single only for left mouse clicks
- if event.button == 1:
- self.app.inform.emit(_("Painting polygon..."))
- self.app.plotcanvas.vis_disconnect('mouse_press', doit)
- pos = self.app.plotcanvas.translate_coords(event.pos)
- if self.app.grid_status() == True:
- pos = self.app.geo_editor.snap(pos[0], pos[1])
- self.paint_poly(self.paint_obj,
- inside_pt=[pos[0], pos[1]],
- tooldia=tooldia_list,
- overlap=overlap,
- connect=connect,
- contour=contour)
- self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
- self.app.plotcanvas.vis_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
- self.app.plotcanvas.vis_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_connect('mouse_press', doit)
- elif select_method == "area":
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Click the start point of the paint area."))
- # use the first tool in the tool table; get the diameter
- # tooldia = float('%.4f' % float(self.tools_table.item(0, 1).text()))
- # To be called after clicking on the plot.
- def on_mouse_release(event):
- # do paint single only for left mouse clicks
- if event.button == 1:
- if not self.first_click:
- self.first_click = True
- self.app.inform.emit('[WARNING_NOTCL] %s' %
- _("Click the end point of the paint area."))
- self.cursor_pos = self.app.plotcanvas.translate_coords(event.pos)
- if self.app.grid_status() == True:
- self.cursor_pos = self.app.geo_editor.snap(self.cursor_pos[0], self.cursor_pos[1])
- else:
- self.app.inform.emit(_("Zone added. Click to start adding next zone or right click to finish."))
- self.app.delete_selection_shape()
- curr_pos = self.app.plotcanvas.translate_coords(event.pos)
- if self.app.grid_status() == True:
- curr_pos = self.app.geo_editor.snap(curr_pos[0], curr_pos[1])
- x0, y0 = self.cursor_pos[0], self.cursor_pos[1]
- x1, y1 = curr_pos[0], curr_pos[1]
- pt1 = (x0, y0)
- pt2 = (x1, y0)
- pt3 = (x1, y1)
- pt4 = (x0, y1)
- self.sel_rect.append(Polygon([pt1, pt2, pt3, pt4]))
- self.first_click = False
- return
- # modifiers = QtWidgets.QApplication.keyboardModifiers()
- #
- # if modifiers == QtCore.Qt.ShiftModifier:
- # mod_key = 'Shift'
- # elif modifiers == QtCore.Qt.ControlModifier:
- # mod_key = 'Control'
- # else:
- # mod_key = None
- #
- # if mod_key == self.app.defaults["global_mselect_key"]:
- # self.first_click = False
- # return
- #
- # self.sel_rect = cascaded_union(self.sel_rect)
- # self.paint_poly_area(obj=self.paint_obj,
- # tooldia=tooldia_list,
- # sel_obj= self.sel_rect,
- # outname=o_name,
- # overlap=overlap,
- # connect=connect,
- # contour=contour)
- #
- # self.app.plotcanvas.vis_disconnect('mouse_release', on_mouse_release)
- # self.app.plotcanvas.vis_disconnect('mouse_move', on_mouse_move)
- #
- # self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
- # self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
- # self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
- elif event.button == 2 and self.mouse_is_dragging is False:
- self.first_click = False
- self.app.plotcanvas.vis_disconnect('mouse_release', on_mouse_release)
- self.app.plotcanvas.vis_disconnect('mouse_move', on_mouse_move)
- self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
- self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
- if len(self.sel_rect) == 0:
- return
- self.sel_rect = cascaded_union(self.sel_rect)
- self.paint_poly_area(obj=self.paint_obj,
- tooldia=tooldia_list,
- sel_obj=self.sel_rect,
- outname=o_name,
- overlap=overlap,
- connect=connect,
- contour=contour)
- # called on mouse move
- def on_mouse_move(event):
- curr_pos = self.app.plotcanvas.translate_coords(event.pos)
- self.app.app_cursor.enabled = False
- # detect mouse dragging motion
- if event.is_dragging is True:
- self.mouse_is_dragging = True
- else:
- self.mouse_is_dragging = False
- # update the cursor position
- if self.app.grid_status() == True:
- self.app.app_cursor.enabled = True
- # Update cursor
- curr_pos = self.app.geo_editor.snap(curr_pos[0], curr_pos[1])
- self.app.app_cursor.set_data(np.asarray([(curr_pos[0], curr_pos[1])]),
- symbol='++', edge_color='black', size=20)
- # draw the utility geometry
- if self.first_click:
- self.app.delete_selection_shape()
- self.app.draw_moving_selection_shape(old_coords=(self.cursor_pos[0], self.cursor_pos[1]),
- coords=(curr_pos[0], curr_pos[1]),
- face_alpha=0.0)
- self.app.plotcanvas.vis_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
- self.app.plotcanvas.vis_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
- self.app.plotcanvas.vis_connect('mouse_release', on_mouse_release)
- self.app.plotcanvas.vis_connect('mouse_move', on_mouse_move)
- elif select_method == 'ref':
- self.bound_obj_name = self.box_combo.currentText()
- # Get source object.
- try:
- self.bound_obj = self.app.collection.get_by_name(self.bound_obj_name)
- except Exception as e:
- self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
- (_("Could not retrieve object"),
- self.obj_name))
- return "Could not retrieve object: %s" % self.obj_name
- self.paint_poly_ref(obj=self.paint_obj,
- sel_obj=self.bound_obj,
- tooldia=tooldia_list,
- overlap=overlap,
- outname=o_name,
- connect=connect,
- contour=contour)
- def paint_poly(self, obj,
- inside_pt=None,
- tooldia=None,
- overlap=None,
- order=None,
- margin=None,
- method=None,
- outname=None,
- connect=None,
- contour=None,
- tools_storage=None,
- plot=True,
- run_threaded=True):
- """
- Paints a polygon selected by clicking on its interior or by having a point coordinates given
- Note:
- * The margin is taken directly from the form.
- :param obj: painted object
- :param inside_pt: [x, y]
- :param tooldia: Diameter of the painting tool
- :param overlap: Overlap of the tool between passes.
- :param order: if the tools are ordered and how
- :param margin: a border around painting area
- :param outname: Name of the resulting Geometry Object.
- :param connect: Connect lines to avoid tool lifts.
- :param contour: Paint around the edges.
- :param method: choice out of 'seed', 'normal', 'lines'
- :param tools_storage: whether to use the current tools_storage self.paints_tools or a different one.
- Usage of the different one is related to when this function is called from a TcL command.
- :return: None
- """
- # Which polygon.
- # poly = find_polygon(self.solid_geometry, inside_pt)
- if isinstance(obj, FlatCAMGerber):
- if self.app.defaults["gerber_buffering"] == 'no':
- self.app.inform.emit('%s %s' %
- (_("Paint Tool. Normal painting polygon task started."),
- _("Buffering geometry...")))
- else:
- self.app.inform.emit(_("Paint Tool. Normal painting polygon task started."))
- else:
- self.app.inform.emit(_("Paint Tool. Normal painting polygon task started."))
- if isinstance(obj, FlatCAMGerber):
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- if isinstance(obj.solid_geometry, list):
- obj.solid_geometry = MultiPolygon(obj.solid_geometry).buffer(0)
- else:
- obj.solid_geometry = obj.solid_geometry.buffer(0)
- poly = self.find_polygon(point=inside_pt, geoset=obj.solid_geometry)
- paint_method = method if method is None else self.paintmethod_combo.get_value()
- if margin is not None:
- paint_margin = margin
- else:
- try:
- paint_margin = float(self.paintmargin_entry.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- paint_margin = float(self.paintmargin_entry.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- return
- # determine if to use the progressive plotting
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- prog_plot = True
- else:
- prog_plot = False
- # No polygon?
- if poly is None:
- self.app.log.warning('No polygon found.')
- self.app.inform.emit('[WARNING] %s' % _('No polygon found.'))
- return
- proc = self.app.proc_container.new(_("Painting polygon..."))
- self.app.inform.emit('%s: %s' % (_("Paint Tool. Painting polygon at location"), str(inside_pt)))
- name = outname if outname is not None else self.obj_name + "_paint"
- over = overlap if overlap is not None else float(self.app.defaults["tools_paintoverlap"])
- conn = connect if connect is not None else self.app.defaults["tools_pathconnect"]
- cont = contour if contour is not None else self.app.defaults["tools_paintcontour"]
- order = order if order is not None else self.order_radio.get_value()
- sorted_tools = []
- if tooldia is not None:
- try:
- sorted_tools = [float(eval(dia)) for dia in tooldia.split(",") if dia != '']
- except AttributeError:
- if not isinstance(tooldia, list):
- sorted_tools = [float(tooldia)]
- else:
- sorted_tools = tooldia
- else:
- for row in range(self.tools_table.rowCount()):
- sorted_tools.append(float(self.tools_table.item(row, 1).text()))
- if tools_storage is not None:
- tools_storage = tools_storage
- else:
- tools_storage = self.paint_tools
- # Initializes the new geometry object
- def gen_paintarea(geo_obj, app_obj):
- # assert isinstance(geo_obj, FlatCAMGeometry), \
- # "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
- # assert isinstance(app_obj, App)
- tool_dia = None
- if order == 'fwd':
- sorted_tools.sort(reverse=False)
- elif order == 'rev':
- sorted_tools.sort(reverse=True)
- else:
- pass
- def paint_p(polyg, tooldiameter):
- cpoly = None
- try:
- if paint_method == "seed":
- # Type(cp) == FlatCAMRTreeStorage | None
- cpoly = self.clear_polygon2(polyg,
- tooldia=tooldiameter,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "lines":
- # Type(cp) == FlatCAMRTreeStorage | None
- cpoly = self.clear_polygon3(polyg,
- tooldia=tooldiameter,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- else:
- # Type(cp) == FlatCAMRTreeStorage | None
- cpoly = self.clear_polygon(polyg,
- tooldia=tooldiameter,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- except FlatCAMApp.GracefulException:
- return "fail"
- except Exception as e:
- log.debug("ToolPaint.paint_poly().gen_paintarea().paint_p() --> %s" % str(e))
- if cpoly is not None:
- geo_obj.solid_geometry += list(cpoly.get_objects())
- return cpoly
- else:
- app_obj.inform.emit('[ERROR_NOTCL] %s' %
- _('Geometry could not be painted completely'))
- return None
- try:
- a, b, c, d = poly.bounds
- geo_obj.options['xmin'] = a
- geo_obj.options['ymin'] = b
- geo_obj.options['xmax'] = c
- geo_obj.options['ymax'] = d
- except Exception as e:
- log.debug("ToolPaint.paint_poly.gen_paintarea() bounds error --> %s" % str(e))
- return
- total_geometry = []
- current_uid = int(1)
- geo_obj.solid_geometry = []
- for tool_dia in sorted_tools:
- # find the tooluid associated with the current tool_dia so we know where to add the tool solid_geometry
- for k, v in tools_storage.items():
- if float('%.4f' % v['tooldia']) == float('%.4f' % tool_dia):
- current_uid = int(k)
- break
- try:
- poly_buf = poly.buffer(-paint_margin)
- if isinstance(poly_buf, MultiPolygon):
- cp = []
- for pp in poly_buf:
- cp.append(paint_p(pp, tooldia=tool_dia))
- else:
- cp = paint_p(poly_buf, tooldia=tool_dia)
- if cp is not None:
- if isinstance(cp, list):
- for x in cp:
- total_geometry += list(x.get_objects())
- else:
- total_geometry = list(cp.get_objects())
- except FlatCAMApp.GracefulException:
- return "fail"
- except Exception as e:
- log.debug("Could not Paint the polygons. %s" % str(e))
- app_obj.inform.emit('[ERROR] %s\n%s' %
- (_("Could not do Paint. Try a different combination of parameters. "
- "Or a different strategy of paint"),
- str(e)
- )
- )
- return "fail"
- # add the solid_geometry to the current too in self.paint_tools (tools_storage)
- # dictionary and then reset the temporary list that stored that solid_geometry
- tools_storage[current_uid]['solid_geometry'] = deepcopy(total_geometry)
- tools_storage[current_uid]['data']['name'] = name
- total_geometry[:] = []
- # clean the progressive plotted shapes if it was used
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- self.temp_shapes.clear(update=True)
- # delete tools with empty geometry
- keys_to_delete = []
- # look for keys in the tools_storage dict that have 'solid_geometry' values empty
- for uid in tools_storage:
- # if the solid_geometry (type=list) is empty
- if not tools_storage[uid]['solid_geometry']:
- keys_to_delete.append(uid)
- # actual delete of keys from the tools_storage dict
- for k in keys_to_delete:
- tools_storage.pop(k, None)
- geo_obj.options["cnctooldia"] = str(tool_dia)
- # this turn on the FlatCAMCNCJob plot for multiple tools
- geo_obj.multigeo = True
- geo_obj.multitool = True
- geo_obj.tools.clear()
- geo_obj.tools = dict(tools_storage)
- # test if at least one tool has solid_geometry. If no tool has solid_geometry we raise an Exception
- has_solid_geo = 0
- for tooluid in geo_obj.tools:
- if geo_obj.tools[tooluid]['solid_geometry']:
- has_solid_geo += 1
- if has_solid_geo == 0:
- self.app.inform.emit('[ERROR] %s' %
- _("There is no Painting Geometry in the file.\n"
- "Usually it means that the tool diameter is too big for the painted geometry.\n"
- "Change the painting parameters and try again."))
- return
- self.app.inform.emit('[success] %s' % _("Paint Single Done."))
- # Experimental...
- # print("Indexing...", end=' ')
- # geo_obj.make_index()
- # if errors == 0:
- # print("[success] Paint single polygon Done")
- # self.app.inform.emit("[success] Paint single polygon Done")
- # else:
- # print("[WARNING] Paint single polygon done with errors")
- # self.app.inform.emit("[WARNING] Paint single polygon done with errors. "
- # "%d area(s) could not be painted.\n"
- # "Use different paint parameters or edit the paint geometry and correct"
- # "the issue."
- # % errors)
- def job_thread(app_obj):
- try:
- app_obj.new_object("geometry", name, gen_paintarea, plot=plot)
- except FlatCAMApp.GracefulException:
- proc.done()
- return
- except Exception as e:
- proc.done()
- self.app.inform.emit('[ERROR_NOTCL] %s --> %s' %
- (_('PaintTool.paint_poly()'),
- str(e)))
- return
- proc.done()
- # focus on Selected Tab
- self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
- self.app.inform.emit(_("Polygon Paint started ..."))
- # Promise object with the new name
- self.app.collection.promise(name)
- if run_threaded:
- # Background
- self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
- else:
- job_thread(app_obj=self.app)
- def paint_poly_all(self, obj,
- tooldia=None,
- overlap=None,
- order=None,
- margin=None,
- method=None,
- outname=None,
- connect=None,
- contour=None,
- tools_storage=None,
- plot=True,
- run_threaded=True):
- """
- Paints all polygons in this object.
- :param obj: painted object
- :param tooldia: a tuple or single element made out of diameters of the tools to be used
- :param overlap: value by which the paths will overlap
- :param order: if the tools are ordered and how
- :param margin: a border around painting area
- :param outname: name of the resulting object
- :param connect: Connect lines to avoid tool lifts.
- :param contour: Paint around the edges.
- :param method: choice out of 'seed', 'normal', 'lines'
- :param tools_storage: whether to use the current tools_storage self.paints_tools or a different one.
- Usage of the different one is related to when this function is called from a TcL command.
- :return:
- """
- paint_method = method if method is None else self.paintmethod_combo.get_value()
- if margin is not None:
- paint_margin = margin
- else:
- try:
- paint_margin = float(self.paintmargin_entry.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- paint_margin = float(self.paintmargin_entry.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- return
- # determine if to use the progressive plotting
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- prog_plot = True
- else:
- prog_plot = False
- proc = self.app.proc_container.new(_("Painting polygons..."))
- name = outname if outname is not None else self.obj_name + "_paint"
- over = overlap if overlap is not None else float(self.app.defaults["tools_paintoverlap"])
- conn = connect if connect is not None else self.app.defaults["tools_pathconnect"]
- cont = contour if contour is not None else self.app.defaults["tools_paintcontour"]
- order = order if order is not None else self.order_radio.get_value()
- sorted_tools = []
- if tooldia is not None:
- try:
- sorted_tools = [float(eval(dia)) for dia in tooldia.split(",") if dia != '']
- except AttributeError:
- if not isinstance(tooldia, list):
- sorted_tools = [float(tooldia)]
- else:
- sorted_tools = tooldia
- else:
- for row in range(self.tools_table.rowCount()):
- sorted_tools.append(float(self.tools_table.item(row, 1).text()))
- if tools_storage is not None:
- tools_storage = tools_storage
- else:
- tools_storage = self.paint_tools
- # This is a recursive generator of individual Polygons.
- # Note: Double check correct implementation. Might exit
- # early if it finds something that is not a Polygon?
- # def recurse(geo):
- # try:
- # for subg in geo:
- # for subsubg in recurse(subg):
- # yield subsubg
- # except TypeError:
- # if isinstance(geo, Polygon):
- # yield geo
- #
- # raise StopIteration
- def recurse(geometry, reset=True):
- """
- Creates a list of non-iterable linear geometry objects.
- Results are placed in self.flat_geometry
- :param geometry: Shapely type or list or list of list of such.
- :param reset: Clears the contents of self.flat_geometry.
- """
- if self.app.abort_flag:
- # graceful abort requested by the user
- raise FlatCAMApp.GracefulException
- if geometry is None:
- return
- if reset:
- self.flat_geometry = []
- # ## If iterable, expand recursively.
- try:
- for geo in geometry:
- if geo is not None:
- recurse(geometry=geo, reset=False)
- # ## Not iterable, do the actual indexing and add.
- except TypeError:
- if isinstance(geometry, LinearRing):
- g = Polygon(geometry)
- self.flat_geometry.append(g)
- else:
- self.flat_geometry.append(geometry)
- return self.flat_geometry
- # Initializes the new geometry object
- def gen_paintarea(geo_obj, app_obj):
- # assert isinstance(geo_obj, FlatCAMGeometry), \
- # "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
- log.debug("Paint Tool. Normal painting all task started.")
- if isinstance(obj, FlatCAMGerber):
- if app_obj.defaults["gerber_buffering"] == 'no':
- app_obj.inform.emit('%s %s' %
- (_("Paint Tool. Normal painting all task started."),
- _("Buffering geometry...")))
- else:
- app_obj.inform.emit(_("Paint Tool. Normal painting all task started."))
- else:
- app_obj.inform.emit(_("Paint Tool. Normal painting all task started."))
- tool_dia = None
- if order == 'fwd':
- sorted_tools.sort(reverse=False)
- elif order == 'rev':
- sorted_tools.sort(reverse=True)
- else:
- pass
- if isinstance(obj, FlatCAMGerber):
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- if isinstance(obj.solid_geometry, list):
- obj.solid_geometry = MultiPolygon(obj.solid_geometry).buffer(0)
- else:
- obj.solid_geometry = obj.solid_geometry.buffer(0)
- try:
- a, b, c, d = obj.bounds()
- geo_obj.options['xmin'] = a
- geo_obj.options['ymin'] = b
- geo_obj.options['xmax'] = c
- geo_obj.options['ymax'] = d
- except Exception as e:
- log.debug("ToolPaint.paint_poly.gen_paintarea() bounds error --> %s" % str(e))
- return
- total_geometry = []
- current_uid = int(1)
- geo_obj.solid_geometry = []
- for tool_dia in sorted_tools:
- log.debug("Starting geometry processing for tool: %s" % str(tool_dia))
- app_obj.inform.emit(
- '[success] %s %s%s %s' % (_('Painting with tool diameter = '),
- str(tool_dia),
- self.units.lower(),
- _('started'))
- )
- app_obj.proc_container.update_view_text(' %d%%' % 0)
- # find the tooluid associated with the current tool_dia so we know where to add the tool solid_geometry
- for k, v in tools_storage.items():
- if float('%.4f' % v['tooldia']) == float('%.4f' % tool_dia):
- current_uid = int(k)
- break
- painted_area = recurse(obj.solid_geometry)
- # variables to display the percentage of work done
- geo_len = len(painted_area)
- old_disp_number = 0
- log.warning("Total number of polygons to be cleared. %s" % str(geo_len))
- pol_nr = 0
- for geo in painted_area:
- try:
- # Polygons are the only really paintable geometries, lines in theory have no area to be painted
- if not isinstance(geo, Polygon):
- continue
- poly_buf = geo.buffer(-paint_margin)
- if paint_method == "seed":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon2(poly_buf,
- tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "lines":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon3(poly_buf,
- tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- else:
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon(poly_buf,
- tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- if cp is not None:
- total_geometry += list(cp.get_objects())
- except FlatCAMApp.GracefulException:
- return "fail"
- except Exception as e:
- log.debug("Could not Paint the polygons. %s" % str(e))
- self.app.inform.emit('[ERROR] %s\n%s' %
- (_("Could not do Paint All. Try a different combination of parameters. "
- "Or a different Method of paint"),
- str(e)))
- return "fail"
- pol_nr += 1
- disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
- # log.debug("Polygons cleared: %d" % pol_nr)
- if old_disp_number < disp_number <= 100:
- app_obj.proc_container.update_view_text(' %d%%' % disp_number)
- old_disp_number = disp_number
- # log.debug("Polygons cleared: %d. Percentage done: %d%%" % (pol_nr, disp_number))
- # add the solid_geometry to the current too in self.paint_tools (tools_storage)
- # dictionary and then reset the temporary list that stored that solid_geometry
- tools_storage[current_uid]['solid_geometry'] = deepcopy(total_geometry)
- tools_storage[current_uid]['data']['name'] = name
- total_geometry[:] = []
- # clean the progressive plotted shapes if it was used
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- self.temp_shapes.clear(update=True)
- # delete tools with empty geometry
- keys_to_delete = []
- # look for keys in the tools_storage dict that have 'solid_geometry' values empty
- for uid in tools_storage:
- # if the solid_geometry (type=list) is empty
- if not tools_storage[uid]['solid_geometry']:
- keys_to_delete.append(uid)
- # actual delete of keys from the tools_storage dict
- for k in keys_to_delete:
- tools_storage.pop(k, None)
- geo_obj.options["cnctooldia"] = str(tool_dia)
- # this turn on the FlatCAMCNCJob plot for multiple tools
- geo_obj.multigeo = True
- geo_obj.multitool = True
- geo_obj.tools.clear()
- geo_obj.tools = dict(tools_storage)
- # test if at least one tool has solid_geometry. If no tool has solid_geometry we raise an Exception
- has_solid_geo = 0
- for tooluid in geo_obj.tools:
- if geo_obj.tools[tooluid]['solid_geometry']:
- has_solid_geo += 1
- if has_solid_geo == 0:
- self.app.inform.emit('[ERROR] %s' %
- _("There is no Painting Geometry in the file.\n"
- "Usually it means that the tool diameter is too big for the painted geometry.\n"
- "Change the painting parameters and try again."))
- return
- # Experimental...
- # print("Indexing...", end=' ')
- # geo_obj.make_index()
- self.app.inform.emit('[success] %s' % _("Paint All Done."))
- # Initializes the new geometry object
- def gen_paintarea_rest_machining(geo_obj, app_obj):
- assert isinstance(geo_obj, FlatCAMGeometry), \
- "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
- log.debug("Paint Tool. Rest machining painting all task started.")
- if isinstance(obj, FlatCAMGerber):
- if app_obj.defaults["gerber_buffering"] == 'no':
- app_obj.inform.emit('%s %s' %
- (_("Paint Tool. Rest machining painting all task started."),
- _("Buffering geometry...")))
- else:
- app_obj.inform.emit(_("Paint Tool. Rest machining painting all task started."))
- else:
- app_obj.inform.emit(_("Paint Tool. Rest machining painting all task started."))
- tool_dia = None
- sorted_tools.sort(reverse=True)
- cleared_geo = []
- current_uid = int(1)
- geo_obj.solid_geometry = []
- if isinstance(obj, FlatCAMGerber):
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- if isinstance(obj.solid_geometry, list):
- obj.solid_geometry = MultiPolygon(obj.solid_geometry).buffer(0)
- else:
- obj.solid_geometry = obj.solid_geometry.buffer(0)
- try:
- a, b, c, d = obj.bounds()
- geo_obj.options['xmin'] = a
- geo_obj.options['ymin'] = b
- geo_obj.options['xmax'] = c
- geo_obj.options['ymax'] = d
- except Exception as e:
- log.debug("ToolPaint.paint_poly.gen_paintarea() bounds error --> %s" % str(e))
- return
- for tool_dia in sorted_tools:
- log.debug("Starting geometry processing for tool: %s" % str(tool_dia))
- app_obj.inform.emit(
- '[success] %s %s%s %s' % (_('Painting with tool diameter = '),
- str(tool_dia),
- self.units.lower(),
- _('started'))
- )
- app_obj.proc_container.update_view_text(' %d%%' % 0)
- painted_area = recurse(obj.solid_geometry)
- # variables to display the percentage of work done
- geo_len = int(len(painted_area) / 100)
- old_disp_number = 0
- log.warning("Total number of polygons to be cleared. %s" % str(geo_len))
- pol_nr = 0
- for geo in painted_area:
- try:
- geo = Polygon(geo) if not isinstance(geo, Polygon) else geo
- poly_buf = geo.buffer(-paint_margin)
- cp = None
- if paint_method == "standard":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon(poly_buf, tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over, contour=cont, connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "seed":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon2(poly_buf, tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over, contour=cont, connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "lines":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon3(poly_buf, tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over, contour=cont, connect=conn,
- prog_plot=prog_plot)
- if cp is not None:
- cleared_geo += list(cp.get_objects())
- except FlatCAMApp.GracefulException:
- return "fail"
- except Exception as e:
- log.debug("Could not Paint the polygons. %s" % str(e))
- self.app.inform.emit('[ERROR] %s\n%s' %
- (_("Could not do Paint All. Try a different combination of parameters. "
- "Or a different Method of paint"),
- str(e)))
- return "fail"
- pol_nr += 1
- disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
- # log.debug("Polygons cleared: %d" % pol_nr)
- if old_disp_number < disp_number <= 100:
- app_obj.proc_container.update_view_text(' %d%%' % disp_number)
- old_disp_number = disp_number
- # log.debug("Polygons cleared: %d. Percentage done: %d%%" % (pol_nr, disp_number))
- # find the tooluid associated with the current tool_dia so we know where to add the tool solid_geometry
- for k, v in tools_storage.items():
- if float('%.4f' % v['tooldia']) == float('%.4f' % tool_dia):
- current_uid = int(k)
- break
- # add the solid_geometry to the current too in self.paint_tools (or tools_storage) dictionary and
- # then reset the temporary list that stored that solid_geometry
- tools_storage[current_uid]['solid_geometry'] = deepcopy(cleared_geo)
- tools_storage[current_uid]['data']['name'] = name
- cleared_geo[:] = []
- geo_obj.options["cnctooldia"] = str(tool_dia)
- # this turn on the FlatCAMCNCJob plot for multiple tools
- geo_obj.multigeo = True
- geo_obj.multitool = True
- geo_obj.tools.clear()
- geo_obj.tools = dict(tools_storage)
- # clean the progressive plotted shapes if it was used
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- self.temp_shapes.clear(update=True)
- # test if at least one tool has solid_geometry. If no tool has solid_geometry we raise an Exception
- has_solid_geo = 0
- for tooluid in geo_obj.tools:
- if geo_obj.tools[tooluid]['solid_geometry']:
- has_solid_geo += 1
- if has_solid_geo == 0:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("There is no Painting Geometry in the file.\n"
- "Usually it means that the tool diameter is too big for the painted geometry.\n"
- "Change the painting parameters and try again."))
- return
- # Experimental...
- # print("Indexing...", end=' ')
- # geo_obj.make_index()
- self.app.inform.emit('[success] %s' % _("Paint All with Rest-Machining done."))
- def job_thread(app_obj):
- try:
- if self.rest_cb.isChecked():
- app_obj.new_object("geometry", name, gen_paintarea_rest_machining, plot=plot)
- else:
- app_obj.new_object("geometry", name, gen_paintarea, plot=plot)
- except FlatCAMApp.GracefulException:
- proc.done()
- return
- except Exception as e:
- proc.done()
- traceback.print_stack()
- return
- proc.done()
- # focus on Selected Tab
- self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
- self.app.inform.emit(_("Polygon Paint started ..."))
- # Promise object with the new name
- self.app.collection.promise(name)
- if run_threaded:
- # Background
- self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
- else:
- job_thread(app_obj=self.app)
- def paint_poly_area(self, obj, sel_obj,
- tooldia=None,
- overlap=None,
- order=None,
- margin=None,
- method=None,
- outname=None,
- connect=None,
- contour=None,
- tools_storage=None,
- plot=True,
- run_threaded=True):
- """
- Paints all polygons in this object that are within the sel_obj object
- :param obj: painted object
- :param sel_obj: paint only what is inside this object bounds
- :param tooldia: a tuple or single element made out of diameters of the tools to be used
- :param overlap: value by which the paths will overlap
- :param order: if the tools are ordered and how
- :param margin: a border around painting area
- :param outname: name of the resulting object
- :param connect: Connect lines to avoid tool lifts.
- :param contour: Paint around the edges.
- :param method: choice out of 'seed', 'normal', 'lines'
- :param tools_storage: whether to use the current tools_storage self.paints_tools or a different one.
- Usage of the different one is related to when this function is called from a TcL command.
- :return:
- """
- paint_method = method if method is None else self.paintmethod_combo.get_value()
- if margin is not None:
- paint_margin = margin
- else:
- try:
- paint_margin = float(self.paintmargin_entry.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- paint_margin = float(self.paintmargin_entry.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("Wrong value format entered, use a number."))
- return
- # determine if to use the progressive plotting
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- prog_plot = True
- else:
- prog_plot = False
- proc = self.app.proc_container.new(_("Painting polygons..."))
- name = outname if outname is not None else self.obj_name + "_paint"
- over = overlap if overlap is not None else float(self.app.defaults["tools_paintoverlap"])
- conn = connect if connect is not None else self.app.defaults["tools_pathconnect"]
- cont = contour if contour is not None else self.app.defaults["tools_paintcontour"]
- order = order if order is not None else self.order_radio.get_value()
- sorted_tools = []
- if tooldia is not None:
- try:
- sorted_tools = [float(eval(dia)) for dia in tooldia.split(",") if dia != '']
- except AttributeError:
- if not isinstance(tooldia, list):
- sorted_tools = [float(tooldia)]
- else:
- sorted_tools = tooldia
- else:
- for row in range(self.tools_table.rowCount()):
- sorted_tools.append(float(self.tools_table.item(row, 1).text()))
- if tools_storage is not None:
- tools_storage = tools_storage
- else:
- tools_storage = self.paint_tools
- def recurse(geometry, reset=True):
- """
- Creates a list of non-iterable linear geometry objects.
- Results are placed in self.flat_geometry
- :param geometry: Shapely type or list or list of list of such.
- :param reset: Clears the contents of self.flat_geometry.
- """
- if self.app.abort_flag:
- # graceful abort requested by the user
- raise FlatCAMApp.GracefulException
- if geometry is None:
- return
- if reset:
- self.flat_geometry = []
- # ## If iterable, expand recursively.
- try:
- for geo in geometry:
- if geo is not None:
- recurse(geometry=geo, reset=False)
- # ## Not iterable, do the actual indexing and add.
- except TypeError:
- if isinstance(geometry, LinearRing):
- g = Polygon(geometry)
- self.flat_geometry.append(g)
- else:
- self.flat_geometry.append(geometry)
- return self.flat_geometry
- # Initializes the new geometry object
- def gen_paintarea(geo_obj, app_obj):
- # assert isinstance(geo_obj, FlatCAMGeometry), \
- # "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
- log.debug("Paint Tool. Normal painting area task started.")
- if isinstance(obj, FlatCAMGerber):
- if app_obj.defaults["gerber_buffering"] == 'no':
- app_obj.inform.emit('%s %s' %
- (_("Paint Tool. Normal painting area task started."),
- _("Buffering geometry...")))
- else:
- app_obj.inform.emit(_("Paint Tool. Normal painting area task started."))
- else:
- app_obj.inform.emit(_("Paint Tool. Normal painting area task started."))
- tool_dia = None
- if order == 'fwd':
- sorted_tools.sort(reverse=False)
- elif order == 'rev':
- sorted_tools.sort(reverse=True)
- else:
- pass
- # this is were heavy lifting is done and creating the geometry to be painted
- target_geo = MultiPolygon(obj.solid_geometry)
- if isinstance(obj, FlatCAMGerber):
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- if isinstance(target_geo, list):
- target_geo = MultiPolygon(target_geo).buffer(0)
- else:
- target_geo = target_geo.buffer(0)
- geo_to_paint = target_geo.intersection(sel_obj)
- painted_area = recurse(geo_to_paint)
- try:
- a, b, c, d = self.paint_bounds(geo_to_paint)
- geo_obj.options['xmin'] = a
- geo_obj.options['ymin'] = b
- geo_obj.options['xmax'] = c
- geo_obj.options['ymax'] = d
- except Exception as e:
- log.debug("ToolPaint.paint_poly.gen_paintarea() bounds error --> %s" % str(e))
- return
- total_geometry = []
- current_uid = int(1)
- geo_obj.solid_geometry = []
- for tool_dia in sorted_tools:
- log.debug("Starting geometry processing for tool: %s" % str(tool_dia))
- app_obj.inform.emit(
- '[success] %s %s%s %s' % (_('Painting with tool diameter = '),
- str(tool_dia),
- self.units.lower(),
- _('started'))
- )
- app_obj.proc_container.update_view_text(' %d%%' % 0)
- # find the tooluid associated with the current tool_dia so we know where to add the tool solid_geometry
- for k, v in tools_storage.items():
- if float('%.4f' % v['tooldia']) == float('%.4f' % tool_dia):
- current_uid = int(k)
- break
- # variables to display the percentage of work done
- geo_len = len(painted_area)
- old_disp_number = 0
- log.warning("Total number of polygons to be cleared. %s" % str(geo_len))
- pol_nr = 0
- for geo in painted_area:
- try:
- # Polygons are the only really paintable geometries, lines in theory have no area to be painted
- if not isinstance(geo, Polygon):
- continue
- poly_buf = geo.buffer(-paint_margin)
- if paint_method == "seed":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon2(poly_buf,
- tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "lines":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon3(poly_buf,
- tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- else:
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon(poly_buf,
- tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over,
- contour=cont,
- connect=conn,
- prog_plot=prog_plot)
- if cp is not None:
- total_geometry += list(cp.get_objects())
- except FlatCAMApp.GracefulException:
- return "fail"
- except Exception as e:
- log.debug("Could not Paint the polygons. %s" % str(e))
- self.app.inform.emit('[ERROR] %s\n%s' %
- (_("Could not do Paint All. Try a different combination of parameters. "
- "Or a different Method of paint"), str(e)))
- return
- pol_nr += 1
- disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
- # log.debug("Polygons cleared: %d" % pol_nr)
- if old_disp_number < disp_number <= 100:
- app_obj.proc_container.update_view_text(' %d%%' % disp_number)
- old_disp_number = disp_number
- # log.debug("Polygons cleared: %d. Percentage done: %d%%" % (pol_nr, disp_number))
- # add the solid_geometry to the current too in self.paint_tools (tools_storage)
- # dictionary and then reset the temporary list that stored that solid_geometry
- tools_storage[current_uid]['solid_geometry'] = deepcopy(total_geometry)
- tools_storage[current_uid]['data']['name'] = name
- total_geometry[:] = []
- # clean the progressive plotted shapes if it was used
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- self.temp_shapes.clear(update=True)
- # delete tools with empty geometry
- keys_to_delete = []
- # look for keys in the tools_storage dict that have 'solid_geometry' values empty
- for uid in tools_storage:
- # if the solid_geometry (type=list) is empty
- if not tools_storage[uid]['solid_geometry']:
- keys_to_delete.append(uid)
- # actual delete of keys from the tools_storage dict
- for k in keys_to_delete:
- tools_storage.pop(k, None)
- geo_obj.options["cnctooldia"] = str(tool_dia)
- # this turn on the FlatCAMCNCJob plot for multiple tools
- geo_obj.multigeo = True
- geo_obj.multitool = True
- geo_obj.tools.clear()
- geo_obj.tools = dict(tools_storage)
- # test if at least one tool has solid_geometry. If no tool has solid_geometry we raise an Exception
- has_solid_geo = 0
- for tooluid in geo_obj.tools:
- if geo_obj.tools[tooluid]['solid_geometry']:
- has_solid_geo += 1
- if has_solid_geo == 0:
- self.app.inform.emit('[ERROR] %s' %
- _("There is no Painting Geometry in the file.\n"
- "Usually it means that the tool diameter is too big for the painted geometry.\n"
- "Change the painting parameters and try again."))
- return
- # Experimental...
- # print("Indexing...", end=' ')
- # geo_obj.make_index()
- self.app.inform.emit('[success] %s' % _("Paint Area Done."))
- # Initializes the new geometry object
- def gen_paintarea_rest_machining(geo_obj, app_obj):
- assert isinstance(geo_obj, FlatCAMGeometry), \
- "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
- log.debug("Paint Tool. Rest machining painting area task started.")
- if isinstance(obj, FlatCAMGerber):
- if app_obj.defaults["gerber_buffering"] == 'no':
- app_obj.inform.emit('%s %s' %
- (_("Paint Tool. Rest machining painting area task started."),
- _("Buffering geometry...")))
- else:
- app_obj.inform.emit(_("Paint Tool. Rest machining painting area task started."))
- else:
- app_obj.inform.emit(_("Paint Tool. Rest machining painting area task started."))
- tool_dia = None
- sorted_tools.sort(reverse=True)
- cleared_geo = []
- current_uid = int(1)
- geo_obj.solid_geometry = []
- # this is were heavy lifting is done and creating the geometry to be painted
- target_geo = obj.solid_geometry
- if isinstance(obj, FlatCAMGerber):
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- if isinstance(target_geo, list):
- target_geo = MultiPolygon(target_geo).buffer(0)
- else:
- target_geo = target_geo.buffer(0)
- geo_to_paint = target_geo.intersection(sel_obj)
- painted_area = recurse(geo_to_paint)
- try:
- a, b, c, d = obj.bounds()
- geo_obj.options['xmin'] = a
- geo_obj.options['ymin'] = b
- geo_obj.options['xmax'] = c
- geo_obj.options['ymax'] = d
- except Exception as e:
- log.debug("ToolPaint.paint_poly.gen_paintarea() bounds error --> %s" % str(e))
- return
- for tool_dia in sorted_tools:
- log.debug("Starting geometry processing for tool: %s" % str(tool_dia))
- app_obj.inform.emit(
- '[success] %s %s%s %s' % (_('Painting with tool diameter = '),
- str(tool_dia),
- self.units.lower(),
- _('started'))
- )
- app_obj.proc_container.update_view_text(' %d%%' % 0)
- # variables to display the percentage of work done
- geo_len = len(painted_area)
- old_disp_number = 0
- log.warning("Total number of polygons to be cleared. %s" % str(geo_len))
- pol_nr = 0
- for geo in painted_area:
- try:
- geo = Polygon(geo) if not isinstance(geo, Polygon) else geo
- poly_buf = geo.buffer(-paint_margin)
- cp = None
- if paint_method == "standard":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon(poly_buf, tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over, contour=cont, connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "seed":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon2(poly_buf, tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over, contour=cont, connect=conn,
- prog_plot=prog_plot)
- elif paint_method == "lines":
- # Type(cp) == FlatCAMRTreeStorage | None
- cp = self.clear_polygon3(poly_buf, tooldia=tool_dia,
- steps_per_circle=self.app.defaults["geometry_circle_steps"],
- overlap=over, contour=cont, connect=conn,
- prog_plot=prog_plot)
- if cp is not None:
- cleared_geo += list(cp.get_objects())
- except FlatCAMApp.GracefulException:
- return "fail"
- except Exception as e:
- log.debug("Could not Paint the polygons. %s" % str(e))
- self.app.inform.emit('[ERROR] %s\n%s' %
- (_("Could not do Paint All. Try a different combination of parameters. "
- "Or a different Method of paint"), str(e)))
- return
- pol_nr += 1
- disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
- # log.debug("Polygons cleared: %d" % pol_nr)
- if old_disp_number < disp_number <= 100:
- app_obj.proc_container.update_view_text(' %d%%' % disp_number)
- old_disp_number = disp_number
- # log.debug("Polygons cleared: %d. Percentage done: %d%%" % (pol_nr, disp_number))
- # find the tooluid associated with the current tool_dia so we know where to add the tool solid_geometry
- for k, v in tools_storage.items():
- if float('%.4f' % v['tooldia']) == float('%.4f' % tool_dia):
- current_uid = int(k)
- break
- # add the solid_geometry to the current too in self.paint_tools (or tools_storage) dictionary and
- # then reset the temporary list that stored that solid_geometry
- tools_storage[current_uid]['solid_geometry'] = deepcopy(cleared_geo)
- tools_storage[current_uid]['data']['name'] = name
- cleared_geo[:] = []
- geo_obj.options["cnctooldia"] = str(tool_dia)
- # this turn on the FlatCAMCNCJob plot for multiple tools
- geo_obj.multigeo = True
- geo_obj.multitool = True
- geo_obj.tools.clear()
- geo_obj.tools = dict(self.paint_tools)
- # clean the progressive plotted shapes if it was used
- if self.app.defaults["tools_paint_plotting"] == 'progressive':
- self.temp_shapes.clear(update=True)
- # test if at least one tool has solid_geometry. If no tool has solid_geometry we raise an Exception
- has_solid_geo = 0
- for tooluid in geo_obj.tools:
- if geo_obj.tools[tooluid]['solid_geometry']:
- has_solid_geo += 1
- if has_solid_geo == 0:
- self.app.inform.emit('[ERROR_NOTCL] %s' %
- _("There is no Painting Geometry in the file.\n"
- "Usually it means that the tool diameter is too big for the painted geometry.\n"
- "Change the painting parameters and try again."))
- return
- # Experimental...
- # print("Indexing...", end=' ')
- # geo_obj.make_index()
- self.app.inform.emit('[success] %s' % _("Paint All with Rest-Machining done."))
- def job_thread(app_obj):
- try:
- if self.rest_cb.isChecked():
- app_obj.new_object("geometry", name, gen_paintarea_rest_machining, plot=plot)
- else:
- app_obj.new_object("geometry", name, gen_paintarea, plot=plot)
- except FlatCAMApp.GracefulException:
- proc.done()
- return
- except Exception as e:
- proc.done()
- traceback.print_stack()
- return
- proc.done()
- # focus on Selected Tab
- self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
- self.app.inform.emit(_("Polygon Paint started ..."))
- # Promise object with the new name
- self.app.collection.promise(name)
- if run_threaded:
- # Background
- self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
- else:
- job_thread(app_obj=self.app)
- def paint_poly_ref(self, obj, sel_obj,
- tooldia=None,
- overlap=None,
- order=None,
- margin=None,
- method=None,
- outname=None,
- connect=None,
- contour=None,
- tools_storage=None,
- plot=True,
- run_threaded=True):
- """
- Paints all polygons in this object that are within the sel_obj object
- :param obj: painted object
- :param sel_obj: paint only what is inside this object bounds
- :param tooldia: a tuple or single element made out of diameters of the tools to be used
- :param overlap: value by which the paths will overlap
- :param order: if the tools are ordered and how
- :param margin: a border around painting area
- :param outname: name of the resulting object
- :param connect: Connect lines to avoid tool lifts.
- :param contour: Paint around the edges.
- :param method: choice out of 'seed', 'normal', 'lines'
- :param tools_storage: whether to use the current tools_storage self.paints_tools or a different one.
- Usage of the different one is related to when this function is called from a TcL command.
- :return:
- """
- geo = sel_obj.solid_geometry
- try:
- if isinstance(geo, MultiPolygon):
- env_obj = geo.convex_hull
- elif (isinstance(geo, MultiPolygon) and len(geo) == 1) or \
- (isinstance(geo, list) and len(geo) == 1) and isinstance(geo[0], Polygon):
- env_obj = cascaded_union(self.bound_obj.solid_geometry)
- else:
- env_obj = cascaded_union(self.bound_obj.solid_geometry)
- env_obj = env_obj.convex_hull
- sel_rect = env_obj.buffer(distance=0.0000001, join_style=base.JOIN_STYLE.mitre)
- except Exception as e:
- log.debug("ToolPaint.on_paint_button_click() --> %s" % str(e))
- self.app.inform.emit('[ERROR_NOTCL] %s' % _("No object available."))
- return
- self.paint_poly_area(obj=obj,
- sel_obj=sel_rect,
- tooldia=tooldia,
- overlap=overlap,
- order=order,
- margin=margin,
- method=method,
- outname=outname,
- connect=connect,
- contour=contour,
- tools_storage=tools_storage,
- plot=plot,
- run_threaded=run_threaded)
- @staticmethod
- def paint_bounds(geometry):
- def bounds_rec(o):
- if type(o) is list:
- minx = Inf
- miny = Inf
- maxx = -Inf
- maxy = -Inf
- for k in o:
- try:
- minx_, miny_, maxx_, maxy_ = bounds_rec(k)
- except Exception as e:
- log.debug("ToolPaint.bounds() --> %s" % str(e))
- return
- minx = min(minx, minx_)
- miny = min(miny, miny_)
- maxx = max(maxx, maxx_)
- maxy = max(maxy, maxy_)
- return minx, miny, maxx, maxy
- else:
- # it's a Shapely object, return it's bounds
- return o.bounds
- return bounds_rec(geometry)
- def reset_fields(self):
- self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|