def node_item_factory():
from .qt_node_item import QtNodeItem
return QtNodeItem
def node_content_factory():
from .qt_node_content import QtNodeContent
return QtNodeContent
def node_socket_factory():
from .qt_node_socket import QtNodeSocket
return QtNodeSocket
def edge_item_factory():
from .qt_edge_item import QtEdgeItem
return QtEdgeItem
# Inject the factory
QT_FACTORIES.update({
'NodeItem': node_item_factory,
'NodeContent': node_content_factory,
'NodeSocket': node_socket_factory,
'EdgeItem': edge_item_factory,
'NodeGraphicsScene': node_graphics_scene_factory,
'GraphicsScene': graphics_scene_factory,
'GraphicsView': graphics_view_factory,
})
'Fuse': occ_fuse_factory,
'Fillet': occ_fillet_factory,
'Chamfer': occ_chamfer_factory,
'Offset': occ_offset_factory,
'ThickSolid': occ_thick_solid_factory,
'Pipe': occ_pipe_factory,
'ThruSections': occ_thru_sections_factory,
'Transform': occ_transform_factory,
#: Draw
'Point': occ_point_factory,
'Vertex': occ_vertex_factory,
'Line': occ_line_factory,
'Segment': occ_segment_factory,
'Arc': occ_arc_factory,
'Circle': occ_circle_factory,
'Ellipse': occ_ellipse_factory,
'Hyperbola': occ_hyperbola_factory,
'Parabola': occ_parabola_factory,
'Polygon': occ_polygon_factory,
'Bezier': occ_bezier_factory,
'BSpline': occ_bspline_factory,
'Wire': occ_wire_factory,
'Text': occ_text_factory,
'Svg': occ_svg_factory,
}
#: Install it
QT_FACTORIES.update(OCC_FACTOIRES)
"""
Copyright (c) 2017-2018, Jairus Martin.
Distributed under the terms of the GPL v3 License.
The full license is in the file LICENSE, distributed with this software.
Created on Dec 13, 2017
@author: jrm
"""
from enaml.qt.qt_factories import QT_FACTORIES
def occ_viewer_factory():
from .qt_occ_viewer import QtOccViewer
return QtOccViewer
def occ_viewer_clipped_plane_factory():
from .qt_occ_viewer import QtOccViewerClippedPlane
return QtOccViewerClippedPlane
QT_FACTORIES.update({
'OccViewer': occ_viewer_factory,
'OccViewerClippedPlane': occ_viewer_clipped_plane_factory,
})
pass
@d_func
def mouse_release_event(self, event):
pass
@d_func
def mouse_move_event(self, event):
pass
def image_view_ex_factory():
return QtImageViewEx
QT_FACTORIES.update({'ImageViewEx': image_view_ex_factory})
def _get_pixmap(image):
if image is None:
return None
# w, h, format = _get_image_size_n_format(image)
# qimage = QImage(image.data, w, h, image.strides[0], format)
qimage = ndarray_to_qimage(image)
return QPixmap.fromImage(qimage)
def _get_image_size_n_format(image):
if image.dtype != np.uint8:
raise NotImplementedError(f"Unsupported image dtype {image.dtype}")
h, w = image.shape[0:2]
"""
Created on Dec 13, 2017
@author: jrm
"""
from enaml.qt.qt_factories import QT_FACTORIES
def occ_viewer_factory():
from .qt_occ_viewer import QtOccViewer
return QtOccViewer
QT_FACTORIES.update({
'OccViewer': occ_viewer_factory,
})
def tree_widget_item_factory():
from .qt_tree_widget import QtTreeWidgetItem
return QtTreeWidgetItem
def tree_widget_col_factory():
from .qt_tree_widget import QtTreeWidgetColumn
return QtTreeWidgetColumn
# Inject the factory
QT_FACTORIES.update({
'KeyEvent': key_event_factory,
'TableView': table_view_factory,
'TableViewItem': table_view_item_factory,
'TableViewRow': table_view_row_factory,
'TableViewColumn': table_view_col_factory,
'TableWidget': table_widget_factory,
'TableWidgetItem': table_widget_item_factory,
'TableWidgetRow': table_widget_row_factory,
'TableWidgetColumn': table_widget_col_factory,
'TreeView': tree_view_factory,
'TreeViewItem': tree_view_item_factory,
'TreeViewColumn': tree_view_col_factory,
'TreeWidget': tree_widget_factory,
'TreeWidgetItem': tree_widget_item_factory,
'TreeWidgetColumn': tree_widget_col_factory,
})
# Inject the factory
QT_FACTORIES.update({
'Console': console_factory,
'DoubleSpinBox': double_spin_box_factory,
'KeyEvent': key_event_factory,
'OccViewer': occ_viewer_factory,
'PlotArea': plot_area_factory,
'PlotItem2D': plot_item_2d_factory,
'PlotItem3D': plot_item_3d_factory,
'PlotItemArray': plot_item_array_factory,
'PlotItemArray3D': plot_item_array_3d_factory,
'PlotItemList': plot_item_list_factory,
'PlotItemDict': plot_item_dict_factory,
'TableView': table_view_factory,
'TableViewItem': table_view_item_factory,
'TableViewRow': table_view_row_factory,
'TableViewColumn': table_view_col_factory,
'TableWidget': table_widget_factory,
'TableWidgetItem': table_widget_item_factory,
'TableWidgetRow': table_widget_row_factory,
'TableWidgetColumn': table_widget_col_factory,
'TreeView': tree_view_factory,
'TreeViewItem': tree_view_item_factory,
'TreeViewColumn': tree_view_col_factory,
'TreeWidget': tree_widget_factory,
'TreeWidgetItem': tree_widget_item_factory,
'TreeWidgetColumn': tree_widget_col_factory,
})