mirror of
https://github.com/Denvi/FlatCAM.git
synced 2026-03-03 06:54:19 +01:00
0f9628803d
Pool object belongs to FlatCAMApp, and recreated on 'Plot clear' & 'File new' functions to free memory (of main & pool additional processes). FlatCAMObj has 'delete()' method to free resources. FlatCAMObj deletes properly now on 'del *.ojb' in TreeItems.
129 lines
4.2 KiB
Python
129 lines
4.2 KiB
Python
############################################################
|
|
# FlatCAM: 2D Post-processing for Manufacturing #
|
|
# http://caram.cl/software/flatcam #
|
|
# Author: Juan Pablo Caram (c) #
|
|
# Date: 2/5/2014 #
|
|
# MIT Licence #
|
|
############################################################
|
|
|
|
from PyQt4 import QtCore
|
|
|
|
import logging
|
|
from VisPyCanvas import VisPyCanvas
|
|
from VisPyVisuals import ShapeGroup, ShapeCollection, TextCollection, TextGroup, Cursor
|
|
from vispy.scene.visuals import InfiniteLine
|
|
import numpy as np
|
|
from vispy.geometry import Rect
|
|
|
|
log = logging.getLogger('base')
|
|
|
|
|
|
class PlotCanvas(QtCore.QObject):
|
|
"""
|
|
Class handling the plotting area in the application.
|
|
"""
|
|
|
|
def __init__(self, container, app):
|
|
"""
|
|
The constructor configures the Matplotlib figure that
|
|
will contain all plots, creates the base axes and connects
|
|
events to the plotting area.
|
|
|
|
:param container: The parent container in which to draw plots.
|
|
:rtype: PlotCanvas
|
|
"""
|
|
|
|
super(PlotCanvas, self).__init__()
|
|
|
|
self.app = app
|
|
|
|
# Parent container
|
|
self.container = container
|
|
|
|
# Attach to parent
|
|
self.vispy_canvas = VisPyCanvas()
|
|
self.vispy_canvas.create_native()
|
|
self.vispy_canvas.native.setParent(self.app.ui)
|
|
self.container.addWidget(self.vispy_canvas.native)
|
|
|
|
self.vline = InfiniteLine(pos=0, color=(0.0, 0.0, 0.0, 1.0), vertical=True,
|
|
parent=self.vispy_canvas.view.scene)
|
|
|
|
self.hline = InfiniteLine(pos=0, color=(0.0, 0.0, 0.0, 1.0), vertical=False,
|
|
parent=self.vispy_canvas.view.scene)
|
|
|
|
# self.shape_collections = []
|
|
|
|
self.shape_collection = self.new_shape_collection()
|
|
self.app.pool_recreated.connect(self.on_pool_recreated)
|
|
self.text_collection = self.new_text_collection()
|
|
|
|
# TODO: Should be setting to show/hide CNC job annotations (global or per object)
|
|
self.text_collection.enabled = False
|
|
|
|
def vis_connect(self, event_name, callback):
|
|
return getattr(self.vispy_canvas.events, event_name).connect(callback)
|
|
|
|
def vis_disconnect(self, event_name, callback):
|
|
getattr(self.vispy_canvas.events, event_name).disconnect(callback)
|
|
|
|
def zoom(self, factor, center=None):
|
|
"""
|
|
Zooms the plot by factor around a given
|
|
center point. Takes care of re-drawing.
|
|
|
|
:param factor: Number by which to scale the plot.
|
|
:type factor: float
|
|
:param center: Coordinates [x, y] of the point around which to scale the plot.
|
|
:type center: list
|
|
:return: None
|
|
"""
|
|
self.vispy_canvas.view.camera.zoom(factor, center)
|
|
|
|
def new_shape_group(self):
|
|
return ShapeGroup(self.shape_collection)
|
|
|
|
def new_shape_collection(self, **kwargs):
|
|
# sc = ShapeCollection(parent=self.vispy_canvas.view.scene, pool=self.app.pool, **kwargs)
|
|
# self.shape_collections.append(sc)
|
|
# return sc
|
|
return ShapeCollection(parent=self.vispy_canvas.view.scene, pool=self.app.pool, **kwargs)
|
|
|
|
def new_cursor(self):
|
|
c = Cursor(pos=np.empty((0, 2)), parent=self.vispy_canvas.view.scene)
|
|
c.antialias = 0
|
|
return c
|
|
|
|
def new_text_group(self):
|
|
return TextGroup(self.text_collection)
|
|
|
|
def new_text_collection(self, **kwargs):
|
|
return TextCollection(parent=self.vispy_canvas.view.scene, **kwargs)
|
|
|
|
def fit_view(self, rect=None):
|
|
|
|
# Lock updates in other threads
|
|
self.shape_collection.lock_updates()
|
|
|
|
if not rect:
|
|
rect = Rect(0, 0, 10, 10)
|
|
try:
|
|
rect.left, rect.right = self.shape_collection.bounds(axis=0)
|
|
rect.bottom, rect.top = self.shape_collection.bounds(axis=1)
|
|
except TypeError:
|
|
pass
|
|
|
|
self.vispy_canvas.view.camera.rect = rect
|
|
|
|
self.shape_collection.unlock_updates()
|
|
|
|
def clear(self):
|
|
pass
|
|
|
|
def redraw(self):
|
|
self.shape_collection.redraw([])
|
|
self.text_collection.redraw()
|
|
|
|
def on_pool_recreated(self, pool):
|
|
self.shape_collection.pool = pool
|