def __init__(self, lon0, lat0, lon1, lat1, pixmap, parent=None):
"""Constructor.
Args:
longitude(float): Longitude of the upper left corner
latitude(float): Latitude of the upper left corner
longitude(float): Longitude of the lower right corner
latitude(float): Latitude of the lower right corner
pixmap(QPixmap): Pixmap.
scene(MapGraphicsScene): Scene the item belongs to.
parent(QGraphicsItem): Parent item.
Show a pixamp with geo-registered corners
"""
QGraphicsPixmapItem.__init__(self, parent=parent)
MapItem.__init__(self)
self._lon0 = lon0
self._lat0 = lat0
self._lon1 = lon1
self._lat1 = lat1
self._xsize = 0
self._ysize = 0
self.orig_pixmap = pixmap
self.setPixmap(pixmap)
self.setShapeMode(1)
self.x_mult = 1
self.y_mult = 1
class ImageButton(QGraphicsObject):
'''
Custom Image Holder Class that Allows for event handling -
- QGraphicsPixmapItem is not a QObject, so cannot catch events.
'''
QtParentClass = QGraphicsObject
clicked = Signal(int)
def __init__(self, img, parent=None):
'''
Args:
img: A QPixmap instance
Keyword Args:
parent: A pyqt window instance
'''
QGraphicsObject.__init__(self, parent=parent)
self.image = QGraphicsPixmapItem(img, parent=self)
self.parent = parent
def paint(self, painter, option, widget):
self.image.paint(painter, option, widget)
def boundingRect(self):
return self.image.boundingRect()
def mouseReleaseEvent(self, event):
self.clicked.emit(1)
def mousePressEvent(self, event):
pass
class GraphicsView(QGraphicsView):
def __init__(self, **kwargs):
super(GraphicsView, self).__init__(**kwargs)
self._scene = QGraphicsScene(parent=self)
self.current_image = QGraphicsPixmapItem()
self._scene.addItem(self.current_image)
self.setScene(self._scene)
self.fitInView(self._scene.sceneRect(), Qt.KeepAspectRatio)
self.setTransformationAnchor(QGraphicsView.AnchorUnderMouse)
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.setMouseTracking(True)
self.setRenderHint(QPainter.Antialiasing)
def update_image(self, img: np.ndarray):
rgb_image = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
h, w, ch = rgb_image.shape
bytes_per_line = ch * w
qimg = QImage(rgb_image.data, w, h, bytes_per_line,
QImage.Format_RGB888)
pixmap = QPixmap.fromImage(qimg)
self.current_image.setPixmap(pixmap)
self._scene.setSceneRect(0, 0, w, h)
self.fitInView(self._scene.sceneRect(), Qt.KeepAspectRatio)
def resizeEvent(self, e):
self.fitInView(self._scene.sceneRect(), Qt.KeepAspectRatio)
def wheelEvent(self, e):
if e.angleDelta().y() > 0:
self.scale(1.25, 1.25)
elif e.angleDelta().y() < 0:
self.scale(0.85, 0.85)
def mousePressEvent(self, e):
# シーン上のマウス位置を取得する
pos = self.mapToScene(e.pos())
if e.button() == Qt.LeftButton:
if e.modifiers() == Qt.NoModifier:
pass
elif e.modifiers() == Qt.AltModifier:
self.setDragMode(QGraphicsView.ScrollHandDrag)
QGraphicsView.mousePressEvent(self, e)
def mouseReleaseEvent(self, e):
QGraphicsView.mouseReleaseEvent(self, e)
if e.button() == Qt.LeftButton:
self.setDragMode(QGraphicsView.NoDrag)
def calc_offset(self, p: QPoint):
offset_x = p.x() - int(self.viewport().width() / 2)
offset_y = p.y() - int(self.viewport().height() / 2)
return QPoint(offset_x, offset_y)
def __init__(self, img, parent=None):
'''
Args:
img: A QPixmap instance
Keyword Args:
parent: A pyqt window instance
'''
QGraphicsObject.__init__(self, parent=parent)
self.image = QGraphicsPixmapItem(img, parent=self)
self.parent = parent
def __init__(self, **kwargs):
super(GraphicsView, self).__init__(**kwargs)
self._scene = QGraphicsScene(parent=self)
self.current_image = QGraphicsPixmapItem()
self._scene.addItem(self.current_image)
self.setScene(self._scene)
self.fitInView(self._scene.sceneRect(), Qt.KeepAspectRatio)
self.setTransformationAnchor(QGraphicsView.AnchorUnderMouse)
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.setMouseTracking(True)
self.setRenderHint(QPainter.Antialiasing)
def __init__(self,
canvas: 'PMGraphicsScene',
node_id,
text: str = '',
input_ports: List[CustomPort] = None,
output_ports: List[CustomPort] = None,
icon_path=r'',
look: Dict[str, str] = None):
super(Node, self).__init__()
assert isinstance(look, dict)
self.look = look
direction = self.look.get('direction')
self.direction = 'E' if direction is None else direction # 输出端口的方向
assert_in(self.direction, ['E', 'W'])
self.look['direction'] = self.direction
self.id = node_id
self.text = text if text != '' else node_id
self.base_rect = CustomRect(self)
icon_path = os.path.join(get_parent_path(__file__), 'icons',
'logo.png') if icon_path == '' else icon_path
self.icon_path = icon_path
pix_map = QPixmap(icon_path)
self.pix_map_item = QGraphicsPixmapItem(pix_map, parent=self.base_rect)
self.pix_map_item.setPos(20, 20)
self.text_item = QGraphicsTextItem(parent=self.base_rect)
start_left = 50
self.text_item.setPos(start_left, 10)
self.text_item.setPlainText(self.text)
self.internal_value_text = QGraphicsTextItem(parent=self.base_rect)
self.internal_value_text.setPos(start_left, 30)
self.internal_value_text.setPlainText('')
self.status_text = QGraphicsTextItem(parent=self.base_rect)
self.status_text.setPos(start_left, 100)
self.status_text.setPlainText('wait')
self.input_ports = input_ports
self.output_ports = output_ports
self.input_ports_dic = {p.id: p for p in input_ports}
self.output_ports_dic = {p.id: p for p in output_ports}
self.canvas = canvas
# self.set_content(content)
self.setup()
def __init__(self, longitude, latitude, pixmap, parent=None):
"""Constructor.
Args:
longitude(float): Longitude of the origin of the pixmap.
latitude(float): Latitude of the center of the pixmap.
pixmap(QPixmap): Pixmap.
scene(MapGraphicsScene): Scene the item belongs to.
parent(QGraphicsItem): Parent item.
"""
QGraphicsPixmapItem.__init__(self, parent=parent)
MapItem.__init__(self)
self._lon = longitude
self._lat = latitude
self.setPixmap(pixmap)
def __init__(self, longitude, latitude, pixmap, parent=None):
"""Constructor.
Args:
longitude(float): Longitude of the center of the pixmap
latitude(float): Latitude of the center of the pixmap.
pixmap(QPixmap): Pixmap.
scene(MapGraphicsScene): Scene the item belongs to.
parent(QGraphicsItem): Parent item.
"""
QGraphicsPixmapItem.__init__(self, parent=parent)
MapItem.__init__(self)
self._lon = longitude
self._lat = latitude
self.setPixmap(pixmap)
def __init__(self,
lon0,
lat0,
lon1,
lat1,
lon2,
lat2,
lon3,
lat3,
pixmap,
parent=None):
"""Constructor.
Args:
lon0(float): Longitude (decimal degrees) of the upper left corner of the image
lat0(float): Latitude of the upper left corner of the image
lon1(float): longitude of the next point clockwise
lat0(float): latitude of the next point clockwise
lon2(float): longitude of the next point clockwise
lat2(float): longitude of the next point clockwise
lon3(float): latitude of the next point clockwise
lat3(float): latitude of the next point clockwise
pixmap(QPixmap): Pixmap.
scene(MapGraphicsScene): Scene the item belongs to.
parent(QGraphicsItem): Parent item.
Show a pixamp with geo-registered corners
"""
QGraphicsPixmapItem.__init__(self, parent=parent)
MapItem.__init__(self)
self._lon0 = lon0
self._lat0 = lat0
self._lon1 = lon1
self._lat1 = lat1
self._lon2 = lon2
self._lat2 = lat2
self._lon3 = lon3
self._lat3 = lat3
self._xsize = 0
self._ysize = 0
self.setPixmap(pixmap)
self.setShapeMode(1)
self.x_mult = 1
self.y_mult = 1
def showImage(self, posterPath: str):
self.removeImage()
imgData = requests.get(self.__poster_url__ + posterPath)
pix = QPixmap()
pix.loadFromData(imgData.content)
item = QGraphicsPixmapItem(pix)
self.__scene__ = QGraphicsScene(self)
self.__scene__.addItem(item)
self.ui.graphicsView.setScene(self.__scene__)
self.resizeImage()
def helpWindow(self):
self.pixmap = QtGui.QPixmap()
#self.pixmap.setAlignment(QtCore.Qt.AlignCenter)
self.pixmap.load(self.helpFile)
self.pixmap.setDevicePixelRatio(
self.pixel_ratio
) # https://stackoverflow.com/questions/50127246/pyqt-5-10-enabling-high-dpi-support-for-macos-poor-pixmap-quality
#self.pixmap4 = self.pixmap.scaled(self.width * self.pixel_ratio, (self.height * self.pixel_ratio)-45, Qt.KeepAspectRatio)
self.pixmap4 = self.pixmap.scaled(self.width * self.pixel_ratio,
(self.height * self.pixel_ratio),
Qt.KeepAspectRatio)
try:
self.scene.removeItem(self.item)
except:
print("failed to remove item")
self.item = QGraphicsPixmapItem(self.pixmap4)
self.scene.addItem(self.item)
#myapp.setWindowTitle(os.path.basename("Instructions"))
self.setWindowTitle(os.path.basename("Instructions"))
self.ui.graphicsView.setScene(self.scene)
class Node(QObject):
"""
Node是一个节点的抽象
"""
def __init__(self,
canvas: 'PMGraphicsScene',
node_id,
text: str = '',
input_ports: List[CustomPort] = None,
output_ports: List[CustomPort] = None,
icon_path=r'',
look: Dict[str, str] = None):
super(Node, self).__init__()
assert isinstance(look, dict)
self.look = look
direction = self.look.get('direction')
self.direction = 'E' if direction is None else direction # 输出端口的方向
assert_in(self.direction, ['E', 'W'])
self.look['direction'] = self.direction
self.id = node_id
self.text = text if text != '' else node_id
self.base_rect = CustomRect(self)
icon_path = os.path.join(get_parent_path(__file__), 'icons',
'logo.png') if icon_path == '' else icon_path
self.icon_path = icon_path
pix_map = QPixmap(icon_path)
self.pix_map_item = QGraphicsPixmapItem(pix_map, parent=self.base_rect)
self.pix_map_item.setPos(20, 20)
self.text_item = QGraphicsTextItem(parent=self.base_rect)
start_left = 50
self.text_item.setPos(start_left, 10)
self.text_item.setPlainText(self.text)
self.internal_value_text = QGraphicsTextItem(parent=self.base_rect)
self.internal_value_text.setPos(start_left, 30)
self.internal_value_text.setPlainText('')
self.status_text = QGraphicsTextItem(parent=self.base_rect)
self.status_text.setPos(start_left, 100)
self.status_text.setPlainText('wait')
self.input_ports = input_ports
self.output_ports = output_ports
self.input_ports_dic = {p.id: p for p in input_ports}
self.output_ports_dic = {p.id: p for p in output_ports}
self.canvas = canvas
# self.set_content(content)
self.setup()
def display_internal_values(self, val_str: str = ''):
self.internal_value_text.setPlainText(val_str)
def set_content(self, content: 'FlowContentEditableFunction' = None):
self.content: 'FlowContentEditableFunction' = content if content is not None else FlowContentEditableFunction(
self, '')
self.content.signal_exec_finished.connect(self.on_exec_finished)
self.content.signal_exec_started.connect(self.on_exec_started)
self.content.signal_error_occurs.connect(self.on_error_occurs)
self.display_internal_values(self.content.format_param())
def on_error_occurs(self, error: FlowContentError):
flow_widget: PMFlowWidget = self.canvas.flow_widget
flow_widget.on_error_occurs(error)
def on_exec_started(self, content):
"""
执行前进行的操作
:param content:
:return:
"""
self.status_text.setPlainText(content)
def on_exec_finished(self, content: str):
"""
执行完成后进行的操作
:param content:
:return:
"""
self.status_text.setPlainText(content)
#
def reset(self):
self.status_text.setPlainText('wait')
def get_port_index(self, port: CustomPort):
"""
获取端口的索引
:param port:
:return:
"""
return self.input_ports.index(
port) if port.port_type == 'input' else self.output_ports.index(
port)
def set_icon(self, icon_path: str):
pass
def set_pos(self, x: int, y: int):
"""
设置位置，左上角角点
:param x:
:param y:
:return:
"""
self.base_rect.setPos(x, y)
self.refresh_pos()
def get_pos(self) -> Tuple[int, int]:
"""
获取位置，左上角角点
:return:
"""
pos = self.base_rect.pos()
return pos.x(), pos.y()
def refresh_pos(self):
"""
刷新位置。当节点被拖动的时候，此方法会被触发。
"""
y = self.base_rect.y()
dy_input = self.base_rect.boundingRect().height() / (
1 + len(self.input_ports))
dy_output = self.base_rect.boundingRect().height() / (
1 + len(self.output_ports))
if self.direction == 'E':
x_input = self.base_rect.x() + 5
x_output = self.base_rect.x() + self.base_rect.boundingRect(
).width() - 15
elif self.direction == 'W':
x_input = self.base_rect.x() + self.base_rect.boundingRect().width(
) - 15
x_output = self.base_rect.x() + 5
else:
raise NotImplementedError
for i, p in enumerate(self.input_ports):
p.setPos(QPointF(x_input, y + int(dy_input * (1 + i))))
for i, p in enumerate(self.output_ports):
p.setPos(QPointF(x_output, y + int(dy_output * (1 + i))))
self.canvas.signal_item_dragged.emit('')
def setup(self):
self.base_rect.setPos(80, 80)
self.canvas.signal_clear_selection.connect(
self.base_rect.on_clear_selection)
self.canvas.addItem(self.base_rect)
for p in self.input_ports + self.output_ports:
self.canvas.addItem(p)
p.port_clicked.connect(self.on_port_clicked)
p.node = self
self.refresh_pos()
def on_port_clicked(self, port: 'CustomPort'):
if self.canvas.drawing_lines:
if self.canvas.line_start_port is not port:
if not port.port_type == self.canvas.line_start_port.port_type:
if self.canvas.allow_multiple_input or (
(not self.canvas.allow_multiple_input)
and len(port.connected_lines) == 0):
self.canvas.connect_port(port)
else:
if port.port_type == 'output':
self.canvas.drawing_lines = True
# if self.canvas.item
self.canvas.line_start_point = port.center_pos
self.canvas.line_start_port = port
def on_delete(self):
for port in self.input_ports + self.output_ports:
port.canvas = self.canvas
port.on_delete()
self.canvas.removeItem(self.base_rect)
self.canvas.nodes.remove(self)
self.deleteLater()
def __repr__(self):
s = super(Node, self).__repr__()
return s + repr(self.input_ports) + repr(self.output_ports)
def get_port(self, port_id: str) -> 'CustomPort':
for port in self.input_ports + self.output_ports:
if port_id == port.id:
return port
return None
def add_port(self, port: CustomPort):
"""
添加一个端口
:param port:
:return:
"""
port_type = port.port_type
if port_type == 'input':
self.input_ports.append(port)
self.input_ports_dic[port.id] = port
elif port_type == 'output':
self.output_ports.append(port)
self.output_ports_dic[port.id] = port
else:
raise ValueError('port type invalid')
if port.node is None:
port.node = self
self.refresh_pos()
self.canvas.addItem(port)
port.port_clicked.connect(self.on_port_clicked)
return port
def remove_port(self, port: CustomPort):
"""
删除一个端口
:param port:
:return:
"""
port_type = port.port_type
if port_type == 'input':
self.input_ports.remove(port)
self.input_ports_dic.pop(port.id)
elif port_type == 'output':
self.output_ports.remove(port)
self.output_ports_dic.pop(port.id)
else:
raise ValueError('port type invalid')
port.on_delete()
self.refresh_pos()
def change_property(self, property: Dict[str, Union[int, str]]):
"""
改变各个端口的文字、端口的数目以及文字。
:param property:
:return:
"""
self.text = property['text']
self.text_item.setPlainText(self.text)
self.content.update_settings(property)
self.valid_port_ids = []
if property.get('inputs') is not None:
for input_id, input_text in zip(property['inputs'][0],
property['inputs'][1]):
self.valid_port_ids.append(input_id)
p = self.get_port(input_id)
if p is None:
p = self.add_port(
CustomPort(port_id=input_id,
text=input_text,
port_type='input'))
p.set_text(input_text)
for p in self.input_ports:
if p.id not in self.valid_port_ids:
self.remove_port(p)
if property.get('outputs') is not None:
for output_id, output_text in zip(property['outputs'][0],
property['outputs'][1]):
self.valid_port_ids.append(output_id)
p = self.get_port(output_id)
if p is None:
p = self.add_port(
CustomPort(port_id=output_id,
text=output_text,
port_type='output'))
p.set_text(output_text)
for p in self.output_ports:
if p.id not in self.valid_port_ids:
self.remove_port(p)
def change_ports_property(self, property: Dict[str, Union[int, str]]):
"""
改变各个端口的文字、端口的数目以及文字。
:param property:
:return:
"""
self.text = property['text']
self.text_item.setPlainText(self.text)
self.valid_port_ids = []
if property.get('inputs') is not None:
for input_id, input_text in zip(property['inputs'][0],
property['inputs'][1]):
self.valid_port_ids.append(input_id)
p = self.get_port(input_id)
if p is None:
p = self.add_port(
CustomPort(port_id=input_id,
text=input_text,
port_type='input'))
p.set_text(input_text)
for p in self.input_ports:
if p.id not in self.valid_port_ids:
self.remove_port(p)
if property.get('outputs') is not None:
for output_id, output_text in zip(property['outputs'][0],
property['outputs'][1]):
self.valid_port_ids.append(output_id)
p = self.get_port(output_id)
if p is None:
p = self.add_port(
CustomPort(port_id=output_id,
text=output_text,
port_type='output'))
p.set_text(output_text)
for p in self.output_ports:
if p.id not in self.valid_port_ids:
self.remove_port(p)
def on_edit_ports(self):
from pmgwidgets import PMGPanel
from qtpy.QtWidgets import QDialog, QVBoxLayout
dlg = QDialog(self.base_rect.scene().flow_widget)
sp = PMGPanel()
p: 'CustomPort' = None
input_ids, output_ids = [], []
input_texts, output_texts = [], []
self._last_var = 0
def new_id_input():
node_id = self.id
max_val = 0
for p in self.input_ports:
n, t, p = p.parse_id()
if max_val < int(p):
max_val = int(p)
self._last_var += 1
return '%s:input:%d' % (node_id, max_val + self._last_var)
def new_id_output():
node_id = self.id
max_val = 0
for p in self.output_ports:
n, t, p = p.parse_id()
if max_val < int(p):
max_val = int(p)
self._last_var += 1
return '%s:output:%d' % (node_id, max_val + self._last_var)
for p in self.input_ports:
input_ids.append(p.id)
input_texts.append(p.text)
for p in self.output_ports:
output_ids.append(p.id)
output_texts.append(p.text)
views = []
# views += self.content.get_settings_params()
views += [
('line_ctrl', 'text', 'Node Text', self.text),
]
if self.content.ports_changable[0]:
views.append(('list_ctrl', 'inputs', 'Set Inputs',
[input_ids, input_texts], new_id_input))
if self.content.ports_changable[1]:
views.append(('list_ctrl', 'outputs', 'Set Outputs',
[output_ids, output_texts], new_id_output))
sp.set_items(views)
dlg.setLayout(QVBoxLayout())
dlg.layout().addWidget(sp)
dlg.exec_()
dic = sp.get_value()
self.change_ports_property(dic)
def on_edit_properties_requested(self):
"""
Show Settings Panel.
If It was customized node,it will call the function of the content.
Or the default settings parameters will be got.
Returns:
"""
from pmgwidgets import PMGPanel
from qtpy.QtWidgets import QDialog, QVBoxLayout
if isinstance(self.content, FlowContentForFunction):
dlg = QDialog(self.base_rect.scene().flow_widget)
sp = PMGPanel()
p: 'CustomPort' = None
input_ids, output_ids = [], []
input_texts, output_texts = [], []
self._last_var = 0
def new_id_input():
node_id = self.id
max_val = 0
for p in self.input_ports:
n, t, p = p.parse_id()
if max_val < int(p):
max_val = int(p)
self._last_var += 1
return '%s:input:%d' % (node_id, max_val + self._last_var)
def new_id_output():
node_id = self.id
max_val = 0
for p in self.output_ports:
n, t, p = p.parse_id()
if max_val < int(p):
max_val = int(p)
self._last_var += 1
return '%s:output:%d' % (node_id, max_val + self._last_var)
for p in self.input_ports:
input_ids.append(p.id)
input_texts.append(p.text)
for p in self.output_ports:
output_ids.append(p.id)
output_texts.append(p.text)
views = []
views += self.content.get_settings_params()
views += [
('line_ctrl', 'text', 'Node Text', self.text),
]
if self.content.ports_changable[0]:
views.append(('list_ctrl', 'inputs', 'Set Inputs',
[input_ids, input_texts], new_id_input))
if self.content.ports_changable[1]:
views.append(('list_ctrl', 'outputs', 'Set Outputs',
[output_ids, output_texts], new_id_output))
sp.set_items(views)
dlg.setLayout(QVBoxLayout())
dlg.layout().addWidget(sp)
dlg.exec_()
dic = sp.get_value()
self.change_property(dic)
else:
try:
self.content.settings_window_requested(self.canvas.flow_widget)
except Exception as e:
import traceback
exc = traceback.format_exc()
print(exc)
QMessageBox.warning(self.canvas.flow_widget, 'Error', str(e))
def invert(self):
self.direction = 'W' if self.direction == 'E' else 'E'
self.refresh_pos()
self.look['direction'] = self.direction