Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 7 additions & 41 deletions cq_editor/main_window.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,6 @@
import sys

from typing import Optional
from PyQt5.QtCore import pyqtSlot
from PyQt5.QtWidgets import (QApplication, QLabel, QMainWindow, QToolBar, QDockWidget, QAction)
from PyQt5.QtWidgets import (QLabel, QMainWindow, QToolBar, QDockWidget, QAction)
from logbook import Logger
import cadquery as cq

Expand Down Expand Up @@ -67,8 +65,6 @@ def __init__(self,parent=None, filename=None):

self.restoreComponentState()

self.on_idle()

def closeEvent(self,event):

self.saveWindow()
Expand Down Expand Up @@ -206,7 +202,7 @@ def prepare_toolbar(self):
self.toolbar = QToolBar('Main toolbar',self,objectName='Main toolbar')

for c in self.components.values():
add_actions(self.toolbar, c.toolbarActions())
add_actions(self.toolbar,c.toolbarActions())

self.addToolBar(self.toolbar)

Expand All @@ -217,25 +213,18 @@ def prepare_statusbar(self):

def prepare_actions(self):

self.components['debugger'].sigRenderStarted \
.connect(self.on_render_start)
self.components['debugger'].sigRendered\
.connect(self.components['object_tree'].addObjects)
self.components['debugger'].sigTraceback\
.connect(self.components['traceback_viewer'].addTraceback)
self.components['debugger'].sigRendered \
.connect(lambda _: self.on_idle())
self.components['debugger'].sigTraceback \
.connect(lambda _: self.on_idle())

self.components['debugger'].sigLocals\
.connect(self.components['variables_viewer'].update_frame)
self.components['debugger'].sigLocals\
.connect(self.components['console'].push_vars)

self.components['object_tree'].sigObjectsAdded[list, list]\
.connect(lambda objects, names: self.components['viewer'].display_many(objects, None, names))
self.components['object_tree'].sigObjectsAdded[list, bool, list]\
self.components['object_tree'].sigObjectsAdded[list]\
.connect(self.components['viewer'].display_many)
self.components['object_tree'].sigObjectsAdded[list,bool]\
.connect(self.components['viewer'].display_many)
self.components['object_tree'].sigItemChanged.\
connect(self.components['viewer'].update_item)
Expand All @@ -250,8 +239,6 @@ def prepare_actions(self):

self.components['viewer'].sigObjectSelected\
.connect(self.components['object_tree'].handleGraphicalSelection)
self.components['viewer'].sigDisplayProgress \
.connect(self.on_display_progress)

self.components['traceback_viewer'].sigHighlightLine\
.connect(self.components['editor'].go_to_line)
Expand Down Expand Up @@ -357,27 +344,6 @@ def handle_filename_change(self, fname):
new_title = fname if fname else "*"
self.setWindowTitle(f"{self.name}: {new_title}")

def on_idle(self):
self.components['debugger'].set_rendering_state(False)
self.set_status_message('Idle', '#000000')
if __name__ == "__main__":

@pyqtSlot()
def on_render_start(self):
self.components['debugger'].set_rendering_state(True)
self.set_status_message('Rendering...', '#ff0000')

@pyqtSlot(int, int, str)
def on_display_progress(self, current: int, total: int, name: Optional[str]):
if current == total:
self.on_idle()
else:
message = f'Displaying Shape {current + 1} / {total}'
if name:
message += f' ({name})'
self.set_status_message(message, '#0000ff')

def set_status_message(self, message: str, color: str):
self.statusBar().showMessage(message)
self.statusBar().setStyleSheet(f'color: {color}')
# required because rendering is currently done on the main thread
QApplication.processEvents()
pass
7 changes: 0 additions & 7 deletions cq_editor/widgets/debugger.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,6 @@ class Debugger(QObject,ComponentMixin):
])


sigRenderStarted = pyqtSignal()
sigRendered = pyqtSignal(dict)
sigLocals = pyqtSignal(dict)
sigTraceback = pyqtSignal(object,str)
Expand Down Expand Up @@ -264,7 +263,6 @@ def _cleanup_locals(self,module,injected_names):

@pyqtSlot(bool)
def render(self):
self.sigRenderStarted.emit()

seed(59798267586177)
if self.preferences['Reload CQ']:
Expand Down Expand Up @@ -296,11 +294,6 @@ def render(self):
sys.last_traceback = exc_info[-1]
self.sigTraceback.emit(exc_info, cq_script)

def set_rendering_state(self, rendering):
render_action = self._actions['Run'][0]
render_action.setCheckable(rendering)
render_action.setChecked(rendering)

@property
def breakpoints(self):
return [ el[0] for el in self.get_breakpoints()]
Expand Down
20 changes: 8 additions & 12 deletions cq_editor/widgets/object_tree.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ class ObjectTree(QWidget,ComponentMixin):
{'name': 'Clear all before each run', 'type': 'bool', 'value': True},
{'name': 'STL precision','type': 'float', 'value': .1}])

sigObjectsAdded = pyqtSignal([list, list],[list, bool, list])
sigObjectsAdded = pyqtSignal([list],[list,bool])
sigObjectsRemoved = pyqtSignal(list)
sigCQObjectSelected = pyqtSignal(object)
sigAISObjectsSelected = pyqtSignal(list)
Expand Down Expand Up @@ -201,7 +201,6 @@ def addLines(self):

origin = (0,0,0)
ais_list = []
names = []

for name,color,direction in zip(('X','Y','Z'),
('red','lawngreen','blue'),
Expand All @@ -215,9 +214,8 @@ def addLines(self):
ais=line))

ais_list.append(line)
names.append(name)

self.sigObjectsAdded.emit(ais_list, names)
self.sigObjectsAdded.emit(ais_list)

def _current_properties(self):

Expand Down Expand Up @@ -250,7 +248,6 @@ def addObjects(self,objects,clean=False,root=None):
self.removeObjects()

ais_list = []
names = []

#remove empty objects
objects_f = {k:v for k,v in objects.items() if not is_obj_empty(v.shape)}
Expand All @@ -269,14 +266,13 @@ def addObjects(self,objects,clean=False,root=None):

if child.properties['Visible']:
ais_list.append(ais)
names.append(name)


root.addChild(child)

if request_fit_view:
self.sigObjectsAdded[list, bool, list].emit(ais_list, True, names)
self.sigObjectsAdded[list,bool].emit(ais_list,True)
else:
self.sigObjectsAdded[list, list].emit(ais_list, names)
self.sigObjectsAdded[list].emit(ais_list)

@pyqtSlot(object,str,object)
def addObject(self,obj,name='',options=None):
Expand All @@ -285,15 +281,15 @@ def addObject(self,obj,name='',options=None):

root = self.CQ

ais, shape_display = make_AIS(obj, options)
ais,shape_display = make_AIS(obj, options)

root.addChild(ObjectTreeItem(name,
shape=obj,
shape_display=shape_display,
ais=ais,
sig=self.sigObjectPropertiesChanged))

self.sigObjectsAdded.emit([ais], [name])
self.sigObjectsAdded.emit([ais])

@pyqtSlot(list)
@pyqtSlot()
Expand All @@ -317,7 +313,7 @@ def stashObjects(self,action : bool):
self.removeObjects()
self.CQ.addChildren(self._stash)
ais_list = [el.ais for el in self._stash]
self.sigObjectsAdded.emit(ais_list, [''] * len(ais_list))
self.sigObjectsAdded.emit(ais_list)

@pyqtSlot()
def removeSelected(self):
Expand Down
3 changes: 2 additions & 1 deletion cq_editor/widgets/traceback_viewer.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,8 @@ def __init__(self,parent):

self.tree = TracebackTree(self)
self.current_exception = QLabel(self)
self.current_exception.setStyleSheet("QLabel {color : red; }");
self.current_exception.setStyleSheet(\
"QLabel {color : red; }");

layout(self,
(self.current_exception,
Expand Down
37 changes: 21 additions & 16 deletions cq_editor/widgets/viewer.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from PyQt5.QtWidgets import QWidget, QDialog, QTreeWidgetItem, QApplication, QAction

from typing import Optional, List
from PyQt5.QtCore import pyqtSlot, pyqtSignal
from PyQt5.QtGui import QIcon

Expand Down Expand Up @@ -48,7 +47,6 @@ class OCCViewer(QWidget,ComponentMixin):
IMAGE_EXTENSIONS = 'png'

sigObjectSelected = pyqtSignal(list)
sigDisplayProgress = pyqtSignal(int, int, str)

def __init__(self,parent=None):

Expand Down Expand Up @@ -181,23 +179,31 @@ def clear(self):
context.PurgeDisplay()
context.RemoveAll(True)

def _display(self,shape):

ais = make_AIS(shape)
self.canvas.context.Display(shape,True)

self.displayed_shapes.append(shape)
self.displayed_ais.append(ais)

#self.canvas._display.Repaint()

@pyqtSlot(object)
def display(self, ais):
self.display_many([ais])
def display(self,ais):

context = self._get_context()
context.Display(ais,True)

if self.preferences['Fit automatically']: self.fit()

@pyqtSlot(list)
@pyqtSlot(list, bool, list)
def display_many(self, ais_list, fit: Optional[bool] = None, names: Optional[List] = None):
if names is None:
names = [None] * len(ais_list)
assert len(ais_list) == len(names)
@pyqtSlot(list,bool)
def display_many(self,ais_list,fit=None):

context = self._get_context()
num_objects = len(ais_list)
for i, (ais, name) in enumerate(zip(ais_list, names)):
self.sigDisplayProgress.emit(i, num_objects, name)
context.Display(ais, True)
self.sigDisplayProgress.emit(num_objects, num_objects, None)
for ais in ais_list:
context.Display(ais,True)

if self.preferences['Fit automatically'] and fit is None:
self.fit()
Expand All @@ -217,8 +223,7 @@ def update_item(self,item,col):
def remove_items(self,ais_items):

ctx = self._get_context()
for ais in ais_items:
ctx.Erase(ais,True)
for ais in ais_items: ctx.Erase(ais,True)

@pyqtSlot()
def redraw(self):
Expand Down
Loading