def set_scene(self, scene):
self.scene = scene
self.grScene = self.scene.grScene
self.layout.removeWidget(self.view)
self.addItemWindow.close()
self.menu_bar.close()
self.view = QDMGraphicsView(self.grScene, self)
self.layout.addWidget(self.view)
for i in self.scene.links:
i.update_positions()
self.addItemWindow = AddItemWindow(self)
self.addItemWindow.show()
self.menu_bar = MainMenu(self, self)
def initUI(self):
self.setGeometry(200, 200, 800, 800)
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
self.scene = Scene()
self.grScene = self.scene.grScene
self.view = QDMGraphicsView(self.grScene, self)
self.layout.addWidget(self.view)
node1 = Node(self.scene, "First Node", inputs=[1, 1, 1],
outputs=[1]) # 数字代表 socket 类型， socket index 按数组顺序
node2 = Node(self.scene, "Second Node", inputs=[1, 1, 1], outputs=[1])
node3 = Node(self.scene, "Third Node", inputs=[1, 1, 1], outputs=[1])
node1.setPos(-350, 100)
node2.setPos(-75, 0)
node3.setPos(200, -100)
edge1 = Edge(self.scene, node1.outputs[0], node2.inputs[0],
EDGE_TYPE_BEZIER)
edge2 = Edge(self.scene, node2.outputs[0], node3.inputs[1],
EDGE_TYPE_BEZIER)
self.setWindowTitle("Node Editor")
self.show()
def initUI(self):
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
self.scene = self.__class__.Scene_class()
self.view = QDMGraphicsView(self.scene.grScene, self)
self.layout.addWidget(self.view)
def initUI(self):
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
# crate graphics scene
self.scene = Scene()
# create graphics view
self.view = QDMGraphicsView(self.scene.grScene, self)
self.layout.addWidget(self.view)
def initUI(self):
self.setGeometry(200, 300, 800, 600)
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
# create graphics scene
self.scene = Scene()
self.grScene = self.scene.grScene
# create graphics view
self.view = QDMGraphicsView(self.grScene, self)
self.layout.addWidget(self.view)
self.setWindowTitle("Node Editor")
self.show()
self.addNodes()
self.addItemWindow = AddItemWindow(self)
self.addItemWindow.show()
self.menu_bar = MainMenu(self, self)
def initUI(self):
self.setGeometry(800, 600, 1280, 720)
# 创建画布
self.scene = Scene()
# self.grScene = self.scene.grScene
self.view = QDMGraphicsView(self.scene.grScene, self)
node = Node(self.scene, "My First Node")
box_layout = QVBoxLayout()
box_layout.setContentsMargins(5, 5, 5, 5)
box_layout.addWidget(self.view)
self.setLayout(box_layout)
self.setWindowTitle("Node Editor")
self.show()
def initUI(self):
self.setGeometry(560, 240, 800, 600)
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
# graphics scene
self.scene = Scene()
self.grScene=self.scene.grScene
# graphics view
self.view = QDMGraphicsView(self.grScene,self)
self.layout.addWidget(self.view)
self.setWindowTitle("Node Editor")
self.show()
self.addDebugContent()
def initUI(self):
self.setGeometry(200, 200, 800, 600) # set Geometry of the Window
self.layout = QVBoxLayout()
self.layout.setContentsMargins(
0, 0, 0,
0) # removes the frame (Rahmen) from the editor/ sets it to 0
self.setLayout(self.layout)
# crate graphics scene
self.scene = Scene()
# self.grScene = self.scene.grScene
# add Nodes to the scene
self.addNodes()
# create graphics view
self.view = QDMGraphicsView(self.scene.grScene, self)
self.layout.addWidget(self.view)
# set Title of Window to Node Editor
self.setWindowTitle("Node Editor")
self.show()
class NodeEditorWnd(QWidget):
IS_THIRD_PARTY = False
def __init__(self, parent=None):
super().__init__(parent)
self.thread_pool = QThreadPool()
self.stylesheet_filename = 'qss/nodestyle.qss'
self.loadStylesheet(self.stylesheet_filename)
self.initUI()
self._run_third_party()
def initUI(self):
self.setGeometry(200, 300, 800, 600)
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
# create graphics scene
self.scene = Scene()
self.grScene = self.scene.grScene
# create graphics view
self.view = QDMGraphicsView(self.grScene, self)
self.layout.addWidget(self.view)
self.setWindowTitle("Node Editor")
self.show()
self.addNodes()
self.addItemWindow = AddItemWindow(self)
self.addItemWindow.show()
self.menu_bar = MainMenu(self, self)
# self.addDebugContent()
def set_scene(self, scene):
self.scene = scene
self.grScene = self.scene.grScene
self.layout.removeWidget(self.view)
self.addItemWindow.close()
self.menu_bar.close()
self.view = QDMGraphicsView(self.grScene, self)
self.layout.addWidget(self.view)
for i in self.scene.links:
i.update_positions()
self.addItemWindow = AddItemWindow(self)
self.addItemWindow.show()
self.menu_bar = MainMenu(self, self)
def addDebugContent(self):
green_brush = QBrush(Qt.green)
line_pen = QPen(Qt.black)
line_pen.setWidth(2)
rectangle = self.grScene.addRect(-100, -100, 80, 100, line_pen,
green_brush)
rectangle.setFlag(QGraphicsItem.ItemIsMovable)
text = self.grScene.addText("H E L L O M E T E O R", QFont("Arial"))
text.setDefaultTextColor(QColor("#e3b0ac"))
text.setFlag(QGraphicsItem.ItemIsMovable)
text.setFlag(QGraphicsItem.ItemIsSelectable)
widget1 = QPushButton("Push Me")
proxy1 = self.grScene.addWidget(widget1)
proxy1.setPos(0, -50)
widget2 = QTextEdit("Edit me")
proxy2 = self.grScene.addWidget(widget2)
proxy2.setPos(0, 100)
line = self.grScene.addLine(-100, -100, 400, 400, line_pen)
line.setFlag(QGraphicsItem.ItemIsMovable)
def loadStylesheet(self, filename):
print("Loading STYLE: {0}".format(filename))
file = QFile(filename)
file.open(QFile.ReadOnly | QFile.Text)
stylesheet = file.readAll()
QApplication.instance().setStyleSheet(str(stylesheet,
encoding='utf-8'))
def addNodes(self):
pass
def focus_on_node(self, node: Node = None, index: int = None):
if index is not None and node is None and index in range(
0, len(self.scene.nodes)):
node = self.scene.nodes[index]
if node is not None:
self.view.centerOn(node.position)
else:
print(
"focus_on_node(): invalid parameters. Please add node or index."
)
def select_node(self, node: Node = None, index: int = None):
if index is not None and node is None and index in range(
0, len(self.scene.nodes)):
node = self.scene.nodes[index]
if node is not None:
for i in self.scene.nodes:
i.grNode.setSelected(False)
node.grNode.setSelected(True)
node.grNode.setFocus()
else:
print(
"select_node(): invalid parameters. Please add node or index.")
def _run_third_party(self):
if not NodeEditorWnd.IS_THIRD_PARTY:
return
worker = node_thirdparty_worker.Worker(self.on_third_party_exec)
worker.signals.result.connect(self.on_third_party_result)
worker.signals.finished.connect(self.on_third_party_finished)
self.thread_pool.start(worker)
def run_third_party_cmd(self, cmd: str):
if cmd.startswith("focus"):
index = int(cmd.split(" ")[1])
self.focus_on_node(index=index)
elif cmd.startswith("select"):
index = int(cmd.split(" ")[1])
self.select_node(index=index)
elif cmd.startswith("stop"):
NodeEditorWnd.IS_THIRD_PARTY = False
print("third party loop stopped.")
elif cmd.startswith("exit"):
NodeEditorWnd.IS_THIRD_PARTY = False
self.close()
elif cmd.startswith("load"):
file = cmd.replace("load ", "")
print(file)
scene = node_serialization.deserialize_scene(file)
self.set_scene(scene)
else:
print("unrecognised command {0}".format(cmd))
def on_third_party_exec(self, progress_callback=None):
time.sleep(1)
cmd = input("user command: ")
return cmd
def on_third_party_result(self, obj):
self.run_third_party_cmd(obj)
def on_third_party_finished(self):
self._run_third_party()